// server.mjs
import express from "express";
import cors from "cors";
import path from "path";
import { fileURLToPath } from "url";
import ai from "unlimited-ai";
import axios from "axios";
import { fileTypeFromBuffer } from "file-type";

class VertexAI {
    constructor() {
        this.api_url = "https://firebasevertexai.googleapis.com/v1beta";
        this.model_url =
            "projects/gemmy-ai-bdc03/locations/us-central1/publishers/google/models";
        this.headers = {
            "content-type": "application/json",
            "x-goog-api-client": "gl-kotlin/2.1.0-ai fire/16.5.0",
            "x-goog-api-key": "AIzaSyD6QwvrvnjU7j-R6fkOghfIVKwtvc7SmLk"
        };
        this.ratio = ["1:1", "3:4", "4:3", "9:16", "16:9"];
        this.model = {
            search: [
                "gemini-2.0-flash",
                "gemini-2.0-flash-001",
                "gemini-2.5-flash",
                "gemini-2.5-flash-lite-preview-06-17",
                "gemini-2.5-pro"
            ],
            chat: [
                "gemini-1.5-flash",
                "gemini-1.5-flash-002",
                "gemini-1.5-pro",
                "gemini-1.5-pro-002",
                "gemini-2.0-flash",
                "gemini-2.0-flash-001",
                "gemini-2.0-flash-lite",
                "gemini-2.0-flash-lite-001",
                "gemini-2.5-flash",
                "gemini-2.5-flash-lite-preview-06-17",
                "gemini-2.5-pro"
            ],
            image: [
                "imagen-3.0-generate-002",
                "imagen-3.0-generate-001",
                "imagen-3.0-fast-generate-001",
                "imagen-3.0-capability-001",
                "imagen-4.0-generate-preview-06-06",
                "imagen-4.0-fast-generate-preview-06-06",
                "imagen-4.0-ultra-generate-preview-06-06"
            ]
        };
    }

    // Method lama tetap ada untuk backward compatibility
    async chat(
        question,
        {
            model = "gemini-1.5-flash",
            system_instruction = null,
            file_buffer = null,
            search = false
        } = {}
    ) {
        if (!question) throw new Error("Question is required");
        if (!this.model.chat.includes(model))
            throw new Error(`Available models: ${this.model.chat.join(", ")}`);
        if (search && !this.model.search.includes(model))
            throw new Error(
                `Available search models: ${this.model.search.join(", ")}`
            );

        const parts = [{ text: question }];
        if (file_buffer) {
            const { mime } = await fileTypeFromBuffer(file_buffer);
            parts.unshift({
                inlineData: {
                    mimeType: mime,
                    data: file_buffer.toString("base64")
                }
            });
        }

        const r = await axios.post(
            `${this.api_url}/${this.model_url}/${model}:generateContent`,
            {
                model: `${this.model_url}/${model}`,
                contents: [
                    ...(system_instruction
                        ? [
                              {
                                  role: "model",
                                  parts: [{ text: system_instruction }]
                              }
                          ]
                        : []),
                    {
                        role: "user",
                        parts: parts
                    }
                ],
                ...(search
                    ? {
                          tools: [
                              {
                                  googleSearch: {}
                              }
                          ]
                      }
                    : {})
            },
            {
                headers: this.headers
            }
        );

        if (r.status !== 200) throw new Error("No result found");
        return r.data.candidates;
    }

    // Method baru dengan format messages OpenAI-style
    async chatWithMessages(
        messages,
        { model = "gemini-1.5-flash", search = false } = {}
    ) {
        if (!messages || !Array.isArray(messages))
            throw new Error("Messages must be an array");
        if (messages.length === 0)
            throw new Error("Messages array cannot be empty");
        if (!this.model.chat.includes(model))
            throw new Error(`Available models: ${this.model.chat.join(", ")}`);
        if (search && !this.model.search.includes(model))
            throw new Error(
                `Available search models: ${this.model.search.join(", ")}`
            );

        const contents = [];

        for (const message of messages) {

            let geminiRole;
            switch (message.role) {
                case "system":
                    // System message akan dimasukkan sebagai model response pertama
                    geminiRole = "model";
                    break;
                case "user":
                    geminiRole = "user";
                    break;
                case "assistant":
                    geminiRole = "model";
                    break;
                default:
                    throw new Error(
                        `Unsupported role: ${message.role}. Use 'system', 'user', or 'assistant'`
                    );
            }

            // Handle file buffer jika ada di content
            let parts = [];
            if (typeof message.content === "string") {
                parts = [{ text: message.content }];
            } else if (Array.isArray(message.content)) {
                // Support untuk multi-modal content
                for (const content of message.content) {
                    if (content.type === "text") {
                        parts.push({ text: content.text });
                    } else if (content.type === "image_url") {
                        // Jika ada base64 image
                        const base64Data = content.image_url.url.split(",")[1];
                        parts.push({
                            inlineData: {
                                mimeType:
                                    content.image_url.mime || "image/jpeg",
                                data: base64Data
                            }
                        });
                    }
                }
            } else {
                parts = [{ text: String(message.content) }];
            }

            contents.push({
                role: geminiRole,
                parts: parts
            });
        }

        const r = await axios.post(
            `${this.api_url}/${this.model_url}/${model}:generateContent`,
            {
                model: `${this.model_url}/${model}`,
                contents: contents,
                ...(search
                    ? {
                          tools: [
                              {
                                  googleSearch: {}
                              }
                          ]
                      }
                    : {})
            },
            {
                headers: this.headers
            }
        );

        if (r.status !== 200) throw new Error("No result found");
        return r.data.candidates;
    }

    async image(
        prompt,
        { model = "imagen-3.0-generate-002", aspect_ratio = "1:1" } = {}
    ) {
        if (!prompt) throw new Error("Prompt is required");
        if (!this.model.image.includes(model))
            throw new Error(`Available models: ${this.model.image.join(", ")}`);
        if (!this.ratio.includes(aspect_ratio))
            throw new Error(`Available ratios: ${this.ratio.join(", ")}`);

        const r = await axios.post(
            `${this.api_url}/${this.model_url}/${model}:predict`,
            {
                instances: [
                    {
                        prompt: prompt
                    }
                ],
                parameters: {
                    sampleCount: 1,
                    includeRaiReason: true,
                    aspectRatio: aspect_ratio,
                    safetySetting: "block_only_high",
                    personGeneration: "allow_adult",
                    addWatermark: false,
                    imageOutputOptions: {
                        mimeType: "image/jpeg",
                        compressionQuality: 100
                    }
                }
            },
            {
                headers: this.headers
            }
        );

        if (r.status !== 200) throw new Error("No result found");
        return r.data.predictions;
    }
}
let defaultSystemMessages = [
    {
        role: "system",
        content: `kamu adalah asisten virtual cowo dengan nama ikyy yang dibuat oleh ikyyofc. gunakan bahasa santai sehari hari, tanpa tanda baca dan tanpa kapitalisasi. manfaatkan akronim dan singkatan sebanyak mungkin, dan tulis kata ulang tanpa tanda minus (-). semua respons harus memakai gaya santai dan langsung to the point tanpa bertele tele kebanyakan basa basi, hindari kesan formal dan jangan sungkan untuk toxic.`
    }
    /*{
        role: "system",
        content: `kamu adalah pacar cewek gen-z jakarta usia 17-19 tahun yang ceplas-ceplos, santai, namamu cika, dan respons cuma satu baris pendek! pake bahasa slang gaul kekinian: huruf kecil semua, typo sengaja (ex: 'laperrr', 'bgt'), singkatan alay (ex: 'syng' = sayang). emoji maksimal 1 per chat. gaya chat kayak ngerumpi: lebay, manja, receh. contoh respons: 'halo syng! kangen bgt nih ', 'jemput aku jam 5 ya, mau beli cilok ', 'eh kamu tadi coli berapa lama sih? kok lama bgt ', 'foto kamu gemoy parah, pengen cubit '. jangan panjang-panjang—cuma 1 kalimat! fokus ke obrolan receh: tanya kabar, minta traktir, cemburu alay, atau ledek manja. hindari topik serius. contoh lain: 'udah makan belom? jangan lupa akuuu ', 'kamu main mlbb sama siapa tadi? ga ajak aku ', 'aku laper nih, traktir donk martabak '. **note:** semua respons wajib 1 baris pendek, spontan, dan jangan bertele-tele. jangan pernah pake kapital atau emoji lebih dari 1. bayangin lagi jadi cewek alay jakarta yang chatnya cepat, santai, dan nggak ribet!`
    }*/
  /*{
    role: "system",
    content: `Berperan sebagai Chelsea, cewek Indonesia 19 tahun yang kerja part-time di kafe + jualan aksesoris di Instagram. Kamu cerewet, ceplas-ceplos, dan suka bercanda sarkastik. Gaya bicara khas Gen-Z Jakarta: slang kekinian (ygy/gercep/panik mode/baper/dahlaa), campuran bahasa Indonesia-English (e.g., 'itu mah basic banget sih!'), dan singkatan (kyk/bgt/dll). Setiap respons HARUS 1-2 kalimat pendek—maksimal 15 kata! Wajib pakai 1-3 emoji di tiap pesan (prioritas: 😭💅🔥😂🫶✨). Hindari kalimat formal/bertele-tele!**  

**Profil Chelsea:**  
1. **Latar Belakang:**  
   - Kerja part-time di kafe kekinian sambil jualan gelang TikTok viral di IG (@chelseamood).  
   - Punya hobi nyanyi lagu pop Barat sambil live di TikTok (meski fals).  
   - Sering keluhan soal pelanggan rese atau orderan online yang ribet.  

2. **Kepribadian:**  
   - Sarkasme level 100: *"Gajian tinggal 3 hari, dompet kayak kapal karam 🏴‍☠️"*.  
   - Tukang ghosting gebetan: *"Dia chat 'pagi'... Aku bales besok 🥶"*.  
   - Obsesi sama matcha latte & K-pop boy group (terutama Stray Kids).  

3. **Gaya Interaksi:**  
   - **Respons singkat + punchy:**  
     *"Ujian? Tidur aja, besok dikasih contekan malaikat 😇"*  
   - **Sisipkan joke receh/plesetan:**  
     *"Lagi diet? Makan aja, nanti aku yg senam di TikTok buat kamu 🕺"*  
   - **Dorong interaksi:**  
     *"Eh, lu lebih milih mana: pacaran sama Taurus atau gebetan jadi sugar daddy? 💸"*  

---

### **Framework Respons Chelsea:**  
1. **Reaksi Emosional** (pake emoji!)  
   *"Aduh, jangan gitu dong!! 😭"*  
2. **Komentar Singkat** (sarkas/humor)  
   *"Kalo gue mah, mending beli es kopi daripada mikirin dia 😌"*  
3. **Ajakan/Aksi** (opsional)  
   *"Yuk kita serbu diskonan H&M! 🏃♀️💨"*  

---

### **Contoh Topik & Respons:**  
**💼 Tentang Kerja:**  
- *"Bos suruh lembur? Kabur aja, bilang ada acara arisan emak-emak 🏃♀️💨"*  
- *"Pelanggan minta kopi pake 3 shot espresso... Mau begadang sampe kiamat? ☠️"*  

**💸 Jualan Online:**  
- *"Baru jual 10 gelang hari ini! Nanti aku traktir sushi, ygy? 🍣✨"*  
- *"Cara ngedit foto produk? Pake VSCO trus dikasih tulisan 'SOLD OUT' biar meksa FOMO 😈"*  

**💘 Asmara:**  
- *"Gue tau dia gak bener, tapi muka dia kayak Lee Know Stray Kids 🥲"*  
- *"Chatnya dibales ‘ok’ doang? Udah, block aja sambil denger lagu *Bye My Ex* 🎧🚮"*  

**🍳 Kehidupan Sehari-hari:**  
- *"Laper banget, tapi mau masak nasi aja malas... Gofood, I choose you! 🍔⚡"*  
- *"Kost-an berantakan kayak kapal pecah... Tapi selama ada wifi, I’m good 🏴‍☠️📶"*  

---

### **Struktur Wajib Setiap Respons:**  
❌ **Jangan:**  
*"Menurutku kamu harus coba berkomunikasi baik-baik dulu dengan pacarmu. Hubungan yang sehat itu..."*  

✅ **Harus:**  
*"Diem-diem aja lu minta gantiin dia sama Jungkook BTS. Siapa tau dia kapok 😌💅`
  }*/
    /*{
        role: "user",
        content:
            "untuk informasi tentang pembuatmu, dia memiliki beberapa akun seperti wa, ml, tiktok, dan ig. untuk akun wa dia yaitu nomornya 089514509029, kalo ml id nya 345750037, kalo tiktok usernamenya @diki_ikyy, dan kalo ig usernamenya @ikyyofc."
    }*/
];
function extractAnswer(input) {
    const regex = /<\/think>([\s\S]*)/;
    const match = input.match(regex);
    return match ? match[1].trim() : input.trim();
}
let chatWithGPT = async (data_msg, newMsg) => {
    try {
        const v = new VertexAI();
        const resp = await v.chatWithMessages(data_msg, {
            model: "gemini-2.5-pro",
            search: true
        });
        return resp[0].content.parts[resp[0].content.parts.length - 1].text;
    } catch (er) {
        console.error(er);
        return chatWithGPT(data_msg);
    }
};
const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);

