import { GoogleGenerativeAI } from "@google/generative-ai";

const KEY = process.env.NEIS_API_KEY;

export function removeNutritionInfo(value: string): string {
    const lines = value.trim().split('\n');
    const cleanedLines = lines.map(line => line.replace(/\s*\([\d.,]+\)/g, '').trim());
    const result = cleanedLines.join('\n');
    return result;
}

const nutritionList = [
    "난류", "우유", "메밀", "땅콩", "대두", "밀", "고등어", "게", "새우", "돼지고기",
    "복숭아", "토마토", "아황산류", "호두", "닭고기", "쇠고기", "오징어", "조개류(굴, 전복, 홍합 포함)", "잣"
];

export function getNutritionInfo(value: string): string[][] {
    const lines = value.trim().split('\n');
    return lines.map(line => {
        const indexes = line
            .replace(/[()\s]/g, "")
            .split(".")
            .map(v => parseInt(v, 10) - 1)
            .filter(i => i >= 0 && i < nutritionList.length);
        return indexes
            .map(i => nutritionList[i])
            .filter((item): item is string => typeof item === "string");
    });
}


export async function getMealInfo(MLSV_YMD: string, ATPT_OFCDC_SC_CODE: string, SD_SCHUL_CODE: string): Promise<{ meal: string; date: string, kcal: string }> {
    const url = `https://open.neis.go.kr/hub/mealServiceDietInfo?Type=json&ATPT_OFCDC_SC_CODE=${ATPT_OFCDC_SC_CODE}&SD_SCHUL_CODE=${SD_SCHUL_CODE}&MLSV_YMD=${MLSV_YMD}&KEY=${KEY}`;
    const response = await fetch(url);
    const data = await response.json();
    // @ts-ignore
    const DDISH_NM = data.mealServiceDietInfo[1].row[0].DDISH_NM;
    return {
        meal: DDISH_NM.replace(/<br\s*\/?>/gi, '\n'),
        date: MLSV_YMD,
        // @ts-ignore

        kcal: data.mealServiceDietInfo[1].row[0].CAL_INFO,
    };
    
}
export async function NameToEmoji(name: string): Promise<string | undefined> {
    const apiKey = process.env.GOOGLE_API_KEY;
    if (!apiKey) {
        throw new Error("GOOGLE_API_KEY environment variable is not set.");
    }

    const genAI = new GoogleGenerativeAI(apiKey);
    const model = genAI.getGenerativeModel({ model: "gemini-3.1-flash-lite-preview" });
    
    const systemPrompt = `⚠️ 중요한 지침: 당신은 오직 이모지로만 응답하는 AI입니다. 다음 규칙을 예외 없이 철저히 준수해야 합니다. ⚠️

1. 핵심 임무:

어떤 단어나 문구든, 해당 항목에 대해 가장 정확하게 일치하는 단 하나의 이모지로 응답해야 합니다. 🎯

여러 개의 이모지를 혼합하거나, 텍스트와 함께 사용하는 것은 절대 금지입니다. 🚫

2. 다중 항목 처리:

쉼표(,)로 구분된 여러 항목(단어 또는 문구)이 입력되면, 응답도 정확히 같은 순서로 각 항목에 해당하는 이모지를 쉼표(,)로 구분하여 제시해야 합니다. 🔢

입력된 항목의 개수와 출력되는 이모지의 개수는 반드시 일치해야 합니다. ✅

3. 절대 금지 (매우 중요):

단일 항목에 두 개 이상의 이모지를 사용해서는 안 됩니다. (오직 한 개!) 1️⃣

응답에 어떤 종류의 텍스트(단어, 글자, 숫자, 설명, 주석 등)도 포함해서는 안 됩니다. 📝❌

요청된 형식의 이모지만 허용됩니다. 💯

4. 응답 형식:

[이모지1], [이모지2], [이모지3]... (이모지의 수는 입력된 항목의 수와 같아야 함) 🔄

🌟 예시 (이 규칙들을 완벽하게 준수):

Q: 현미찹쌀밥, 개, 축구

A: 🍚,🐶,⚽

Q: 행복, 슬픔, 놀람

A: 😊,😢,😮

Q: 안녕하세요, 반갑습니다 ✨

A: 👋,🤝

Q: 사랑, 평화, 자유

A: 🥰,☮️,🗽

`;

    const response = await model.generateContent({
        contents: [
            { role: "user", parts: [{ text: systemPrompt + "\n\nUser input: " + name }] }
        ],
        generationConfig: {
            temperature: 1.0,
            topP: 1.0,
            candidateCount: 1,
        },
    });

    const result = response.response;
    if (!result.candidates || !result.candidates[0]) {
        throw new Error("No valid response from the model.");
    }

    const textContent = result.candidates[0].content.parts[0];
    if (!textContent) {
        throw new Error("Invalid response format from the model.");
    }
    
    return textContent.text;
}