Upload index.js

server/index.js

@@ -9,211 +9,119 @@ const app = express();
 const server = http.createServer(app);
 const wss = new WebSocketServer({ server });
 
-// --------------------------------------------------------------------------
-// 1. تنظیمات و متغیرها
-// --------------------------------------------------------------------------
+// بخش خواندن دستورالعمل‌های شخصیت از Secrets
 const instructionSecretNames = {
-    default: 'PERSONALITY_DEFAULT',
-    teacher: 'PERSONALITY_TEACHER',
-    poetic: 'PERSONALITY_POETIC',
-    funny: 'PERSONALITY_FUNNY',
+  default: 'PERSONALITY_DEFAULT',
+  teacher: 'PERSONALITY_TEACHER',
+  poetic: 'PERSONALITY_POETIC',
+  funny: 'PERSONALITY_FUNNY',
 };
-
 const personalityInstructions = {};
-console.log('🔄 در حال بارگذاری تنظیمات...');
+console.log('در حال خواندن دستورالعمل‌های شخصیت از Secrets...');
 Object.keys(instructionSecretNames).forEach(key => {
-    const secretName = instructionSecretNames[key];
-    const instruction = process.env[secretName];
-    if (instruction) {
-        personalityInstructions[key] = instruction;
-    } else {
-        personalityInstructions[key] = ""; // مقدار پیش‌فرض خالی برای جلوگیری از خطا
-    }
+  const secretName = instructionSecretNames[key];
+  const instruction = process.env[secretName];
+  if (instruction) {
+    personalityInstructions[key] = instruction;
+    console.log(`- دستورالعمل '${key}' با موفقیت خوانده شد.`);
+  } else {
+    personalityInstructions[key] = `دستورالعمل '${key}' از سرور بارگذاری نشد.`;
+    console.warn(`** هشدار: Secret با نام '${secretName}' یافت نشد! **`);
+  }
 });
 
-// مدیریت کلیدهای API
+// بخش مدیریت کلیدهای API (کد هوشمند شما بدون تغییر)
 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 را در تنظیمات اسپیس وارد کنید.');
-    // سرور را متوقف نمیکنیم تا حداقل صفحه بالا بیاید و خطا دهد
-} else {
-    console.log(`✅ تعداد ${apiKeys.length} کلید API شناسایی شد.`);
+  console.error('🔴 خطای حیاتی: Secret با نام ALL_GEMINI_API_KEYS یافت نشد!');
+  process.exit(1);
 }
-
+console.log(`✅ تعداد ${apiKeys.length} کلید API بارگذاری شد.`);
 let currentKeyIndex = 0;
