// ==UserScript==
// @name         百度搜索广告清理
// @namespace    local.baidu.ad.filter
// @version      1.3.0
// @description  自动隐藏百度搜索结果页的商业推广、右侧广告、品牌广告；支持一键临时显示、按日统计
// @author       local
// @match        https://www.baidu.com/*
// @match        https://m.baidu.com/*
// @match        https://baidu.com/*
// @run-at       document-start
// @noframes
// @grant        GM_getValue
// @grant        GM_setValue
// @grant        GM_registerMenuCommand
// ==/UserScript==

/*
 * v1.2.0 相对 1.1.0 的关键修复
 * 1. 脚本猫等管理器在 document-start 注入时，页面文档可能随后被替换，
 *    导致注入的 <style> 和 MutationObserver 一起失效（表现为「脚本已触发但什么都没做」）。
 *    现在改为：注入后校验节点是否仍在文档里，不在就重新注入；监听器绑定到当前 documentElement，
 *    前 6 秒每 500ms 校正一次，并绑定 DOMContentLoaded / load。
 * 2. 隐藏判定改为「命中竞价容器就直接隐藏」，不再因为「容器里结果太多」而整体放弃。
 * 3. 基础选择器与 :has() 增强选择器拆成两张样式表，前者解析失败也不影响后者，反之亦然。
 * 4. 用 data-badf-hidden 标记已隐藏节点，避免同一条广告被重复计数。
 */

