country-geo-locations/api/v1/server.go

122 lines
2.6 KiB
Go

package api_v1
import (
"context"
"net/http"
"time"
"git.ar21.de/yolokube/country-geo-locations/pkg/geoloc"
"github.com/go-chi/chi/v5"
"github.com/go-chi/render"
)
func ApiRouter() chi.Router {
r := chi.NewRouter()
r.Use(render.SetContentType(render.ContentTypeJSON))
r.Get("/ping", func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("pong"))
})
r.Route("/location", func(r chi.Router) {
r.Route("/{ipAddress}", func(r chi.Router) {
r.Use(searchIPCtx)
r.Get("/", searchIP)
})
})
return r
}
func searchIP(w http.ResponseWriter, r *http.Request) {
ipinfo := r.Context().Value(keyIPInfo).(*geoloc.IPInfo)
if err := render.Render(w, r, newIPInfoResponse(ipinfo)); err != nil {
render.Render(w, r, errRender(err))
return
}
}
func searchIPCtx(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
var ipinfo *geoloc.IPInfo
var ipnetnum uint
var err error
if ipAddress := chi.URLParam(r, "ipAddress"); ipAddress != "" {
ipnetnum, err = geoloc.GetIPInfo(ipAddress)
if err != nil {
render.Render(w, r, errInvalidRequest(err))
return
}
newipnet, found := geoloc.GetCacheContent(ipnetnum)
if found {
ipnetnum = newipnet
}
ipinfo, err = geoloc.SearchIPNet(ipnetnum)
if err != nil {
render.Render(w, r, errNotFound)
return
}
if !found {
geoloc.SetCacheContent(ipnetnum, ipinfo.IPNumFrom, 2*time.Minute)
}
}
ctx := context.WithValue(r.Context(), keyIPInfo, ipinfo)
next.ServeHTTP(w, r.WithContext(ctx))
})
}
type errResponse struct {
Err error `json:"-"`
HTTPStatusCode int `json:"-"`
StatusText string `json:"status"`
AppCode int64 `json:"code,omitempty"`
ErrorText string `json:"error,omitempty"`
}
func (e *errResponse) Render(w http.ResponseWriter, r *http.Request) error {
render.Status(r, e.HTTPStatusCode)
return nil
}
func errInvalidRequest(err error) render.Renderer {
return &errResponse{
Err: err,
HTTPStatusCode: 400,
StatusText: "Invalid request",
ErrorText: err.Error(),
}
}
func errRender(err error) render.Renderer {
return &errResponse{
Err: err,
HTTPStatusCode: 422,
StatusText: "Error rendering response",
ErrorText: err.Error(),
}
}
var errNotFound = &errResponse{HTTPStatusCode: 404, StatusText: "Resource not found"}
type key int
const keyIPInfo key = iota
type ipInfoResponse struct {
*geoloc.IPInfo
}
func (ir *ipInfoResponse) Render(w http.ResponseWriter, r *http.Request) error {
return nil
}
func newIPInfoResponse(ipinfo *geoloc.IPInfo) *ipInfoResponse {
return &ipInfoResponse{IPInfo: ipinfo}
}