@taole/dd-playwright 0.0.0-stage → 0.1.1

package/src/auth/diandian-auth.mjs

@@ -0,0 +1,279 @@
+import { chromium } from "playwright";
+
+export const DIANDIAN_HOME_URL = "https://y.tuwan.com";
+export const DIANDIAN_USER_INFO_URL =
+  "https://yapi.tuwan.com/activityapi/getUserInfo?format=json";
+export const DIANDIAN_LOGIN_TIMEOUT_MS = 5 * 60 * 1000;
+
+export function cookiesToHeaderString(cookies) {
+  if (!Array.isArray(cookies)) return "";
+  return cookies
+    .filter((c) => c && typeof c.name === "string" && c.value != null)
+    .map((c) => `${c.name}=${c.value}`)
+    .join("; ");
+}
+
+export class DiandianAuthManager {
+  /**
+   * @param {object} options
+   * @param {import('../account-store.mjs').AccountStore} options.accountStore
+   * @param {any} [options.chromiumLauncher] 可选自定义 chromium 启动器 (测试用)
+   */
+  constructor({ accountStore, chromiumLauncher = chromium } = {}) {
+    this.accountStore = accountStore;
+    this.chromium = chromiumLauncher;
+
+    this.browser = null;
+    this.context = null;
+    this.page = null;
+    this.startupPromise = null;
+    this.timeoutTimer = null;
+    this.scanStartedAt = 0;
+  }
+
+  get isScanning() {
+    return Boolean(this.page && !this.page.isClosed());
+  }
+
+  async startLogin({ headless = true, forceNew = false } = {}) {
+    if (this.isScanning && !forceNew) {
+      return { status: "scanning", message: "扫码登录会话已在进行中" };
+    }
+    if (this.startupPromise) return this.startupPromise;
+
+    this.startupPromise = (async () => {
+      await this.close();
+
+      const instance = await this.chromium.launch({ headless });
+      const ctx = await instance.newContext({
+        viewport: { width: 1100, height: 850 },
+        userAgent:
+          "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36",
+      });
+      const tab = await ctx.newPage();
+
+      this.browser = instance;
+      this.context = ctx;
+      this.page = tab;
+      this.scanStartedAt = Date.now();
+
+      clearTimeout(this.timeoutTimer);
+      this.timeoutTimer = setTimeout(() => {
+        this.close().catch(() => {});
+      }, DIANDIAN_LOGIN_TIMEOUT_MS);
+
+      // 导航至点点首页
+      await tab
+        .goto(DIANDIAN_HOME_URL, {
+          waitUntil: "domcontentloaded",
+          timeout: 30000,
+        })
+        .catch(() => {});
+
+      // 自动触发登录弹窗并主动切换至「扫码登录」模式
+      try {
+        const loginBtn = await tab
+          .waitForSelector(".header-iconbtns-login", { timeout: 12000 })
+          .catch(() => null);
+        if (loginBtn) {
+          await loginBtn.click().catch(() => {});
+        }
+
+        await tab
+          .waitForSelector(".login-window-content", { timeout: 10000 })
+          .catch(() => null);
+        await tab.waitForTimeout(600);
+
+        await tab
+          .evaluate(() => {
+            if (window.$) {
+              window
+                .$('.login-window-other-mobile[data-logintype="qrcode"]')
+                .trigger("click");
+              window.$(".login-window-other-mobile").trigger("click");
+            }
+            const qrcodeBtn =
+              document.querySelector(
+                '.login-window-other-mobile[data-logintype="qrcode"]',
+              ) ||
+              Array.from(document.querySelectorAll("*")).find(
+                (el) =>
+                  (el.innerText || "").trim() === "扫码登录" &&
+                  el.children.length === 0,
+              );
+            if (qrcodeBtn) qrcodeBtn.click();
+          })
+          .catch(() => {});
+
+        await tab.waitForTimeout(1000);
+      } catch {}
+
+      return {
+        status: "scanning",
+        message: "扫码登录页面已就绪，请使用微信或点点 App 扫码登录",
+      };
+    })();
+
+    try {
+      return await this.startupPromise;
+    } finally {
+      this.startupPromise = null;
+    }
+  }
+
+  async getScreenshot() {
+    if (!this.page || this.page.isClosed()) {
+      throw new Error("扫码登录会话未启动或已关闭");
+    }
+
+    // 自愈检查：确保在扫码模式
+    try {
+      await this.page
+        .evaluate(() => {
+          const text =
+            document.querySelector(".login-window-content")?.innerText || "";
+          if (
+            !text.includes("扫码登录") ||
+            text.includes("请输入用户名/手机号") ||
+            text.includes("忘记密码")
+          ) {
+            if (window.$) {
+              window
+                .$('.login-window-other-mobile[data-logintype="qrcode"]')
+                .trigger("click");
+            }
+            document
+              .querySelector(
+                '.login-window-other-mobile[data-logintype="qrcode"]',
+              )
+              ?.click();
+          }
+        })
+        .catch(() => {});
+    } catch {}
+    // 1. 优先截取纯净的二维码元素（放大高清，便于手机扫码识别）
+    try {
+      const qrElement =
+        (await this.page.$(".login-window-qrcodelogin")) ||
+        (await this.page.$(".login-window-content img[src^='data:image']"));
+      if (qrElement) {
+        const box = await qrElement.boundingBox().catch(() => null);
+        if (box && box.width > 50 && box.height > 50) {
+          return await qrElement.screenshot({ type: "png", timeout: 8000 });
+        }
+      }
+    } catch {}
+
+    // 2. 兜底截取登录弹窗主体
+    try {
+      const modal = await this.page.$(".login-window-content");
+      if (modal) {
+        const box = await modal.boundingBox().catch(() => null);
+        if (box && box.width > 150 && box.height > 150) {
+          return await modal.screenshot({ type: "png", timeout: 8000 });
+        }
+      }
+    } catch {}
+    return this.page.screenshot({
+      type: "png",
+      timeout: 10000,
+    });
+  }
+
+  async verifyCookie(cookieHeader) {
+    if (!cookieHeader || typeof cookieHeader !== "string") return null;
+    try {
+      const res = await fetch(DIANDIAN_USER_INFO_URL, {
+        headers: {
+          Cookie: cookieHeader,
+          Accept: "application/json",
+          "User-Agent":
+            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36",
+        },
+      });
+      if (!res.ok) return null;
+      const json = await res.json();
+      if (json && (json.code === 0 || json.error === 0) && json.data) {
+        const d = json.data;
+        return {
+          uid: String(d.uid || d.id || ""),
+          nickname: d.nickname || d.username || "点点用户",
+          avatar: d.avatar || d.headimgurl || "",
+        };
+      }
+      return null;
+    } catch {
+      return null;
+    }
+  }
+
+  async checkStatus() {
+    if (!this.context || !this.page || this.page.isClosed()) {
+      return { status: "not_started", message: "未开始扫码登录" };
+    }
+
+    try {
+      const cookies = await this.context.cookies([
+        "https://y.tuwan.com",
+        "https://tuwan.com",
+      ]);
+
+      const passportCookie = cookies.find((c) => c.name === "Tuwan_Passport");
+      if (!passportCookie || !passportCookie.value) {
+        return { status: "scanning", message: "等待扫码或登录中..." };
+      }
+
+      const storageState = await this.context.storageState();
+      const cookieHeader = cookiesToHeaderString(
+        storageState.cookies || cookies,
+      );
+
+      const userInfo = await this.verifyCookie(cookieHeader);
+      const uid = userInfo?.uid || passportCookie.value.slice(0, 10);
+      const nickname = userInfo?.nickname || "点点用户";
+      const avatar = userInfo?.avatar || "";
+
+      let savedAccount = null;
+      if (this.accountStore) {
+        savedAccount = this.accountStore.saveAccount({
+          accountId: uid,
+          nickname,
+          avatar,
+          cookieHeader,
+          storageState,
+          isActive: true,
+        });
+      }
+
+      // 扫码成功后释放浏览器资源
+      await this.close();
+
+      return {
+        status: "ready",
+        message: "点点账号绑定成功",
+        user: { uid, nickname, avatar },
+        account: savedAccount,
+      };
+    } catch (err) {
+      return {
+        status: "scanning",
+        message: `检查登录状态中: ${err.message}`,
+      };
+    }
+  }
+
+  async close() {
+    clearTimeout(this.timeoutTimer);
+    this.timeoutTimer = null;
+    this.scanStartedAt = 0;
+
+    const b = this.browser;
+    this.browser = null;
+    this.context = null;
+    this.page = null;
+
+    if (b) {
+      await b.close().catch(() => {});
+    }
+  }
+}