Update app.py

app.py

@@ -1,8 +1,54 @@
-from fastapi import FastAPI
+import os
+import asyncio
+import pty
+import subprocess
+from fastapi import FastAPI, WebSocket, WebSocketDisconnect
+from fastapi.responses import HTMLResponse
+from fastapi.middleware.cors import CORSMiddleware
 
 app = FastAPI()
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+# Permitir CORS para qualquer origem
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@app.websocket("/ws")
+async def websocket_endpoint(websocket: WebSocket):
+    await websocket.accept()
+
+    # Cria um pseudo-terminal com bash
+    pid, fd = pty.fork()
+
+    if pid == 0:
+        # Processo filho: executa o shell
+        shell = os.environ.get("SHELL", "/bin/bash")
+        os.execvp(shell, [shell])
+    else:
+        loop = asyncio.get_event_loop()
+
+        async def read_from_terminal():
+            while True:
+                try:
+                    data = await loop.run_in_executor(None, os.read, fd, 1024)
+                    if data:
+                        await websocket.send_text(data.decode(errors="ignore"))
+                except Exception as e:
+                    print("Erro ao ler do terminal:", e)
+                    break
+
+        async def write_to_terminal():
+            try:
+                while True:
+                    msg = await websocket.receive_text()
+                    os.write(fd, msg.encode())
+            except WebSocketDisconnect:
+                print("Cliente desconectado")
+                os.close(fd)
+
+        # Executa leitura e escrita concorrentes
+        await asyncio.gather(read_from_terminal(), write_to_terminal())