import { Task, WorkflowContext } from '../../task';
export interface WorkflowHookContext<T extends Task<any, any>[]> {
    workflow: Workflow<T>;
    taskStates: Map<string, TaskState>;
}
export interface TaskHookContext {
    task: Task<any, any>;
    taskState: TaskState;
    ctx?: WorkflowContext;
}
export interface WorkflowHooks<T extends Task<any, any>[]> {
    onWorkflowStart?: (context: WorkflowHookContext<T>) => void;
    onWorkflowEnd?: (context: WorkflowHookContext<T>) => void;
    onWorkflowError?: (error: unknown, context: WorkflowHookContext<T>) => void;
    onTaskStart?: (context: TaskHookContext) => void;
    onTaskEnd?: (error: unknown | undefined, result: unknown | undefined, context: TaskHookContext) => void;
    /**
     * Return true if the error means we should skip retrying the task
     */
    onTaskError?: (error: unknown, context: TaskHookContext) => boolean;
    onTaskSuccess?: (result: unknown, context: TaskHookContext) => void;
    onTaskRetry?: (context: TaskHookContext) => void;
}
export interface Workflow<T extends Task<any, any>[]> {
    name: string;
    uniqueTasks: Map<keyof T[number]['name'], Task<any, any>>;
    hooks?: WorkflowHooks<T>;
}
export interface WorkflowResult {
    taskStates: Map<string, TaskState>;
    error?: unknown;
    status: 'completed' | 'failed';
    duration: number;
}
export interface TaskExecutionContext<T extends Task<any, any>[]> {
    workflow: Workflow<T>;
    taskStates: Map<string, TaskState>;
    initialData: unknown;
    context?: WorkflowContext;
    uniqueTasks: Map<keyof T[number]['name'], Task<any, any>>;
}
export interface TaskExecutionResult {
    success: boolean;
    result?: unknown;
    error?: unknown;
    task?: Task<any, any>;
}
export interface TaskState {
    status: 'pending' | 'running' | 'completed' | 'failed';
    result?: unknown;
    error?: unknown;
    attempts: number;
}