hugo-cms/handlers/post.go

package handlers

import (
	"errors"
	"fmt"
	"github.com/davecgh/go-spew/spew"
	"github.com/gofiber/fiber/v3"
	"lmika.dev/lmika/hugo-cms/models"
	"lmika.dev/lmika/hugo-cms/services/posts"
	"net/http"
	"strings"
)

type Post struct {
	Post *posts.Service
}

func (h *Post) Posts(c fiber.Ctx) error {
	site := GetSite(c)

	posts, err := h.Post.ListPostOfSite(c.Context(), site)
	if err != nil {
		return err
	}

	return c.Render("posts/index", fiber.Map{
		"posts": posts,
	}, "layouts/site")
}

func (h *Post) New(c fiber.Ctx) error {
	return c.Render("posts/new", fiber.Map{
		"post": models.Post{},
	}, "layouts/site")
}

func (h *Post) Create(c fiber.Ctx) error {
	site := GetSite(c)

	req, err := h.parseReqToNewPost(c)
	if err != nil {
		return err
	}

	_, err = h.Post.Create(c.Context(), site, req)
	if err != nil {
		return err
	}

	return c.Redirect().To(fmt.Sprintf("/sites/%v/posts", site.ID))
}

func (h *Post) NewLinkPost(c fiber.Ctx) error {
	return c.Render("posts/new-link", fiber.Map{}, "layouts/site")
}

func (h *Post) CreateLinkPost(c fiber.Ctx) error {
	site := GetSite(c)

	var req posts.NewLinkPost
	if err := c.Bind().Body(&req); err != nil {
		return err
	}

	post, err := h.Post.CreateLinkPost(c.Context(), site, req)
	if err != nil {
		return err
	}

	return c.Render("posts/new", fiber.Map{
		"post": post,
	}, "layouts/site")
}

func (h *Post) Edit(c fiber.Ctx) error {
	site := GetSite(c)

	postID := fiber.Params[int](c, "postId")
	if postID == 0 {
		return errors.New("postId is required")
	}

	post, err := h.Post.GetPost(c.Context(), postID)
	if err != nil {
		return err
	} else if post.SiteID != site.ID {
		return fmt.Errorf("post id %v not equal to site id %v", postID, site.ID)
	}

	return c.Render("posts/new", fiber.Map{
		"post": post,
	}, "layouts/site")
}

func (h *Post) Update(c fiber.Ctx) error {
	site := GetSite(c)

	postID := fiber.Params[int](c, "postId")
	if postID == 0 {
		return errors.New("postId is required")
	}

	req, err := h.parseReqToNewPost(c)
	if err != nil {
		return err
	}

	spew.Dump(req)

	post, err := h.Post.GetPost(c.Context(), postID)
	if err != nil {
		return err
	} else if post.SiteID != site.ID {
		return fmt.Errorf("post id %v not equal to site id %v", postID, site.ID)
	}

	post.Title = req.Title
	post.Body = req.Body

	if err := h.Post.Save(c.Context(), site, &post); err != nil {
		return err
	}

	return c.Redirect().To(fmt.Sprintf("/sites/%v/posts", site.ID))
}

func (h *Post) Delete(c fiber.Ctx) error {
	site := GetSite(c)

	postID := fiber.Params[int](c, "postId")
	if postID == 0 {
		return errors.New("postId is required")
	}

	if err := h.Post.DeletePost(c.Context(), site, postID); err != nil {
		return err
	}

	return Select(c,
		HTMX(func(c fiber.Ctx) error {
			return c.Status(http.StatusOK).SendString("")
		}),
		Otherwise(func(c fiber.Ctx) error {
			return c.Redirect().To(fmt.Sprintf("/sites/%v/posts", site.ID))
		}),
	)
}

func (h *Post) parseReqToNewPost(c fiber.Ctx) (req posts.NewPost, err error) {
	reqMap := make(map[string]string)
	if err := c.Bind().Body(&reqMap); err != nil {
		return req, err
	}

	for k, v := range reqMap {
		switch {
		case k == "title":
			req.Title = v
		case k == "body":
			req.Body = v
		case strings.HasPrefix(k, "params."):
			kk := strings.TrimPrefix(k, "params.")
			if req.Params == nil {
				req.Params = make(map[string]string)
			}
			req.Params[kk] = v
		}
	}
	return req, nil
}