Initial commit

app/tool/protoc-gen-bm/codegenerator/BUILD

@@ -0,0 +1,32 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["generator.go"],
+    importpath = "go-common/app/tool/protoc-gen-bm/codegenerator",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "@com_github_golang_protobuf//proto:go_default_library",
+        "@com_github_golang_protobuf//protoc-gen-go/plugin:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/tool/protoc-gen-bm/codegenerator/generator.go

@@ -0,0 +1,39 @@
+package codegenerator
+
+import (
+	"fmt"
+	"io"
+	"io/ioutil"
+
+	"github.com/golang/protobuf/proto"
+	plugin "github.com/golang/protobuf/protoc-gen-go/plugin"
+)
+
+// ParseRequest parses a code generator request from a proto Message.
+func ParseRequest(r io.Reader) (*plugin.CodeGeneratorRequest, error) {
+	input, err := ioutil.ReadAll(r)
+	if err != nil {
+		return nil, fmt.Errorf("failed to read code generator request: %v", err)
+	}
+	req := new(plugin.CodeGeneratorRequest)
+	if err = proto.Unmarshal(input, req); err != nil {
+		return nil, fmt.Errorf("failed to unmarshal code generator request: %v", err)
+	}
+	return req, nil
+}
+
+// WriteResponse write a code generator response
+func WriteResponse(w io.Writer, files []*plugin.CodeGeneratorResponse_File, inErr error) error {
+	var perrMsg *string
+	if inErr != nil {
+		errMsg := inErr.Error()
+		perrMsg = &errMsg
+	}
+	resp := &plugin.CodeGeneratorResponse{Error: perrMsg, File: files}
+	buf, err := proto.Marshal(resp)
+	if err != nil {
+		return err
+	}
+	_, err = w.Write(buf)
+	return err
+}