mirror of
https://github.com/zoriya/vex.git
synced 2026-05-27 08:13:23 +00:00
Add users services and jwt
This commit is contained in:
@@ -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)
|
||||
}
|
||||
Reference in New Issue
Block a user