feat: Add user register

Author: devkcud

cmd/api/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"github.com/foss-opensolace/api.opensolace.com/internal/api/service"
 	"github.com/foss-opensolace/api.opensolace.com/internal/app"
 	"github.com/foss-opensolace/api.opensolace.com/internal/config"
 	"github.com/foss-opensolace/api.opensolace.com/internal/db"
@@ -9,5 +10,6 @@ import (
 func main() {
 	config.New()
 	db.New()
+	service.New()
 	app.New()
 }

internal/api/controller/router.go

@@ -21,4 +21,5 @@ func (c *controller) v1() {
 	group := c.instance.Group("/v1")
 
 	v1.NewMetricRouter(group)
+	v1.NewAuthRouter(group)
 }

internal/api/controller/v1/auth.go

@@ -0,0 +1,53 @@
+package v1
+
+import (
+	"errors"
+
+	"github.com/foss-opensolace/api.opensolace.com/internal/api/model/dto"
+	"github.com/foss-opensolace/api.opensolace.com/internal/api/service"
+	"github.com/foss-opensolace/api.opensolace.com/pkg/jwt"
+	"github.com/foss-opensolace/api.opensolace.com/pkg/validate"
+	"github.com/gofiber/fiber/v2"
+	"github.com/hashicorp/go-multierror"
+	"gorm.io/gorm"
+)
+
+func NewAuthRouter(router fiber.Router) {
+	group := router.Group("/auth")
+
+	group.Post("/register", authRegisterHandler())
+}
+
+func authRegisterHandler() fiber.Handler {
+	return func(c *fiber.Ctx) error {
+		var body dto.UserRegister
+
+		if err := c.BodyParser(&body); err != nil {
+			return err
+		}
+
+		if err := validate.Struct(&body); err != nil {
+			if errs, ok := err.(*multierror.Error); ok {
+				return c.Status(fiber.StatusBadRequest).JSON(errs.Errors)
+			}
+
+			return fiber.ErrInternalServerError
+		}
+
+		user, err := service.User.Create(&body)
+		if err != nil {
+			if errors.Is(err, gorm.ErrDuplicatedKey) {
+				return c.Status(fiber.StatusConflict).SendString("A user with that username or email already exists")
+			}
+
+			return err
+		}
+
+		token, err := jwt.GenerateJWT(user.ID)
+		if err != nil {
+			return err
+		}
+
+		return c.Status(fiber.StatusCreated).JSON(fiber.Map{"token": token})
+	}
+}

internal/api/service/repository/user.go

@@ -0,0 +1,59 @@
+package repository
+
+import (
+	"github.com/foss-opensolace/api.opensolace.com/internal/api/model"
+	"github.com/foss-opensolace/api.opensolace.com/internal/api/model/dto"
+	"golang.org/x/crypto/bcrypt"
+	"gorm.io/gorm"
+)
+
+type userRepository struct {
+	db *gorm.DB
+}
+
+type UserRepository interface {
+	Create(obj *dto.UserRegister) (*model.User, error)
+}
+
+func NewUserRepository(postgres *gorm.DB) UserRepository {
+	return userRepository{db: postgres}
+}
+
+func (u userRepository) Create(obj *dto.UserRegister) (*model.User, error) {
+	var displayName *string
+
+	if obj.DisplayName != "" {
+		displayName = &obj.DisplayName
+	}
+
+	newUser := model.User{
+		DisplayName: displayName,
+		Username:    obj.Username,
+		Email:       obj.Email,
+	}
+
+	var user *model.User
+	if err := u.db.First(&user, "username = ? OR email = ?", newUser.Username, newUser.Email).Error; err == nil {
+		return nil, gorm.ErrDuplicatedKey
+	}
+
+	if hashed, err := bcrypt.GenerateFromPassword([]byte(obj.Password), bcrypt.DefaultCost); err != nil {
+		return nil, err
+	} else {
+		newUser.Password = string(hashed)
+	}
+
+	transaction := u.db.Begin()
+
+	if err := transaction.Create(&newUser).Error; err != nil {
+		transaction.Rollback()
+		return nil, err
+	}
+
+	if err := transaction.Commit().Error; err != nil {
+		transaction.Rollback()
+		return nil, err
+	}
+
+	return &newUser, nil
+}

internal/api/service/service.go

@@ -0,0 +1,14 @@
+package service
+
+import (
+	"github.com/foss-opensolace/api.opensolace.com/internal/api/service/repository"
+	"github.com/foss-opensolace/api.opensolace.com/internal/db"
+)
+
+var (
+	User repository.UserRepository
+)
+
+func New() {
+	User = repository.NewUserRepository(db.Postgres)
+}