import {z} from 'zod'

export type VmGpu = {vendor?: string | undefined, mode?: string | undefined, device_id?: number | undefined};

export type DiskAddress = {device_index: number, adapter_type: string};

export type VmBootDevice = {disk_address?: DiskAddress | undefined, mac_address?: string | undefined};

export type VmBootConfig = {boot_device_order_list?: Array<string> | undefined, boot_device?: VmBootDevice | undefined};

export type CustomGenericKeyValuePair = {[key: string]: string};

export type GuestCustomization = {cloud_init?: ({meta_data?: string | undefined, user_data?: string | undefined, custom_key_values?: CustomGenericKeyValuePair | undefined}) | undefined, is_overridable?: boolean | undefined, sysprep?: ({install_type?: string | undefined, unattend_xml?: string | undefined, custom_key_values?: CustomGenericKeyValuePair | undefined}) | undefined} | {[key: string]: string};

export type VmGuestPowerStateTransitionConfig = {should_fail_on_script_failure?: boolean | undefined, enable_script_exec?: boolean | undefined};

export type VmPowerStateMechanism = {guest_transition_config?: VmGuestPowerStateTransitionConfig | undefined, mechanism?: string | undefined};

export type Reference = {kind: string, uuid: unknown, name?: string | undefined};

export type VmDisk = {volume_group_reference?: Reference | undefined, device_properties?: ({disk_address?: DiskAddress | undefined, device_type?: string | undefined}) | undefined, uuid?: unknown | undefined, data_source_reference?: Reference | undefined, disk_size_mib?: number | undefined, disk_size_bytes?: number | undefined};

export type VmVnumaConfig = {num_vnuma_nodes?: number | undefined};

export type IpAddress = {ip?: string | undefined, type?: string | undefined, prefix_length?: number | undefined, gateway_address_list?: Array<string> | undefined, ip_type?: string | undefined};

export type NetworkFunctionChainReference = {kind: string, name?: string | undefined, uuid: unknown};

export type SubnetReference = {kind: string, name?: string | undefined, uuid: unknown};

export type VmNic = {nic_type?: string | undefined, uuid?: unknown | undefined, ip_endpoint_list?: Array<IpAddress> | undefined, network_function_chain_reference?: NetworkFunctionChainReference | undefined, network_function_nic_type?: string | undefined, mac_address?: string | undefined, subnet_reference?: SubnetReference | undefined, model?: string | undefined, is_connected?: boolean | undefined};

export type SerialPort = {index?: number | undefined, is_connected?: boolean | undefined};

export type Credentials = {username: string, password?: string | undefined};

export type NutanixGuestToolsSpec = {state?: string | undefined, version?: string | undefined, ngt_state?: string | undefined, credentials?: Credentials | undefined, iso_mount_state?: string | undefined, enabled_capability_list?: Array<string> | undefined};

export type GuestToolsSpec = {nutanix_guest_tools?: NutanixGuestToolsSpec | undefined};

export type StorageQosPolicyConfigInput = {provisioned_iops: number, throttled_iops: number};

export type VmResources = {power_state?: string | undefined, num_vcpus_per_socket?: number | undefined, num_sockets?: number | undefined, gpu_list?: Array<VmGpu> | undefined, memory_size_mib?: number | undefined, boot_config?: VmBootConfig | undefined, hardware_clock_timezone?: string | undefined, guest_customization?: GuestCustomization | undefined, power_state_mechanism?: VmPowerStateMechanism | undefined, vga_console_enabled?: boolean | undefined, disk_list?: Array<VmDisk> | undefined, vnuma_config?: VmVnumaConfig | undefined, nic_list?: Array<VmNic> | undefined, guest_os_id?: string | undefined, serial_port_list?: Array<SerialPort> | undefined, guest_tools?: GuestToolsSpec | undefined, storage_qos_policy_config?: StorageQosPolicyConfigInput | undefined, parent_reference?: Reference | undefined};

export type ClusterReference = {kind: string, name?: string | undefined, uuid: unknown};

export type AvailabilityZoneReference = {kind: string, name?: string | undefined, uuid: unknown};

export type Vm = {description?: string | undefined, resources: VmResources, cluster_reference?: ClusterReference | undefined, name: string, availability_zone_reference?: AvailabilityZoneReference | undefined};

export type UserReference = {kind: string, name?: string | undefined, uuid: unknown};

export type VmMetadata = {last_update_time?: string | undefined, kind: string, uuid?: unknown | undefined, spec_version?: number | undefined, creation_time?: string | undefined, spec_hash?: string | undefined, should_force_translate?: boolean | undefined, owner_reference?: UserReference | undefined, categories?: {[key: string]: string} | undefined, name?: string | undefined};

export type VmIntentInput = {spec: Vm, api_version?: string | undefined, metadata: VmMetadata};

