+ ideas

api/endpoints/idea_router.py

@@ -0,0 +1,61 @@
+from typing import List, Optional
+from fastapi import APIRouter, Depends, HTTPException, Query
+from api.dependency import get_idea_service, get_current_user, get_project_id, get_generation_service
+from api.service.idea_service import IdeaService
+from api.service.generation_service import GenerationService
+from models.Idea import Idea
+from api.models.GenerationRequest import GenerationResponse
+
+router = APIRouter(prefix="/ideas", tags=["ideas"])
+
+@router.post("", response_model=Idea)
+async def create_idea(
+    name: str,
+    project_id: str = Depends(get_project_id),
+    current_user: dict = Depends(get_current_user),
+    idea_service: IdeaService = Depends(get_idea_service)
+):
+    if not project_id:
+        raise HTTPException(status_code=400, detail="Project ID header is required")
+    
+    return await idea_service.create_idea(name, project_id, str(current_user["_id"]))
+
+@router.get("", response_model=List[Idea])
+async def get_ideas(
+    project_id: str = Depends(get_project_id),
+    limit: int = 20,
+    offset: int = 0,
+    idea_service: IdeaService = Depends(get_idea_service)
+):
+    if not project_id:
+        raise HTTPException(status_code=400, detail="Project ID header is required")
+    return await idea_service.get_ideas(project_id, limit, offset)
+
+@router.get("/{idea_id}", response_model=Idea)
+async def get_idea(
+    idea_id: str,
+    idea_service: IdeaService = Depends(get_idea_service)
+):
+    idea = await idea_service.get_idea(idea_id)
+    if not idea:
+        raise HTTPException(status_code=404, detail="Idea not found")
+    return idea
+
+@router.delete("/{idea_id}")
+async def delete_idea(
+    idea_id: str,
+    idea_service: IdeaService = Depends(get_idea_service)
+):
+    success = await idea_service.delete_idea(idea_id)
+    if not success:
+        raise HTTPException(status_code=404, detail="Idea not found or could not be deleted")
+    return {"status": "success"}
+
+@router.get("/{idea_id}/generations", response_model=List[GenerationResponse])
+async def get_idea_generations(
+    idea_id: str,
+    limit: int = 50,
+    offset: int = 0,
+    generation_service: GenerationService = Depends(get_generation_service)
+):
+    return await generation_service.get_generations(idea_id=idea_id, limit=limit, offset=offset)