Beispiel #1
0
func execute(gen genex.Generator, src, name string, params map[string]interface{}) error {
	// Format the output.
	funcMap := map[string]interface{}{
		"join": func(s []string) string {
			return strings.Join(s, ",")
		},
	}
	t := template.Must(template.New("").Funcs(funcMap).Parse(src))
	var buf bytes.Buffer
	t.Execute(&buf, params)
	gen.Printf("%s", buf.String())
	formattedSrc := gen.Format()

	// Write to file.
	return ioutil.WriteFile(name, formattedSrc, 0644)
}
Beispiel #2
0
func run() int {
	var (
		args = flag.Args()
		g    genex.Generator
	)

	// Print the header and package clause.
	g.Printf("// Code generated by gen-config.\n")
	// Run generate for each type.
	filename := []string{}
	config := map[string]interface{}{}
	for _, fp := range genex.MustParseGlobs(args[:]) {
		if strings.HasSuffix(fp, ".tml") {
			if _, err := toml.DecodeFile(fp, &config); err == nil {
				g.Printf("// %s\n", fp)
				filename = append(filename, fp)
			}
		}
	}
	g.Printf("// DO NOT EDIT\n")
	g.Printf("\npackage %s\n", *pkg)

	// Generate two source files.
	params := map[string]interface{}{}
	params["ConfigDir"] = *confDir
	params["TypeBody"] = printer(config)
	params["Filename"] = func() (ret []string) {
		for _, v := range filename {
			ret = append(ret, strings.Replace(v, *confDir+"/", "", -1))
		}
		return
	}()
	params["StructName"] = func(cfg map[string]interface{}) (ret []map[string]string) {
		for k, _ := range cfg {
			elm := map[string]string{}
			elm["Struct"] = snaker.SnakeToCamel(k)
			elm["Tag"] = fmt.Sprintf("`toml:\"%s\"`", k)
			ret = append(ret, elm)
		}
		return
	}(config)

	// Format the output.
	t := template.Must(template.New("").Parse(text))
	var buf bytes.Buffer
	t.Execute(&buf, params)
	g.Printf("%s", buf.String())
	src := g.Format()

	// Write to file.
	outputName := *output
	if outputName == "" {
		baseName := OUTPUT
		outputName = filepath.Join(*srcDir, strings.ToLower(baseName))
	}
	err := ioutil.WriteFile(outputName, src, 0644)
	if err != nil {
		log.Fatalf("writing output: %s", err)
	}

	return 0
}
Beispiel #3
0
func run() int {
	var (
		args = flag.Args()
		mg   genex.Generator
		g    genex.Generator
	)

	// Print the header and package clause.
	mg.Printf("// Code generated by gen-routes.\n// DO NOT EDIT\n")
	mg.Printf("\npackage %s\n", *pkg)

	// Print the header and package clause.
	g.Printf("// Code generated by gen-routes.\n")
	// Set Routes from args
	list, filepaths := MustParseGlobs(args[:])
	for _, fp := range filepaths {
		g.Printf("// %s\n", fp)
	}
	g.Printf("// DO NOT EDIT\n")
	g.Printf("\npackage %s\n", *pkg)

	// Generate two source files.
	params := map[string]interface{}{}
	type Element struct {
		// Essentials
		Method, Path, Controller string
		// Controller
		ControllerPack, ControllerName, ControllerFunc string
		// Param
		Param []string
	}
	data := []Element{}
	statics := []Static{}
	htmls := []HTML{}
	for _, routes := range list {
		for _, route := range routes.Endpoint.Routes {
			elm := Element{}
			elm.Param = []string{}
			// Essentials
			elm.Method = route.Method
			elm.Path = path.Join("/", route.Path)
			elm.Controller = route.Controller

			// Controller
			_, t, fn := parsePackage(route.Controller)
			elm.ControllerFunc = fn
			_, elm.ControllerPack, elm.ControllerName = parsePackage(t)

			// Param
			elm.Param = parsePatternPath(elm.Path)
			// To be c.Param("%s")
			for i := 0; i < len(elm.Param); i++ {
				elm.Param[i] = fmt.Sprintf("c.Param(\"%s\")", elm.Param[i])
			}

			// append
			data = append(data, elm)
		}
		statics = append(statics, routes.Statics...)
		htmls = append(htmls, routes.HTML...)
	}
	params["data"] = data
	params["package"] = func() (ret []string) {
		ins := map[string]bool{}
		for _, routes := range list {
			for _, v := range routes.Packages {
				if ok, found := ins[v.Name]; found && ok {
					continue
				}
				ins[v.Name] = true
				ret = append(ret, v.Name)
			}
		}
		return
	}()
	params["static"] = statics
	params["html"] = htmls
	params["appName"] = filepath.Base(strings.Replace(*srcDir, "/internal/app", "", 1))

	// Write to routes_gen.go
	outputName := *output
	if outputName == "" {
		baseName := OUTPUT
		outputName = filepath.Join(*srcDir, strings.ToLower(baseName))
	}
	if !strings.HasSuffix(outputName, ".go") {
		outputName = outputName + ".go"
	}
	if err := execute(g, routesSource, outputName, params); err != nil {
		log.Fatalf("writing output: %s", err)
	}

	// Write to main_gen.go
	outputName = filepath.Join(*srcDir, "main_gen.go")
	var src string
	if v := os.Getenv("SOURCE_GAE"); v != "" {
		src = gaeSource
	} else {
		src = mainSource
	}
	if err := execute(mg, src, outputName, params); err != nil {
		log.Fatalf("writing output: %s", err)
	}

	return 0
}