// code from goatstore
// api: https://store.agi.bd
const axios = require("axios");

const API_URL = "https://vireonix.ai/v1/chat/completions";

// ==========================================
// 🤖 SHAKIB AI
// বাংলা + English + Banglish
// Reply Chain + Conversation Memory
// ==========================================

const chatHistory = new Map();

const MAX_HISTORY = 12;
const HISTORY_TIMEOUT = 30 * 60 * 1000;


// ==========================================
// ⌨️ Typing Indicator
// ==========================================
async function typing(api, threadID) {
  try {
    if (api.sendTypingIndicator) {
      await api.sendTypingIndicator(threadID, true);

      setTimeout(async () => {
        try {
          await api.sendTypingIndicator(threadID, false);
        } catch {}
      }, 1200);
    }
  } catch {}
}


// ==========================================
// 🧠 Conversation Memory
// ==========================================
function getHistory(key) {
  const data = chatHistory.get(key);

  if (!data) return [];

  if (Date.now() - data.updatedAt > HISTORY_TIMEOUT) {
    chatHistory.delete(key);
    return [];
  }

  return data.messages || [];
}


function saveHistory(key, messages) {
  chatHistory.set(key, {
    messages: messages.slice(-MAX_HISTORY),
    updatedAt: Date.now()
  });
}


// ==========================================
// 🔗 Proper Reply Chain
// ==========================================
function setReply(info, event, historyKey) {
  if (!info?.messageID) return;

  global.GoatBot.onReply.set(info.messageID, {
    commandName: "baby",
    messageID: info.messageID,
    author: event.senderID,
    threadID: event.threadID,
    type: "reply",
    historyKey
  });
}


// ==========================================
// 📩 Reply Helper
// ==========================================
async function sendBotReply(message, text) {
  try {
    return await message.reply(text);
  } catch (error) {
    console.error("❌ Reply Error:", error.message);

    // Fallback
    try {
      return await message.send(text);
    } catch {
      return null;
    }
  }
}


// ==========================================
// 😂 Random Replies
// ==========================================
const randomReplies = [
  "𝐀𝐬𝐬𝐚𝐥𝐚𝐦𝐮 𝐰𝐚𝐥𝐚𝐢𝐤𝐮𝐦 ♥",
  "বলেন sir__😌",
  "𝐁𝐨𝐥𝐨 𝐣𝐚𝐧 𝐤𝐢 𝐤𝐨𝐫𝐭𝐞 𝐩𝐚𝐫𝐢 𝐭𝐨𝐦𝐫 𝐣𝐨𝐧𝐧𝐨 🐸",
  "𝐋𝐞𝐛𝐮 𝐤𝐡𝐚𝐰 𝐝𝐚𝐤𝐭𝐞 𝐝𝐚𝐤𝐭𝐞 𝐭𝐨 𝐡𝐚𝐩𝐚𝐲 𝐠𝐞𝐬𝐨 🫴🍋",
  "এতো ডাকিস না তারেক জিয়ার সাথে মিটিংয়ে আছি 😒🖐🏻",
  "ডিসটার্ব করিস না জামাই আদর করতেছে-🌚💋",
  "আমাকেই তো ডাকবি সারাদিন আমার জান এর যে বউ লাগব একটা খুজে দে 😌",
  "একা বাঁচতে শিখো দেখবে পৃথিবী অনেক সুন্দর ✨",
  "──‎ 𝐇𝐮𝐌..? 👉👈",
  "আম গাছে আম নাই ঢিল কেন মারো, তোমার সাথে প্রেম নাই জান্নাত কেন ডাকো 😒🐸",
  "এত ডাকিস না এমন থাপ্পর দিমু পেন্টে মুইতা দিবি-😾👋🏻 🤣",
  "𝐓𝐫𝐮𝐬𝐭 𝐦𝐞 𝐢𝐚𝐦 SHAKIB 𝐟𝐫𝐨𝐦 MAYA JANNAT🧃",
  "𝗛𝗲𝘆 𝘅𝗮𝗻 𝗶𝗮𝗺 MAYA JANNAT ✨",
  "𝐓𝐨𝐫 𝐣𝐧𝐧𝐨 𝐛𝐬𝐢 𝐚𝐜𝐡𝐢, 𝐣𝐥𝐝𝐢 𝐛𝐨𝐥 𝐤𝐢 𝐝𝐫𝐤𝐚𝐫 ✨",
  "আমার এডমিনের সাথে প্রেম করলে দশ টাকা দিমু 🌚🥀",
  "কিরে তোর কি খেয়ে দেয়ে কাজ নাই সারাদিন জান্নাত জান্নাত করিস কেন..? 😙🤏",
  "𝙜𝙤𝙥 𝙜𝙤𝙥 𝙜𝙤𝙥 🙊",
"ওনেক দিন পরে ডাকলা তোমাকে ওনেক মিছ করছি তুমি কেমন আছো তুমি😑👍"
];


