standard header

Signed-off-by: Jeff Carr <jcarr@wit.com>

main.go

@@ -85,8 +85,7 @@ func main() {
 	marshal(sortmap)
 }
 
-func header(w io.Writer, names map[string]string) {
-	fmt.Fprintln(w, "package "+names["package"])
+func headerComment(w io.Writer) {
 	fmt.Fprintln(w, "")
 	fmt.Fprintln(w, "// this file was autogenerated with autogenpb")
 	fmt.Fprintln(w, "//")
@@ -94,6 +93,11 @@ func header(w io.Writer, names map[string]string) {
 	fmt.Fprintln(w, "//")
 	fmt.Fprintln(w, "// go install go.wit.com/apps/autogenpb@latest")
 	fmt.Fprintln(w, "")
+}
+
+func header(w io.Writer, names map[string]string) {
+	fmt.Fprintln(w, "package "+names["package"])
+	headerComment(w)
 	fmt.Fprintln(w, "import (")
 	fmt.Fprintln(w, "	\"fmt\"")
 	fmt.Fprintln(w, "	\"os\"")

marshal.go

@@ -20,10 +20,7 @@ func marshal(names map[string]string) {
 	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, "")
+	headerComment(w)
 	fmt.Fprintln(w, "import (")
 	fmt.Fprintln(w, "	\"google.golang.org/protobuf/encoding/protojson\"")
 	fmt.Fprintln(w, "	\"google.golang.org/protobuf/encoding/prototext\"")

testautogen/Makefile

@@ -3,7 +3,7 @@ BUILDTIME = $(shell date +%Y.%m.%d)
 
 test: vet
 
-all: clean test.pb.go run goimports vet
+all: clean test.pb.go run vet
 
 run:
 	../autogenpb --proto test.proto --lobase gitTag --upbase GitTag --sort "ByPath,Refname"