Forgot to implement file arg reading

ilkka · Jan 22, 2015 · e72f373 · e72f373
1 parent f4cbe84
commit e72f373
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/substenv.go b/substenv.go
@@ -41,8 +41,11 @@ func RegexParserExpand(line string) string {
 func main() {
     app.Version("1.0.0")
     app.Parse(os.Args[1:])
-
-    var bio = bufio.NewReader(os.Stdin)
+    var file = os.Stdin
+    if *input != nil {
+        file = *input
+    }
+    var bio = bufio.NewReader(file)
     for {
         var line, err = bio.ReadString('\n')
         if err != nil {