Fuyu/jsonenums
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Allow setting a custom output file which will contain all types

Browse Source 
Fixes #9
This commit is contained in:
Sean Smith 2017-06-03 22:04:33 -07:00
parent 7058b320c2
commit de36c3860c
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
jsonenums.go
View File

@ -91,6 +91,7 @@ var (
typeNames = flag.String("type", "", "comma-separated list of type names; must be set") typeNames = flag.String("type", "", "comma-separated list of type names; must be set")
outputPrefix = flag.String("prefix", "", "prefix to be added to the output file") outputPrefix = flag.String("prefix", "", "prefix to be added to the output file")
outputSuffix = flag.String("suffix", "_jsonenums", "suffix to be added to the output file") outputSuffix = flag.String("suffix", "_jsonenums", "suffix to be added to the output file")
outputFile = flag.String("file", "", "Specify the exact filename to output with. Will ignore output prefix/suffix values.")
) )
func main() { func main() {
@ -152,6 +153,10 @@ func main() {
output := strings.ToLower(*outputPrefix + typeName + output := strings.ToLower(*outputPrefix + typeName +
*outputSuffix + ".go") *outputSuffix + ".go")
if *outputFile != "" {
// This will make sure that .go is at the end whether or not a user specifies it
output = strings.TrimSuffix(*outputFile, ".go") + ".go"
}
outputPath := filepath.Join(dir, output) outputPath := filepath.Join(dir, output)
if err := ioutil.WriteFile(outputPath, src, 0644); err != nil { if err := ioutil.WriteFile(outputPath, src, 0644); err != nil {
log.Fatalf("writing output: %s", err) log.Fatalf("writing output: %s", err)