import { callZhipuAPI } from "@/utils";
import { NextRequest, NextResponse } from "next/server";

async function getLocale(text: string) {
    console.log(`get locale of ${text}`);
    const messages = [
        {
            role: 'user', content: `
请推断以下文本的的locale，以JSON格式返回
[${text}]
结果如：
{
    "locale": "zh-CN"
}
注意：
直接返回json文本，
locale如果可能有多个，选取最可能的一个，其中使用符号"-"，
locale如果推断失败，就返回{"locale": "en-US"}
`
        }];
    try {
        const response = await callZhipuAPI(messages);
        let to_parse = response.choices[0].message.content.trim() as string;
        if (to_parse.startsWith('`')) to_parse = to_parse.slice(7, to_parse.length - 3);
        if (to_parse.length === 0) throw Error('ai啥也每说');
        return JSON.parse(to_parse);
    } catch (error) {
        console.error(error);
        return null;
    }
}

export async function GET(request: NextRequest) {
    try {
        const searchParams = request.nextUrl.searchParams;
        const text = searchParams.get('text');

        if (!text) {
            return NextResponse.json(
                { error: "查询参数错误", message: "text参数是必需的" },
                { status: 400 }
            );
        }

        const textInfo = await getLocale(text.slice(0, 30));
        if (!textInfo) {
            return NextResponse.json(
                { error: "服务暂时不可用", message: "LLM API 请求失败" },
                { status: 503 }
            );
        }

        return NextResponse.json(textInfo, { status: 200 });

    } catch (error) {
        console.error('API 错误:', error);
        return NextResponse.json(
            { error: "服务器内部错误", message: "请稍后重试" },
            { status: 500 }
        );
    }
}