hugo-cms/handlers/site.go
2025-01-27 21:48:40 +11:00

70 lines
1.3 KiB
Go

package handlers
import (
"context"
"fmt"
"github.com/gofiber/fiber/v2"
"lmika.dev/lmika/hugo-cms/services/sites"
"time"
)
type Site struct {
Site *sites.Service
}
func (s *Site) Create() fiber.Handler {
return func(c *fiber.Ctx) error {
site, err := s.Site.CreateSite(c.UserContext(), "New Site "+time.Now().Format("2006-01-02 15:04:05"))
if err != nil {
return err
}
return c.Redirect(fmt.Sprintf("/sites/%v/posts", site.ID))
}
}
func (s *Site) Show() fiber.Handler {
return func(c *fiber.Ctx) error {
id, err := c.ParamsInt("siteId")
if err != nil {
return err
}
site, err := s.Site.GetSite(c.UserContext(), id)
if err != nil {
return err
}
return c.Render("sites/index", fiber.Map{
"site": site,
}, "layouts/main")
}
}
func (s *Site) Rebuild() fiber.Handler {
return func(c *fiber.Ctx) error {
if err := s.Site.Rebuild(c.UserContext(), GetSite(c)); err != nil {
return err
}
return c.Redirect(fmt.Sprintf("/sites/%v/posts", GetSite(c).ID))
}
}
func (s *Site) WithSite() fiber.Handler {
return func(c *fiber.Ctx) error {
id, err := c.ParamsInt("siteId")
if err != nil {
return err
}
site, err := s.Site.GetSite(c.UserContext(), id)
if err != nil {
return err
}
c.SetUserContext(context.WithValue(c.UserContext(), siteKey, site))
return c.Next()
}
}