Update server/Server.js

server/Server.js

@@ -2,59 +2,47 @@ const express = require('express');
 const path = require('node:path');
 const { WebSocketServer, WebSocket } = require('ws');
 const http = require('node:http');
-require('dotenv').config(); // برای خواندن GEMINI_API_KEY از .env
+require('dotenv').config();
 
 const app = express();
 const server = http.createServer(app);
 
-// --- 👇 دامنه مجاز WebSocket (باید با ALLOWED_ORIGIN در App.tsx یکی باشد) 👇 ---
 const ALLOWED_WEBSOCKET_ORIGIN = "https://aisada.ir"; // یا "http://aisada.ir"
-// --- 👆 ---
 
-// --- 👇 استفاده از verifyClient برای بررسی Origin قبل از نهایی کردن اتصال 👇 ---
 const wss = new WebSocketServer({
-  server, // سرور HTTP که به آن متصل می‌شود
+  server,
   verifyClient: (info, cb) => {
-    // info.origin حاوی مقدار هدر Origin فرستاده شده توسط مرورگر است
     const origin = info.origin;
-    console.log(`[WebSocket Verify] Verifying origin: ${origin}`); // لاگ برای دیباگ
-
+    // --- 👇 لاگ Verify را نگه می‌داریم چون مهم است 👇 ---
+    console.log(`[WebSocket Verify] Verifying origin: ${origin}`);
     if (origin === ALLOWED_WEBSOCKET_ORIGIN) {
       console.log(`[WebSocket Verify] Origin allowed: ${origin}`);
-      cb(true); // اتصال مجاز است، ادامه بده
+      cb(true);
     } else {
-      // اگر origin تعریف نشده بود (مثلاً یک کلاینت غیرمرورگری) یا مطابقت نداشت
       console.warn(`[WebSocket Verify] Forbidden origin: ${origin}. Rejecting connection.`);
-      // cb(accept, code, message)
-      cb(false, 403, 'Forbidden Origin'); // اتصال را با کد 403 رد کن
+      cb(false, 403, 'Forbidden Origin');
     }
   }
 });
-// --- 👆 ---
 
-// Serve static files from the React app build directory
-// این مسیر باید درست باشد نسبت به محل اجرای server.js
 app.use(express.static(path.join(__dirname, '../build')));
 
-// خواندن API Key از متغیرهای محیطی
 const GEMINI_API_KEY = process.env.GEMINI_API_KEY;
-
 if (!GEMINI_API_KEY) {
   console.error('FATAL ERROR: GEMINI_API_KEY environment variable is not set!');
-  process.exit(1); // خروج از برنامه اگر کلید وجود ندارد
+  process.exit(1);
 }
 
