Allow the disassembler to accept input from stdin

nlowe · Dec 12, 2019 · 6963bf4 · 6963bf4
1 parent ca46f81
commit 6963bf4
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/intcode/disassembler/main.go b/intcode/disassembler/main.go
@@ -20,7 +20,15 @@ func main() {
 		os.Exit(1)
 	}
 
-	f, err := ioutil.ReadFile(os.Args[1])
+	var f []byte
+	var err error
+
+	if os.Args[1] == "-" {
+		f, err = ioutil.ReadAll(os.Stdin)
+	} else {
+		f, err = ioutil.ReadFile(os.Args[1])
+	}
+
 	if err != nil {
 		panic(err)
 	}