Rename ipd -> echoip

cmd/echoip/main.go

@@ -0,0 +1,55 @@
+package main
+
+import (
+	"log"
+
+	flags "github.com/jessevdk/go-flags"
+
+	"os"
+
+	"github.com/mpolden/echoip/http"
+	"github.com/mpolden/echoip/iputil"
+	"github.com/mpolden/echoip/iputil/geo"
+)
+
+func main() {
+	var opts struct {
+		CountryDBPath string   `short:"f" long:"country-db" description:"Path to GeoIP country database" value-name:"FILE" default:""`
+		CityDBPath    string   `short:"c" long:"city-db" description:"Path to GeoIP city database" value-name:"FILE" default:""`
+		Listen        string   `short:"l" long:"listen" description:"Listening address" value-name:"ADDR" default:":8080"`
+		ReverseLookup bool     `short:"r" long:"reverse-lookup" description:"Perform reverse hostname lookups"`
+		PortLookup    bool     `short:"p" long:"port-lookup" description:"Enable port lookup"`
+		Template      string   `short:"t" long:"template" description:"Path to template" default:"index.html" value-name:"FILE"`
+		IPHeaders     []string `short:"H" long:"trusted-header" description:"Header to trust for remote IP, if present (e.g. X-Real-IP)" value-name:"NAME"`
+	}
+	_, err := flags.ParseArgs(&opts, os.Args)
+	if err != nil {
+		os.Exit(1)
+	}
+
+	log := log.New(os.Stderr, "echoip: ", 0)
+	r, err := geo.Open(opts.CountryDBPath, opts.CityDBPath)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	server := http.New(r)
+	server.Template = opts.Template
+	server.IPHeaders = opts.IPHeaders
+	if opts.ReverseLookup {
+		log.Println("Enabling reverse lookup")
+		server.LookupAddr = iputil.LookupAddr
+	}
+	if opts.PortLookup {
+		log.Println("Enabling port lookup")
+		server.LookupPort = iputil.LookupPort
+	}
+	if len(opts.IPHeaders) > 0 {
+		log.Printf("Trusting header(s) %+v to contain correct remote IP", opts.IPHeaders)
+	}
+
+	log.Printf("Listening on http://%s", opts.Listen)
+	if err := server.ListenAndServe(opts.Listen); err != nil {
+		log.Fatal(err)
+	}
+}