import type {
  AgentConfig,
  AgentContextMode,
  IsolationMode,
  JoinMode,
  ThinkingLevel,
} from "./types.js";

interface AgentInvocationParams {
  model?: string;
  thinking?: string;
  max_turns?: number;
  run_in_background?: boolean;
  context?: AgentContextMode;
  isolated?: boolean;
  isolation?: IsolationMode;
}

export function resolveAgentInvocationConfig(
  agentConfig: AgentConfig | undefined,
  params: AgentInvocationParams
): {
  modelInput?: string;
  modelFromParams: boolean;
  thinking?: ThinkingLevel;
  maxTurns?: number;
  inheritContext: boolean;
  runInBackground: boolean;
  isolated: boolean;
  isolation?: IsolationMode;
} {
  return {
    modelInput: agentConfig?.model ?? params.model,
    modelFromParams: agentConfig?.model == null && params.model != null,
    thinking: (agentConfig?.thinking ?? params.thinking) as
      | ThinkingLevel
      | undefined,
    maxTurns: agentConfig?.maxTurns ?? params.max_turns,
    inheritContext:
      (agentConfig?.context ?? params.context ?? "fresh") === "fork",
    runInBackground:
      agentConfig?.runInBackground ?? params.run_in_background ?? false,
    isolated: agentConfig?.isolated ?? params.isolated ?? false,
    isolation: agentConfig?.isolation ?? params.isolation,
  };
}

export function resolveJoinMode(
  defaultJoinMode: JoinMode,
  runInBackground: boolean
): JoinMode | undefined {
  return runInBackground ? defaultJoinMode : undefined;
}