拆分腾讯会话页面交互辅助逻辑

2026-05-04 14:18:55 +08:00
parent 37299a2797
commit 635e0138fe
3 changed files with 285 additions and 157 deletions
@@ -0,0 +1,188 @@
+// @ts-nocheck
+
+import fs from 'node:fs/promises'
+import path from 'node:path'
+
+import { isRetryableQrCaptureError, logBrowserSessionDebug } from './session-login-shared.js'
+import { captureQqQrImage, ensureQqLoginReady, extractQqQrState } from './session-qq.js'
+import { captureWxQrImage, ensureWxLoginReady, extractWxQrState } from './session-wx.js'
+
+export async function captureSessionQr(
+  session,
+  {
+    ensureLoginReady = true,
+    ensureLoginTab: ensureLoginTabImpl = ensureLoginTab,
+    captureQrImageWithRetry: captureQrImageWithRetryImpl = captureQrImageWithRetry,
+    readFile = fs.readFile,
+    now = () => new Date().toISOString(),
+  } = {},
+) {
+  if (ensureLoginReady) {
+    await ensureLoginTabImpl(session.page, session.loginType)
+  }
+
+  const qrImagePath = path.join(session.sessionDir, `${session.loginType}-qr.png`)
+  await captureQrImageWithRetryImpl(session.page, session.loginType, qrImagePath, {
+    ensureLoginTab: ensureLoginTabImpl,
+  })
+
+  session.qrImagePath = qrImagePath
+  session.qrImageBase64 = await readFile(qrImagePath, 'base64')
+  session.qrUpdatedAt = now()
+  session.updatedAt = session.qrUpdatedAt
+}
+
+export async function captureQrImageWithRetry(
+  page,
+  loginType,
+  qrImagePath,
+  {
+    captureQqQrImage: captureQqQrImageImpl = captureQqQrImage,
+    captureWxQrImage: captureWxQrImageImpl = captureWxQrImage,
+    ensureLoginTab: ensureLoginTabImpl = ensureLoginTab,
+    isRetryableQrCaptureError: isRetryableQrCaptureErrorImpl = isRetryableQrCaptureError,
+    logBrowserSessionDebug: logBrowserSessionDebugImpl = logBrowserSessionDebug,
+  } = {},
+) {
+  const maxAttempts = loginType === 'qq' ? 4 : 3
+  let lastError = null
+
+  for (let attempt = 1; attempt <= maxAttempts; attempt += 1) {
+    try {
+      logBrowserSessionDebugImpl('captureQrImageWithRetry.start', {
+        loginType,
+        attempt,
+        maxAttempts,
+        qrImagePath,
+      })
+
+      if (loginType === 'qq') {
+        await captureQqQrImageImpl(page, qrImagePath)
+      } else {
+        await captureWxQrImageImpl(page, qrImagePath)
+      }
+
+      logBrowserSessionDebugImpl('captureQrImageWithRetry.success', {
+        loginType,
+        attempt,
+        qrImagePath,
+      })
+
+      return
+    } catch (error) {
+      lastError = error
+      logBrowserSessionDebugImpl('captureQrImageWithRetry.error', {
+        loginType,
+        attempt,
+        message: error instanceof Error ? error.message : String(error),
+      })
+
+      if (!isRetryableQrCaptureErrorImpl(error) || attempt === maxAttempts) {
+        throw error
+      }
+
+      await page.waitForTimeout(500)
+      await ensureLoginTabImpl(page, loginType)
+    }
+  }
+
+  if (lastError) {
+    throw lastError
+  }
+
+  throw new Error('二维码截图失败')
+}
+
+export async function ensureLoginTab(
+  page,
+  loginType,
+  {
+    ensureQqLoginReady: ensureQqLoginReadyImpl = ensureQqLoginReady,
+    ensureWxLoginReady: ensureWxLoginReadyImpl = ensureWxLoginReady,
+  } = {},
+) {
+  const tabSelector = loginType === 'wx' ? '.wx-tab' : '.qc-tab'
+  const loginButton = page.locator('#unlogin')
+  let interacted = false
+
+  if (await loginButton.count()) {
+    try {
+      await loginButton.first().click({ timeout: 2_000 })
+      interacted = true
+    } catch {
+      // ignore when the login layer is already open
+    }
+  }
+
+  const tab = page.locator(tabSelector).first()
+
+  if (await tab.count()) {
+    try {
+      await tab.click({ timeout: 2_000 })
+      interacted = true
+    } catch {
+      // ignore tab click failures
+    }
+  }
+
+  if (interacted) {
+    await page.waitForTimeout(250)
+  }
+
+  if (loginType === 'qq') {
+    await ensureQqLoginReadyImpl(page)
+    return
+  }
+
+  if (loginType === 'wx') {
+    await ensureWxLoginReadyImpl(page)
+  }
+}
+
+export async function extractHostState(page) {
+  return page.evaluate(() => {
+    const isVisible = (element) => {
+      if (!element) {
+        return false
+      }
+
+      const style = window.getComputedStyle(element)
+
+      return style.display !== 'none' && style.visibility !== 'hidden' && style.opacity !== '0'
+    }
+
+    const unlogin = document.querySelector('#unlogin')
+    const logined = document.querySelector('#logined')
+
+    return {
+      unloginVisible: isVisible(unlogin),
+      loginedVisible: isVisible(logined),
+      loginText: String(logined?.textContent || '').trim(),
+    }
+  })
+}
+
+export async function reloadActivityPageForPresentation(page, activityUrl) {
+  try {
+    await page.reload({ waitUntil: 'domcontentloaded' })
+  } catch {
+    await page.goto(activityUrl, { waitUntil: 'domcontentloaded' })
+  }
+
+  await page.waitForTimeout(2_500)
+}
+
+export async function extractQrState(
+  page,
+  loginType,
+  {
+    extractQqQrState: extractQqQrStateImpl = extractQqQrState,
+    extractWxQrState: extractWxQrStateImpl = extractWxQrState,
+  } = {},
+) {
+  if (loginType === 'qq') {
+    return extractQqQrStateImpl(page)
+  }
+
+  return extractWxQrStateImpl(page)
+}