-const getNextKeyIndex = () => {
-    const index = currentKeyIndex;
-    currentKeyIndex = (currentKeyIndex + 1) % apiKeys.length;
-    return index;
+const getStartingKeyIndex = () => {
+  const index = currentKeyIndex % apiKeys.length;
+  currentKeyIndex++;
+  return index;
 };
-
-// --------------------------------------------------------------------------
-// 2. توابع کمکی جیمینای
-// --------------------------------------------------------------------------
-
+// توابع attachGeminiEventHandlers و tryConnectToGemini شما اینجا بدون تغییر قرار می‌گیرند.
+// ... (برای کوتاهی اینجا حذف شده، اما شما باید آنها را نگه دارید) ...
 function attachGeminiEventHandlers(clientWs, geminiWs, apiKeyUsed) {
-    // وقتی پیامی از جیمینای آمد، به کلاینت (کاربر) بفرست
-    const handleGeminiMessage = (data) => {
-        if (clientWs.readyState === WebSocket.OPEN) {
-            // نکته مهم: ارسال به صورت باینری (Blob) تا کلاینت بتواند پردازش کند
-            clientWs.send(data, { binary: true });
-        }
-    };
-
-    const handleGeminiError = (error) => {
-        console.error(`🔴 خطای جیمینای:`, error.message);
-        if (clientWs.readyState === WebSocket.OPEN) {
-            clientWs.close();
-        }
-    };
-
-    const handleGeminiClose = () => {
-        if (clientWs.readyState === WebSocket.OPEN) {
-            clientWs.close();
-        }
-    };
-
-    geminiWs.on('message', handleGeminiMessage);
-    geminiWs.on('error', handleGeminiError);
-    geminiWs.on('close', handleGeminiClose);
-
-    // *** بخش مهم برای رفع نشت حافظه ***
-    // وقتی کلاینت قطع شد، تمام لیسنرهای جیمینای را پاک کن و اتصال را ببند
-    clientWs.on('close', () => {
-        geminiWs.off('message', handleGeminiMessage);
-        geminiWs.off('error', handleGeminiError);
-        geminiWs.off('close', handleGeminiClose);
-        
-        if (geminiWs.readyState === WebSocket.OPEN || geminiWs.readyState === WebSocket.CONNECTING) {
-            geminiWs.close();
-        }
-    });
+  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('close', code => {
+    if (code !== 1000) console.log(`🟡 اتصال جیمینای بسته شد (کلید ...${apiKeyUsed.slice(-4)}). کد: ${code}`);
+    clientWs.readyState === WebSocket.OPEN && clientWs.close();
+  });
 }
-
 async function tryConnectToGemini(clientWs, setupData, startIndex, attemptCount = 0) {
-    if (attemptCount >= apiKeys.length) {
-        console.error(`❌ همه کلیدها تست شدند و ناموفق بودند.`);
-        return null;
-    }
-
-    const keyIndexToTry = (startIndex + attemptCount) % apiKeys.length;
-    const apiKey = apiKeys[keyIndexToTry];
-
-    return new Promise(resolve => {
-        const url = `wss://generativelanguage.googleapis.com/ws/google.ai.generativelanguage.v1alpha.GenerativeService.BidiGenerateContent?key=${apiKey}`;
-        
-        let geminiWs;
-        try {
-            geminiWs = new WebSocket(url);
-        } catch (e) {
-            console.error("خطا در ایجاد سوکت:", e);
-            resolve(tryConnectToGemini(clientWs, setupData, startIndex, attemptCount + 1));
-            return;
-        }
-
-        // تایم‌اوت اتصال (10 ثانیه)
-        const timeout = setTimeout(() => {
-            console.warn(`⏳ تایم‌اوت اتصال (کلید ${keyIndexToTry}). امتحان کلید بعدی...`);
-            try { geminiWs.close(); } catch(e){}
-            resolve(tryConnectToGemini(clientWs, setupData, startIndex, attemptCount + 1));
-        }, 10000);
-
-        geminiWs.on('open', () => {
-            clearTimeout(timeout);
-            
-            // اگر کاربر در حین اتصال قطع شد
-            if (clientWs.readyState !== WebSocket.OPEN) {
-                geminiWs.close();
-                resolve(null);
-                return;
-            }
-
-            console.log(`🔗 متصل به جیمینای (کلید ${keyIndexToTry})`);
-            
-            try {
-                geminiWs.send(JSON.stringify(setupData));
-                attachGeminiEventHandlers(clientWs, geminiWs, apiKey);
-                resolve(geminiWs);
-            } catch (e) {
-                console.error("خطا در ارسال Setup:", e);
-                geminiWs.close();
-                resolve(null);
-            }
-        });
-
-        geminiWs.on('error', (err) => {
-            clearTimeout(timeout);
-            console.warn(`⚠️ خطا در اتصال (کلید ${keyIndexToTry}):`, err.message);
-            resolve(tryConnectToGemini(clientWs, setupData, startIndex, attemptCount + 1));
-        });
+  if (attemptCount >= apiKeys.length) {
+    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 timeout = setTimeout(() => {
+      geminiWs.removeAllListeners() && geminiWs.close();
+      console.warn(`🟡 اتصال با کلید ...${apiKey.slice(-4)} زمان‌بر شد. امتحان کلید بعدی...`);
+      resolve(tryConnectToGemini(clientWs, setupData, startIndex, attemptCount + 1));
+    }, 8000);
+    geminiWs.on('open', () => {
+      clearTimeout(timeout);
+      console.log(`✅ اتصال با کلید اندیس ${keyIndexToTry} موفق بود.`);
+      try {
+        geminiWs.send(JSON.stringify(setupData));
+      } catch (e) {
+        clientWs.close();
+        resolve(null);
+        return;
+      }
+      attachGeminiEventHandlers(clientWs, geminiWs, apiKey);
+      resolve(geminiWs);
     });
+    geminiWs.on('error', () => {
+      clearTimeout(timeout);
+      geminiWs.removeAllListeners();
+      console.warn(`🟡 اتصال با کلید ...${apiKey.slice(-4)} ناموفق بود. امتحان کلید بعدی...`);
+      resolve(tryConnectToGemini(clientWs, setupData, startIndex, attemptCount + 1));
+    });
+  });
 }
 
-// --------------------------------------------------------------------------
-// 3. سرور و سوکت اصلی
-// --------------------------------------------------------------------------
-
+// سرو کردن فایل‌های استاتیک
 app.use(express.static(path.join(__dirname, '../build')));
-app.get('/api/instructions', (req, res) => res.json(personalityInstructions));
-
-wss.on('connection', function connection(ws) {
-    const myKeyIndex = getNextKeyIndex();
-    console.log(`🔌 کلاینت جدید وصل شد.`);
-
-    let geminiWs = null;
 
-    ws.on('message', async (message) => {
-        try {
-            // اگر پیام باینری است (مثل داده‌های صوتی)، مستقیماً بفرست
-            if (Buffer.isBuffer(message)) {
-                if (geminiWs && geminiWs.readyState === WebSocket.OPEN) {
-                    geminiWs.send(message);
-                }
-                return;
-            }
-
-            // پیام‌های متنی (JSON)
-            const strMsg = message.toString();
-            let data;
-            try {
-                data = JSON.parse(strMsg);
-            } catch (e) {
-                // اگر JSON نیست، شاید پیام متنی ساده باشد، بفرست بره
-                if (geminiWs && geminiWs.readyState === WebSocket.OPEN) {
-                    geminiWs.send(strMsg);
-                }
-                return;
-            }
-
-            // اگر پیام Setup (شروع مکالمه) است
-            if (data.setup) {
-                if (geminiWs) return; // اگر قبلاً وصل شده، دوباره وصل نشو
-                
-                geminiWs = await tryConnectToGemini(ws, data, myKeyIndex);
-            } 
-            // سایر پیام‌های JSON
-            else if (geminiWs && geminiWs.readyState === WebSocket.OPEN) {
-                geminiWs.send(strMsg);
-            }
-
-        } catch (error) {
-            console.error("خطا در پردازش پیام:", error);
-        }
-    });
-
-    ws.on('close', () => {
-        console.log(`👋 کلاینت قطع شد.`);
-        if (geminiWs) {
-            try { geminiWs.close(); } catch(e){}
-        }
-    });
+// API Endpoint جدید برای ارسال دستورالعمل‌ها
+app.get('/api/instructions', (req, res) => res.json(personalityInstructions));
 
-    ws.on('error', (err) => console.error('خطای سوکت کلاینت:', err));
+// مدیریت اتصال WebSocket کلاینت (بدون تغییر)
+wss.on('connection', ws => {
+  const startingKeyIndex = getStartingKeyIndex();
+  console.log(`🔌 کلاینت جدید متصل شد. کلید شروع: اندیس ${startingKeyIndex}`);
+  let geminiWs = null;
+  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));
+      }
+    } catch (e) { /* خطا */ }
+  });
+  ws.on('close', () => console.log(`🔌 کلاینت با کلید شروع اندیس ${startingKeyIndex} قطع شد.`) && geminiWs?.close());
+  ws.on('error', error => console.error(`🔴 خطای WebSocket کلاینت:`, error) && geminiWs?.close());
 });
 
+// رسیدگی به سایر درخواست‌ها
 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}`));