import {z} from 'zod'

export type SnowflakeIntegrationAccountPrivateKeyAuthType = 'snowflake_private_key';

export type SnowflakeIntegrationAccountAuthenticationUpdate = {auth_type: SnowflakeIntegrationAccountPrivateKeyAuthType, private_key?: string | undefined, private_key_name?: string | undefined, private_key_passphrase?: (string | null) | undefined};

export type SnowflakeAccountUsageMetricsIntegrationDataflowSettingsRequest = {account_usage_metrics_aggregate_last_24h?: boolean | undefined};

export type SnowflakeAccountUsageMetricsIntegrationDataflowRequest = {enabled?: boolean | undefined, settings?: SnowflakeAccountUsageMetricsIntegrationDataflowSettingsRequest | undefined};

export type SnowflakeCloudCostMetricsIntegrationDataflowSettingsRequest = {query_tags?: string | undefined};

export type SnowflakeCloudCostMetricsIntegrationDataflowRequest = {enabled?: boolean | undefined, settings?: SnowflakeCloudCostMetricsIntegrationDataflowSettingsRequest | undefined};

export type SnowflakeDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest = {do_table_crawler_cron?: string | undefined, sync_snowflake_system_database?: boolean | undefined};

export type SnowflakeDataObservabilityQualityMonitoringIntegrationDataflowRequest = {enabled?: boolean | undefined, settings?: SnowflakeDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest | undefined};

export type SnowflakeEventTableLogsIntegrationDataflowSettingsRequest = {event_table_events_enabled?: boolean | undefined, event_table_logs_enabled?: boolean | undefined, event_table_logs_interval_min?: number | undefined, event_table_span_events_enabled?: boolean | undefined, event_table_spans_enabled?: boolean | undefined};

export type SnowflakeEventTableLogsIntegrationDataflowRequest = {enabled?: boolean | undefined, settings?: SnowflakeEventTableLogsIntegrationDataflowSettingsRequest | undefined};

export type SnowflakeOrganizationUsageMetricsIntegrationDataflowSettingsRequest = {organization_usage_metrics_aggregate_last_24h?: boolean | undefined};

export type SnowflakeOrganizationUsageMetricsIntegrationDataflowRequest = {enabled?: boolean | undefined, settings?: SnowflakeOrganizationUsageMetricsIntegrationDataflowSettingsRequest | undefined};

export type SnowflakeQueryHistoryLogsIntegrationDataflowSettingsRequest = {join_query_history_with_access_history_enabled?: boolean | undefined, query_history_logs_interval_min?: number | undefined};

export type SnowflakeQueryHistoryLogsIntegrationDataflowRequest = {enabled?: boolean | undefined, settings?: SnowflakeQueryHistoryLogsIntegrationDataflowSettingsRequest | undefined};

export type SnowflakeSecurityLogsIntegrationDataflowSettingsRequest = {security_logs_interval_min?: number | undefined};

export type SnowflakeSecurityLogsIntegrationDataflowRequest = {enabled?: boolean | undefined, settings?: SnowflakeSecurityLogsIntegrationDataflowSettingsRequest | undefined};

export type SnowflakeTaskHistoryLogsIntegrationDataflowSettingsRequest = {task_history_logs_interval_min?: number | undefined};

export type SnowflakeTaskHistoryLogsIntegrationDataflowRequest = {enabled?: boolean | undefined, settings?: SnowflakeTaskHistoryLogsIntegrationDataflowSettingsRequest | undefined};

export type SnowflakeIntegrationDataflowsRequest = {'snowflake-account-usage-metrics'?: SnowflakeAccountUsageMetricsIntegrationDataflowRequest | undefined, 'snowflake-cloud-cost-metrics'?: SnowflakeCloudCostMetricsIntegrationDataflowRequest | undefined, 'snowflake-data-observability-quality-monitoring'?: SnowflakeDataObservabilityQualityMonitoringIntegrationDataflowRequest | undefined, 'snowflake-event-table-logs'?: SnowflakeEventTableLogsIntegrationDataflowRequest | undefined, 'snowflake-organization-usage-metrics'?: SnowflakeOrganizationUsageMetricsIntegrationDataflowRequest | undefined, 'snowflake-query-history-logs'?: SnowflakeQueryHistoryLogsIntegrationDataflowRequest | undefined, 'snowflake-security-logs'?: SnowflakeSecurityLogsIntegrationDataflowRequest | undefined, 'snowflake-task-history-logs'?: SnowflakeTaskHistoryLogsIntegrationDataflowRequest | undefined};

