feat: allowing users to delete images from lists

backend/stacks/handler.go

@@ -13,11 +13,15 @@ import (
 
 	"github.com/charmbracelet/log"
 	"github.com/go-chi/chi/v5"
+	"github.com/google/uuid"
 )
 
 type StackHandler struct {
-	logger        *log.Logger
-	stackModel    models.ListModel
+	logger *log.Logger
+
+	imageModel models.ImageModel
+	stackModel models.ListModel
+
 	limitsManager limits.LimitsManagerMethods
 }
 
@@ -94,6 +98,47 @@ func (h *StackHandler) deleteStack(w http.ResponseWriter, r *http.Request) {
 	w.WriteHeader(http.StatusOK)
 }
 
+func (h *StackHandler) deleteImageFromStack(w http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+
+	stringImageID := chi.URLParam(r, "imageID")
+	stringListID := chi.URLParam(r, "listID")
+
+	imageID, err := uuid.Parse(stringImageID)
+	if err != nil {
+		w.WriteHeader(http.StatusBadRequest)
+		return
+	}
+
+	listID, err := uuid.Parse(stringListID)
+	if err != nil {
+		w.WriteHeader(http.StatusBadRequest)
+		return
+	}
+
+	// TODO: this should be extracted into a middleware of sorts
+	userID, err := middleware.GetUserID(ctx, h.logger, w)
+	if err != nil {
+		w.WriteHeader(http.StatusUnauthorized)
+		return
+	}
+
+	isAuthorized := h.imageModel.IsUserAuthorized(ctx, imageID, userID)
+	if !isAuthorized {
+		w.WriteHeader(http.StatusUnauthorized)
+		return
+	}
+
+	err = h.stackModel.DeleteImage(ctx, listID, imageID)
+	if err != nil {
+		h.logger.Warn("failed to delete image from list", "error", err)
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+
+	w.WriteHeader(http.StatusOK)
+}
+
 type CreateStackBody struct {
 	Title string `json:"title"`
 
@@ -149,15 +194,18 @@ func (h *StackHandler) CreateRoutes(r chi.Router) {
 		r.Post("/", middleware.WithLimit(h.logger, h.limitsManager.HasReachedStackLimit, middleware.WithValidatedPost(h.createStack)))
 		r.Patch("/{listID}", middleware.WithValidatedPost(h.editStack))
 		r.Delete("/{listID}", h.deleteStack)
+		r.Delete("/{listID}/{imageID}", h.deleteImageFromStack)
 	})
 }
 
 func CreateStackHandler(db *sql.DB, limitsManager limits.LimitsManagerMethods) StackHandler {
 	stackModel := models.NewListModel(db)
+	imageModel := models.NewImageModel(db)
 	logger := log.New(os.Stdout).WithPrefix("Stacks")
 
 	return StackHandler{
 		logger,
+		imageModel,
 		stackModel,
 		limitsManager,
 	}