diff options
author | Martin Polden <mpolden@mpolden.no> | 2018-02-09 23:57:33 +0300 |
---|---|---|
committer | Martin Polden <mpolden@mpolden.no> | 2018-02-09 23:57:33 +0300 |
commit | ddb1aa2bb9ff69a4333d6a1f82c1ebd6be0544e4 (patch) | |
tree | 1061e9ec4598baf296f17a8a81721e4b5c814cf0 /cmd | |
parent | a7ff603ab10b24599689aa1eab96522ea9659316 (diff) |
Move main to subpackage
Diffstat (limited to 'cmd')
-rw-r--r-- | cmd/ipd/main.go | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/cmd/ipd/main.go b/cmd/ipd/main.go new file mode 100644 index 0000000..d896ce7 --- /dev/null +++ b/cmd/ipd/main.go @@ -0,0 +1,71 @@ +package main + +import ( + "net/http" + + flags "github.com/jessevdk/go-flags" + + "os" + + "github.com/sirupsen/logrus" + + "github.com/mpolden/ipd/api" +) + +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"` + IPHeader string `short:"H" long:"trusted-header" description:"Header to trust for remote IP, if present (e.g. X-Real-IP)" value-name:"NAME"` + LogLevel string `short:"L" long:"log-level" description:"Log level to use" default:"info" choice:"debug" choice:"info" choice:"warn" choice:"error" choice:"fatal" choice:"panic"` + } + _, err := flags.ParseArgs(&opts, os.Args) + if err != nil { + os.Exit(1) + } + + log := logrus.New() + level, err := logrus.ParseLevel(opts.LogLevel) + if err != nil { + log.Fatal(err) + } + log.Level = level + + oracle := api.NewOracle() + if opts.ReverseLookup { + log.Println("Enabling reverse lookup") + oracle.EnableLookupAddr() + } + if opts.PortLookup { + log.Println("Enabling port lookup") + oracle.EnableLookupPort() + } + if opts.CountryDBPath != "" { + log.Printf("Enabling country lookup (using database: %s)", opts.CountryDBPath) + if err := oracle.EnableLookupCountry(opts.CountryDBPath); err != nil { + log.Fatal(err) + } + } + if opts.CityDBPath != "" { + log.Printf("Enabling city lookup (using database: %s)", opts.CityDBPath) + if err := oracle.EnableLookupCity(opts.CityDBPath); err != nil { + log.Fatal(err) + } + } + if opts.IPHeader != "" { + log.Printf("Trusting header %s to contain correct remote IP", opts.IPHeader) + } + + api := api.New(oracle, log) + api.Template = opts.Template + api.IPHeader = opts.IPHeader + + log.Printf("Listening on http://%s", opts.Listen) + if err := http.ListenAndServe(opts.Listen, api.Router()); err != nil { + log.Fatal(err) + } +} |