vendor/github.com/golang/protobuf/protoc-gen-go/main.go

   1 // Go support for Protocol Buffers - Google's data interchange format
   2 //
   3 // Copyright 2010 The Go Authors.  All rights reserved.
   4 // https://github.com/golang/protobuf
   5 //
   6 // Redistribution and use in source and binary forms, with or without
   7 // modification, are permitted provided that the following conditions are
   8 // met:
   9 //
  10 //     * Redistributions of source code must retain the above copyright
  11 // notice, this list of conditions and the following disclaimer.
  12 //     * Redistributions in binary form must reproduce the above
  13 // copyright notice, this list of conditions and the following disclaimer
  14 // in the documentation and/or other materials provided with the
  15 // distribution.
  16 //     * Neither the name of Google Inc. nor the names of its
  17 // contributors may be used to endorse or promote products derived from
  18 // this software without specific prior written permission.
  19 //
  20 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  21 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  22 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  23 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  24 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  25 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  26 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  27 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  28 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  29 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  30 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31
  32 // protoc-gen-go is a plugin for the Google protocol buffer compiler to generate
  33 // Go code.  Run it by building this program and putting it in your path with
  34 // the name
  35 //      protoc-gen-go
  36 // That word 'go' at the end becomes part of the option string set for the
  37 // protocol compiler, so once the protocol compiler (protoc) is installed
  38 // you can run
  39 //      protoc --go_out=output_directory input_directory/file.proto
  40 // to generate Go bindings for the protocol defined by file.proto.
  41 // With that input, the output will be written to
  42 //      output_directory/file.pb.go
  43 //
  44 // The generated code is documented in the package comment for
  45 // the library.
  46 //
  47 // See the README and documentation for protocol buffers to learn more:
  48 //      https://developers.google.com/protocol-buffers/
  49 package main
  50
  51 import (
  52         "io/ioutil"
  53         "os"
  54
  55         "github.com/golang/protobuf/proto"
  56         "github.com/golang/protobuf/protoc-gen-go/generator"
  57 )
  58
  59 func main() {
  60         // Begin by allocating a generator. The request and response structures are stored there
  61         // so we can do error handling easily - the response structure contains the field to
  62         // report failure.
  63         g := generator.New()
  64
  65         data, err := ioutil.ReadAll(os.Stdin)
  66         if err != nil {
  67                 g.Error(err, "reading input")
  68         }
  69
  70         if err := proto.Unmarshal(data, g.Request); err != nil {
  71                 g.Error(err, "parsing input proto")
  72         }
  73
  74         if len(g.Request.FileToGenerate) == 0 {
  75                 g.Fail("no files to generate")
  76         }
  77
  78         g.CommandLineParameters(g.Request.GetParameter())
  79
  80         // Create a wrapped version of the Descriptors and EnumDescriptors that
  81         // point to the file that defines them.
  82         g.WrapTypes()
  83
  84         g.SetPackageNames()
  85         g.BuildTypeNameMap()
  86
  87         g.GenerateAllFiles()
  88
  89         // Send back the results.
  90         data, err = proto.Marshal(g.Response)
  91         if err != nil {
  92                 g.Error(err, "failed to marshal output proto")
  93         }
  94         _, err = os.Stdout.Write(data)
  95         if err != nil {
  96                 g.Error(err, "failed to write output proto")
  97         }
  98 }