tt-help-cli-ycl 1.3.31 → 1.3.33

package/src/scraper/modules/follow-extractor.js

@@ -3,6 +3,36 @@ import { scrollAndCollect } from "./scroll-collector.js";
 
 const FILTER_WORDS = ["主页", "已关注", "粉丝", "推荐"];
 
+const FOLLOW_TRIGGER_SELECTORS = [
+  "[data-e2e=following]",
+  'a[href$="/following"]',
+  'a[href*="/following"]',
+  '[data-e2e*="following"]',
+];
+
+async function waitForFollowTrigger(page, timeout = 15000) {
+  await page
+    .waitForFunction(
+      (selectors) => {
+        for (const selector of selectors) {
+          if (document.querySelector(selector)) return true;
+        }
+
+        const textMatchers = [/^关注$/, /^Following$/i, /^已关注$/];
+        const nodes = document.querySelectorAll("a,button,div,span");
+        for (const node of nodes) {
+          const text = (node.textContent || "").trim();
+          if (textMatchers.some((matcher) => matcher.test(text))) return true;
+        }
+
+        return false;
+      },
+      FOLLOW_TRIGGER_SELECTORS,
+      { timeout },
+    )
+    .catch(() => {});
+}
+
 async function waitForListContent(page, minChildren = 1, timeout = 15000) {
   await page
     .waitForFunction(
@@ -19,14 +49,58 @@ async function waitForListContent(page, minChildren = 1, timeout = 15000) {
 }
 
 async function openFollowModal(page) {
-  const el = await page.$("[data-e2e=following]");
-  if (!el) {
+  const tryOpen = async () =>
+    page.evaluate((selectors) => {
+      const clickTarget = (node) => {
+        if (!node) return false;
+        const clickable =
+          node.closest('a,button,[role="button"]') ||
+          node.parentElement ||
+          node;
+        clickable.click();
+        return true;
+      };
+
+      for (const selector of selectors) {
+        const node = document.querySelector(selector);
+        if (clickTarget(node)) return selector;
+      }
+
+      const textMatchers = [/^关注$/, /^Following$/i, /^已关注$/];
+      const nodes = Array.from(document.querySelectorAll("a,button,div,span"));
+      for (const node of nodes) {
+        const text = (node.textContent || "").trim();
+        if (!text) continue;
+        if (textMatchers.some((matcher) => matcher.test(text))) {
+          if (clickTarget(node)) return `text:${text}`;
+        }
+      }
+
+      return null;
+    }, FOLLOW_TRIGGER_SELECTORS);
+
+  let opened = null;
+  for (let attempt = 1; attempt <= 3; attempt++) {
+    await waitForFollowTrigger(page, attempt === 1 ? 15000 : 8000);
+    opened = await tryOpen();
+    if (opened) break;
+
+    await page
+      .evaluate(() => {
+        window.scrollTo({ top: 0, behavior: "instant" });
+      })
+      .catch(() => {});
+    await delay(800, 1500);
+  }
+
+  if (!opened) {
     throw new Error(
-      "未找到 [data-e2e=following] 元素，请确认当前页面为用户主页",
+      "未找到关注入口元素，请确认当前页面为用户主页或页面结构已变化",
     );
   }
-  await el.evaluate((el) => el.parentElement.click());
-  await page.waitForSelector("[class*=DivUserListContainer]", { timeout: 30000 });
+  await page.waitForSelector("[class*=DivUserListContainer]", {
+    timeout: 30000,
+  });
   await waitForListContent(page, 1, 5000);
 }
 
@@ -41,7 +115,9 @@ async function switchToFollowersTab(page) {
     }
     throw new Error("未找到粉丝 Tab");
   });
-  await page.waitForSelector("[class*=DivUserListContainer]", { timeout: 30000 });
+  await page.waitForSelector("[class*=DivUserListContainer]", {
+    timeout: 30000,
+  });
   await waitForListContent(page, 1, 5000);
 }