package redis_test

import (
	"fmt"
	"net"
	"os"
	"testing"
	"time"

	"apigo.cc/go/config"
	"apigo.cc/go/redis"
)

type userInfo struct {
	Id    int
	Name  string
	Phone string
	Time  time.Time
}

func TestMain(m *testing.M) {
	// 检查 Redis 是否可用，不可用则跳过测试
	os.Setenv("REDIS_TEST", "redis://:@localhost:6379/2?timeout=100ms&logSlow=10us")
	conn, err := net.DialTimeout("tcp", "localhost:6379", 500*time.Millisecond)
	if err != nil {
		fmt.Println("Redis server is not running at localhost:6379, skipping redis tests.")
		os.Exit(0)
	}
	_ = conn.Close()

	_ = config.Load("redis", nil)
	os.Exit(m.Run())
}

func TestBase(t *testing.T) {
	rd := redis.GetRedis("test", nil)
	if rd.Error != nil {
		t.Fatal("GetRedis error", rd.Error)
	}
	rd.DEL("redisName", "redisUser", "redisIds")

	// 测试不存在的 Key
	r := rd.GET("redisNotExists")
	if r.Error != nil || r.String() != "" || r.Int() != 0 {
		t.Fatal("GET NotExists should return empty", r.Error, r.String(), r.Int())
	}

	// 测试 EXISTS
	exists := rd.EXISTS("redisName")
	if exists {
		t.Fatal("EXISTS should be false")
	}

	// 测试 SET/GET
	rd.SET("redisName", "12345")
	if rd.GET("redisName").String() != "12345" {
		t.Fatal("GET mismatch")
	}

	// 测试 GETSET
	r = rd.GETSET("redisName", 12345)
	if r.String() != "12345" {
		t.Fatal("GETSET String mismatch", r.String())
	}
	if rd.GET("redisName").Int() != 12345 {
		t.Fatal("Int conversion mismatch")
	}

	// 测试 Expire
	rd.SET("redisExpire", "val")
	rd.EXPIRE("redisExpire", 1)
	time.Sleep(1100 * time.Millisecond)
	if rd.EXISTS("redisExpire") {
		t.Fatal("Key should have expired")
	}

	// 测试 Struct 自动序列化
	info := userInfo{
		Name: "aaa",
		Id:   123,
		Time: time.Now().Truncate(time.Second),
	}
	rd.SET("redisUser", info)
	var ru userInfo
	rd.GET("redisUser").To(&ru)
	if ru.Name != info.Name || ru.Id != info.Id || !ru.Time.Equal(info.Time) {
		t.Fatalf("Struct mismatch: expected %+v, got %+v", info, ru)
	}

	// 测试 MSET/MGET
	rd.MSET("redisK1", "V1", "redisK2", "V2")
	results := rd.MGET("redisK1", "redisK2")
	if len(results) != 2 || results[0].String() != "V1" || results[1].String() != "V2" {
		t.Fatal("MGET mismatch")
	}

	// 清理
	rd.DEL("redisName", "redisUser", "redisK1", "redisK2")
}

func TestIdMaker(t *testing.T) {
	rd := redis.GetRedis("test", nil)
	maker := redis.NewIdMaker(rd)

	// 测试生成唯一性
	ids := make(map[string]bool)
	for i := 0; i < 200; i++ {
		id := maker.Get(10)
		if ids[id] {
			t.Fatalf("Duplicate ID generated: %s", id)
		}
		ids[id] = true
	}

	// 测试针对数据库优化的版本
	idMysql := maker.GetForMysql(16)
	if len(idMysql) != 16 {
		t.Fatalf("Invalid MySQL ID length: %d", len(idMysql))
	}

	idPg := maker.GetForPostgreSQL(16)
	if len(idPg) != 16 {
		t.Fatalf("Invalid PostgreSQL ID length: %d", len(idPg))
	}
}

func TestGenerics(t *testing.T) {
	rd := redis.GetRedis("test", nil)
	rd.SET("gen_test", userInfo{Name: "Generics", Id: 888})
	defer rd.DEL("gen_test")

	r := rd.GET("gen_test")
	user := redis.To[userInfo](r)
	if user.Name != "Generics" || user.Id != 888 {
		t.Fatal("Generics To[T] mismatch", user)
	}
}