content.js

// X-Vision: Judgment Day v5.0.1
// 修复：hover 绿框消失 | 自己被标绿 | /following 右侧推荐关注误标红

(() => {
  const STORE_KEY = "my_following_roster";
  const STYLE_ID = "uh-style-v501";
  const MAX_ROSTER = 60000;

  const SAVE_DEBOUNCE_MS = 2000;
  const NAV_SCAN_DELAY_MS = 800;
  const NAV_POLL_MS = 8000;

  const DEBUG = false;

  let followingSet = new Set();
  let rosterReady = false;

  let saveTimer = null;
  let lastPath = window.location.pathname;

  let myHandle = null;

  // key: element, value: signature
  let sigMap = new WeakMap();

  const queue = new Set();
  let rafScheduled = false;

  function log(...args) {
    if (DEBUG) console.log("[XVJD]", ...args);
  }

  function injectStylesOnce() {
    if (document.getElementById(STYLE_ID)) return;

    const styleEl = document.createElement("style");
    styleEl.id = STYLE_ID;

    // 用 inset box-shadow，避免 hover 时 outline 被站点样式压掉
    styleEl.textContent = `
      .uh-highlight-new{
        box-shadow: 0 0 0 2px #00e676 inset !important;
        border-radius: 12px !important;
      }
      .uh-alert-unfollow{
        box-shadow: 0 0 0 3px #ff4444 inset !important;
        background-color: rgba(255, 68, 68, 0.03) !important;
        border-radius: 12px !important;
      }
    `;

    (document.head || document.documentElement).appendChild(styleEl);
  }

  function loadRoster() {
    return new Promise((resolve) => {
      chrome.storage.local.get([STORE_KEY], (result) => {
        const arr = Array.isArray(result[STORE_KEY]) ? result[STORE_KEY] : [];
        followingSet = new Set(arr.map((x) => String(x).toLowerCase()));
        rosterReady = true;
        log("roster loaded", followingSet.size);
        resolve();
      });
    });
  }

  function scheduleSave() {
    if (saveTimer) clearTimeout(saveTimer);
    saveTimer = setTimeout(() => {
      chrome.storage.local.set({ [STORE_KEY]: Array.from(followingSet) });
      saveTimer = null;
      log("roster saved", followingSet.size);
    }, SAVE_DEBOUNCE_MS);
  }

  function isFollowingPage() {
    return window.location.pathname.includes("/following");
  }

  function parseHandleFromHref(hrefRaw) {
    if (!hrefRaw) return null;
    const href = String(hrefRaw);
    if (!href.startsWith("/")) return null;

    const clean = href.split("?")[0].split("#")[0];
    const parts = clean.split("/").filter(Boolean);
    if (parts.length < 1) return null;

    const first = `/${parts[0].toLowerCase()}`;
    const blocked = new Set([
      "/home",
      "/explore",
      "/notifications",
      "/messages",
      "/search",
      "/settings",
      "/compose",
      "/i",
      "/intent"
    ]);
    if (blocked.has(first)) return null;

    return parts[0].toLowerCase();
  }

  // 识别当前账号 handle，用于排除自己
  function detectMyHandle() {
    // 侧边栏个人资料入口通常稳定
    const a = document.querySelector('a[data-testid="AppTabBar_Profile_Link"][href^="/"]');
    if (a) {
      const h = parseHandleFromHref(a.getAttribute("href"));
      if (h) return h;
    }
    return null;
  }

  // /following 主栏过滤：避免右侧推荐关注被处理
  function isUserCellInFollowingTimeline(card) {
    const primary = card.closest('div[data-testid="primaryColumn"]');
    if (!primary) return false;

    const sidebar = card.closest('div[data-testid="sidebarColumn"], aside');
    if (sidebar) return false;

    // /following 页面主时间线通常在一个带 aria-label 的 section 内
    const sec = card.closest("section[aria-label]");
    if (!sec) return true;

    const label = sec.getAttribute("aria-label") || "";
    // 常见：Timeline: Following / 时间线: 正在关注
    if (label.includes("Following") || label.includes("正在关注")) return true;

    // 其他 section 先跳过，避免误伤推荐模块
    return false;
  }

  // 在 UserCell 内抓 handle
  function getHandleFromUserCell(card) {
    if (!card || !card.querySelectorAll) return null;
    const links = Array.from(card.querySelectorAll('a[href^="/"]'));
    for (const a of links) {
      const h = parseHandleFromHref(a.getAttribute("href"));
      if (!h) continue;
      // 排除帖子链接 /user/status/xxx
      const href = String(a.getAttribute("href") || "");
      if (href.includes("/status/")) continue;
      return h;
    }
    return null;
  }

  // 时间线抓作者 handle：优先用时间戳链接 /user/status/id，hover 不会变
  function getHandleFromArticle(article) {
    if (!article) return null;

    const timeEl = article.querySelector('a[href*="/status/"] time');
    if (timeEl) {
      const a = timeEl.closest("a");
      if (a) {
        const href = String(a.getAttribute("href") || "");
        const clean = href.split("?")[0].split("#")[0];
        const parts = clean.split("/").filter(Boolean);

        // 期待格式：user / status / id
        if (parts.length >= 3 && parts[1] === "status") {
          const h = parts[0].toLowerCase();
          if (h && h !== "i") return h;
        }
      }
    }

    // 兜底：找 User-Name 区块里的主页链接，限制在当前 article
    const blocks = Array.from(article.querySelectorAll('[data-testid="User-Name"]'))
      .filter((el) => el.closest("article") === article);

    for (const block of blocks) {
      const a = block.querySelector('a[href^="/"]');
      const h = a ? parseHandleFromHref(a.getAttribute("href")) : null;
      if (h) return h;
    }

    return null;
  }

  function enqueue(el) {
    if (!el || el.nodeType !== 1) return;
    queue.add(el);
    if (rafScheduled) return;
    rafScheduled = true;
    requestAnimationFrame(flushQueue);
  }

  function flushQueue() {
    rafScheduled = false;
    if (!rosterReady) {
      queue.clear();
      return;
    }

    const following = isFollowingPage();

    for (const el of queue) {
      if (following) {
        if (el.matches && el.matches('[data-testid="UserCell"]')) {
          applyFollowingCard(el);
        } else if (el.querySelectorAll) {
          el.querySelectorAll('[data-testid="UserCell"]').forEach(applyFollowingCard);
        }
      } else {
        if (el.matches && el.matches("article")) {
          applyTimelineArticle(el);
        } else {
          const art = el.closest && el.closest("article");
          if (art) applyTimelineArticle(art);
          if (el.querySelectorAll) el.querySelectorAll("article").forEach(applyTimelineArticle);
        }
      }
    }

    queue.clear();
  }

  // /following 页面逻辑
  function applyFollowingCard(card) {
    if (!card) return;

    if (!isUserCellInFollowingTimeline(card)) return;

    const handle = getHandleFromUserCell(card);
    if (!handle) {
      card.classList.remove("uh-alert-unfollow");
      return;
    }

    if (myHandle && handle === myHandle) {
      card.classList.remove("uh-alert-unfollow");
      return;
    }

    // 学习 roster，仅对真正的 following 列表做学习
    if (!followingSet.has(handle) && followingSet.size < MAX_ROSTER) {
      followingSet.add(handle);
      scheduleSave();
    }

    // 未回关判断仍用局部文案匹配
    const text = card.innerText || "";
    const followerFlag =
      text.includes("关注了你") ||
      text.includes("Follows you") ||
      text.includes("互相关注");

    const sig = `f|${handle}|${followerFlag ? "ok" : "no"}`;
    const prev = sigMap.get(card);
    if (prev === sig) return;

    if (followerFlag) card.classList.remove("uh-alert-unfollow");
    else card.classList.add("uh-alert-unfollow");

    sigMap.set(card, sig);
  }

  // 时间线与评论区：只给生人加绿框
  function applyTimelineArticle(article) {
    if (!article) return;

    const handle = getHandleFromArticle(article);
    if (!handle) return;

    if (myHandle && handle === myHandle) {
      // 清理残留
      article.classList.remove("uh-highlight-new");
      sigMap.set(article, `t|${handle}|self`);
      return;
    }

    const hasRoster = followingSet.size > 0;
    const isNew = hasRoster && !followingSet.has(handle);

    const sig = `t|${handle}|${isNew ? "new" : "old"}`;
    const prev = sigMap.get(article);
    if (prev === sig) return;

    article.classList.remove("uh-highlight-new");
    if (isNew) article.classList.add("uh-highlight-new");

    sigMap.set(article, sig);
  }

  function forceScanAll() {
    sigMap = new WeakMap();

    // 尝试刷新 myHandle
    myHandle = myHandle || detectMyHandle();
    log("myHandle", myHandle);

    if (isFollowingPage()) {
      document.querySelectorAll('[data-testid="UserCell"]').forEach((el) => enqueue(el));
    } else {
      document.querySelectorAll("article").forEach((el) => enqueue(el));
    }
  }

  const observer = new MutationObserver((mutations) => {
    if (!rosterReady) return;

    for (const m of mutations) {
      for (const n of m.addedNodes) {
        if (n.nodeType !== 1) continue;
        enqueue(n);
      }
    }
  });

  function checkUrlChange() {
    const current = window.location.pathname;
    if (current === lastPath) return;
    lastPath = current;
    setTimeout(forceScanAll, NAV_SCAN_DELAY_MS);
  }

  function hookHistory() {
    if (window.__xvjdHistoryHooked) return;
    window.__xvjdHistoryHooked = true;

    const fire = () => window.dispatchEvent(new Event("xvjd:navigate"));

    const wrap = (method) => {
      const orig = history[method];
      history[method] = function () {
        const ret = orig.apply(this, arguments);
        fire();
        return ret;
      };
    };

    wrap("pushState");
    wrap("replaceState");

    window.addEventListener("popstate", fire);
    window.addEventListener("xvjd:navigate", checkUrlChange);
  }

  async function init() {
    injectStylesOnce();
    await loadRoster();

    // 初始化时尽早拿到自己的 handle
    myHandle = detectMyHandle();
    // 有些页面侧栏渲染慢，补一次延迟检测
    setTimeout(() => {
      myHandle = myHandle || detectMyHandle();
    }, 1500);

    observer.observe(document.body, { childList: true, subtree: true });
    hookHistory();

    setTimeout(forceScanAll, NAV_SCAN_DELAY_MS);
    setInterval(checkUrlChange, NAV_POLL_MS);
  }

  init();
})();