-// تابع برای ایجاد اتصال WebSocket به Gemini (بدون تغییر نسبت به قبل)
 const createGeminiWebSocket = (clientWs) => {
-   const geminiWsUrl = `wss://generativelanguage.googleapis.com/v1beta/models/gemini-pro:streamGenerateContent?key=${GEMINI_API_KEY}&alt=sse`; // URL ممکن است نیاز به آپدیت داشته باشد
-   // یا URL قبلی: `wss://generativelanguage.googleapis.com/ws/google.ai.generativelanguage.v1alpha.GenerativeService.BidiGenerateContent?key=${GEMINI_API_KEY}`
-   console.log(`[Gemini Connect] Attempting connection to Gemini: ${geminiWsUrl.split('key=')[0]}...`); // عدم لاگ کلید
-   const geminiWs = new WebSocket(geminiWsUrl); // از URL صحیح استفاده کنید
+   const geminiWsUrl = `wss://generativelanguage.googleapis.com/v1beta/models/gemini-pro:streamGenerateContent?key=${GEMINI_API_KEY}&alt=sse`;
+   console.log(`[Gemini Connect] Attempting connection to Gemini...`);
+   const geminiWs = new WebSocket(geminiWsUrl);
 
    geminiWs.on('open', () => {
      console.log('[Gemini Connect] Connected to Gemini API');
      if (geminiWs.pendingSetup) {
-       console.log('[Gemini Connect] Sending pending setup:', geminiWs.pendingSetup);
+       // لاگ pendingSetup ممکن است حجیم باشد، خلاصه‌تر می‌کنیم
+       console.log('[Gemini Connect] Sending pending setup...');
        geminiWs.send(JSON.stringify(geminiWs.pendingSetup));
        geminiWs.pendingSetup = null;
      }
@@ -62,14 +50,12 @@ const createGeminiWebSocket = (clientWs) => {
 
    geminiWs.on('message', (data) => {
      try {
-       console.log('[Gemini Receive] Received from Gemini'); // حذف لاگ کامل محتوا برای امنیت/حجم
-       // ارسال پیام به کلاینت اصلی
+       // --- 👇 لاگ دریافت از Gemini حذف شد 👇 ---
+       // console.log('[Gemini Receive] Received from Gemini');
        if (clientWs.readyState === WebSocket.OPEN) {
-            // اگر بافر بود، مستقیم بفرست
             if (Buffer.isBuffer(data)) {
                  clientWs.send(data, { binary: true });
             } else {
-                 // اگر متن بود، تبدیل به بافر و ارسال (مطابق کد قبلی)
                   const blob = Buffer.from(data.toString());
                   clientWs.send(blob, { binary: true });
             }
@@ -85,9 +71,7 @@ const createGeminiWebSocket = (clientWs) => {
    return geminiWs;
 };
 
-// رویداد connection حالا فقط برای درخواست‌های مجاز اجرا می‌شود
 wss.on('connection', (ws, req) => {
-  // req.headers.origin را برای لاگ استفاده می‌کنیم
   const origin = req.headers.origin || 'N/A';
   console.log(`[WebSocket Event] Client connected (Origin: ${origin})`);
   let geminiWs = null;
@@ -96,12 +80,13 @@ wss.on('connection', (ws, req) => {
     try {
        const messageString = message.toString();
        const data = JSON.parse(messageString);
-       console.log('[Client Receive] Received from client'); // حذف لاگ کامل محتوا
+       // --- 👇 لاگ دریافت از Client حذف شد 👇 ---
+       // console.log('[Client Receive] Received from client');
 
       if (data.setup) {
+        // لاگ setup را نگه می‌داریم چون مهم است
         console.log('[Client Receive] Initializing Gemini connection...');
         geminiWs = createGeminiWebSocket(ws);
-        // مدیریت پیام setup که قبل از باز شدن اتصال gemini می‌رسد
         if (geminiWs.readyState !== WebSocket.OPEN) {
           geminiWs.pendingSetup = data;
         } else {
@@ -110,19 +95,17 @@ wss.on('connection', (ws, req) => {
         return;
       }
 
-      // ارسال سایر پیام‌ها به Gemini
       if (geminiWs && geminiWs.readyState === WebSocket.OPEN) {
         geminiWs.send(JSON.stringify(data));
       } else if (geminiWs) {
+        // لاگ انتظار را نگه می‌داریم
         console.log('[Client Receive] Gemini connection not ready, waiting...');
-        // اینجا می‌توان پیام را در صف گذاشت یا خطا داد
       } else {
         console.error('[Client Receive] No Gemini connection established. Closing client connection.');
-        ws.close(1011, 'Backend connection not ready'); // بستن اتصال کلاینت
+        ws.close(1011, 'Backend connection not ready');
       }
     } catch (error) {
-      console.error('[Client Receive] Error processing message:', error, message.toString().substring(0, 100) + '...'); // لاگ خلاصه پیام خطا
-      // بستن اتصال در صورت خطای parse یا پردازش ناموفق
+      console.error('[Client Receive] Error processing message:', error, message.toString().substring(0, 100) + '...');
       ws.close(1008, 'Message processing error');
     }
   });
@@ -133,18 +116,16 @@ wss.on('connection', (ws, req) => {
       console.log('[WebSocket Event] Closing associated Gemini connection.');
       geminiWs.close();
     }
-    geminiWs = null; // پاک کردن رفرنس
+    geminiWs = null;
   });
 
   ws.on('error', (error) => {
      console.error('[WebSocket Event] Client WebSocket error:', error);
-     // بستن اتصال جمینی در صورت خطای کلاینت
      if (geminiWs && geminiWs.readyState !== WebSocket.CLOSED && geminiWs.readyState !== WebSocket.CLOSING) {
         console.log('[WebSocket Event] Closing associated Gemini connection due to client error.');
         geminiWs.close();
      }
      geminiWs = null;
-     // خود ws معمولا به طور خودکار بسته می‌شود، اما برای اطمینان:
      if (ws.readyState !== WebSocket.CLOSED && ws.readyState !== WebSocket.CLOSING) {
          ws.close(1011, 'Client error');
      }
@@ -152,14 +133,12 @@ wss.on('connection', (ws, req) => {
 
 });
 
-// سرو کردن فایل‌های استاتیک React (مسیر build)
 app.get('*', (req, res) => {
-  res.sendFile(path.resolve(__dirname, '../build', 'index.html')); // استفاده از path.resolve برای اطمینان
+  res.sendFile(path.resolve(__dirname, '../build', 'index.html'));
 });
 
-// اجرای سرور
 const PORT = process.env.PORT || 3001;
 server.listen(PORT, () => {
   console.log(`Server listening on port ${PORT}`);
-  console.log(`Allowed WebSocket Origin: ${ALLOWED_WEBSOCKET_ORIGIN}`); // نمایش دامنه مجاز در لاگ
+  console.log(`Allowed WebSocket Origin: ${ALLOWED_WEBSOCKET_ORIGIN}`);
 });