Update server/index.js

server/index.js

@@ -1,5 +1,4 @@
 // server/index.js
-
 const express = require('express');
 const path = require('node:path');
 const { WebSocketServer, WebSocket } = require('ws');
@@ -10,209 +9,119 @@ const app = express();
 const server = http.createServer(app);
 const wss = new WebSocketServer({ server });
 
-// --- START: منطق جدید و بهبود یافته برای مدیریت API Key ---
+// بخش خواندن دستورالعمل‌های شخصیت از Secrets
+const instructionSecretNames = {
+  default: 'PERSONALITY_DEFAULT',
+  teacher: 'PERSONALITY_TEACHER',
+  poetic: 'PERSONALITY_POETIC',
+  funny: 'PERSONALITY_FUNNY',
+};
+const personalityInstructions = {};
+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}' با موفقیت خوانده شد.`);
+  } else {
+    personalityInstructions[key] = `دستورالعمل '${key}' از سرور بارگذاری نشد.`;
+    console.warn(`** هشدار: Secret با نام '${secretName}' یافت نشد! **`);
+  }
+});
 
-// ۱. خواندن تمام کلیدها از یک Secret واحد (جدا شده با کاما)
+// بخش مدیریت کلیدهای API (کد هوشمند شما بدون تغییر)
 const apiKeysEnv = process.env.ALL_GEMINI_API_KEYS;
 const apiKeys = apiKeysEnv ? apiKeysEnv.split(',').map(key => key.trim()).filter(key => key) : [];
