Update app.py

app.py

@@ -7,30 +7,31 @@ import select
 import subprocess
 import fcntl
 import signal
+import asyncio
+import time
 
 app = FastAPI()
 
-# Global Session State
-SESSION = {"fd": None, "process": None}
+# Global Session
+SESSION = {"fd": None, "process": None, "last_activity": time.time()}
 
 class InputData(BaseModel):
     data: str
 
 @app.post("/start")
 def start_terminal():
-    """Starts the root shell session"""
     # Kill existing
     if SESSION["process"]:
         try:
             os.kill(SESSION["process"].pid, signal.SIGTERM)
         except: pass
 
-    # Create PTY
     master, slave = pty.openpty()
     
-    # Start bash as root (sudo)
+    # FIX 1: Add '-i' for Interactive mode (Fixes "no job control")
+    # FIX 2: Use setsid to detach correctly
     p = subprocess.Popen(
-        ["sudo", "bash"],
+        ["sudo", "bash", "-i"],
         stdin=slave,
         stdout=slave,
         stderr=slave,
@@ -39,41 +40,49 @@ def start_terminal():
     )
     os.close(slave)
 
-    # Set non-blocking
+    # Non-blocking read
     fl = fcntl.fcntl(master, fcntl.F_GETFL)
     fcntl.fcntl(master, fcntl.F_SETFL, fl | os.O_NONBLOCK)
 
     SESSION["fd"] = master
     SESSION["process"] = p
+    SESSION["last_activity"] = time.time()
     return {"status": "started"}
 
-def stream_generator():
-    """Yields data instantly as it appears in the terminal"""
+async def stream_generator():
+    """Yields data. Sends a 'ping' if idle to prevent disconnects."""
     fd = SESSION["fd"]
     while True:
         if fd:
-            # Wait for data (timeout 0.1s to prevent CPU 100%)
+            # Check for data
             r, _, _ = select.select([fd], [], [], 0.1)
             if fd in r:
                 try:
-                    # Read chunk
-                    data = os.read(fd, 1024)
+                    data = os.read(fd, 4096)
                     if data:
+                        SESSION["last_activity"] = time.time()
                         yield data
                 except OSError:
                     break
+            else:
+                # FIX 3: Heartbeat
+                # If no data for 10 seconds, yield a null byte to keep HTTP open
+                # The client will ignore this byte.
+                if time.time() - SESSION["last_activity"] > 10:
+                    yield b'\x00' # Null byte (Heartbeat)
+                    SESSION["last_activity"] = time.time()
+                
+                await asyncio.sleep(0.01)
         else:
-            # If no session, wait a bit and retry
-            select.select([], [], [], 0.5)
+            await asyncio.sleep(1)
 
 @app.get("/stream")
 def stream_output():
-    """The magic endpoint: Keeps connection open and streams text"""
-    return StreamingResponse(stream_generator(), media_type="text/plain")
+    return StreamingResponse(stream_generator(), media_type="application/octet-stream")
 
 @app.post("/write")
 def write_input(payload: InputData):
-    """Sends keystrokes to the terminal"""
     if SESSION["fd"]:
+        # Write data exactly as received
         os.write(SESSION["fd"], payload.data.encode())
     return {"status": "ok"}