const Router = require('koa-router');
const fetch = require('node-fetch');

const router = new Router();

const DEEPSEEK_API_KEY = 'sk-21cc85a144874d9b902edffe6af8a971';
const DEEPSEEK_API_URL = 'https://api.deepseek.com';

router.post('/generate-product', async (ctx) => {
  try {
    const { imageUrl, keywords } = ctx.request.body;

    let prompt = '你是一个专业的便利店商品管理助手。';

    if (imageUrl) {
      prompt += `\n请分析这张商品图片：${imageUrl}`;
    }

    if (keywords) {
      prompt += `\n关键词：${keywords}`;
    }

    prompt += `
请生成商品的详细信息，返回JSON格式，不要包含其他内容：
{
  "name": "商品名称（简洁明了，2-10字）",
  "category": "商品分类（请从以下选择：饮料,零食,日用品,食品,生鲜,烟酒,其他）",
  "description": "商品详细描述（50-100字，突出产品特点）",
  "suggestedPrice": 建议售价（数字）
}`;

    const response = await fetch(DEEPSEEK_API_URL, {
      method: 'POST',
      headers: {
        'Authorization': `Bearer ${DEEPSEEK_API_KEY}`,
        'Content-Type': 'application/json'
      },
      body: JSON.stringify({
        model: 'deepseek-v4-flash',
        messages: [
          {
            role: 'user',
            content: prompt
          }
        ],
        temperature: 0.7,
        max_tokens: 500
      }),
      timeout: 30000
    });

    if (!response.ok) {
      const errorText = await response.text();
      console.error('DeepSeek API Error:', response.status, errorText);

      let errorMsg = 'AI 服务调用失败';
      if (response.status === 401) {
        errorMsg = 'API Key 无效，请检查密钥配置';
      } else if (response.status === 403) {
        errorMsg = 'API 调用被拒绝，请检查账户权限';
      } else if (response.status === 429) {
        errorMsg = 'API 调用次数超限，请稍后重试';
      } else if (response.status === 503) {
        errorMsg = 'AI 服务暂时不可用，请稍后重试';
      }

      ctx.status = response.status;
      ctx.body = {
        code: response.status,
        message: errorMsg
      };
      return;
    }

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

    if (!aiResponse) {
      ctx.status = 500;
      ctx.body = {
        code: 500,
        message: 'AI 服务返回为空'
      };
      return;
    }

    const jsonMatch = aiResponse.match(/\{[\s\S]*\}/);
    if (!jsonMatch) {
      ctx.status = 500;
      ctx.body = {
        code: 500,
        message: '无法解析 AI 响应格式'
      };
      return;
    }

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

    ctx.body = {
      code: 200,
      message: '生成成功',
      data: productInfo
    };

  } catch (error) {
    console.error('生成商品信息失败:', error);

    let errorMsg = '生成失败，请稍后重试';
    if (error.message.includes('timeout')) {
      errorMsg = 'AI 服务响应超时，请检查网络或稍后重试';
    } else if (error.message.includes('ENOTFOUND')) {
      errorMsg = '无法连接到 AI 服务，请检查网络设置';
    } else if (error.message.includes('ECONNRESET')) {
      errorMsg = 'AI 服务连接中断，请稍后重试';
    }

    ctx.status = 503;
    ctx.body = {
      code: 503,
      message: errorMsg
    };
  }
});

router.post('/recognize-product', async (ctx) => {
  try {
    const { imageBase64 } = ctx.request.body;

    if (!imageBase64) {
      ctx.status = 400;
      ctx.body = {
        code: 400,
        message: '请提供商品图片'
      };
      return;
    }

    const prompt = `你是一个专业的便利店商品识别助手。请分析这张商品图片，识别出商品信息。

请返回JSON格式的商品信息，只返回一个最可能的商品，不要返回多个：
{
  "name": "商品名称（根据图片识别，如果无法确定则返回空字符串）",
  "category": "商品分类（从以下选择：饮料,零食,日用品,食品,生鲜,烟酒,其他，如果无法确定则返回空字符串）",
  "description": "商品描述（根据图片识别，突出产品特点，如果无法确定则返回空字符串）",
  "suggestedPrice": 数字（根据市场价估算，如果无法确定则返回0）,
  "confidence": 0到1之间的数字（识别置信度）
}`;

    const response = await fetch(DEEPSEEK_API_URL, {
      method: 'POST',
      headers: {
        'Authorization': `Bearer ${DEEPSEEK_API_KEY}`,
        'Content-Type': 'application/json'
      },
      body: JSON.stringify({
        model: 'deepseek-v4-flash',
        messages: [
          {
            role: 'user',
            content: [
              {
                type: 'image_url',
                image_url: {
                  url: `data:image/jpeg;base64,${imageBase64}`
                }
              },
              {
                type: 'text',
                text: prompt
              }
            ]
          }
        ],
        temperature: 0.3,
        max_tokens: 500
      }),
      timeout: 60000
    });

    if (!response.ok) {
      const errorText = await response.text();
      console.error('DeepSeek API Error:', response.status, errorText);

      let errorMsg = 'AI 服务调用失败';
      if (response.status === 401) {
        errorMsg = 'API Key 无效，请检查密钥配置';
      } else if (response.status === 403) {
        errorMsg = 'API 调用被拒绝，请检查账户权限';
      } else if (response.status === 429) {
        errorMsg = 'API 调用次数超限，请稍后重试';
      } else if (response.status === 503) {
        errorMsg = 'AI 服务暂时不可用，请稍后重试';
      }

      ctx.status = response.status;
      ctx.body = {
        code: response.status,
        message: errorMsg
      };
      return;
    }

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

    if (!aiResponse) {
      ctx.status = 500;
      ctx.body = {
        code: 500,
        message: 'AI 服务返回为空'
      };
      return;
    }

    const jsonMatch = aiResponse.match(/\{[\s\S]*\}/);
    if (!jsonMatch) {
      ctx.status = 500;
      ctx.body = {
        code: 500,
        message: '无法解析 AI 响应格式'
      };
      return;
    }

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

    ctx.body = {
      code: 200,
      message: '识别成功',
      data: productInfo
    };

  } catch (error) {
    console.error('识别商品失败:', error);

    let errorMsg = '识别失败，请稍后重试';
    if (error.message.includes('timeout')) {
      errorMsg = 'AI 服务响应超时，请检查网络或稍后重试';
    } else if (error.message.includes('ENOTFOUND')) {
      errorMsg = '无法连接到 AI 服务，请检查网络设置';
    } else if (error.message.includes('ECONNRESET')) {
      errorMsg = 'AI 服务连接中断，请稍后重试';
    }

    ctx.status = 503;
    ctx.body = {
      code: 503,
      message: errorMsg
    };
  }
});

module.exports = router.routes();