main.go

package main // import "moul.io/protoc-gen-gotemplate"

import (
	"io/ioutil"
	"os"

	"github.com/golang/protobuf/proto"                   // nolint:staticcheck
	"github.com/golang/protobuf/protoc-gen-go/generator" // nolint:staticcheck

	pgghelpers "moul.io/protoc-gen-gotemplate/helpers"
)

func main() {
	g := generator.New()

	data, err := ioutil.ReadAll(os.Stdin)
	if err != nil {
		g.Error(err, "reading input")
	}

	if err = proto.Unmarshal(data, g.Request); err != nil {
		g.Error(err, "parsing input proto")
	}

	if len(g.Request.FileToGenerate) == 0 {
		g.Fail("no files to generate")
	}

	g.CommandLineParameters(g.Request.GetParameter())

	pgghelpers.ParseParams(g)

	// Generate the protobufs
	g.GenerateAllFiles()

	data, err = proto.Marshal(g.Response)
	if err != nil {
		g.Error(err, "failed to marshal output proto")
	}

	_, err = os.Stdout.Write(data)
	if err != nil {
		g.Error(err, "failed to write output proto")
	}
}