export type CreateApiVmsArgs = {body: VmIntentInput};

export type CreateApiVmsOptions = {baseUrl: string, basicAuth?: {username: string, password: string} | undefined, fetch?: typeof fetch | undefined};

export const availabilityZoneReference = z.object({kind: z.string(), name: z.string().optional(), uuid: z.string()});

export const messageResource = z.object({message: z.string(), reason: z.string(), details: z.record(z.string(), z.string()).optional()});

export const clusterReference = z.object({kind: z.string(), name: z.string().optional(), uuid: z.string()});

export const vmGpuOutputStatus = z.object({frame_buffer_size_mib: z.number().int().optional(), vendor: z.string().optional(), uuid: z.string().optional(), num_virtual_display_heads: z.number().int().optional(), pci_address: z.string().optional(), mode: z.string().optional(), fraction: z.number().int().optional(), guest_driver_version: z.string().optional(), device_id: z.number().int().optional(), name: z.string().optional()});

export const diskAddress = z.object({device_index: z.number().int(), adapter_type: z.string()});

export const vmBootDevice = z.object({disk_address: diskAddress.optional(), mac_address: z.string().optional()});

export const vmBootConfig = z.object({boot_device_order_list: z.array(z.string()).optional(), boot_device: vmBootDevice.optional()});

export const customGenericKeyValuePair = z.record(z.string(), z.string());

export const guestCustomizationStatus = z.object({cloud_init: z.object({meta_data: z.string().optional(), user_data: z.string().optional(), custom_key_values: customGenericKeyValuePair.optional()}).optional(), is_overridable: z.boolean().optional(), sysprep: z.object({install_type: z.string().optional(), unattend_xml: z.string().optional(), custom_key_values: customGenericKeyValuePair.optional()}).optional()}).and(z.record(z.string(), z.string()));

export const vmGuestPowerStateTransitionConfig = z.object({should_fail_on_script_failure: z.boolean().optional(), enable_script_exec: z.boolean().optional()});

export const vmPowerStateMechanism = z.object({guest_transition_config: vmGuestPowerStateTransitionConfig.optional(), mechanism: z.string().optional()});

export const reference = z.object({kind: z.string(), uuid: z.string(), name: z.string().optional()});

export const vmDisk = z.object({volume_group_reference: reference.optional(), device_properties: z.object({disk_address: diskAddress.optional(), device_type: z.string().optional()}).optional(), uuid: z.string().optional(), data_source_reference: reference.optional(), disk_size_mib: z.number().int().gte(1).optional(), disk_size_bytes: z.number().int().gte(1).optional()});

export const vmVnumaConfig = z.object({num_vnuma_nodes: z.number().int().optional()});

export const ipAddress = z.object({ip: z.string().optional(), type: z.string().optional(), prefix_length: z.number().int().gte(0).lte(32).optional(), gateway_address_list: z.array(z.string()).optional(), ip_type: z.string().optional()});

export const networkFunctionChainReference = z.object({kind: z.string(), name: z.string().optional(), uuid: z.string()});

export const subnetReference = z.object({kind: z.string(), name: z.string().optional(), uuid: z.string()});

export const vmNicOutputStatus = z.object({nic_type: z.string().optional(), uuid: z.string().optional(), ip_endpoint_list: z.array(ipAddress).optional(), network_function_chain_reference: networkFunctionChainReference.optional(), floating_ip: z.string().optional(), network_function_nic_type: z.string().optional(), mac_address: z.string().optional(), subnet_reference: subnetReference.optional(), model: z.string().optional(), is_connected: z.boolean().optional()});

export const serialPort = z.object({index: z.number().int().optional(), is_connected: z.boolean().optional()});

export const credentials = z.object({username: z.string(), password: z.string().optional()});

export const nutanixGuestToolsStatus = z.object({available_version: z.string().optional(), ngt_state: z.string().optional(), iso_mount_state: z.string().optional(), state: z.string().optional(), version: z.string().optional(), enabled_capability_list: z.array(z.string()).optional(), credentials: credentials.optional(), guest_os_version: z.string().optional(), vss_snapshot_capable: z.boolean().optional(), is_reachable: z.boolean().optional(), vm_mobility_drivers_installed: z.boolean().optional()});

export const guestToolsStatus = z.object({nutanix_guest_tools: nutanixGuestToolsStatus.optional()});

export const storageQosPolicyConfigOutput = z.object({provisioned_iops: z.number().int().optional(), throttled_iops: z.number().int().optional()});

export const vmResourcesDefStatus = z.object({power_state: z.string().optional(), num_vcpus_per_socket: z.number().int().optional(), num_sockets: z.number().int().optional(), gpu_list: z.array(vmGpuOutputStatus).optional(), protection_type: z.string().optional(), memory_size_mib: z.number().int().optional(), boot_config: vmBootConfig.optional(), hardware_clock_timezone: z.string().optional(), guest_customization: guestCustomizationStatus.optional(), power_state_mechanism: vmPowerStateMechanism.optional(), vga_console_enabled: z.boolean().optional(), disk_list: z.array(vmDisk).optional(), vnuma_config: vmVnumaConfig.optional(), nic_list: z.array(vmNicOutputStatus).optional(), host_reference: reference.optional(), guest_os_id: z.string().optional(), serial_port_list: z.array(serialPort).optional(), guest_tools: guestToolsStatus.optional(), hypervisor_type: z.string().optional(), storage_qos_policy_config: storageQosPolicyConfigOutput.optional(), parent_reference: reference.optional()});

export const vmDefStatus = z.object({state: z.string().optional(), name: z.string(), availability_zone_reference: availabilityZoneReference.optional(), message_list: z.array(messageResource).optional(), cluster_reference: clusterReference.optional(), resources: vmResourcesDefStatus, description: z.string().optional()});

export const vmGpu = z.object({vendor: z.string().optional(), mode: z.string().optional(), device_id: z.number().int().optional()});

export const guestCustomization = z.object({cloud_init: z.object({meta_data: z.string().max(32000).optional(), user_data: z.string().max(32000).optional(), custom_key_values: customGenericKeyValuePair.optional()}).optional(), is_overridable: z.boolean().optional(), sysprep: z.object({install_type: z.string().optional(), unattend_xml: z.string().max(32000).optional(), custom_key_values: customGenericKeyValuePair.optional()}).optional()}).and(z.record(z.string(), z.string()));

export const vmNic = z.object({nic_type: z.string().optional(), uuid: z.string().optional(), ip_endpoint_list: z.array(ipAddress).optional(), network_function_chain_reference: networkFunctionChainReference.optional(), network_function_nic_type: z.string().optional(), mac_address: z.string().optional(), subnet_reference: subnetReference.optional(), model: z.string().optional(), is_connected: z.boolean().optional()});

export const nutanixGuestToolsSpec = z.object({state: z.string().optional(), version: z.string().optional(), ngt_state: z.string().optional(), credentials: credentials.optional(), iso_mount_state: z.string().optional(), enabled_capability_list: z.array(z.string()).optional()});

export const guestToolsSpec = z.object({nutanix_guest_tools: nutanixGuestToolsSpec.optional()});

export const storageQosPolicyConfigInput = z.object({provisioned_iops: z.number().int(), throttled_iops: z.number().int()});

export const vmResources = z.object({power_state: z.string().optional(), num_vcpus_per_socket: z.number().int().gte(1).optional(), num_sockets: z.number().int().gte(1).optional(), gpu_list: z.array(vmGpu).optional(), memory_size_mib: z.number().int().gte(1).optional(), boot_config: vmBootConfig.optional(), hardware_clock_timezone: z.string().max(64).optional(), guest_customization: guestCustomization.optional(), power_state_mechanism: vmPowerStateMechanism.optional(), vga_console_enabled: z.boolean().optional(), disk_list: z.array(vmDisk).optional(), vnuma_config: vmVnumaConfig.optional(), nic_list: z.array(vmNic).optional(), guest_os_id: z.string().max(64).optional(), serial_port_list: z.array(serialPort).optional(), guest_tools: guestToolsSpec.optional(), storage_qos_policy_config: storageQosPolicyConfigInput.optional(), parent_reference: reference.optional()});

export const vm = z.object({description: z.string().max(1000).optional(), resources: vmResources, cluster_reference: clusterReference.optional(), name: z.string().max(64), availability_zone_reference: availabilityZoneReference.optional()});

export const userReference = z.object({kind: z.string(), name: z.string().optional(), uuid: z.string()});

export const vmMetadata = z.object({last_update_time: z.string().optional(), kind: z.string(), uuid: z.string().optional(), spec_version: z.number().int().optional(), creation_time: z.string().optional(), spec_hash: z.string().optional(), should_force_translate: z.boolean().optional(), owner_reference: userReference.optional(), categories: z.record(z.string(), z.string()).optional(), name: z.string().max(64).optional()});

export const vmIntentResponse = z.object({status: vmDefStatus.optional(), spec: vm.optional(), api_version: z.string(), metadata: vmMetadata});

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
};

/**
 * Create a VM
 * 
 * POST /vms
 */
export const createApiVms = async (args: CreateApiVmsArgs, options: CreateApiVmsOptions) => {
const {baseUrl, fetch: fetchFn = fetch, basicAuth} = options ?? {}

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

if (basicAuth !== undefined) headers.set('authorization', `Basic ${btoa(`${basicAuth.username}:${basicAuth.password}`)}`)

const url = `${baseUrl.replace(/\/+$/, '')}/vms${query.size > 0 ? `?${query}` : ''}`

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