Update server/index.js

server/index.js

@@ -1,278 +1,218 @@
+// server/index.js
+
 const express = require('express');
 const path = require('node:path');
 const { WebSocketServer, WebSocket } = require('ws');
 const http = require('node:http');
-require('dotenv').config(); // برای تست محلی ممکن است نیاز باشد
+require('dotenv').config();
 
 const app = express();
 const server = http.createServer(app);
 const wss = new WebSocketServer({ server });
 
-// --- START: منطق جدید برای چرخش API Key ---
+// --- START: منطق جدید و بهبود یافته برای مدیریت API Key ---
 
-// ۱. خواندن تمام کلیدهای API از Secrets (متغیرهای محیطی در هاگینگ فیس)
-const apiKeys = [];
-let i = 1;
-while (process.env[`GEMINI_API_KEY_${i}`]) {
-  apiKeys.push(process.env[`GEMINI_API_KEY_${i}`]);
-  i++;
-}
+// ۱. خواندن تمام کلیدها از یک Secret واحد (جدا شده با کاما)
+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 با نام GEMINI_API_KEY_n یافت نشد!' +
-      ' لطفاً Secret ها را مانند GEMINI_API_KEY_1, GEMINI_API_KEY_2, ... ' +
-      'در تنظیمات Space خود اضافه کنید.'
+    'خطای حیاتی: هیچ Secret با نام ALL_GEMINI_API_KEYS یافت نشد یا خالی است!' +
+    ' لطفاً کلیدهای خود را به صورت جدا شده با کاما در این Secret در تنظیمات Space خود اضافه کنید.'
   );
-  process.exit(1); // برنامه را متوقف کن
+  process.exit(1);
 } else {
-  console.log(`تعداد ${numKeys} کلید API جیمینای بارگذاری شد.`);
+  console.log(`✅ تعداد ${numKeys} کلید API جیمینای با موفقیت بارگذاری شد.`);
 }
 
-// ۲. شمارنده سراسری برای انتخاب کلید بعدی (شروع از اندیس ۰)
+// ۲. شمارنده سراسری برای توزیع بار بین کاربران (انتخاب کلید شروع)
 let currentKeyIndex = 0;
 
-// ۳. تابع برای گرفتن کلید بعدی به صورت چرخشی
-function getNextApiKey() {
-  if (numKeys === 0) {
-    console.error('تلاش برای گرفتن کلید API در حالی که هیچ کلیدی بارگذاری نشده است.');
-    return null; // یا یک خطا پرتاب کنید
-  }
-  // محاسبه اندیس کلید با استفاده از باقیمانده تقسیم
-  const keyIndexToUse = currentKeyIndex % numKeys;
-  const selectedKey = apiKeys[keyIndexToUse];
-  console.log(`اختصاص کلید API با اندیس: ${keyIndexToUse}`); // لاگ برای اشکال‌زدایی
-
-  // افزایش شمارنده برای درخواست بعدی
+function getStartingKeyIndex() {
+  const startingIndex = currentKeyIndex % numKeys;
   currentKeyIndex++;
-
-  // اختیاری: جلوگیری از خیلی بزرگ شدن شمارنده (گرچه جاوااسکریپت اعداد بزرگ را مدیریت می‌کند)
-  // if (currentKeyIndex >= Number.MAX_SAFE_INTEGER - 10) { // نزدیک به حداکثر عدد امن
-  //     currentKeyIndex = currentKeyIndex % numKeys;
-  // }
-
-  return selectedKey;
+  return startingIndex;
 }
 
-// --- END: منطق جدید برای چرخش API Key ---
+// --- END: منطق جدید مدیریت API Key ---
 
-// سرو کردن فایل‌های استاتیک از پوشه بیلد React
-app.use(express.static(path.join(__dirname, '../build')));
 
-// این تابع حالا کلید API انتخاب شده را به عنوان ورودی می‌گیرد
-const createGeminiWebSocket = (clientWs, apiKey) => {
-  // بررسی اینکه آیا کلید معتبری داده شده است
-  if (!apiKey) {
-    console.error('امکان ایجاد WebSocket جیمینای وجود ندارد: کلید API ارائه نشده است.');
-    // می‌توانید یک پیام خطا به کلاینت بفرستید و اتصالش را ببندید
-    if (clientWs.readyState === WebSocket.OPEN) {
+// این تابع رویدادهای لازم را به یک اتصال موفق جیمینای متصل می‌کند
+function attachGeminiEventHandlers(clientWs, geminiWs, apiKeyUsed) {
+    // ارسال پیام از جیمینای به کلاینت
+    geminiWs.on('message', (data) => {
         try {
-             clientWs.send(JSON.stringify({ error: "خطای داخلی سرور: عدم امکان دریافت کلید API." }));
-        } catch (sendError) {
-             console.error("خطا در ارسال پیام خطا به کلاینت:", sendError);
+            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();
+    });
+}
+
+
+// --- 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();
         }
-        clientWs.close();
+        return null;
     }
-    return null; // برگرداندن null نشان‌دهنده شکست است
-  }
 
-  console.log(`ایجاد اتصال به جیمینای با کلید: ...${apiKey.slice(-4)}`); // نمایش ۴ کاراکتر آخر کلید برای تایید
+    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));
+        });
+    });
+}
 
