Update app.py

app.py

@@ -72,6 +72,94 @@ def _detect_browser() -> str:
 BROWSER = _detect_browser()
 print(f"🌐 Browser: {BROWSER}")
 
+# ════════════════════════════════════════════════════════════════
+# ── Firefox Profile (per-user, isolated, crash-recovery DISABLED) ─
+# ════════════════════════════════════════════════════════════════
+# المشكلة الجذرية التي هذا القسم يحلّها:
+# عند قتل Firefox (حتى بـ pkill "لطيف")، لا يُغلق بشكل نظيف دائماً على
+# سيرفر بموارد محدودة/Xvfb، فتُسجَّل الجلسة على أنها "crashed" داخل
+# sessionstore.jsonlz4 الخاص بالبروفايل. في المرة التالية يفتح Firefox
+# ويعرض شاشة "Sorry. We're having trouble getting your pages back" بدل
+# الصفحة المطلوبة — وهذه هي الشاشة السوداء/الفاشلة التي تصل كلقطة "ناجحة"
+# تقنياً (ليست سوداء فعلاً) لكنها لا تحتوي على المحتوى المطلوب أبداً.
+# الحل الجذري: بروفايل خاص بكل مستخدم (isolated) مع user.js يعطّل تماماً:
+#   - استرجاع الجلسة بعد التحطّم (session restore prompt)
+#   - أي محاولة لإعادة فتح تبويبات سابقة
+# بهذا، كل فتح لفايرفوكس يبدأ صفحة نظيفة فارغة دائماً، بغض النظر عن كيفية
+# إغلاقه سابقاً.
+
+FIREFOX_PROFILE_PREFS = """
+// ── تعطيل استرجاع الجلسة نهائياً (سبب شاشة "Restore Session") ──
+user_pref("browser.sessionstore.resume_from_crash", false);
+user_pref("browser.sessionstore.resume_session_once", false);
+user_pref("browser.sessionstore.max_resumed_crashes", -1);
+user_pref("browser.sessionstore.restore_on_demand", false);
+user_pref("browser.sessionstore.enabled", false);
+user_pref("browser.sessionstore.privacy_level", 2);
+user_pref("browser.startup.page", 0);
+user_pref("browser.startup.homepage_override.mstone", "ignore");
+user_pref("toolkit.startup.max_resumed_crashes", -1);
+// تعطيل نافذة "Restore previous session" ونوافذ الأزمات المختلفة
+user_pref("browser.sessionstore.max_tabs_undo", 0);
+user_pref("browser.sessionstore.max_windows_undo", 0);
+user_pref("browser.tabs.crashReporting.sendReport", false);
+user_pref("browser.crashReports.unsubmittedCheck.autoSubmit2", false);
+user_pref("browser.crashReports.unsubmittedCheck.enabled", false);
+user_pref("toolkit.crashreporter.infoURL", "");
+// تعطيل شاشة "Restore" وكذلك أي إشعار تحطّم
+user_pref("browser.sessionstore.resumeFromCrash", false);
+user_pref("dom.ipc.plugins.flash.subprocess.crashreporter.enabled", false);
+// إلغاء استعادة آخر جلسة تماماً + دائماً about:blank
+user_pref("browser.startup.homepage", "about:blank");
+// تعطيل تحديثات ورسائل onboarding التي قد تبطئ أو تحجب اللقطة
+user_pref("browser.shell.checkDefaultBrowser", false);
+user_pref("browser.aboutwelcome.enabled", false);
+user_pref("browser.startup.firstrunSkipsHomepage", true);
+user_pref("startup.homepage_welcome_url", "");
+user_pref("startup.homepage_welcome_url.additional", "");
+user_pref("browser.uitour.enabled", false);
+user_pref("browser.newtabpage.activity-stream.feeds.telemetry", false);
+user_pref("datareporting.policy.dataSubmissionEnabled", false);
+user_pref("app.normandy.enabled", false);
+user_pref("app.update.enabled", false);
+user_pref("app.update.auto", false);
+"""
+
+def _ensure_firefox_profile(user_id: str, display: str) -> str:
+    """
+    يُنشئ (أو يُعيد استخدام) بروفايل Firefox مخصص ونظيف لهذا المستخدم،
+    مع إعدادات تعطّل استرجاع الجلسة (crash recovery) نهائياً.
+    يُعيد المسار الكامل للبروفايل.
+    """
+    safe_id = re.sub(r"[^a-zA-Z0-9_.-]", "_", user_id) or "anon"
+    profile_dir = os.path.expanduser(f"~/.zai_ff_profiles/{safe_id}")
+    try:
+        os.makedirs(profile_dir, exist_ok=True)
+        prefs_path = os.path.join(profile_dir, "user.js")
+        with open(prefs_path, "w", encoding="utf-8") as f:
+            f.write(FIREFOX_PROFILE_PREFS)
+    except Exception as e:
+        print(f"[profile] ⚠️ failed to prepare profile for {user_id}: {e}")
+    return profile_dir
+
+
+def _wipe_firefox_session_data(profile_dir: str):
+    """
+    يمسح ملفات الجلسة الفاسدة (sessionstore) + أي lock متبقٍّ.
+    هذا ضروري لأن مجرد تعطيل session restore في user.js لا يمسح ملفات
+    جلسة سابقة موجودة بالفعل على القرص من قبل تفعيل هذه الإعدادات.
+    """
+    try:
+        subprocess.run(
+            f"rm -f '{profile_dir}'/lock '{profile_dir}'/.parentlock "
+            f"'{profile_dir}'/sessionstore.jsonlz4 "
+            f"'{profile_dir}'/sessionstore-backups/*.jsonlz4 "
+            f"'{profile_dir}'/sessionCheckpoints.json 2>/dev/null",
+            shell=True, timeout=5, capture_output=True
+        )
+    except Exception:
+        pass
+
 
 # ════════════════════════════════════════════════════════════════
 # ── Xvfb Management (per-display) ────────────────────────────
