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

let userSession = {};
let botMessages = {};

module.exports = {
  config: {
    name: "segs",
    version: "1.5",
    author: "Azadx69x",
    role: 1,
    category: "18+",
    shortDescription: "𝐇𝐃 𝐕𝐢𝐝𝐞𝐨 𝐒𝐞𝐚𝐫𝐜𝐡 & 𝐃𝐨𝐰𝐧𝐥𝐨𝐚𝐝",
    longDescription: "𝐒𝐞𝐚𝐫𝐜𝐡 𝐟𝐨𝐫 𝐇𝐃 𝐯𝐢𝐝𝐞𝐨𝐬 𝐚𝐧𝐝 𝐝𝐨𝐰𝐧𝐥𝐨𝐚𝐝"
  },

  onStart: async ({ api, event, args }) => {
    const keyword = args.join(" ");
    const thread = event.threadID;
    const sender = event.senderID;
      
    if (botMessages[thread]) {
      clearBotMessages(api, thread);
    }
    botMessages[thread] = [];

    if (!keyword) {
      const msg = await api.sendMessage("🔴 𝙺𝙴𝚈𝚆𝙾𝚁𝙳 ", thread);
      botMessages[thread].push(msg.messageID);
      return api.setMessageReaction("❌", event.messageID, () => {}, true);
    }

    const searchMsg = await api.sendMessage("🔍 𝚂𝙴𝙰𝚁𝙲𝙷𝙸𝙽𝙶...", thread);
    botMessages[thread].push(searchMsg.messageID);
    api.setMessageReaction("⏳", event.messageID, () => {}, true);

    try {
      const res = await axios.get(
        `https://azadx69x-segs.vercel.app/api/search?q=${encodeURIComponent(keyword)}`
      );

      const results = res.data.list;

      if (!results.length) {
        const noResultMsg = await api.sendMessage("❌ 𝙽𝙾 𝚁𝙴𝚂𝚄𝙻𝚃𝚂 ", thread);
        botMessages[thread].push(noResultMsg.messageID);
        return api.setMessageReaction("❌", event.messageID, () => {}, true);
      }

      userSession[sender] = {
        results,
        expires: Date.now() + 90_000,
        threadID: thread,
        keyword: keyword
      };
        
      setTimeout(() => {
        api.unsendMessage(searchMsg.messageID);
      }, 500);

      sendList(api, thread, sender, event.messageID);

    } catch (e) {
      const errorMsg = await api.sendMessage("❌ 𝙴𝚁𝚁𝙾𝚁 ", thread);
      botMessages[thread].push(errorMsg.messageID);
      api.setMessageReaction("❌", event.messageID, () => {}, true);
    }
  },

  onChat: async ({ api, event }) => {
    const sender = event.senderID;
    const thread = event.threadID;
    const msg = event.body.trim();

    if (!userSession[sender]) return;
    
    if (userSession[sender].threadID !== thread) return;

    if (Date.now() > userSession[sender].expires) {
      clearBotMessages(api, thread);
      delete userSession[sender];
      const timeoutMsg = await api.sendMessage("⏰ 𝚃𝙸𝙼𝙴𝙾𝚄𝚃 ", thread);
      setTimeout(() => {
        api.unsendMessage(timeoutMsg.messageID);
      }, 3000);
      return api.setMessageReaction("⏳", event.messageID, () => {}, true);
    }

    const session = userSession[sender];

    if (/^\d+$/.test(msg)) {
      const number = parseInt(msg);
      const index = number - 1;

      if (number < 1 || number > session.results.length || !session.results[index]) {
        const invalidMsg = await api.sendMessage("❌ 𝙸𝙽𝚅𝙰𝙻𝙸𝙳 𝙽𝚄𝙼𝙱𝙴𝚁 ❌", thread);
        setTimeout(() => {
          api.unsendMessage(invalidMsg.messageID);
        }, 3000);
        return api.setMessageReaction("❌", event.messageID, () => {}, true);
      }

      const item = session.results[index];

      api.setMessageReaction("📤", event.messageID, () => {}, true);
        
      clearBotMessages(api, thread);

      try {
        const filePath = __dirname + "/video.mp4";

        const video = await axios.get(item.video, {
          responseType: "arraybuffer",
          headers: { "User-Agent": "Mozilla/5.0" }
        });

        fs.writeFileSync(filePath, video.data);

        const videoInfo = `━━━━━━━━━━━━━━━━━━━━
𝐕𝐈𝐃𝐄𝐎 𝐃𝐎𝐖𝐍𝐋𝐎𝐀𝐃𝐄𝐃 ✅
╭─╼━━━━━━━━━━━━━━╾─╮
│ 🎬  ${item.name}
│ ⏱  ${item.time}
│ 🔢  ${number}
│ 📊  ${session.results.length} results
│ 🔍  "${session.keyword}"
╰─━━━━━━━━━━━━━━━╾─╯
━━━━━━━━━━━━━━━━━━━━`;

        const videoMsg = await api.sendMessage(
          {
            body: videoInfo,
            attachment: fs.createReadStream(filePath)
          },
          thread
        );

        fs.unlinkSync(filePath);
        api.setMessageReaction("✅", event.messageID, () => {}, true);
          
        if (!botMessages[thread]) botMessages[thread] = [];
        botMessages[thread].push(videoMsg.messageID);
          
        delete userSession[sender];

      } catch (e) {
        const errorMsg = await api.sendMessage("❌ 𝙳𝙾𝚆𝙽𝙻𝙾𝙰𝙳 𝙴𝚁𝚁𝙾𝚁 ❌", thread);
        botMessages[thread].push(errorMsg.messageID);
        api.setMessageReaction("❌", event.messageID, () => {}, true);
      }

      return;
    }

    const invalidMsg = await api.sendMessage("❌ 𝙸𝙽𝚅𝙰𝙻𝙸𝙳 𝙸𝙽𝙿𝚄𝚃 ❌", thread);
    setTimeout(() => {
      api.unsendMessage(invalidMsg.messageID);
    }, 3000);
    api.setMessageReaction("❌", event.messageID, () => {}, true);
  }
};

function sendList(api, thread, user, messageID) {
  const s = userSession[user];
  
  let listMessage = `━━━━━━━━━━━━━━━━━━━━
𝐒𝐄𝐀𝐑𝐂𝐇 𝐑𝐄𝐒𝐔𝐋𝐓𝐒
━━━━━━━━━━━━━━━━━━━━\n\n`;
  
  const showCount = Math.min(20, s.results.length);
  
  for (let i = 0; i < showCount; i++) {
    const item = s.results[i];
    const number = i + 1;
    
    const title = item.name.length > 50 ? item.name.substring(0, 47) + "..." : item.name;
    
    listMessage += `【${number}】 ${title}\n      ⏱ ${item.time}\n\n`;
  }

  listMessage += `━━━━━━━━━━━━━━━━━━━━
𝐒𝐄𝐋𝐄𝐂𝐓: 𝟏-${showCount}
━━━━━━━━━━━━━━━━━━━━`;

  api.sendMessage(listMessage, thread).then(msg => {
    if (!botMessages[thread]) botMessages[thread] = [];
    botMessages[thread].push(msg.messageID);
    
    if (messageID) {
      api.setMessageReaction("✅", messageID, () => {}, true);
    }
  });
}

function clearBotMessages(api, thread) {
  if (botMessages[thread]) {
    botMessages[thread].forEach(messageID => {
      try {
        api.unsendMessage(messageID);
      } catch (e) {
      }
    });
    botMessages[thread] = [];
  }
}