Added user authentication
This commit is contained in:
parent
d7e7af5a10
commit
cb54057305
40 changed files with 710 additions and 218 deletions
69
handlers/auth.go
Normal file
69
handlers/auth.go
Normal file
|
|
@ -0,0 +1,69 @@
|
|||
package handlers
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"github.com/gofiber/fiber/v2"
|
||||
"lmika.dev/lmika/hugo-cms/models"
|
||||
"lmika.dev/lmika/hugo-cms/services/users"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
type AuthHandler struct {
|
||||
UserService *users.Service
|
||||
}
|
||||
|
||||
func (h *AuthHandler) ShowLogin(c *fiber.Ctx) error {
|
||||
return c.Render("auth/login", fiber.Map{}, "layouts/login")
|
||||
}
|
||||
|
||||
func (h *AuthHandler) Login(c *fiber.Ctx) error {
|
||||
var req struct {
|
||||
Email string `form:"email"`
|
||||
Password string `form:"password"`
|
||||
}
|
||||
|
||||
if err := c.BodyParser(&req); err != nil {
|
||||
return errors.New("invalid email or password")
|
||||
}
|
||||
|
||||
user, err := h.UserService.VerifyLogin(c.UserContext(), req.Email, req.Password)
|
||||
if err != nil {
|
||||
return errors.New("invalid email or password")
|
||||
}
|
||||
|
||||
bts, err := json.Marshal(models.AuthCookie{UserID: user.ID})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
c.Cookie(&fiber.Cookie{
|
||||
Name: models.AuthCookieName,
|
||||
Value: string(bts),
|
||||
})
|
||||
return c.Redirect("/", http.StatusFound)
|
||||
}
|
||||
|
||||
func (h *AuthHandler) RequireAuth(c *fiber.Ctx) error {
|
||||
user, err := h.readAuthCookie(c)
|
||||
if err != nil {
|
||||
return c.Redirect("/auth/login", http.StatusFound)
|
||||
}
|
||||
|
||||
c.Locals("user", user)
|
||||
return c.Next()
|
||||
}
|
||||
|
||||
func (h *AuthHandler) readAuthCookie(c *fiber.Ctx) (user models.User, err error) {
|
||||
authData := c.Cookies(models.AuthCookieName)
|
||||
if authData == "" {
|
||||
return models.User{}, errors.New("no auth cookie")
|
||||
}
|
||||
|
||||
var ac models.AuthCookie
|
||||
if err := json.Unmarshal([]byte(authData), &ac); err != nil {
|
||||
return models.User{}, err
|
||||
}
|
||||
|
||||
return h.UserService.GetUserByID(c.UserContext(), ac.UserID)
|
||||
}
|
||||
|
|
@ -1,14 +1,23 @@
|
|||
package handlers
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"github.com/gofiber/fiber/v2"
|
||||
"lmika.dev/lmika/hugo-cms/models"
|
||||
)
|
||||
|
||||
type siteKeyType struct{}
|
||||
|
||||
var siteKey siteKeyType
|
||||
func GetUser(c *fiber.Ctx) models.User {
|
||||
u, ok := c.Locals("user").(models.User)
|
||||
if !ok {
|
||||
panic(errors.New("user not found in context"))
|
||||
}
|
||||
return u
|
||||
}
|
||||
|
||||
func GetSite(c *fiber.Ctx) models.Site {
|
||||
return c.UserContext().Value(siteKey).(models.Site)
|
||||
s, ok := c.Locals("site").(models.Site)
|
||||
if !ok {
|
||||
panic(errors.New("no site in context"))
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
|
|
|||
37
handlers/mimeselectors.go
Normal file
37
handlers/mimeselectors.go
Normal file
|
|
@ -0,0 +1,37 @@
|
|||
package handlers
|
||||
|
||||
import "github.com/gofiber/fiber/v2"
|
||||
|
||||
type mimeTypeHandler interface {
|
||||
CanHandle(c *fiber.Ctx) bool
|
||||
Handle(c *fiber.Ctx) error
|
||||
}
|
||||
|
||||
type HTMX func(c *fiber.Ctx) error
|
||||
|
||||
func (h HTMX) CanHandle(c *fiber.Ctx) bool {
|
||||
return c.Get("Hx-request") == "true"
|
||||
}
|
||||
|
||||
func (h HTMX) Handle(c *fiber.Ctx) error {
|
||||
return h(c)
|
||||
}
|
||||
|
||||
type Otherwise func(c *fiber.Ctx) error
|
||||
|
||||
func (h Otherwise) CanHandle(c *fiber.Ctx) bool {
|
||||
return true
|
||||
}
|
||||
|
||||
func (h Otherwise) Handle(c *fiber.Ctx) error {
|
||||
return h(c)
|
||||
}
|
||||
|
||||
func Select(c *fiber.Ctx, mimeTypes ...mimeTypeHandler) error {
|
||||
for _, mt := range mimeTypes {
|
||||
if mt.CanHandle(c) {
|
||||
return mt.Handle(c)
|
||||
}
|
||||
}
|
||||
return c.Status(fiber.StatusInternalServerError).SendString("cant handle response")
|
||||
}
|
||||
186
handlers/post.go
186
handlers/post.go
|
|
@ -5,121 +5,125 @@ import (
|
|||
"github.com/gofiber/fiber/v2"
|
||||
"lmika.dev/lmika/hugo-cms/models"
|
||||
"lmika.dev/lmika/hugo-cms/services/posts"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
type Post struct {
|
||||
Post *posts.Service
|
||||
}
|
||||
|
||||
func (h *Post) Posts() fiber.Handler {
|
||||
return func(c *fiber.Ctx) error {
|
||||
site := GetSite(c)
|
||||
func (h *Post) Posts(c *fiber.Ctx) error {
|
||||
site := GetSite(c)
|
||||
|
||||
posts, err := h.Post.ListPostOfSite(c.UserContext(), site)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return c.Render("posts/index", fiber.Map{
|
||||
"site": site,
|
||||
"posts": posts,
|
||||
}, "layouts/site")
|
||||
posts, err := h.Post.ListPostOfSite(c.UserContext(), site)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return c.Render("posts/index", fiber.Map{
|
||||
"posts": posts,
|
||||
}, "layouts/site")
|
||||
}
|
||||
|
||||
func (h *Post) New() fiber.Handler {
|
||||
return func(c *fiber.Ctx) error {
|
||||
site := GetSite(c)
|
||||
|
||||
return c.Render("posts/new", fiber.Map{
|
||||
"site": site,
|
||||
"post": models.Post{},
|
||||
}, "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() fiber.Handler {
|
||||
type Req struct {
|
||||
func (h *Post) Create(c *fiber.Ctx) error {
|
||||
site := GetSite(c)
|
||||
|
||||
var req struct {
|
||||
Title string `json:"title" form:"title"`
|
||||
Body string `json:"body" form:"body"`
|
||||
}
|
||||
|
||||
return func(c *fiber.Ctx) error {
|
||||
site := GetSite(c)
|
||||
|
||||
var req Req
|
||||
if err := c.BodyParser(&req); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, err := h.Post.Create(c.UserContext(), site, posts.NewPost{
|
||||
Title: req.Title,
|
||||
Body: req.Body,
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return c.Redirect(fmt.Sprintf("/sites/%v/posts", site.ID))
|
||||
if err := c.BodyParser(&req); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, err := h.Post.Create(c.UserContext(), site, posts.NewPost{
|
||||
Title: req.Title,
|
||||
Body: req.Body,
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return c.Redirect(fmt.Sprintf("/sites/%v/posts", site.ID))
|
||||
}
|
||||
|
||||
func (h *Post) Edit() fiber.Handler {
|
||||
return func(c *fiber.Ctx) error {
|
||||
site := GetSite(c)
|
||||
func (h *Post) Edit(c *fiber.Ctx) error {
|
||||
site := GetSite(c)
|
||||
|
||||
postID, err := c.ParamsInt("postId")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
post, err := h.Post.GetPost(c.UserContext(), 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{
|
||||
"site": site,
|
||||
"post": post,
|
||||
}, "layouts/site")
|
||||
postID, err := c.ParamsInt("postId")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
post, err := h.Post.GetPost(c.UserContext(), 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() fiber.Handler {
|
||||
type Req struct {
|
||||
func (h *Post) Update(c *fiber.Ctx) error {
|
||||
site := GetSite(c)
|
||||
|
||||
postID, err := c.ParamsInt("postId")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var req struct {
|
||||
Title string `json:"title" form:"title"`
|
||||
Body string `json:"body" form:"body"`
|
||||
}
|
||||
|
||||
return func(c *fiber.Ctx) error {
|
||||
site := GetSite(c)
|
||||
|
||||
postID, err := c.ParamsInt("postId")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var req Req
|
||||
if err := c.BodyParser(&req); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
post, err := h.Post.GetPost(c.UserContext(), 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.UserContext(), site, &post); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return c.Redirect(fmt.Sprintf("/sites/%v/posts", site.ID))
|
||||
if err := c.BodyParser(&req); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
post, err := h.Post.GetPost(c.UserContext(), 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.UserContext(), site, &post); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return c.Redirect(fmt.Sprintf("/sites/%v/posts", site.ID))
|
||||
}
|
||||
|
||||
func (h *Post) Delete(c *fiber.Ctx) error {
|
||||
site := GetSite(c)
|
||||
|
||||
postID, err := c.ParamsInt("postId")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err := h.Post.DeletePost(c.UserContext(), 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(fmt.Sprintf("/sites/%v/posts", site.ID), http.StatusSeeOther)
|
||||
}),
|
||||
)
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,7 +1,6 @@
|
|||
package handlers
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"github.com/gofiber/fiber/v2"
|
||||
"lmika.dev/lmika/hugo-cms/services/sites"
|
||||
|
|
@ -12,43 +11,39 @@ 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
|
||||
}
|
||||
func (s *Site) Create(c *fiber.Ctx) error {
|
||||
user := GetUser(c)
|
||||
|
||||
return c.Redirect(fmt.Sprintf("/sites/%v/posts", site.ID))
|
||||
site, err := s.Site.CreateSite(c.UserContext(), user, "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) Show(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) Rebuild(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 {
|
||||
|
|
@ -63,7 +58,7 @@ func (s *Site) WithSite() fiber.Handler {
|
|||
return err
|
||||
}
|
||||
|
||||
c.SetUserContext(context.WithValue(c.UserContext(), siteKey, site))
|
||||
c.Locals("site", site)
|
||||
return c.Next()
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue