/**
 * 更新数据库中的 User Prompt Template
 * 将旧版本（包含"要求"和"输出格式"）更新为新版本（只包含任务描述和变量）
 * 
 * 使用方法：
 * cd backend
 * npx ts-node scripts/update-prompt-templates.ts
 */

import prisma from '../src/utils/prisma';
import { promptConfigService } from '../src/services/promptConfigService';

async function updatePromptTemplates() {
  try {
    console.log('🔄 开始更新数据库中的 User Prompt Template...\n');

    const types: ('summary' | 'outline' | 'content')[] = ['summary', 'outline', 'content'];

    for (const promptType of types) {
      console.log(`\n📝 处理类型: ${promptType}`);

      // 获取数据库中的配置
      const config = await prisma.aIPromptConfig.findUnique({
        where: { promptType },
      });

      if (!config) {
        console.log(`  ⚠️  数据库中没有 ${promptType} 类型的配置，跳过`);
        continue;
      }

      // 获取新的默认配置
      const defaultConfig = promptConfigService.getPromptConfig(promptType);
      const newUserPromptTemplate = defaultConfig.userPromptTemplate;

      // 检查是否需要更新
      const oldUserPromptTemplate = config.userPromptTemplate || '';
      
      // 检查是否包含"要求"或"输出格式"（旧版本的特征）
      const hasOldFormat = oldUserPromptTemplate.includes('要求：') || 
                           oldUserPromptTemplate.includes('输出格式');

      if (!hasOldFormat && oldUserPromptTemplate.trim() === newUserPromptTemplate.trim()) {
        console.log(`  ✅ ${promptType} 已经是新版本，无需更新`);
        continue;
      }

      console.log(`  📋 旧版本 User Prompt Template:`);
      console.log(`     ${oldUserPromptTemplate.substring(0, 100)}...`);
      console.log(`  📋 新版本 User Prompt Template:`);
      console.log(`     ${newUserPromptTemplate}`);

      // 更新数据库
      await prisma.aIPromptConfig.update({
        where: { promptType },
        data: {
          userPromptTemplate: newUserPromptTemplate,
          version: config.version + 1,
        },
      });

      console.log(`  ✅ ${promptType} 已更新到新版本`);
    }

    console.log('\n✅ 所有配置更新完成！');
    console.log('\n💡 提示：服务器会自动重新加载配置到内存缓存');

  } catch (error: any) {
    console.error('❌ 更新失败:', error.message);
    process.exit(1);
  } finally {
    await prisma.$disconnect();
  }
}

// 执行更新
updatePromptTemplates();