Add users services and jwt

This commit is contained in:
2024-05-04 16:37:19 +02:00
parent e127360ecf
commit d281c81dfa
7 changed files with 199 additions and 8 deletions
+106
View File
@@ -0,0 +1,106 @@
package main
import (
"net/http"
"github.com/golang-jwt/jwt"
"github.com/google/uuid"
"github.com/labstack/echo/v4"
"github.com/zoriya/vex"
)
type LoginDto struct {
Email string `json:"email" validate:"required"`
Password string `json:"password" validate:"required"`
}
type RegisterDto struct {
Name string `json:"name" validate:"required"`
Email string `json:"email" validate:"required"`
Password string `json:"password" validate:"required,max(60)"`
}
func (h *Handler) Login(c echo.Context) error {
var req LoginDto
err := c.Bind(&req)
if err != nil {
return echo.NewHTTPError(http.StatusBadRequest, err.Error())
}
if err = c.Validate(&req); err != nil {
return err
}
user := h.users.GetByEmail(req.Email)
if user == nil {
return echo.NewHTTPError(403, "Invalid email")
}
if !h.users.CheckPassword(req.Password, user.Password) {
return echo.NewHTTPError(403, "Invalid password")
}
return h.CreateToken(c, user)
}
func (h *Handler) CreateToken(c echo.Context, user *vex.User) error {
claims := &jwt.StandardClaims{
Subject: user.Id.String(),
}
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
t, err := token.SignedString(h.jwtSecret)
if err != nil {
return err
}
return c.JSON(http.StatusOK, echo.Map{
"token": t,
})
}
func (h *Handler) Register(c echo.Context) error {
var req RegisterDto
err := c.Bind(&req)
if err != nil {
return echo.NewHTTPError(http.StatusBadRequest, err.Error())
}
if err = c.Validate(&req); err != nil {
return err
}
user, err := h.users.Create(req.Name, req.Email, req.Password)
if err != nil {
return err
}
return h.CreateToken(c, &user)
}
func (h *Handler) GetMe(c echo.Context) error {
id, err := GetCurrentUserId(c)
if err != nil {
return err
}
user := h.users.GetById(id)
if user == nil {
return echo.NewHTTPError(500, "Internal server error")
}
return c.JSON(200, user)
}
func GetCurrentUserId(c echo.Context) (uuid.UUID, error) {
user := c.Get("user").(*jwt.Token)
if user == nil {
return uuid.UUID{}, echo.NewHTTPError(401, "Unauthorized")
}
claims := user.Claims.(*jwt.StandardClaims)
if claims == nil {
return uuid.UUID{}, echo.NewHTTPError(403, "Missing claims")
}
ret, err := uuid.Parse(claims.Subject)
if err != nil {
return uuid.UUID{}, echo.NewHTTPError(403, "Invalid id")
}
return ret, nil
}
func (h *Handler) RegisterLoginRoutes(e *echo.Echo, r *echo.Group) {
e.POST("/login", h.Login)
e.POST("/register", h.Register)
r.GET("/register", h.GetMe)
}