修改权限

2024-04-09 12:11:46 +08:00
parent ce06c5a0d4
commit 51ac40758e
4 changed files with 121 additions and 13 deletions
@@ -32,6 +32,7 @@ type LoginLoginVerify struct {
 // @Param LoginLoginVerify body LoginLoginVerify true "登陆验证信息"
 // @Tags user
 // @Router /login [post]
+
 func (l LoginApi) Login(c *gin.Context) {
 	param := LoginLoginVerify{}
 	if err := c.ShouldBindJSON(&param); err != nil {
@@ -111,3 +112,111 @@ func (l *LoginApi) Logout(c *gin.Context) {
 	global.CUserToken.Delete(cToken)
 	apiReturn.Success(c)
 }
+
+/**
+ * @Author: wintsa
+ * @Date: 2024-04-09 11:01:55
+ * @LastEditors: wintsa
+ * @Description: token生成
+ * @return {*}
+ */
+func (l LoginApi) GetToken(c *gin.Context) {
+	param := LoginLoginVerify{}
+	if err := c.ShouldBindJSON(&param); err != nil {
+		apiReturn.ErrorParamFomat(c, err.Error())
+		return
+	}
+
+	if errMsg, err := base.ValidateInputStruct(param); err != nil {
+		apiReturn.ErrorParamFomat(c, errMsg)
+		return
+	}
+
+	settings := systemSetting.ApplicationSetting{}
+	global.SystemSetting.GetValueByInterface("system_application", &settings)
+
+	mUser := models.User{}
+	var (
+		err  error
+		info models.User
+	)
+	bToken := ""
+	param.Username = strings.TrimSpace(param.Username)
+	info, err1 := mUser.GetUserInfoByUsername(param.Username)
+
+	if err1 != nil {
+		// 未找到记录 账号
+
+		if err == gorm.ErrRecordNotFound {
+			return
+
+			// 	mUser := models.User{
+			// 		Username:  strings.TrimSpace(param.Username),
+			// 		Password:  cmn.PasswordEncryption("zw.123456"),
+			// 		Name:      param.Username,
+			// 		HeadImage: "",
+			// 		Status:    1,
+			// 		Role:      2,
+			// 		// Mail:      param.Username, 不再保存邮箱账号字段
+			// 	}
+			// 	userInfo, err := mUser.CreateOne()
+			// 	if err != nil {
+			// 		apiReturn.ErrorDatabase(c, err.Error())
+			// 		return
+			// 	}
+			// 	info = userInfo
+		} else {
+			// 未知错误
+			apiReturn.Error(c, err.Error())
+			return
+		}
+
+	}
+
+	// 停用或未激活
+	if info.Status != 1 {
+		apiReturn.ErrorByCode(c, 1004)
+		return
+	}
+
+	bToken = info.Token
+	if info.Token == "" {
+		// 生成token
+		buildTokenOver := false
+		for !buildTokenOver {
+			bToken = cmn.BuildRandCode(32, cmn.RAND_CODE_MODE2)
+			if _, err := mUser.GetUserInfoByToken(bToken); err != nil {
+				// 保存token
+				mUser.UpdateUserInfoByUserId(info.ID, map[string]interface{}{
+					"token": bToken,
+				})
+				buildTokenOver = true
+			}
+		}
+		info.Token = bToken
+	}
+	info.Password = ""
+	info.ReferralCode = ""
+
+	// global.UserToken.SetDefault(bToken, info)
+	cToken := uuid.NewString() + "-" + cmn.Md5(cmn.Md5("userId"+strconv.Itoa(int(info.ID))))
+	global.CUserToken.SetDefault(cToken, bToken)
+	global.Logger.Debug("token:", cToken, "|", bToken)
+	global.Logger.Debug(global.CUserToken.Get(cToken))
+
+	// 设置当前用户信息
+	c.Set("userInfo", info)
+	info.Token = cToken // 重要 采用cToken,隐藏真实token
+	apiReturn.SuccessData(c, info)
+}
+
+/**
+ * @Author: wintsa
+ * @Date: 2024-04-09 11:02:30
+ * @LastEditors: wintsa
+ * @Description: token验证
+ * @return {*}
+ */
+// func (l LoginApi) tokenValidate(c *gin.Context) {
+
+// }