import {z} from 'zod'

export type GetApiInstanceArgs = {'manager-id'?: string | undefined, 'manager-type'?: 'instance-pool' | undefined, 'ip-address'?: string | undefined, labels?: string | undefined};

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

export const publicIpAssignment = z.enum(["inet4", "dual", "none"]);

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

export const securityGroupResource = z.object({id: z.string().optional(), name: z.string().min(1).max(255).optional(), visibility: z.enum(["private", "public"]).optional()});

export const securityGroupRule = z.object({description: z.string().max(255).optional(), "start-port": z.number().int().gte(1).lte(65535).optional(), protocol: z.enum(["tcp", "esp", "icmp", "udp", "gre", "ah", "ipip", "icmpv6"]).optional(), icmp: z.object({code: z.number().int().gte(-1).lte(254).optional(), type: z.number().int().gte(-1).lte(254).optional()}).optional(), "end-port": z.number().int().gte(1).lte(65535).optional(), "security-group": securityGroupResource.optional(), id: z.string().optional(), network: z.string().optional(), "flow-direction": z.enum(["ingress", "egress"]).optional()});

export const securityGroup = z.object({id: z.string().optional(), name: z.string().min(1).max(255).optional(), description: z.string().max(255).optional(), "external-sources": z.array(z.string()).optional(), rules: z.array(securityGroupRule).optional()});

export const zoneName = z.enum(["ch-dk-2", "de-muc-1", "ch-gva-2", "at-vie-1", "de-fra-1", "bg-sof-1", "at-vie-2", "hr-zag-1"]);

export const instanceType = z.object({id: z.string().optional(), size: z.enum(["large", "huge", "jumbo", "medium", "mega", "small", "extra-large", "titan48c", "titan", "micro", "colossus", "tiny"]).optional(), family: z.enum(["gpu3", "gpua30", "gpu3080ti", "gpu2", "gpu", "memory", "gpua5000", "gpub300", "gpurtx6000pro", "storage", "standard", "colossus", "cpu"]).optional(), cpus: z.number().int().gt(0).optional(), gpus: z.number().int().gt(0).optional(), authorized: z.boolean().optional(), memory: z.number().int().gt(0).optional(), zones: z.array(zoneName).optional()});

export const template = z.object({"application-consistent-snapshot-enabled": z.boolean().optional(), maintainer: z.string().optional(), description: z.string().max(255).optional(), "ssh-key-enabled": z.boolean().optional(), family: z.string().optional(), name: z.string().min(1).max(255).optional(), "default-user": z.string().min(1).max(255).optional(), size: z.number().int().gt(0).optional(), "password-enabled": z.boolean().optional(), build: z.string().optional(), checksum: z.string().optional(), "boot-mode": z.enum(["legacy", "uefi"]).optional(), id: z.string().optional(), zones: z.array(zoneName).optional(), url: z.string().optional(), version: z.string().optional(), "created-at": z.string().optional(), visibility: z.enum(["private", "public"]).optional()});

export const instanceState = z.enum(["expunging", "starting", "destroying", "running", "stopping", "stopped", "migrating", "error", "destroyed"]);

export const sshKey = z.object({name: z.string().min(1).max(255).optional(), fingerprint: z.string().optional()});

export const manager = z.object({id: z.string().optional(), type: z.enum(["sks-nodepool", "instance-pool"]).optional()});

export const getApiInstanceResponse = z.object({instances: z.array(z.object({"public-ip-assignment": publicIpAssignment.optional(), labels: labels.optional(), "security-groups": z.array(securityGroup).optional(), name: z.string().min(1).max(255).optional(), "instance-type": instanceType.optional(), "private-networks": z.array(z.object({id: z.string().optional(), "mac-address": z.string().optional()})).optional(), template: template.optional(), state: instanceState.optional(), "ssh-key": sshKey.optional(), "mac-address": z.string().optional(), manager: manager.optional(), "ipv6-address": z.string().optional(), id: z.string().optional(), "ssh-keys": z.array(sshKey).optional(), "created-at": z.string().optional(), "public-ip": z.string().optional()})).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
};

/**
 * List Compute instances
 * 
 * GET /instance
 */
export const getApiInstance = async (args: GetApiInstanceArgs, options?: GetApiInstanceOptions) => {
const {baseUrl = "https://api-ch-gva-2.exoscale.com/v2", fetch: fetchFn = fetch} = options ?? {}

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

if (args['manager-id'] != null) query.append("manager-id", String(args['manager-id']))
if (args['manager-type'] != null) query.append("manager-type", String(args['manager-type']))
if (args['ip-address'] != null) query.append("ip-address", String(args['ip-address']))
if (args.labels != null) query.append("labels", String(args.labels))

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

const res = await fetchFn(url, {method: "GET", headers})

if (!res.ok) throw new Error(`Request failed with status ${res.status}`, {cause: {status: res.status, body: await res.text()}})

return validateResponse(getApiInstanceResponse, await res.json())
};