// ==========================================
// 🧠 AI Request
// ==========================================
async function askAI(text, history = []) {
  try {

    const messages = [
      {
        role: "system",
        content:
          "You are ARIYAN AI, a friendly Messenger chatbot. " +
          "You understand Bangla, English and Banglish. " +
          "If the user writes Bangla, answer naturally in Bangla. " +
          "If the user writes English, answer naturally in English. " +
          "If the user mixes Bangla and English, reply naturally in the same style. " +
          "Remember previous conversation context when relevant. " +
          "Keep casual replies reasonably short."
      },

      ...history,

      {
        role: "user",
        content: text
      }
    ];


    const response = await axios.post(
      API_URL,
      {
        model: "auto",
        messages
      },
      {
        headers: {
          "Content-Type": "application/json"
        },
        timeout: 30000
      }
    );


    const answer =
      response.data?.choices?.[0]?.message?.content;


    if (!answer) return null;

    return answer.trim();

  } catch (error) {

    console.error(
      "❌ SHAKIB AI ERROR:",
      error.response?.status,
      error.response?.data || error.message
    );

    return null;
  }
}


// ==========================================
// 📦 COMMAND
// ==========================================
module.exports = {

  config: {
    name: "baby",

    version: "14.0",

    author: "ARIYAN AHMED SABBIR",

    countDown: 2,

    role: 0,

    shortDescription: {
      en: "Chat with ARIYAN AI"
    },

    longDescription: {
      en:
        "Bangla + English AI with reply chain and conversation memory"
    },

    category: "AI",

    guide: {
      en:
        "{pn} hello\n" +
        "{pn} কেমন আছো\n" +
        "{pn} how are you"
    },

    aliases: [
      "bby",
      "bbe",
      "babe",
      "sam",
      "mari",
      "maria",
      "hippi",
      "xan",
      "bbz"
    ]
  },


  // ==========================================
  // ▶️ ON START
  // ==========================================
  onStart: async function ({
    api,
    event,
    args,
    message
  }) {

    const text =
      args.join(" ").trim();

    const historyKey =
      `${event.threadID}_${event.senderID}`;


    // ========================================
    // শুধু baby লিখলে
    // ========================================
    if (!text) {

      const reply =
        randomReplies[
          Math.floor(
            Math.random() *
            randomReplies.length
          )
        ];


      const info =
        await sendBotReply(
          message,
          reply
        );


      setReply(
        info,
        event,
        historyKey
      );

      return;
    }


    await typing(
      api,
      event.threadID
    );


    const history =
      getHistory(historyKey);


    const answer =
      await askAI(
        text,
        history
      );


    if (!answer) {

      await sendBotReply(
        message,
        "⚠️ SHAKIB AI এখন উত্তর দিতে পারছে না। একটু পরে আবার চেষ্টা করো।"
      );

      return;
    }


    // ========================================
    // 🧠 Memory Update
    // ========================================
    saveHistory(
      historyKey,
      [
        ...history,

        {
          role: "user",
          content: text
        },

        {
          role: "assistant",
          content: answer
        }
      ]
    );


    // ========================================
    // ⭐ USER MESSAGE-এর REPLY
    // ========================================
    const info =
      await sendBotReply(
        message,
        `🤖 SHAKIB AI\n\n${answer}`
      );


    // পরের chain
    setReply(
      info,
      event,
      historyKey
    );
  },


  // ==========================================
  // 💬 ON REPLY
  // ==========================================
  onReply: async function ({
    api,
    event,
    Reply,
    message
  }) {

    const text =
      event.body?.trim();


    if (!text) return;


    const historyKey =
      Reply?.historyKey ||
      `${event.threadID}_${event.senderID}`;


    await typing(
      api,
      event.threadID
    );


    const history =
      getHistory(historyKey);


    const answer =
      await askAI(
        text,
        history
      );


    if (!answer) {

      await sendBotReply(
        message,
        "⚠️ উত্তর দিতে একটু সমস্যা হচ্ছে 😵‍💫"
      );

      return;
    }


    // ========================================
    // 🧠 Memory Update
    // ========================================
    saveHistory(
      historyKey,
      [
        ...history,

        {
          role: "user",
          content: text
        },

        {
          role: "assistant",
          content: answer
        }
      ]
    );


    // ========================================
    // ⭐ USER-এর REPLY MESSAGE-এর REPLY
    // ========================================
    const info =
      await sendBotReply(
        message,
        `🤖 SHAKIB AI\n\n${answer}`
      );


    // Chain continue
    setReply(
      info,
      event,
      historyKey
    );
  },


  // ==========================================
  // 👀 ON CHAT
  // ==========================================
  onChat: async function ({
    api,
    event,
    message
  }) {

    const body =
      event.body?.trim();


    if (!body) return;


    const lower =
      body.toLowerCase();


    const triggers = [
      "baby",
      "maya",
      "jannat",
      "bow",
      "shakib",
      "noman",
      "kolija",
      "nusrat",
      "xan",
      "bbz",
      "sadiya",
      "bot"
    ];


    const prefixes = [
      "baby ",
      "bby ",
      "bbe ",
      "babe ",
      "sam ",
      "mari ",
      "maria ",
      "hippi ",
      "xan ",
      "bbz ",
      "মারিয়া ",
      "bot "
    ];


    // ========================================
    // শুধু trigger
    // ========================================
    if (triggers.includes(lower)) {

      const reply =
        randomReplies[
          Math.floor(
            Math.random() *
            randomReplies.length
          )
        ];


      const historyKey =
        `${event.threadID}_${event.senderID}`;


      const info =
        await sendBotReply(
          message,
          reply
        );


      setReply(
        info,
        event,
        historyKey
      );

      return;
    }


    // ========================================
    // Prefix Detect
    // ========================================
    let userMessage = null;


    for (const prefix of prefixes) {

      if (lower.startsWith(prefix)) {

        userMessage =
          body
            .slice(prefix.length)
            .trim();

        break;
      }
    }


    if (!userMessage) return;


    // ========================================
    // শুধু baby / bot
    // ========================================
    if (!userMessage.length) {

      const reply =
        randomReplies[
          Math.floor(
            Math.random() *
            randomReplies.length
          )
        ];


      const historyKey =
        `${event.threadID}_${event.senderID}`;


      const info =
        await sendBotReply(
          message,
          reply
        );


      setReply(
        info,
        event,
        historyKey
      );

      return;
    }


    // ========================================
    // 🧠 AI
    // ========================================
    const historyKey =
      `${event.threadID}_${event.senderID}`;


    await typing(
      api,
      event.threadID
    );


    const history =
      getHistory(historyKey);


    const answer =
      await askAI(
        userMessage,
        history
      );


    if (!answer) {

      await sendBotReply(
        message,
        "⚠️ SHAKIB AI এখন একটু ব্যস্ত 😵‍💫"
      );

      return;
    }


    // ========================================
    // 🧠 Memory
    // ========================================
    saveHistory(
      historyKey,
      [
        ...history,

        {
          role: "user",
          content: userMessage
        },

        {
          role: "assistant",
          content: answer
        }
      ]
    );


    // ========================================
    // ⭐ ORIGINAL USER MESSAGE-এর REPLY
    // ========================================
    const info =
      await sendBotReply(
        message,
        `🤖 SHAKIB AI\n\n${answer}`
      );


    // Chain continue
    setReply(
      info,
      event,
      historyKey
    );
  }
};