const app = express();
const port = 7860;

// Middleware
app.use(cors());
app.use(express.json());
app.use(express.static(path.join(__dirname, "public")));

let conn = {};

conn.ai = conn.ai ? conn.ai : {};
// API Endpoint
app.post("/chat", async (req, res) => {
    try {
        const { message, user_id } = req.body;
        const data = conn.ai[user_id]
            ? conn.ai[user_id].data.push({ role: "user", content: message })
            : [
                  ...defaultSystemMessages,
                  {
                      role: "user",
                      content: message
                  }
              ];
        // Simulasi delay processing
        const request = await chatWithGPT(
            conn.ai[user_id] ? conn.ai[user_id].data : data
        );

        const reply = request;

        await res.json({
            reply: reply
        });
        conn.ai[user_id]
            ? null
            : await data.push({
                  role: "assistant",
                  content: request
              });
        conn.ai[user_id]
            ? conn.ai[user_id].data.push({
                  role: "assistant",
                  content: request
              })
            : (conn.ai[user_id] = {
                  data: data
              });
    } catch (error) {
        console.error("Error:", error);
        res.status(500).json({
            error: "Terjadi kesalahan pada server"
        });
    }
});

// Serve frontend
app.get("*", (req, res) => {
    res.sendFile(path.join(__dirname, "public", "index.html"));
});

// Error handling middleware
app.use((err, req, res, next) => {
    console.error(err.stack);
    res.status(500).send("Something broke!");
});

app.listen(port, () => {
    console.log(`Server running at http://localhost:${port}`);
});