Create app.py

app.py

@@ -0,0 +1,111 @@
+import os
+import uuid
+import subprocess
+import resource
+import multiprocessing
+import signal
+import time
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+import uvicorn
+
+# --- FastAPI app ---
+app = FastAPI()
+
+# Allow CORS for frontend testing (optional)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Store running sandboxes
+sandboxes = {}
+
+# --- Sandbox worker ---
+def sandbox_worker(sandbox_id: str, code: str):
+    user_dir = f"/tmp/sandbox_{sandbox_id}"
+    os.makedirs(user_dir, exist_ok=True)
+    os.chdir(user_dir)
+
+    # Resource limits
+    resource.setrlimit(resource.RLIMIT_AS, (4 * 1024**3, 4 * 1024**3))  # 4GB RAM
+    resource.setrlimit(resource.RLIMIT_CPU, (7200, 7200))  # 2h CPU time
+
+    # Save user code
+    code_file = os.path.join(user_dir, "user_code.py")
+    with open(code_file, "w") as f:
+        f.write(code)
+
+    # Run code
+    try:
+        result = subprocess.run(
+            ["python3", code_file],
+            capture_output=True,
+            text=True,
+            timeout=7200  # 2h wall clock
+        )
+        # Save logs
+        with open(os.path.join(user_dir, "stdout.log"), "w") as f:
+            f.write(result.stdout)
+        with open(os.path.join(user_dir, "stderr.log"), "w") as f:
+            f.write(result.stderr)
+    except subprocess.TimeoutExpired:
+        with open(os.path.join(user_dir, "stderr.log"), "w") as f:
+            f.write("Execution timed out (2h limit reached).")
+
+
+# --- API endpoints ---
+@app.post("/launch")
+def launch_sandbox(code: str):
+    sandbox_id = str(uuid.uuid4())[:8]
+
+    p = multiprocessing.Process(target=sandbox_worker, args=(sandbox_id, code))
+    p.start()
+
+    sandboxes[sandbox_id] = {"pid": p.pid, "start_time": time.time()}
+    return {"sandbox_id": sandbox_id}
+
+
+@app.get("/logs/{sandbox_id}")
+def get_logs(sandbox_id: str):
+    user_dir = f"/tmp/sandbox_{sandbox_id}"
+    if not os.path.exists(user_dir):
+        raise HTTPException(status_code=404, detail="Sandbox not found")
+
+    stdout_file = os.path.join(user_dir, "stdout.log")
+    stderr_file = os.path.join(user_dir, "stderr.log")
+
+    stdout = open(stdout_file).read() if os.path.exists(stdout_file) else ""
+    stderr = open(stderr_file).read() if os.path.exists(stderr_file) else ""
+
+    return {"stdout": stdout, "stderr": stderr}
+
+
+@app.post("/kill/{sandbox_id}")
+def kill_sandbox(sandbox_id: str):
+    if sandbox_id not in sandboxes:
+        raise HTTPException(status_code=404, detail="Sandbox not found")
+
+    pid = sandboxes[sandbox_id]["pid"]
+    try:
+        os.kill(pid, signal.SIGKILL)
+        del sandboxes[sandbox_id]
+        return {"status": "killed"}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+@app.get("/status/{sandbox_id}")
+def status(sandbox_id: str):
+    if sandbox_id not in sandboxes:
+        return {"status": "not found"}
+    elapsed = time.time() - sandboxes[sandbox_id]["start_time"]
+    return {"status": "running", "elapsed_sec": elapsed}
+
+
+# --- Main entrypoint ---
+if __name__ == "__main__":
+    uvicorn.run("app:app", host="0.0.0.0", port=8000, reload=True)