import {z} from 'zod'

export type AgentSkillDef = {id?: string | undefined, name?: string | undefined, description?: string | undefined, tags?: Array<string> | undefined};

export type AgentCardSettings = {inputModes?: Array<string> | undefined, outputModes?: Array<string> | undefined, descriptionOverride?: string | undefined, skills?: Array<AgentSkillDef> | undefined};

export type A2aPushSettings = {pushNotificationsEnabled?: boolean | undefined, allowPrivateNetworkEgress?: boolean | undefined};

export type A2aDiscoveredSkill = {id?: string | undefined, name?: string | undefined, description?: string | undefined, tags?: Array<string> | undefined};

export type ConnectionMTLSSettings = {enabled?: boolean | undefined, skipSslVerification?: boolean | undefined, keyStoreId?: string | undefined, trustStoreId?: string | undefined, certificateId?: string | undefined, supportedProtocolList?: (Array<'TLS_1_3' | 'TLS_1_2' | 'TLS_1_1' | 'TLS_1_0' | 'SSL_3_0'>) | undefined, hostnameVerifierType?: ('NOOP' | 'DEFAULT' | 'STRICT' | 'BROWSER_COMPAT') | undefined};

export type ConnectionConfigA2a = {id?: string | undefined, projectId?: string | undefined, name?: string | undefined, description?: string | undefined, enabled?: boolean | undefined, projectName?: string | undefined, environmentId?: string | undefined, circuitBreakerEnabled?: boolean | undefined, errorThresholdValue?: number | undefined, sleepWindow?: number | undefined, endpoint?: string | undefined, a2aProtocolVersion?: string | undefined, allowPrivateNetworkEgress?: boolean | undefined, authScheme?: ('BEARER' | 'OAUTH2_CLIENT_CREDENTIALS' | 'MTLS' | 'NONE' | 'API_KEY' | 'BASIC' | 'OAUTH2_TOKEN_EXCHANGE') | undefined, bearerToken?: string | undefined, credentialRef?: string | undefined, tokenEndpoint?: string | undefined, clientId?: string | undefined, clientSecret?: string | undefined, scope?: string | undefined, tokenExchangeAudience?: string | undefined, apiKeyHeaderName?: string | undefined, apiKey?: string | undefined, basicUsername?: string | undefined, basicPassword?: string | undefined, bearerTokenSet?: boolean | undefined, clientSecretSet?: boolean | undefined, apiKeySet?: boolean | undefined, basicPasswordSet?: boolean | undefined, agentCardCacheTtlSeconds?: number | undefined, agentCardCacheEnabled?: boolean | undefined, discoveredSkills?: Array<A2aDiscoveredSkill> | undefined, skillsParsedAt?: number | undefined, healthStatus?: string | undefined, lastHealthCheckAt?: string | undefined, healthMessage?: string | undefined, driftPolicy?: ('WARN_ONLY' | 'FAIL_CLOSED') | undefined, pinnedCatalogHash?: string | undefined, driftDetected?: boolean | undefined, driftMessage?: string | undefined, driftStatePushPending?: boolean | undefined, resolvedEndpoint?: string | undefined, resolvedTokenEndpoint?: string | undefined, resolvedClientId?: string | undefined, resolvedApiKeyHeaderName?: string | undefined, resolvedBasicUsername?: string | undefined, resolvedTokenExchangeAudience?: string | undefined, resolvedScope?: string | undefined, mTLSSettings?: ConnectionMTLSSettings | undefined};

export type AgentTokenForwardSettings = {enabled?: boolean | undefined, source?: 'CLIENT_AUTHORIZATION_HEADER' | undefined, audienceAllowList?: Array<string> | undefined, stripGatewayCredential?: boolean | undefined};

export type IdentityAclRule = {principalType?: ('CREDENTIAL' | 'ROLE' | 'ORGANIZATION') | undefined, principalName?: string | undefined, allow?: Array<string> | undefined, deny?: Array<string> | undefined};

export type A2aRouting = {routingMode?: ('AGENT_EXPOSE' | 'PASSTHROUGH') | undefined, protocolVersion?: string | undefined, agentCard?: AgentCardSettings | undefined, pushSettings?: A2aPushSettings | undefined, backendServers?: Array<ConnectionConfigA2a> | undefined, allowedSkillNames?: Array<string> | undefined, forwardClientToken?: boolean | undefined, tokenForwardSettings?: AgentTokenForwardSettings | undefined, identityAcl?: Array<IdentityAclRule> | undefined, effectiveTokenForwardSettings?: AgentTokenForwardSettings | undefined, effectiveBackendA2aServers?: Array<ConnectionConfigA2a> | undefined, routingModeRaw?: ('AGENT_EXPOSE' | 'PASSTHROUGH') | undefined};

export type UpdateApiApiopsProjectsProjectNameApiProxiesApiProxyNameA2aRoutingArgs = {projectName: string, apiProxyName: string, body: A2aRouting};

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

export const deploymentResult = z.object({envName: z.string().optional(), podName: z.string().optional(), podIp: z.string().optional(), success: z.boolean().optional(), detail: z.string().optional(), responseTime: z.number().int().optional(), detailList: z.array(z.unknown()).optional(), firstFromApiResultList: z.unknown()});

export const apiOpsAdvisoryDTO = z.object({level: z.string().optional(), key: z.string().optional(), params: z.record(z.string(), z.unknown()).optional()});

export const apiOpsResponseSuccess = z.object({status: z.enum(["SUCCESS", "FAILURE"]).optional(), resultMessage: z.string().optional(), resultList: z.array(z.string()).optional(), resultCount: z.number().int().optional(), deploymentResult: deploymentResult.optional(), unresolvedReferences: z.array(z.string()).optional(), advisories: z.array(apiOpsAdvisoryDTO).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 the A2A Routing config of an A2A API Proxy
 * 
 * PUT /apiops/projects/{projectName}/apiProxies/{apiProxyName}/a2a-routing/
 */
export const updateApiApiopsProjectsProjectNameApiProxiesApiProxyNameA2aRouting = async (args: UpdateApiApiopsProjectsProjectNameApiProxiesApiProxyNameA2aRoutingArgs, options?: UpdateApiApiopsProjectsProjectNameApiProxiesApiProxyNameA2aRoutingOptions) => {
const {baseUrl = "http://167.86.118.166:32080", fetch: fetchFn = fetch} = options ?? {}

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

const url = `${baseUrl.replace(/\/+$/, '')}/apiops/projects/${encodeURIComponent(String(args.projectName))}/apiProxies/${encodeURIComponent(String(args.apiProxyName))}/a2a-routing/${query.size > 0 ? `?${query}` : ''}`

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