Update server/index.js

server/index.js

@@ -6,122 +6,233 @@ const http = require('node:http');
 require('dotenv').config();
 
 const app = express();
-const server = http.createServer(app);
-const wss = new WebSocketServer({ server });
+// افزایش محدودیت‌های سرور برای هندل کردن تعداد بالای درخواست
+const server = http.createServer({
+    maxHeaderSize: 16384 // 16KB
+}, app);
 
-// بخش خواندن دستورالعمل‌های شخصیت از Secrets
+const wss = new WebSocketServer({ 
+    server,
+    clientTracking: true // برای مدیریت هارت‌بیت ضروری است
+});
+
+// --- بخش مدیریت تنظیمات شخصیت‌ها ---
 const instructionSecretNames = {
   default: 'PERSONALITY_DEFAULT',
   teacher: 'PERSONALITY_TEACHER',
   poetic: 'PERSONALITY_POETIC',
   funny: 'PERSONALITY_FUNNY',
 };
+
 const personalityInstructions = {};
-console.log('در حال خواندن دستورالعمل‌های شخصیت از Secrets...');
+console.log('🔄 در حال خواندن دستورالعمل‌های شخصیت از Secrets...');
 Object.keys(instructionSecretNames).forEach(key => {
   const secretName = instructionSecretNames[key];
   const instruction = process.env[secretName];
   if (instruction) {
     personalityInstructions[key] = instruction;
-    console.log(`- دستورالعمل '${key}' با موفقیت خوانده شد.`);
+    console.log(`✅ دستورالعمل '${key}' با موفقیت خوانده شد.`);
   } else {
-    personalityInstructions[key] = `دستورالعمل '${key}' از سرور بارگذاری نشد.`;
-    console.warn(`** هشدار: Secret با نام '${secretName}' یافت نشد! **`);
+    personalityInstructions[key] = `دستورالعمل '${key}' یافت نشد.`;
+    console.warn(`⚠️ هشدار: Secret با نام '${secretName}' تنظیم نشده است.`);
   }
 });
 
-// بخش مدیریت کلیدهای API (کد هوشمند شما بدون تغییر)
+// --- بخش مدیریت کلیدهای API (Load Balancing) ---
 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('🔴 خطای حیاتی: Secret با نام ALL_GEMINI_API_KEYS یافت نشد!');
+  console.error('🔴 خطای حیاتی: هیچ کلید API در متغیر ALL_GEMINI_API_KEYS یافت نشد!');
   process.exit(1);
 }
-console.log(`✅ تعداد ${apiKeys.length} کلید API بارگذاری شد.`);
+console.log(`🚀 سرور با ${apiKeys.length} کلید API آماده‌سازی شد.`);
+
 let currentKeyIndex = 0;
