import {z} from 'zod'

export type ArgumentConstraintRule = {argumentPath?: string | undefined, ruleType?: ('REGEX' | 'MAX_LENGTH' | 'ENUM' | 'REQUIRED') | undefined, value?: string | undefined};

export type McpExposedTool = {proxyId?: string | undefined, toolName?: string | undefined, toolDescription?: string | undefined, overrideSchema?: string | undefined, apiMethodHttpMethod?: string | undefined, apiMethodPath?: string | undefined, requiredScopes?: Array<string> | undefined, downstreamIdentity?: ('SERVICE' | 'DELEGATED') | undefined, riskTier?: ('LOW' | 'MEDIUM' | 'HIGH') | undefined, argumentConstraints?: Array<ArgumentConstraintRule> | undefined};

export type McpDiscoveredTool = {name?: string | undefined, description?: string | undefined, inputSchemaJson?: 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 ConnectionConfigMcp = {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, transport?: ('HTTP_SSE' | 'STDIO') | undefined, mcpProtocolVersion?: 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, capabilityCacheTtlSeconds?: number | undefined, capabilityCacheEnabled?: boolean | undefined, trustToolAnnotations?: boolean | undefined, relayElicitation?: boolean | undefined, discoveredTools?: Array<McpDiscoveredTool> | undefined, toolsParsedAt?: number | undefined, credentialMappingRef?: string | 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 McpForwardedHeader = {name?: string | undefined, required?: boolean | undefined, errorMessage?: string | undefined};

export type McpToolOutputFilter = {enabled?: boolean | undefined, maskPii?: boolean | undefined, denyPatterns?: Array<string> | undefined};

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

export type McpToolArgumentConstraints = {toolName?: string | undefined, rules?: Array<ArgumentConstraintRule> | undefined};

export type McpRiskPolicy = {low?: ('ALLOW' | 'REQUIRE_APPROVAL' | 'DENY') | undefined, medium?: ('ALLOW' | 'REQUIRE_APPROVAL' | 'DENY') | undefined, high?: ('ALLOW' | 'REQUIRE_APPROVAL' | 'DENY') | undefined};

export type McpToolRiskOverride = {toolName?: string | undefined, riskTier?: ('LOW' | 'MEDIUM' | 'HIGH') | undefined};

export type McpRouting = {routingMode?: ('TOOL_EXPOSE' | 'PASSTHROUGH') | undefined, protocolVersion?: string | undefined, serverJsonOverride?: string | undefined, exposedTools?: Array<McpExposedTool> | undefined, backendServers?: Array<ConnectionConfigMcp> | undefined, allowedToolNames?: Array<string> | undefined, tokenForwardSettings?: AgentTokenForwardSettings | undefined, forwardedHeaders?: Array<McpForwardedHeader> | undefined, toolOutputFilter?: McpToolOutputFilter | undefined, identityAcl?: Array<IdentityAclRule> | undefined, argumentConstraints?: Array<McpToolArgumentConstraints> | undefined, riskPolicy?: McpRiskPolicy | undefined, riskPolicyExemptions?: Array<IdentityAclRule> | undefined, toolRiskOverrides?: Array<McpToolRiskOverride> | undefined, authProfile?: ('LEGACY' | 'STANDARD_MCP_OAUTH') | undefined, serviceClientName?: string | undefined, standardMcpOAuthProfile?: boolean | undefined, routingModeRaw?: ('TOOL_EXPOSE' | 'PASSTHROUGH') | undefined};

export type UpdateApiApiopsProjectsProjectNameApiProxiesApiProxyNameMcpRoutingArgs = {projectName: string, apiProxyName: string, body: McpRouting};

export type UpdateApiApiopsProjectsProjectNameApiProxiesApiProxyNameMcpRoutingOptions = {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 MCP Routing config of an MCP API Proxy
 * 
 * PUT /apiops/projects/{projectName}/apiProxies/{apiProxyName}/mcp-routing/
 */
export const updateApiApiopsProjectsProjectNameApiProxiesApiProxyNameMcpRouting = async (args: UpdateApiApiopsProjectsProjectNameApiProxiesApiProxyNameMcpRoutingArgs, options?: UpdateApiApiopsProjectsProjectNameApiProxiesApiProxyNameMcpRoutingOptions) => {
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))}/mcp-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())
};