Update app.py

app.py

@@ -3,136 +3,116 @@ import os
 from flask import Flask, request, jsonify
 
 app = Flask(__name__)
-# ──────────────────────────────────────────────────────────────────────────────
-# محرك اتصال خارجي: Direct أولاً، ثم Relay (WebSocket) كخطة B
-# ألصقه بعد الاستيرادات وقبل بدء خدامك (RPC/Flask)
-# ──────────────────────────────────────────────────────────────────────────────
-import os, time, threading, socket, random, requests, importlib
-MODE = None                 # "direct" أو "relay"
+# ─────────────────────────────────────────────────────────────────────
+# محرك اتصال: سيرفر عشوائي من peer_discovery + منافذ 0001..9999 بالتسلسل
+# يتوقف فقط عند أول تسجيل ناجح
+# ─────────────────────────────────────────────────────────────────────
+import os, time, threading, socket, requests, importlib, random, sys
+
 CURRENT_SERVER = None
-PORT = None                 # منفذك المحلي لخدمة /run أو RPC
+PORT = None
 CONNECTED = threading.Event()
-RELAY_SIO = None
 
-# ← جِب المرشحين من peer_discovery في كل دورة (حتى لو بيولدهم ديناميكياً)
-def _load_candidates():
+def _load_peer_discovery():
     import peer_discovery as pd
     importlib.reload(pd)
+    return pd
+
+def _pick_random_server(pd):
+    # إن كان لديك pd.random_server() استخدمه، وإلا اختر عشوائياً من القائمة
+    if hasattr(pd, "random_server"):
+        return pd.random_server()
     servers = list(getattr(pd, "CENTRAL_REGISTRY_SERVERS", []))
-    # منفذك الذي ستشغل عليه الـ RPC محليًا (وليس منفذ السيرفر العام)
-    # يفضل أن يكون ما يولده peer_discovery؛ وإن لم يوجد استخدم 7860/8000/5000 احتياط
-    ports = list(getattr(pd, "rport", [])) or ["7860", "8000", "5000"]
-    get_ip = getattr(pd, "get_local_ip", lambda: "127.0.0.1")
-    random.shuffle(servers); random.shuffle(ports)
-    return servers, [int(p) for p in ports], get_ip
-
-def _is_port_free(p, host="0.0.0.0"):
+    if not servers:
+        raise RuntimeError("لا توجد سيرفرات في peer_discovery.CENTRAL_REGISTRY_SERVERS")
+    random.shuffle(servers)
+    return servers[0]
+
+def _get_local_ip(pd):
+    if hasattr(pd, "get_local_ip"):
+        return pd.get_local_ip()
+    # احتياط
+    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    try:
+        s.connect(("8.8.8.8", 80))
+        return s.getsockname()[0]
+    except Exception:
+        return "127.0.0.1"
+    finally:
+        s.close()
+
+def _lan_port(pd):
+    return int(getattr(pd, "LAN_PORT", os.getenv("LAN_PORT", 7520)))
+
+def _can_bind_low_ports():
+    # على لينكس: تحتاج root للمنافذ <1024
+    if os.name != "nt":
+        try:
+            return os.geteuid() == 0  # type: ignore[attr-defined]
+        except Exception:
+            return False
+    return True  # ويندوز عادة يسمح
+
+def _is_port_free(port, host="0.0.0.0"):
     with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
         s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
         try:
-            s.bind((host, int(p)))
+            s.bind((host, int(port)))
             return True
         except OSError:
             return False
 
-def _try_direct_register(server, port, get_ip):
-    """
-    يحاول التسجيل المباشر عند السيرفر: POST /register
-    إذا نجح، نستخدم هذا المنفذ لتشغيل خادمنا المحلي لاحقًا.
-    """
-    info = {
-        "node_id": os.getenv("NODE_ID", socket.gethostname()),
-        "ip": get_ip(),
-        "port": int(port),
-    }
-    r = requests.post(f"{server}/register", json=info, timeout=6)
-    r.raise_for_status()  # يرمي استثناء لو فشل
-    return True
-
-def _start_relay_client(server):
-    """
-    وضع Relay: قناة Socket.IO دائمة مع السيرفر (على 443 غالبًا).
-    لا تحتاج أي منفذ inbound. إذا السيرفر يدعم relay، يُعتبر اتصال خارجي ناجح.
-    """
-    global RELAY_SIO
-    import socketio  # python-socketio client
-    sio = socketio.Client(reconnection=True, reconnection_attempts=0)
-
-    node_id = os.getenv("NODE_ID", socket.gethostname())
-
-    @sio.event
-    def connect():
-        print(f"🔌 Relay connected to {server}")
-        sio.emit("register_node", {"node_id": node_id})
-        # اعتبر الاتصال ناجح خارجيًا
-        # نُبقي PORT لاستخدام LAN/RPC محلي (لو احتجته). تستطيع ضبطه لاحقًا.
-    
-    @sio.event
-    def disconnect():
-        print("⚠️ Relay disconnected; will auto-reconnect…")
-
-    # أمثلة أحداث (عدّل على مزاجك):
-    @sio.on("ping_node")
-    def _on_ping(msg):
-        sio.emit("pong_node", {"node_id": node_id, "t": time.time()})
+def _try_register(server, port, node_id, ip):
+    payload = {"node_id": node_id, "ip": ip, "port": int(port)}
+    r = requests.post(f"{server}/register", json=payload, timeout=6)
+    r.raise_for_status()
 
