initial commit

backend/routers/documents.py

@@ -0,0 +1,151 @@
+from typing import Any, Dict, List, Optional
+
+from fastapi import APIRouter, Depends, HTTPException
+from models.document import DocumentNode
+from models.user import User
+from pydantic import BaseModel
+from services.document_service import document_service
+
+from routers.auth import get_current_user
+
+router = APIRouter()
+
+
+class CreateDocumentRequest(BaseModel):
+    title: str
+    type: str  # "book", "chapter", "scene"
+    parent_id: Optional[str] = None
+    content: str = ""
+
+
+class UpdateDocumentRequest(BaseModel):
+    title: Optional[str] = None
+    content: Optional[str] = None
+    parent_id: Optional[str] = None  # Для переноса
+    order: Optional[int] = None
+
+
+def _get_user_project(project_id: str, current_user: User) -> None:
+    """Проверить что проект принадлежит пользователю"""
+    from services.project_service import project_service
+
+    project = project_service.get_project(project_id)
+    if not project:
+        raise HTTPException(status_code=404, detail="Project not found")
+    if project.user_id != current_user.id:
+        raise HTTPException(status_code=403, detail="Access denied")
+
+
+@router.get(
+    "/projects/{project_id}/documents/tree", response_model=List[Dict[str, Any]]
+)
+async def get_documents_tree(
+    project_id: str, current_user: User = Depends(get_current_user)
+):
+    """Получить документы в виде дерева"""
+    _get_user_project(project_id, current_user)
+    return document_service.get_documents_tree(project_id, current_user.id)
+
+
+@router.get("/projects/{project_id}/documents", response_model=List[DocumentNode])
+async def get_all_documents(
+    project_id: str, current_user: User = Depends(get_current_user)
+):
+    """Получить все документы (плоский список)"""
+    _get_user_project(project_id, current_user)
+    return document_service.get_all_documents(project_id, current_user.id)
+
+
+@router.get(
+    "/projects/{project_id}/documents/{document_id}", response_model=DocumentNode
+)
+async def get_document(
+    project_id: str,
+    document_id: str,
+    current_user: User = Depends(get_current_user),
+):
+    """Получить документ по ID"""
+    _get_user_project(project_id, current_user)
+    document = document_service.get_document(project_id, document_id, current_user.id)
+    if not document:
+        raise HTTPException(status_code=404, detail="Document not found")
+    return document
+
+
+@router.post(
+    "/projects/{project_id}/documents", response_model=DocumentNode, status_code=201
+)
+async def create_document(
+    project_id: str,
+    request: CreateDocumentRequest,
+    current_user: User = Depends(get_current_user),
+):
+    """Создать новый документ"""
+    _get_user_project(project_id, current_user)
+    try:
+        return document_service.create_document(
+            project_id=project_id,
+            title=request.title,
+            doc_type=request.type,
+            user_id=current_user.id,
+            parent_id=request.parent_id,
+            content=request.content,
+        )
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+
+@router.put(
+    "/projects/{project_id}/documents/{document_id}", response_model=DocumentNode
+)
+async def update_document(
+    project_id: str,
+    document_id: str,
+    request: UpdateDocumentRequest,
+    current_user: User = Depends(get_current_user),
+):
+    """Обновить документ (включая перенос)"""
+    _get_user_project(project_id, current_user)
+    try:
+        document = document_service.update_document(
+            project_id=project_id,
+            document_id=document_id,
+            user_id=current_user.id,
+            title=request.title,
+            content=request.content,
+            parent_id=request.parent_id,
+            order=request.order,
+        )
+        if not document:
+            raise HTTPException(status_code=404, detail="Document not found")
+        return document
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+
+@router.delete("/projects/{project_id}/documents/{document_id}")
+async def delete_document(
+    project_id: str,
+    document_id: str,
+    current_user: User = Depends(get_current_user),
+):
+    """Удалить документ с каскадом"""
+    _get_user_project(project_id, current_user)
+    success = document_service.delete_document(project_id, document_id, current_user.id)
+    if not success:
+        raise HTTPException(status_code=404, detail="Document not found")
+    return {"status": "deleted", "document_id": document_id}
+
+
+@router.get(
+    "/projects/{project_id}/documents/{parent_id}/children",
+    response_model=List[DocumentNode],
+)
+async def get_children(
+    project_id: str,
+    parent_id: str,
+    current_user: User = Depends(get_current_user),
+):
+    """Получить прямых детей узла"""
+    _get_user_project(project_id, current_user)
+    return document_service.get_children(project_id, parent_id, current_user.id)