Update server/index.js

server/index.js

@@ -1,4 +1,4 @@
-// server/index.js
+// server/index.js (نسخه بهینه شده: انتخاب تصادفی کلیدها + پایداری بالا)
 const express = require('express');
 const path = require('node:path');
 const { WebSocketServer, WebSocket } = require('ws');
@@ -6,14 +6,16 @@ const http = require('node:http');
 require('dotenv').config();
 
 const app = express();
-// افزایش محدودیت‌های سرور برای هندل کردن تعداد بالای درخواست
+
+// افزایش محدودیت‌ها برای جلوگیری از خطای هدر در درخواست‌های سنگین
 const server = http.createServer({
     maxHeaderSize: 16384 // 16KB
 }, app);
 
 const wss = new WebSocketServer({ 
     server,
-    clientTracking: true // برای مدیریت هارت‌بیت ضروری است
+    clientTracking: true, // برای مدیریت هارت‌بیت و بستن اتصالات مرده
+    perMessageDeflate: false // غیرفعال کردن فشرده‌سازی برای کاهش بار CPU و حافظه
 });
 
 // --- بخش مدیریت تنظیمات شخصیت‌ها ---
@@ -31,33 +33,24 @@ Object.keys(instructionSecretNames).forEach(key => {
   const instruction = process.env[secretName];
   if (instruction) {
     personalityInstructions[key] = instruction;
-    console.log(`✅ دستورالعمل '${key}' با موفقیت خوانده شد.`);
   } else {
     personalityInstructions[key] = `دستورالعمل '${key}' یافت نشد.`;
-    console.warn(`⚠️ هشدار: Secret با نام '${secretName}' تنظیم نشده است.`);
   }
 });
 
-// --- بخش مدیریت کلیدهای API (Load Balancing) ---
+// --- بخش مدیریت کلیدهای API (Random Selection) ---
 const apiKeysEnv = process.env.ALL_GEMINI_API_KEYS;
+// تبدیل رشته کلیدها به آرایه و حذف فاصله‌های اضافی
 const apiKeys = apiKeysEnv ? apiKeysEnv.split(',').map(key => key.trim()).filter(key => key) : [];
 
 if (apiKeys.length === 0) {
-  console.error('🔴 خطای حیاتی: هیچ کلید API در متغیر ALL_GEMINI_API_KEYS یافت نشد!');
+  console.error('🔴 خطای حیاتی: هیچ کلید API یافت نشد! لطفا متغیر ALL_GEMINI_API_KEYS را تنظیم کنید.');
   process.exit(1);
 }
-console.log(`🚀 سرور با ${apiKeys.length} کلید API آماده‌سازی شد.`);
-
-let currentKeyIndex = 0;
-const getNextKeyIndex = () => {
-    // چرخش ساده بین کلیدها (Round Robin)
-    const index = currentKeyIndex;
-    currentKeyIndex = (currentKeyIndex + 1) % apiKeys.length;
-    return index;
-};
+console.log(`🚀 سرور با ${apiKeys.length} کلید API آماده‌سازی شد. (حالت انتخاب تصادفی)`);
 
-// --- مکانیزم Heartbeat برای جلوگیری از قطعی سرور ---
-// این تابع اتصالات مرده را شناسایی و حذف می‌کند
+// --- مکانیزم Heartbeat برای جلوگیری از قطعی و هنگ کردن سرور ---
+// این تابع هر ۳۰ ثانیه اتصالات مرده را پاک می‌کند تا حافظه آزاد شود
 function heartbeat() {
   this.isAlive = true;
 }
