dashboard-nanobot/backend/api/workspace_router.py

from typing import List, Optional

from fastapi import APIRouter, Depends, File, HTTPException, Request, UploadFile
from sqlmodel import Session

from core.database import get_session
from models.bot import BotInstance
from schemas.system import WorkspaceFileUpdateRequest
from services.workspace_service import (
    get_workspace_tree_data,
    read_workspace_text_file,
    serve_workspace_file,
    update_workspace_markdown_file,
    upload_workspace_files_to_workspace,
)

router = APIRouter()


@router.get("/api/bots/{bot_id}/workspace/tree")
def get_workspace_tree(
    bot_id: str,
    path: Optional[str] = None,
    recursive: bool = False,
    session: Session = Depends(get_session),
):
    bot = session.get(BotInstance, bot_id)
    if not bot:
        raise HTTPException(status_code=404, detail="Bot not found")
    return get_workspace_tree_data(bot_id, path=path, recursive=recursive)

@router.get("/api/bots/{bot_id}/workspace/file")
def read_workspace_file(
    bot_id: str,
    path: str,
    max_bytes: int = 200000,
    session: Session = Depends(get_session),
):
    bot = session.get(BotInstance, bot_id)
    if not bot:
        raise HTTPException(status_code=404, detail="Bot not found")
    return read_workspace_text_file(bot_id, path=path, max_bytes=max_bytes)

@router.put("/api/bots/{bot_id}/workspace/file")
def update_workspace_file(
    bot_id: str,
    path: str,
    payload: WorkspaceFileUpdateRequest,
    session: Session = Depends(get_session),
):
    bot = session.get(BotInstance, bot_id)
    if not bot:
        raise HTTPException(status_code=404, detail="Bot not found")
    return update_workspace_markdown_file(bot_id, path=path, content=payload.content)

@router.get("/api/bots/{bot_id}/workspace/download")
def download_workspace_file(
    bot_id: str,
    path: str,
    download: bool = False,
    request: Request = None,
    session: Session = Depends(get_session),
):
    bot = session.get(BotInstance, bot_id)
    if not bot:
        raise HTTPException(status_code=404, detail="Bot not found")
    return serve_workspace_file(
        bot_id=bot_id,
        path=path,
        download=download,
        request=request,
        public=False,
        redirect_html_to_raw=True,
    )

@router.get("/public/bots/{bot_id}/workspace/download")
def public_download_workspace_file(
    bot_id: str,
    path: str,
    download: bool = False,
    request: Request = None,
    session: Session = Depends(get_session),
):
    bot = session.get(BotInstance, bot_id)
    if not bot:
        raise HTTPException(status_code=404, detail="Bot not found")
    return serve_workspace_file(
        bot_id=bot_id,
        path=path,
        download=download,
        request=request,
        public=True,
        redirect_html_to_raw=True,
    )

@router.get("/api/bots/{bot_id}/workspace/raw/{path:path}")
def raw_workspace_file(
    bot_id: str,
    path: str,
    download: bool = False,
    request: Request = None,
    session: Session = Depends(get_session),
):
    bot = session.get(BotInstance, bot_id)
    if not bot:
        raise HTTPException(status_code=404, detail="Bot not found")
    return serve_workspace_file(
        bot_id=bot_id,
        path=path,
        download=download,
        request=request,
        public=False,
        redirect_html_to_raw=False,
    )

@router.get("/public/bots/{bot_id}/workspace/raw/{path:path}")
def public_raw_workspace_file(
    bot_id: str,
    path: str,
    download: bool = False,
    request: Request = None,
    session: Session = Depends(get_session),
):
    bot = session.get(BotInstance, bot_id)
    if not bot:
        raise HTTPException(status_code=404, detail="Bot not found")
    return serve_workspace_file(
        bot_id=bot_id,
        path=path,
        download=download,
        request=request,
        public=True,
        redirect_html_to_raw=False,
    )

@router.post("/api/bots/{bot_id}/workspace/upload")
async def upload_workspace_files(
    bot_id: str,
    files: List[UploadFile] = File(...),
    path: Optional[str] = None,
    session: Session = Depends(get_session),
):
    bot = session.get(BotInstance, bot_id)
    if not bot:
        raise HTTPException(status_code=404, detail="Bot not found")
    return await upload_workspace_files_to_workspace(bot_id, files=files, path=path)