87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.orm import Session
|
|
from typing import List, Optional
|
|
from uuid import UUID
|
|
from app.core.database import get_db
|
|
from app.core.security import get_current_user
|
|
from app.models.user import User
|
|
from app.models.flow_template import FlowTemplate
|
|
from app.models.flow import Flow, TriggerType
|
|
from app.schemas.flow_template import FlowTemplateCreate, FlowTemplateResponse
|
|
|
|
router = APIRouter(prefix="/api/flow-templates", tags=["flow-templates"])
|
|
|
|
|
|
@router.get("", response_model=List[FlowTemplateResponse])
|
|
def list_templates(
|
|
category: Optional[str] = None,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
query = db.query(FlowTemplate).filter(FlowTemplate.is_public == True)
|
|
if category:
|
|
query = query.filter(FlowTemplate.category == category)
|
|
return query.all()
|
|
|
|
|
|
@router.post("", response_model=FlowTemplateResponse)
|
|
def create_template(
|
|
request: FlowTemplateCreate,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
template = FlowTemplate(
|
|
**request.model_dump(),
|
|
created_by=current_user.id,
|
|
)
|
|
db.add(template)
|
|
db.commit()
|
|
db.refresh(template)
|
|
return template
|
|
|
|
|
|
@router.post("/{template_id}/use")
|
|
def use_template(
|
|
template_id: UUID,
|
|
flow_name: str,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
"""Create a new flow from template"""
|
|
template = db.query(FlowTemplate).filter(FlowTemplate.id == template_id).first()
|
|
if not template:
|
|
raise HTTPException(status_code=404, detail="Template not found")
|
|
|
|
flow = Flow(
|
|
name=flow_name,
|
|
description=f"Created from template: {template.name}",
|
|
trigger_type=TriggerType.KEYWORD,
|
|
nodes=template.nodes,
|
|
edges=template.edges,
|
|
variables=template.variables,
|
|
is_active=False,
|
|
)
|
|
db.add(flow)
|
|
db.commit()
|
|
db.refresh(flow)
|
|
|
|
return {"success": True, "flow_id": str(flow.id)}
|
|
|
|
|
|
@router.delete("/{template_id}")
|
|
def delete_template(
|
|
template_id: UUID,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
template = db.query(FlowTemplate).filter(FlowTemplate.id == template_id).first()
|
|
if not template:
|
|
raise HTTPException(status_code=404, detail="Template not found")
|
|
|
|
if template.created_by != current_user.id:
|
|
raise HTTPException(status_code=403, detail="Not authorized")
|
|
|
|
db.delete(template)
|
|
db.commit()
|
|
return {"success": True}
|