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

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

module.exports = {
  config: {
    name: "latti2", // ⚠️ Don't change name
    version: "2.0",
    author: "🔰𝐑𝐀𝐇𝐀𝐓 𝐈𝐒𝐋𝐀𝐌🔰", // ⚠️ Don't change credit
    countDown: 5,
    role: 0,
    category: "image",
    shortDescription: "image generate",
    longDescription: "Generate latti2 image",
    guide: "{pn} [@mention/reply/uid/link]"
  },

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

    // ==============================
    // 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 Frame
    // ==============================
    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=latti2` +
          `&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 কাজ করছে না")
      );
    }

    // ==============================
    // Main
    // ==============================
    try {
      let mention;
      let mentionName;

      // Reply
      if (event.messageReply) {
        mention = event.messageReply.senderID;
      }

      // Argument
      else if (args && args.length > 0) {
        const input = args.join(" ");

        // Facebook profile link
        if (input.includes(".com/")) {
          mention = await api.getUID(input);
        }

        // @mention
        else if (input.includes("@")) {
          mention =
            Object.keys(event.mentions || {})[0];

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

        // UID
        else {
          mention = args[0];
        }
      }

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

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

      // Get 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
      );

      // Get API list
      const apiConfigList =
        await getApiList("latti2");

      // Generate image
      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,
        `latti2_${Date.now()}.png`
      );

      fs.writeFileSync(
        outPath,
        imageBuffer
      );

      // Remove loading
      try {
        await api.unsendMessage(
          waiting.messageID
        );
      } catch (e) {}

      // Send image
      const messageInfo =
        await api.sendMessage(
          {
            body: finalCaption,
            mentions: [
              {
                tag: mentionName,
                id: mention
              }
            ],
            attachment:
              fs.createReadStream(outPath)
          },
          event.threadID,
          event.messageID
        );

      // Delete after 2 minutes
      setTimeout(async () => {
        try {
          if (messageInfo?.messageID) {
            await api.unsendMessage(
              messageInfo.messageID
            );
          }

          if (fs.existsSync(outPath)) {
            fs.unlinkSync(outPath);
          }
        } catch (e) {}
      }, 120000);

    } catch (error) {
      console.error(
        "latti2 error:",
        error
      );

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