Function main

github/gen-accessors.go:83–127

()

}

func main() {
	flag.Parse()

	// For debugging purposes, processing just a single or a few files is helpful:
	var processOnly map[string]bool
	if *verbose { // Only create the map if args are provided.
		for _, arg := range flag.Args() {
			if processOnly == nil {
				processOnly = map[string]bool{}
			}
			processOnly[arg] = true
		}
	}

	fset := token.NewFileSet()

	pkgs, err := parser.ParseDir(fset, ".", sourceFilter, 0)
	if err != nil {
		log.Fatal(err)
		return
	}

	for pkgName, pkg := range pkgs {
		t := &templateData{
			filename: pkgName + fileSuffix,
			Year:     2017,
			Package:  pkgName,
			Imports:  map[string]string{},
		}
		for filename, f := range pkg.Files {
			if *verbose && processOnly != nil && !processOnly[filename] {
				continue
			}

			logf("Processing %v...", filename)
			if err := t.processAST(f); err != nil {
				log.Fatal(err)
			}
		}
		if err := t.dump(); err != nil {
			log.Fatal(err)
		}
	}
	logf("Done.")
}

func (t *templateData) processAST(f *ast.File) error {
	for _, decl := range f.Decls {