45 lines
876 B
Go
45 lines
876 B
Go
package auth
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/huoxue1/qinglong-go/internal/cache"
|
|
"github.com/huoxue1/qinglong-go/models"
|
|
)
|
|
|
|
const (
|
|
CacheAuthKey = "auth_info"
|
|
)
|
|
|
|
func InitAuthInfo(username, password string) error {
|
|
m := new(models.AuthFile)
|
|
m.Username = username
|
|
m.Password = password
|
|
m.IsTwoFactorChecking = false
|
|
m.Tokens.Mobile = ""
|
|
m.Tokens.Desktop = ""
|
|
data, _ := json.Marshal(m)
|
|
return cache.SetBytes(CacheAuthKey, data)
|
|
}
|
|
|
|
func GetAuthInfo() (*models.AuthFile, error) {
|
|
m := new(models.AuthFile)
|
|
data, err := cache.GetBytes(CacheAuthKey)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = json.Unmarshal(data, m)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
func UpdateAuthInfo(file *models.AuthFile) error {
|
|
data, _ := json.Marshal(file)
|
|
return cache.SetBytes(CacheAuthKey, data)
|
|
}
|
|
|
|
func IsInit() bool {
|
|
return cache.Exists(CacheAuthKey)
|
|
}
|