using Quartz;
using Service.AI;

namespace Service.Jobs;

/// <summary>
/// 分类图标生成定时任务
/// 每10分钟扫描一次，为没有图标的分类生成 5 个 SVG 图标
/// </summary>
public class CategoryIconGenerationJob(
    ITransactionCategoryRepository categoryRepository,
    IOpenAiService openAiService,
    ILogger<CategoryIconGenerationJob> logger) : IJob
{
    private static readonly SemaphoreSlim _semaphore = new(1, 1);

    public async Task Execute(IJobExecutionContext context)
    {
        // 尝试获取锁，如果失败则跳过本次执行
        if (!await _semaphore.WaitAsync(0))
        {
            logger.LogInformation("上一个分类图标生成任务尚未完成，跳过本次执行");
            return;
        }

        try
        {

            logger.LogInformation("开始执行分类图标生成任务");

            // 查询所有分类，然后过滤出没有图标的
            var allCategories = await categoryRepository.GetAllAsync();
            var categoriesWithoutIcon = allCategories
                .Where(c => string.IsNullOrEmpty(c.Icon))
                .ToList();

            if (categoriesWithoutIcon.Count == 0)
            {
                logger.LogInformation("所有分类都已有图标，跳过本次任务");
                return;
            }

            logger.LogInformation("发现 {Count} 个分类没有图标，开始生成", categoriesWithoutIcon.Count);

            // 为每个分类生成图标
            foreach (var category in categoriesWithoutIcon)
            {
                try
                {
                    await GenerateIconsForCategoryAsync(category);
                }
                catch (Exception ex)
                {
                    logger.LogError(ex, "为分类 {CategoryName}(ID:{CategoryId}) 生成图标失败",
                        category.Name, category.Id);
                }
            }

            logger.LogInformation("分类图标生成任务执行完成");
        }
        catch (Exception ex)
        {
            logger.LogError(ex, "分类图标生成任务执行出错");
            throw;
        }
        finally
        {
            _semaphore.Release();
        }
    }

    /// <summary>
    /// 为单个分类生成 5 个 SVG 图标
    /// </summary>
    private async Task GenerateIconsForCategoryAsync(TransactionCategory category)
    {
        logger.LogInformation("正在为分类 {CategoryName}(ID:{CategoryId}) 生成图标",
            category.Name, category.Id);

        var typeText = category.Type == TransactionType.Expense ? "支出" : "收入";

        var systemPrompt = """
            你是一个专业的 SVG 图标设计师，擅长创作精美、富有表现力的图标。
            请根据分类名称和类型，生成 5 个风格迥异、视觉效果突出的 SVG 图标。
            
            设计要求：
            1. 尺寸：24x24，viewBox="0 0 24 24"
            2. 色彩：使用丰富的渐变色和多色搭配，让图标更有吸引力和辨识度
               - 可以使用 <linearGradient> 或 <radialGradient> 创建渐变效果
               - 不同元素使用不同颜色，增加层次感
               - 根据分类含义选择合适的配色方案（如餐饮用暖色系、交通用蓝色系等）
            3. 设计风格：5 个图标必须风格明显不同，避免雷同
               - 第1个：扁平化风格，色彩鲜明，使用渐变
               - 第2个：线性风格，多色描边，细节丰富
               - 第3个：3D立体风格，使用阴影和高光效果
               - 第4个：卡通可爱风格，圆润造型，活泼配色
               - 第5个：现代简约风格，几何与曲线结合，优雅配色
            4. 细节丰富：不要只用简单的几何图形，添加特征性的细节元素
               - 例如：餐饮可以加刀叉、蒸汽、食材纹理等
               - 交通可以加轮胎、车窗、尾气等
               - 每个图标要有独特的视觉记忆点
            5. 图标要直观表达分类含义，让人一眼就能识别
            6. 只返回 JSON 数组格式，包含 5 个完整的 SVG 字符串，不要有任何其他文字说明
            
            重要：每个 SVG 必须是自包含的完整代码，包含所有必要的 gradient 定义。
            """;

        var userPrompt = $"""
            分类名称：{category.Name}
            分类类型：{typeText}
            
            请为这个分类生成 5 个精美的、风格各异的彩色 SVG 图标。
            确保每个图标都有独特的视觉特征，不会与其他图标混淆。
            
            返回格式（纯 JSON 数组，无其他内容）：
            ["<svg>...</svg>", "<svg>...</svg>", "<svg>...</svg>", "<svg>...</svg>", "<svg>...</svg>"]
            """;

        var response = await openAiService.ChatAsync(systemPrompt, userPrompt, timeoutSeconds: 60 * 10);

        if (string.IsNullOrWhiteSpace(response))
        {
            logger.LogWarning("AI 未返回有效的图标数据，分类: {CategoryName}", category.Name);
            return;
        }

        // 验证返回的是有效的 JSON 数组
        try
        {
            var icons = JsonSerializer.Deserialize<List<string>>(response);
            if (icons == null || icons.Count != 5)
            {
                logger.LogWarning("AI 返回的图标数量不正确（期望5个），分类: {CategoryName}", category.Name);
                return;
            }

            // 保存图标到数据库
            category.Icon = response;
            await categoryRepository.UpdateAsync(category);

            logger.LogInformation("成功为分类 {CategoryName}(ID:{CategoryId}) 生成并保存了 5 个图标",
                category.Name, category.Id);
        }
        catch (JsonException ex)
        {
            logger.LogError(ex, "解析 AI 返回的图标数据失败，分类: {CategoryName}，响应内容: {Response}",
                category.Name, response);
        }
    }
}