import {z} from 'zod'

export type AnswerCheckPolicy = {enabled?: boolean | undefined, language?: (string | null) | undefined, rewrite_instruction?: (string | null) | undefined};

export type CodeToolSettings = {disabled_methods?: (Array<string> | null) | undefined, image_model_id?: (string | null) | undefined, requires_user_confirmation?: (boolean | null) | undefined, video_model_id?: (string | null) | undefined};

export type CodeToolConfig = {name: string, settings?: (CodeToolSettings | null) | undefined, type?: 'code' | undefined};

export type McpToolPermission = {requires_user_confirmation?: (boolean | null) | undefined, tool_name: string};

export type McpToolSettings = {allowed_tools?: (Array<McpToolPermission> | null) | undefined, requires_user_confirmation?: (boolean | null) | undefined};

export type McpToolConfigInput = {name: string, settings?: (McpToolSettings | null) | undefined, type?: 'mcp' | undefined};

export type AgentToolSettings = {a2a_url?: (string | null) | undefined, agent_id?: (string | null) | undefined, auth_secret_name?: (string | null) | undefined, description_override?: (string | null) | undefined, requires_user_confirmation?: (boolean | null) | undefined};

export type AgentToolConfig = {name: string, settings?: (AgentToolSettings | null) | undefined, type?: 'agent' | undefined};

export type OpenApiToolSettings = {allowed_tools?: (Array<string> | null) | undefined, load_mode?: ('explicit' | 'searchable' | null) | undefined, openapi_connection_id?: (string | null) | undefined, requires_user_confirmation?: (boolean | null) | undefined};

export type OpenApiToolConfig = {name: string, settings?: (OpenApiToolSettings | null) | undefined, type?: 'openapi' | undefined};

export type TriggerBatchWindow = {max_events?: (number | null) | undefined, window?: (string | null) | undefined};

export type TriggerSession = {continue?: (boolean | null) | undefined, idle_timeout?: (string | null) | undefined};

export type TriggerSpec = {batch?: (TriggerBatchWindow | null) | undefined, conditions?: {[key: string]: unknown} | undefined, cron_expression?: (string | null) | undefined, description?: string | undefined, enabled?: boolean | undefined, event_filter?: ({[key: string]: unknown} | null) | undefined, failure_threshold?: number | undefined, name: string, session?: (TriggerSession | null) | undefined, stream_id?: (string | null) | undefined, task_definition_id?: (string | null) | undefined, task_parameters?: {[key: string]: unknown} | undefined, timezone?: string | undefined, trigger_type: 'cron' | 'stream'};

export type AgentCreateRequest = {a2ui_enabled?: (boolean | null) | undefined, agent_type?: ('stateless' | 'stateful') | undefined, answer_check?: (AnswerCheckPolicy | null) | undefined, client_id?: (string | null) | undefined, description?: string | undefined, instruction?: string | undefined, model_id?: (string | null) | undefined, name: string, planning?: (boolean | null) | undefined, reasoning_effort?: (string | null) | undefined, reasoning_max_tokens?: (number | null) | undefined, skill_ids?: (Array<string> | null) | undefined, tools: Array<CodeToolConfig | McpToolConfigInput | AgentToolConfig | OpenApiToolConfig>, triggers?: Array<TriggerSpec> | undefined};

export type CreateApiV1WorkspacesWorkspaceAgentsArgs = {workspace: string, body: AgentCreateRequest};

export type CreateApiV1WorkspacesWorkspaceAgentsOptions = {baseUrl: string, HTTPBearer?: string | undefined, fetch?: typeof fetch | undefined};

export const answerCheckPolicy = z.object({enabled: z.boolean().optional(), language: z.string().min(2).max(8).nullable().optional(), rewrite_instruction: z.string().min(1).max(4000).nullable().optional()});

export const agentClientSummary = z.object({default_model: z.string().nullable(), id: z.string(), kind: z.string(), name: z.string(), online: z.boolean()});

export const codeToolSettings = z.object({disabled_methods: z.array(z.string()).nullable().optional(), image_model_id: z.string().nullable().optional(), requires_user_confirmation: z.boolean().nullable().optional(), video_model_id: z.string().nullable().optional()});

export const codeToolConfig = z.object({name: z.string(), settings: codeToolSettings.nullable().optional(), type: z.literal("code").optional()});

export const mcpToolPermission = z.object({requires_user_confirmation: z.boolean().nullable().optional(), tool_name: z.string()});

