do marshal for Bases and Base messages

1 files changed, 8 insertions, 0 deletions

diff --git a/marshal.go b/marshal.go
index 9709d5a..9cb1a4e 100644
--- a/marshal.go
+++ b/marshal.go
@@ -25,6 +25,14 @@ func (pb *Files) marshal(f *File) {
 	fmt.Fprintln(w, ")")
 	fmt.Fprintln(w, "")
 
+	if f.Bases.DoMarshal {
+		marshalThing(w, f.Bases.Name)
+	}
+
+	if f.Base.DoMarshal {
+		marshalThing(w, f.Base.Name)
+	}
+
 	for _, msg := range f.MsgNames {
 		if msg.DoMarshal {
 			marshalThing(w, msg.Name)