topnic-https 1.2.4 → 1.2.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "topnic-https",
-  "version": "1.2.4",
+  "version": "1.2.6",
   "main": "./src/main/index.js",
   "type": "commonjs",
   "scripts": {
@@ -91,6 +91,7 @@
     "chokidar": "^4.0.1",
     "commander": "^12.1.0",
     "crypto": "^1.0.1",
+    "dns": "^0.2.2",
     "electron": "^33.2.0",
     "fs": "^0.0.1-security",
     "https": "^1.0.0",

package/src/sharing/directServer.js

@@ -3,75 +3,60 @@ const WebSocket = require('ws');
 const { nanoid } = require('nanoid/non-secure');
 const os = require('os');
 const net = require('net');
+const dns = require('dns');
 
 class DirectServer {
   constructor() {
     this.connections = new Map();
     this.server = http.createServer(this.handleRequest.bind(this));
     this.wss = new WebSocket.Server({ server: this.server });
-    this.publicIp = this.getPublicIp();
+    this.publicIp = null;
   }
 
-  getPublicIp() {
-    const interfaces = os.networkInterfaces();
-    for (const iface of Object.values(interfaces)) {
-      const nonLocalIpv4 = iface.find(addr => addr.family === 'IPv4' && !addr.internal);
-      if (nonLocalIpv4) return nonLocalIpv4.address;
-    }
-    return 'localhost';
-  }
-
-  async isPortAvailable(port) {
+  async getPublicIp() {
     return new Promise((resolve) => {
-      const tester = net.createServer()
-        .once('error', () => resolve(false))
-        .once('listening', () => {
-          tester.once('close', () => resolve(true)).close();
-        })
-        .listen(port);
-    });
-  }
-
-  async start(port = 3500) {
-    if (!(await this.isPortAvailable(port))) {
-      console.log(`المنفذ ${port} مشغول، جاري المحاولة على منفذ آخر...`);
-      port = await this.findAvailablePort(3501);
-    }
+      // أولاً نحاول الحصول على IP المحلي
+      const interfaces = os.networkInterfaces();
+      let localIp = 'localhost';
+      
+      for (const iface of Object.values(interfaces)) {
+        const nonLocalIpv4 = iface.find(addr => 
+          addr.family === 'IPv4' && 
+          !addr.internal &&
+          addr.address !== '127.0.0.1'
+        );
+        if (nonLocalIpv4) {
+          localIp = nonLocalIpv4.address;
+          break;
+        }
+      }
 
-    return new Promise((resolve) => {
-      this.server.listen(port, '0.0.0.0', () => {
-        console.log(`🚀 خادم المشاركة يعمل على المنفذ ${port}`);
-        console.log(`📡 عنوان IP: ${this.publicIp}:${port}`);
-        resolve(port);
+      // نحاول الحصول على IP العام
+      dns.lookup(os.hostname(), (err, address, family) => {
+        if (!err && address && address !== '127.0.0.1') {
+          resolve(address);
+        } else {
+          resolve(localIp);
+        }
       });
     });
   }
 
-  async findAvailablePort(startPort) {
-    let port = startPort;
-    while (!(await this.isPortAvailable(port))) {
-      port++;
-    }
-    return port;
-  }
-
-  async verifyTargetPort(port) {
+  async start(port = 3500) {
     try {
-      const response = await new Promise((resolve, reject) => {
-        const req = http.request({
-          hostname: 'localhost',
-          port: port,
-          path: '/',
-          method: 'GET',
-          timeout: 2000
-        }, resolve);
-        
-        req.on('error', reject);
-        req.end();
+      this.publicIp = await this.getPublicIp();
+      
+      return new Promise((resolve) => {
+        this.server.listen(port, '0.0.0.0', () => {
+          console.log(`🚀 خادم المشاركة يعمل على المنفذ ${port}`);
+          console.log(`📡 عنوان IP: ${this.publicIp}:${port}`);
+          console.log(`🌐 يمكنك الوصول محلياً عبر: http://localhost:${port}`);
+          resolve(port);
+        });
       });
-      return true;
     } catch (error) {
-      return false;
+      console.error('خطأ في بدء الخادم:', error);
+      throw error;
     }
   }
 
@@ -91,21 +76,23 @@ class DirectServer {
       const connectionId = urlParts[1];
       const connection = this.connections.get(connectionId);
 
-      if (!connection) {
-        res.writeHead(404);
-        res.end('رابط المشاركة غير صالح');
+      res.setHeader('Access-Control-Allow-Origin', '*');
+      res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
+      res.setHeader('Access-Control-Allow-Headers', '*');
+
+      if (req.method === 'OPTIONS') {
+        res.writeHead(200);
+        res.end();
         return;
       }
 
-      // التحقق من أن الخدمة المحلية تعمل
-      if (!(await this.verifyTargetPort(connection.port))) {
-        res.writeHead(502);
-        res.end('الخدمة المحلية غير متاحة');
+      if (!connection) {
+        res.writeHead(404);
+        res.end('رابط المشاركة غير صالح');
         return;
       }
 
-      // إعادة بناء المسار الأصلي
-      const originalPath = '/' + urlParts.slice(2).join('/');
+      const originalPath = urlParts.length > 2 ? '/' + urlParts.slice(2).join('/') : '/';
 
       const options = {
         hostname: 'localhost',
@@ -114,19 +101,28 @@ class DirectServer {
         method: req.method,
         headers: {
           ...req.headers,
+          'X-Forwarded-Host': req.headers.host,
+          'X-Forwarded-Proto': 'http',
+          'X-Real-IP': req.socket.remoteAddress,
           host: `localhost:${connection.port}`
         }
       };
 
       const proxyReq = http.request(options, (proxyRes) => {
-        res.writeHead(proxyRes.statusCode, proxyRes.headers);
+        Object.keys(proxyRes.headers).forEach(key => {
+          res.setHeader(key, proxyRes.headers[key]);
+        });
+        
+        res.writeHead(proxyRes.statusCode);
         proxyRes.pipe(res);
       });
 
       proxyReq.on('error', (error) => {
         console.error('خطأ في التوجيه:', error);
-        res.writeHead(502);
-        res.end('خطأ في الاتصال بالتطبيق المحلي');
+        if (!res.headersSent) {
+          res.writeHead(502);
+          res.end('خطأ في الاتصال بالتطبيق المحلي');
+        }
       });
 
       if (!req.readableEnded) {
@@ -138,8 +134,10 @@ class DirectServer {
       connection.visitors++;
     } catch (error) {
       console.error('خطأ في معالجة الطلب:', error);
-      res.writeHead(500);
-      res.end('خطأ داخلي في الخادم');
+      if (!res.headersSent) {
+        res.writeHead(500);
+        res.end('خطأ داخلي في الخادم');
+      }
     }
   }