Allow custom placeholders (#464)

Author: reaper47

.idea/inspectionProfiles/Project_Default.xml

@@ -9,6 +9,7 @@ import (
 	"github.com/briandowns/spinner"
 	"github.com/disintegration/imaging"
 	"github.com/gen2brain/webp"
+	"github.com/reaper47/recipya/web"
 	"image/jpeg"
 	"io"
 	"net"
@@ -97,6 +98,44 @@ func moveFileStructure() {
 	if count == len(dirs) {
 		_ = os.RemoveAll(filepath.Join(exeDir, "data"))
 	}
+
+	// Move placeholder.webp to images dir
+	placeholdersDir := filepath.Join(ImagesDir, "Placeholders")
+	err = os.MkdirAll(placeholdersDir, os.ModePerm)
+	if err != nil {
+		panic(err)
+	}
+	
+	_, err = os.Stat(filepath.Join(placeholdersDir, "placeholder.recipe.webp"))
+	if os.IsNotExist(err) {
+		openFile, err := web.StaticFS.Open("static/img/recipes/placeholder.webp")
+		if err != nil {
+			return
+		}
+		defer openFile.Close()
+
+		fRecipe, err := os.Create(filepath.Join(placeholdersDir, "placeholder.recipe.webp"))
+		if err != nil {
+			panic(err)
+		}
+		defer fRecipe.Close()
+
+		io.Copy(fRecipe, openFile)
+
+		openFileCookbook, err := web.StaticFS.Open("static/img/cookbooks-new/placeholder.webp")
+		if err != nil {
+			panic(err)
+		}
+		defer openFileCookbook.Close()
+
+		fCookbook, err := os.Create(filepath.Join(placeholdersDir, "placeholder.cookbook.webp"))
+		if err != nil {
+			panic(err)
+		}
+		defer fCookbook.Close()
+
+		io.Copy(fCookbook, openFileCookbook)
+	}
 }
 
 func moveFiles(srcDir, destDir string) error {

internal/app/setup.go

@@ -2,6 +2,7 @@ package server
 
 import (
 	"errors"
+	"github.com/reaper47/recipya/web"
 	"io"
 	"log/slog"
 	"net/http"
@@ -80,6 +81,114 @@ func notFoundHandler(w http.ResponseWriter, r *http.Request) {
 	_ = components.SimplePage("Page Not Found", "The page you requested to view is not found. Please go back to the main page.").Render(r.Context(), w)
 }
 
+func (s *Server) placeholderPostHandler() http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		userID := getUserID(r)
+		userIDAttr := slog.Int64("userID", userID)
+
+		name := r.FormValue("name")
+		if name != "recipe" && name != "cookbook" {
+			s.Brokers.SendToast(models.NewErrorFormToast("Only the recipe or cookbook placeholder can be updated."), userID)
+			w.WriteHeader(http.StatusBadRequest)
+			return
+		}
+
+		r.Body = http.MaxBytesReader(w, r.Body, 1<<24)
+
+		err := r.ParseMultipartForm(1 << 24)
+		if err != nil {
+			msg := "Could not parse the form."
+			slog.Error(msg, userIDAttr, "error", err)
+			s.Brokers.SendToast(models.NewErrorFormToast(msg), userID)
+			w.WriteHeader(http.StatusBadRequest)
+			return
+		}
+
+		images, ok := r.MultipartForm.File["images"]
+		if !ok {
+			msg := "Could not retrieve the image from the form."
+			slog.Error(msg, userIDAttr, "error", err)
+			s.Brokers.SendToast(models.NewErrorFormToast(msg), userID)
+			w.WriteHeader(http.StatusBadRequest)
+			return
+		}
+
+		if len(images) == 0 {
+			msg := "No image has been uploaded."
+			slog.Error(msg, userIDAttr, "error", err)
+			s.Brokers.SendToast(models.NewErrorFormToast(msg), userID)
+			w.WriteHeader(http.StatusBadRequest)
+			return
+		}
+
+		f, err := images[0].Open()
+		if err != nil {
+			msg := "Could not open the image from the form."
+			slog.Error(msg, "error", err, userIDAttr)
+			s.Brokers.SendToast(models.NewErrorFormToast(msg), userID)
+			w.WriteHeader(http.StatusBadRequest)
+			return
+		}
+		defer f.Close()
+
+		imageUUID, err := s.Files.UploadImage(f)
+		if err != nil {
+			msg := "Error uploading image."
+			slog.Error(msg, "error", err, userIDAttr)
+			s.Brokers.SendToast(models.NewErrorFilesToast(msg), userID)
+			w.WriteHeader(http.StatusInternalServerError)
+			return
+		}
+		imageUUIDAttr := slog.String("imageUUID", imageUUID.String())
+
+		from := filepath.Join(app.ImagesDir, imageUUID.String()+app.ImageExt)
+		to := filepath.Join(app.ImagesDir, "Placeholders", "placeholder."+name+app.ImageExt)
+		err = os.Rename(from, to)
+		if err != nil {
+			msg := "Error moving compressed placeholder image."
+			slog.Error(msg, "error", err, imageUUIDAttr, userIDAttr)
+			s.Brokers.SendToast(models.NewErrorFilesToast(msg), userID)
+			w.WriteHeader(http.StatusInternalServerError)
+			return
+		}
+	})
+}
+
+func (s *Server) restorePlaceholderPostHandler() http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		name := "recipe"
+		path := "static/img/recipes/placeholder.webp"
+		if r.FormValue("name") == "cookbook" {
+			name = "cookbook"
+			path = "static/img/cookbooks-new/placeholder.webp"
+		}
+
+		userID := getUserID(r)
+		userIDAttr := slog.Int64("userID", userID)
+
+		openFile, err := web.StaticFS.Open(path)
+		if err != nil {
+			msg := "Error opening the public file."
+			s.Brokers.SendToast(models.NewErrorGeneralToast(msg), userID)
+			slog.Error(msg, "error", err, userIDAttr)
+			return
+		}
+		defer openFile.Close()
+
+		f, err := os.Create(filepath.Join(app.ImagesDir, "Placeholders", "placeholder."+name+".webp"))
+		if err != nil {
+			msg := "Error creating placeholder file."
+			s.Brokers.SendToast(models.NewErrorGeneralToast(msg), userID)
+			slog.Error(msg, "error", err, userIDAttr)
+			w.WriteHeader(http.StatusInternalServerError)
+			return
+		}
+		defer f.Close()
+
+		io.Copy(f, openFile)
+	}
+}
+
 func (s *Server) userInitialsHandler() http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
 		_, _ = w.Write([]byte(s.Repository.UserInitials(getUserID(r))))

internal/server/handlers_cookbooks_test.go

@@ -155,6 +155,16 @@ func TestHandlers_General_Index(t *testing.T) {
 	})
 }
 
+func TestHandlers_General_Placeholder(t *testing.T) {
+	srv, ts, c := createWSServer()
+	defer c.CloseNow()
+
+	t.Run("must be logged in", func(t *testing.T) {
+		assertMustBeLoggedIn(t, srv, http.MethodPost, ts.URL+"/placeholder")
+		assertMustBeLoggedIn(t, srv, http.MethodPost, ts.URL+"/placeholder/restore")
+	})
+}
+
 func TestHandlers_General_Update(t *testing.T) {
 	srv, ts, c := createWSServer()
 	defer c.CloseNow()