initial commit

utils.go

@@ -0,0 +1,71 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"os"
+	"regexp"
+	"strings"
+)
+
+type Verbosity int
+
+const (
+	Info Verbosity = iota
+	Warning
+	Error
+	Debug
+)
+
+func debugOut(level Verbosity, text string) {
+	switch level {
+	case Info:
+		if verbose {
+			fmt.Println("[INFO] " + text)
+		} else {
+			fmt.Println(text)
+		}
+	case Warning:
+		if verbose {
+			fmt.Println("[WARNING] " + text)
+		}
+	case Error:
+		fmt.Println("[ERROR] " + text)
+		os.Exit(1)
+	case Debug:
+		if verbose {
+			fmt.Println("[DEBUG] " + text)
+		}
+	}
+}
+
+func CheckValidIP(ip string) {
+	re := regexp.MustCompile(`^(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])$`)
+	if !re.MatchString(ip) {
+		debugOut(Error, "Please enter a valid IP address")
+	}
+}
+
+func parseFlags() {
+	_ip := flag.String("ip", "", "IP address to lookup")
+	_props := flag.String("p", "country", "Properties to retrieve")
+	_verbose := flag.Bool("v", false, "Verbose output")
+	flag.Parse()
+	_loose := flag.Args()
+
+	if *_verbose {
+		verbose = true
+	}
+	if *_ip == "" {
+		if len(_loose) == 0 {
+			debugOut(Error, "Please enter an IP address or use -h for help")
+		} else {
+			ip = _loose[0]
+		}
+	} else {
+		ip = *_ip
+	}
+	if _props != nil {
+		properties = strings.Split(*_props, ",")
+	}
+}