Switches to filename argument

whoerau · Jun 2, 2016 · 965c87f · 965c87f
1 parent 71fe44f
commit 965c87f
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/gron b/gron
diff --git a/main.go b/main.go
@@ -2,14 +2,27 @@ package main
 
 import (
 	"encoding/json"
+	"flag"
 	"fmt"
 	"io/ioutil"
 	"log"
+	"os"
 	"regexp"
 )
 
+const (
+	exitOK = iota
+	exitInvalidArgs
+)
+
 func main() {
-	filename := "test-input.json"
+	flag.Parse()
+
+	filename := flag.Arg(0)
+	if filename == "" {
+		os.Exit(exitInvalidArgs)
+	}
+
 	raw, err := ioutil.ReadFile(filename)
 	if err != nil {
 		todo(err)

diff --git a/test-input.json → testdata/one.json b/test-input.json → testdata/one.json