Update index.js

index.js

@@ -5,6 +5,10 @@ import dotenv from 'dotenv';
 import cors from 'cors';
 import puppeteer from 'puppeteer';
 import { v4 as uuidv4 } from 'uuid';
+import fs from 'fs'; 
+//path
+
+import path from 'path';
 
 dotenv.config();
 // 配置常量
@@ -16,13 +20,13 @@ const CONFIG = {
     },
     API: {
         BASE_URL: "https://grok.com",
-        API_KEY: process.env.API_KEY || "sk-123456",
+        API_KEY: process.env.API_KEY_GUIJI || "linux.do",
         SSO_TOKEN: null,//登录时才有的认证cookie,这里暂时用不到，之后可能需要
         SIGNATURE_COOKIE: null,
         PICGO_KEY: process.env.PICGO_KEY || null //想要生图的话需要填入这个PICGO图床的key
     },
     SERVER: {
-        PORT: process.env.PORT || 3000,
+        PORT:  9300,
         BODY_LIMIT: '5mb'
     },
     RETRY: {
@@ -30,7 +34,7 @@ const CONFIG = {
         DELAY_BASE: 1000 // 基础延迟时间（毫秒）
     },
     ISSHOW_SEARCH_RESULTS: process.env.ISSHOW_SEARCH_RESULTS === 'true',//是否显示搜索结果,默认关闭
-    CHROME_PATH: process.env.CHROME_PATH || "/usr/bin/chromium"//chrome路径
+    CHROME_PATH: process.env.CHROME_PATH || '/usr/bin/google-chrome-stable'
 };
 
 // 请求头配置
@@ -51,8 +55,70 @@ const DEFAULT_HEADERS = {
     'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36',
     'baggage': 'sentry-public_key=b311e0f2690c81f25e2c4cf6d4f7ce1c'
 };
+// Add this class for managing authentication state
+class AuthManager {
+    constructor() {
+        this.authPromise = null;
+        this.lastAuthTime = null;
+        this.AUTH_LIFETIME = 30 * 60 * 1000; // 30 minutes
+        this.isRefreshing = false;
+        this.waitingQueue = [];
+    }
+
+    async getAuth() {
+        // Check if we need to refresh
+        if (this.shouldRefreshAuth()) {
+            return this.refreshAuth();
+        }
+        return CONFIG.API.SIGNATURE_COOKIE;
+    }
+
+    shouldRefreshAuth() {
+        return !CONFIG.API.SIGNATURE_COOKIE || 
+               !this.lastAuthTime || 
+               (Date.now() - this.lastAuthTime > this.AUTH_LIFETIME);
+    }
+
+    async refreshAuth() {
+        // If already refreshing, wait for it
+        if (this.isRefreshing) {
+            return new Promise((resolve) => {
+                this.waitingQueue.push(resolve);
+            });
+        }
+
+        try {
+            this.isRefreshing = true;
+            const headers = await Utils.extractGrokHeaders();
+            
+            if (!headers) {
+                throw new Error('Failed to extract headers');
+            }
+
+            CONFIG.API.SIGNATURE_COOKIE = {
+                cookie: `x-anonuserid=${headers["x-anonuserid"]}; x-challenge=${headers["x-challenge"]}; x-signature=${headers["x-signature"]}`
+            };
+            this.lastAuthTime = Date.now();
+
+            // Resolve all waiting requests
+            this.waitingQueue.forEach(resolve => resolve(CONFIG.API.SIGNATURE_COOKIE));
+            this.waitingQueue = [];
+
+            return CONFIG.API.SIGNATURE_COOKIE;
+        } catch (error) {
+            console.error('Auth refresh failed:', error);
+            this.waitingQueue.forEach(resolve => resolve(null));
+            this.waitingQueue = [];
+            throw error;
+        } finally {
+            this.isRefreshing = false;
+        }
+    }
+}
 
 class Utils {
+    static authManager = new AuthManager();
+
     static async extractGrokHeaders() {
         console.log("开始提取头信息");
         try {
@@ -94,6 +160,9 @@ class Utils {
         }
     }
     static async get_signature() {
+        return Utils.authManager.getAuth();
+    }
+    static async get_signature_bak() {
         if (CONFIG.API.SIGNATURE_COOKIE) {
             return CONFIG.API.SIGNATURE_COOKIE;
         }
@@ -104,7 +173,7 @@ class Utils {
             if (headers) {
                 console.log("获取认证信息成功");
                 CONFIG.API.SIGNATURE_COOKIE = { cookie: `x-anonuserid=${headers["x-anonuserid"]}; x-challenge=${headers["x-challenge"]}; x-signature=${headers["x-signature"]}` };
-                return CONFIG.API.SIGNATURE_COOKIE;
+                return;
             }
             retryCount++;
             if (retryCount >= CONFIG.RETRY.MAX_ATTEMPTS) {
@@ -411,7 +480,7 @@ app.use(cors({
     allowedHeaders: ['Content-Type', 'Authorization']
 }));
 // API路由
-app.get('/hf/v1/models', (req, res) => {
+app.get('/v1/models', (req, res) => {
     res.json({
         object: "list",
         data: Object.keys(CONFIG.MODELS).map((model, index) => ({
@@ -423,8 +492,9 @@ app.get('/hf/v1/models', (req, res) => {
     });
 });
 
-app.post('/hf/v1/chat/completions', async (req, res) => {
+app.post('/v1/chat/completions', async (req, res) => {
     const authToken = req.headers.authorization?.replace('Bearer ', '');
+    console.log(authToken);
     if (authToken !== CONFIG.API.API_KEY) {
         return res.status(401).json({ error: 'Unauthorized' });
     }
@@ -508,6 +578,7 @@ async function handleStreamResponse(response, model, res) {
                 const trimmedLine = line.trim();
                 if (trimmedLine.startsWith('data: ')) {
                     const data = trimmedLine.substring(6);
+                    
                     if(data === "[DONE]"){
                         console.log("流结束");
                         res.write('data: [DONE]\n\n');
@@ -517,10 +588,16 @@ async function handleStreamResponse(response, model, res) {
                         if(!data.trim())continue;
                         const linejosn = JSON.parse(data);
                         if (linejosn?.error?.name === "RateLimitError") {
-                            var responseData = MessageProcessor.createChatResponse(`${linejosn.error.name},请重新对话`, model, true);
-                            fs.unlinkSync(path.resolve(process.cwd(), 'signature.json'));
+                            var responseData = MessageProcessor.createChatResponse(
+                                `${linejosn.error.name},请重新对话`, 
+                                model, 
+                                true
+                            );
+                            
+                            // Remove file deletion - it's not needed
                             CONFIG.API.SIGNATURE_COOKIE = null;
-                            console.log("认证信息已删除");
+                            console.log("认证信息已重置");
+                            
                             res.write(`data: ${JSON.stringify(responseData)}\n\n`);
                             res.write('data: [DONE]\n\n');
                             return res.end();
@@ -563,12 +640,6 @@ async function handleStreamResponse(response, model, res) {
                 }
             }
         });
-        stream.on('error', (error) => {
-            console.error('流处理错误:', error);
-            res.write('data: [DONE]\n\n');
-            res.end();
-        });
-
     } catch (error) {
         console.error('处理响应错误:', error);
         res.write('data: [DONE]\n\n');
@@ -706,4 +777,4 @@ app.use((req, res) => {
 // 启动服务器
 app.listen(CONFIG.SERVER.PORT, () => {
     console.log(`服务器已启动，监听端口: ${CONFIG.SERVER.PORT}`);
-});
+});