tt-help-cli-ycl 1.3.60 → 1.3.62

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tt-help-cli-ycl",
-  "version": "1.3.60",
+  "version": "1.3.62",
   "description": "TikTok user & video data scraper - extract ttSeller, verified, locationCreated from HTML source",
   "type": "module",
   "bin": {

package/src/lib/browser/page.js

@@ -66,92 +66,92 @@ export async function withBrowserRecovery(fn, browser, page, cdpOptions, port) {
   }
 }
 
+const DOM_CHECK_TIMEOUT = 20000; // 单次 DOM 检测超时 20 秒
+const DOM_CHECK_RETRIES = 3; // DOM 检测最大重试次数
+
 /**
  * 判断登录状态：Cookie 为主，DOM 验真为辅。
  * - 无 sessionid Cookie → 未登录
- * - sessionid Cookie 已过期 → 未登录
  * - 有 Cookie + DOM 确认已登录 → 已登录
  * - 有 Cookie + DOM 确认未登录（出现登录按钮）→ 未登录
- * - 有 Cookie + DOM 无法判断（超时/元素未找到）→ 信任 Cookie，判定已登录
+ * - 有 Cookie + DOM 无法判断（超时/元素未找到），重试后仍无法判断 → 信任 Cookie，判定已登录
  */
 export async function isLoggedIn(page) {
   const cookies = await page.context().cookies("https://www.tiktok.com");
-  const sessionCookie = cookies.find((c) => c.name === "sessionid");
-
-  if (!sessionCookie) {
-    console.error("[登录检测] 无 sessionid Cookie，判定未登录");
-    return false;
-  }
-
-  // Cookie 过期检查：TikTok sessionid 可能是 session cookie（expires=-1）
-  // 只有当 expires > 0 且已过期时才判定无效
-  if (
-    sessionCookie.expires > 0 &&
-    sessionCookie.expires < Math.floor(Date.now() / 1000)
-  ) {
-    console.error("[登录检测] sessionid Cookie 已过期，判定未登录");
-    return false;
-  }
-
-  const domResult = await isLoggedInByDom(page);
-  // domResult: true=已登录, false=明确未登录, null=无法判断
-  if (domResult === true) {
-    console.error("[登录检测] DOM 验真确认已登录");
-    return true;
-  }
-  if (domResult === false) {
-    console.error("[登录检测] DOM 验真发现登录按钮，判定未登录（Cookie 可能已失效）");
-    return false;
+  const hasSessionId = cookies.some((c) => c.name === "sessionid");
+  if (!hasSessionId) return false;
+
+  // 重试 DOM 检测，直到得到明确结果或耗尽重试次数
+  for (let attempt = 1; attempt <= DOM_CHECK_RETRIES; attempt++) {
+    const domResult = await isLoggedInByDom(page);
+    // domResult: true=已登录, false=明确未登录, null=无法判断
+    if (domResult === true) return true;
+    if (domResult === false) return false;
+    // null: DOM 无法判断，刷新页面后重试
+    if (attempt < DOM_CHECK_RETRIES) {
+      console.error(
+        `  [登录检测] DOM 无法判断，刷新页面后重试 (${attempt}/${DOM_CHECK_RETRIES})...`,
+      );
+      await page.reload({ waitUntil: "domcontentloaded" });
+    }
   }
-  // null: DOM 无法判断，信任 Cookie
-  console.error("[登录检测] DOM 验真超时，信任 Cookie 判定已登录");
+  // 重试后仍无法判断，信任 Cookie
+  console.error(
+    `  [登录检测] DOM 检测 ${DOM_CHECK_RETRIES} 次均未明确结果，信任 Cookie 判定为已登录`,
+  );
   return true;
 }
 
 /**
  * 通过 DOM 元素判断登录状态（验真方案）
+ * 使用 locator API + state: 'attached' 来避免 CDP 连接下 waitForSelector 的可见性问题
  * @returns {boolean|null} true=已登录, false=明确未登录, null=无法判断
  */
 export async function isLoggedInByDom(page) {
-  // 先等客户端渲染完成：登录态元素或登录按钮，哪个先出现就停止等待
-  const loginOrLoggedInSelector = [
-    '[class*="DivProfileContainer"]',
-    '[class*="DivUserContainer"]',
-    '[class*="UserMenu"]',
-    '[class*="CurrentUserInfo"]',
-    'button:has-text("登录")',
-    'button:has-text("Log in")',
-    'button:has-text("Sign in")',
-  ].join(", ");
-
-  // 分阶段等待：先 5 秒，超时再重试 10 秒
-  const timeouts = [5000, 10000];
-  let selectorFound = false;
-
-  for (let i = 0; i < timeouts.length; i++) {
-    const found = await page
-      .waitForSelector(loginOrLoggedInSelector, { timeout: timeouts[i] })
-      .then(() => true)
-      .catch(() => false);
-
-    if (found) {
-      selectorFound = true;
-      break;
-    }
-    // 非最后一次，输出重试日志
-    if (i < timeouts.length - 1) {
-      console.error(
-        `[登录检测] DOM 元素等待超时 (${timeouts[i]}ms)，重试中...`,
-      );
-    }
+  // 使用 Promise.race 等待：已登录元素 或 登录按钮，哪个先出现就停止
+  const loggedInLocators = [
+    page.locator('[class*="DivProfileContainer"]'),
+    page.locator('[class*="DivUserContainer"]'),
+    page.locator('[class*="UserMenu"]'),
+    page.locator('[class*="CurrentUserInfo"]'),
+  ];
+
+  const loginButtonLocators = [
+    page.getByText("登录", { exact: true }),
+    page.getByText("Log in", { exact: true }),
+    page.getByText("Sign in", { exact: true }),
+  ];
+
+  // 并发等待：已登录元素 vs 登录按钮
+  const waitForLoggedIn = Promise.any(
+    loggedInLocators.map((loc) =>
+      loc.first().waitFor({ state: "attached", timeout: DOM_CHECK_TIMEOUT }),
+    ),
+  )
+    .then(() => "loggedIn")
+    .catch(() => null);
+
+  const waitForLoginButton = Promise.any(
+    loginButtonLocators.map((loc) =>
+      loc.first().waitFor({ state: "attached", timeout: DOM_CHECK_TIMEOUT }),
+    ),
+  )
+    .then(() => "loginButton")
+    .catch(() => null);
+
+  // 哪个先完成就返回哪个结果
+  const result = await Promise.race([waitForLoggedIn, waitForLoginButton]);
+
+  if (result === "loginButton") {
+    // 明确看到登录按钮 → 未登录
+    return false;
   }
-
-  if (!selectorFound) {
-    // 所有阶段都超时，DOM 无法判断
-    console.error(`[登录检测] DOM 验真超时 (${timeouts[timeouts.length - 1]}ms)，无法判断`);
-    return null;
+  if (result === "loggedIn") {
+    // 看到已登录元素 → 已登录
+    return true;
   }
 
+  // 都超时了，回退到 evaluate 做最终判断
   return page.evaluate(() => {
     const hasProfileContainer = !!document.querySelector(
       '[class*="DivProfileContainer"], [class*="DivUserContainer"]',
@@ -167,7 +167,7 @@ export async function isLoggedInByDom(page) {
     if (hasLoginButton) return false;
     // 看到已登录元素 → 已登录
     if (hasProfileContainer || hasUserMenu) return true;
-    // 元素已出现但都不是登录/未登录标志 → 无法判断
+    // 都无法判断
     return null;
   });
 }