GithubProxy/coai
1
0
Fork 0
mirror of https://github.com/coaidev/coai.git synced 2025-06-06 22:00:25 +09:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: fix deeptrain login

Browse Source
This commit is contained in:
Zhang Minghan 2024-03-07 12:03:23 +08:00
parent 4140fec71c
commit c3d5928414
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
auth/auth.go
View File

@ -202,8 +202,13 @@ func DeepLogin(c *gin.Context, token string) (string, error) {
// register
password := utils.GenerateChar(64)
_ = globals.QueryRowDb(db, "INSERT INTO auth (bind_id, username, token, password) VALUES (?, ?, ?, ?)",
user.ID, user.Username, token, password)
_, err := globals.QueryDb(db, "INSERT INTO auth (bind_id, username, token, password) VALUES (?, ?, ?, ?)",
user.ID, user.Username, utils.Extract(token, 255, ""), utils.Sha2Encrypt(password))
if err != nil {
return "", err
}
u := &User{
Username: user.Username,
Password: password,
@ -214,7 +219,7 @@ func DeepLogin(c *gin.Context, token string) (string, error) {
}
// login
_ = globals.QueryRowDb(db, "UPDATE auth SET token = ? WHERE username = ?", token, user.Username)
_ = globals.QueryRowDb(db, "UPDATE auth SET token = ? WHERE username = ?", utils.Extract(token, 255, ""), user.Username)
var password string
err := globals.QueryRowDb(db, "SELECT password FROM auth WHERE username = ?", user.Username).Scan(&password)
if err != nil {