Update index.js

index.js

@@ -10,6 +10,7 @@ const Tokens = [];
 let tokenManager;
 let redisClient;
 let currentIndex = 0;
+let sessionId = null;
 const CONFIG = {
   API: {
     BASE_URL: "https://partyrock.aws",
@@ -95,8 +96,9 @@ class TokenManager {
   }
 
   async updateCacheTokens() {
+    sessionId = Utils.uuidv4();
     CONFIG.DEFAULT_HEADERS["anti-csrftoken-a2z"] = Tokens[currentIndex].anti_csrftoken_a2z;
-    CONFIG.DEFAULT_HEADERS.Cookie = `idToken=${Tokens[currentIndex].idToken}; pr_refresh_token=${Tokens[currentIndex].pr_refresh_token};aws-waf-token=${Tokens[currentIndex].aws_waf_token}`;
+    CONFIG.DEFAULT_HEADERS.Cookie = `idToken=${Tokens[currentIndex].idToken}; pr_refresh_token=${Tokens[currentIndex].pr_refresh_token};aws-waf-token=${Tokens[currentIndex].aws_waf_token};cwr_s=${Tokens[currentIndex].cwr_s};cwr_u=${sessionId}`;
     CONFIG.DEFAULT_HEADERS.referer = Tokens[currentIndex].refreshUrl;
   }
 
@@ -108,7 +110,6 @@ class TokenManager {
         Tokens[currentIndex].aws_waf_token = wafToken;
         await this.updateCacheTokens();
         console.log("更新后的token",JSON.stringify(Tokens[currentIndex],null,2));
-        console.log("更新后cookie",CONFIG.DEFAULT_HEADERS.Cookie);
         this.updateRedisTokens();
         console.log("更新成功",)
         return;
@@ -138,7 +139,7 @@ class TokenManager {
 class Utils {
   static async extractWaf() {
     const browser = await puppeteer.launch({
-      headless: true,
+      headless: false,
       args: [
         '--no-sandbox',
         '--disable-setuid-sandbox',
@@ -150,11 +151,22 @@ class Utils {
     try {
       const page = await browser.newPage();
       await page.setExtraHTTPHeaders({
-        cookie: `pr_refresh_token=${Tokens[currentIndex].pr_refresh_token};idToken=${Tokens[currentIndex].idToken}`
+        cookie: `pr_refresh_token=${Tokens[currentIndex].pr_refresh_token};idToken=${Tokens[currentIndex].idToken};aws-waf-token=${Tokens[currentIndex].aws_waf_token};cwr_s=${Tokens[currentIndex].cwr_s};cwr_u=${Utils.uuidv4()}`
       });
-
-      await page.goto(Tokens[currentIndex].refreshUrl, { waitUntil: 'networkidle0' });
-
+      await page.setUserAgent(
+        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.0 Safari/605.1.15"
+      )
+      await page.goto(Tokens[currentIndex].refreshUrl, { 
+        waitUntil: 'networkidle2',
+        timeout: 30000
+      });
+      await page.evaluate(() => {
+        // 随机滚动
+        window.scrollBy(0, Math.random() * 500)
+      })
+      await page.evaluate(() => {
+        return new Promise(resolve => setTimeout(resolve, 2000))
+      })
       // 直接从页面 cookies 中提取 aws-waf-token
       const awsWafToken = (await page.cookies()).find(
         cookie => cookie.name.toLowerCase() === 'aws-waf-token'
@@ -236,7 +248,8 @@ async function initializeService() {
           anti_csrftoken_a2z: parsedRedis.anti_csrftoken_a2z,
           pr_refresh_token: parsedRedis.pr_refresh_token,
           aws_waf_token: parsedRedis.aws_waf_token,
-          idToken: parsedRedis.idToken
+          idToken: parsedRedis.idToken,
+          cwr_s: parsedRedis.cwr_s
         });
         console.log(`成功添加第 ${index} 组 Token`);
       } catch (error) {
@@ -259,13 +272,16 @@ async function initializeService() {
           anti_csrftoken_a2z,
           pr_refresh_token: cookies["pr_refresh_token"],
           aws_waf_token: cookies["aws-waf-token"],
-          idToken: cookies["idToken"]
+          idToken: cookies["idToken"],
+          cwr_s: cookies["cwr_s"]
         });
       }
     }
     index++;
   }
   console.log('服务初始化完毕');
+  console.log(JSON.stringify(Tokens[0],null,2));
+  
 }
 
 await initializeService();
@@ -515,9 +531,7 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
             ...CONFIG.DEFAULT_HEADERS
           },
           body: JSON.stringify(requestPayload)
-        });
-        console.log("请求的头部",JSON.stringify(CONFIG.DEFAULT_HEADERS,null,2));
-        
+        });     
         reqStatus = response.status;
         switch (reqStatus) {
           case 200: