package cache

import (
	"context"
	"time"
)

// Cache 缓存接口
type Cache interface {
	// Get 获取缓存
	Get(ctx context.Context, key string) (*CacheEntry, error)

	// Set 设置缓存
	Set(ctx context.Context, entry *CacheEntry) error

	// Delete 删除缓存
	Delete(ctx context.Context, key string) error

	// Clear 清空缓存
	Clear(ctx context.Context) error

	// Stats 获取缓存统计信息
	Stats(ctx context.Context) (*CacheStats, error)

	// Cleanup 清理过期缓存
	Cleanup(ctx context.Context) error

	// Close 关闭缓存
	Close() error
}

// CacheEntry 缓存条目
type CacheEntry struct {
	ID               int64     `json:"id"`
	CacheKey         string    `json:"cache_key"`
	OriginalText     string    `json:"original_text"`
	TranslatedText   string    `json:"translated_text"`
	FromLang         string    `json:"from_lang"`
	ToLang           string    `json:"to_lang"`
	Model            string    `json:"model"`
	PromptName       string    `json:"prompt_name,omitempty"`
	PromptContent    string    `json:"prompt_content,omitempty"`
	PromptTokens     int       `json:"prompt_tokens"`
	CompletionTokens int       `json:"completion_tokens"`
	TotalTokens      int       `json:"total_tokens"`
	CreatedAt        time.Time `json:"created_at"`
	LastUsedAt       time.Time `json:"last_used_at"`
}

// CacheStats 缓存统计信息
type CacheStats struct {
	TotalRecords       int       `json:"total_records"`
	TotalSizeBytes     int64     `json:"total_size_bytes"`
	OldestRecord       time.Time `json:"oldest_record"`
	NewestRecord       time.Time `json:"newest_record"`
	AvgTokensPerRecord float64   `json:"avg_tokens_per_record"`
}

// CacheConfig 缓存配置
type CacheConfig struct {
	Enabled    bool   `yaml:"enabled"`
	MaxRecords int    `yaml:"max_records"`
	ExpireDays int    `yaml:"expire_days"`
	DBPath     string `yaml:"db_path"`
}

// NewCacheConfig 创建默认缓存配置
func NewCacheConfig() *CacheConfig {
	return &CacheConfig{
		Enabled:    true,
		MaxRecords: 10000,
		ExpireDays: 30,
		DBPath:     "~/.config/yoyo/cache.db",
	}
}