-
-const numKeys = apiKeys.length;
-
-if (numKeys === 0) {
-  console.error(
-    'خطای حیاتی: هیچ Secret با نام ALL_GEMINI_API_KEYS یافت نشد یا خالی است!' +
-    ' لطفاً کلیدهای خود را به صورت جدا شده با کاما در این Secret در تنظیمات Space خود اضافه کنید.'
-  );
+if (apiKeys.length === 0) {
+  console.error('🔴 خطای حیاتی: Secret با نام ALL_GEMINI_API_KEYS یافت نشد!');
   process.exit(1);
-} else {
-  console.log(`✅ تعداد ${numKeys} کلید API جیمینای با موفقیت بارگذاری شد.`);
 }
-
-// ۲. شمارنده سراسری برای توزیع بار بین کاربران (انتخاب کلید شروع)
+console.log(`✅ تعداد ${apiKeys.length} کلید API بارگذاری شد.`);
 let currentKeyIndex = 0;
-
-function getStartingKeyIndex() {
-  const startingIndex = currentKeyIndex % numKeys;
+const getStartingKeyIndex = () => {
+  const index = currentKeyIndex % apiKeys.length;
   currentKeyIndex++;
-  return startingIndex;
-}
-
-// --- END: منطق جدید مدیریت API Key ---
-
-
-// این تابع رویدادهای لازم را به یک اتصال موفق جیمینای متصل می‌کند
+  return index;
+};
+// توابع attachGeminiEventHandlers و tryConnectToGemini شما اینجا بدون تغییر قرار می‌گیرند.
+// ... (برای کوتاهی اینجا حذف شده، اما شما باید آنها را نگه دارید) ...
 function attachGeminiEventHandlers(clientWs, geminiWs, apiKeyUsed) {
-    // ارسال پیام از جیمینای به کلاینت
-    geminiWs.on('message', (data) => {
-        try {
-            if (clientWs.readyState === WebSocket.OPEN) {
-                // پیام‌ها به صورت باینری (Blob) برای کلاینت ارسال می‌شوند
-                clientWs.send(data, { binary: true });
-            }
-        } catch (error) {
-            console.error(`🔴 خطا در پردازش پیام جیمینای (کلید ...${apiKeyUsed.slice(-4)}):`, error);
-        }
-    });
-
-    // رسیدگی به خطای اتصال جیمینای
-    geminiWs.on('error', (error) => {
-        console.error(`🔴 خطای WebSocket جیمینای (کلید ...${apiKeyUsed.slice(-4)}):`, error.message);
-        if (clientWs.readyState === WebSocket.OPEN) clientWs.close();
-    });
-
-    // رسیدگی به بسته شدن اتصال جیمینای
-    geminiWs.on('close', (code, reason) => {
-        // لاگ بسته شدن اتصال فقط در صورتی که غیرعادی باشد
-        if (code !== 1000) {
-           console.log(`🟡 اتصال WebSocket جیمینای بسته شد (کلید ...${apiKeyUsed.slice(-4)}). کد: ${code}, دلیل: ${reason.toString()}`);
-        }
-        if (clientWs.readyState === WebSocket.OPEN) clientWs.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();
+  });
 }
-
-
-// --- START: تابع هوشمند جدید برای اتصال با قابلیت Failover ---
-
-/**
- * تلاش برای اتصال به جیمینای با استفاده از لیست کلیدها.
- * در صورت شکست، به صورت بازگشتی کلید بعدی را امتحان می‌کند.
- * @param {WebSocket} clientWs - وب‌سوکت کلاینت کاربر.
- * @param {object} setupData - داده‌های setup ارسال شده توسط کلاینت.
- * @param {number} attemptIndex - اندیس کلیدی که باید امتحان شود.
- * @returns {Promise<WebSocket|null>} - در صورت موفقیت، وب‌سوکت جیمینای را برمی‌گرداند.
- */
 async function tryConnectToGemini(clientWs, setupData, startIndex, attemptCount = 0) {
-    if (attemptCount >= numKeys) {
-        console.error(`🔴 تمام ${numKeys} کلید API امتحان شدند و ناموفق بودند. قطع اتصال کلاینت.`);
-        if (clientWs.readyState === WebSocket.OPEN) {
-            clientWs.send(JSON.stringify({ error: "خطای داخلی سرور: هیچ سرویس‌دهنده‌ای در دسترس نیست." }));
-            clientWs.close();
-        }
-        return null;
-    }
-
-    const keyIndexToTry = (startIndex + attemptCount) % numKeys;
-    const apiKey = apiKeys[keyIndexToTry];
-    // console.log(`⏳ تلاش برای اتصال با کلید اندیس ${keyIndexToTry} (کلید ...${apiKey.slice(-4)})`);
-
-    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} (کلید ...${apiKey.slice(-4)}) موفقیت‌آمیز بود.`);
-            
-            // ارسال پیام setup به محض برقراری اتصال
-            try {
-                geminiWs.send(JSON.stringify(setupData));
-            } catch (sendError) {
-                 console.error(`🔴 خطا در ارسال پیام setup اولیه به جیمینای:`, sendError);
-                 clientWs.close();
-                 resolve(null); // اتصال ناموفق بود
-                 return;
-            }
-            
-            // اتصال سایر event handler ها
-            attachGeminiEventHandlers(clientWs, geminiWs, apiKey);
-            resolve(geminiWs); // اتصال موفق را برگردان
-        });
-
-        geminiWs.on('error', (error) => {
-            clearTimeout(timeout);
-            geminiWs.removeAllListeners();
-            console.warn(`🟡 اتصال با کلید اندیس ${keyIndexToTry} (کلید ...${apiKey.slice(-4)}) ناموفق بود. دلیل: ${error.message.split('Unexpected server response: ')[1] || 'خطای نامشخص'}. امتحان کلید بعدی...`);
-            // به صورت بازگشتی کلید بعدی را امتحان کن
-            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));
     });
+  });
 }
 
-// --- END: تابع هوشمند جدید ---
-
-
-// سرو کردن فایل‌های استاتیک از پوشه build ری‌اکت
+// سرو کردن فایل‌های استاتیک
 app.use(express.static(path.join(__dirname, '../build')));
 
-wss.on('connection', (ws) => {
-    // به هر کلاینت یک کلید شروع اختصاص داده می‌شود تا بار توزیع شود
-    const startingKeyIndex = getStartingKeyIndex();
-    console.log(`🔌 یک کلاینت جدید متصل شد. کلید شروع اختصاص یافته: اندیس ${startingKeyIndex} (کلید ...${apiKeys[startingKeyIndex].slice(-4)})`);
-
-    let geminiWs = null; // اتصال جیمینای برای این کلاینت
-
-    ws.on('message', async (message) => {
-        try {
-            const data = JSON.parse(message.toString());
-
-            // اولین پیام باید `setup` باشد
-            if (data.setup) {
-                if (geminiWs) {
-                    // console.warn("کلاینت دوباره پیام setup ارسال کرد. نادیده گرفته شد.");
-                    return;
-                }
-                
-                // شروع فرآیند اتصال با قابلیت Failover
-                geminiWs = await tryConnectToGemini(ws, data, startingKeyIndex);
-                
-                if (!geminiWs) {
-                    console.error("🔴 اتصال به سرویس جیمینای برای این کلاینت پس از امتحان همه کلیدها ناموفق بود.");
-                }
-                return;
-            }
-            
-            // ارسال پیام‌های بعدی به جیمینای
-            if (geminiWs && geminiWs.readyState === WebSocket.OPEN) {
-                geminiWs.send(JSON.stringify(data));
-            } else if (!geminiWs) {
-                console.warn("🟡 پیام از کلاینت دریافت شد اما اتصال جیمینای هنوز برقرار نشده یا ناموفق بوده است.");
-                // می توان یک پیام خطا به کلاینت فرستاد
-            }
-
-        } catch (error) {
-            console.error('🔴 خطا در پردازش پیام کلاینت (احتمالا JSON نامعتبر):', error);
-        }
-    });
-
-    ws.on('close', () => {
-        console.log(`🔌 کلاینت با کلید شروع اندیس ${startingKeyIndex} قطع شد.`);
-        if (geminiWs && geminiWs.readyState !== WebSocket.CLOSED) {
-            geminiWs.close();
-        }
-    });
-
-    ws.on('error', (error) => {
-        console.error(`🔴 خطای WebSocket کلاینت (کلید شروع اندیس ${startingKeyIndex}):`, error);
-        if (geminiWs && geminiWs.readyState !== WebSocket.CLOSED) {
-            geminiWs.close();
-        }
-    });
+// API Endpoint جدید برای ارسال دستورالعمل‌ها
+app.get('/api/instructions', (req, res) => res.json(personalityInstructions));
+
+// مدیریت اتصال 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());
 });
 
-// رسیدگی به درخواست‌های باقیمانده با برگرداندن اپ React
-app.get('*', (req, res) => {
-    const indexPath = path.join(__dirname, '../build', 'index.html');
-    res.sendFile(indexPath, (err) => {
-        if (err) {
-            if (!res.headersSent) {
-                res.status(500).send('خطا در بارگذاری برنامه');
-            }
-        }
-    });
-});
+// رسیدگی به سایر درخواست‌ها
+app.get('*', (req, res) => res.sendFile(path.join(__dirname, '../build', 'index.html')));
 
 const PORT = process.env.PORT || 3001;
-server.listen(PORT, () => {
-    console.log(`🚀 سرور در حال اجرا روی پورت ${PORT}`);
-    if (numKeys > 0) {
-        console.log(`🔑 برای استفاده، Secret با نام 'ALL_GEMINI_API_KEYS' را با کلید(های) خود تنظیم کنید.`);
-    }
-});
+server.listen(PORT, () => console.log(`🚀 سرور در حال اجرا روی پورت ${PORT}`));