Update main.py

main.py

@@ -30,10 +30,91 @@ class WriteFileRequest(BaseModel):
     path: str
     content: str
 
+class RunCodeRequest(BaseModel):
+    language: str = "python"  # python, bash, node, etc.
+    code: str
+    timeout: int = 30
+
 @app.get("/")
 def read_root():
     return {"status": "online", "workspace": WORKSPACE, "auth_enabled": True}
 
+# 语言到执行命令的映射
+LANGUAGE_RUNNERS = {
+    "python": "python3",
+    "python3": "python3",
+    "py": "python3",
+    "bash": "bash",
+    "sh": "sh",
+    "node": "node",
+    "javascript": "node",
+    "js": "node",
+}
+
+# 语言到文件扩展名的映射
+LANGUAGE_EXTENSIONS = {
+    "python": ".py",
+    "python3": ".py",
+    "py": ".py",
+    "bash": ".sh",
+    "sh": ".sh",
+    "node": ".js",
+    "javascript": ".js",
+    "js": ".js",
+}
+
+@app.post("/run_code", dependencies=[Depends(verify_token)])
+def run_code(req: RunCodeRequest):
+    """一站式代码执行：自动写入临时文件并执行"""
+    lang = req.language.lower()
+    
+    if lang not in LANGUAGE_RUNNERS:
+        raise HTTPException(
+            status_code=400, 
+            detail=f"Unsupported language: {req.language}. Supported: {list(LANGUAGE_RUNNERS.keys())}"
+        )
+    
+    runner = LANGUAGE_RUNNERS[lang]
+    ext = LANGUAGE_EXTENSIONS[lang]
+    
+    # 创建临时文件
+    import uuid
+    temp_filename = f"_run_{uuid.uuid4().hex[:8]}{ext}"
+    temp_path = os.path.join(WORKSPACE, temp_filename)
+    
+    try:
+        # 写入代码
+        with open(temp_path, "w", encoding="utf-8") as f:
+            f.write(req.code)
+        
+        # 执行代码
+        result = subprocess.run(
+            f"{runner} {temp_filename}",
+            shell=True,
+            cwd=WORKSPACE,
+            capture_output=True,
+            text=True,
+            timeout=req.timeout,
+            encoding='utf-8',
+            errors='replace'
+        )
+        
+        return {
+            "stdout": result.stdout,
+            "stderr": result.stderr,
+            "exit_code": result.returncode,
+            "temp_file": temp_filename
+        }
+    except subprocess.TimeoutExpired:
+        raise HTTPException(status_code=408, detail="Code execution timed out")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+    finally:
+        # 清理临时文件（可选，保留以便调试）
+        # if os.path.exists(temp_path):
+        #     os.remove(temp_path)
+        pass
+
 @app.post("/execute", dependencies=[Depends(verify_token)])
 def execute(req: ExecuteRequest):
     try: