hello world

storage/api.go

@@ -0,0 +1,103 @@
+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
+}