1 files changed, 71 insertions, 0 deletions
diff --git a/marshal.go b/marshal.go
new file mode 100644
index 0000000..462468c
--- /dev/null
+++ b/marshal.go
@@ -0,0 +1,71 @@
+package main
+
+import (
+	"fmt"
+	"io"
+	"os"
+
+	"go.wit.com/log"
+)
+
+func marshal(names map[string]string) {
+
+	if argv.DryRun {
+		for k, v := range names {
+			log.Info(k, "=", v)
+		}
+		os.Exit(0)
+	}
+
+	w, _ := os.OpenFile(names["protobase"]+".marshal.pb.go", os.O_WRONLY|os.O_CREATE, 0600)
+
+	fmt.Fprintln(w, "package "+names["package"])
+	fmt.Fprintln(w, "")
+	fmt.Fprintln(w, "// todo: autogen this")
+	fmt.Fprintln(w, "// functions to import and export the protobuf")
+	fmt.Fprintln(w, "")
+	fmt.Fprintln(w, "import (")
+	fmt.Fprintln(w, "	\"google.golang.org/protobuf/encoding/protojson\"")
+	fmt.Fprintln(w, "	\"google.golang.org/protobuf/encoding/prototext\"")
+	fmt.Fprintln(w, "	\"google.golang.org/protobuf/proto\"")
+	fmt.Fprintln(w, ")")
+	fmt.Fprintln(w, "")
+
+	marshalThing(w, names["Base"])
+	marshalThing(w, names["Bases"])
+}
+
+func marshalThing(w io.Writer, thing string) {
+	fmt.Fprintln(w, "// human readable JSON")
+	fmt.Fprintln(w, "func (r *"+thing+") FormatJSON() string {")
+	fmt.Fprintln(w, "	return protojson.Format(r)")
+	fmt.Fprintln(w, "}")
+	fmt.Fprintln(w, "")
+	fmt.Fprintln(w, "// apparently this isn't supposed to be used?")
+	fmt.Fprintln(w, "// https://protobuf.dev/reference/go/faq/#unstable-text")
+	fmt.Fprintln(w, "// this is a shame because this is much nicer output than JSON Format()")
+	fmt.Fprintln(w, "func (r *"+thing+") FormatTEXT() string {")
+	fmt.Fprintln(w, "	return prototext.Format(r)")
+	fmt.Fprintln(w, "}")
+	fmt.Fprintln(w, "")
+	fmt.Fprintln(w, "// marshal json")
+	fmt.Fprintln(w, "func (r *"+thing+") MarshalJSON() ([]byte, error) {")
+	fmt.Fprintln(w, "	return protojson.Marshal(r)")
+	fmt.Fprintln(w, "}")
+	fmt.Fprintln(w, "")
+	fmt.Fprintln(w, "// unmarshal")
+	fmt.Fprintln(w, "func (r *"+thing+") UnmarshalJSON(data []byte) error {")
+	fmt.Fprintln(w, "	return protojson.Unmarshal(data, r)")
+	fmt.Fprintln(w, "}")
+	fmt.Fprintln(w, "")
+	fmt.Fprintln(w, "// marshal to wire")
+	fmt.Fprintln(w, "func (r *"+thing+") Marshal() ([]byte, error) {")
+	fmt.Fprintln(w, "	return proto.Marshal(r)")
+	fmt.Fprintln(w, "}")
+	fmt.Fprintln(w, "")
+	fmt.Fprintln(w, "// unmarshal from wire")
+	fmt.Fprintln(w, "func (r *"+thing+") Unmarshal(data []byte) error {")
+	fmt.Fprintln(w, "	return proto.Unmarshal(data, r)")
+	fmt.Fprintln(w, "}")
+	fmt.Fprintln(w, "")
+}