vendor files

vendor/github.com/mailru/easyjson/parser/parser.go

@@ -0,0 +1,97 @@
+package parser
+
+import (
+	"go/ast"
+	"go/parser"
+	"go/token"
+	"os/exec"
+	"strings"
+)
+
+const structComment = "easyjson:json"
+
+type Parser struct {
+	PkgPath     string
+	PkgName     string
+	StructNames []string
+	AllStructs  bool
+}
+
+type visitor struct {
+	*Parser
+
+	name     string
+	explicit bool
+}
+
+func (p *Parser) needType(comments string) bool {
+	for _, v := range strings.Split(comments, "\n") {
+		if strings.HasPrefix(v, structComment) {
+			return true
+		}
+	}
+	return false
+}
+
+func (v *visitor) Visit(n ast.Node) (w ast.Visitor) {
+	switch n := n.(type) {
+	case *ast.Package:
+		return v
+	case *ast.File:
+		v.PkgName = n.Name.String()
+		return v
+
+	case *ast.GenDecl:
+		v.explicit = v.needType(n.Doc.Text())
+
+		if !v.explicit && !v.AllStructs {
+			return nil
+		}
+		return v
+	case *ast.TypeSpec:
+		v.name = n.Name.String()
+
+		// Allow to specify non-structs explicitly independent of '-all' flag.
+		if v.explicit {
+			v.StructNames = append(v.StructNames, v.name)
+			return nil
+		}
+		return v
+	case *ast.StructType:
+		v.StructNames = append(v.StructNames, v.name)
+		return nil
+	}
+	return nil
+}
+
+func (p *Parser) Parse(fname string, isDir bool) error {
+	var err error
+	if p.PkgPath, err = getPkgPath(fname, isDir); err != nil {
+		return err
+	}
+
+	fset := token.NewFileSet()
+	if isDir {
+		packages, err := parser.ParseDir(fset, fname, nil, parser.ParseComments)
+		if err != nil {
+			return err
+		}
+
+		for _, pckg := range packages {
+			ast.Walk(&visitor{Parser: p}, pckg)
+		}
+	} else {
+		f, err := parser.ParseFile(fset, fname, nil, parser.ParseComments)
+		if err != nil {
+			return err
+		}
+
+		ast.Walk(&visitor{Parser: p}, f)
+	}
+	return nil
+}
+
+func getDefaultGoPath() (string, error) {
+	output, err := exec.Command("go", "env", "GOPATH").Output()
+	return string(output), err
+}