-  // استفاده از کلید API اختصاص‌یافته به این کاربر
-  const geminiWs = new WebSocket(
-    `wss://generativelanguage.googleapis.com/ws/google.ai.generativelanguage.v1alpha.GenerativeService.BidiGenerateContent?key=${apiKey}`
-  );
+// --- END: تابع هوشمند جدید ---
 
-  // --- بقیه event handler های geminiWs (open, message, error, close) بدون تغییر باقی می‌مانند ---
-  geminiWs.on('open', () => {
-    console.log(`اتصال به جیمینای برای کلاینت با کلید ...${apiKey.slice(-4)} برقرار شد.`);
-    // ارسال پیام setup معلق اگر وجود دارد
-    if (geminiWs.pendingSetup) {
-      console.log('ارسال پیام setup معلق:', geminiWs.pendingSetup);
-      try {
-        geminiWs.send(JSON.stringify(geminiWs.pendingSetup));
-      } catch (sendError) {
-          console.error("خطا در ارسال پیام setup معلق به جیمینای:", sendError);
-          clientWs.close(); // بستن اتصال کلاینت اگر نتوانستیم با جیمینای ارتباط اولیه برقرار کنیم
-      }
-      geminiWs.pendingSetup = null; // پاک کردن پیام معلق
-    }
-  });
-
-  geminiWs.on('message', (data) => {
-    try {
-      // تبدیل پیام به رشته قبل از ارسال به کلاینت (کلاینت شما انتظار Blob داشت، مطمئن شوید هنوز همینطور است)
-      const message = data.toString();
-      console.log(`دریافت از جیمینای (کلید ...${apiKey.slice(-4)}):`, message); // نمایش ۴ کاراکتر آخر کلید در لاگ
-
-      // کلاینت شما انتظار Blob داشت. اگر هنوز نیاز به Blob دارید، آن را حفظ کنید:
-      const blob = Buffer.from(message); // ایجاد Buffer (شبیه Blob در Node.js)
-      if (clientWs.readyState === WebSocket.OPEN) {
-          clientWs.send(blob, { binary: true }); // ارسال به عنوان داده باینری
-      }
-
-      // اگر کلاینت شما حالا انتظار JSON یا رشته دارد، کد بالا را تغییر دهید:
-      // if (clientWs.readyState === WebSocket.OPEN) {
-      //     clientWs.send(message); // ارسال به عنوان رشته
-      // }
-
-    } catch (error) {
-      console.error(`خطا در پردازش پیام جیمینای (کلید ...${apiKey.slice(-4)}):`, error);
-    }
-  });
-
-  geminiWs.on('error', (error) => {
-    console.error(`خطای WebSocket جیمینای (کلید ...${apiKey.slice(-4)}):`, error);
-     // بستن اتصال کلاینت مرتبط در صورت خطای جیمینای
-     if (clientWs.readyState === WebSocket.OPEN) {
-         clientWs.close();
-     }
-  });
-
-  geminiWs.on('close', (code, reason) => {
-    console.log(`اتصال WebSocket جیمینای بسته شد (کلید ...${apiKey.slice(-4)}):`, code, reason.toString());
-     // بستن اتصال کلاینت مرتبط وقتی اتصال جیمینای بسته می‌شود
-     if (clientWs.readyState === WebSocket.OPEN) {
-         clientWs.close();
-     }
-  });
-
-  return geminiWs;
-};
+
+// سرو کردن فایل‌های استاتیک از پوشه build ری‌اکت
+app.use(express.static(path.join(__dirname, '../build')));
 
 wss.on('connection', (ws) => {
-  console.log('یک کلاینت جدید متصل شد.');
-
-  // --- اختصاص کلید API به محض اتصال کلاینت ---
-  const assignedApiKey = getNextApiKey(); // گرفتن کلید بعدی برای *این* کلاینت
-
-  // اگر به دلایلی کلید دریافت نشد (نباید اتفاق بیفتد اگر چک اولیه انجام شده)
-  if (!assignedApiKey) {
-       console.error("خطا: عدم موفقیت در اختصاص کلید API به کلاینت جدید. بستن اتصال.");
-       ws.close();
-       return; // ادامه نده برای این کلاینت
-  }
-  // لاگ کردن کلید اختصاص یافته (فقط ۴ کاراکتر آخر برای امنیت)
-  console.log(`کلید API اختصاص یافته به کلاینت: ...${assignedApiKey.slice(-4)}`);
-
-  let geminiWs = null; // اتصال جیمینای مخصوص *این* کلاینت
-
-  ws.on('message', async (message) => {
-    try {
-        // بررسی کنیم پیام باینری است یا متنی
-        let data;
-        if (message instanceof Buffer) {
-            // اگر کلاینت پیام باینری می‌فرستد، آن را به رشته تبدیل کنید (فرض بر JSON بودن)
-            data = JSON.parse(message.toString());
-        } else {
-            data = JSON.parse(message); // اگر پیام متنی است
-        }
+    // به هر کلاینت یک کلید شروع اختصاص داده می‌شود تا بار توزیع شود
+    const startingKeyIndex = getStartingKeyIndex();
+    console.log(`🔌 یک کلاینت جدید متصل شد. کلید شروع اختصاص ��افته: اندیس ${startingKeyIndex} (کلید ...${apiKeys[startingKeyIndex].slice(-4)})`);
 
-        console.log('دریافت از کلاینت:', data);
-
-        // مقداردهی اولیه اتصال جیمینای برای این کلاینت با استفاده از کلید اختصاص یافته
-        if (data.setup) {
-          // جلوگیری از مقداردهی مجدد اگر کلاینت دوباره پیام setup فرستاد
-          if (geminiWs) {
-              console.warn("کلاینت دوباره پیام setup ارسال کرد. نادیده گرفته شد.");
-              return;
-          }
-          console.log('مقداردهی اولیه اتصال جیمینای با تنظیمات:', data.setup);
-          // ارسال کلید اختصاص یافته این کلاینت به تابع ایجاد اتصال
-          geminiWs = createGeminiWebSocket(ws, assignedApiKey);
-
-          // ذخیره پیام setup برای ارسال پس از برقراری اتصال (اگر اتصال هنوز آماده نیست)
-          if (geminiWs && geminiWs.readyState !== WebSocket.OPEN) {
-             // مهم: پیام معلق را به نمونه WebSocket *این* کلاینت متصل کنید
-             geminiWs.pendingSetup = data;
-          } else if (geminiWs) {
-              // اگر اتصال بلافاصله برقرار شد، پیام setup را بفرست
-              try {
-                  geminiWs.send(JSON.stringify(data));
-              } catch (sendError) {
-                   console.error("خطا در ارسال پیام setup اولیه به جیمینای:", sendError);
-                   ws.close(); // بستن اتصال کلاینت
-              }
-          } else {
-              // اگر createGeminiWebSocket ناموفق بود (مثلا به خاطر خطای کلید در بالا)
-              console.error("ایجاد اتصال WebSocket جیمینای پس از پیام setup ناموفق بود.");
-              // ws قبلا باید بسته شده باشد توسط createGeminiWebSocket
-          }
-          return; // پردازش پیام setup تمام شد
-        }
+    let geminiWs = null; // اتصال جیمینای برای این کلاینت
 
-        // ارسال پیام به جیمینای اگر اتصال برای این کلاینت وجود دارد و باز است
-        if (geminiWs && geminiWs.readyState === WebSocket.OPEN) {
-          console.log('ارسال به جیمینای:', data);
-          try {
-            geminiWs.send(JSON.stringify(data));
-          } catch (sendError) {
-              console.error("خطا در ارسال پیام به جیمینای:", sendError);
-              ws.close(); // بستن اتصال کلاینت
-          }
-        } else if (geminiWs) {
-          // اتصال جیمینای هنوز برقرار نشده یا بسته شده
-          console.log('اتصال جیمینای آماده نیست، امکان ارسال پیام وجود ندارد.');
-          // TODO: می‌توانید پیام‌ها را در صف قرار دهید یا خطا به کلاینت برگردانید
-        } else {
-          // هنوز پیام setup دریافت نشده یا اتصال ناموفق بوده
-          console.error('امکان ارسال پیام وجود ندارد: اتصال جیمینای برای این کلاینت برقرار نشده است.');
-          // می‌توانید خطا به کلاینت برگردانید
-           if (ws.readyState === WebSocket.OPEN) {
-               try {
-                   ws.send(JSON.stringify({ error: "ارتباط با سرویس برقرار نیست. لطفاً دوباره تلاش کنید." }));
-               } catch (sendError) {
-                   console.error("خطا در ارسال پیام خطا به کلاینت:", sendError);
-               }
-           }
+    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);
         }
-    } catch (error) {
-        // خطای کلی در پردازش پیام (مثلا JSON نامعتبر)
-        console.error('خطا در پردازش پیام کلاینت:', error);
-        // از کرش کردن سرور جلوگیری کنید
-    }
-  });
+    });
 
-  ws.on('close', () => {
-    console.log(`کلاینت با کلید ...${assignedApiKey ? assignedApiKey.slice(-4) : 'N/A'} قطع شد.`);
-    // بستن اتصال جیمینای مربوط به این کلاینت، اگر باز است
-    if (geminiWs && geminiWs.readyState !== WebSocket.CLOSED && geminiWs.readyState !== WebSocket.CLOSING) {
-      geminiWs.close();
-    }
-  });
-
-  ws.on('error', (error) => {
-      console.error(`خطای WebSocket کلاینت (کلید ...${assignedApiKey ? assignedApiKey.slice(-4) : 'N/A'}):`, error);
-      // بستن اتصال جیمینای مربوطه در صورت خطای کلاینت
-      if (geminiWs && geminiWs.readyState !== WebSocket.CLOSED && geminiWs.readyState !== WebSocket.CLOSING) {
-          geminiWs.close();
-      }
-      // ws احتمالا به طور خودکار بسته می‌شود یا در شرف بسته شدن است
-  });
+    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();
+        }
+    });
 });
 
 // رسیدگی به درخواست‌های باقیمانده با برگرداندن اپ React
 app.get('*', (req, res) => {
-  // اطمینان حاصل کنید که مسیر درست است
-  const indexPath = path.join(__dirname, '../build', 'index.html');
-  res.sendFile(indexPath, (err) => {
-      if (err) {
-          console.error("خطا در ارسال فایل index.html:", err);
-          // ارسال یک پاسخ خطای ساده اگر فایل پیدا نشد
-          if (!res.headersSent) {
-              res.status(err.status || 500).send('خطا در بارگذاری برنامه');
-          }
-      }
-  });
+    const indexPath = path.join(__dirname, '../build', 'index.html');
+    res.sendFile(indexPath, (err) => {
+        if (err) {
+            if (!res.headersSent) {
+                res.status(500).send('خطا در بارگذاری برنامه');
+            }
+        }
+    });
 });
 
-
-const PORT = process.env.PORT || 3001; // در هاگینگ فیس معمولا پورت 7860 استفاده می‌شود، اما اینجا 3001 است
+const PORT = process.env.PORT || 3001;
 server.listen(PORT, () => {
-  console.log(`سرور در حال اجرا روی پورت ${PORT}`);
-  console.log(`لطفاً ${numKeys} کلید API با نام‌های GEMINI_API_KEY_1 تا GEMINI_API_KEY_${numKeys} را در Secrets تنظیم کرده باشید.`);
+    console.log(`🚀 سرور در حال اجرا روی پورت ${PORT}`);
+    if (numKeys > 0) {
+        console.log(`🔑 برای استفاده، Secret با نام 'ALL_GEMINI_API_KEYS' را با کلید(های) خود تنظیم کنید.`);
+    }
 });