Add support for reading from mailq directly

main.go

@@ -2,22 +2,58 @@ package main
 
 import (
 	"bufio"
+	"flag"
 	"fmt"
 	"mailq-inspector/parser"
 	"os"
+	"os/exec"
 )
 
+var settings = make(map[string]string)
+
+func argParse() {
+	inputModePtr := flag.String("inputMode", "stdin", "How to get mailq output. Available options: \"stdin\", 'cmd'")
+	flag.Parse()
+	settings["inputMode"] = *inputModePtr
+}
+
 func main() {
-	// Read data from stdin
-	scanner := bufio.NewScanner(os.Stdin)
-	entries, err := parser.ParseMailQ(scanner)
-	if err != nil {
-		fmt.Printf("Got error from parser: %s\n", err.Error())
-		os.Exit(1)
+	argParse()
+	var entries []parser.QEntry
+	var err error
+	if settings["inputMode"] == "stdin" {
+		entries, err = readFromStdin()
+	} else if settings["inputMode"] == "cmd" {
+		entries, err = readFromCmd(exec.Command("mailq"))
 	} else {
-		for _, entry := range entries {
-			fmt.Printf("%s\n", entry.String())
-		}
-		os.Exit(0)
+		fmt.Printf("Unsupported inputMode!")
+		flag.PrintDefaults()
+		os.Exit(1)
+	}
+
+	if err != nil {
+		fmt.Printf("Error reading: %s\n", err.Error())
+		os.Exit(1)
+	}
+
+	for _, line := range entries {
+		fmt.Printf("Read: %s", line)
 	}
 }
+
+func readFromStdin() ([]parser.QEntry, error) {
+	scanner := bufio.NewScanner(os.Stdin)
+	return parser.ParseMailQ(scanner)
+}
+
+func readFromCmd(cmd *exec.Cmd) ([]parser.QEntry, error) {
+	stdout, err := cmd.StdoutPipe()
+	err = cmd.Start()
+	scanner := bufio.NewScanner(stdout)
+	result, resErr := parser.ParseMailQ(scanner)
+	err = cmd.Wait()
+	if err != nil {
+		fmt.Printf("Oh no: %s", err.Error())
+	}
+	return result, resErr
+}