Level sets dependency graph to consume etcd 3.1.5

vendor/github.com/gogo/protobuf/vanity/command/command.go

@@ -1,7 +1,7 @@
-// Extensions for Protocol Buffers to create more go like structures.
+// Protocol Buffers for Go with Gadgets
 //
-// Copyright (c) 2015, Vastech SA (PTY) LTD. All rights reserved.
-// http://github.com/gogo/protobuf/gogoproto
+// Copyright (c) 2015, The GoGo Authors. All rights reserved.
+// http://github.com/gogo/protobuf
 //
 // Redistribution and use in source and binary forms, with or without
 // modification, are permitted provided that the following conditions are
@@ -29,14 +29,11 @@
 package command
 
 import (
+	"fmt"
+	"go/format"
 	"io/ioutil"
 	"os"
-
-	"github.com/gogo/protobuf/proto"
-	"github.com/gogo/protobuf/protoc-gen-gogo/generator"
-	plugin "github.com/gogo/protobuf/protoc-gen-gogo/plugin"
-
-	_ "github.com/gogo/protobuf/protoc-gen-gogo/grpc"
+	"strings"
 
 	_ "github.com/gogo/protobuf/plugin/compare"
 	_ "github.com/gogo/protobuf/plugin/defaultcheck"
@@ -51,13 +48,13 @@ import (
 	_ "github.com/gogo/protobuf/plugin/populate"
 	_ "github.com/gogo/protobuf/plugin/size"
 	_ "github.com/gogo/protobuf/plugin/stringer"
+	"github.com/gogo/protobuf/plugin/testgen"
 	_ "github.com/gogo/protobuf/plugin/union"
 	_ "github.com/gogo/protobuf/plugin/unmarshal"
-
-	"github.com/gogo/protobuf/plugin/testgen"
-
-	"go/format"
-	"strings"
+	"github.com/gogo/protobuf/proto"
+	"github.com/gogo/protobuf/protoc-gen-gogo/generator"
+	_ "github.com/gogo/protobuf/protoc-gen-gogo/grpc"
+	plugin "github.com/gogo/protobuf/protoc-gen-gogo/plugin"
 )
 
 func Read() *plugin.CodeGeneratorRequest {
@@ -77,6 +74,44 @@ func Read() *plugin.CodeGeneratorRequest {
 	return g.Request
 }
 
+// filenameSuffix replaces the .pb.go at the end of each filename.
+func GeneratePlugin(req *plugin.CodeGeneratorRequest, p generator.Plugin, filenameSuffix string) *plugin.CodeGeneratorResponse {
+	g := generator.New()
+	g.Request = req
+	if len(g.Request.FileToGenerate) == 0 {
+		g.Fail("no files to generate")
+	}
+
+	g.CommandLineParameters(g.Request.GetParameter())
+
+	g.WrapTypes()
+	g.SetPackageNames()
+	g.BuildTypeNameMap()
+	g.GeneratePlugin(p)
+
+	for i := 0; i < len(g.Response.File); i++ {
+		g.Response.File[i].Name = proto.String(
+			strings.Replace(*g.Response.File[i].Name, ".pb.go", filenameSuffix, -1),
+		)
+	}
+	if err := goformat(g.Response); err != nil {
+		g.Error(err)
+	}
+	return g.Response
+}
+
+func goformat(resp *plugin.CodeGeneratorResponse) error {
+	for i := 0; i < len(resp.File); i++ {
+		formatted, err := format.Source([]byte(resp.File[i].GetContent()))
+		if err != nil {
+			return fmt.Errorf("go format error: %v", err)
+		}
+		fmts := string(formatted)
+		resp.File[i].Content = &fmts
+	}
+	return nil
+}
+
 func Generate(req *plugin.CodeGeneratorRequest) *plugin.CodeGeneratorResponse {
 	// Begin by allocating a generator. The request and response structures are stored there
 	// so we can do error handling easily - the response structure contains the field to
@@ -95,46 +130,20 @@ func Generate(req *plugin.CodeGeneratorRequest) *plugin.CodeGeneratorResponse {
 
 	g.GenerateAllFiles()
 
-	gtest := generator.New()
-
-	data, err := proto.Marshal(req)
-	if err != nil {
-		g.Error(err, "failed to marshal modified proto")
-	}
-	if err := proto.Unmarshal(data, gtest.Request); err != nil {
-		g.Error(err, "parsing modified proto")
+	if err := goformat(g.Response); err != nil {
+		g.Error(err)
 	}
 
-	if len(gtest.Request.FileToGenerate) == 0 {
-		gtest.Fail("no files to generate")
-	}
+	testReq := proto.Clone(req).(*plugin.CodeGeneratorRequest)
 
-	gtest.CommandLineParameters(gtest.Request.GetParameter())
+	testResp := GeneratePlugin(testReq, testgen.NewPlugin(), "pb_test.go")
 
-	// Create a wrapped version of the Descriptors and EnumDescriptors that
-	// point to the file that defines them.
-	gtest.WrapTypes()
-
-	gtest.SetPackageNames()
-	gtest.BuildTypeNameMap()
-
-	gtest.GeneratePlugin(testgen.NewPlugin())
-
-	for i := 0; i < len(gtest.Response.File); i++ {
-		if strings.Contains(*gtest.Response.File[i].Content, `//These tests are generated by github.com/gogo/protobuf/plugin/testgen`) {
-			gtest.Response.File[i].Name = proto.String(strings.Replace(*gtest.Response.File[i].Name, ".pb.go", "pb_test.go", -1))
-			g.Response.File = append(g.Response.File, gtest.Response.File[i])
+	for i := 0; i < len(testResp.File); i++ {
+		if strings.Contains(*testResp.File[i].Content, `//These tests are generated by github.com/gogo/protobuf/plugin/testgen`) {
+			g.Response.File = append(g.Response.File, testResp.File[i])
 		}
 	}
 
-	for i := 0; i < len(g.Response.File); i++ {
-		formatted, err := format.Source([]byte(g.Response.File[i].GetContent()))
-		if err != nil {
-			g.Error(err, "go format error")
-		}
-		fmts := string(formatted)
-		g.Response.File[i].Content = &fmts
-	}
 	return g.Response
 }