zevairouter 1.0.35 → 1.0.36

package/src/lib/oauth/services/codebuddyBulkImportManager.js

@@ -96,17 +96,17 @@ async function defaultSaveCodeBuddyConnection({ tokens, email }) {
   return { connection };
 }
 
-async function defaultRequestDeviceCode(providerId) {
+export async function defaultRequestDeviceCode(providerId) {
   const { requestDeviceCode } = await import("../providers.js");
   return requestDeviceCode(providerId);
 }
 
-async function defaultPollForToken(providerId, deviceCode) {
+export async function defaultPollForToken(providerId, deviceCode) {
   const { pollForToken } = await import("../providers.js");
   return pollForToken(providerId, deviceCode);
 }
 
-async function fetchCodeBuddyLoginAccount(accessToken, state, domain = "www.codebuddy.ai") {
+export async function fetchCodeBuddyLoginAccount(accessToken, state, domain = "www.codebuddy.ai") {
   if (!accessToken || !state) return null;
   try {
     const response = await fetch(`https://${domain}/v2/plugin/login/account?state=${encodeURIComponent(state)}`, {
@@ -530,7 +530,7 @@ export async function createCodeBuddyApiKey(page, email, onStep, {
   throw error;
 }
 
-function createCodeBuddyPollPromise({
+export function createCodeBuddyPollPromise({
   deviceCode,
   pollToken,
   onStep,
@@ -538,6 +538,8 @@ function createCodeBuddyPollPromise({
   timeoutMs = CODEBUDDY_POLL_TIMEOUT_MS,
   pollIntervalMs = CODEBUDDY_POLL_INTERVAL_MS,
   maxTransientErrors = CODEBUDDY_MAX_TRANSIENT_POLL_ERRORS,
+  providerId = CODEBUDDY_PROVIDER_ID,
+  domain = "www.codebuddy.ai",
 }) {
   return (async () => {
     const startedAt = Date.now();
@@ -550,10 +552,10 @@ function createCodeBuddyPollPromise({
         lastStepAt = Date.now();
       }
 
-      const result = await pollToken(CODEBUDDY_PROVIDER_ID, deviceCode);
+      const result = await pollToken(providerId, deviceCode);
       if (result.success) {
         const accessToken = result.tokens?.accessToken || result.tokens?.access_token;
-        const loginAccount = await fetchLoginAccount(accessToken, deviceCode);
+        const loginAccount = await fetchLoginAccount(accessToken, deviceCode, domain);
         return {
           tokens: {
             ...result.tokens,
@@ -564,7 +566,7 @@ function createCodeBuddyPollPromise({
               ...(loginAccount?.enterpriseId ? { enterpriseId: loginAccount.enterpriseId } : {}),
               ...(loginAccount?.enterpriseName ? { enterpriseName: loginAccount.enterpriseName } : {}),
               ...(loginAccount?.rawAuth ? { rawAuth: loginAccount.rawAuth } : {}),
-              domain: "www.codebuddy.ai",
+              domain,
               oauthState: deviceCode,
             },
           },