Add Plus SUB2API session access strategy

2026-05-19 06:14:35 +08:00
parent a3c36e6f15
commit de17884a30
25 changed files with 2380 additions and 49 deletions
@@ -2,8 +2,42 @@
  root.MultiPageBackgroundGoPayManualConfirm = factory();
 })(typeof self !== 'undefined' ? self : globalThis, function createBackgroundGoPayManualConfirmModule() {
  const PLUS_CHECKOUT_SOURCE = 'plus-checkout';
+  const GOPAY_CONFIRM_NODE_ID = 'gopay-subscription-confirm';
+  const SUB2API_SESSION_IMPORT_NODE_ID = 'sub2api-session-import';
  const DEFAULT_CONFIRM_TITLE = 'GoPay 订阅确认';
-  const DEFAULT_CONFIRM_MESSAGE = 'GoPay 订阅页已打开。请先手动完成订阅，完成后确认继续 OAuth 登录。';
+  const OAUTH_CONTINUATION_LABEL = 'OAuth 登录';
+  const SUB2API_SESSION_CONTINUATION_LABEL = '导入当前 ChatGPT 会话到 SUB2API';
+
+  function normalizeString(value = '') {
+    return String(value || '').trim();
+  }
+
+  function getContinuationActionLabel(state = {}, options = {}) {
+    const { getNodeIdsForState = null } = options;
+    if (typeof getNodeIdsForState === 'function') {
+      const nodeIds = getNodeIdsForState(state)
+        .map((nodeId) => normalizeString(nodeId))
+        .filter(Boolean);
+      const currentNodeId = normalizeString(state?.nodeId || GOPAY_CONFIRM_NODE_ID) || GOPAY_CONFIRM_NODE_ID;
+      const currentNodeIndex = nodeIds.indexOf(currentNodeId);
+      const nextNodeId = currentNodeIndex >= 0
+        ? normalizeString(nodeIds[currentNodeIndex + 1])
+        : '';
+
+      if (
+        nextNodeId === SUB2API_SESSION_IMPORT_NODE_ID
+        || (currentNodeIndex < 0 && nodeIds.includes(SUB2API_SESSION_IMPORT_NODE_ID))
+      ) {
+        return SUB2API_SESSION_CONTINUATION_LABEL;
+      }
+    }
+    return OAUTH_CONTINUATION_LABEL;
+  }
+
+  function buildDefaultConfirmMessage(state = {}, options = {}) {
+    const continuationActionLabel = getContinuationActionLabel(state, options);
+    return `GoPay 订阅页已打开。请先手动完成订阅，完成后确认继续${continuationActionLabel}。`;
+  }

  function createGoPayManualConfirmExecutor(deps = {}) {
    const {
@@ -12,6 +46,7 @@
      chrome,
      createAutomationTab = null,
      getTabId,
+      getNodeIdsForState = null,
      isTabAlive,
      registerTab,
      setState,
@@ -40,7 +75,7 @@
        }
      }

-      const checkoutUrl = String(state?.plusCheckoutUrl || '').trim();
+      const checkoutUrl = normalizeString(state?.plusCheckoutUrl);
      if (!checkoutUrl) {
        throw new Error('步骤 7：未检测到 GoPay 订阅页，请先执行步骤 6。');
      }
@@ -63,19 +98,21 @@
    }

    async function executeGoPayManualConfirm(state = {}) {
+      const visibleStep = Number(state?.visibleStep) || 7;
      const tabId = await resolveCheckoutTabId(state);
      if (chrome?.tabs?.update && tabId) {
        await chrome.tabs.update(tabId, { active: true }).catch(() => {});
      }

+      const continuationActionLabel = getContinuationActionLabel(state, { getNodeIdsForState });
      const payload = {
        plusCheckoutTabId: tabId,
        plusManualConfirmationPending: true,
        plusManualConfirmationRequestId: buildRequestId(),
-        plusManualConfirmationStep: 7,
+        plusManualConfirmationStep: visibleStep,
        plusManualConfirmationMethod: 'gopay',
        plusManualConfirmationTitle: DEFAULT_CONFIRM_TITLE,
-        plusManualConfirmationMessage: DEFAULT_CONFIRM_MESSAGE,
+        plusManualConfirmationMessage: buildDefaultConfirmMessage(state, { getNodeIdsForState }),
      };

      await setState(payload);
@@ -83,7 +120,10 @@
        broadcastDataUpdate(payload);
      }

-      await addLog('步骤 7：正在等待手动完成 GoPay 订阅，确认后继续 OAuth 登录。', 'info');
+      await addLog(
+        `步骤 ${visibleStep}：正在等待手动完成 GoPay 订阅，确认后继续${continuationActionLabel}。`,
+        'info'
+      );
    }

    return {
@@ -0,0 +1,185 @@
+(function attachBackgroundSub2ApiSessionImport(root, factory) {
+  root.MultiPageBackgroundSub2ApiSessionImport = factory();
+})(typeof self !== 'undefined' ? self : globalThis, function createBackgroundSub2ApiSessionImportModule() {
+  const PLUS_CHECKOUT_SOURCE = 'plus-checkout';
+  const PLUS_CHECKOUT_INJECT_FILES = ['content/utils.js', 'content/operation-delay.js', 'content/plus-checkout.js'];
+
+  function createSub2ApiSessionImportExecutor(deps = {}) {
+    const {
+      addLog: rawAddLog = async () => {},
+      chrome,
+      completeNodeFromBackground,
+      ensureContentScriptReadyOnTabUntilStopped,
+      getTabId,
+      isTabAlive,
+      normalizeSub2ApiUrl = (value) => value,
+      registerTab,
+      sendTabMessageUntilStopped,
+      sleepWithStop = async () => {},
+      throwIfStopped = () => {},
+      waitForTabCompleteUntilStopped = async () => {},
+      DEFAULT_SUB2API_GROUP_NAME = 'codex',
+    } = deps;
+
+    let sub2ApiApi = null;
+
+    function addStepLog(step, message, level = 'info') {
+      return rawAddLog(message, level, {
+        step,
+        stepKey: 'sub2api-session-import',
+      });
+    }
+
+    function getSub2ApiApi() {
+      if (sub2ApiApi) {
+        return sub2ApiApi;
+      }
+      const factory = deps.createSub2ApiApi
+        || self.MultiPageBackgroundSub2ApiApi?.createSub2ApiApi;
+      if (typeof factory !== 'function') {
+        throw new Error('SUB2API 接口模块未加载，无法导入当前 ChatGPT 会话。');
+      }
+      sub2ApiApi = factory({
+        addLog: rawAddLog,
+        normalizeSub2ApiUrl,
+        DEFAULT_SUB2API_GROUP_NAME,
+      });
+      return sub2ApiApi;
+    }
+
+    function normalizeString(value = '') {
+      return String(value || '').trim();
+    }
+
+    function resolveVisibleStep(state = {}) {
+      const visibleStep = Math.floor(Number(state?.visibleStep) || 0);
+      return visibleStep > 0 ? visibleStep : 10;
+    }
+
+    function isSupportedChatGptSessionUrl(url = '') {
+      try {
+        const parsed = new URL(String(url || ''));
+        if (!/^https?:$/i.test(parsed.protocol)) {
+          return false;
+        }
+        const hostname = String(parsed.hostname || '').trim().toLowerCase();
+        return /(^|\.)chatgpt\.com$/.test(hostname)
+          || hostname === 'chat.openai.com'
+          || /(^|\.)openai\.com$/.test(hostname);
+      } catch {
+        return false;
+      }
+    }
+
+    async function resolveSessionTabId(state = {}) {
+      const registeredTabId = typeof getTabId === 'function'
+        ? await getTabId(PLUS_CHECKOUT_SOURCE)
+        : null;
+      if (registeredTabId && typeof isTabAlive === 'function' && await isTabAlive(PLUS_CHECKOUT_SOURCE)) {
+        return Number(registeredTabId) || 0;
+      }
+
+      const storedTabId = Number(state?.plusCheckoutTabId) || 0;
+      if (storedTabId && chrome?.tabs?.get) {
+        const tab = await chrome.tabs.get(storedTabId).catch(() => null);
+        if (tab?.id) {
+          if (typeof registerTab === 'function') {
+            await registerTab(PLUS_CHECKOUT_SOURCE, tab.id);
+          }
+          return tab.id;
+        }
+      }
+
+      throw new Error('未找到可读取 ChatGPT 会话的 Plus 标签页，请先完成当前 Plus 支付链路。');
+    }
+
+    async function getResolvedSessionTab(tabId, visibleStep) {
+      const tab = await chrome?.tabs?.get?.(tabId).catch(() => null);
+      if (!tab?.id) {
+        throw new Error(`步骤 ${visibleStep}：Plus 会话标签页不存在或已关闭，无法继续导入 SUB2API。`);
+      }
+      if (!isSupportedChatGptSessionUrl(tab.url)) {
+        throw new Error(`步骤 ${visibleStep}：当前标签页不在 ChatGPT / OpenAI 页面，无法读取当前登录会话。`);
+      }
+      return tab;
+    }
+
+    async function readCurrentChatGptSession(tabId, visibleStep) {
+      await waitForTabCompleteUntilStopped(tabId);
+      await sleepWithStop(1000);
+      await ensureContentScriptReadyOnTabUntilStopped(PLUS_CHECKOUT_SOURCE, tabId, {
+        inject: PLUS_CHECKOUT_INJECT_FILES,
+        injectSource: PLUS_CHECKOUT_SOURCE,
+        logMessage: `步骤 ${visibleStep}：正在等待 ChatGPT 会话页完成加载，再继续读取当前登录会话...`,
+      });
+
+      const sessionResult = await sendTabMessageUntilStopped(tabId, PLUS_CHECKOUT_SOURCE, {
+        type: 'PLUS_CHECKOUT_GET_STATE',
+        source: 'background',
+        payload: {
+          includeSession: true,
+          includeAccessToken: true,
+        },
+      });
+      if (sessionResult?.error) {
+        throw new Error(sessionResult.error);
+      }
+
+      const session = sessionResult?.session && typeof sessionResult.session === 'object' && !Array.isArray(sessionResult.session)
+        ? sessionResult.session
+        : null;
+      const accessToken = normalizeString(
+        sessionResult?.accessToken
+        || session?.accessToken
+      );
+      if (!session && !accessToken) {
+        throw new Error(`步骤 ${visibleStep}：未读取到有效的 ChatGPT 会话或 accessToken，请确认当前标签页仍处于已登录状态。`);
+      }
+
+      return {
+        session,
+        accessToken,
+      };
+    }
+
+    async function executeSub2ApiSessionImport(state = {}) {
+      throwIfStopped();
+      const visibleStep = resolveVisibleStep(state);
+      const api = getSub2ApiApi();
+
+      await addStepLog(visibleStep, '正在定位当前 Plus 会话页并准备导入 SUB2API...', 'info');
+      const tabId = await resolveSessionTabId(state);
+      const tab = await getResolvedSessionTab(tabId, visibleStep);
+      if (chrome?.tabs?.update) {
+        await chrome.tabs.update(tab.id, { active: true }).catch(() => {});
+      }
+
+      await addStepLog(visibleStep, '正在读取当前 ChatGPT 登录会话...', 'info');
+      const sessionState = await readCurrentChatGptSession(tab.id, visibleStep);
+      throwIfStopped();
+
+      const result = await api.importCurrentChatGptSession({
+        ...state,
+        session: sessionState.session,
+        accessToken: sessionState.accessToken,
+      }, {
+        visibleStep,
+        logLabel: `步骤 ${visibleStep}`,
+        logOptions: { step: visibleStep, stepKey: 'sub2api-session-import' },
+        timeoutMs: 120000,
+        importTimeoutMs: 120000,
+      });
+
+      await completeNodeFromBackground(state?.nodeId || 'sub2api-session-import', result);
+    }
+
+    return {
+      executeSub2ApiSessionImport,
+      isSupportedChatGptSessionUrl,
+    };
+  }
+
+  return {
+    createSub2ApiSessionImportExecutor,
+  };
+});