tt-help-cli-ycl 1.3.33 → 1.3.34

package/src/lib/browser/page.js

@@ -1,183 +1,183 @@
-import os from "os";
-import path from "path";
-import { delay } from "../delay.js";
-import { retryWithBackoff } from "../retry.js";
-import { getDelayConfig } from "../delay.js";
-import { ensureBrowserReady, killEdgeProcesses } from "./cdp.js";
-
-const DEFAULT_USER_DATA_DIR = path.join(
-  os.homedir(),
-  "Library",
-  "Application Support",
-  "Microsoft Edge For Testing",
-);
-
-const BROWSER_CLOSED_PATTERNS = [
-  "Target page, context or browser has been closed",
-  "Target closed",
-  "Browser has been closed",
-  "Protocol error",
-  "Connection closed",
-  "net::ERR_CONNECTION",
-];
-
-export function isBrowserClosedError(error) {
-  if (!error) return false;
-  const msg = error.message || error.toString() || "";
-  return BROWSER_CLOSED_PATTERNS.some((p) => msg.includes(p));
-}
-
-export async function relaunchBrowser(cdpOptions, port) {
-  console.error(`  [浏览器] 浏览器已关闭，正在重启 (端口 ${port})...`);
-  const targetDir = cdpOptions.userDataDir || DEFAULT_USER_DATA_DIR;
-  // kill 并清理 Singleton 锁文件，确保 Edge 能启动新实例
-  await killEdgeProcesses(targetDir);
-  await new Promise((r) => setTimeout(r, 3000));
-  // 确保端口已释放后再启动
-  let retries = 0;
-  while (retries < 5) {
-    try {
-      return await ensureBrowserReady(cdpOptions);
-    } catch (e) {
-      if (e.message && e.message.includes("setDownloadBehavior")) {
-        retries++;
-        console.error(`  [浏览器] CDP 连接异常，重试 ${retries}/5...`);
-        await new Promise((r) => setTimeout(r, 3000));
-        await killEdgeProcesses(targetDir);
-        await new Promise((r) => setTimeout(r, 5000));
-        continue;
-      }
-      throw e;
-    }
-  }
-  throw new Error("浏览器重启失败，CDP 连接异常");
-}
-
-export async function withBrowserRecovery(fn, browser, page, cdpOptions, port) {
-  try {
-    return await fn();
-  } catch (e) {
-    if (isBrowserClosedError(e)) {
-      const newBrowser = await relaunchBrowser(cdpOptions, port);
-      const newPage = await getOrCreatePage(newBrowser);
-      return await fn(newBrowser, newPage);
-    }
-    throw e;
-  }
-}
-
-/**
- * 通过 Cookie 判断登录状态（推荐，最可靠）
- * 已登录：存在 sessionid Cookie
- * 未登录：不存在 sessionid Cookie
- */
-export async function isLoggedIn(page) {
-  const cookies = await page.context().cookies("https://www.tiktok.com");
-  return cookies.some((c) => c.name === "sessionid");
-}
-
-/**
- * 通过 DOM 元素判断登录状态（备用方案）
- * 依赖页面渲染完成，可能存在误判
- * @deprecated 推荐使用 isLoggedIn（基于 Cookie）
- */
-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(", ");
-
-  await page
-    .waitForSelector(loginOrLoggedInSelector, { timeout: 10000 })
-    .catch(() => {});
-
-  return page.evaluate(() => {
-    const hasProfileContainer = !!document.querySelector(
-      '[class*="DivProfileContainer"], [class*="DivUserContainer"]',
-    );
-    const hasUserMenu = !!document.querySelector(
-      '[class*="UserMenu"], [class*="user-menu"], [class*="CurrentUserInfo"]',
-    );
-    const hasLoginButton = Array.from(
-      document.querySelectorAll('button, [role="button"]'),
-    ).some((el) => /^(登录|Log in|Sign in)$/i.test(el.textContent.trim()));
-
-    return (hasProfileContainer || hasUserMenu) && !hasLoginButton;
-  });
-}
-
-export async function closeCommentPanel(page) {
-  await page.evaluate(() => {
-    const rightPanel = document.querySelector('[class*="RightPanelContainer"]');
-    if (rightPanel) {
-      const tabContainer = rightPanel.querySelector('[class*="TabContainer"]');
-      if (tabContainer) {
-        const closeOverlay = tabContainer.querySelector("div:last-child");
-        if (closeOverlay) closeOverlay.click();
-      }
-    }
-  });
-}
-
-export async function ensureTikTokPage(browser, url) {
-  const contexts = browser.contexts();
-  let page = null;
-
-  for (const ctx of contexts) {
-    for (const p of ctx.pages()) {
-      if (p.url().includes("tiktok.com")) {
-        page = p;
-        break;
-      }
-    }
-    if (page) break;
-  }
-
-  if (!page) {
-    console.error("未找到 TikTok 页面，正在打开...");
-    const defaultCtx = browser.contexts()[0];
-    page = await defaultCtx.newPage();
-    await retryWithBackoff(() =>
-      page.goto(url, { waitUntil: "load", timeout: 30000 }),
-    );
-    const config = getDelayConfig();
-    await delay(Math.round(config.switchMax * 0.5), config.switchMax);
-    console.error("TikTok 页面已打开");
-  }
-
-  return page;
-}
-
-export async function findTikTokPage(browser) {
-  const contexts = browser.contexts();
-  for (const ctx of contexts) {
-    for (const p of ctx.pages()) {
-      if (p.url().includes("tiktok.com")) return p;
-    }
-  }
-  return null;
-}
-
-export async function getOrCreatePage(browser) {
-  let page = await findTikTokPage(browser);
-  if (!page) {
-    const defaultCtx = browser.contexts()[0] || (await browser.newContext());
-    page = await defaultCtx.newPage();
-  }
-  return page;
-}
-
-export function assertPageUrl(page, expectedPath) {
-  const actual = page.url();
-  if (!actual.includes(expectedPath)) {
-    throw new Error(
-      `[代理错误] 预期访问 ${expectedPath}，实际跳转到了 ${actual}`,
-    );
-  }
-}
+import os from "os";
+import path from "path";
+import { delay } from "../delay.js";
+import { retryWithBackoff } from "../retry.js";
+import { getDelayConfig } from "../delay.js";
+import { ensureBrowserReady, killEdgeProcesses } from "./cdp.js";
+
+const DEFAULT_USER_DATA_DIR = path.join(
+  os.homedir(),
+  "Library",
+  "Application Support",
+  "Microsoft Edge For Testing",
+);
+
+const BROWSER_CLOSED_PATTERNS = [
+  "Target page, context or browser has been closed",
+  "Target closed",
+  "Browser has been closed",
+  "Protocol error",
+  "Connection closed",
+  "net::ERR_CONNECTION",
+];
+
+export function isBrowserClosedError(error) {
+  if (!error) return false;
+  const msg = error.message || error.toString() || "";
+  return BROWSER_CLOSED_PATTERNS.some((p) => msg.includes(p));
+}
+
+export async function relaunchBrowser(cdpOptions, port) {
+  console.error(`  [浏览器] 浏览器已关闭，正在重启 (端口 ${port})...`);
+  const targetDir = cdpOptions.userDataDir || DEFAULT_USER_DATA_DIR;
+  // kill 并清理 Singleton 锁文件，确保 Edge 能启动新实例
+  await killEdgeProcesses(targetDir);
+  await new Promise((r) => setTimeout(r, 3000));
+  // 确保端口已释放后再启动
+  let retries = 0;
+  while (retries < 5) {
+    try {
+      return await ensureBrowserReady(cdpOptions);
+    } catch (e) {
+      if (e.message && e.message.includes("setDownloadBehavior")) {
+        retries++;
+        console.error(`  [浏览器] CDP 连接异常，重试 ${retries}/5...`);
+        await new Promise((r) => setTimeout(r, 3000));
+        await killEdgeProcesses(targetDir);
+        await new Promise((r) => setTimeout(r, 5000));
+        continue;
+      }
+      throw e;
+    }
+  }
+  throw new Error("浏览器重启失败，CDP 连接异常");
+}
+
+export async function withBrowserRecovery(fn, browser, page, cdpOptions, port) {
+  try {
+    return await fn();
+  } catch (e) {
+    if (isBrowserClosedError(e)) {
+      const newBrowser = await relaunchBrowser(cdpOptions, port);
+      const newPage = await getOrCreatePage(newBrowser);
+      return await fn(newBrowser, newPage);
+    }
+    throw e;
+  }
+}
+
+/**
+ * 通过 Cookie 判断登录状态（推荐，最可靠）
+ * 已登录：存在 sessionid Cookie
+ * 未登录：不存在 sessionid Cookie
+ */
+export async function isLoggedIn(page) {
+  const cookies = await page.context().cookies("https://www.tiktok.com");
+  return cookies.some((c) => c.name === "sessionid");
+}
+
+/**
+ * 通过 DOM 元素判断登录状态（备用方案）
+ * 依赖页面渲染完成，可能存在误判
+ * @deprecated 推荐使用 isLoggedIn（基于 Cookie）
+ */
+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(", ");
+
+  await page
+    .waitForSelector(loginOrLoggedInSelector, { timeout: 10000 })
+    .catch(() => {});
+
+  return page.evaluate(() => {
+    const hasProfileContainer = !!document.querySelector(
+      '[class*="DivProfileContainer"], [class*="DivUserContainer"]',
+    );
+    const hasUserMenu = !!document.querySelector(
+      '[class*="UserMenu"], [class*="user-menu"], [class*="CurrentUserInfo"]',
+    );
+    const hasLoginButton = Array.from(
+      document.querySelectorAll('button, [role="button"]'),
+    ).some((el) => /^(登录|Log in|Sign in)$/i.test(el.textContent.trim()));
+
+    return (hasProfileContainer || hasUserMenu) && !hasLoginButton;
+  });
+}
+
+export async function closeCommentPanel(page) {
+  await page.evaluate(() => {
+    const rightPanel = document.querySelector('[class*="RightPanelContainer"]');
+    if (rightPanel) {
+      const tabContainer = rightPanel.querySelector('[class*="TabContainer"]');
+      if (tabContainer) {
+        const closeOverlay = tabContainer.querySelector("div:last-child");
+        if (closeOverlay) closeOverlay.click();
+      }
+    }
+  });
+}
+
+export async function ensureTikTokPage(browser, url) {
+  const contexts = browser.contexts();
+  let page = null;
+
+  for (const ctx of contexts) {
+    for (const p of ctx.pages()) {
+      if (p.url().includes("tiktok.com")) {
+        page = p;
+        break;
+      }
+    }
+    if (page) break;
+  }
+
+  if (!page) {
+    console.error("未找到 TikTok 页面，正在打开...");
+    const defaultCtx = browser.contexts()[0];
+    page = await defaultCtx.newPage();
+    await retryWithBackoff(() =>
+      page.goto(url, { waitUntil: "load", timeout: 30000 }),
+    );
+    const config = getDelayConfig();
+    await delay(Math.round(config.switchMax * 0.5), config.switchMax);
+    console.error("TikTok 页面已打开");
+  }
+
+  return page;
+}
+
+export async function findTikTokPage(browser) {
+  const contexts = browser.contexts();
+  for (const ctx of contexts) {
+    for (const p of ctx.pages()) {
+      if (p.url().includes("tiktok.com")) return p;
+    }
+  }
+  return null;
+}
+
+export async function getOrCreatePage(browser) {
+  let page = await findTikTokPage(browser);
+  if (!page) {
+    const defaultCtx = browser.contexts()[0] || (await browser.newContext());
+    page = await defaultCtx.newPage();
+  }
+  return page;
+}
+
+export function assertPageUrl(page, expectedPath) {
+  const actual = page.url();
+  if (!actual.includes(expectedPath)) {
+    throw new Error(
+      `[代理错误] 预期访问 ${expectedPath}，实际跳转到了 ${actual}`,
+    );
+  }
+}