mirror of
https://github.com/coaidev/coai.git
synced 2025-05-19 21:10:18 +09:00
25 lines
631 B
Go
25 lines
631 B
Go
package auth
|
|
|
|
import (
|
|
"chat/utils"
|
|
"database/sql"
|
|
"fmt"
|
|
)
|
|
|
|
func (u *User) CreateApiKey(db *sql.DB) string {
|
|
salt := utils.Sha2Encrypt(fmt.Sprintf("%s-%s", u.Username, utils.GenerateChar(utils.GetRandomInt(720, 1024))))
|
|
key := fmt.Sprintf("sk-%s", salt[:64]) // 64 bytes
|
|
if _, err := db.Exec("INSERT INTO apikey (user_id, api_key) VALUES (?, ?)", u.GetID(db), key); err != nil {
|
|
return ""
|
|
}
|
|
return key
|
|
}
|
|
|
|
func (u *User) GetApiKey(db *sql.DB) string {
|
|
var key string
|
|
if err := db.QueryRow("SELECT api_key FROM apikey WHERE user_id = ?", u.GetID(db)).Scan(&key); err != nil {
|
|
return u.CreateApiKey(db)
|
|
}
|
|
return key
|
|
}
|