103 lines
1.8 KiB
Go
103 lines
1.8 KiB
Go
package storage
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"git.zomo.dev/zomo/discord-retokenizer/util"
|
|
"github.com/go-redis/redis/v9"
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
func UpdateUsername(username string) {
|
|
if username != "" {
|
|
client.Set(ctx, "username", username, 0)
|
|
}
|
|
}
|
|
|
|
func UpdatePassword(password string) {
|
|
if password != "" {
|
|
passHash, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
client.Set(ctx, "password", string(passHash), 0)
|
|
}
|
|
}
|
|
|
|
func CheckLogin(username string, password string) (bool, string) {
|
|
if username == "" || password == "" {
|
|
return false, ""
|
|
}
|
|
|
|
user, err := client.Get(ctx, "username").Result()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
pass, err := client.Get(ctx, "password").Result()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if user != username {
|
|
return false, ""
|
|
}
|
|
|
|
err = bcrypt.CompareHashAndPassword([]byte(pass), []byte(password))
|
|
if err != nil {
|
|
return false, ""
|
|
}
|
|
|
|
return true, createLoginToken()
|
|
}
|
|
|
|
func createLoginToken() string {
|
|
token := util.GeneratePassword(32)
|
|
|
|
member := redis.Z{
|
|
Score: float64(time.Now().Unix() + 4 * 60 * 60),
|
|
Member: token,
|
|
}
|
|
|
|
err := client.ZAdd(ctx, "loginTokens", member).Err()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return token
|
|
}
|
|
|
|
func CheckLoginToken(token string) bool {
|
|
|
|
expired, err := client.ZRangeByScore(ctx, "loginTokens", &redis.ZRangeBy{
|
|
Min: "-inf",
|
|
Max: fmt.Sprintf("%d", time.Now().Unix()),
|
|
}).Result()
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
for _, e := range expired {
|
|
client.ZRem(ctx, "loginTokens", e)
|
|
}
|
|
|
|
current, err := client.ZRangeByScore(ctx, "loginTokens", &redis.ZRangeBy{
|
|
Min: fmt.Sprintf("%d", time.Now().Unix()),
|
|
Max: "inf",
|
|
}).Result()
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
for _, c := range current {
|
|
fmt.Println(c)
|
|
if c == token {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
} |