export type SnowflakeIntegrationAccountSettingsUpdate = {snowflake_account_identifier?: string | undefined, username?: string | undefined};

export type SnowflakeIntegrationAccountUpdateAttributes = {authentication?: SnowflakeIntegrationAccountAuthenticationUpdate | undefined, dataflows?: SnowflakeIntegrationDataflowsRequest | undefined, name?: string | undefined, settings?: SnowflakeIntegrationAccountSettingsUpdate | undefined};

export type IntegrationAccountType = 'integration-account';

export type SnowflakeIntegrationAccountUpdateData = {attributes: SnowflakeIntegrationAccountUpdateAttributes, id: string, type: IntegrationAccountType};

export type SnowflakeIntegrationAccountUpdateRequest = {data: SnowflakeIntegrationAccountUpdateData};

export type PatchApiApiV2IntegrationInterfacesSnowflakeAccountsAccountIdArgs = {account_id: string, body: SnowflakeIntegrationAccountUpdateRequest};

export type PatchApiApiV2IntegrationInterfacesSnowflakeAccountsAccountIdOptions = {baseUrl?: string | undefined, apiKeyAuth?: string | undefined, appKeyAuth?: string | undefined, fetch?: typeof fetch | undefined};

export const snowflakeIntegrationAccountPrivateKeyAuthType = z.literal("snowflake_private_key");

export const snowflakeIntegrationAccountAuthenticationResponse = z.object({auth_type: snowflakeIntegrationAccountPrivateKeyAuthType, private_key_name: z.string()});

export const snowflakeAccountUsageMetricsIntegrationDataflowSettingsResponse = z.object({account_usage_metrics_aggregate_last_24h: z.boolean().optional()});

export const snowflakeAccountUsageMetricsIntegrationDataflowResponse = z.object({enabled: z.boolean().optional(), settings: snowflakeAccountUsageMetricsIntegrationDataflowSettingsResponse.optional()});

export const snowflakeCloudCostMetricsIntegrationDataflowSettingsResponse = z.object({query_tags: z.string().optional()});

export const snowflakeCloudCostMetricsIntegrationDataflowResponse = z.object({enabled: z.boolean().optional(), settings: snowflakeCloudCostMetricsIntegrationDataflowSettingsResponse.optional()});

export const snowflakeDataObservabilityQualityMonitoringIntegrationDataflowSettingsResponse = z.object({do_table_crawler_cron: z.string().optional(), sync_snowflake_system_database: z.boolean().optional()});

export const snowflakeDataObservabilityQualityMonitoringIntegrationDataflowResponse = z.object({enabled: z.boolean().optional(), settings: snowflakeDataObservabilityQualityMonitoringIntegrationDataflowSettingsResponse.optional()});

export const snowflakeEventTableLogsIntegrationDataflowSettingsResponse = z.object({event_table_events_enabled: z.boolean().optional(), event_table_logs_enabled: z.boolean().optional(), event_table_logs_interval_min: z.number().int().optional(), event_table_span_events_enabled: z.boolean().optional(), event_table_spans_enabled: z.boolean().optional()});

export const snowflakeEventTableLogsIntegrationDataflowResponse = z.object({enabled: z.boolean().optional(), settings: snowflakeEventTableLogsIntegrationDataflowSettingsResponse.optional()});

export const snowflakeOrganizationUsageMetricsIntegrationDataflowSettingsResponse = z.object({organization_usage_metrics_aggregate_last_24h: z.boolean().optional()});

export const snowflakeOrganizationUsageMetricsIntegrationDataflowResponse = z.object({enabled: z.boolean().optional(), settings: snowflakeOrganizationUsageMetricsIntegrationDataflowSettingsResponse.optional()});

