const axios = require("axios");
const path = require("path");
const fs = require("fs");

const baseApiUrl = async () => {
  const base = await axios.get(
    "https://raw.githubusercontent.com/Mostakim0978/D1PT0/refs/heads/main/baseApiUrl.json"
  );
  return base.data.api;
};

module.exports = {
  config: {
    name: "album",
    version: "1.0.0",
    role: 0, // Goat-Bot-V2 uses "role" instead of "hasPermssion" (0 = everyone, 1 = group admin, 2 = bot admin)
    author: "Dipto Modified By SAHU",
    countDown: 5, // Goat-Bot-V2 uses "countDown" instead of "cooldowns"
    description: "Displays album options for selection.",
    category: "media", // Goat-Bot-V2 categories are conventionally lowercase
    guide: {
      en:
        "{pn} to see the list\n" +
        "{pn} add [cartoon/photo/lofi/sad/islamic/funny/horny/anime/aesthetic/cat/lyrics/love/sigma] (reply to a media)",
    },
  },

  onStart: async function ({ api, event, args, message }) {
    if (!args[0]) {
      api.setMessageReaction("😘", event.messageID, () => {}, true);

      const albumOptions = [
        "𝗙𝘂𝗻𝗻𝘆 𝘃𝗶𝗱𝗲𝗼", "𝗜𝘀𝗹𝗮𝗺𝗶𝗰 𝘃𝗶𝗱𝗲𝗼", "𝗦𝗮𝗱 𝘃𝗶𝗱𝗲𝗼", "𝗔𝗻𝗶𝗺𝗲 𝘃𝗶𝗱𝗲𝗼",
        "𝗖𝗮𝗿𝘁𝗼𝗼𝗻 𝘃𝗶𝗱𝗲𝗼", "𝗟𝗼𝗙𝗶 𝗩𝗶𝗱𝗲𝗼", "𝗛𝗼𝗿𝗻𝘆 𝘃𝗶𝗱𝗲𝗼", "𝗖𝗼𝘂𝗽𝗹𝗲 𝗩𝗶𝗱𝗲𝗼",
        "𝗙𝗹𝗼𝘄𝗲𝗿 𝗩𝗶𝗱𝗲𝗼", "𝗥𝗮𝗻𝗱𝗼𝗺 𝗣𝗵𝗼𝘁𝗼", "𝗔𝗲𝘀𝘁𝗵𝗲𝘁𝗶𝗰 𝗩𝗶𝗱𝗲𝗼", "𝗦𝗶𝗴𝗺𝗮 𝗥𝘂𝗹𝗲",
        "𝗟𝘆𝗿𝗶𝗰𝘀 𝗩𝗶𝗱𝗲𝗼", "𝗖𝗮𝘁 𝗩𝗶𝗱𝗲𝗼", "18+ 𝘃𝗶𝗱𝗲𝗼", "𝗙𝗿𝗲𝗲 𝗙𝗶𝗿𝗲 𝘃𝗶𝗱𝗲𝗼",
        "𝗙𝗼𝗼𝘁𝗕𝗮𝗹𝗹 𝘃𝗶𝗱𝗲𝗼", "𝗚𝗶𝗿𝗹 𝘃𝗶𝗱𝗲𝗼", "𝗙𝗿𝗶𝗲𝗻𝗱𝘀 𝗩𝗶𝗱𝗲𝗼", "𝗖𝗿𝗶𝗰𝗸𝗲𝘁 𝘃𝗶𝗱𝗲𝗼",
      ];

      const msg =
        "╔══════════════════════╗\n" +
        "║ 🎵 𝗔𝗹𝗯𝘂𝗺 𝗩𝗶𝗱𝗲𝗼 𝗟𝗶𝘀𝘁 🎶\n" +
        "╠══════════════════════╣\n" +
        "║ 🎬 𝟬𝟭. 𝗙𝘂𝗻𝗻𝘆 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 🌙𝟬 𝟮. 𝗜𝘀𝗹𝗮𝗺𝗶𝗰 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 💔𝟬 𝟯. 𝗦𝗮𝗱 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 🎎 𝟬𝟰. 𝗔𝗻𝗶𝗺𝗲 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 🐾𝟬 𝟱. 𝗖𝗮𝗿𝘁𝗼𝗼𝗻 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 🎧 𝟬𝟲. 𝗟𝗼𝗙𝗶 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 🔥 𝟬𝟳. 𝗛𝗼𝗿𝗻𝘆 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 💑 𝟬𝟴. 𝗖𝗼𝘂𝗽𝗹𝗲 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 🌹 𝟬𝟵. 𝗙𝗹𝗼𝘄𝗲𝗿 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 🖼️ 𝟭𝟬. 𝗥𝗮𝗻𝗱𝗼𝗺 𝗣𝗵𝗼𝘁𝗼 🐤\n" +
        "║ 🌌 𝟭𝟭. 𝗔𝗲𝘀𝘁𝗵𝗲𝘁𝗶𝗰 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 🦁 𝟭𝟮. 𝗦𝗶𝗴𝗺𝗮 𝗥𝘂𝗹𝗲 🐤\n" +
        "║ 🎶 𝟭𝟯. 𝗟𝘆𝗿𝗶𝗰𝘀 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 🐱 𝟭𝟰. 𝗖𝗮𝘁 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 🚫 𝟭𝟱. 18+ 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 🎮 𝟭𝟲. 𝗙𝗿𝗲𝗲 𝗙𝗶𝗿𝗲 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ ⚽ 𝟭𝟳. 𝗙𝗼𝗼𝘁𝗯𝗮𝗹𝗹 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 👧 𝟭𝟴. 𝗚𝗶𝗿𝗹 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 🤝 𝟭𝟵. 𝗙𝗿𝗶𝗲𝗻𝗱𝘀 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "║ 🏏 𝟮𝟬. 𝗖𝗿𝗶𝗰𝗸𝗲𝘁 𝗩𝗶𝗱𝗲𝗼 🐤\n" +
        "╠══════════════════════╣\n" +
        "║ 🔰আপনি যে ক্যাটাগরির ভিডিও\n" +
        "║ দেখতে চান সেটির নাম্বার লিখুন!\n" +
        "║ ◀️ উদাহরণস্বরূপ: 11\n" +
        "╚══════════════════════╝";

      // message.reply returns the sent message's info (contains messageID)
      const info = await message.reply(msg);

      // Goat-Bot-V2 registers reply-listeners on global.GoatBot.onReply (a Map),
      // keyed by the sent message's messageID
      global.GoatBot.onReply.set(info.messageID, {
        commandName: this.config.name,
        messageID: info.messageID,
        author: event.senderID,
        link: albumOptions,
      });
      return;
    }

    // ------------ Video Add via URL ------------
    const d1 = args[1] ? args[1].toLowerCase() : (args[0] ? args[0].toLowerCase() : "");
    const validCommands = [
      "cartoon", "photo", "lofi", "sad", "islamic", "funny", "horny",
      "anime", "love", "baby", "lyrics", "sigma", "aesthetic",
      "cat", "flower", "ff", "sex", "football", "girl", "friend", "cricket",
    ];
    if (!d1 || !validCommands.includes(d1)) return;
    if (!event.messageReply || !event.messageReply.attachments) return;

    const attachment = event.messageReply.attachments[0].url;
    const URL = attachment;
    let queryMap = {
      cartoon: "addVideo",
      photo: "addPhoto",
      lofi: "addLofi",
      sad: "addSad",
      funny: "addFunny",
      islamic: "addIslamic",
      horny: "addHorny",
      anime: "addAnime",
      love: "addLove",
      lyrics: "addLyrics",
      flower: "addFlower",
      sigma: "addSigma",
      aesthetic: "addAesthetic",
      cat: "addCat",
      ff: "addFf",
      sex: "addSex",
      football: "addFootball",
      girl: "addGirl",
      friend: "addFriend",
      cricket: "addCricket",
    };
    const query = queryMap[d1];

    try {
      const response = await axios.get(
        `${await baseApiUrl()}/drive?url=${encodeURIComponent(URL)}`
      );
      const fileUrl = response.data.fileUrl;
      const fileExt = path.extname(fileUrl) || ".mp4";

      let queryType = [".jpg", ".jpeg", ".png"].includes(fileExt) ? "addPhoto" : query;

      const saveRes = await axios.get(
        `${await baseApiUrl()}/album?add=${queryType}&url=${fileUrl}`
      );

      message.reply(
        `✅ | ${saveRes.data.data}\n🔰 | ${saveRes.data.data2}\n🔥 | URL: ${fileUrl}`
      );
    } catch (error) {
      console.error(error);
      message.reply(`Failed to upload media.\nError: ${error.message || error}`);
    }
  },

  onReply: async function ({ api, event, Reply, message }) {
    // Only the original requester's reply should be processed
    if (event.senderID !== Reply.author) return;
    if (event.type !== "message_reply") return;

    // remove the original prompt message (equivalent to api.unsendMessage(handleReply.messageID))
    api.unsendMessage(Reply.messageID);

    const admin = "100001039692046";

    const reply = parseInt(event.body);
    if (isNaN(reply) || reply < 1 || reply > 20) {
      return message.reply("Please reply with a number between 1 and 20");
    }

    let queryMap = {
      1: ["funny", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗙𝘂𝗻𝗻𝘆 𝘃𝗶𝗱𝗲𝗼 <🤣"],
      2: ["islamic", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗜𝘀𝗹𝗮𝗺𝗶𝗰 𝘃𝗶𝗱𝗲𝗼 <😇"],
      3: ["sad", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗦𝗮𝗱 𝘃𝗶𝗱𝗲𝗼 <🥺"],
      4: ["anime", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗮𝗻𝗶𝗺 𝘃𝗶𝗱𝗲𝗼 <😘"],
      5: ["cartoon", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗖𝗮𝗿𝘁𝗼𝗼𝗻 𝘃𝗶𝗱𝗲𝗼 <😇"],
      6: ["lofi", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗟𝗼𝗳𝗶 𝘃𝗶𝗱𝗲𝗼 <😇"],
      7: ["horny", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗛𝗼𝗿𝗻𝘆 𝘃𝗶𝗱𝗲𝗼 <🥵"],
      8: ["love", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗟𝗼𝘃𝗲 𝘃𝗶𝗱𝗲𝗼 <😍"],
      9: ["flower", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗙𝗹𝗼𝘄𝗲𝗿 𝘃𝗶𝗱𝗲𝗼 <🌷🌸"],
      10: ["photo", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗥𝗮𝗻𝗱𝗼𝗺 𝗣𝗵𝗼𝘁𝗼 <😙"],
      11: ["aesthetic", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗔𝗲𝘀𝘁𝗵𝗲𝘁𝗶𝗰 𝗩𝗶𝗱𝗲𝗼 <😙"],
      12: ["sigma", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗦𝗶𝗴𝗺𝗮 𝘃𝗶𝗱𝗲𝗼 <🐤"],
      13: ["lyrics", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗟𝘆𝗿𝗶𝗰𝘀 𝘃𝗶𝗱𝗲𝗼 <🥰"],
      14: ["cat", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗖𝗮𝘁 𝗩𝗶𝗱𝗲𝗼 <😙"],
      15: ["sex", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗦𝗲𝘅 𝘃𝗶𝗱𝗲𝗼 <😙"],
      16: ["ff", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗙𝗿𝗲𝗲 𝗙𝗶𝗿𝗲 𝗩𝗶𝗱𝗲𝗼 <😙"],
      17: ["football", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗙𝗼𝗼𝘁𝗯𝗮𝗹𝗹 𝘃𝗶𝗱𝗲𝗼 <😙"],
      18: ["girl", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗚𝗶𝗿𝗹 𝘃𝗶𝗱𝗲𝗼 <😙"],
      19: ["friend", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗙𝗿𝗶𝗲𝗻𝗱𝘀 𝘃𝗶𝗱𝗲𝗼 <😙"],
      20: ["cricket", "𝗡𝗮𝘄 𝗕𝗮𝗯𝘆 𝗖𝗿𝗶𝗰𝗸𝗲𝘁 𝘃𝗶𝗱𝗲𝗼 <😙"],
    };

    let [query, cp] = queryMap[reply];

    if ((reply === 7 || reply === 15) && event.senderID !== admin) {
      return message.reply("Only admin can use it!");
    }

    try {
      const res = await axios.get(`${await baseApiUrl()}/album?type=${query}`);
      const imgUrl = res.data.data;

      const imgRes = await axios.get(imgUrl, {
        responseType: "arraybuffer",
        headers: { "User-Agent": "Mozilla/5.0" },
      });
      const filename = path.join(__dirname, `cache/dipto_${Date.now()}.mp4`);
      fs.writeFileSync(filename, Buffer.from(imgRes.data, "binary"));

      await message.reply({
        body: cp,
        attachment: fs.createReadStream(filename),
      });
      fs.unlinkSync(filename);

      // done handling this reply chain, remove it from the onReply map
      global.GoatBot.onReply.delete(Reply.messageID);
    } catch (error) {
      console.error(error);
      message.reply(`An error occurred while fetching the media.\n${error.message || error}`);
    }
  },
};