goiplookup.go

// Package main is the main application
package main

import (
	"fmt"
	"os"

	"github.com/axllent/ghru"
	flag "github.com/spf13/pflag"
)

// Flags
var (
	country       bool
	iso           bool
	showHelp      bool
	verboseOutput bool
	showVersion   bool
	dataDir       string
	licenseKey    string // GeoLite2 license key for updating
	version       = "dev"
)

// URLs
const (
	releaseURL = "https://api.github.com/repos/axllent/goiplookup/releases/latest"
)

// Main function
func main() {

	p := "/usr/share/GeoIP"
	if isDir("/usr/local/share/GeoIP") {
		// alternate default path for OSX or custom
		p = "/usr/local/share/GeoIP"
	}
	flag.StringVarP(&dataDir, "dir", "d", p, "database directory or file")

	flag.BoolVarP(&country, "country", "c", false, "return country name")
	flag.BoolVarP(&iso, "iso", "i", false, "return country iso code")
	flag.BoolVarP(&showHelp, "help", "h", false, "show help")
	flag.BoolVarP(&verboseOutput, "verbose", "v", false, "verbose/debug output")
	flag.BoolVarP(&showVersion, "version", "V", false, "show version number")

	// parse flags
	flag.Parse()

	if showVersion {
		fmt.Println(fmt.Sprintf("Version %s", version))

		latest, _, _, err := ghru.Latest("axllent/goiplookup", "goiplookup")
		if err == nil && ghru.GreaterThan(latest, version) {
			fmt.Printf("Update available: %s\nRun `%s self-update` to update\n", latest, os.Args[0])
		}
		os.Exit(0)
	}

	if len(flag.Args()) != 1 || showHelp {
		showUsage()
		return
	}

	lookup := flag.Args()[0]

	if lookup == "db-update" {
		// update database
		UpdateGeoLite2Country()
	} else if lookup == "self-update" {
		// update app if needed
		rel, err := ghru.Update("axllent/goiplookup", "goiplookup", version)
		if err != nil {
			fmt.Println(err.Error())
			os.Exit(1)
		}
		fmt.Printf("Updated %s to version %s\n", os.Args[0], rel)
		os.Exit(0)
	} else {
		// lookup ip/hostname
		Lookup(lookup)
	}
}

// ShowUsage prints the help function
var showUsage = func() {
	fmt.Printf("Usage: %s [-i] [-c] [-d <database directory>] <ipaddress|hostname|db-update|self-update>\n", os.Args[0])
	fmt.Println("\nGoiplookup uses the GeoLite2-Country database to find the Country that an IP address or hostname originates from.")
	fmt.Println("\nOptions:")
	flag.PrintDefaults()
	fmt.Println("\nExamples:")
	fmt.Printf("%s 8.8.8.8               # Return the country ISO code and name\n", os.Args[0])
	fmt.Printf("%s -d ~/GeoIP 8.8.8.8    # Use a different database directory\n", os.Args[0])
	fmt.Printf("%s -i 8.8.8.8            # Return just the country ISO code\n", os.Args[0])
	fmt.Printf("%s -c 8.8.8.8            # Return just the country name\n", os.Args[0])
	fmt.Printf("%s db-update             # Update the GeoLite2-Country database (do not run more than once a month)\n", os.Args[0])
	fmt.Printf("%s self-update           # Update the GoIpLookup binary with the latest release\n", os.Args[0])
}