from fastapi import FastAPI, HTTPException, UploadFile, File, Query, Body, status
from fastapi.responses import JSONResponse, FileResponse
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
from typing import List, Optional
import os
import shutil
import tempfile

app = FastAPI()

class FileUpdate(BaseModel):
    operation: str
    name: Optional[str] = None
    target: Optional[str] = None
    ids: Optional[List[str]] = None

class NewFile(BaseModel):
    name: str
    type: str

class ResponseModel(BaseModel):
    error: Optional[str] = None
    result: Optional[dict] = None

class ResponseMultiModel(BaseModel):
    error: Optional[str] = None
    result: Optional[List[dict]] = None

# Configuration and global variables
UPLOAD_LIMIT = 10_000_000
ROOT_DIR = tempfile.mkdtemp()

# CORS middleware
origins = ["*"]
app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_methods=["*"],
    allow_headers=["*"],
)

def get_full_path(relative_path: str) -> str:
    return os.path.join(ROOT_DIR, relative_path)

@app.get("/files")
async def list_files():
    try:
        files = os.listdir(ROOT_DIR)
        return {"files": files}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))

@app.get("/files/{path:path}")
async def get_file(path: str):
    full_path = get_full_path(path)
    if not os.path.exists(full_path):
        raise HTTPException(status_code=404, detail="File not found")
    return FileResponse(full_path)

@app.post("/files")
async def create_file(data: NewFile):
    full_path = get_full_path(data.name)
    if os.path.exists(full_path):
        raise HTTPException(status_code=400, detail="File already exists")
    if data.type == "folder":
        os.makedirs(full_path, exist_ok=True)
    else:
        open(full_path, 'a').close()
    return {"result": {"name": data.name}}

@app.put("/files/{id:path}")
async def update_file(id: str, data: FileUpdate):
    full_path = get_full_path(id)
    if not os.path.exists(full_path):
        raise HTTPException(status_code=404, detail="File not found")
    if data.operation == "rename":
        if not data.name:
            raise HTTPException(status_code=400, detail="'name' parameter must be provided")
        new_path = get_full_path(data.name)
        os.rename(full_path, new_path)
        return {"result": {"id": new_path, "name": data.name}}
    else:
        raise HTTPException(status_code=400, detail="Operation is not supported")

@app.delete("/files/{id:path}")
async def delete_file(id: str):
    full_path = get_full_path(id)
    if not os.path.exists(full_path):
        raise HTTPException(status_code=404, detail="File not found")
    if os.path.isdir(full_path):
        shutil.rmtree(full_path)
    else:
        os.remove(full_path)
    return JSONResponse(status_code=status.HTTP_204_NO_CONTENT)

@app.post("/upload")
async def upload_file(file: UploadFile = File(...), id: Optional[str] = Query("")):
    directory = get_full_path(id)
    if not os.path.exists(directory):
        raise HTTPException(status_code=404, detail="Directory not found")
    file_path = os.path.join(directory, file.filename)
    with open(file_path, "wb") as buffer:
        shutil.copyfileobj(file.file, buffer)
    return {"result": {"name": file.filename}}

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=3200)