summaryrefslogtreecommitdiffstats
path: root/vendor/github.com/mailru/easyjson/easyjson/main.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/mailru/easyjson/easyjson/main.go')
-rw-r--r--vendor/github.com/mailru/easyjson/easyjson/main.go106
1 files changed, 106 insertions, 0 deletions
diff --git a/vendor/github.com/mailru/easyjson/easyjson/main.go b/vendor/github.com/mailru/easyjson/easyjson/main.go
new file mode 100644
index 000000000..1cd30bb36
--- /dev/null
+++ b/vendor/github.com/mailru/easyjson/easyjson/main.go
@@ -0,0 +1,106 @@
+package main
+
+import (
+ "errors"
+ "flag"
+ "fmt"
+ "os"
+ "path/filepath"
+ "strings"
+
+ "github.com/mailru/easyjson/bootstrap"
+ // Reference the gen package to be friendly to vendoring tools,
+ // as it is an indirect dependency.
+ // (The temporary bootstrapping code uses it.)
+ _ "github.com/mailru/easyjson/gen"
+ "github.com/mailru/easyjson/parser"
+)
+
+var buildTags = flag.String("build_tags", "", "build tags to add to generated file")
+var snakeCase = flag.Bool("snake_case", false, "use snake_case names instead of CamelCase by default")
+var lowerCamelCase = flag.Bool("lower_camel_case", false, "use lowerCamelCase names instead of CamelCase by default")
+var noStdMarshalers = flag.Bool("no_std_marshalers", false, "don't generate MarshalJSON/UnmarshalJSON funcs")
+var omitEmpty = flag.Bool("omit_empty", false, "omit empty fields by default")
+var allStructs = flag.Bool("all", false, "generate marshaler/unmarshalers for all structs in a file")
+var leaveTemps = flag.Bool("leave_temps", false, "do not delete temporary files")
+var stubs = flag.Bool("stubs", false, "only generate stubs for marshaler/unmarshaler funcs")
+var noformat = flag.Bool("noformat", false, "do not run 'gofmt -w' on output file")
+var specifiedName = flag.String("output_filename", "", "specify the filename of the output")
+var processPkg = flag.Bool("pkg", false, "process the whole package instead of just the given file")
+
+func generate(fname string) (err error) {
+ fInfo, err := os.Stat(fname)
+ if err != nil {
+ return err
+ }
+
+ p := parser.Parser{AllStructs: *allStructs}
+ if err := p.Parse(fname, fInfo.IsDir()); err != nil {
+ return fmt.Errorf("Error parsing %v: %v", fname, err)
+ }
+
+ var outName string
+ if fInfo.IsDir() {
+ outName = filepath.Join(fname, p.PkgName+"_easyjson.go")
+ } else {
+ if s := strings.TrimSuffix(fname, ".go"); s == fname {
+ return errors.New("Filename must end in '.go'")
+ } else {
+ outName = s + "_easyjson.go"
+ }
+ }
+
+ if *specifiedName != "" {
+ outName = *specifiedName
+ }
+
+ var trimmedBuildTags string
+ if *buildTags != "" {
+ trimmedBuildTags = strings.TrimSpace(*buildTags)
+ }
+
+ g := bootstrap.Generator{
+ BuildTags: trimmedBuildTags,
+ PkgPath: p.PkgPath,
+ PkgName: p.PkgName,
+ Types: p.StructNames,
+ SnakeCase: *snakeCase,
+ LowerCamelCase: *lowerCamelCase,
+ NoStdMarshalers: *noStdMarshalers,
+ OmitEmpty: *omitEmpty,
+ LeaveTemps: *leaveTemps,
+ OutName: outName,
+ StubsOnly: *stubs,
+ NoFormat: *noformat,
+ }
+
+ if err := g.Run(); err != nil {
+ return fmt.Errorf("Bootstrap failed: %v", err)
+ }
+ return nil
+}
+
+func main() {
+ flag.Parse()
+
+ files := flag.Args()
+
+ gofile := os.Getenv("GOFILE")
+ if *processPkg {
+ gofile = filepath.Dir(gofile)
+ }
+
+ if len(files) == 0 && gofile != "" {
+ files = []string{gofile}
+ } else if len(files) == 0 {
+ flag.Usage()
+ os.Exit(1)
+ }
+
+ for _, fname := range files {
+ if err := generate(fname); err != nil {
+ fmt.Fprintln(os.Stderr, err)
+ os.Exit(1)
+ }
+ }
+}