country-geo-locations/api/v1/server.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}
}