Auto reg user, event map

Signed-off-by: Asai Neko <sugar@sne.moe>

service/auth/magic.go

@@ -1,14 +1,15 @@
 package auth
 
 import (
-	"net/http"
 	"nixcn-cms/data"
 	"nixcn-cms/internal/cryptography"
 	"nixcn-cms/pkgs/email"
 	"nixcn-cms/pkgs/magiclink"
 	"nixcn-cms/pkgs/turnstile"
 
+	"github.com/google/uuid"
 	log "github.com/sirupsen/logrus"
+	"gorm.io/gorm"
 
 	"github.com/gin-gonic/gin"
 	"github.com/spf13/viper"
@@ -74,14 +75,41 @@ func VerifyMagicLink(c *gin.Context) {
 		return
 	}
 
-	// Generate jwt
-	userInfo := new(data.User)
-	err := userInfo.GetByEmail(email)
+	// Verify if user exists
+	user := new(data.User)
+	err := user.GetByEmail(email)
+
 	if err != nil {
-		c.JSON(http.StatusUnauthorized, gin.H{"status": "user not found"})
+		if err == gorm.ErrRecordNotFound {
+			// Create user
+			newUUID, err := uuid.NewUUID()
+			if err != nil {
+				c.JSON(500, gin.H{"status": "internal server error"})
+				return
+			}
+			newUserId, err := uuid.NewUUID()
+			if err != nil {
+				c.JSON(500, gin.H{"status": "internal server error"})
+				return
+			}
+			user.UUID = newUUID
+			user.UserId = newUserId
+			user.Email = email
+			user.Type = "Normal"
+			user.PermissionLevel = 10
+			if err := user.Create(); err != nil {
+				c.JSON(500, gin.H{"status": "internal server error"})
+				return
+			}
+		} else {
+			c.JSON(500, gin.H{"status": "internal server error"})
+			return
+		}
 	}
+
+	// Generate jwt
 	JwtTool := cryptography.Token{
-		UserID:      userInfo.UserId,
+		UserID:      user.UserId,
 		Application: viper.GetString("server.application"),
 	}
 	accessToken, refreshToken, err := JwtTool.IssueTokens()

service/auth/refresh.go

@@ -1,7 +1,6 @@
 package auth
 
 import (
-	"net/http"
 	"nixcn-cms/internal/cryptography"
 
 	"github.com/gin-gonic/gin"
@@ -14,7 +13,7 @@ func Refresh(c *gin.Context) {
 	}
 
 	if err := c.ShouldBindJSON(&req); err != nil {
-		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid request"})
+		c.JSON(400, gin.H{"error": "invalid request"})
 		return
 	}
 
@@ -24,11 +23,11 @@ func Refresh(c *gin.Context) {
 
 	access, err := JwtTool.RefreshAccessToken(req.RefreshToken)
 	if err != nil {
-		c.JSON(http.StatusUnauthorized, gin.H{"error": "invalid refresh token"})
+		c.JSON(401, gin.H{"error": "invalid refresh token"})
 		return
 	}
 
-	c.JSON(http.StatusOK, gin.H{
+	c.JSON(200, gin.H{
 		"access_token": access,
 	})
 }