import {z} from 'zod'

export type SystemContextVariableSpec = {name: string, type: 'string' | 'int' | 'float' | 'bool' | 'enum', enum_values?: (Array<string> | null) | undefined, description?: (string | null) | undefined};

export type LegacySystemContextVariableSpec = {name: string, type: 'string' | 'int' | 'float' | 'bool' | 'enum', enum_values?: (Array<string> | null) | undefined, description?: (string | null) | undefined, source?: ('system' | null) | undefined, reask_cap: unknown};

export type ModelConfig = {temperature?: (number | null) | undefined, max_tokens?: (number | null) | undefined, top_p?: (number | null) | undefined, stop_sequences?: (Array<string> | null) | undefined, reasoning_effort?: ('none' | 'minimal' | 'low' | 'medium' | 'high' | 'xhigh' | 'max' | null) | undefined, request_timeout_ms?: (number | null) | undefined, thinking_budget?: (number | null) | undefined};

export type ModelChainEntry = {model: string, config?: ModelConfig | undefined};

export type CurrentTimeSettings = {enabled?: (boolean | null) | undefined, timezone?: (string | null) | undefined};

export type InputArgType = 'default' | 'pinned';

export type InputArgConfig = {value: unknown, type?: InputArgType | undefined};

export type ToolRef = {kind: 'api' | 'integration' | 'util' | 'kb', id?: (string | null) | undefined, slug?: (string | null) | undefined, output_bindings?: ({[key: string]: string} | null) | undefined, input_defaults?: ({[key: string]: InputArgConfig} | null) | undefined};

export type KnowledgeBaseRef = {id: string, description?: (string | null) | undefined, clearance_level?: (number | null) | undefined};

export type RevisionConfigUpdate = {base_version?: (number | null) | undefined, persona?: (string | null) | undefined, system_prompt?: (string | null) | undefined, system_context_variables?: (Array<SystemContextVariableSpec> | null) | undefined, context_variables?: (Array<LegacySystemContextVariableSpec> | null) | undefined, variable_defaults?: ({[key: string]: unknown} | null) | undefined, channel_specific_config?: ({[key: string]: unknown} | null) | undefined, router_config_id?: (string | null) | undefined, model_settings?: (ModelChainEntry | null) | undefined, analyzer_model_settings?: (ModelChainEntry | null) | undefined, workflow_enabled?: boolean | undefined, memory_enabled?: boolean | undefined, current_time?: (CurrentTimeSettings | null) | undefined, kb_clearance_level?: (number | null) | undefined, kb_audience_tags?: (Array<string> | null) | undefined, tool_refs?: (Array<ToolRef> | null) | undefined, knowledge_base_refs?: (Array<KnowledgeBaseRef> | null) | undefined, platform_tools?: ({[key: string]: unknown} | null) | undefined, continue_after_flow?: (boolean | null) | undefined};

export type PatchApiV1AssistantsAgentIdRevisionsRevisionIdArgs = {agent_id: string, revision_id: string, body: RevisionConfigUpdate};

export type PatchApiV1AssistantsAgentIdRevisionsRevisionIdOptions = {baseUrl: string, APIKeyHeader?: string | undefined, fetch?: typeof fetch | undefined};

export const contextVarSpec = z.object({name: z.string(), type: z.enum(["string", "int", "float", "bool", "enum"]), source: z.enum(["system", "derived"]).optional(), enum_values: z.array(z.string()).nullable().optional(), reask_cap: z.number().int().nullable().optional(), description: z.string().nullable().optional()});

export const firstSpeakingConfig = z.object({first_speaker: z.enum(["agent", "user"]).optional(), mode: z.enum(["static", "dynamic"]).optional(), text: z.string().max(1000).nullable().optional(), instructions: z.string().max(2000).nullable().optional(), interruptible: z.boolean().optional(), ai_disclosure_text: z.string().max(500).nullable().optional()});

export const warmTransferChannelConfig = z.object({briefing_template: z.string().max(4000).nullable().optional()});

export const channelConfig = z.object({prompt: z.string().nullable().optional(), first_speaking_config: firstSpeakingConfig.nullable().optional(), warm_transfer: warmTransferChannelConfig.nullable().optional()});

