feat: more shelf functions

internal/api/shelves.go

@@ -19,6 +19,7 @@ func (s *APIServer) setupShelfRoutes() func(chi.Router) {
 			r.Use(s.ShelfCtx)
 			r.Get("/", makeHandler(s.getShelf))
 			r.Delete("/", makeHandler(s.deleteShelf))
+			r.Put("/", makeHandler(s.updateShelf))
 			r.Get("/count", makeHandler(s.getShelfAssetCount))
 		})
 	}
@@ -141,3 +142,35 @@ func (s *APIServer) getShelfAssetCount(w http.ResponseWriter, r *http.Request) e
 		Count: count,
 	})
 }
+
+func (s *APIServer) updateShelf(w http.ResponseWriter, r *http.Request) error {
+	ctx := r.Context()
+	eShelf, ok := ctx.Value("shelf").(*types.ShelfLocation)
+	if !ok {
+		return render.Render(w, r, errUnprocessable)
+	}
+
+	data := &types.CreateShelfRequest{}
+	if err := render.Bind(r, data); err != nil {
+		return render.Render(w, r, errBadRequest(err))
+	}
+
+	shelf := &types.ShelfLocation{
+		ID:         eShelf.ID,
+		Name:       data.Name,
+		BuildingID: data.BuildingID,
+		RoomNumber: data.RoomNumber,
+	}
+
+	err := s.db.UpdateShelfLocation(eShelf.ID, shelf)
+	if err != nil {
+		return err
+	}
+
+	return render.Render(w, r, &types.ShelfResponse{
+		Response: &types.Response{
+			HTTPStatusCode: http.StatusOK,
+		},
+		ShelfLocation: shelf,
+	})
+}

internal/storage/datastore.go

@@ -68,6 +68,11 @@ func (s *DataStore) CreateShelfLocation(shelf *types.ShelfLocation) error {
 	return result.Error
 }
 
+func (s *DataStore) UpdateShelfLocation(id uint64, update *types.ShelfLocation) error {
+	result := s.db.Model(&types.ShelfLocation{}).Where("id = ?", id).Select("*").Updates(update)
+	return result.Error
+}
+
 func (s *DataStore) GetAssetByID(id uint64) (*types.Asset, error) {
 	var result types.Asset
 	tx := s.db.Model(&types.Asset{}).Where("id = ?", id).Preload(clause.Associations).First(&result)