import type { StudioRuntimeTurnPlan } from '../domain/types'
import type { StudioToolRegistry } from '../tools/registry'
import type { StudioTurnPlanResolver } from './turn-plan-resolver'
import { parseStudioTurnIntent } from './turn-plan-intent'
import { insertStudioReminders } from './insert-reminders'
import { resolveStudioTurnPolicy } from './turn-plan-policy'
import { buildAgentAssistantText } from './turn-plan-text'

const DEFAULT_ENABLED_TOOL_NAMES = ['skill', 'task', 'read', 'glob', 'grep', 'ls']

interface CreateStudioDefaultTurnPlanResolverOptions {
  registry: StudioToolRegistry
  enabledToolNames?: string[]
}

export function createStudioDefaultTurnPlanResolver(
  options: CreateStudioDefaultTurnPlanResolverOptions
): StudioTurnPlanResolver {
  const enabledToolNames = new Set(options.enabledToolNames ?? DEFAULT_ENABLED_TOOL_NAMES)

  return async (input) => {
    const intent = parseStudioTurnIntent(input.inputText)
    const agentToolNames = new Set(options.registry.listForAgent(input.session.agentType, input.session.studioKind).map((tool) => tool.name))
    const supportedToolNames = new Set(
      [...agentToolNames].filter((toolName) => enabledToolNames.has(toolName))
    )
    const unsupportedRequestedTools = intent.requestedToolNames.filter(
      (toolName) => !supportedToolNames.has(toolName)
    )

    const policyDecision = resolveStudioTurnPolicy({
      agentType: input.session.agentType,
      inputText: input.inputText,
      intent,
      supportedToolNames,
      workContext: input.workContext
    })

    const assistantText = insertStudioReminders({
      assistantText: buildAgentAssistantText({
        agentType: input.session.agentType,
        inputText: input.inputText,
        intent,
        policyDecision
      }),
      agentType: input.session.agentType,
      unsupportedRequestedTools,
      workContext: input.workContext,
      policyDecision
    })

    const plan: StudioRuntimeTurnPlan = {
      assistantText,
      toolCalls: policyDecision.toolCalls.length ? policyDecision.toolCalls : undefined
    }

    return plan
  }
}