@@ -65,14 +58,14 @@ function heartbeat() {
 const interval = setInterval(function ping() {
   wss.clients.forEach(function each(ws) {
     if (ws.isAlive === false) {
-        console.log('💀 قطع اتصال مرده (Zombie Connection) برای آزادسازی حافظه.');
+        // اگر کلاینت پاسخ نداد، اتصال را قطع کن تا حافظه آزاد شود
         return ws.terminate();
     }
     
     ws.isAlive = false;
     ws.ping();
   });
-}, 30000); // چک کردن هر ۳۰ ثانیه
+}, 30000);
 
 wss.on('close', function close() {
   clearInterval(interval);
@@ -81,7 +74,7 @@ wss.on('close', function close() {
 // --- توابع اتصال به گوگل ---
 
 /**
- * اتصال کلاینت و جمینای را به هم متصل می‌کند
+ * اتصال سوکت کلاینت و سوکت جمینای را به هم متصل می‌کند
  */
 function attachGeminiEventHandlers(clientWs, geminiWs, apiKeyUsed) {
   // انتقال پیام از گوگل به کلاینت
@@ -93,27 +86,32 @@ function attachGeminiEventHandlers(clientWs, geminiWs, apiKeyUsed) {
 
   geminiWs.on('error', (error) => {
     console.error(`🔴 خطای جمینای (کلید ...${apiKeyUsed.slice(-4)}):`, error.message);
+    // در صورت خطا، سوکت کلاینت را نمی‌بندیم تا شاید بتواند دوباره تلاش کند، 
+    // اما معمولا کلاینت خودش قطع می‌شود.
     if (clientWs.readyState === WebSocket.OPEN) clientWs.close();
   });
 
   geminiWs.on('close', (code) => {
-    if (code !== 1000 && code !== 1005) {
-        console.log(`🟡 گوگل اتصال را بست (کلید ...${apiKeyUsed.slice(-4)}). کد: ${code}`);
-    }
     if (clientWs.readyState === WebSocket.OPEN) clientWs.close();
   });
   
-  // هن��ل کردن پونگ برای زنده نگه داشتن اتصال گوگل (اختیاری ولی مفید)
-  geminiWs.on('ping', () => geminiWs.pong());
+  geminiWs.on('ping', () => {
+      try { geminiWs.pong(); } catch (e) {}
+  });
 }
 
 /**
- * تلاش برای اتصال به جمینای با قابلیت امتحان مجدد (Retry Logic)
+ * تلاش برای اتصال به جمینای با انتخاب تصادفی و امتحان مجدد
+ * @param {WebSocket} clientWs سوکت کاربر
+ * @param {Object} setupData داده‌های اولیه (کانفیگ)
+ * @param {Number} startIndex ایندکس شروع (که به صورت تصادفی انتخاب شده)
+ * @param {Number} attemptCount شمارنده تلاش‌ها
  */
 async function tryConnectToGemini(clientWs, setupData, startIndex, attemptCount = 0) {
-  // اگر کلاینت در حین تلاش قطع شد، ادامه نده
+  // اگر کلاینت در حین تلاش قطع شد، ادامه نده (جلوگیری از لیک حافظه)
   if (clientWs.readyState !== WebSocket.OPEN) return null;
 
+  // اگر تمام کلیدها تست شدند و هیچکدام کار نکرد
   if (attemptCount >= apiKeys.length) {
     console.error(`⛔ تمام ${apiKeys.length} کلید API با شکست مواجه شدند.`);
     if (clientWs.readyState === WebSocket.OPEN) {
@@ -123,29 +121,33 @@ async function tryConnectToGemini(clientWs, setupData, startIndex, attemptCount
     return null;
   }
 
+  // منطق انتخاب کلید:
+  // بار اول: ایندکس کاملا تصادفی (startIndex)
+  // دفعات بعد (در صورت خطا): کلید بعدی در لیست به صورت چرخشی
   const keyIndexToTry = (startIndex + attemptCount) % apiKeys.length;
   const apiKey = apiKeys[keyIndexToTry];
+  
   const url = `wss://generativelanguage.googleapis.com/ws/google.ai.generativelanguage.v1alpha.GenerativeService.BidiGenerateContent?key=${apiKey}`;
 
   return new Promise((resolve) => {
     const geminiWs = new WebSocket(url);
     
-    // تایم‌اوت اتصال (اگر گوگل تا ۸ ثانیه جواب نداد، برو کلید بعدی)
+    // تایم‌اوت سخت: اگر گوگل تا ۸ ثانیه هیچ پاسخی نداد (هنگ کرد)، برو کلید بعدی
     const timeout = setTimeout(() => {
-      console.warn(`⏳ تایم‌اوت اتصال (کلید ...${apiKey.slice(-4)}). تلاش مجدد...`);
-      geminiWs.terminate(); // کشتن اتصال گیر کرده
+      console.warn(`⏳ تایم‌اوت اتصال (کلید ...${apiKey.slice(-4)}). رفتن به کلید بعدی...`);
+      geminiWs.terminate(); 
       resolve(tryConnectToGemini(clientWs, setupData, startIndex, attemptCount + 1));
     }, 8000);
 
     geminiWs.on('open', () => {
       clearTimeout(timeout);
-      // بررسی مجدد وضعیت کلاینت قبل از ارسال
+      
       if (clientWs.readyState !== WebSocket.OPEN) {
           geminiWs.close();
           return resolve(null);
       }
       
-      console.log(`🔗 متصل شد به گوگل با کلید اندیس ${keyIndexToTry}`);
+      console.log(`🔗 اتصال موفق با کلید شماره ${keyIndexToTry} (تلاش ${attemptCount + 1})`);
       
       try {
         geminiWs.send(JSON.stringify(setupData));
@@ -160,8 +162,8 @@ async function tryConnectToGemini(clientWs, setupData, startIndex, attemptCount
 
     geminiWs.on('error', (err) => {
       clearTimeout(timeout);
-      // فقط لاگ کن و برو بعدی، هندلرهای اصلی هنوز ست نشدن
-      console.warn(`⚠️ خطای اتصال (کلید ...${apiKey.slice(-4)}): ${err.message || 'Unknown'}`);
+      console.warn(`⚠️ خطای اتصال (کلید ...${apiKey.slice(-4)}): ${err.message || 'Unknown'}. تلاش با کلید بعدی...`);
+      // فراخوانی بازگشتی برای تست کلید بعدی
       resolve(tryConnectToGemini(clientWs, setupData, startIndex, attemptCount + 1));
     });
   });
@@ -172,40 +174,39 @@ app.use(express.static(path.join(__dirname, '../build')));
 
 // API Endpoint دستورالعمل‌ها
 app.get('/api/instructions', (req, res) => {
-    res.setHeader('Cache-Control', 'no-store'); // جلوگیری از کش شدن قدیمی
+    res.setHeader('Cache-Control', 'no-store'); 
     res.json(personalityInstructions);
 });
 
 // --- مدیریت WebSocket کلاینت ---
 wss.on('connection', (ws, req) => {
-  ws.isAlive = true; // نشان‌گذاری به عنوان زنده
-  ws.on('pong', heartbeat); // پاسخ به پینگ سرور
+  ws.isAlive = true;
+  ws.on('pong', heartbeat);
 
-  const startingKeyIndex = getNextKeyIndex();
-  // const clientIp = req.socket.remoteAddress; // برای لاگ کردن IP در صورت نیاز
+  // *تغییر اصلی:* انتخاب یک نقطه شروع کاملاً تصادفی برای هر کاربر
+  // این کار باعث می‌شود بار روی سرور پخش شود و نیاز به حافظه برای نگهداری نوبت نباشد.
+  const randomStartIndex = Math.floor(Math.random() * apiKeys.length);
   
   let geminiWs = null;
   let isConnecting = false;
 
   ws.on('message', async (message) => {
     try {
-      // اگر پیام متنی است (JSON)
       if (!Buffer.isBuffer(message) || message[0] === 123) { 
         const msgStr = message.toString();
-        // چک کردن ساده برای تشخیص JSON
         if (msgStr.startsWith('{')) {
             const data = JSON.parse(msgStr);
             if (data.setup) {
-                if (isConnecting || geminiWs) return; // جلوگیری از دوبار وصل شدن
+                if (isConnecting || geminiWs) return; 
                 isConnecting = true;
-                geminiWs = await tryConnectToGemini(ws, data, startingKeyIndex);
+                // ارسال ایندکس تصادفی به تابع اتصال
+                geminiWs = await tryConnectToGemini(ws, data, randomStartIndex);
                 isConnecting = false;
                 return;
             }
         }
       }
       
-      // اگر به گوگل وصلیم، پیام را بفرست
       if (geminiWs && geminiWs.readyState === WebSocket.OPEN) {
         geminiWs.send(message);
       }
@@ -214,20 +215,21 @@ wss.on('connection', (ws, req) => {
     }
   });
 
-  // وقتی کاربر مرورگر را می‌بندد یا اینترنتش قطع می‌شود
   ws.on('close', () => {
-    // console.log('👋 کلاینت قطع شد.');
     if (geminiWs) {
-        if (geminiWs.readyState === WebSocket.OPEN || geminiWs.readyState === WebSocket.CONNECTING) {
-            geminiWs.close();
-        }
+        // بستن اجباری اتصال گوگل برای آزادسازی منابع
+        try {
+            geminiWs.terminate(); 
+        } catch(e) {}
         geminiWs = null;
     }
   });
 
   ws.on('error', (error) => {
-    console.error(`خطای سوکت کلاینت: ${error.message}`);
-    if (geminiWs) geminiWs.close();
+    if (geminiWs) {
+        try { geminiWs.terminate(); } catch(e) {}
+        geminiWs = null;
+    }
   });
 });
 
@@ -235,4 +237,4 @@ wss.on('connection', (ws, req) => {
 app.get('*', (req, res) => res.sendFile(path.join(__dirname, '../build', 'index.html')));
 
 const PORT = process.env.PORT || 3001;
-server.listen(PORT, () => console.log(`🚀 سرور (نسخه پایدار) روی پورت ${PORT} اجرا شد.`));
+server.listen(PORT, () => console.log(`🚀 سرور (حالت تصادفی) روی پورت ${PORT} اجرا شد.`));