export const mcpToolSettings = z.object({allowed_tools: z.array(mcpToolPermission).nullable().optional(), requires_user_confirmation: z.boolean().nullable().optional()});

export const mcpToolConfigOutput = z.object({name: z.string(), settings: mcpToolSettings.nullable().optional(), type: z.literal("mcp").optional()});

export const agentToolSettings = z.object({a2a_url: z.string().nullable().optional(), agent_id: z.string().nullable().optional(), auth_secret_name: z.string().nullable().optional(), description_override: z.string().nullable().optional(), requires_user_confirmation: z.boolean().nullable().optional()});

export const agentToolConfig = z.object({name: z.string(), settings: agentToolSettings.nullable().optional(), type: z.literal("agent").optional()});

export const openApiToolSettings = z.object({allowed_tools: z.array(z.string()).nullable().optional(), load_mode: z.enum(["explicit", "searchable"]).nullable().optional(), openapi_connection_id: z.string().nullable().optional(), requires_user_confirmation: z.boolean().nullable().optional()});

export const openApiToolConfig = z.object({name: z.string(), settings: openApiToolSettings.nullable().optional(), type: z.literal("openapi").optional()});

export const agentResponse = z.object({a2a_url: z.string().nullable().optional(), a2ui_enabled: z.boolean().nullable().optional(), agent_type: z.string().optional(), answer_check: answerCheckPolicy.nullable().optional(), client: agentClientSummary.nullable().optional(), client_id: z.string().nullable().optional(), description: z.string().nullable().optional(), id: z.string(), instruction: z.string().nullable().optional(), is_catalog: z.boolean().optional(), model_id: z.string().nullable().optional(), name: z.string(), planning: z.boolean().nullable().optional(), reasoning_effort: z.string().nullable().optional(), reasoning_max_tokens: z.number().int().nullable().optional(), registry_item_id: z.string().nullable().optional(), skills: z.array(z.record(z.string(), z.unknown())).nullable().optional(), slug: z.string(), status: z.string(), tools: z.array(z.discriminatedUnion("type", [codeToolConfig, mcpToolConfigOutput, agentToolConfig, openApiToolConfig])).nullable().optional(), update_available: z.boolean().optional()});

export const validateResponse = async <Output>(
  schema: {
    readonly '~standard': {
      readonly validate: (value: unknown) => ({ readonly value: Output; readonly issues?: undefined } | { readonly issues: ReadonlyArray<{ readonly message: string; readonly path?: ReadonlyArray<PropertyKey | { readonly key: PropertyKey }> | undefined }> }) | Promise<{ readonly value: Output; readonly issues?: undefined } | { readonly issues: ReadonlyArray<{ readonly message: string; readonly path?: ReadonlyArray<PropertyKey | { readonly key: PropertyKey }> | undefined }> }>
    }
  },
  value: unknown
): Promise<Output> => {
  const result = await schema['~standard'].validate(value)

  if (result.issues !== undefined) {
    const detail = result.issues
      .map(({ message, path }) =>
        path?.length
          ? `${path.map((segment) => String(typeof segment === 'object' ? segment.key : segment)).join('.')}: ${message}`
          : message
      )
      .join('; ')

    throw new Error(`Response does not match the schema: ${detail}`, { cause: { issues: result.issues } })
  }

  return result.value
};

/**
 * Create Agent
 * 
 * POST /v1/workspaces/{workspace}/agents/
 */
export const createApiV1WorkspacesWorkspaceAgents = async (args: CreateApiV1WorkspacesWorkspaceAgentsArgs, options: CreateApiV1WorkspacesWorkspaceAgentsOptions) => {
const {baseUrl, fetch: fetchFn = fetch, HTTPBearer} = options ?? {}

const query = new URLSearchParams()
const headers = new Headers()
headers.set('content-type', "application/json")
headers.set('accept', "application/json")

if (HTTPBearer !== undefined) headers.set('authorization', `Bearer ${HTTPBearer}`)

const url = `${baseUrl.replace(/\/+$/, '')}/v1/workspaces/${encodeURIComponent(String(args.workspace))}/agents/${query.size > 0 ? `?${query}` : ''}`

const res = await fetchFn(url, {method: "POST", headers, body: JSON.stringify(args.body)})

if (!res.ok) throw new Error(`Request failed with status ${res.status}`, {cause: {status: res.status, body: await res.text()}})

return validateResponse(agentResponse, await res.json())
};