package benchmark

import (
	"testing"
	"time"

	"github.com/landaiqing/go-pixelnebula"
	"github.com/landaiqing/go-pixelnebula/cache"
	"github.com/landaiqing/go-pixelnebula/style"
)

// BenchmarkDefaultCacheVsNoCache 对比有无默认缓存的性能差异
func BenchmarkDefaultCacheVsNoCache(b *testing.B) {
	// 不使用缓存的基准测试
	b.Run("NoCache", func(b *testing.B) {
		b.ResetTimer()

		for i := 0; i < b.N; i++ {
			pn := pixelnebula.NewPixelNebula()
			pn.WithStyle(style.GirlStyle)
			pn.WithSize(231, 231)

			_, err := pn.Generate("benchmark-cache-test", false).ToSVG()
			if err != nil {
				b.Fatal(err)
			}
		}
	})

	// 使用默认缓存的基准测试
	b.Run("DefaultCache", func(b *testing.B) {
		// 创建一个带默认缓存的实例
		pn := pixelnebula.NewPixelNebula()
		pn.WithStyle(style.GirlStyle)
		pn.WithSize(231, 231)
		pn.WithDefaultCache()

		b.ResetTimer()

		for i := 0; i < b.N; i++ {
			_, err := pn.Generate("benchmark-cache-test", false).ToSVG()
			if err != nil {
				b.Fatal(err)
			}
		}
	})
}

// BenchmarkCacheSizes 测试不同缓存大小对性能的影响
func BenchmarkCacheSizes(b *testing.B) {
	cacheSizes := []int{10, 100, 1000}

	for _, size := range cacheSizes {
		b.Run("CacheSize_"+Itoa(size), func(b *testing.B) {
			// 创建自定义缓存配置
			pn := pixelnebula.NewPixelNebula()
			pn.WithStyle(style.GirlStyle)
			pn.WithSize(231, 231)
			pn.WithCache(cache.CacheOptions{
				Size:       size,
				Expiration: 3600 * time.Second,
			})

			// 预热缓存，生成一些不同的头像
			for i := 0; i < size/2; i++ {
				_, _ = pn.Generate("preload-"+Itoa(i), false).ToSVG()
			}

			b.ResetTimer()

			// 测试缓存命中和未命中的混合场景
			for i := 0; i < b.N; i++ {
				id := "benchmark-" + Itoa(i%size) // 循环使用ID，确保部分缓存命中
				_, err := pn.Generate(id, false).ToSVG()
				if err != nil {
					b.Fatal(err)
				}
			}
		})
	}
}

// BenchmarkCacheCompression 测试缓存压缩对性能的影响
func BenchmarkCacheCompression(b *testing.B) {
	compressionLevels := []struct {
		name    string
		options cache.CompressOptions
	}{
		{
			name: "NoCompression",
			options: cache.CompressOptions{
				Enabled: false,
			},
		},
		{
			name: "DefaultCompression",
			options: cache.CompressOptions{
				Enabled:          true,
				Level:            5,    // 默认压缩级别
				MinSize:          1024, // 最小压缩大小
				CompressionRatio: 0.8,  // 最小压缩比率
			},
		},
		{
			name: "BestCompression",
			options: cache.CompressOptions{
				Enabled:          true,
				Level:            9,    // 最高压缩级别
				MinSize:          1024, // 最小压缩大小
				CompressionRatio: 0.8,  // 最小压缩比率
			},
		},
	}

	for _, cl := range compressionLevels {
		b.Run(cl.name, func(b *testing.B) {
			// 创建带压缩缓存的实例
			pn := pixelnebula.NewPixelNebula()
			pn.WithStyle(style.GirlStyle)
			pn.WithSize(231, 231)
			pn.WithDefaultCache()

			if cl.options.Enabled {
				pn.WithCompression(cl.options)
			}

			b.ResetTimer()

			for i := 0; i < b.N; i++ {
				_, err := pn.Generate("benchmark-compress", false).ToSVG()
				if err != nil {
					b.Fatal(err)
				}
			}
		})
	}
}

// BenchmarkCacheExpiry 测试不同缓存过期时间的性能影响
func BenchmarkCacheExpiry(b *testing.B) {
	expiryTimes := []struct {
		name string
		time time.Duration
	}{
		{"Short_1m", 1 * time.Minute},
		{"Medium_1h", 1 * time.Hour},
		{"Long_24h", 24 * time.Hour},
	}

	for _, et := range expiryTimes {
		b.Run(et.name, func(b *testing.B) {
			// 创建带自定义过期时间的缓存实例
			pn := pixelnebula.NewPixelNebula()
			pn.WithStyle(style.GirlStyle)
			pn.WithSize(231, 231)
			pn.WithCache(cache.CacheOptions{
				Size:       100,
				Expiration: et.time,
			})

			b.ResetTimer()

			for i := 0; i < b.N; i++ {
				_, err := pn.Generate("benchmark-expiry", false).ToSVG()
				if err != nil {
					b.Fatal(err)
				}
			}
		})
	}
}