-const getStartingKeyIndex = () => {
-  const index = currentKeyIndex % apiKeys.length;
-  currentKeyIndex++;
-  return index;
+const getNextKeyIndex = () => {
+    // چرخش ساده بین کلیدها (Round Robin)
+    const index = currentKeyIndex;
+    currentKeyIndex = (currentKeyIndex + 1) % apiKeys.length;
+    return index;
 };
-// توابع attachGeminiEventHandlers و tryConnectToGemini شما اینجا بدون تغییر قرار می‌گیرند.
-// ... (برای کوتاهی اینجا حذف شده، اما شما باید آنها را نگه دارید) ...
+
+// --- مکانیزم Heartbeat برای جلوگیری از قطعی سرور ---
+// این تابع اتصالات مرده را شناسایی و حذف می‌کند
+function heartbeat() {
+  this.isAlive = true;
+}
+
+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); // چک کردن هر ۳۰ ثانیه
+
+wss.on('close', function close() {
+  clearInterval(interval);
+});
+
+// --- توابع اتصال به گوگل ---
+
+/**
+ * اتصال کلاینت و جمینای را به هم متصل می‌کند
+ */
 function attachGeminiEventHandlers(clientWs, geminiWs, apiKeyUsed) {
-  geminiWs.on('message', data => clientWs.readyState === WebSocket.OPEN && clientWs.send(data, { binary: true }));
-  geminiWs.on('error', error => {
-    console.error(`🔴 خطای WebSocket جیمینای (کلید ...${apiKeyUsed.slice(-4)}):`, error.message);
-    clientWs.readyState === WebSocket.OPEN && clientWs.close();
+  // انتقال پیام از گوگل به کلاینت
+  geminiWs.on('message', (data) => {
+    if (clientWs.readyState === WebSocket.OPEN) {
+        clientWs.send(data, { binary: true });
+    }
   });
-  geminiWs.on('close', code => {
-    if (code !== 1000) console.log(`🟡 اتصال جیمینای بسته شد (کلید ...${apiKeyUsed.slice(-4)}). کد: ${code}`);
-    clientWs.readyState === WebSocket.OPEN && clientWs.close();
+
+  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());
 }
+
+/**
+ * تلاش برای اتصال به جمینای با قابلیت امتحان مجدد (Retry Logic)
+ */
 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) clientWs.send(JSON.stringify({ error: "خطای داخلی سرور." })) && clientWs.close();
+    console.error(`⛔ تمام ${apiKeys.length} کلید API با شکست مواجه شدند.`);
+    if (clientWs.readyState === WebSocket.OPEN) {
+        clientWs.send(JSON.stringify({ error: "سرویس موقتاً در دسترس نیست (ظرفیت تکمیل)." }));
+        clientWs.close();
+    }
     return null;
   }
+
   const keyIndexToTry = (startIndex + attemptCount) % apiKeys.length;
   const apiKey = apiKeys[keyIndexToTry];
-  return new Promise(resolve => {
-    const geminiWs = new WebSocket(`wss://generativelanguage.googleapis.com/ws/google.ai.generativelanguage.v1alpha.GenerativeService.BidiGenerateContent?key=${apiKey}`);
+  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(() => {
-      geminiWs.removeAllListeners() && geminiWs.close();
-      console.warn(`🟡 اتصال با کلید ...${apiKey.slice(-4)} زمان‌بر شد. امتحان کلید بعدی...`);
+      console.warn(`⏳ تایم‌اوت اتصال (کلید ...${apiKey.slice(-4)}). تلاش مجدد...`);
+      geminiWs.terminate(); // کشتن اتصال گیر کرده
       resolve(tryConnectToGemini(clientWs, setupData, startIndex, attemptCount + 1));
     }, 8000);
+
     geminiWs.on('open', () => {
       clearTimeout(timeout);
-      console.log(`✅ اتصال با کلید اندیس ${keyIndexToTry} موفق بود.`);
+      // بررسی مجدد وضعیت کلاینت قبل از ارسال
+      if (clientWs.readyState !== WebSocket.OPEN) {
+          geminiWs.close();
+          return resolve(null);
+      }
+      
+      console.log(`🔗 متصل شد به گوگل با کلید اندیس ${keyIndexToTry}`);
+      
       try {
         geminiWs.send(JSON.stringify(setupData));
+        attachGeminiEventHandlers(clientWs, geminiWs, apiKey);
+        resolve(geminiWs);
       } catch (e) {
-        clientWs.close();
-        resolve(null);
-        return;
+        console.error("خطا در ارسال تنظیمات اولیه:", e);
+        geminiWs.close();
+        resolve(tryConnectToGemini(clientWs, setupData, startIndex, attemptCount + 1));
       }
-      attachGeminiEventHandlers(clientWs, geminiWs, apiKey);
-      resolve(geminiWs);
     });
-    geminiWs.on('error', () => {
+
+    geminiWs.on('error', (err) => {
       clearTimeout(timeout);
-      geminiWs.removeAllListeners();
-      console.warn(`🟡 اتصال با کلید ...${apiKey.slice(-4)} ناموفق بود. امتحان کلید بعدی...`);
+      // فقط لاگ کن و برو بعدی، هندلرهای اصلی هنوز ست نشدن
+      console.warn(`⚠️ خطای اتصال (کلید ...${apiKey.slice(-4)}): ${err.message || 'Unknown'}`);
       resolve(tryConnectToGemini(clientWs, setupData, startIndex, attemptCount + 1));
     });
   });
 }
 
-// سرو کردن فایل‌های استاتیک
+// --- سرو کردن فایل‌های استاتیک ---
 app.use(express.static(path.join(__dirname, '../build')));
 
-// API Endpoint جدید برای ارسال دستورالعمل‌ها
-app.get('/api/instructions', (req, res) => res.json(personalityInstructions));
+// API Endpoint دستورالعمل‌ها
+app.get('/api/instructions', (req, res) => {
+    res.setHeader('Cache-Control', 'no-store'); // جلوگیری از کش شدن قدیمی
+    res.json(personalityInstructions);
+});
+
+// --- مدیریت WebSocket کلاینت ---
+wss.on('connection', (ws, req) => {
+  ws.isAlive = true; // نشان‌گذاری به عنوان زنده
+  ws.on('pong', heartbeat); // پاسخ به پینگ سرور
 
-// مدیریت اتصال WebSocket کلاینت (بدون تغییر)
-wss.on('connection', ws => {
-  const startingKeyIndex = getStartingKeyIndex();
-  console.log(`🔌 کلاینت جدید متصل شد. کلید شروع: اندیس ${startingKeyIndex}`);
+  const startingKeyIndex = getNextKeyIndex();
+  // const clientIp = req.socket.remoteAddress; // برای لاگ کردن IP در صورت نیاز
+  
   let geminiWs = null;
-  ws.on('message', async message => {
+  let isConnecting = false;
+
+  ws.on('message', async (message) => {
     try {
-      const data = JSON.parse(message.toString());
-      if (data.setup) {
-        if (geminiWs) return;
-        geminiWs = await tryConnectToGemini(ws, data, startingKeyIndex);
-      } else if (geminiWs?.readyState === WebSocket.OPEN) {
-        geminiWs.send(JSON.stringify(data));
+      // اگر پیام متنی است (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; // جلوگیری از دوبار وصل شدن
+                isConnecting = true;
+                geminiWs = await tryConnectToGemini(ws, data, startingKeyIndex);
+                isConnecting = false;
+                return;
+            }
+        }
+      }
+      
+      // اگر به گوگل وصلیم، پیام را بفرست
+      if (geminiWs && geminiWs.readyState === WebSocket.OPEN) {
+        geminiWs.send(message);
       }
-    } catch (e) { /* خطا */ }
+    } catch (e) {
+      console.error("خطا در پردازش پیام کلاینت:", e.message);
+    }
+  });
+
+  // وقتی کاربر مرورگر را می‌بندد یا اینترنتش قطع می‌شود
+  ws.on('close', () => {
+    // console.log('👋 کلاینت قطع شد.');
+    if (geminiWs) {
+        if (geminiWs.readyState === WebSocket.OPEN || geminiWs.readyState === WebSocket.CONNECTING) {
+            geminiWs.close();
+        }
+        geminiWs = null;
+    }
+  });
+
+  ws.on('error', (error) => {
+    console.error(`خطای سوکت کلاینت: ${error.message}`);
+    if (geminiWs) geminiWs.close();
   });
-  ws.on('close', () => console.log(`🔌 کلاینت با کلید شروع اندیس ${startingKeyIndex} قطع شد.`) && geminiWs?.close());
-  ws.on('error', error => console.error(`🔴 خطای WebSocket کلاینت:`, error) && geminiWs?.close());
 });
 
-// رسیدگی به سایر درخواست‌ها
+// ارسال تمام درخواست‌های دیگر به React
 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} اجرا شد.`));