import { NextRequest, NextResponse } from "next/server";
import { batchTranslateWithAI, translateWithAI } from "@/lib/ai-translate";

// 单个翻译
export async function GET(request: NextRequest) {
  const { searchParams } = new URL(request.url);
  const word = searchParams.get("word");

  if (!word) {
    return NextResponse.json({ error: "缺少 word 参数" }, { status: 400 });
  }

  try {
    const translation = await translateWithAI(word);

    if (!translation) {
      return NextResponse.json({ error: "翻译失败" }, { status: 500 });
    }

    return NextResponse.json({
      word,
      english: translation,
    });
  } catch (error) {
    console.error("AI translation error:", error);
    return NextResponse.json(
      { error: String(error) },
      { status: 500 }
    );
  }
}

// 批量翻译
export async function POST(request: NextRequest) {
  try {
    const body = await request.json();
    const { words } = body as { words: string[] };

    if (!Array.isArray(words) || words.length === 0) {
      return NextResponse.json({ error: "缺少 words 参数（数组）" }, { status: 400 });
    }

    // 检查 API Key
    if (!process.env.AI_API_KEY) {
      return NextResponse.json({ error: "AI 配置未完成，请先在设置中配置 AI" }, { status: 500 });
    }

    const results = await batchTranslateWithAI(words);

    // 转换为响应格式
    const translations: Record<string, string | null> = {};
    for (const word of words) {
      translations[word] = results.get(word) || null;
    }

    return NextResponse.json({
      translations,
      total: words.length,
      success: results.size,
    });
  } catch (error) {
    console.error("AI batch translation error:", error);
    return NextResponse.json(
      { error: String(error) },
      { status: 500 }
    );
  }
}