@skrillex1224/playwright-toolkit 2.1.49 → 2.1.51

package/browser.d.ts

@@ -0,0 +1,3 @@
+export * from './index.d.ts';
+
+export declare function usePlaywrightToolKit(): BrowserModule;

package/dist/browser.js

@@ -1,4 +1,10 @@
-// src/logger.js
+var __defProp = Object.defineProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+
+// src/internals/logger.js
 var formatLine = (prefix, icon, message) => {
   const parts = [];
   if (prefix) parts.push(`[${prefix}]`);
@@ -12,22 +18,22 @@ var fallbackLog = {
   error: (...args) => console.error(...args),
   debug: (...args) => console.debug ? console.debug(...args) : console.log(...args)
 };
-var resolveLogMethod = (logger, name) => {
-  if (logger && typeof logger[name] === "function") {
-    return logger[name].bind(logger);
+var resolveLogMethod = (logger2, name) => {
+  if (logger2 && typeof logger2[name] === "function") {
+    return logger2[name].bind(logger2);
   }
-  if (name === "warning" && logger && typeof logger.warn === "function") {
-    return logger.warn.bind(logger);
+  if (name === "warning" && logger2 && typeof logger2.warn === "function") {
+    return logger2.warn.bind(logger2);
   }
   return fallbackLog[name];
 };
 var defaultLogger = null;
-var setDefaultLogger = (logger) => {
-  defaultLogger = logger;
+var setDefaultLogger = (logger2) => {
+  defaultLogger = logger2;
 };
-var resolveLogger = (logger) => {
-  if (logger && typeof logger.info === "function") {
-    return logger;
+var resolveLogger = (explicitLogger) => {
+  if (explicitLogger && typeof explicitLogger.info === "function") {
+    return explicitLogger;
   }
   if (defaultLogger && typeof defaultLogger.info === "function") {
     return defaultLogger;
@@ -43,31 +49,62 @@ var ANSI = {
   blue: "\x1B[34m",
   cyan: "\x1B[36m"
 };
-var stripAnsi = (input) => {
-  if (!input) return "";
-  return String(input).replace(/\x1b\[[0-9;]*m/g, "");
-};
 var colorize = (text, color) => {
   if (!text || !color) return text;
   return `${color}${text}${ANSI.reset}`;
 };
-var createBaseLogger = (prefix = "", logger) => {
+var createBaseLogger = (prefix = "", explicitLogger) => {
   const name = prefix ? String(prefix) : "";
-  const targetLogger = resolveLogger(logger);
-  const info = resolveLogMethod(targetLogger, "info");
-  const warning = resolveLogMethod(targetLogger, "warning");
-  const error = resolveLogMethod(targetLogger, "error");
-  const debug = resolveLogMethod(targetLogger, "debug");
+  const dispatch = (methodName, icon, message, color) => {
+    const logger2 = resolveLogger(explicitLogger);
+    const logFn = resolveLogMethod(logger2, methodName);
+    logFn(colorize(formatLine(name, icon, message), color));
+  };
   return {
-    info: (message) => info(colorize(formatLine(name, "\u{1F4D6}", message), ANSI.cyan)),
-    success: (message) => info(colorize(formatLine(name, "\u2705", message), ANSI.green)),
-    warning: (message) => warning(colorize(formatLine(name, "\u26A0\uFE0F", message), ANSI.yellow)),
-    warn: (message) => warning(colorize(formatLine(name, "\u26A0\uFE0F", message), ANSI.yellow)),
-    error: (message) => error(colorize(formatLine(name, "\u274C", message), ANSI.red)),
-    debug: (message) => debug(colorize(formatLine(name, "\u{1F539}", message), ANSI.gray)),
-    start: (message) => info(colorize(formatLine(name, "\u{1F537}", message), ANSI.blue))
+    info: (message) => dispatch("info", "\u{1F4D6}", message, ANSI.cyan),
+    success: (message) => dispatch("info", "\u2705", message, ANSI.green),
+    warning: (message) => dispatch("warning", "\u26A0\uFE0F", message, ANSI.yellow),
+    warn: (message) => dispatch("warning", "\u26A0\uFE0F", message, ANSI.yellow),
+    error: (message) => dispatch("error", "\u274C", message, ANSI.red),
+    debug: (message) => dispatch("debug", "\u{1F539}", message, ANSI.gray),
+    start: (message) => dispatch("info", "\u{1F537}", message, ANSI.blue)
   };
 };
+function createInternalLogger(moduleName, explicitLogger) {
+  const baseLogger = createBaseLogger(moduleName, explicitLogger);
+  return {
+    start(methodName, params = "") {
+      const paramStr = params ? ` (${params})` : "";
+      baseLogger.start(`${methodName} \u5F00\u59CB${paramStr}`);
+    },
+    success(methodName, result = "") {
+      const resultStr = result ? ` (${result})` : "";
+      baseLogger.success(`${methodName} \u5B8C\u6210${resultStr}`);
+    },
+    fail(methodName, error) {
+      const message = error instanceof Error ? error.message : error;
+      baseLogger.error(`${methodName} \u5931\u8D25: ${message}`);
+    },
+    debug(message) {
+      baseLogger.debug(message);
+    },
+    warn(message) {
+      baseLogger.warning(message);
+    },
+    warning(message) {
+      baseLogger.warning(message);
+    },
+    info(message) {
+      baseLogger.info(message);
+    }
+  };
+}
+
+// src/logger.js
+var stripAnsi = (input) => {
+  if (!input) return "";
+  return String(input).replace(/\x1b\[[0-9;]*m/g, "");
+};
 var STEP_PREFIX = "\u6B65\u9AA4:";
 var STEP_SEPARATOR = " | ";
 var STEP_EMOJIS = [
@@ -762,9 +799,9 @@ var createTemplateLogger = (baseLogger = createBaseLogger()) => {
     ...definitionMethods
   };
 };
-var getDefaultBaseLogger = () => createBaseLogger("", defaultLogger);
+var getDefaultBaseLogger = () => createBaseLogger("");
 var Logger = {
-  setLogger: (logger) => setDefaultLogger(logger),
+  setLogger: (logger2) => setDefaultLogger(logger2),
   info: (message) => getDefaultBaseLogger().info(message),
   success: (message) => getDefaultBaseLogger().success(message),
   warning: (message) => getDefaultBaseLogger().warning(message),
@@ -772,19 +809,272 @@ var Logger = {
   error: (message) => getDefaultBaseLogger().error(message),
   debug: (message) => getDefaultBaseLogger().debug(message),
   start: (message) => getDefaultBaseLogger().start(message),
-  useTemplate: (logger) => createTemplateLogger(createBaseLogger("", logger || defaultLogger))
+  useTemplate: (logger2) => {
+    if (logger2) return createTemplateLogger(createBaseLogger("", logger2));
+    return createTemplateLogger();
+  }
+};
+
+// src/browser-onlys/index.js
+var browser_onlys_exports = {};
+__export(browser_onlys_exports, {
+  PLATFORM_CONFIG: () => PLATFORM_CONFIG,
+  TokenForm: () => TokenForm
+});
+
+// src/utils.js
+import delay from "delay";
+var logger = createInternalLogger("Utils");
+var Utils = {
+  /**
+  * 解析 Cookie 字符串为 Playwright 格式的 Cookie 数组
+  * @param {string} cookieString - Cookie 字符串
+  * @param {string} [domain] - Cookie 域名 (可选)
+  * @returns {Array} Cookie 数组
+  */
+  parseCookies(cookieString, domain) {
+    const cookies = [];
+    const pairs = cookieString.split(";").map((c) => c.trim());
+    for (const pair of pairs) {
+      const [name, ...valueParts] = pair.split("=");
+      if (name && valueParts.length > 0) {
+        const cookie = {
+          name: name.trim(),
+          value: valueParts.join("=").trim(),
+          path: "/"
+        };
+        if (domain) {
+          cookie.domain = domain;
+        }
+        cookies.push(cookie);
+      }
+    }
+    logger.success("parseCookies", `parsed ${cookies.length} cookies`);
+    return cookies;
+  },
+  /**
+   * 全页面滚动截图
+   * 自动检测页面所有可滚动元素，取最大高度，强制展开后截图
+   * 
+   * @param {import('playwright').Page} page - Playwright page 对象
+   * @param {Object} [options] - 配置选项
+   * @param {number} [options.buffer] - 额外缓冲高度 (默认: 视口高度的一半)
+   * @param {boolean} [options.restore] - 截图后是否恢复页面高度和样式 (默认: false)
+   * @param {number} [options.maxHeight] - 最大截图高度 (默认: 8000px)
+   * @returns {Promise<string>} - base64 编码的 PNG 图片
+   */
+  async fullPageScreenshot(page, options = {}) {
+    var _a, _b, _c;
+    logger.start("fullPageScreenshot", "detecting scrollable elements");
+    const originalViewport = page.viewportSize();
+    const defaultBuffer = Math.round(((originalViewport == null ? void 0 : originalViewport.height) || 1080) / 2);
+    const buffer = (_a = options.buffer) != null ? _a : defaultBuffer;
+    const restore = (_b = options.restore) != null ? _b : false;
+    const maxHeight = (_c = options.maxHeight) != null ? _c : 8e3;
+    try {
+      const maxScrollHeight = await page.evaluate(() => {
+        let maxHeight2 = document.body.scrollHeight;
+        document.querySelectorAll("*").forEach((el) => {
+          const style = window.getComputedStyle(el);
+          const overflowY = style.overflowY;
+          if ((overflowY === "auto" || overflowY === "scroll") && el.scrollHeight > el.clientHeight) {
+            if (el.scrollHeight > maxHeight2) {
+              maxHeight2 = el.scrollHeight;
+            }
+            el.dataset.pkOrigOverflow = el.style.overflow;
+            el.dataset.pkOrigHeight = el.style.height;
+            el.dataset.pkOrigMaxHeight = el.style.maxHeight;
+            el.classList.add("__pk_expanded__");
+            el.style.overflow = "visible";
+            el.style.height = "auto";
+            el.style.maxHeight = "none";
+          }
+        });
+        return maxHeight2;
+      });
+      const targetHeight = Math.min(maxScrollHeight + buffer, maxHeight);
+      await page.setViewportSize({
+        width: (originalViewport == null ? void 0 : originalViewport.width) || 1280,
+        height: targetHeight
+      });
+      await delay(1e3);
+      const buffer_ = await page.screenshot({
+        fullPage: true,
+        type: "png"
+      });
+      logger.success("fullPageScreenshot", `captured ${Math.round(buffer_.length / 1024)} KB`);
+      return buffer_.toString("base64");
+    } finally {
+      if (restore) {
+        await page.evaluate(() => {
+          document.querySelectorAll(".__pk_expanded__").forEach((el) => {
+            el.style.overflow = el.dataset.pkOrigOverflow || "";
+            el.style.height = el.dataset.pkOrigHeight || "";
+            el.style.maxHeight = el.dataset.pkOrigMaxHeight || "";
+            delete el.dataset.pkOrigOverflow;
+            delete el.dataset.pkOrigHeight;
+            delete el.dataset.pkOrigMaxHeight;
+            el.classList.remove("__pk_expanded__");
+          });
+        });
+        if (originalViewport) {
+          await page.setViewportSize(originalViewport);
+        }
+      }
+    }
+  }
+};
+
+// src/browser-onlys/token-form.js
+var PLATFORM_CONFIG = {
+  kimi: {
+    name: "Kimi (Moonshot)",
+    tokenName: "kimi_auth",
+    serviceUrl: "https://kimi.moonshot.cn/",
+    tokenFields: ["kimi_auth"],
+    parse: (token) => {
+      const val = token ? token.trim() : "";
+      return val ? { kimi_auth: val } : {};
+    },
+    format: (values) => {
+      const val = values.kimi_auth;
+      return val === void 0 || val === null ? "" : String(val).trim();
+    }
+  },
+  doubao: {
+    name: "\u8C46\u5305 (Doubao)",
+    tokenName: "sessionid",
+    serviceUrl: "https://www.doubao.com/",
+    tokenFields: ["sessionid"],
+    parse: (token) => {
+      const val = token ? token.trim() : "";
+      return val ? { sessionid: val } : {};
+    },
+    format: (values) => {
+      const val = values.sessionid;
+      return val === void 0 || val === null ? "" : String(val).trim();
+    }
+  },
+  deepseek: {
+    name: "DeepSeek",
+    tokenName: "userToken",
+    serviceUrl: "https://chat.deepseek.com/",
+    guide: '\u8BF7\u8F93\u5165\u5B8C\u6574 JSON\uFF0C\u4F8B\u5982 {"value":"xxxxx"}\uFF0C\u7CFB\u7EDF\u4F1A\u81EA\u52A8\u63D0\u53D6 value\u3002',
+    tokenFields: ["userToken"],
+    parse: (token) => {
+      const val = token ? token.trim() : "";
+      return val ? { userToken: val } : {};
+    },
+    format: (values) => {
+      const raw = values.userToken ? String(values.userToken).trim() : "";
+      if (!raw) return "";
+      if (raw.startsWith("{")) {
+        try {
+          const parsed = JSON.parse(raw);
+          if (parsed && typeof parsed.value === "string") return parsed.value.trim();
+        } catch (e) {
+          return "";
+        }
+        return "";
+      }
+      return raw;
+    }
+  },
+  yuanbao: {
+    name: "\u5143\u5B9D (Tencent)",
+    tokenName: "hy_user, hy_token",
+    serviceUrl: "https://yuanbao.tencent.com/",
+    guide: "\u5206\u522B\u586B\u5165\u4ECE\u6D4F\u89C8\u5668\u83B7\u53D6\u7684 hy_user \u548C hy_token\uFF0C\u7CFB\u7EDF\u5C06\u81EA\u52A8\u62FC\u63A5\u3002",
+    tokenFields: ["hy_user", "hy_token"],
+    parse: (token) => {
+      const cookieList = Utils.parseCookies(token || "");
+      const pairs = {};
+      cookieList.forEach((c) => {
+        pairs[c.name] = c.value;
+      });
+      return {
+        hy_user: pairs["hy_user"] || "",
+        hy_token: pairs["hy_token"] || ""
+      };
+    },
+    format: (values) => {
+      const u = values.hy_user ? String(values.hy_user).trim() : "";
+      const t = values.hy_token ? String(values.hy_token).trim() : "";
+      if (!u || !t) return "";
+      return `hy_user=${u}; hy_token=${t}`;
+    }
+  },
+  qwen: {
+    name: "\u901A\u4E49\u5343\u95EE (Qwen)",
+    tokenName: "tongyi_sso_ticket",
+    serviceUrl: "https://tongyi.aliyun.com/",
+    tokenFields: ["tongyi_sso_ticket"],
+    parse: (token) => {
+      const val = token ? token.trim() : "";
+      return val ? { tongyi_sso_ticket: val } : {};
+    },
+    format: (values) => {
+      const val = values.tongyi_sso_ticket;
+      return val === void 0 || val === null ? "" : String(val).trim();
+    }
+  },
+  erine: {
+    name: "\u6587\u5FC3\u4E00\u8A00 (Erine)",
+    tokenName: "BDUSS",
+    serviceUrl: "https://yiyan.baidu.com/",
+    tokenFields: ["BDUSS"],
+    parse: (token) => {
+      const val = token ? token.trim() : "";
+      return val ? { BDUSS: val } : {};
+    },
+    format: (values) => {
+      const val = values.BDUSS;
+      return val === void 0 || val === null ? "" : String(val).trim();
+    }
+  }
+};
+function createTokenForm() {
+  return {
+    getPlatformConfig(actor) {
+      if (!actor) throw new Error("Actor parameter is required to get platform config");
+      if (PLATFORM_CONFIG[actor]) return PLATFORM_CONFIG[actor];
+      const key = Object.keys(PLATFORM_CONFIG).find((k) => actor.includes(k));
+      if (key) return PLATFORM_CONFIG[key];
+      throw new Error(`\u672A\u627E\u5230\u5E73\u53F0\u914D\u7F6E: ${actor}. \u8BF7\u786E\u4FDD playwright-toolkit/browser-onlys/token-form.js \u4E2D\u5DF2\u5B9A\u4E49\u8BE5\u5E73\u53F0\u3002`);
+    },
+    parseToken(token, config) {
+      if (!config) return {};
+      return config.parse(token);
+    },
+    formatToken(values, config) {
+      if (!config) return "";
+      return config.format(values);
+    }
+  };
+}
+var instance = null;
+async function useTokenForm() {
+  if (!instance) {
+    instance = createTokenForm();
+  }
+  return instance;
+}
+var TokenForm = {
+  useTokenForm
 };
 
 // entrys/browser.js
 var usePlaywrightToolKit = () => {
-  throw new Error("usePlaywrightToolKit \u4EC5\u652F\u6301\u5728 Node \u73AF\u5883\u4E2D\u4F7F\u7528\u3002");
+  return {
+    Logger,
+    ...browser_onlys_exports,
+    $Internals: {
+      LOG_TEMPLATES,
+      stripAnsi
+    }
+  };
 };
-var browser = { Logger, LOG_TEMPLATES, stripAnsi };
 export {
-  LOG_TEMPLATES,
-  Logger,
-  browser,
-  stripAnsi,
   usePlaywrightToolKit
 };
 //# sourceMappingURL=browser.js.map