From 059db6372cce76f446112d33281c76c8b91f6c05 Mon Sep 17 00:00:00 2001
From: Martin Polden <mpolden@mpolden.no>
Date: Tue, 14 Aug 2018 21:00:46 +0200
Subject: [PATCH] Rename package database -> geo

---
 cmd/ipd/main.go                             |  6 +++---
 http/http.go                                | 14 +++++++-------
 http/http_test.go                           | 10 +++++-----
 iputil/{database/database.go => geo/geo.go} |  6 +++---
 4 files changed, 18 insertions(+), 18 deletions(-)
 rename iputil/{database/database.go => geo/geo.go} (94%)

diff --git a/cmd/ipd/main.go b/cmd/ipd/main.go
index 70d5c52..4d659bf 100644
--- a/cmd/ipd/main.go
+++ b/cmd/ipd/main.go
@@ -9,7 +9,7 @@ import (
 
 	"github.com/mpolden/ipd/http"
 	"github.com/mpolden/ipd/iputil"
-	"github.com/mpolden/ipd/iputil/database"
+	"github.com/mpolden/ipd/iputil/geo"
 )
 
 func main() {
@@ -28,12 +28,12 @@ func main() {
 	}
 
 	log := log.New(os.Stderr, "ipd: ", 0)
-	db, err := database.New(opts.CountryDBPath, opts.CityDBPath)
+	r, err := geo.Open(opts.CountryDBPath, opts.CityDBPath)
 	if err != nil {
 		log.Fatal(err)
 	}
 
-	server := http.New(db)
+	server := http.New(r)
 	server.Template = opts.Template
 	server.IPHeaders = opts.IPHeaders
 	if opts.ReverseLookup {
diff --git a/http/http.go b/http/http.go
index 5639239..82295ed 100644
--- a/http/http.go
+++ b/http/http.go
@@ -7,7 +7,7 @@ import (
 	"path/filepath"
 
 	"github.com/mpolden/ipd/iputil"
-	"github.com/mpolden/ipd/iputil/database"
+	"github.com/mpolden/ipd/iputil/geo"
 	"github.com/mpolden/ipd/useragent"
 
 	"net"
@@ -25,7 +25,7 @@ type Server struct {
 	IPHeaders  []string
 	LookupAddr func(net.IP) (string, error)
 	LookupPort func(net.IP, uint64) error
-	db         database.Client
+	gr         geo.Reader
 }
 
 type Response struct {
@@ -43,8 +43,8 @@ type PortResponse struct {
 	Reachable bool   `json:"reachable"`
 }
 
-func New(db database.Client) *Server {
-	return &Server{db: db}
+func New(db geo.Reader) *Server {
+	return &Server{gr: db}
 }
 
 func ipFromRequest(headers []string, r *http.Request) (net.IP, error) {
@@ -75,8 +75,8 @@ func (s *Server) newResponse(r *http.Request) (Response, error) {
 		return Response{}, err
 	}
 	ipDecimal := iputil.ToDecimal(ip)
-	country, _ := s.db.Country(ip)
-	city, _ := s.db.City(ip)
+	country, _ := s.gr.Country(ip)
+	city, _ := s.gr.City(ip)
 	var hostname string
 	if s.LookupAddr != nil {
 		hostname, _ = s.LookupAddr(ip)
@@ -264,7 +264,7 @@ func (s *Server) Handler() http.Handler {
 	r.Route("GET", "/", s.CLIHandler).MatcherFunc(cliMatcher)
 	r.Route("GET", "/", s.CLIHandler).Header("Accept", textMediaType)
 	r.Route("GET", "/ip", s.CLIHandler)
-	if !s.db.IsEmpty() {
+	if !s.gr.IsEmpty() {
 		r.Route("GET", "/country", s.CLICountryHandler)
 		r.Route("GET", "/country-iso", s.CLICountryISOHandler)
 		r.Route("GET", "/city", s.CLICityHandler)
diff --git a/http/http_test.go b/http/http_test.go
index d55fba7..c30bb05 100644
--- a/http/http_test.go
+++ b/http/http_test.go
@@ -8,7 +8,7 @@ import (
 	"net/http/httptest"
 	"testing"
 
-	"github.com/mpolden/ipd/iputil/database"
+	"github.com/mpolden/ipd/iputil/geo"
 )
 
 func lookupAddr(net.IP) (string, error) { return "localhost", nil }
@@ -16,15 +16,15 @@ func lookupPort(net.IP, uint64) error   { return nil }
 
 type testDb struct{}
 
-func (t *testDb) Country(net.IP) (database.Country, error) {
-	return database.Country{Name: "Elbonia", ISO: "EB"}, nil
+func (t *testDb) Country(net.IP) (geo.Country, error) {
+	return geo.Country{Name: "Elbonia", ISO: "EB"}, nil
 }
 
 func (t *testDb) City(net.IP) (string, error) { return "Bornyasherk", nil }
 func (t *testDb) IsEmpty() bool               { return false }
 
 func testServer() *Server {
-	return &Server{db: &testDb{}, LookupAddr: lookupAddr, LookupPort: lookupPort}
+	return &Server{gr: &testDb{}, LookupAddr: lookupAddr, LookupPort: lookupPort}
 }
 
 func httpGet(url string, acceptMediaType string, userAgent string) (string, int, error) {
@@ -87,7 +87,7 @@ func TestDisabledHandlers(t *testing.T) {
 	server := testServer()
 	server.LookupPort = nil
 	server.LookupAddr = nil
-	server.db, _ = database.New("", "")
+	server.db, _ = geo.New("", "")
 	s := httptest.NewServer(server.Handler())
 
 	var tests = []struct {
diff --git a/iputil/database/database.go b/iputil/geo/geo.go
similarity index 94%
rename from iputil/database/database.go
rename to iputil/geo/geo.go
index c431312..4f1705f 100644
--- a/iputil/database/database.go
+++ b/iputil/geo/geo.go
@@ -1,4 +1,4 @@
-package database
+package geo
 
 import (
 	"net"
@@ -6,7 +6,7 @@ import (
 	geoip2 "github.com/oschwald/geoip2-golang"
 )
 
-type Client interface {
+type Reader interface {
 	Country(net.IP) (Country, error)
 	City(net.IP) (string, error)
 	IsEmpty() bool
@@ -22,7 +22,7 @@ type geoip struct {
 	city    *geoip2.Reader
 }
 
-func New(countryDB, cityDB string) (Client, error) {
+func Open(countryDB, cityDB string) (Reader, error) {
 	var country, city *geoip2.Reader
 	if countryDB != "" {
 		r, err := geoip2.Open(countryDB)