Update app.py

app.py

@@ -1,132 +1,92 @@
-from fastapi import FastAPI, Request, UploadFile, File
+from fastapi import FastAPI, UploadFile, File, Request
 from fastapi.responses import HTMLResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
-from pydantic import BaseModel
-import os, subprocess, shutil
+import shutil, os, json
 
 app = FastAPI()
 
-# Global storage
-CONFIG = {
-    "username": "",
-    "password": "",
-    "startcmd": "",
-    "running": False,
-    "process": None
-}
+# Buat direktori yang dibutuhkan
+UPLOAD_DIR = "uploads"
+STATIC_DIR = "static"
+os.makedirs(UPLOAD_DIR, exist_ok=True)
+os.makedirs(STATIC_DIR, exist_ok=True)
 
-# Base dir
-USER_DIR = "/tmp/user"
-os.makedirs(USER_DIR, exist_ok=True)
-
-# Mount static
-app.mount("/static", StaticFiles(directory="static"), name="static")
+# Serve static files
+app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static")
+app.mount("/uploads", StaticFiles(directory=UPLOAD_DIR), name="uploads")
 
+# Serve halaman utama
 @app.get("/", response_class=HTMLResponse)
-def index():
-    with open("index.html", "r", encoding="utf-8") as f:
+async def root():
+    with open("index.html", encoding="utf-8") as f:
         return f.read()
 
-# ================= SETUP ====================
-
-class SetupData(BaseModel):
-    username: str
-    password: str
-    startcmd: str
-
-@app.post("/setup")
-def save_setup(data: SetupData):
-    CONFIG.update({
-        "username": data.username,
-        "password": data.password,
-        "startcmd": data.startcmd
-    })
-    return {"status": "ok"}
-
-# ================= CONSOLE ====================
-
-@app.post("/console/start")
-def start_console():
-    if CONFIG["running"]:
-        return {"status": "already running"}
-    with open(f"{USER_DIR}/main.py", "w") as f:
-        f.write("")
-    CONFIG["process"] = subprocess.Popen(["python3", "-i", f"{USER_DIR}/main.py"], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
-    CONFIG["running"] = True
-    return {"status": "started"}
-
-@app.post("/console/stop")
-def stop_console():
-    if CONFIG["process"]:
-        CONFIG["process"].terminate()
-    CONFIG["running"] = False
-    return {"status": "stopped"}
-
-class InputData(BaseModel):
-    input: str
-
-@app.post("/console/input")
-def console_input(data: InputData):
-    if not CONFIG["running"] or not CONFIG["process"]:
-        return {"output": "Console not running"}
-    try:
-        CONFIG["process"].stdin.write(data.input + "\n")
-        CONFIG["process"].stdin.flush()
-        output = CONFIG["process"].stdout.readline()
-        return {"output": output}
-    except Exception as e:
-        return {"output": str(e)}
-
-# ================= FILE MANAGER ====================
+# Upload file
+@app.post("/upload")
+async def upload(file: UploadFile = File(...)):
+    dest = os.path.join(UPLOAD_DIR, file.filename)
+    with open(dest, "wb") as f:
+        shutil.copyfileobj(file.file, f)
+    return {"filename": file.filename}
 
+# Ambil list file/folder dari uploads
 @app.get("/files")
-def list_files():
-    result = []
-    for root, dirs, files in os.walk(USER_DIR):
-        for name in dirs + files:
-            full = os.path.join(root, name)
-            rel = os.path.relpath(full, USER_DIR)
-            result.append(rel)
-    return result
-
-class FileData(BaseModel):
-    name: str
-
-@app.post("/file/create")
-def create_file(data: FileData):
-    full = os.path.join(USER_DIR, data.name)
-    open(full, "w").close()
-    return {"status": "file created"}
-
-@app.post("/folder/create")
-def create_folder(data: FileData):
-    full = os.path.join(USER_DIR, data.name)
-    os.makedirs(full, exist_ok=True)
-    return {"status": "folder created"}
+async def files():
+    return os.listdir(UPLOAD_DIR)
 
+# Rename file
+@app.post("/rename")
+async def rename_file(request: Request):
+    data = await request.json()
+    src = os.path.join(UPLOAD_DIR, data["old"])
+    dst = os.path.join(UPLOAD_DIR, data["new"])
+    if os.path.exists(src):
+        os.rename(src, dst)
+        return {"status": "renamed"}
+    return {"status": "error", "detail": "file not found"}
+
+# Delete file
 @app.post("/delete")
-def delete_item(data: FileData):
-    full = os.path.join(USER_DIR, data.name)
-    if os.path.isdir(full):
-        shutil.rmtree(full)
-    elif os.path.isfile(full):
-        os.remove(full)
-    return {"status": "deleted"}
-
-class RenameData(BaseModel):
-    old_name: str
-    new_name: str
+async def delete_file(request: Request):
+    data = await request.json()
+    path = os.path.join(UPLOAD_DIR, data["filename"])
+    if os.path.exists(path):
+        if os.path.isfile(path):
+            os.remove(path)
+        elif os.path.isdir(path):
+            shutil.rmtree(path)
+        return {"status": "deleted"}
+    return {"status": "error", "detail": "file not found"}
+
+# Buat file kosong
+@app.post("/create_file")
+async def create_file(request: Request):
+    data = await request.json()
+    path = os.path.join(UPLOAD_DIR, data["filename"])
+    with open(path, "w") as f:
+        f.write("")
+    return {"status": "created"}
 
-@app.post("/rename")
-def rename_item(data: RenameData):
-    src = os.path.join(USER_DIR, data.old_name)
-    dst = os.path.join(USER_DIR, data.new_name)
-    os.rename(src, dst)
-    return {"status": "renamed"}
+# Buat folder
+@app.post("/create_folder")
+async def create_folder(request: Request):
+    data = await request.json()
+    path = os.path.join(UPLOAD_DIR, data["foldername"])
+    os.makedirs(path, exist_ok=True)
+    return {"status": "created"}
 
-@app.post("/upload")
-async def upload(file: UploadFile = File(...)):
-    dest = os.path.join(USER_DIR, file.filename)
-    with open(dest, "wb") as f:
-        f.write(await file.read())
-    return {"status": "uploaded"}
+# Simpan setup
+@app.post("/setup")
+async def setup_config(request: Request):
+    data = await request.json()
+    with open("setup.json", "w") as f:
+        json.dump(data, f, indent=2)
+    return {"status": "ok", "received": data}
+
+# Ambil setup
+@app.get("/setup")
+async def get_setup():
+    if os.path.exists("setup.json"):
+        with open("setup.json", "r") as f:
+            return json.load(f)
+    return {"username": "", "password": "", "start_cmd": ""}