from fastapi import APIRouter, HTTPException, Depends, Query
from typing import List, Any, Optional
from pydantic import BaseModel
from flowsint_core.core.registry import TransformRegistry
from flowsint_core.core.celery import celery
from flowsint_core.core.types import Node, Edge, FlowBranch
from flowsint_core.core.models import Profile
from app.api.deps import get_current_user


class FlowComputationRequest(BaseModel):
    nodes: List[Node]
    edges: List[Edge]
    inputType: Optional[str] = None


class FlowComputationResponse(BaseModel):
    flowBranches: List[FlowBranch]
    initialData: Any


class StepSimulationRequest(BaseModel):
    flowBranches: List[FlowBranch]
    currentStepIndex: int


class launchTransformPayload(BaseModel):
    values: List[str]
    sketch_id: str


router = APIRouter()


# Get the list of all transforms
@router.get("/")
def get_transforms(
    category: Optional[str] = Query(None),
    # current_user: Profile = Depends(get_current_user),
):
    if category is not None and category != "undefined":
        transforms = TransformRegistry.list_by_input_type(
            category, exclude=["n8n_connector"]
        )
    else:
        transforms = TransformRegistry.list(exclude=["n8n_connector"])
    return transforms


@router.post("/{transform_name}/launch")
async def launch_transform(
    transform_name: str,
    payload: launchTransformPayload,
    current_user: Profile = Depends(get_current_user),
):
    try:
        task = celery.send_task(
            "run_transform",
            args=[
                transform_name,
                payload.values,
                payload.sketch_id,
                str(current_user.id),
            ],
        )
        return {"id": task.id}

    except Exception as e:
        print(e)
        raise HTTPException(status_code=404, detail="Transform not found")