@@ -160,11 +248,12 @@ def _kill_proc(proc: "subprocess.Popen | None"):
 def _kill_display_processes(display: str):
     """يقتل كل العمليات المرتبطة بـ display معين."""
     env_d = {**os.environ, "DISPLAY": display}
-    # قتل المتصفح
+    # قتل المتصفح — SIGKILL مباشر (-9) لتفادي أي حالة إغلاق نصفي تُسجَّل
+    # لاحقاً كتحطّم (crash) داخل Firefox وتُنتج شاشة "Restore Session"
     for b in ["firefox", "firefox-esr", "chromium", "chrome"]:
         try:
             subprocess.run(
-                ["pkill", "-f", f"[{b[0]}]{b[1:]}.*{display}"],
+                ["pkill", "-9", "-f", f"[{b[0]}]{b[1:]}.*{display}"],
                 timeout=3, capture_output=True
             )
         except Exception:
@@ -230,8 +319,16 @@ async def reset_user_computer(user_id: str) -> dict:
         display = sess["display"]
         print(f"[reset] 🔄 Resetting computer for user '{user_id}' on {display}")
 
-        # قتل المتصفح
-        _kill_proc(sess.get("browser_proc"))
+        # قتل المتصفح — SIGKILL مباشر (بدل terminate اللطيف) لأن الأخير قد
+        # يترك فايرفوكس في حالة نصف-مغلقة تُسجَّل كتحطّم (crash) في المرة
+        # القادمة، وهذا بالضبط ما يُنتج شاشة "Restore Session" لاحقاً.
+        browser_proc = sess.get("browser_proc")
+        if browser_proc:
+            try:
+                browser_proc.kill()
+                browser_proc.wait(timeout=3)
+            except Exception:
+                pass
         sess["browser_proc"] = None
 
         # قتل Xvfb
@@ -241,14 +338,22 @@ async def reset_user_computer(user_id: str) -> dict:
         # تنظيف شامل لهذا الـ display
         _kill_display_processes(display)
 
