func MustParseGlobs(args []string) (list []Routes, filepaths []string) {
for _, fp := range genex.MustParseGlobs(args[:]) {
if strings.HasSuffix(fp, ".tml") {
var routes Routes
if _, err := toml.DecodeFile(fp, &routes); err != nil {
panic(err)
}
filepaths = append(filepaths, fp)
list = append(list, routes)
}
}
return
}
func main() {
if flag.NArg() < 1 {
flag.Usage()
}
for _, f := range genex.MustParseGlobs(flag.Args()[:]) {
var tmp interface{}
md, err := toml.DecodeFile(f, &tmp)
if err != nil {
log.Fatalf("Error in '%s': %s", f, err)
}
if flagTypes {
printTypes(md)
}
}
}
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
}
