diff options
Diffstat (limited to 'cmd/binapi-generator/main.go')
| -rw-r--r-- | cmd/binapi-generator/main.go | 21 |
1 files changed, 16 insertions, 5 deletions
diff --git a/cmd/binapi-generator/main.go b/cmd/binapi-generator/main.go index 8045212..b73a699 100644 --- a/cmd/binapi-generator/main.go +++ b/cmd/binapi-generator/main.go @@ -20,13 +20,13 @@ import ( "flag" "fmt" "io/ioutil" - "log" "os" "os/exec" "path/filepath" "strings" "github.com/bennyscetbun/jsongo" + "github.com/sirupsen/logrus" ) var ( @@ -38,15 +38,26 @@ var ( continueOnError = flag.Bool("continue-onerror", false, "Wheter to continue with next file on error.") ) +func init() { + flag.Parse() + if *debug { + logrus.SetLevel(logrus.DebugLevel) + } +} + func logf(f string, v ...interface{}) { if *debug { - log.Printf(f, v...) + logrus.Debugf(f, v...) } } -func main() { - flag.Parse() +var log = logrus.Logger{ + Level: logrus.InfoLevel, + Formatter: &logrus.TextFormatter{}, + Out: os.Stdout, +} +func main() { if *inputFile == "" && *inputDir == "" { fmt.Fprintln(os.Stderr, "ERROR: input-file or input-dir must be specified") os.Exit(1) @@ -143,7 +154,7 @@ func generateFromFile(inputFile, outputDir string) error { // count number of lines in generated output file cmd = exec.Command("wc", "-l", ctx.outputFile) if output, err := cmd.CombinedOutput(); err != nil { - log.Printf("wc command failed: %v\n%s", err, string(output)) + log.Warnf("wc command failed: %v\n%s", err, string(output)) } else { logf("generated lines: %s", output) } |