1 files changed, 106 insertions, 0 deletions
diff --git a/grus.go b/grus.go
new file mode 100644
index 0000000..d4a63ad
--- /dev/null
+++ b/grus.go
@@ -0,0 +1,106 @@
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+
+	"tildegit.org/andinus/grus/lexical"
+)
+
+func grus() {
+	if len(os.Args) == 1 {
+		fmt.Println("Usage: grus <word> <dictionaries>")
+		os.Exit(1)
+	}
+
+	version := "v0.2.0"
+
+	if os.Args[1] == "version" {
+		fmt.Printf("Grus %s\n", version)
+		os.Exit(0)
+	}
+
+	dicts := []string{
+		"/usr/local/share/dict/words",
+		"/usr/share/dict/words",
+		"/usr/share/dict/special/4bsd",
+		"/usr/share/dict/special/math",
+	}
+
+	// User has specified dictionaries, prepend them to dicts
+	// list.
+	if len(os.Args) >= 3 {
+		dicts = append(os.Args[2:], dicts...)
+	}
+
+	// Check if user has asked to search in all dictionaries.
+	searchAll := false
+	searchAllEnv := os.Getenv("GRUS_SEARCH_ALL")
+	if searchAllEnv == "true" ||
+		searchAllEnv == "1" {
+		searchAll = true
+	}
+
+	// Check if user wants anagrams.
+	anagrams := false
+	anagramsEnv := os.Getenv("GRUS_ANAGRAMS")
+	if anagramsEnv == "true" ||
+		anagramsEnv == "1" {
+		anagrams = true
+	}
+
+	for _, dict := range dicts {
+		if _, err := os.Stat(dict); err != nil &&
+			!os.IsNotExist(err) {
+			// Error is not nil & also it's not path
+			// doesn't exist error. We do it this way to
+			// avoid another level of indentation.
+			panic(err)
+		} else if err != nil &&
+			os.IsNotExist(err) {
+			// If file doesn't exist then continue with
+			// next dictionary.
+			continue
+		}
+
+		file, err := os.Open(dict)
+		panicOnErr(err)
+		defer file.Close()
+
+		// We use this to record if the word was unjumbled.
+		unjumbled := false
+
+		scanner := bufio.NewScanner(file)
+		for scanner.Scan() {
+			// Filter words by comparing length first &
+			// run lexical.Sort on it only if the length
+			// is equal.
+			if len(scanner.Text()) == len(os.Args[1]) &&
+				lexical.Sort(scanner.Text()) == lexical.Sort(os.Args[1]) {
+				fmt.Println(scanner.Text())
+				// If the user doesn't want anagrams
+				// then exit the program.
+				if !anagrams {
+					os.Exit(0)
+				}
+				unjumbled = true
+			}
+		}
+		panicOnErr(scanner.Err())
+
+		// If word was unjumbled & user hasn't asked to search
+		// in all dictionaries then exit the program otherwise
+		// keep searching in other dictionaries.
+		if unjumbled &&
+			!searchAll {
+			os.Exit(0)
+		}
+	}
+}
+
+func panicOnErr(err error) {
+	if err != nil {
+		panic(err)
+	}
+}