-    try:
-        # Socket.IO URL ← عادة نفس الدومين على /socket.io
-        sio.connect(server, transports=["websocket", "polling"], wait_timeout=10)
-        RELAY_SIO = sio
-        return True
-    except Exception as e:
-        print(f"❌ Relay connect failed: {e}")
-        return False
+def _sequential_ports(pd):
+    """يولّد 0001..9999 مع مراعاة الـ LAN_PORT وتخطي <1024 إذا غير مسموح."""
+    lan_p = _lan_port(pd)
+    start = 1 if _can_bind_low_ports() else 1025
+    for p in range(start, 10000):
+        if p == lan_p:
+            continue
+        yield p
 
 def _connect_until_success():
-    global MODE, CURRENT_SERVER, PORT
+    global CURRENT_SERVER, PORT
     backoff = 1
     while True:
-        servers, ports, get_ip = _load_candidates()
-
-        # 1) جرّب DIRECT على كل السيرفرات مع مجموعة منافذك
-        for s in servers:
-            for p in ports:
-                # تأكد أن المنفذ متاح محليًا لتشغيل خادمك لاحقًا
-                if not _is_port_free(p):
-                    continue
-                try:
-                    _try_direct_register(s, p, get_ip)
-                    MODE, CURRENT_SERVER, PORT = "direct", s, int(p)
-                    print(f"✅ DIRECT connected: {s} (node port {PORT})")
-                    CONNECTED.set()
-                    return
-                except Exception as e:
-                    print(f"⚠️ DIRECT failed on {s} / {p}: {e}")
-
-        # 2) إذا فشلت المباشرة على الكل → جرّب RELAY لكل سيرفر بالترتيب
-        for s in servers:
-            ok = _start_relay_client(s)
-            if ok:
-                MODE, CURRENT_SERVER = "relay", s
-                # PORT اختياري هنا (لـ LAN فقط). لو عندك RPC محلّي شغّله على أي منفذ متاح.
-                if ports:
-                    # اختر أول منفذ متاح محليًا للتشغيل الداخلي
-                    for p in ports:
-                        if _is_port_free(p):
-                            PORT = int(p)
-                            break
-                print(f"✅ RELAY connected via {s} (no inbound port needed)")
+        pd = _load_peer_discovery()
+        server = _pick_random_server(pd)
+        node_id = os.getenv("NODE_ID", socket.gethostname())
+        ip = _get_local_ip(pd)
+        print(f"🔎 Trying server: {server}")
+
+        any_attempt = False
+        for p in _sequential_ports(pd):
+            any_attempt = True
+            if not _is_port_free(p):
+                continue
+            try:
+                _try_register(server, p, node_id, ip)
+                CURRENT_SERVER, PORT = server, int(p)
                 CONNECTED.set()
+                print(f"✅ Connected to {server} with port {PORT:04d}")
                 return
+            except Exception as e:
+                # اطبع على فترات لتجنب الضوضاء
+                if p % 1000 == 0:
+                    print(f"…still scanning (last error on {server}:{p:04d}) -> {e}")
 
-        # لا نجاح بعد التجربتين → انتظر وزِد المهلة وأعد المحاولة
+        # لم ينجح أي منفذ مع هذا السيرفر → جرّب سيرفر جديد
+        if not any_attempt:
+            print("⚠️ لم تُجرَ محاولات منافذ (تحقق من صلاحيات المنافذ المنخفضة).")
+
+        print(f"↻ No success on {server}. Picking a new server…")
         time.sleep(backoff)
-        backoff = min(backoff * 2, 30)
+        backoff = min(backoff * 2, 10)  # backoff بسيط لتخفيف الضغط
 
 def start_connect_loop():
     threading.Thread(target=_connect_until_success, daemon=True).start()
 
-# شغّل حلقة الاتصال مبكرًا:
-start_connect_loop()
-# ──────────────────────────────────────────────────────────────────────────────
-
 # مسار اختبار سريع
 @app.get("/")
 def hello():