(function () {
  'use strict';

  /* ==================== 1. 配置区（要改就改这里） ==================== */
  const CONFIG = {
    // 只隐藏右栏里的广告块；「相关搜索」「百度热搜」等正常内容不动
    hideRightColumnAds: true,
    showCounter: true,      // 右下角浮标（含「显示/隐藏」按钮）
    debug: true             // 在控制台打印一行自检信息；确认正常后可改成 false
  };

  /* ==================== 2. 广告特征 ==================== */
  const AD_MARKERS = ['.EC_result', '[data-cmatchid]', '[data-general-xst]'];
  const RESULT_SEL = '.c-container, .result-op, .result, .c-result';
  const AD_TEXT = /^(广告|商业推广|品牌广告|广告[·•].{0,6})$/;
  const HIDDEN_ATTR = 'data-badf-hidden';

  /* ==================== 3. 样式 ==================== */
  // 基础规则：只用最普通的 CSS，任何浏览器都能解析
  const CSS_BASIC = [
    '.EC_result, [data-cmatchid], [data-general-xst],',
    '#top_ads, #brand_ads, .ec-pc-brand-ad, .ec-brand-ad',
    ' { display: none !important; }'
  ].join('\n');

  // 增强规则：用 :has() 把「带广告角标的结果」和「广告外壳」也一并收掉
  const CSS_HAS = [
    '#content_left .c-container:has(.ec-tuiguang),',
    '#content_left .result:has(.ec-tuiguang),',
    '#content_left .result-op:has(.ec-tuiguang),',
    '.c-result:has(.ec-tuiguang),',
    '.c-result:has([data-tuiguang]),',
    '[data-placeid]:has(.EC_result),',
    '[data-aggregation-area]:has(.EC_result),',
    '[data-placeid]:has(.ec-tuiguang),',
    '[data-aggregation-area]:has(.ec-tuiguang)',
    ' { display: none !important; }'
  ].join('\n');

  const BADGE_CSS = [
    '#badf-badge {',
    '  position: fixed; right: 16px; bottom: 16px; z-index: 2147483647;',
    '  display: none; align-items: center; gap: 6px;',
    '  padding: 7px 11px; border-radius: 999px;',
    '  background: rgba(24,28,36,.92); color: #e8eaed;',
    '  font: 12px/1.4 -apple-system, "Segoe UI", "Microsoft YaHei", sans-serif;',
    '  box-shadow: 0 4px 16px rgba(0,0,0,.22); user-select: none;',
    '  transition: opacity .25s ease; opacity: 1;',
    '}',
    '#badf-badge.badf-on { display: flex; }',
    '#badf-badge.badf-faded { opacity: .38; }',
    '#badf-badge.badf-faded:hover { opacity: 1; }',
    '#badf-badge .badf-dot { width: 7px; height: 7px; border-radius: 50%; background: #41d18a; }',
    '#badf-badge.badf-paused .badf-dot { background: #f0a531; }',
    '#badf-badge .badf-count { font-weight: 700; color: #41d18a; }',
    '#badf-badge.badf-paused .badf-count { color: #f0a531; }',
    '#badf-badge .badf-tip { cursor: pointer; }',
    '#badf-badge .badf-toggle {',
    '  margin-left: 2px; padding: 2px 8px; border: 1px solid rgba(255,255,255,.28);',
    '  border-radius: 999px; background: transparent; color: #e8eaed;',
    '  font: inherit; font-size: 11px; cursor: pointer;',
    '}',
    '#badf-badge .badf-toggle:hover { background: rgba(255,255,255,.14); }',
    '#badf-panel {',
    '  position: absolute; right: 0; bottom: 38px; display: none;',
    '  min-width: 148px; padding: 8px 10px; border-radius: 10px;',
    '  background: rgba(24,28,36,.97); box-shadow: 0 6px 20px rgba(0,0,0,.3);',
    '  font-size: 12px; line-height: 1.9; white-space: nowrap;',
    '}',
    '#badf-panel.badf-open { display: block; }',
    '#badf-panel .badf-row { display: flex; justify-content: space-between; gap: 18px; }',
    '#badf-panel b { color: #41d18a; font-weight: 700; }'
  ].join('\n');

  /* ==================== 4. 本地统计 ==================== */
  const STORE_KEY = 'badf_stats_v1';
  const useGM = typeof GM_getValue === 'function' && typeof GM_setValue === 'function';

  function readStats() {
    try {
      const raw = useGM ? GM_getValue(STORE_KEY, '') : localStorage.getItem(STORE_KEY);
      if (!raw) return {};
      const data = typeof raw === 'string' ? JSON.parse(raw) : raw;
      return data && typeof data === 'object' ? data : {};
    } catch (e) {
      return {};
    }
  }

  function writeStats(data) {
    try {
      const raw = JSON.stringify(data);
      if (useGM) GM_setValue(STORE_KEY, raw);
      else localStorage.setItem(STORE_KEY, raw);
    } catch (e) {
      /* 存储不可用时跳过，不影响去广告 */
    }
  }

  // 用本地日期而不是 UTC，避免早上 8 点前被算到前一天
  function dayKey(date) {
    const d = date || new Date();
    const p = function (n) { return n < 10 ? '0' + n : '' + n; };
    return d.getFullYear() + '-' + p(d.getMonth() + 1) + '-' + p(d.getDate());
  }

  function bumpStats(n) {
    if (!n) return;
    const data = readStats();
    const k = dayKey();
    data[k] = (Number(data[k]) || 0) + n;
    writeStats(data);
  }

  function statsSummary() {
    const data = readStats();
    let total = 0;
    for (const k in data) {
      if (Object.prototype.hasOwnProperty.call(data, k)) total += Number(data[k]) || 0;
    }
    return { today: Number(data[dayKey()]) || 0, total: total };
  }

  /* ==================== 5. 状态 ==================== */
  const prevDisplay = new WeakMap();
  let hiddenList = [];   // 计入条数的记录
  let extraList = [];    // 不计入条数（右侧整列、广告外壳）
  let count = 0;
  let paused = false;
  let cssBasic = null;
  let cssHas = null;
  let badge = null;
  let panel = null;
  let fadeTimer = 0;
  let observer = null;
  let observedRoot = null;
  let booted = false;
  let reported = false;

  /* ==================== 6. 样式注入（可自愈） ==================== */
  function addStyle(css, id) {
    let el = document.getElementById(id);
    if (!el) {
      el = document.createElement('style');
      el.id = id;
    }
    el.textContent = css;
    const host = document.head || document.documentElement;
    if (host && el.parentNode !== host) host.appendChild(el);
    return el;
  }

  // 文档可能被替换，每次扫描都确认一次样式还在
  function ensureStyles() {
    if (!cssBasic || !cssBasic.isConnected) cssBasic = addStyle(CSS_BASIC, 'badf-style');
    if (!cssHas || !cssHas.isConnected) cssHas = addStyle(CSS_HAS, 'badf-style-has');
  }

  /* ==================== 7. 隐藏与还原 ==================== */
  // 把单条广告定位到「外壳」上（能上浮才上浮，绝不返回 null 放弃）
  function slotOf(el) {
    let node = el;
    for (let i = 0; i < 3; i++) {
      const parent = node.parentElement;
      if (!parent) break;
      const isSlot = parent.hasAttribute('data-placeid') ||
                     parent.hasAttribute('data-aggregation-area');
      if (!isSlot) break;
      const ads = parent.querySelectorAll('.EC_result, [data-cmatchid], .ec-tuiguang').length;
      const results = parent.querySelectorAll(RESULT_SEL).length;
      if (ads > 2 || results > 1) break;   // 这层已经包了别的东西
      node = parent;
    }
    return node;
  }

  function hide(el, tick) {
    if (!el || el.nodeType !== 1) return false;
    if (el.hasAttribute(HIDDEN_ATTR)) return false;
    if (el.closest('[' + HIDDEN_ATTR + ']')) return false;  // 已被上层容器隐藏

    prevDisplay.set(el, el.style.getPropertyValue('display'));
    el.setAttribute(HIDDEN_ATTR, '1');
    el.style.setProperty('display', 'none', 'important');
    (tick ? hiddenList : extraList).push(el);

    if (tick) {
      count++;
      bumpStats(1);
      updateBadge();
    }
    return true;
  }

  function show(el) {
    const prev = prevDisplay.get(el) || '';
    if (prev) el.style.setProperty('display', prev, 'important');
    else el.style.removeProperty('display');
    el.removeAttribute(HIDDEN_ATTR);
  }

  function liveRecords() {
    hiddenList = hiddenList.filter(function (el) { return el.isConnected; });
    extraList = extraList.filter(function (el) { return el.isConnected; });
    return hiddenList.concat(extraList);
  }

  function restoreAll() {
    paused = true;
    if (cssBasic) cssBasic.disabled = true;
    if (cssHas) cssHas.disabled = true;
    for (const el of liveRecords()) show(el);
    updateBadge();
  }

  function rehideAll() {
    paused = false;
    if (cssBasic) cssBasic.disabled = false;
    if (cssHas) cssHas.disabled = false;
    for (const el of liveRecords()) {
      el.setAttribute(HIDDEN_ATTR, '1');
      el.style.setProperty('display', 'none', 'important');
    }
    updateBadge();
    run();
  }

  /* ==================== 8. 扫描 ==================== */
  function scanAds() {
    // ① 竞价容器：命中就隐藏，不再「疑似容器就放弃」
    const ads = document.querySelectorAll('.EC_result, [data-cmatchid], [data-general-xst]');
    for (const ad of ads) {
      if (ad.closest('[' + HIDDEN_ATTR + ']')) continue;
      // 右栏只处理其中的广告块，「相关搜索」「百度热搜」等正常内容保留
      if (!CONFIG.hideRightColumnAds && ad.closest('#content_right')) continue;
      hide(slotOf(ad), true);
    }

    // ② 官方广告角标：向上找它所在的那条结果，顺便清掉外壳留白
    for (const label of document.querySelectorAll('.ec-tuiguang, [data-tuiguang]')) {
      if (label.closest('[' + HIDDEN_ATTR + ']')) continue;
      if (!CONFIG.hideRightColumnAds && label.closest('#content_right')) continue;
      const box = label.closest(RESULT_SEL);
      if (box) hide(slotOf(box), true);
    }

    // ③ 文字角标兜底：class 全改名时仍能识别（要求处在商业推广上下文里）
    for (const label of document.querySelectorAll('#content_left span, #content_left i, #content_left em, .c-result span')) {
      if (label.children.length) continue;
      const text = (label.textContent || '').trim();
      if (!text || text.length > 8 || !AD_TEXT.test(text)) continue;
      const ctx = label.closest('[data-placeid], [data-aggregation-area], [data-tuiguang], ' +
        'a[href*="e.baidu.com"], a[href*="baidu.php"]');
      if (!ctx) continue;
      if (label.closest('[' + HIDDEN_ATTR + ']')) continue;
      const box = label.closest(RESULT_SEL);
      if (box) hide(slotOf(box), true);
    }
  }

  function run() {
    if (paused || !document.body) return;
    ensureStyles();

    scanAds();

    for (const el of document.querySelectorAll('#top_ads, #brand_ads')) hide(el, false);

    if (!reported && document.readyState !== 'loading') {
      reported = true;
      if (CONFIG.debug) {
        console.log('[百度广告清理] v1.3 广告容器 ' +
          document.querySelectorAll('.EC_result, [data-cmatchid]').length +
          ' 个，已隐藏 ' + hiddenList.length + ' 条，样式生效 ' +
          !!(cssBasic && cssBasic.isConnected && cssBasic.sheet && cssBasic.sheet.cssRules.length));
      }
    }
  }

  /* ==================== 9. 右下角浮标 ==================== */
  function buildBadge() {
    badge = document.createElement('div');
    badge.id = 'badf-badge';
    badge.innerHTML = [
      '<span class="badf-dot"></span>',
      '<span class="badf-tip"><span class="badf-label">已净化</span> ',
      '<span class="badf-count">0</span> 条广告</span>',
      '<button type="button" class="badf-toggle">显示</button>',
      '<div id="badf-panel"></div>'
    ].join('');

    panel = badge.querySelector('#badf-panel');
    const toggle = badge.querySelector('.badf-toggle');
    const tip = badge.querySelector('.badf-tip');

    toggle.addEventListener('click', function (e) {
      e.stopPropagation();
      if (paused) rehideAll();
      else restoreAll();
    });

    tip.addEventListener('click', function (e) {
      e.stopPropagation();
      renderPanel();
      panel.classList.toggle('badf-open');
    });

    document.addEventListener('click', function () {
      if (panel) panel.classList.remove('badf-open');
    });

    document.body.appendChild(badge);
  }

  function renderPanel() {
    const s = statsSummary();
    const rows = [['今日', s.today], ['累计', s.total], ['本次浏览', count]];
    panel.textContent = '';
    for (const item of rows) {
      const row = document.createElement('div');
      row.className = 'badf-row';
      const k = document.createElement('span');
      k.textContent = item[0];
      const v = document.createElement('b');
      v.textContent = String(item[1]);
      row.appendChild(k);
      row.appendChild(v);
      panel.appendChild(row);
    }
  }

  function updateBadge() {
    if (!CONFIG.showCounter || !document.body) return;
    if (!count && !paused) {
      if (badge) badge.classList.remove('badf-on');
      return;
    }
    if (!badge) {
      addStyle(BADGE_CSS, 'badf-style-badge');
      buildBadge();
    }

    badge.classList.add('badf-on');
    badge.classList.toggle('badf-paused', paused);
    badge.querySelector('.badf-label').textContent = paused ? '已还原' : '已净化';
    liveRecords();
    badge.querySelector('.badf-count').textContent = String(hiddenList.length);
    badge.querySelector('.badf-toggle').textContent = paused ? '隐藏' : '显示';
    if (paused && panel) panel.classList.remove('badf-open');

    clearTimeout(fadeTimer);
    badge.classList.remove('badf-faded');
    if (!paused) {
      fadeTimer = setTimeout(function () {
        if (badge && !paused) badge.classList.add('badf-faded');
      }, 3000);
    }
  }

  /* ==================== 10. 启动与自愈 ==================== */
  let guardTimer = 0;

  function schedule() {
    clearTimeout(guardTimer);
    guardTimer = setTimeout(run, 120);
  }

  function boot() {
    if (!document.documentElement) return;
    ensureStyles();

    // 文档被替换过就重新挂监听
    if (observedRoot !== document.documentElement) {
      if (observer) observer.disconnect();
      observer = new MutationObserver(schedule);
      observedRoot = document.documentElement;
      observer.observe(observedRoot, { childList: true, subtree: true });
    }

    run();

    if (!booted) {
      booted = true;
      window.addEventListener('load', schedule);
      window.addEventListener('popstate', schedule);
      [200, 600, 1200, 2500].forEach(function (t) { setTimeout(run, t); });
    }
  }

  function start() {
    // 尽早注入一次，正常环境（如油猴）能直接不闪
    ensureStyles();
    run();
    boot();

    if (document.readyState === 'loading') {
      document.addEventListener('DOMContentLoaded', boot, { once: true });
    }

    // 前 6 秒每 500ms 校正一次，兜住「document-start 后文档被替换」的情况
    let ticks = 0;
    const timer = setInterval(function () {
      if (++ticks > 12) { clearInterval(timer); return; }
      if (document.readyState !== 'loading') boot();
    }, 500);

    if (typeof GM_registerMenuCommand === 'function') {
      GM_registerMenuCommand('查看广告净化统计', function () {
        const s = statsSummary();
        alert('百度广告清理统计\n\n今日：' + s.today + ' 条\n累计：' + s.total + ' 条\n本次浏览：' + count + ' 条');
      });
    }
  }

  if (document.documentElement) start();
  else document.addEventListener('DOMContentLoaded', start, { once: true });
})();