Update app.py

app.py

@@ -5,70 +5,85 @@ import psutil
 import httpx
 import shutil
 from fastapi import FastAPI, WebSocket, WebSocketDisconnect, UploadFile, File, Body
-from fastapi.responses import HTMLResponse, JSONResponse
+from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
 
 app = FastAPI()
 
-# Environment variable থেকে পাসওয়ার্ড নেওয়া হয়েছে, ডিফল্ট "admin"
+# Environment variable থেকে পাসওয়ার্ড নেওয়া
 ACCESS_PASSWORD = os.getenv("ACCESS_PASSWORD", "admin")
 
-# সেশন ডিরেক্টরি ট্র্যাকিং
-current_working_dir = os.getcwd()
+# সিস্টেম রুট থেকে শুরু করার অনুমতি দেওয়া হয়েছে
+current_working_dir = "/"
 
 @app.get("/")
 async def get():
     with open("index.html", "r", encoding="utf-8") as f:
         return HTMLResponse(content=f.read())
 
-# --- ফাইল ম্যানেজার API ---
+# --- ফাইল ম্যানেজার API (Root Access Enabled) ---
 @app.get("/api/files")
-async def list_files(path: str = "."):
+async def list_files(path: str = "/"):
     try:
-        abs_path = os.path.abspath(os.path.join(current_working_dir, path))
+        # পাথ যদি খালি থাকে তবে রুটে নিয়ে যাবে
+        target_path = path if path.startswith("/") else "/"
         files = []
-        for entry in os.scandir(abs_path):
-            files.append({
-                "name": entry.name,
-                "is_dir": entry.is_dir(),
-                "path": os.path.relpath(entry.path, current_working_dir),
-                "size": f"{entry.stat().st_size / 1024:.1f} KB" if entry.is_file() else "-"
-            })
+        for entry in os.scandir(target_path):
+            try:
+                files.append({
+                    "name": entry.name,
+                    "is_dir": entry.is_dir(),
+                    "path": os.path.abspath(entry.path),
+                    "size": f"{entry.stat().st_size / 1024:.1f} KB" if entry.is_file() else "-"
+                })
+            except PermissionError:
+                continue # পারমিশন নেই এমন ফাইল স্কিপ করবে
         return sorted(files, key=lambda x: not x['is_dir'])
     except Exception as e:
         return JSONResponse(status_code=500, content={"error": str(e)})
 
 @app.get("/api/read")
 async def read_file(path: str):
-    target = os.path.join(current_working_dir, path)
-    with open(target, "r", encoding="utf-8") as f:
-        return {"content": f.read()}
+    try:
+        with open(path, "r", encoding="utf-8") as f:
+            return {"content": f.read()}
+    except Exception as e:
+        return {"content": f"Error: {str(e)}"}
 
 @app.post("/api/save")
 async def save_file(path: str = Body(...), content: str = Body(...)):
-    target = os.path.join(current_working_dir, path)
-    with open(target, "w", encoding="utf-8") as f:
+    with open(path, "w", encoding="utf-8") as f:
         f.write(content)
     return {"status": "saved"}
 
 @app.post("/api/create")
 async def create_item(name: str = Body(...), is_dir: bool = Body(...), path: str = Body(...)):
-    target = os.path.join(current_working_dir, path, name)
+    target = os.path.join(path, name)
     if is_dir: 
         os.makedirs(target, exist_ok=True)
     else: 
         with open(target, "w") as f: f.write("")
     return {"status": "created"}
 
+@app.post("/api/upload")
+async def upload_file(file: UploadFile = File(...), path: str = "/"):
+    target_path = os.path.join(path, file.filename)
+    with open(target_path, "wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
+    return {"message": "Uploaded"}
+
+@app.get("/api/download")
+async def download_file(path: str):
+    return FileResponse(path)
+
 @app.delete("/api/delete")
 async def delete_file(path: str):
-    target = os.path.join(current_working_dir, path)
-    if os.path.isdir(target):
-        shutil.rmtree(target)
+    if os.path.isdir(path):
+        shutil.rmtree(path)
     else:
-        os.remove(target)
+        os.remove(path)
     return {"message": "Deleted"}
 
-# --- ওয়েব প্রিভিউ প্রক্সি ---
+# --- প্রক্সি ---
 @app.get("/proxy")
 async def proxy(url: str = "http://localhost:8080"):
     async with httpx.AsyncClient() as client:
@@ -78,7 +93,7 @@ async def proxy(url: str = "http://localhost:8080"):
         except Exception as e:
             return HTMLResponse(content=f"<div style='color:red;padding:20px;'>Error: {str(e)}</div>")
 
-# --- টার্মিনাল সেশন (WebSocket) ---
+# --- টার্মিনাল সেশন ---
 @app.websocket("/ws")
 async def websocket_endpoint(websocket: WebSocket):
     global current_working_dir
@@ -93,13 +108,12 @@ async def websocket_endpoint(websocket: WebSocket):
         
         while True:
             command = await websocket.receive_text()
-            # 'cd' কমান��ড হ্যান্ডলিং
             if command.startswith("cd "):
-                new_path = command[3:].strip()
+                new_p = command[3:].strip()
                 try:
-                    os.chdir(os.path.expanduser(new_path))
+                    os.chdir(os.path.expanduser(new_p))
                     current_working_dir = os.getcwd()
-                    await websocket.send_text(f"Directory: {current_working_dir}")
+                    await websocket.send_text(f"CWD: {current_working_dir}")
                 except Exception as e:
                     await websocket.send_text(str(e))
                 continue
@@ -108,7 +122,7 @@ async def websocket_endpoint(websocket: WebSocket):
                 command, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE, cwd=current_working_dir
             )
             stdout, stderr = await process.communicate()
-            output = stdout.decode().strip() or stderr.decode().strip() or "Command Executed."
+            output = stdout.decode().strip() or stderr.decode().strip() or "Done."
             await websocket.send_text(output)
     except WebSocketDisconnect:
         pass