forked from deckarep/golang-set
/
main.go
50 lines (41 loc) · 1.05 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package main
import (
"flag"
"fmt"
"os"
)
func generateSet() error {
var structName = flag.String("struct_name", "", "name of struct to generate")
var importPath = flag.String("import_path", "", "go")
var defaultValue = flag.String("default_value", "", "default value of struct")
var makeDefaults = flag.Bool("make_defaults", false, "helper to run a series of pre-defined basic types")
flag.Parse()
templateTypes := MakeTemplateTypes()
if *makeDefaults {
setTypes := MakeDefaultSetTypes()
for _, setType := range setTypes {
err := CreateSet(setType, templateTypes)
if err != nil {
return err
}
}
return nil
}
if *structName == "" {
return NewEmptyFlagError("struct_name")
}
if *defaultValue == "" {
return NewEmptyFlagError("default_value")
}
setType := NewSetType(*structName, *importPath, *defaultValue)
return CreateSet(setType, templateTypes)
}
func main() {
var exitcode int = 0
err := generateSet()
if err != nil {
os.Stderr.WriteString(fmt.Sprintf("Error, Exiting: %v", err))
exitcode = 1
}
os.Exit(exitcode)
}