now can also read from stdin

2024-11-20 21:29:35 +01:00
1 changed files with 10 additions and 3 deletions
@@ -68,9 +68,16 @@ func main() {
 		os.Exit(1)
 	}
 	filename := os.Args[1]
-
-	file, _ := os.Open(filename)
+	var file *os.File
+	if filename != "-" {
+		file, err := os.Open(filename)
+		if err != nil {
+			panic(err)
+		}
 		defer file.Close()
+	} else {
+		file = os.Stdin
+	}

 	scanner := bufio.NewScanner(file)
 	lineno := 0