190 lines
6.6 KiB
Python
190 lines
6.6 KiB
Python
from typing import Any, Dict, Optional
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, Query
|
|
from sqlalchemy.orm import Session
|
|
|
|
from middleware.auth_middleware import get_current_user
|
|
from services.database import get_db
|
|
from services.story_writer.story_project_service import StoryProjectService
|
|
from ..models_projects import (
|
|
CreateStoryProjectRequest,
|
|
StoryProjectListResponse,
|
|
StoryProjectResponse,
|
|
UpdateStoryProjectRequest,
|
|
)
|
|
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post("/projects", response_model=StoryProjectResponse, status_code=201)
|
|
async def create_story_project(
|
|
request: CreateStoryProjectRequest,
|
|
db: Session = Depends(get_db),
|
|
current_user: Dict[str, Any] = Depends(get_current_user),
|
|
) -> StoryProjectResponse:
|
|
try:
|
|
user_id = current_user.get("user_id") or current_user.get("id")
|
|
if not user_id:
|
|
raise HTTPException(status_code=401, detail="User ID not found")
|
|
|
|
service = StoryProjectService(db)
|
|
|
|
existing = service.get_project(user_id, request.project_id)
|
|
if existing:
|
|
raise HTTPException(status_code=400, detail="Project ID already exists")
|
|
|
|
project = service.create_project(
|
|
user_id=user_id,
|
|
project_id=request.project_id,
|
|
title=request.title,
|
|
story_mode=request.story_mode,
|
|
story_template=request.story_template,
|
|
setup=request.setup,
|
|
)
|
|
|
|
return StoryProjectResponse.model_validate(project)
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Error creating story project: {str(e)}")
|
|
|
|
|
|
@router.get("/projects/{project_id}", response_model=StoryProjectResponse)
|
|
async def get_story_project(
|
|
project_id: str,
|
|
db: Session = Depends(get_db),
|
|
current_user: Dict[str, Any] = Depends(get_current_user),
|
|
) -> StoryProjectResponse:
|
|
try:
|
|
user_id = current_user.get("user_id") or current_user.get("id")
|
|
if not user_id:
|
|
raise HTTPException(status_code=401, detail="User ID not found")
|
|
|
|
service = StoryProjectService(db)
|
|
project = service.get_project(user_id, project_id)
|
|
|
|
if not project:
|
|
raise HTTPException(status_code=404, detail="Project not found")
|
|
|
|
return StoryProjectResponse.model_validate(project)
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Error fetching story project: {str(e)}")
|
|
|
|
|
|
@router.put("/projects/{project_id}", response_model=StoryProjectResponse)
|
|
async def update_story_project(
|
|
project_id: str,
|
|
request: UpdateStoryProjectRequest,
|
|
db: Session = Depends(get_db),
|
|
current_user: Dict[str, Any] = Depends(get_current_user),
|
|
) -> StoryProjectResponse:
|
|
try:
|
|
user_id = current_user.get("user_id") or current_user.get("id")
|
|
if not user_id:
|
|
raise HTTPException(status_code=401, detail="User ID not found")
|
|
|
|
service = StoryProjectService(db)
|
|
|
|
updates = request.model_dump(exclude_unset=True)
|
|
|
|
project = service.update_project(user_id, project_id, **updates)
|
|
|
|
if not project:
|
|
raise HTTPException(status_code=404, detail="Project not found")
|
|
|
|
return StoryProjectResponse.model_validate(project)
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Error updating story project: {str(e)}")
|
|
|
|
|
|
@router.get("/projects", response_model=StoryProjectListResponse)
|
|
async def list_story_projects(
|
|
status: Optional[str] = Query(None, description="Filter by status"),
|
|
favorites_only: bool = Query(False, description="Only favorites"),
|
|
limit: int = Query(50, ge=1, le=200),
|
|
offset: int = Query(0, ge=0),
|
|
order_by: str = Query("updated_at", description="Order by: updated_at or created_at"),
|
|
db: Session = Depends(get_db),
|
|
current_user: Dict[str, Any] = Depends(get_current_user),
|
|
) -> StoryProjectListResponse:
|
|
try:
|
|
user_id = current_user.get("user_id") or current_user.get("id")
|
|
if not user_id:
|
|
raise HTTPException(status_code=401, detail="User ID not found")
|
|
|
|
if order_by not in ["updated_at", "created_at"]:
|
|
raise HTTPException(status_code=400, detail="order_by must be 'updated_at' or 'created_at'")
|
|
|
|
service = StoryProjectService(db)
|
|
projects, total = service.list_projects(
|
|
user_id=user_id,
|
|
status=status,
|
|
favorites_only=favorites_only,
|
|
limit=limit,
|
|
offset=offset,
|
|
order_by=order_by,
|
|
)
|
|
|
|
return StoryProjectListResponse(
|
|
projects=[StoryProjectResponse.model_validate(p) for p in projects],
|
|
total=total,
|
|
limit=limit,
|
|
offset=offset,
|
|
)
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Error listing story projects: {str(e)}")
|
|
|
|
|
|
@router.delete("/projects/{project_id}", status_code=204)
|
|
async def delete_story_project(
|
|
project_id: str,
|
|
db: Session = Depends(get_db),
|
|
current_user: Dict[str, Any] = Depends(get_current_user),
|
|
) -> None:
|
|
try:
|
|
user_id = current_user.get("user_id") or current_user.get("id")
|
|
if not user_id:
|
|
raise HTTPException(status_code=401, detail="User ID not found")
|
|
|
|
service = StoryProjectService(db)
|
|
deleted = service.delete_project(user_id, project_id)
|
|
|
|
if not deleted:
|
|
raise HTTPException(status_code=404, detail="Project not found")
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Error deleting story project: {str(e)}")
|
|
|
|
|
|
@router.post("/projects/{project_id}/favorite", response_model=StoryProjectResponse)
|
|
async def toggle_story_project_favorite(
|
|
project_id: str,
|
|
db: Session = Depends(get_db),
|
|
current_user: Dict[str, Any] = Depends(get_current_user),
|
|
) -> StoryProjectResponse:
|
|
try:
|
|
user_id = current_user.get("user_id") or current_user.get("id")
|
|
if not user_id:
|
|
raise HTTPException(status_code=401, detail="User ID not found")
|
|
|
|
service = StoryProjectService(db)
|
|
project = service.toggle_favorite(user_id, project_id)
|
|
|
|
if not project:
|
|
raise HTTPException(status_code=404, detail="Project not found")
|
|
|
|
return StoryProjectResponse.model_validate(project)
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Error toggling story project favorite: {str(e)}")
|
|
|