mirror of
				https://github.com/zoriya/Kyoo.git
				synced 2025-11-04 03:27:14 -05:00 
			
		
		
		
	Create register
This commit is contained in:
		
							parent
							
								
									dfc411e5f6
								
							
						
					
					
						commit
						306dbbd024
					
				
							
								
								
									
										64
									
								
								auth/auth.go
									
									
									
									
									
								
							
							
						
						
									
										64
									
								
								auth/auth.go
									
									
									
									
									
								
							@ -1,64 +0,0 @@
 | 
				
			|||||||
package main
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
import (
 | 
					 | 
				
			||||||
	"context"
 | 
					 | 
				
			||||||
	"net/http"
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	"github.com/alexedwards/argon2id"
 | 
					 | 
				
			||||||
	"github.com/golang-jwt/jwt/v5"
 | 
					 | 
				
			||||||
	"github.com/labstack/echo/v4"
 | 
					 | 
				
			||||||
	"github.com/zoriya/kyoo/keibi/dbc"
 | 
					 | 
				
			||||||
)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
type LoginDto struct {
 | 
					 | 
				
			||||||
	Login    string `json:"login" validate:"required"`
 | 
					 | 
				
			||||||
	Password string `json:"password" validate:"required"`
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
type RegisterDto struct {
 | 
					 | 
				
			||||||
	Username string `json:"username" validate:"required"`
 | 
					 | 
				
			||||||
	Email    string `json:"email" validate:"required,email"`
 | 
					 | 
				
			||||||
	Password string `json:"password" validate:"required"`
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
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
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	pass, err := argon2id.CreateHash(req.Password, argon2id.DefaultParams)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return echo.NewHTTPError(400, "Invalid password")
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	duser, err := h.db.CreateUser(context.Background(), dbc.CreateUserParams{
 | 
					 | 
				
			||||||
		Username: req.Username,
 | 
					 | 
				
			||||||
		Email:    req.Email,
 | 
					 | 
				
			||||||
		Password: &pass,
 | 
					 | 
				
			||||||
		Claims:   h.config.DefaultClaims,
 | 
					 | 
				
			||||||
	})
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return echo.NewHTTPError(409, "Email or username already taken")
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	user := MapDbUser(&duser)
 | 
					 | 
				
			||||||
	return h.createToken(c, &user)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *Handler) createToken(c echo.Context, user *User) error {
 | 
					 | 
				
			||||||
	claims := &jwt.RegisteredClaims{
 | 
					 | 
				
			||||||
		Subject: user.ID.String(),
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
 | 
					 | 
				
			||||||
	t, err := token.SignedString(h.config.JwtSecret)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	return c.JSON(http.StatusOK, echo.Map{
 | 
					 | 
				
			||||||
		"token": t,
 | 
					 | 
				
			||||||
	})
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
@ -5,12 +5,14 @@ import (
 | 
				
			|||||||
	"crypto/rand"
 | 
						"crypto/rand"
 | 
				
			||||||
	"encoding/base64"
 | 
						"encoding/base64"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"github.com/golang-jwt/jwt/v5"
 | 
				
			||||||
	"github.com/zoriya/kyoo/keibi/dbc"
 | 
						"github.com/zoriya/kyoo/keibi/dbc"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type Configuration struct {
 | 
					type Configuration struct {
 | 
				
			||||||
	JwtSecret     []byte
 | 
						JwtSecret     []byte
 | 
				
			||||||
	DefaultClaims []byte
 | 
						Issuer        string
 | 
				
			||||||
 | 
						DefaultClaims jwt.MapClaims
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
const (
 | 
					const (
 | 
				
			||||||
 | 
				
			|||||||
@ -132,6 +132,7 @@ func main() {
 | 
				
			|||||||
	h.config = conf
 | 
						h.config = conf
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	e.GET("/users", h.ListUsers)
 | 
						e.GET("/users", h.ListUsers)
 | 
				
			||||||
 | 
						e.POST("/users", h.Register)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	e.Logger.Fatal(e.Start(":4568"))
 | 
						e.Logger.Fatal(e.Start(":4568"))
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
				
			|||||||
							
								
								
									
										39
									
								
								auth/session.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										39
									
								
								auth/session.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,39 @@
 | 
				
			|||||||
 | 
					package main
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"maps"
 | 
				
			||||||
 | 
						"net/http"
 | 
				
			||||||
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"github.com/golang-jwt/jwt/v5"
 | 
				
			||||||
 | 
						"github.com/labstack/echo/v4"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type LoginDto struct {
 | 
				
			||||||
 | 
						Login    string `json:"login" validate:"required"`
 | 
				
			||||||
 | 
						Password string `json:"password" validate:"required"`
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (h *Handler) createToken(c echo.Context, user *User) error {
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (h *Handler) CreateJwt(c echo.Context, user *User) error {
 | 
				
			||||||
 | 
						claims := maps.Clone(user.Claims)
 | 
				
			||||||
 | 
						claims["sub"] = user.ID.String()
 | 
				
			||||||
 | 
						claims["iss"] = h.config.Issuer
 | 
				
			||||||
 | 
						claims["exp"] = &jwt.NumericDate{
 | 
				
			||||||
 | 
							Time: time.Now().UTC().Add(time.Hour),
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						claims["iss"] = &jwt.NumericDate{
 | 
				
			||||||
 | 
							Time: time.Now().UTC(),
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
 | 
				
			||||||
 | 
						t, err := token.SignedString(h.config.JwtSecret)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return c.JSON(http.StatusOK, echo.Map{
 | 
				
			||||||
 | 
							"token": t,
 | 
				
			||||||
 | 
						})
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@ -19,5 +19,10 @@ sql:
 | 
				
			|||||||
          go_type:
 | 
					          go_type:
 | 
				
			||||||
            import: "github.com/google/uuid"
 | 
					            import: "github.com/google/uuid"
 | 
				
			||||||
            type: "UUID"
 | 
					            type: "UUID"
 | 
				
			||||||
 | 
					        - column: "users.claims"
 | 
				
			||||||
 | 
					          go_type:
 | 
				
			||||||
 | 
					            import: "github.com/golang-jwt/jwt/v5"
 | 
				
			||||||
 | 
					            package: "jwt"
 | 
				
			||||||
 | 
					            type: "MapClaims"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
				
			|||||||
@ -2,8 +2,11 @@ package main
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"context"
 | 
						"context"
 | 
				
			||||||
 | 
						"net/http"
 | 
				
			||||||
	"time"
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"github.com/alexedwards/argon2id"
 | 
				
			||||||
 | 
						"github.com/golang-jwt/jwt/v5"
 | 
				
			||||||
	"github.com/google/uuid"
 | 
						"github.com/google/uuid"
 | 
				
			||||||
	"github.com/labstack/echo/v4"
 | 
						"github.com/labstack/echo/v4"
 | 
				
			||||||
	"github.com/zoriya/kyoo/keibi/dbc"
 | 
						"github.com/zoriya/kyoo/keibi/dbc"
 | 
				
			||||||
@ -15,6 +18,7 @@ type User struct {
 | 
				
			|||||||
	Email       string                `json:"email"`
 | 
						Email       string                `json:"email"`
 | 
				
			||||||
	CreatedDate time.Time             `json:"createdDate"`
 | 
						CreatedDate time.Time             `json:"createdDate"`
 | 
				
			||||||
	LastSeen    time.Time             `json:"lastSeen"`
 | 
						LastSeen    time.Time             `json:"lastSeen"`
 | 
				
			||||||
 | 
						Claims      jwt.MapClaims         `json:"claims"`
 | 
				
			||||||
	Oidc        map[string]OidcHandle `json:"oidc,omitempty"`
 | 
						Oidc        map[string]OidcHandle `json:"oidc,omitempty"`
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@ -24,6 +28,12 @@ type OidcHandle struct {
 | 
				
			|||||||
	ProfileUrl *string `json:"profileUrl"`
 | 
						ProfileUrl *string `json:"profileUrl"`
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type RegisterDto struct {
 | 
				
			||||||
 | 
						Username string `json:"username" validate:"required"`
 | 
				
			||||||
 | 
						Email    string `json:"email" validate:"required,email"`
 | 
				
			||||||
 | 
						Password string `json:"password" validate:"required"`
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func MapDbUser(user *dbc.User) User {
 | 
					func MapDbUser(user *dbc.User) User {
 | 
				
			||||||
	return User{
 | 
						return User{
 | 
				
			||||||
		ID:          user.ID,
 | 
							ID:          user.ID,
 | 
				
			||||||
@ -31,6 +41,7 @@ func MapDbUser(user *dbc.User) User {
 | 
				
			|||||||
		Email:       user.Email,
 | 
							Email:       user.Email,
 | 
				
			||||||
		CreatedDate: user.CreatedDate,
 | 
							CreatedDate: user.CreatedDate,
 | 
				
			||||||
		LastSeen:    user.LastSeen,
 | 
							LastSeen:    user.LastSeen,
 | 
				
			||||||
 | 
							Claims:      user.Claims,
 | 
				
			||||||
		Oidc:        nil,
 | 
							Oidc:        nil,
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
@ -65,3 +76,31 @@ func (h *Handler) ListUsers(c echo.Context) error {
 | 
				
			|||||||
	}
 | 
						}
 | 
				
			||||||
	return c.JSON(200, ret)
 | 
						return c.JSON(200, ret)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					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
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						pass, err := argon2id.CreateHash(req.Password, argon2id.DefaultParams)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return echo.NewHTTPError(400, "Invalid password")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						duser, err := h.db.CreateUser(context.Background(), dbc.CreateUserParams{
 | 
				
			||||||
 | 
							Username: req.Username,
 | 
				
			||||||
 | 
							Email:    req.Email,
 | 
				
			||||||
 | 
							Password: &pass,
 | 
				
			||||||
 | 
							Claims:   h.config.DefaultClaims,
 | 
				
			||||||
 | 
						})
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return echo.NewHTTPError(409, "Email or username already taken")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						user := MapDbUser(&duser)
 | 
				
			||||||
 | 
						return h.createToken(c, &user)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
				
			|||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user