/**
 * LLM 服务模块 - 使用阿里云通义千问 API 提取招标金额
 */

import { readFileSync } from 'fs';
import { join, dirname } from 'path';
import { fileURLToPath } from 'url';

const __filename = fileURLToPath(import.meta.url);
const __dirname = dirname(__filename);

// 获取 LLM 配置
function getLLMConfig() {
  try {
    const configPath = join(__dirname, '..', 'config.json');
    const configContent = readFileSync(configPath, 'utf-8');
    const config = JSON.parse(configContent);
    return config.llm || null;
  } catch (err) {
    console.error('读取 LLM 配置失败:', err.message);
    return null;
  }
}

// 检查 LLM 是否已启用
export function isLLMEnabled() {
  const config = getLLMConfig();
  return config && config.enabled && config.apiKey;
}

// 使用 LLM 提取招标金额
export async function extractBudgetWithLLM(content) {
  const config = getLLMConfig();

  if (!config || !config.enabled || !config.apiKey) {
    return null;
  }

  // 智能提取包含金额信息的段落，避免截断丢失关键信息
  const maxContentLength = 4000;
  let truncatedContent = content;

  if (content.length > maxContentLength) {
    // 查找金额关键词的位置，提取关键词周围的上下文
    const budgetKeywords = ['预算金额', '项目预算', '采购预算', '控制价', '最高限价', '招标金额', '项目金额', '合同金额', '投标报价', '中标金额', '成交金额', '中标价', '成交价'];
    const contextRadius = 200; // 关键词前后各取200字符
    const extractedContexts = [];

    for (const keyword of budgetKeywords) {
      let pos = content.indexOf(keyword);
      while (pos !== -1) {
        const start = Math.max(0, pos - contextRadius);
        const end = Math.min(content.length, pos + keyword.length + contextRadius);
        extractedContexts.push(content.substring(start, end));
        pos = content.indexOf(keyword, pos + 1);
      }
    }

    if (extractedContexts.length > 0) {
      // 有相关内容，拼接：开头部分 + 提取的上下文
      const headerContent = content.substring(0, 1500);
      const relevantContent = [...new Set(extractedContexts)].join('\n---\n'); // 去重
      truncatedContent = headerContent + '\n\n【以下为金额相关内容】\n' + relevantContent;

      if (truncatedContent.length > maxContentLength) {
        truncatedContent = truncatedContent.substring(0, maxContentLength) + '...(内容已截断)';
      }
    } else {
      // 没找到相关内容，使用原来的截断方式
      truncatedContent = content.substring(0, maxContentLength) + '...(内容已截断)';
    }
  }

  const prompt = `你是一个专业的招标文件分析助手。请从以下招标公告内容中提取预算金额信息。

要求：
1. 优先查找以下字段对应的金额：预算金额、项目预算、采购预算、预算、控制价、最高限价、招标金额、项目金额、合同金额、投标报价、中标金额、成交金额、中标价、成交价
2. 如果有多个金额，优先选择"预算金额"或"项目预算"
3. 金额统一转换为万元单位（如 70万元 = 70，700000元 = 70）
4. 严格按照 JSON 格式返回，不要添加任何其他文字

常见格式示例：
- "预算金额：70万元" → amount: 70
- "预算金额：700000元" → amount: 70
- "项目预算：70.00万元" → amount: 70

返回格式（必须是合法的 JSON）：
{"amount": 数值, "unit": "万元", "text": "原文中的金额描述"}

如果没有找到金额，返回：
{"amount": null, "unit": null, "text": null}

公告内容：
${truncatedContent}`;

  try {
    const response = await fetch(`${config.baseUrl}/chat/completions`, {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json',
        'Authorization': `Bearer ${config.apiKey}`,
      },
      body: JSON.stringify({
        model: config.model || 'qwen-turbo',
        messages: [
          {
            role: 'user',
            content: prompt,
          },
        ],
        temperature: 0.1, // 低温度，保证输出稳定
        max_tokens: 200,
      }),
      signal: AbortSignal.timeout(15000), // 15秒超时
    });

    if (!response.ok) {
      const errorText = await response.text();
      console.error('LLM API 错误:', response.status, errorText);
      return null;
    }

    const data = await response.json();
    const assistantMessage = data.choices?.[0]?.message?.content;

    if (!assistantMessage) {
      console.error('LLM 返回内容为空');
      return null;
    }

    // 解析 JSON 响应
    const jsonMatch = assistantMessage.match(/\{[\s\S]*\}/);
    if (!jsonMatch) {
      console.error('LLM 返回格式异常:', assistantMessage);
      return null;
    }

    const result = JSON.parse(jsonMatch[0]);

    if (result.amount === null || result.amount === undefined) {
      return null;
    }

    // 验证金额合理性
    const amount = parseFloat(result.amount);
    if (isNaN(amount) || amount < 0.01 || amount > 100000000) {
      console.error('LLM 提取的金额不合理:', result.amount);
      return null;
    }

    console.log(`LLM 提取金额成功: ${amount} 万元`);

    return {
      amount: amount,
      unit: '万元',
      text: result.text || `${amount}万元`,
      source: 'llm', // 标记来源
    };
  } catch (err) {
    if (err.name === 'TimeoutError') {
      console.error('LLM API 超时');
    } else {
      console.error('LLM 提取金额失败:', err.message);
    }
    return null;
  }
}

// 测试 LLM 连接
export async function testLLMConnection() {
  const config = getLLMConfig();

  if (!config || !config.apiKey) {
    return { success: false, error: '未配置 API Key' };
  }

  try {
    const response = await fetch(`${config.baseUrl}/chat/completions`, {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json',
        'Authorization': `Bearer ${config.apiKey}`,
      },
      body: JSON.stringify({
        model: config.model || 'qwen-turbo',
        messages: [
          {
            role: 'user',
            content: '请回复"连接成功"',
          },
        ],
        max_tokens: 10,
      }),
      signal: AbortSignal.timeout(10000),
    });

    if (!response.ok) {
      const errorText = await response.text();
      return { success: false, error: `API 错误: ${response.status} - ${errorText}` };
    }

    const data = await response.json();
    const reply = data.choices?.[0]?.message?.content;

    return {
      success: true,
      message: '连接成功',
      model: config.model || 'qwen-turbo',
      reply: reply,
    };
  } catch (err) {
    if (err.name === 'TimeoutError') {
      return { success: false, error: '连接超时' };
    }
    return { success: false, error: err.message };
  }
}

// 获取 LLM 状态
export function getLLMStatus() {
  const config = getLLMConfig();

  if (!config) {
    return {
      configured: false,
      enabled: false,
      model: null,
    };
  }

  return {
    configured: !!config.apiKey,
    enabled: config.enabled && !!config.apiKey,
    model: config.model || 'qwen-turbo',
    baseUrl: config.baseUrl,
  };
}
