qwenproxy-cli 1.4.3 → 1.4.8

package/dist/clean-cache.js

@@ -314,6 +314,7 @@ function runMigrations(db2) {
     "sec_ch_ua_mobile TEXT",
     "sec_ch_ua_platform TEXT",
     "version TEXT",
+    "refresh_token TEXT",
     "captured_at INTEGER DEFAULT 0"
   ];
   for (const col of authSessionCols) {
@@ -392,11 +393,11 @@ function saveAuthSession(accountId, session) {
     INSERT OR REPLACE INTO qwen_auth_sessions (
       account_id, cookie, user_agent, bx_v, bx_ua, bx_umidtoken,
       sec_ch_ua, sec_ch_ua_mobile, sec_ch_ua_platform, version,
-      user_id, token_expires_at, captured_at, updated_at
+      user_id, token_expires_at, refresh_token, captured_at, updated_at
     ) VALUES (
       ?, ?, ?, ?, ?, ?,
       ?, ?, ?, ?,
-      ?, ?, ?, datetime('now')
+      ?, ?, ?, ?, datetime('now')
     )
   `);
   stmt.run(
@@ -412,6 +413,7 @@ function saveAuthSession(accountId, session) {
     session.version || null,
     session.userId || null,
     session.tokenExpiresAt || null,
+    session.refreshToken || null,
     session.capturedAt ?? Date.now()
   );
 }
@@ -420,7 +422,7 @@ function getValidAuthSession(accountId, maxAgeMs = 30 * 24 * 60 * 60 * 1e3) {
   const row = database.prepare(
     `SELECT account_id, cookie, user_agent, bx_v, bx_ua, bx_umidtoken,
               sec_ch_ua, sec_ch_ua_mobile, sec_ch_ua_platform, version,
-              user_id, token_expires_at, captured_at
+              user_id, token_expires_at, refresh_token, captured_at
        FROM qwen_auth_sessions WHERE account_id = ?`
   ).get(accountId);
   if (!row) return null;
@@ -428,7 +430,7 @@ function getValidAuthSession(accountId, maxAgeMs = 30 * 24 * 60 * 60 * 1e3) {
   if (capturedAt <= 0 || Date.now() - capturedAt > maxAgeMs) {
     return null;
   }
-  if (row.token_expires_at) {
+  if (!row.refresh_token && row.token_expires_at) {
     const tokenExpMs = Number(row.token_expires_at) * 1e3;
     if (tokenExpMs <= Date.now() + 5 * 60 * 1e3) {
       return null;
@@ -450,6 +452,7 @@ function getValidAuthSession(accountId, maxAgeMs = 30 * 24 * 60 * 60 * 1e3) {
     version: row.version || void 0,
     userId: row.user_id || void 0,
     tokenExpiresAt: row.token_expires_at ? Number(row.token_expires_at) : void 0,
+    refreshToken: row.refresh_token || void 0,
     capturedAt
   };
 }
@@ -3287,8 +3290,29 @@ function getStealthScript(profile) {
         const _toBlob = HTMLCanvasElement.prototype.toBlob;
         const _getImageData = CanvasRenderingContext2D.prototype.getImageData;
 
+        function isCaptchaCanvas(canvas) {
+          try {
+            if (!canvas) return false;
+            const id = (canvas.id || "").toLowerCase();
+            const cls = (canvas.className || "").toLowerCase();
+            return (
+              id.includes("baxia") ||
+              id.includes("nc_") ||
+              id.includes("captcha") ||
+              id.includes("punish") ||
+              cls.includes("baxia") ||
+              cls.includes("nc_") ||
+              cls.includes("captcha") ||
+              Boolean(canvas.closest?.("#baxia-dialog, #baxia-punish, [class*='baxia'], [class*='nc_'], .auth-layout"))
+            );
+          } catch(e) {
+            return false;
+          }
+        }
+
         function addNoise(canvas) {
           try {
+            if (isCaptchaCanvas(canvas)) return;
             const ctx = canvas.getContext('2d');
             if (!ctx) return;
             const style = ctx.fillStyle;
@@ -3312,6 +3336,9 @@ function getStealthScript(profile) {
 
         CanvasRenderingContext2D.prototype.getImageData = function(x, y, w, h) {
           const imageData = _getImageData.apply(this, arguments);
+          if (isCaptchaCanvas(this.canvas)) {
+            return imageData;
+          }
           const data = imageData.data;
           const maxPixels = Math.min(data.length / 4, 2500);
           for (let i = 0; i < maxPixels * 4; i += 4) {
@@ -3788,6 +3815,7 @@ async function loginToQwen(accountId, email, password) {
             secChUaPlatform: cache.headers["sec-ch-ua-platform"] || void 0,
             version: cache.headers["version"] || void 0,
             tokenExpiresAt: exp || void 0,
+            refreshToken: apiResult.refreshToken,
             capturedAt: Date.now()
           });
         }
@@ -3821,6 +3849,13 @@ async function loginToQwen(accountId, email, password) {
           markAccountHeadersReady(accountId);
           const { saveAuthSession: saveAuthSession2 } = await Promise.resolve().then(() => (init_database(), database_exports));
           const exp = parseJwtExpiry(tokenCookie.value);
+          let rTok = uiResult.refreshToken;
+          if (!rTok) {
+            try {
+              rTok = await page.evaluate(() => localStorage.getItem("refresh_token") || "") || void 0;
+            } catch {
+            }
+          }
           saveAuthSession2(accountId, {
             cookie: cookieStr,
             userAgent: cache.headers["user-agent"] || "",
@@ -3832,6 +3867,7 @@ async function loginToQwen(accountId, email, password) {
             secChUaPlatform: cache.headers["sec-ch-ua-platform"] || void 0,
             version: cache.headers["version"] || void 0,
             tokenExpiresAt: exp || void 0,
+            refreshToken: rTok,
             capturedAt: Date.now()
           });
         }
@@ -3884,37 +3920,52 @@ async function loginToQwen(accountId, email, password) {
 async function loginViaApi(page, email, password) {
   try {
     const hashedPassword = crypto3.createHash("sha256").update(password).digest("hex");
-    const signinUrl = qwenUrl("/api/v2/auths/signin");
-    let signinSuccess = false;
-    let data = null;
-    if (page.request && typeof page.request.post === "function") {
-      try {
-        const response = await page.request.post(signinUrl, {
-          data: {
-            email,
-            password: hashedPassword,
-            login_type: "email"
-          },
-          headers: {
-            "content-type": "application/json",
-            accept: "application/json, text/plain, */*",
-            referer: qwenUrl("/auth"),
-            origin: qwenOrigin(),
-            source: "web"
-          },
-          timeout: 1e4
-        });
-        data = await response.json().catch(() => null);
-        signinSuccess = Boolean(data && (data.success === true || data.token));
-      } catch {
-      }
+    if (!page.url().includes("/auth")) {
+      await page.goto(qwenUrl("/auth"), {
+        waitUntil: "domcontentloaded",
+        timeout: config.timeouts.navigation
+      }).catch(() => {
+      });
+      await sleep2(1e3);
     }
-    if (!signinSuccess && !data) {
-      const requestId = crypto3.randomUUID();
-      const evalRes = await page.evaluate(
-        async ({ email: email2, password: password2, signinUrl: signinUrl2, requestId: requestId2 }) => {
+    const signinUrl = "https://auth.qwen.ai/api/v2/auths/signin";
+    const requestId = crypto3.randomUUID();
+    const evalRes = await page.evaluate(
+      async ({ email: email2, password: password2, signinUrl: signinUrl2, requestId: requestId2 }) => {
+        try {
+          const response = await fetch(signinUrl2, {
+            method: "POST",
+            signal: AbortSignal.timeout(15e3),
+            credentials: "include",
+            headers: {
+              accept: "application/json, text/plain, */*",
+              "content-type": "application/json",
+              source: "web",
+              version: "0.3.11",
+              "x-request-origin": "https://chat.qwen.ai",
+              timezone: (/* @__PURE__ */ new Date()).toString().split(" (")[0],
+              "x-request-id": requestId2
+            },
+            body: JSON.stringify({ email: email2, password: password2 })
+          });
+          const json = await response.json().catch(() => null);
+          return { ok: response.ok, status: response.status, data: json };
+        } catch (e) {
+          return { ok: false, error: e.message };
+        }
+      },
+      { email, password: hashedPassword, signinUrl, requestId }
+    ).catch(() => null);
+    let data = evalRes?.data;
+    let signinSuccess = Boolean(
+      data && (data.success === true || data.token || data.data?.access_token || data.data?.token)
+    );
+    if (!signinSuccess && !data?.data?.code) {
+      const fallbackUrl = qwenUrl("/api/v2/auths/signin");
+      const fallbackRes = await page.evaluate(
+        async ({ email: email2, password: password2, fallbackUrl: fallbackUrl2, requestId: requestId2 }) => {
           try {
-            const response = await fetch(signinUrl2, {
+            const response = await fetch(fallbackUrl2, {
               method: "POST",
               signal: AbortSignal.timeout(1e4),
               credentials: "include",
@@ -3922,10 +3973,11 @@ async function loginViaApi(page, email, password) {
                 accept: "application/json, text/plain, */*",
                 "content-type": "application/json",
                 source: "web",
+                version: "0.3.11",
                 timezone: (/* @__PURE__ */ new Date()).toString().split(" (")[0],
                 "x-request-id": requestId2
               },
-              body: JSON.stringify({ email: email2, password: password2, login_type: "email" })
+              body: JSON.stringify({ email: email2, password: password2 })
             });
             const json = await response.json().catch(() => null);
             return { ok: response.ok, status: response.status, data: json };
@@ -3933,11 +3985,13 @@ async function loginViaApi(page, email, password) {
             return { ok: false, error: e.message };
           }
         },
-        { email, password: hashedPassword, signinUrl, requestId }
+        { email, password: hashedPassword, fallbackUrl, requestId }
       ).catch(() => null);
-      if (evalRes?.data) {
-        data = evalRes.data;
-        signinSuccess = Boolean(data && (data.success === true || data.token));
+      if (fallbackRes?.data) {
+        data = fallbackRes.data;
+        signinSuccess = Boolean(
+          data && (data.success === true || data.token || data.data?.access_token || data.data?.token)
+        );
       }
     }
     if (data?.success === false) {
@@ -3951,7 +4005,8 @@ async function loginViaApi(page, email, password) {
       };
     }
     if (signinSuccess) {
-      const token = data?.data?.token || data?.token;
+      const token = data?.data?.access_token || data?.data?.token || data?.access_token || data?.token;
+      const refreshToken = data?.data?.refresh_token || data?.refresh_token;
       if (token) {
         try {
           await page.context().addCookies([
@@ -3975,18 +4030,23 @@ async function loginViaApi(page, email, password) {
       }).catch(() => {
       });
       if (token) {
-        await page.evaluate((tok) => {
-          try {
-            localStorage.removeItem("qwen_token_logged_out_marker");
-            localStorage.setItem("token", tok);
-            document.cookie = `token=${encodeURIComponent(tok)}; path=/; domain=.qwen.ai; max-age=31536000`;
-          } catch {
-          }
-        }, token).catch(() => {
+        await page.evaluate(
+          ({ tok, rTok }) => {
+            try {
+              localStorage.removeItem("qwen_token_logged_out_marker");
+              localStorage.setItem("token", tok);
+              localStorage.setItem("access_token", tok);
+              if (rTok) localStorage.setItem("refresh_token", rTok);
+              document.cookie = `token=${encodeURIComponent(tok)}; path=/; domain=.qwen.ai; max-age=31536000`;
+            } catch {
+            }
+          },
+          { tok: token, rTok: refreshToken }
+        ).catch(() => {
         });
         await page.reload({ waitUntil: "domcontentloaded" }).catch(() => {
         });
-        return { success: true };
+        return { success: true, token, refreshToken };
       }
     }
     return {
@@ -4010,11 +4070,23 @@ async function loginViaApi(page, email, password) {
 }
 async function loginViaUi(page, email, password) {
   try {
-    await page.goto(qwenUrl("/auth"), {
-      waitUntil: "domcontentloaded",
-      timeout: config.timeouts.navigation
-    });
-    await sleep2(1500);
+    if (!page.url().includes("/auth") && await isPageLoggedIn(page, 2e3).catch(() => false)) {
+      return { success: true };
+    }
+    try {
+      await page.goto(qwenUrl("/auth"), {
+        waitUntil: "domcontentloaded",
+        timeout: Math.min(config.timeouts.navigation, 15e3)
+      });
+      await sleep2(1e3);
+    } catch (gotoErr) {
+      if (!page.url().includes("/auth") && await isPageLoggedIn(page, 3e3).catch(() => false)) {
+        return { success: true };
+      }
+    }
+    if (!page.url().includes("/auth") && await isPageLoggedIn(page, 3e3).catch(() => false)) {
+      return { success: true };
+    }
     const emailSelector = [
       'input[type="email"]',
       'input[name="email"]',
@@ -4024,7 +4096,7 @@ async function loginViaUi(page, email, password) {
     ].join(", ");
     try {
       await page.waitForSelector(emailSelector, {
-        timeout: Math.min(config.timeouts.page, 8e3)
+        timeout: Math.min(config.timeouts.page, 5e3)
       });
     } catch {
       if (!page.url().includes("/auth") && await isPageLoggedIn(page, 3e3)) {
@@ -4037,30 +4109,55 @@ async function loginViaUi(page, email, password) {
       return { success: false, reason: "Campo de e-mail n\xE3o encontrado (poss\xEDvel captcha)" };
     }
     await clearVisibleChallenge(page);
-    const findPasswordModeButton = () => {
-      return page.locator('button, [role="button"], a, div').filter({
-        hasText: /(?:log\s*in\s*with\s*(?:a\s*)?password|fazer\s*login\s*com\s*senha|entrar\s*com\s*(?:uma\s*)?senha|iniciar\s*sesi[oó]n\s*con\s*contrase[nñ]a|密码登录)/i
-      }).first();
+    const isPasswordInputPresent = async () => {
+      return await page.locator('input[type="password"], input[name="password"]').count().catch(() => 0) > 0;
     };
-    try {
-      const pwdModeBtn = findPasswordModeButton();
-      if (await pwdModeBtn.isVisible({ timeout: 2e3 }).catch(() => false)) {
-        await pwdModeBtn.click({ timeout: 2e3 }).catch(() => {
+    const revealPasswordMode = async () => {
+      if (page.url().includes("/forget")) {
+        await page.goto(qwenUrl("/auth"), { waitUntil: "domcontentloaded", timeout: 1e4 }).catch(() => {
         });
-        await sleep2(500);
+        await sleep2(1e3);
       }
-    } catch {
-    }
-    await page.fill(emailSelector, email);
-    await sleep2(300);
-    try {
-      const pwdModeBtn = findPasswordModeButton();
-      if (await pwdModeBtn.isVisible({ timeout: 1500 }).catch(() => false)) {
-        await pwdModeBtn.click({ timeout: 2e3 }).catch(() => {
+      if (await isPasswordInputPresent()) return;
+      const btn = page.locator("button").filter({
+        hasText: /(?:log\s*in\s*with\s*(?:a\s*)?password|fazer\s*login\s*com\s*senha|entrar\s*com\s*(?:uma\s*)?senha|iniciar\s*sesi[oó]n\s*con\s*contrase[nñ]a|密码登录)/i
+      }).filter({
+        hasNotText: /(?:esqueci|forgot|olvid|reset|recover)/i
+      }).first();
+      if (await btn.isVisible({ timeout: 1500 }).catch(() => false)) {
+        await btn.click({ timeout: 2e3 }).catch(() => {
         });
         await sleep2(500);
       }
+      if (await isPasswordInputPresent()) return;
+      await page.evaluate(() => {
+        try {
+          const buttons = Array.from(document.querySelectorAll("button, [role='button'], a"));
+          for (const b of buttons) {
+            const txt = (b.innerText || "").trim().toLowerCase();
+            if (/(?:log\s*in\s*with\s*(?:a\s*)?password|fazer\s*login\s*com\s*senha|entrar\s*com\s*(?:uma\s*)?senha|iniciar\s*sesi[oó]n\s*con\s*contrase[nñ]a|密码登录)/i.test(txt) && !txt.includes("esqueci") && !txt.includes("forgot") && !txt.includes("olvid") && !txt.includes("reset")) {
+              b.click();
+              return true;
+            }
+          }
+        } catch {
+        }
+        return false;
+      }).catch(() => false);
+      await sleep2(500);
+    };
+    await revealPasswordMode();
+    try {
+      await page.fill(emailSelector, email, { timeout: 8e3 });
+      await sleep2(300);
     } catch {
+      if (!page.url().includes("/auth") && await isPageLoggedIn(page, 3e3)) {
+        return { success: true };
+      }
+      throw new Error(`Campo de e-mail n\xE3o encontrado na tela de autentica\xE7\xE3o`);
+    }
+    if (!await isPasswordInputPresent()) {
+      await revealPasswordMode();
     }
     const passwordSelector = 'input[type="password"], input[name="password"]';
     try {
@@ -4079,17 +4176,50 @@ async function loginViaUi(page, email, password) {
         reason: "Campo de senha n\xE3o encontrado na tela de autentica\xE7\xE3o"
       };
     }
-    await page.fill(passwordSelector, password);
-    await sleep2(500);
+    try {
+      await page.fill(passwordSelector, password, { timeout: 8e3 });
+      await sleep2(500);
+    } catch {
+      if (!page.url().includes("/auth") && await isPageLoggedIn(page, 3e3)) {
+        return { success: true };
+      }
+      throw new Error(`Campo de senha n\xE3o encontrado na tela de autentica\xE7\xE3o`);
+    }
+    await page.evaluate(() => {
+      try {
+        const emailEl = document.querySelector('input[name="email"], input[type="email"], input[type="text"]');
+        const passEl = document.querySelector('input[type="password"], input[name="password"]');
+        if (emailEl) {
+          emailEl.dispatchEvent(new Event("input", { bubbles: true }));
+          emailEl.dispatchEvent(new Event("change", { bubbles: true }));
+          emailEl.dispatchEvent(new Event("blur", { bubbles: true }));
+        }
+        if (passEl) {
+          passEl.dispatchEvent(new Event("input", { bubbles: true }));
+          passEl.dispatchEvent(new Event("change", { bubbles: true }));
+          passEl.dispatchEvent(new Event("blur", { bubbles: true }));
+        }
+      } catch {
+      }
+    }).catch(() => {
+    });
     const submitSelector = 'button.qwenchat-auth-pc-submit-button, button[type="submit"], button:has-text("Log in"), button:has-text("Sign in"), button:has-text("Fazer login"), button:has-text("Entrar"), button:has-text("Iniciar sesi\xF3n"), button:has-text("\u767B\u5F55")';
     const submitButton = page.locator(submitSelector).first();
     try {
       await page.waitForSelector('button[type="submit"]:not([disabled])', {
-        timeout: 5e3
+        timeout: 4e3
       });
       await submitButton.click();
     } catch {
-      await page.keyboard.press("Enter");
+      const isDisabled = await submitButton.isDisabled().catch(() => true);
+      if (!isDisabled) {
+        await submitButton.click({ force: true }).catch(() => {
+        });
+      } else {
+        if (!page.url().includes("/auth") && await isPageLoggedIn(page, 2e3)) {
+          return { success: true };
+        }
+      }
     }
     const postSubmitDeadline = Date.now() + 15e3;
     while (Date.now() < postSubmitDeadline) {