♻️ update code structure

service/impl/user_service_impl.go

@@ -1,13 +1,19 @@
 package impl
 
 import (
+	"encoding/gob"
 	"encoding/json"
+	"errors"
+	"github.com/gin-gonic/gin"
+	"github.com/mssola/useragent"
+	"gorm.io/gorm"
 	"schisandra-cloud-album/common/constant"
 	"schisandra-cloud-album/common/redis"
 	"schisandra-cloud-album/dao/impl"
 	"schisandra-cloud-album/global"
 	"schisandra-cloud-album/model"
 	"schisandra-cloud-album/utils"
+	"sync"
 	"time"
 )
 
@@ -15,6 +21,8 @@ var userDao = impl.UserDaoImpl{}
 
 type UserServiceImpl struct{}
 
+var mu = &sync.Mutex{}
+
 // ResponseData 返回数据
 type ResponseData struct {
 	AccessToken  string  `json:"access_token"`
@@ -103,3 +111,113 @@ func (UserServiceImpl) RefreshTokenService(refreshToken string) (*ResponseData,
 	}
 	return &data, true
 }
+
+// HandelUserLogin 处理用户登录
+func (UserServiceImpl) HandelUserLogin(user model.ScaAuthUser, autoLogin bool, c *gin.Context) (*ResponseData, bool) {
+	// 检查 user.UID 是否为 nil
+	if user.UID == nil {
+		return nil, false
+	}
+	if !GetUserLoginDevice(user, c) {
+		return nil, false
+	}
+	accessToken, err := utils.GenerateAccessToken(utils.AccessJWTPayload{UserID: user.UID})
+	if err != nil {
+		return nil, false
+	}
+	var days time.Duration
+	if autoLogin {
+		days = 7 * 24 * time.Hour
+	} else {
+		days = time.Minute * 30
+	}
+
+	refreshToken, expiresAt := utils.GenerateRefreshToken(utils.RefreshJWTPayload{UserID: user.UID}, days)
+	data := ResponseData{
+		AccessToken:  accessToken,
+		RefreshToken: refreshToken,
+		ExpiresAt:    expiresAt,
+		UID:          user.UID,
+	}
+
+	err = redis.Set(constant.UserLoginTokenRedisKey+*user.UID, data, days).Err()
+	if err != nil {
+		return nil, false
+	}
+	gob.Register(ResponseData{})
+	err = utils.SetSession(c, constant.SessionKey, data)
+	if err != nil {
+		return nil, false
+	}
+	return &data, true
+}
+
+// GetUserLoginDevice 获取用户登录设备
+func GetUserLoginDevice(user model.ScaAuthUser, c *gin.Context) bool {
+
+	// 检查user.UID是否为空
+	if user.UID == nil {
+		global.LOG.Errorln("user.UID is nil")
+		return false
+	}
+	userAgent := c.GetHeader("User-Agent")
+	if userAgent == "" {
+		global.LOG.Errorln("user-agent is empty")
+		return false
+	}
+	ua := useragent.New(userAgent)
+
+	ip := utils.GetClientIP(c)
+	location, err := global.IP2Location.SearchByStr(ip)
+	if err != nil {
+		global.LOG.Errorln(err)
+		return false
+	}
+	location = utils.RemoveZeroAndAdjust(location)
+
+	isBot := ua.Bot()
+	browser, browserVersion := ua.Browser()
+	os := ua.OS()
+	mobile := ua.Mobile()
+	mozilla := ua.Mozilla()
+	platform := ua.Platform()
+	engine, engineVersion := ua.Engine()
+
+	device := model.ScaAuthUserDevice{
+		UserID:          user.UID,
+		IP:              &ip,
+		Location:        &location,
+		Agent:           userAgent,
+		Browser:         &browser,
+		BrowserVersion:  &browserVersion,
+		OperatingSystem: &os,
+		Mobile:          &mobile,
+		Bot:             &isBot,
+		Mozilla:         &mozilla,
+		Platform:        &platform,
+		EngineName:      &engine,
+		EngineVersion:   &engineVersion,
+	}
+
+	mu.Lock()
+	defer mu.Unlock()
+
+	userDevice, err := userDeviceDao.GetUserDeviceByUIDIPAgent(*user.UID, ip, userAgent)
+	if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
+		err = userDeviceDao.AddUserDevice(&device)
+		if err != nil {
+			global.LOG.Errorln(err)
+			return false
+		}
+	} else if err != nil {
+		global.LOG.Errorln(err)
+		return false
+	} else {
+		err := userDeviceDao.UpdateUserDevice(userDevice.ID, &device)
+		if err != nil {
+			global.LOG.Errorln(err)
+			return false
+		}
+	}
+	return true
+}