export const modelConfig = z.object({temperature: z.number().gte(0).lte(2).nullable().optional(), max_tokens: z.number().int().gte(1).nullable().optional(), top_p: z.number().gte(0).lte(1).nullable().optional(), stop_sequences: z.array(z.string()).nullable().optional(), reasoning_effort: z.enum(["none", "minimal", "low", "medium", "high", "xhigh", "max"]).nullable().optional(), request_timeout_ms: z.number().int().gte(1000).lte(20000).nullable().optional(), thinking_budget: z.number().int().gte(1).nullable().optional()});

export const modelChainEntry = z.object({model: z.string(), config: modelConfig.optional()});

export const currentTimeSettings = z.object({enabled: z.boolean().nullable().optional(), timezone: z.string().nullable().optional()});

export const systemContextVariableSpec = z.object({name: z.string(), type: z.enum(["string", "int", "float", "bool", "enum"]), enum_values: z.array(z.string()).nullable().optional(), description: z.string().nullable().optional()});

export const agentRevisionResponse = z.object({id: z.string(), agent_id: z.string(), name: z.string(), description: z.string().nullable().optional(), persona: z.string().nullable().optional(), system_prompt: z.string().nullable().optional(), context_variables: z.array(contextVarSpec).nullable().optional(), variable_defaults: z.record(z.string(), z.unknown()).nullable().optional(), tool_refs: z.array(z.record(z.string(), z.unknown())).nullable().optional(), knowledge_base_refs: z.array(z.record(z.string(), z.unknown())).nullable().optional(), platform_tools: z.record(z.string(), z.unknown()).optional(), policies_enabled: z.boolean().optional(), policy_refs: z.array(z.record(z.string(), z.unknown())).nullable().optional(), channel_specific_config: z.record(z.string(), channelConfig).nullable().optional(), router_config_id: z.string().nullable().optional(), model_settings: modelChainEntry.nullable().optional(), analyzer_model_settings: modelChainEntry.nullable().optional(), workflow_enabled: z.boolean().optional(), memory_enabled: z.boolean().optional(), current_time: currentTimeSettings.nullable().optional(), kb_clearance_level: z.number().int().nullable().optional(), kb_audience_tags: z.array(z.string()).optional(), continue_after_flow: z.boolean().nullable().optional(), voice_call_settings: z.record(z.string(), z.unknown()).nullable().optional(), created_by: z.string().nullable().optional(), created_at: z.string(), updated_at: z.string(), published_at: z.string().nullable().optional(), system_context_variables: z.array(systemContextVariableSpec).nullable()});

export const sourceSpan = z.object({start_line: z.number().int(), start_col: z.number().int(), end_line: z.number().int(), end_col: z.number().int(), raw_text_excerpt: z.string().nullable().optional()});

export const validationFinding = z.object({severity: z.enum(["error", "warning", "suggestion"]), rule_id: z.string(), category: z.string(), message: z.string(), node_id: z.string().nullable().optional(), edge_id: z.string().nullable().optional(), related_node_ids: z.array(z.string()).optional(), related_edge_ids: z.array(z.string()).optional(), source_span: sourceSpan.nullable().optional(), suggested_fix: z.string().nullable().optional(), related_spans: z.array(sourceSpan).optional()});

export const revisionResponse = z.object({revision: agentRevisionResponse, version: z.number().int().optional(), prompt_findings: z.array(validationFinding).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
};

/**
 * Update revision config (any revision, in place)
 * 
 * PATCH /v1/assistants/{agent_id}/revisions/{revision_id}
 */
export const patchApiV1AssistantsAgentIdRevisionsRevisionId = async (args: PatchApiV1AssistantsAgentIdRevisionsRevisionIdArgs, options: PatchApiV1AssistantsAgentIdRevisionsRevisionIdOptions) => {
const {baseUrl, fetch: fetchFn = fetch, APIKeyHeader} = options ?? {}

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

if (APIKeyHeader !== undefined) headers.set("x-api-key", APIKeyHeader)

const url = `${baseUrl.replace(/\/+$/, '')}/v1/assistants/${encodeURIComponent(String(args.agent_id))}/revisions/${encodeURIComponent(String(args.revision_id))}${query.size > 0 ? `?${query}` : ''}`

const res = await fetchFn(url, {method: "PATCH", 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(revisionResponse, await res.json())
};