export const snowflakeQueryHistoryLogsIntegrationDataflowSettingsResponse = z.object({join_query_history_with_access_history_enabled: z.boolean().optional(), query_history_logs_interval_min: z.number().int().optional()});

export const snowflakeQueryHistoryLogsIntegrationDataflowResponse = z.object({enabled: z.boolean().optional(), settings: snowflakeQueryHistoryLogsIntegrationDataflowSettingsResponse.optional()});

export const snowflakeSecurityLogsIntegrationDataflowSettingsResponse = z.object({security_logs_interval_min: z.number().int().optional()});

export const snowflakeSecurityLogsIntegrationDataflowResponse = z.object({enabled: z.boolean().optional(), settings: snowflakeSecurityLogsIntegrationDataflowSettingsResponse.optional()});

export const snowflakeTaskHistoryLogsIntegrationDataflowSettingsResponse = z.object({task_history_logs_interval_min: z.number().int().optional()});

export const snowflakeTaskHistoryLogsIntegrationDataflowResponse = z.object({enabled: z.boolean().optional(), settings: snowflakeTaskHistoryLogsIntegrationDataflowSettingsResponse.optional()});

export const snowflakeIntegrationDataflowsResponse = z.object({"snowflake-account-usage-metrics": snowflakeAccountUsageMetricsIntegrationDataflowResponse.optional(), "snowflake-cloud-cost-metrics": snowflakeCloudCostMetricsIntegrationDataflowResponse.optional(), "snowflake-data-observability-quality-monitoring": snowflakeDataObservabilityQualityMonitoringIntegrationDataflowResponse.optional(), "snowflake-event-table-logs": snowflakeEventTableLogsIntegrationDataflowResponse.optional(), "snowflake-organization-usage-metrics": snowflakeOrganizationUsageMetricsIntegrationDataflowResponse.optional(), "snowflake-query-history-logs": snowflakeQueryHistoryLogsIntegrationDataflowResponse.optional(), "snowflake-security-logs": snowflakeSecurityLogsIntegrationDataflowResponse.optional(), "snowflake-task-history-logs": snowflakeTaskHistoryLogsIntegrationDataflowResponse.optional()});

export const snowflakeIntegrationAccountSettingsResponse = z.object({snowflake_account_identifier: z.string(), username: z.string()});

export const snowflakeIntegrationAccountResponseAttributes = z.object({authentication: snowflakeIntegrationAccountAuthenticationResponse.optional(), dataflows: snowflakeIntegrationDataflowsResponse.optional(), name: z.string(), settings: snowflakeIntegrationAccountSettingsResponse});

export const integrationAccountType = z.literal("integration-account");

export const snowflakeIntegrationAccountResponseData = z.object({attributes: snowflakeIntegrationAccountResponseAttributes, id: z.string(), type: integrationAccountType});

export const snowflakeIntegrationAccountResponse = z.object({data: snowflakeIntegrationAccountResponseData});

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 a Snowflake integration account
 * 
 * PATCH /api/v2/integration-interfaces/snowflake/accounts/{account_id}
 * 
 * Also accepts auth: AuthZ
 */
export const patchApiApiV2IntegrationInterfacesSnowflakeAccountsAccountId = async (args: PatchApiApiV2IntegrationInterfacesSnowflakeAccountsAccountIdArgs, options?: PatchApiApiV2IntegrationInterfacesSnowflakeAccountsAccountIdOptions) => {
const {baseUrl = "https://api.datadoghq.com", fetch: fetchFn = fetch, apiKeyAuth, appKeyAuth} = options ?? {}

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

if (apiKeyAuth !== undefined) headers.set("DD-API-KEY", apiKeyAuth)
if (appKeyAuth !== undefined) headers.set("DD-APPLICATION-KEY", appKeyAuth)

const url = `${baseUrl.replace(/\/+$/, '')}/api/v2/integration-interfaces/snowflake/accounts/${encodeURIComponent(String(args.account_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(snowflakeIntegrationAccountResponse, await res.json())
};