-        # تنظيف lock files Firefox
+        # ── تنظيف شامل لبيانات جلسة Firefox الفاسدة (السبب الجذري لشاشة
+        # "Restore Session") — ليس فقط lock files بل sessionstore كاملاً،
+        # في كل من البروفايل الافتراضي وأي بروفايل مخصص لهذا المستخدم ──
         try:
             subprocess.run(
-                "rm -f ~/.mozilla/firefox/*/lock ~/.mozilla/firefox/*/.parentlock 2>/dev/null",
+                "rm -f ~/.mozilla/firefox/*/lock ~/.mozilla/firefox/*/.parentlock "
+                "~/.mozilla/firefox/*/sessionstore.jsonlz4 "
+                "~/.mozilla/firefox/*/sessionstore-backups/*.jsonlz4 "
+                "~/.mozilla/firefox/*/sessionCheckpoints.json 2>/dev/null",
                 shell=True, timeout=5, capture_output=True
             )
         except Exception:
             pass
+        safe_id = re.sub(r"[^a-zA-Z0-9_.-]", "_", user_id) or "anon"
+        custom_profile = os.path.expanduser(f"~/.zai_ff_profiles/{safe_id}")
+        _wipe_firefox_session_data(custom_profile)
 
     # تشغيل Xvfb جديد (خارج الـ lock لأن start_xvfb يستغرق وقتاً)
     new_proc = await asyncio.to_thread(_start_xvfb, display)
@@ -1009,12 +1114,18 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
             await send({"type": "ack", "action": "open_app"})
             return
         if "firefox" in cmd.lower():
+            profile_dir = _ensure_firefox_profile(sess["user_id"], display)
+            # قتل فوري وقوي (SIGKILL) بدل pkill اللطيف: يمنع فايرفوكس من
+            # "التقاط" حالة نصف-مفتوحة تُسجَّل لاحقاً كتحطّم (crash) يُنتج
+            # شاشة "Restore Session" في المرة القادمة.
             await run_cmd(
-                "pkill -f '[f]irefox' 2>/dev/null; sleep 0.5; "
-                "rm -f ~/.mozilla/firefox/*/lock ~/.mozilla/firefox/*/.parentlock 2>/dev/null; "
-                "echo CLEANED",
+                "pkill -9 -f '[f]irefox' 2>/dev/null; sleep 0.6; echo CLEANED",
                 display, timeout=8
             )
+            _wipe_firefox_session_data(profile_dir)
+            # حقن --profile داخل أمر فايرفوكس (إن لم يكن محقوناً مسبقاً)
+            if "--profile" not in cmd and "-P " not in cmd:
+                cmd = cmd.replace("firefox", f"firefox --profile '{profile_dir}' --no-remote", 1)
         env = {**os.environ, "DISPLAY": display}
         proc = subprocess.Popen(cmd, shell=True, env=env,
                                 stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
@@ -1033,16 +1144,19 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
     # ── open_browser ─────────────────────────────────────────
     elif action == "open_browser":
         url = _safe_search(data.get("url", "") or "about:blank")
-        # قتل Firefox القديم الخاص بهذا المستخدم فقط
+        profile_dir = _ensure_firefox_profile(sess["user_id"], display)
+        # قتل فوري وقوي (SIGKILL) بدل pkill اللطيف — يمنع تسجيل حالة
+        # "تحطّم" في sessionstore، وهي السبب الجذري لشاشة "Restore Session"
+        # التي تحجب المحتوى الفعلي وتُلتقط كلقطة شاشة صالحة (غير سوداء)
+        # لكنها ليست الصفحة المطلوبة.
         await run_cmd(
-            "pkill -f '[f]irefox' 2>/dev/null; sleep 0.5; "
-            "rm -f ~/.mozilla/firefox/*/lock ~/.mozilla/firefox/*/.parentlock 2>/dev/null; "
-            "echo CLEANED",
+            "pkill -9 -f '[f]irefox' 2>/dev/null; sleep 0.6; echo CLEANED",
             display, timeout=8
         )
+        _wipe_firefox_session_data(profile_dir)
         env = {**os.environ, "DISPLAY": display}
         proc = subprocess.Popen(
-            [BROWSER, "--no-remote", url], env=env,
+            [BROWSER, "--profile", profile_dir, "--no-remote", url], env=env,
             stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL
         )
         with _display_lock: