import {z} from 'zod'

export type FilterType = 'src_network' | 'dst_network' | 'src_l4_port' | 'dst_l4_port' | 'ip_protocol' | 'src_segment' | 'dst_segment' | 'src_object' | 'dst_object' | 'src_group' | 'dst_group' | 'ingress_vi' | 'tcp_control' | 'is_fragmented' | 'tos' | 'traffic_class' | 'ttl' | 'hop_limit' | 'ip_version' | 'src_dia' | 'dst_dia';

export type RuleFilter = {filter_type: FilterType, values: unknown};

export type NBForwardingPolicy = 'ACCEPT' | 'DROP' | 'ALLOW' | 'DENY' | 'ALLOW_WITH_BLOCKED_SNI' | 'ALLOW_ONLY_SELECTED_SNIS';

export type NBUpdateFirewallRuleRequest = {name?: (string | null) | undefined, priority?: (number | null) | undefined, filters?: (Array<RuleFilter> | null) | undefined, forwarding_policy?: (NBForwardingPolicy | null) | undefined, enable?: (boolean | null) | undefined, description?: (string | null) | undefined, categories?: (Array<string> | null) | undefined, allowed_domains?: (Array<string> | null) | undefined, denied_domains?: (Array<string> | null) | undefined, policy_name?: (string | null) | undefined, add_time_policies?: (Array<string> | null) | undefined, remove_time_policies?: (Array<string> | null) | undefined, add_gate_vi_names?: (Array<string> | null) | undefined, remove_gate_vi_names?: (Array<string> | null) | undefined};

export type UpdateApiServiceTenantNameFirewallRuleRuleNameArgs = {tenant_name: unknown, rule_name: unknown, body: NBUpdateFirewallRuleRequest};

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

export const configurationRequestStatus = z.enum(["received", "in_progress", "success", "failed", "cancelled"]);

export const nBRequestResponse = z.object({id: z.string(), created_at: z.string(), updated_at: z.string(), status: configurationRequestStatus, extra_details: z.record(z.string(), z.unknown()).nullable().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 FW rule
 * 
 * PUT /service/{tenant_name}/firewall_rule/{rule_name}
 */
export const updateApiServiceTenantNameFirewallRuleRuleName = async (args: UpdateApiServiceTenantNameFirewallRuleRuleNameArgs, options: UpdateApiServiceTenantNameFirewallRuleRuleNameOptions) => {
const {baseUrl, fetch: fetchFn = fetch, HTTPBearerFromCookie} = options ?? {}

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

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

const url = `${baseUrl.replace(/\/+$/, '')}/service/${encodeURIComponent(String(args.tenant_name))}/firewall_rule/${encodeURIComponent(String(args.rule_name))}${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(nBRequestResponse, await res.json())
};