- [+] commandlineFlag now deal with bool variables

Author: suntong

test/commandlineFlag.ref

@@ -56,7 +56,11 @@ func init() {
 	flag.IntVar(&Opts.Debug, "debug", 0,
 		"debugging `level`")
 
+	exists := false
 	// Now override those default values from environment variables
+	if _, exists = os.LookupEnv("EASYGEN_HTML"); Opts.HTML|| exists {
+		Opts.HTML = true
+	}
 	if len(Opts.TemplateStr) == 0 ||
 		len(os.Getenv("EASYGEN_TS")) != 0 {
 		Opts.TemplateStr = os.Getenv("EASYGEN_TS")

test/commandlineFlag.tmpl

@@ -37,13 +37,17 @@ func init() {
 	flag.{{clk2uc .Type}}Var(&{{$.StructVar}}.{{.Name}}, "{{.Flag}}", {{.Value}},
 		"{{.Usage}}"){{end}}{{end}}
 
+	exists := false
 	// Now override those default values from environment variables{{range .Options}}{{if eq .Name "SEPARATOR" }}
 
-{{else}}{{if eq .Type "string" }}{{$envVar := printf "%s_%s" (clk2ss $.ProgramName) (clk2ss .Flag)}}
+{{else}}{{$envVar := printf "%s_%s" (clk2ss $.ProgramName) (clk2ss .Flag)}}{{if eq .Type "string" }}
 	if len({{$.StructVar}}.{{.Name}}) == 0 ||
 		len(os.Getenv("{{$envVar}}")) != 0 {
 		{{$.StructVar}}.{{.Name}} = os.Getenv("{{$envVar}}")
-	}{{end}}{{end}}{{end}}
+	}{{else}}{{if eq .Type "bool" }}
+	if _, exists = os.LookupEnv("{{$envVar}}"); {{$.StructVar}}.{{.Name}}|| exists {
+		{{$.StructVar}}.{{.Name}} = true
+	}{{end}}{{end}}{{end}}{{end}}
 
 }