Types

import type {
    ActionDefinition,
    ActionOption,
    ToolAction,
    UseAssistantOptions,
    UseAssistantReturn,
    DynamicButton,
    DynamicButtonGroup,
    DynamicUITree,
    ChatStatus,
} from "modifywithai"

interface ActionDefinition {
    name: string
    description: string
    options?: Record<string, ActionOption>
    approvalRequired?: boolean
}

interface ActionOption {
    type: "string" | "number" | "boolean"
    description?: string
    required?: boolean
}

interface ToolAction {
    name: string
    options?: Record<string, unknown>
}

interface UseAssistantOptions {
    availableActions: ActionDefinition[]
    tokenEndpoint?: string
    apiUrl?: string
    endUserId?: string
    getContext?: () => Record<string, unknown>
    onAction?: (action: ToolAction) => void
    onError?: (error: Error) => void
}

interface UseAssistantReturn {
    // Messages
    messages: Message[]
    sendMessage: (text: string) => void

    // Status
    status: ChatStatus
    error: Error | null
    isReady: boolean
    tokenError: Error | null

    // Input
    input: string
    setInput: (value: string) => void
    handleSubmit: (e: FormEvent) => void

    // Token
    refreshToken: () => Promise<void>

    // Approval
    addToolApprovalResponse: (toolCallId: string, approved: boolean) => void
}

type ChatStatus = "ready" | "streaming" | "submitted" | "error"

interface DynamicButton {
    id: string
    label: string
    icon?: string
    variant?: "default" | "destructive" | "outline" | "secondary" | "ghost"
    actions: ToolAction[]
}

interface DynamicButtonGroup {
    id: string
    label?: string
    direction?: "horizontal" | "vertical"
}

interface DynamicUITree {
    root: string
    elements: Record<string, DynamicUIElement>
}

interface DynamicUIElement {
    key: string
    type: "DynamicButton" | "DynamicButtonGroup"
    props: Record<string, unknown>
    children?: string[]
    parentKey?: string
}

interface UseDynamicUIReturn {
    tree: DynamicUITree | null
    isLoading: boolean
    addButton: (button: DynamicButton) => void
    removeButton: (id: string) => void
    updateButton: (id: string, updates: Partial<DynamicButton>) => void
    addGroup: (group: DynamicButtonGroup) => void
    removeGroup: (id: string) => void
    clearAll: () => void
}

interface Message {
    id: string
    role: "user" | "assistant" | "system"
    parts: MessagePart[]
    createdAt?: Date
}

type MessagePart =
    | { type: "text"; text: string }
    | { type: "tool-invocation"; toolInvocation: ToolInvocation }

interface ToolInvocation {
    toolCallId: string
    toolName: string
    args: Record<string, unknown>
    state: "partial-call" | "call" | "result"
    result?: unknown
}

interface TokenRequest {
    endUserId?: string
    availableActions: ActionDefinition[]
    context?: Record<string, unknown>
}

interface TokenResponse {
    token: string
    expiresAt: string // ISO date string
}

interface CreateAssistantTokenHandlerOptions {
    appId: string
    apiKey?: string
    apiUrl?: string
    getEndUserId?: (request: Request) => Promise<string | null>
}