xiaoxia-saas/apps/api/app/api/routes/projects.py

from typing import Any

from app.auth import AuthenticatedUser, get_current_user
from app.dependencies import get_asset_library_repository, get_project_repository
from app.schemas.project import (
    CreateProjectRequest,
    ListProjectsResponse,
    ProjectResponse,
)
from fastapi import APIRouter, Depends, HTTPException, Response, status
from pydantic import BaseModel

from packages.application import (
    CreateProjectCommand,
    CreateProjectUseCase,
    DeleteProjectUseCase,
    GetProjectUseCase,
    ListProjectsUseCase,
)
from packages.domain import AssetLibraryKind

router = APIRouter()


class DefaultContextResponse(BaseModel):
    """幂等默认上下文响应（Issue #1775）：默认项目 + 各类型默认素材库 ID。"""

    project_id: str
    image_library_id: str
    video_library_id: str
    voice_library_id: str


def _to_project_response(item) -> ProjectResponse:
    return ProjectResponse(
        id=item.id,
        owner_user_id=item.owner_user_id,
        name=item.name,
        description=item.description,
        shared_users=item.shared_users,
    )


@router.get("/{project_id}", response_model=ProjectResponse)
def get_project(
    project_id: str,
    authenticated_user: AuthenticatedUser = Depends(get_current_user),
    project_repository: Any = Depends(get_project_repository),
) -> ProjectResponse:
    use_case = GetProjectUseCase(project_repository)
    project = use_case.execute(project_id)
    if project is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Project not found")
    if not project.can_access(authenticated_user.user.id):
        raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="Access denied to project")
    return _to_project_response(project)


@router.get("", response_model=ListProjectsResponse)
def list_projects(
    authenticated_user: AuthenticatedUser = Depends(get_current_user),
    project_repository: Any = Depends(get_project_repository),
) -> ListProjectsResponse:
    use_case = ListProjectsUseCase(project_repository)
    projects = use_case.execute(authenticated_user.user.id)
    return ListProjectsResponse(items=[_to_project_response(item) for item in projects])


@router.post("", response_model=ProjectResponse)
def create_project(
    request: CreateProjectRequest,
    authenticated_user: AuthenticatedUser = Depends(get_current_user),
    project_repository: Any = Depends(get_project_repository),
) -> ProjectResponse:
    use_case = CreateProjectUseCase(project_repository)
    project = use_case.execute(
        CreateProjectCommand(
            name=request.name,
            description=request.description,
        ),
        owner_user_id=authenticated_user.user.id,
    )
    return _to_project_response(project)


@router.post("/ensure-default", response_model=DefaultContextResponse)
def ensure_default_project_and_libraries(
    authenticated_user: AuthenticatedUser = Depends(get_current_user),
    project_repository: Any = Depends(get_project_repository),
    asset_library_repository: Any = Depends(get_asset_library_repository),
) -> DefaultContextResponse:
    """幂等获取/创建当前用户的默认项目和三类默认素材库（Issue #1775）。

    - 同一用户永远只有一个默认项目（部分唯一索引 uq_projects_owner_default）
    - 同一项目同 kind 永远只有一个默认素材库（唯一约束 uq_asset_libraries_project_kind）
    - 并发调用/失败重试：唯一约束冲突时返回已存在记录，不报 500
    - 项目和素材库的创建各自在仓储事务内幂等，冲突回滚后重查返回同一条
    """
    user_id = authenticated_user.user.id
    project = project_repository.get_or_create_default_project(user_id)

    libraries = {}
    for kind in (AssetLibraryKind.VIDEO, AssetLibraryKind.VOICE, AssetLibraryKind.IMAGE):
        library = asset_library_repository.get_or_create_default_library(project.id, kind)
        libraries[kind] = library.id

    return DefaultContextResponse(
        project_id=project.id,
        image_library_id=libraries[AssetLibraryKind.IMAGE],
        video_library_id=libraries[AssetLibraryKind.VIDEO],
        voice_library_id=libraries[AssetLibraryKind.VOICE],
    )


@router.delete("/{project_id}", status_code=status.HTTP_204_NO_CONTENT, response_model=None, response_class=Response)
def delete_project(
    project_id: str,
    authenticated_user: AuthenticatedUser = Depends(get_current_user),
    project_repository: Any = Depends(get_project_repository),
) -> dict:
    use_case = DeleteProjectUseCase(project_repository)
    try:
        deleted = use_case.execute(project_id, authenticated_user.user.id)
    except PermissionError as _e:
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail="Only the project owner can delete this project",
        ) from _e
    if not deleted:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Project not found")
    return  # type: ignore[return-value]