initial commit

2024-01-18 00:02:55 -06:00
commit 8b850f83ee
15 changed files with 952 additions and 0 deletions
--- a/internal/api/api.go
+++ b/internal/api/api.go
@@ -0,0 +1,77 @@
+package api
+
+import (
+	"encoding/json"
+	"log"
+	"net/http"
+
+	"git.brettb.xyz/goinv/server/internal/storage"
+	"git.brettb.xyz/goinv/server/internal/types"
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
+	"github.com/go-chi/render"
+)
+
+const API_VERSION_MAJOR = 0
+const API_VERSION_MINOR = 0
+const API_VERSION_PATCH = 1
+
+var API_VERSION = types.APIVersion{
+	Major: API_VERSION_MAJOR,
+	Minor: API_VERSION_MINOR,
+	Patch: API_VERSION_PATCH,
+}
+
+type APIFunc func(w http.ResponseWriter, r *http.Request) error
+
+type APIServer struct {
+	listenAddr string
+	db         *storage.DataStore
+}
+
+func NewAPIServer(database *storage.DataStore, listenAddr string) *APIServer {
+	return &APIServer{
+		listenAddr: listenAddr,
+		db:         database,
+	}
+}
+
+func (s *APIServer) Run() {
+	r := chi.NewRouter()
+
+	r.Use(middleware.Logger)
+	r.Use(render.SetContentType(render.ContentTypeJSON))
+
+	s.registerRoutes(r)
+
+	log.Printf("API Server listening on %s", s.listenAddr)
+
+	err := http.ListenAndServe(s.listenAddr, r)
+	if err != nil {
+		panic(err)
+	}
+}
+
+func (s *APIServer) registerRoutes(r *chi.Mux) {
+	r.Get("/", makeHandler(s.handleIndex))
+
+	r.Route("/assets", s.setupAssetRoutes())
+	r.Route("/shelves", s.setupShelfRoutes())
+}
+
+func (s *APIServer) handleIndex(w http.ResponseWriter, r *http.Request) error {
+	return writeJSON(w, http.StatusOK, types.IndexResponse{Version: API_VERSION})
+}
+
+func makeHandler(f APIFunc) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		if err := f(w, r); err != nil {
+			render.Render(w, r, errBadRequest(err))
+		}
+	}
+}
+
+func writeJSON(w http.ResponseWriter, s int, v any) error {
+	w.WriteHeader(s)
+	return json.NewEncoder(w).Encode(v)
+}
--- a/internal/api/assets.go
+++ b/internal/api/assets.go
@@ -0,0 +1,124 @@
+package api
+
+import (
+	"context"
+	"log"
+	"net/http"
+	"strconv"
+
+	"git.brettb.xyz/goinv/server/internal/types"
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/render"
+)
+
+func (s *APIServer) setupAssetRoutes() func(r chi.Router) {
+	return func(r chi.Router) {
+		r.Get("/", makeHandler(s.getAssets))
+		r.Post("/", makeHandler(s.createAsset))
+
+		r.Route("/{assetID}", func(r chi.Router) {
+			r.Use(s.AssetCtx)
+			r.Get("/", makeHandler(s.getAsset))
+			r.Delete("/", makeHandler(s.deleteAsset))
+		})
+	}
+}
+
+func (s *APIServer) AssetCtx(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		assetIdStr := chi.URLParam(r, "assetID")
+		assetId, err := strconv.Atoi(assetIdStr)
+		if err != nil {
+			render.Render(w, r, errNotFound)
+			return
+		}
+		asset, err := s.db.GetAssetByID(uint64(assetId))
+		if err != nil {
+			render.Render(w, r, errNotFound)
+			return
+		}
+		ctx := context.WithValue(r.Context(), "asset", asset)
+		next.ServeHTTP(w, r.WithContext(ctx))
+	})
+}
+
+func (s *APIServer) getAssets(w http.ResponseWriter, r *http.Request) error {
+	assets, err := s.db.GetAssets(0, 50) // TODO: Proper Pagination
+	if err != nil {
+		return err
+	}
+
+	total, err := s.db.TotalAssets()
+	if err != nil {
+		return err
+	}
+
+	return render.Render(w, r, &types.MultipleAssetsResponse{
+		Response: &types.Response{
+			HTTPStatusCode: http.StatusOK,
+		},
+		Assets: assets,
+		Total:  total,
+	})
+}
+
+func (s *APIServer) getAsset(w http.ResponseWriter, r *http.Request) error {
+	ctx := r.Context()
+	asset, ok := ctx.Value("asset").(*types.Asset)
+	if !ok {
+		return render.Render(w, r, errUnprocessable)
+	}
+
+	return render.Render(w, r, &types.AssetResponse{
+		Response: &types.Response{
+			HTTPStatusCode: http.StatusOK,
+		},
+		Asset: asset,
+	})
+}
+
+func (s *APIServer) deleteAsset(w http.ResponseWriter, r *http.Request) error {
+	ctx := r.Context()
+	asset, ok := ctx.Value("asset").(*types.Asset)
+	if !ok {
+		return render.Render(w, r, errUnprocessable)
+	}
+
+	if ok, _ := s.db.DeleteAssetByID(asset.ID); !ok {
+		return render.Render(w, r, errUnprocessable)
+	}
+
+	return render.Render(w, r, &types.AssetResponse{Response: &types.Response{HTTPStatusCode: http.StatusOK}, Asset: asset})
+}
+
+func (s *APIServer) createAsset(w http.ResponseWriter, r *http.Request) error {
+	data := &types.CreateAssetRequest{}
+	if err := render.Bind(r, data); err != nil {
+		log.Printf("ERR: %v\n", err)
+		return render.Render(w, r, errBadRequest(err))
+	}
+
+	asset := &types.Asset{
+		Name:            data.Name,
+		Quantity:        data.Quantity,
+		Length:          data.Length,
+		Manufacturer:    data.Manufacturer,
+		ModelName:       data.ModelName,
+		Price:           data.Price,
+		Comments:        data.Comments,
+		ShelfLocationID: data.ShelfLocationID,
+		CategoryID:      data.CategoryID,
+	}
+
+	err := s.db.CreateAsset(asset)
+	if err != nil {
+		return err
+	}
+
+	return render.Render(w, r, &types.AssetResponse{
+		Response: &types.Response{
+			HTTPStatusCode: http.StatusOK,
+		},
+		Asset: asset,
+	})
+}
--- a/internal/api/shelves.go
+++ b/internal/api/shelves.go
@@ -0,0 +1,96 @@
+package api
+
+import (
+	"context"
+	"net/http"
+	"strconv"
+
+	"git.brettb.xyz/goinv/server/internal/types"
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/render"
+)
+
+func (s *APIServer) setupShelfRoutes() func(chi.Router) {
+	return func(r chi.Router) {
+		r.Get("/", makeHandler(s.getShelves))
+		//r.Post("/", makeHandler(s.createShelf))
+
+		r.Route("/{shelfID}", func(r chi.Router) {
+			r.Use(s.ShelfCtx)
+			r.Get("/", makeHandler(s.getShelf))
+			r.Delete("/", makeHandler(s.deleteShelf))
+		})
+	}
+}
+
+func (s *APIServer) ShelfCtx(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		shelfIdStr := chi.URLParam(r, "shelfID")
+		shelfId, err := strconv.ParseUint(shelfIdStr, 10, 64)
+		if err != nil {
+			render.Render(w, r, errNotFound)
+			return
+		}
+		shelf, err := s.db.GetShelfByID(shelfId)
+		if err != nil {
+			render.Render(w, r, errNotFound)
+			return
+		}
+		ctx := context.WithValue(r.Context(), "shelf", shelf)
+		next.ServeHTTP(w, r.WithContext(ctx))
+	})
+}
+
+func (s *APIServer) getShelves(w http.ResponseWriter, r *http.Request) error {
+	shelves, err := s.db.GetShelves(0, 10)
+	if err != nil {
+		return err
+	}
+
+	total, err := s.db.TotalShelves()
+	if err != nil {
+		return err
+	}
+
+	return render.Render(w, r, &types.MultipleShelfResponse{
+		Response: &types.Response{
+			HTTPStatusCode: http.StatusOK,
+		},
+		ShelfLocations: shelves,
+		Total:          total,
+	})
+}
+
+func (s *APIServer) getShelf(w http.ResponseWriter, r *http.Request) error {
+	ctx := r.Context()
+	shelf, ok := ctx.Value("shelf").(*types.ShelfLocation)
+	if !ok {
+		return render.Render(w, r, errUnprocessable)
+	}
+
+	return render.Render(w, r, &types.ShelfResponse{
+		Response: &types.Response{
+			HTTPStatusCode: http.StatusOK,
+		},
+		ShelfLocation: shelf,
+	})
+}
+
+func (s *APIServer) deleteShelf(w http.ResponseWriter, r *http.Request) error {
+	ctx := r.Context()
+	shelf, ok := ctx.Value("shelf").(*types.ShelfLocation)
+	if !ok {
+		return render.Render(w, r, errUnprocessable)
+	}
+
+	if ok, _ := s.db.DeleteShelfByID(shelf.ID); !ok {
+		return render.Render(w, r, errUnprocessable)
+	}
+
+	return render.Render(w, r, &types.ShelfResponse{
+		Response: &types.Response{
+			HTTPStatusCode: http.StatusOK,
+		},
+		ShelfLocation: shelf,
+	})
+}
--- a/internal/api/types.go
+++ b/internal/api/types.go
@@ -0,0 +1,52 @@
+package api
+
+import (
+	"net/http"
+
+	"git.brettb.xyz/goinv/server/internal/types"
+	"github.com/go-chi/render"
+)
+
+var errNotFound = &types.APIError{
+	Response: &types.Response{
+		HTTPStatusCode: http.StatusNotFound,
+	},
+	Messages: []string{"resource not found"},
+}
+
+var errUnprocessable = &types.APIError{
+	Response: &types.Response{
+		HTTPStatusCode: http.StatusUnprocessableEntity,
+	},
+	Messages: []string{"unable to process"},
+}
+
+func errBadRequest(err error) render.Renderer {
+	return &types.APIError{
+		Response: &types.Response{
+			HTTPStatusCode: http.StatusBadRequest,
+		},
+		Err:      err,
+		Messages: []string{"bad request"},
+	}
+}
+
+func errRender(err error) render.Renderer {
+	return &types.APIError{
+		Response: &types.Response{
+			HTTPStatusCode: http.StatusUnprocessableEntity,
+		},
+		Err:      err,
+		Messages: []string{"error rendering response"},
+	}
+}
+
+func errUnauthorized(err error) render.Renderer {
+	return &types.APIError{
+		Response: &types.Response{
+			HTTPStatusCode: http.StatusUnauthorized,
+		},
+		Err:      err,
+		Messages: []string{"unauthorized"},
+	}
+}