更新用户登录
This commit is contained in:
@@ -4,148 +4,147 @@ import (
|
||||
"crypto/aes"
|
||||
"crypto/cipher"
|
||||
"encoding/base64"
|
||||
"github.com/gin-gonic/gin"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"math/rand"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strconv"
|
||||
"time"
|
||||
"toutoukan/init/config"
|
||||
"toutoukan/init/databaseInit"
|
||||
"toutoukan/model/usermodel"
|
||||
"toutoukan/utill/jwt"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
const wxLoginURL = "https://api.weixin.qq.com/sns/jscode2session"
|
||||
|
||||
func UserLogin(c *gin.Context) {
|
||||
fmt.Println("Request Body:", c.Request.Body)
|
||||
var req usermodel.WxLoginM
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
}
|
||||
fmt.Println("req:123")
|
||||
fmt.Println(req.EncryptedData)
|
||||
fmt.Println(req.Code)
|
||||
fmt.Println(req.Iv)
|
||||
fmt.Println("使用的appid:", config.Conf.WxID)
|
||||
fmt.Println("使用的secret:", config.Conf.Wxsecret)
|
||||
//----------------发送验证请求
|
||||
params := url.Values{}
|
||||
params.Add("appid", config.Conf.WxID)
|
||||
params.Add("secret", config.Conf.Wxsecret)
|
||||
params.Add("js_code", req.Code)
|
||||
params.Add("grant_type", "authorization_code")
|
||||
requestURL := fmt.Sprintf("%s?%s", wxLoginURL, params.Encode())
|
||||
fmt.Println("微信接口请求URL:", requestURL)
|
||||
|
||||
resp, err := http.Get(fmt.Sprintf("%s?%s", wxLoginURL, params.Encode()))
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "调用微信登录接口失败: " + err.Error()})
|
||||
return
|
||||
}
|
||||
fmt.Println("微信登录返回结果：", resp.Body)
|
||||
defer resp.Body.Close()
|
||||
|
||||
var wxResp usermodel.WxLoginResponse
|
||||
if err := json.NewDecoder(resp.Body).Decode(&wxResp); err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "解析微信登录响应失败: " + err.Error(), "code": "10026"})
|
||||
return
|
||||
}
|
||||
|
||||
if wxResp.ErrCode != 0 {
|
||||
c.JSON(http.StatusBadRequest, gin.H{
|
||||
"error": fmt.Sprintf("微信登录失败: %s (错误码: %d)", wxResp.ErrCode, wxResp.ErrMsg),
|
||||
"code": "10027",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
phoneData, err := decryptWxData(wxResp.SessionKey, req.EncryptedData, req.Iv)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "解密失败: " + err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
var phoneInfo usermodel.WxPhoneInfo
|
||||
if err := json.Unmarshal(phoneData, &phoneInfo); err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "解析手机号失败"})
|
||||
return
|
||||
}
|
||||
|
||||
if phoneInfo.Watermark.AppID != config.Conf.WxID {
|
||||
c.JSON(http.StatusForbidden, gin.H{"error": "数据水印验证失败"})
|
||||
return
|
||||
}
|
||||
|
||||
fmt.Println("用户手机号为：", phoneInfo.PhoneNumber)
|
||||
|
||||
openid := wxResp.OpenID
|
||||
|
||||
// 业务逻辑实现
|
||||
var user UserInfo
|
||||
var username string
|
||||
// 查找用户是否存在
|
||||
result := databaseInit.UserDB.Where("uid = ?", openid).First(&user)
|
||||
|
||||
if result.Error != nil {
|
||||
if errors.Is(result.Error, gorm.ErrRecordNotFound) {
|
||||
// 用户不存在，创建新用户
|
||||
username := generateUsername()
|
||||
now := time.Now()
|
||||
newUser := UserInfo{
|
||||
Uid: openid,
|
||||
Gender: 2, // 这里假设 gender 传 2，可根据实际需求修改
|
||||
CreatedTime: now,
|
||||
UpdatedTime: now,
|
||||
Username: username,
|
||||
Telephone: phoneInfo.PhoneNumber,
|
||||
// 若还有 password、avatar_url、birthdate、bio 等字段需要赋值，可在此补充
|
||||
// 比如 Password: "默认密码"（实际中密码应加密存储），AvatarUrl: "默认头像地址" 等
|
||||
}
|
||||
|
||||
if err := databaseInit.UserDB.Create(&newUser).Error; err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{
|
||||
"error": "插入新用户失败: " + err.Error(),
|
||||
"code": "10029",
|
||||
})
|
||||
return
|
||||
}
|
||||
username = newUser.Username
|
||||
} else {
|
||||
// 数据库查询错误
|
||||
c.JSON(http.StatusInternalServerError, gin.H{
|
||||
"error": "查询用户存在性失败: " + result.Error.Error(),
|
||||
"code": "10029",
|
||||
})
|
||||
return
|
||||
}
|
||||
} else {
|
||||
// 用户已存在，获取用户名
|
||||
username = user.Username
|
||||
}
|
||||
|
||||
token, err := jwt.GenerateJWTAndStore(openid)
|
||||
if err != nil {
|
||||
c.JSON(500, gin.H{"error": "生成令牌失败", "code": "10036"})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, gin.H{"result": "success", "error": nil, "code": "20001", "token": token,
|
||||
"userinfo": map[string]string{
|
||||
"username": username,
|
||||
"uid": openid,
|
||||
"telephone": phoneInfo.PhoneNumber,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
//func UserLogin(c *gin.Context) {
|
||||
// fmt.Println("Request Body:", c.Request.Body)
|
||||
// var req usermodel.WxLoginM
|
||||
// if err := c.ShouldBindJSON(&req); err != nil {
|
||||
// c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
// }
|
||||
// fmt.Println("req:123")
|
||||
// fmt.Println(req.EncryptedData)
|
||||
// fmt.Println(req.Code)
|
||||
// fmt.Println(req.Iv)
|
||||
// fmt.Println("使用的appid:", config.Conf.WxID)
|
||||
// fmt.Println("使用的secret:", config.Conf.Wxsecret)
|
||||
// //----------------发送验证请求
|
||||
// params := url.Values{}
|
||||
// params.Add("appid", config.Conf.WxID)
|
||||
// params.Add("secret", config.Conf.Wxsecret)
|
||||
// params.Add("js_code", req.Code)
|
||||
// params.Add("grant_type", "authorization_code")
|
||||
// requestURL := fmt.Sprintf("%s?%s", wxLoginURL, params.Encode())
|
||||
// fmt.Println("微信接口请求URL:", requestURL)
|
||||
//
|
||||
// resp, err := http.Get(fmt.Sprintf("%s?%s", wxLoginURL, params.Encode()))
|
||||
// if err != nil {
|
||||
// c.JSON(http.StatusInternalServerError, gin.H{"error": "调用微信登录接口失败: " + err.Error()})
|
||||
// return
|
||||
// }
|
||||
// fmt.Println("微信登录返回结果：", resp.Body)
|
||||
// defer resp.Body.Close()
|
||||
//
|
||||
// var wxResp usermodel.WxLoginResponse
|
||||
// if err := json.NewDecoder(resp.Body).Decode(&wxResp); err != nil {
|
||||
// c.JSON(http.StatusInternalServerError, gin.H{"error": "解析微信登录响应失败: " + err.Error(), "code": "10026"})
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// if wxResp.ErrCode != 0 {
|
||||
// c.JSON(http.StatusBadRequest, gin.H{
|
||||
// "error": fmt.Sprintf("微信登录失败: %s (错误码: %d)", wxResp.ErrCode, wxResp.ErrMsg),
|
||||
// "code": "10027",
|
||||
// })
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// phoneData, err := decryptWxData(wxResp.SessionKey, req.EncryptedData, req.Iv)
|
||||
// if err != nil {
|
||||
// c.JSON(http.StatusInternalServerError, gin.H{"error": "解密失败: " + err.Error()})
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// var phoneInfo usermodel.WxPhoneInfo
|
||||
// if err := json.Unmarshal(phoneData, &phoneInfo); err != nil {
|
||||
// c.JSON(http.StatusInternalServerError, gin.H{"error": "解析手机号失败"})
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// if phoneInfo.Watermark.AppID != config.Conf.WxID {
|
||||
// c.JSON(http.StatusForbidden, gin.H{"error": "数据水印验证失败"})
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// fmt.Println("用户手机号为：", phoneInfo.PhoneNumber)
|
||||
//
|
||||
// openid := wxResp.OpenID
|
||||
// ctx := c.Request.Context()
|
||||
//
|
||||
// var username string
|
||||
//
|
||||
// var exists bool
|
||||
// query := "SELECT EXISTS(SELECT 1 FROM user_info WHERE uid = ? LIMIT 1)"
|
||||
// err = databaseInit.UserDB.QueryRowContext(ctx, query, openid).Scan(&exists)
|
||||
// if err != nil {
|
||||
// c.JSON(http.StatusInternalServerError, gin.H{
|
||||
// "error": "查询用户存在性失败: " + err.Error(),
|
||||
// "code": "10029",
|
||||
// })
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// if !exists {
|
||||
// username = generateUsername()
|
||||
// now := time.Now()
|
||||
// insertSQL := `
|
||||
// INSERT INTO user_info (
|
||||
// uid, gender,createdtime, updatedtime,username,telephone
|
||||
// ) VALUES (?, ?, ?,?,?,?)
|
||||
// `
|
||||
// _, err := databaseInit.UserDB.ExecContext(
|
||||
// ctx,
|
||||
// insertSQL,
|
||||
// openid, // uid（使用微信 openid）
|
||||
// 2,
|
||||
// now,
|
||||
// now,
|
||||
// username,
|
||||
// phoneInfo.PhoneNumber,
|
||||
// )
|
||||
// if err != nil {
|
||||
// c.JSON(http.StatusInternalServerError, gin.H{
|
||||
// "error": "插入新用户失败: " + err.Error(),
|
||||
// "code": "10029",
|
||||
// })
|
||||
// return
|
||||
// }
|
||||
// } else {
|
||||
// queryUser := "SELECT username FROM user_info WHERE uid = ? LIMIT 1"
|
||||
// err = databaseInit.UserDB.QueryRowContext(ctx, queryUser, openid).Scan(&username)
|
||||
// if err != nil {
|
||||
// c.JSON(http.StatusInternalServerError, gin.H{
|
||||
// "error": "查询用户信息失败: " + err.Error(),
|
||||
// "code": "10030",
|
||||
// })
|
||||
// return
|
||||
// }
|
||||
// }
|
||||
//
|
||||
// token, err := utill.GenerateJWTAndStore(openid)
|
||||
// if err != nil {
|
||||
// c.JSON(500, gin.H{"error": "生成令牌失败", "code": "10036"})
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// c.JSON(http.StatusOK, gin.H{"result": "success", "error": nil, "code": "20001", "token": token,
|
||||
// "userinfo": map[string]string{
|
||||
// "username": username,
|
||||
// "uid": openid,
|
||||
// "telephone": phoneInfo.PhoneNumber,
|
||||
// },
|
||||
// })
|
||||
//}
|
||||
|
||||
func generateUsername() string {
|
||||
rand.Seed(time.Now().UnixNano())
|
||||
randomNum := 10000000 + rand.Intn(90000000)
|
||||
@@ -178,3 +177,22 @@ func pkcs7Unpad(data []byte) []byte {
|
||||
}
|
||||
return data[:len(data)-padding]
|
||||
}
|
||||
|
||||
// 定义与表结构对应的用户模型
|
||||
type UserInfo struct {
|
||||
Uid string `gorm:"column:uid;primaryKey"`
|
||||
Telephone string `gorm:"column:telephone"`
|
||||
Password string `gorm:"column:password"`
|
||||
AvatarUrl string `gorm:"column:avatar_url"`
|
||||
Gender int `gorm:"column:gender"`
|
||||
Birthdate time.Time `gorm:"column:birthdate;type:datetime"`
|
||||
CreatedTime time.Time `gorm:"column:createdtime;type:datetime"`
|
||||
UpdatedTime time.Time `gorm:"column:updatedtime;type:datetime"`
|
||||
Bio string `gorm:"column:bio"`
|
||||
Username string `gorm:"column:username"`
|
||||
}
|
||||
|
||||
// 自定义表名
|
||||
func (UserInfo) TableName() string {
|
||||
return "user_info"
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user