popup.js

// popup.js

/**
 * 存储获取到的所有链接
 */
let allLinks = [];

/**
 * 初始化函数，设置事件监听器
 */
function init() {
  const getLinksBtn = document.getElementById('getLinksBtn');
  const closeLinksBtn = document.getElementById('closeLinksBtn');
  const refreshLinksBtn = document.getElementById('refreshLinksBtn');
  const statusDiv = document.getElementById('status');
  const sameDomainCheckbox = document.getElementById('sameDomainCheckbox');
  const filterInput = document.getElementById('filterInput');
  const excludeInput = document.getElementById('excludeInput');
  const delayInput = document.getElementById('delayInput');
  const limitInput = document.getElementById('limitInput');
  const toggleLinksBtn = document.getElementById('toggleLinksBtn');
  const linksList = document.getElementById('linksList');
  const currentDomainSpan = document.getElementById('currentDomain');

  // 显示当前域名
  async function displayCurrentDomain() {
    try {
      const [tab] = await chrome.tabs.query({ active: true, currentWindow: true });
      if (tab) {
        const currentDomain = new URL(tab.url).hostname;
        currentDomainSpan.textContent = currentDomain;
      } else {
        currentDomainSpan.textContent = '无法获取域名';
      }
    } catch (error) {
      console.error('获取域名失败:', error);
      currentDomainSpan.textContent = '获取域名失败';
    }
  }

  // 页面加载时显示当前域名
  displayCurrentDomain();

  // 页面加载完成后自动获取链接数量
  async function displayLinkCount() {
    try {
      // 获取当前活动标签页
      const [tab] = await chrome.tabs.query({ active: true, currentWindow: true });

      if (!tab) {
        statusDiv.textContent = '无法获取当前标签页';
        return;
      }

      statusDiv.textContent = '正在获取链接数量...';

      // 向content.js发送消息，请求获取链接
      const links = await chrome.tabs.sendMessage(tab.id, { action: 'getLinks' });

      // 存储获取到的链接
      allLinks = links || [];

      // 过滤链接
      const filteredLinks = filterLinks(allLinks, tab.url);

      if (allLinks.length > 0) {
        statusDiv.innerHTML = `当前页面有 <b> ${allLinks.length} </b> 个链接，过滤后有 <b> ${filteredLinks.length} </b> 个`;
        // 初始显示链接列表
        displayLinksList(filteredLinks);
      } else {
        statusDiv.textContent = '当前页面未找到链接';
        linksList.innerHTML = '<p style="margin: 5px 0; color: #666;">没有匹配的链接</p>';
      }
    } catch (error) {
      console.error('获取链接失败:', error);
      statusDiv.textContent = '获取链接数量失败';
    }
  }

  /**
   * 根据条件过滤链接
   * @param {string[]} links - 要过滤的链接数组
   * @param {string} currentUrl - 当前页面URL
   * @returns {string[]} 过滤后的链接数组
   */
  function filterLinks(links, currentUrl) {
    if (!links || links.length === 0) return [];

    const currentDomain = new URL(currentUrl).hostname;
    const filterText = filterInput.value.toLowerCase();
    const excludeKeywords = excludeInput.value.toLowerCase().split(',').map(keyword => keyword.trim()).filter(keyword => keyword !== '');
    const sameDomainOnly = sameDomainCheckbox.checked;

    return links.filter(link => {
      const linkLower = link.toLowerCase();
      
      // 检查是否为本域名
      const linkDomain = new URL(link).hostname;
      const isSameDomain = linkDomain === currentDomain;

      // 如果只允许本域名且当前链接不是本域名，则过滤掉
      if (sameDomainOnly && !isSameDomain) {
        return false;
      }

      // 检查是否匹配筛选文本
      if (filterText !== '' && !linkLower.includes(filterText)) {
        return false;
      }

      // 检查是否包含排除关键字
      if (excludeKeywords.length > 0 && excludeKeywords.some(keyword => linkLower.includes(keyword))) {
        return false;
      }

      return true;
    });
  }

  // 重新获取链接按钮点击事件
  refreshLinksBtn.addEventListener('click', async () => {
    await displayCurrentDomain();
    displayLinkCount();
  });

  // 筛选条件变化事件
  sameDomainCheckbox.addEventListener('change', async () => {
    if (allLinks.length > 0) {
      const [tab] = await chrome.tabs.query({ active: true, currentWindow: true });
      if (tab) {
        const filteredLinks = filterLinks(allLinks, tab.url);
        statusDiv.innerHTML = `当前页面有 <b> ${allLinks.length} </b> 个链接，过滤后有 <b> ${filteredLinks.length} </b> 个`;
        displayLinksList(filteredLinks);
      }
    }
  });

  filterInput.addEventListener('input', async () => {
    if (allLinks.length > 0) {
      const [tab] = await chrome.tabs.query({ active: true, currentWindow: true });
      if (tab) {
        const filteredLinks = filterLinks(allLinks, tab.url);
        statusDiv.innerHTML = `当前页面有 <b> ${allLinks.length} </b> 个链接，过滤后有 <b> ${filteredLinks.length} </b> 个`;
        displayLinksList(filteredLinks);
      }
    }
  });

  // 排除关键字输入变化事件
  excludeInput.addEventListener('input', async () => {
    if (allLinks.length > 0) {
      const [tab] = await chrome.tabs.query({ active: true, currentWindow: true });
      if (tab) {
        const filteredLinks = filterLinks(allLinks, tab.url);
        statusDiv.innerHTML = `当前页面有 <b> ${allLinks.length} </b> 个链接，过滤后有 <b> ${filteredLinks.length} </b> 个`;
        displayLinksList(filteredLinks);
      }
    }
  });

  // 展开/收起链接列表按钮点击事件
  toggleLinksBtn.addEventListener('click', toggleLinksList);

  /**
   * 显示匹配的链接列表
   * @param {string[]} links - 要显示的链接数组
   */
  function displayLinksList(links) {
    if (!links || links.length === 0) {
      linksList.innerHTML = '<p style="margin: 5px 0; color: #666;">没有匹配的链接</p>';
      return;
    }

    const linksHTML = links.map((link, index) => 
      `<div style="margin: 3px 0; padding: 3px; border-bottom: 1px solid #eee; word-break: break-all;">
        <span style="font-weight: bold; margin-right: 5px;">${index + 1}.</span>
        <a href="${link}" target="_blank" style="color: #1a0dab; text-decoration: none;">${link}</a>
      </div>`
    ).join('');

    linksList.innerHTML = linksHTML;
  }

  /**
   * 切换链接列表的展开/收起状态
   */
  function toggleLinksList() {
    if (linksList.style.maxHeight === '0px' || linksList.style.maxHeight === '') {
      // 展开列表
      linksList.style.maxHeight = '200px'; // 设置一个合适的最大高度
      toggleLinksBtn.innerHTML = '🔽 隐藏匹配的链接';
    } else {
      // 收起列表
      linksList.style.maxHeight = '0';
      toggleLinksBtn.innerHTML = '▶️ 显示匹配的链接';
    }
  }

  // 调用函数显示链接数量
  displayLinkCount();

  // 批量关闭按钮点击事件
  closeLinksBtn.addEventListener('click', async () => {
    try {
      // 发送消息到background.js，关闭之前打开的标签页
      const response = await chrome.runtime.sendMessage({ action: 'closeOpenedTabs' });
      if (response && response.success) {
        statusDiv.textContent = '已关闭批量打开的标签页';
      } else {
        statusDiv.textContent = '关闭标签页失败';
      }
    } catch (error) {
      console.error('关闭标签页失败:', error);
      statusDiv.textContent = '关闭标签页失败，请重试';
    }
  });

  // 批量打开按钮点击事件
  getLinksBtn.addEventListener('click', async () => {
    try {
      // 获取当前活动标签页
      const [tab] = await chrome.tabs.query({ active: true, currentWindow: true });

      if (!tab) {
        statusDiv.textContent = '无法获取当前标签页';
        return;
      }

      if (allLinks.length === 0) {
        statusDiv.textContent = '正在获取链接...';
        // 如果没有链接，先获取链接
        allLinks = await chrome.tabs.sendMessage(tab.id, { action: 'getLinks' }) || [];
      }

      // 使用已过滤的链接
      const filteredLinks = filterLinks(allLinks, tab.url);

      if (filteredLinks && filteredLinks.length > 0) {
        // 获取打开方式
        const windowMode = document.querySelector('input[name="openMethod"]:checked').value;
        // 获取延迟时间
        const delayTime = parseInt(delayInput.value) || 20;
        // 获取数量限制
        const limit = parseInt(limitInput.value) || 50;
        
        // 检查是否超过用户设置的限制
        if (filteredLinks.length > limit) {
          // 只打开前limit个链接
          const limitedLinks = filteredLinks.slice(0, limit);
          // 检查是否超过50个链接
          if (limitedLinks > 50) {
            // 显示提示对话框
            if (confirm(`您要打开 ${filteredLinks.length} 个链接，这可能会导致浏览器卡顿。\n\n选择打开方式：\n- 确定：打开全部链接\n- 取消：取消操作`)) {
              // 打开全部链接
              statusDiv.textContent = `正在打开 ${filteredLinks.length} 个链接...`;
              sendOpenLinksMessage(filteredLinks, windowMode, delayTime);
            } else {
              // 取消操作，什么也不做
              statusDiv.textContent = '操作已取消';
            }
          }else{
            statusDiv.textContent = `正在打开 ${limitedLinks.length} 个链接...`;
            sendOpenLinksMessage(limitedLinks, windowMode, delayTime);
          }
        } else {
          // 直接打开链接
          statusDiv.textContent = `正在打开 ${filteredLinks.length} 个链接...`;
          sendOpenLinksMessage(filteredLinks, windowMode, delayTime);
        }
      } else {
        statusDiv.textContent = '没有符合条件的链接';
      }
    } catch (error) {
      console.error('获取链接失败:', error);
      statusDiv.textContent = '获取链接失败，请重试';
    }
  });

  /**
   * 发送打开链接的消息到background.js
   * @param {string[]} links - 要打开的链接数组
   * @param {string} windowMode - 窗口模式（'new'或'current'）
   * @param {number} delay - 延迟时间
   */
  function sendOpenLinksMessage(links, windowMode, delay) {
    if (windowMode === 'new') {
      chrome.runtime.sendMessage({
        action: 'openLinksInNewWindow',
        links: links,
        delay: delay
      });
    } else {
      chrome.runtime.sendMessage({
        action: 'openLinksInCurrentWindow',
        links: links,
        delay: delay
      });
    }
  }
}

// 页面加载完成后初始化
document.addEventListener('DOMContentLoaded', init);