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

module.exports = {
  config: {
    name: "girlfriend",
    version: "2.0",
    author: "🔰𝐑𝐀𝐇𝐀𝐓 𝐈𝐒𝐋𝐀𝐌🔰",
    countDown: 5,
    role: 2,
    shortDescription: {
      en: "Image generate"
    },
    longDescription: {
      en: "Generate girlfriend image"
    },
    category: "Image",
    guide: {
      en: "{pn} @mention\n{pn} UID\n{pn} profile link\nReply to a message and use {pn}"
    }
  },

  onStart: async function ({ api, event, args }) {

    const API_JSON_URL =
      "https://raw.githubusercontent.com/Rahat-Islam10/-Rahat-Boss-/refs/heads/main/api.json";

    // Get UID by full name
    async function getUIDByFullName(threadID, body) {
      if (!body || !body.includes("@")) return null;

      const match = body.match(/@(.+)/);
      if (!match) return null;

      const targetName = match[1]
        .trim()
        .toLowerCase()
        .replace(/\s+/g, " ");

      const threadInfo = await api.getThreadInfo(threadID);
      const users = threadInfo.userInfo || [];

      const user = users.find((u) => {
        if (!u.name) return false;

        const fullName = u.name
          .trim()
          .toLowerCase()
          .replace(/\s+/g, " ");

        return fullName === targetName;
      });

      return user ? user.id : null;
    }

    // Get API list
    async function getApiList(commandName) {
      const res = await axios.get(API_JSON_URL, {
        timeout: 15000
      });

      const data = res.data || {};
      const cmdData = data[commandName];

      if (!cmdData || !cmdData.api) {
        throw new Error(
          `❌ "${commandName}" API পাওয়া যায়নি`
        );
      }

      const apiList = [
        cmdData.api,
        ...(cmdData.backupApis || [])
      ].filter(Boolean);

      if (!apiList.length) {
        throw new Error(
          `❌ "${commandName}" API পাওয়া যায়নি`
        );
      }

      return apiList;
    }

    // Generate image with API fallback
    async function generateFrameWithFallback({
      senderID,
      mention,
      credit,
      apiList
    }) {
      let lastError = null;

      for (const baseApi of apiList) {

        const cleanBase = baseApi.replace(/\/+$/, "");

        const apiUrl =
          `${cleanBase}/api/frame` +
          `?type=girlfriend` +
          `&senderId=${senderID}` +
          `&mentionId=${mention}` +
          `&credit=${encodeURIComponent(credit)}`;

        try {
          const response = await axios.get(apiUrl, {
            timeout: 30000,
            responseType: "json"
          });

          const data = response.data;

          if (data.image && data.captionTemplate) {

            const imageBuffer = Buffer.from(
              data.image,
              "base64"
            );

            return {
              imageBuffer,
              captionTemplate: data.captionTemplate
            };
          }

          throw new Error("❌ API Error");

        } catch (error) {

          lastError = error;

          if (
            error.response?.status === 401 &&
            error.response?.data?.error
          ) {
            throw new Error(
              error.response.data.error
            );
          }
        }
      }

      throw (
        lastError ||
        new Error("❌ API কাজ করছে না")
      );
    }

    try {

      let mention;
      let mentionName;

      // =========================
      // REPLY USER
      // =========================
      if (
        event.messageReply &&
        event.messageReply.senderID
      ) {

        mention =
          event.messageReply.senderID;
      }

      // =========================
      // MENTION USER
      // =========================
      else if (
        event.mentions &&
        Object.keys(event.mentions).length > 0
      ) {

        mention =
          Object.keys(event.mentions)[0];
      }

      // =========================
      // ARGUMENT
      // =========================
      else if (args && args.length > 0) {

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

        // Facebook profile link
        if (input.includes(".com/")) {

          mention =
            await api.getUID(input);
        }

        // @name
        else if (input.includes("@")) {

          mention =
            Object.keys(
              event.mentions || {}
            )[0];

          if (!mention) {

            mention =
              await getUIDByFullName(
                event.threadID,
                input
              );
          }
        }

        // UID
        else {

          mention = args[0];
        }
      }

      // =========================
      // NO USER
      // =========================
      else {

        return api.sendMessage(
          "❌ 𝗣𝗹𝗲𝗮𝘀𝗲 𝗺𝗲𝗻𝘁𝗶𝗼𝗻 𝗮 𝘂𝘀𝗲𝗿",
          event.threadID,
          event.messageID
        );
      }

      // =========================
      // UID CHECK
      // =========================
      if (!mention) {

        return api.sendMessage(
          "❌ 𝗨𝘀𝗲𝗿 𝗻𝗼𝘁 𝗳𝗼𝘂𝗻𝗱 🐸\n" +
          "𝗣𝗹𝗲𝗮𝘀𝗲 𝗰𝗵𝗲𝗰𝗸 𝗵𝗲𝗿 𝗽𝗿𝗼𝗳𝗶𝗹𝗲",
          event.threadID,
          event.messageID
        );
      }

      // =========================
      // USER INFO
      // =========================
      const userInfo =
        await api.getUserInfo(mention);

      mentionName =
        userInfo?.[mention]?.name ||
        "Unknown";

      const senderID =
        event.senderID;

      const credit =
        "🔰𝐑𝐀𝐇𝐀𝐓 𝐈𝐒𝐋𝐀𝐌🔰";

      // =========================
      // LOADING
      // =========================
      const waiting =
        await api.sendMessage(
          "⏳𝗣𝗹𝗲𝗮𝘀𝗲 𝘄𝗮𝗶𝘁....",
          event.threadID
        );

      // =========================
      // API LIST
      // =========================
      const apiConfigList =
        await getApiList("girlfriend");

      // =========================
      // GENERATE
      // =========================
      const {
        imageBuffer,
        captionTemplate
      } =
        await generateFrameWithFallback({
          senderID,
          mention,
          credit,
          apiList: apiConfigList
        });

      // =========================
      // CAPTION
      // =========================
      const finalCaption =
        captionTemplate.replace(
          /{{name}}/g,
          mentionName
        );

      // =========================
      // SAVE IMAGE
      // =========================
      const outPath =
        path.join(
          __dirname,
          `girlfriend_${Date.now()}.png`
        );

      await fs.writeFile(
        outPath,
        imageBuffer
      );

      // =========================
      // DELETE LOADING
      // =========================
      if (waiting?.messageID) {

        await api.unsendMessage(
          waiting.messageID
        );
      }

      // =========================
      // SEND RESULT
      // =========================
      const messageInfo =
        await api.sendMessage(
          {
            body: finalCaption,

            mentions: [
              {
                tag: mentionName,
                id: mention
              }
            ],

            attachment:
              fs.createReadStream(outPath)
          },

          event.threadID,
          event.messageID
        );

      // =========================
      // AUTO DELETE
      // 2 MINUTES
      // =========================
      setTimeout(async () => {

        try {

          if (messageInfo?.messageID) {

            await api.unsendMessage(
              messageInfo.messageID
            );
          }

          if (
            await fs.pathExists(outPath)
          ) {

            await fs.remove(outPath);
          }

        } catch (e) {}

      }, 120000);

    } catch (error) {

      console.error(
        "[girlfriend]",
        error
      );

      return api.sendMessage(
        `⚠️ ${error.message || "Unknown error"}`,
        event.threadID
      );
    }
  }
};