import {z} from 'zod'

export type GetApiTasksIdArgs = {id: string};

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

export const objectVersion = z.object({Index: z.number().int().optional()});

export const pluginPrivilege = z.object({Name: z.string().optional(), Description: z.string().optional(), Value: z.array(z.string()).optional()});

export const mount = z.object({Target: z.string().optional(), Source: z.string().optional(), Type: z.enum(["bind", "volume", "image", "tmpfs", "npipe", "cluster"]).optional(), ReadOnly: z.boolean().optional(), Consistency: z.string().optional(), BindOptions: z.object({Propagation: z.enum(["private", "rprivate", "shared", "rshared", "slave", "rslave"]).optional(), NonRecursive: z.boolean().optional(), CreateMountpoint: z.boolean().optional(), ReadOnlyNonRecursive: z.boolean().optional(), ReadOnlyForceRecursive: z.boolean().optional()}).optional(), VolumeOptions: z.object({NoCopy: z.boolean().optional(), Labels: z.record(z.string(), z.string()).optional(), DriverConfig: z.object({Name: z.string().optional(), Options: z.record(z.string(), z.string()).optional()}).optional(), Subpath: z.string().optional()}).optional(), ImageOptions: z.object({Subpath: z.string().optional()}).optional(), TmpfsOptions: z.object({SizeBytes: z.number().int().optional(), Mode: z.number().int().optional(), Options: z.array(z.array(z.string())).optional()}).optional()});

export const healthConfig = z.object({Test: z.array(z.string()).optional(), Interval: z.number().int().optional(), Timeout: z.number().int().optional(), Retries: z.number().int().optional(), StartPeriod: z.number().int().optional(), StartInterval: z.number().int().optional()});

export const limit = z.object({NanoCPUs: z.number().int().optional(), MemoryBytes: z.number().int().optional(), Pids: z.number().int().optional()});

export const genericResources = z.array(z.object({NamedResourceSpec: z.object({Kind: z.string().optional(), Value: z.string().optional()}).optional(), DiscreteResourceSpec: z.object({Kind: z.string().optional(), Value: z.number().int().optional()}).optional()}));

export const resourceObject = z.object({NanoCPUs: z.number().int().optional(), MemoryBytes: z.number().int().optional(), GenericResources: genericResources.optional()});

export const platform = z.object({Architecture: z.string().optional(), OS: z.string().optional()});

export const networkAttachmentConfig = z.object({Target: z.string().optional(), Aliases: z.array(z.string()).optional(), DriverOpts: z.record(z.string(), z.string()).optional()});

export const taskSpec = z.object({PluginSpec: z.object({Name: z.string().optional(), Remote: z.string().optional(), Disabled: z.boolean().optional(), PluginPrivilege: z.array(pluginPrivilege).optional()}).optional(), ContainerSpec: z.object({Image: z.string().optional(), Labels: z.record(z.string(), z.string()).optional(), Command: z.array(z.string()).optional(), Args: z.array(z.string()).optional(), Hostname: z.string().optional(), Env: z.array(z.string()).optional(), Dir: z.string().optional(), User: z.string().optional(), Groups: z.array(z.string()).optional(), Privileges: z.object({CredentialSpec: z.object({Config: z.string().optional(), File: z.string().optional(), Registry: z.string().optional()}).optional(), SELinuxContext: z.object({Disable: z.boolean().optional(), User: z.string().optional(), Role: z.string().optional(), Type: z.string().optional(), Level: z.string().optional()}).optional(), Seccomp: z.object({Mode: z.enum(["default", "unconfined", "custom"]).optional(), Profile: z.string().optional()}).optional(), AppArmor: z.object({Mode: z.enum(["default", "disabled"]).optional()}).optional(), NoNewPrivileges: z.boolean().optional()}).optional(), TTY: z.boolean().optional(), OpenStdin: z.boolean().optional(), ReadOnly: z.boolean().optional(), Mounts: z.array(mount).optional(), StopSignal: z.string().optional(), StopGracePeriod: z.number().int().optional(), HealthCheck: healthConfig.optional(), Hosts: z.array(z.string()).optional(), DNSConfig: z.object({Nameservers: z.array(z.string()).optional(), Search: z.array(z.string()).optional(), Options: z.array(z.string()).optional()}).optional(), Secrets: z.array(z.object({File: z.object({Name: z.string().optional(), UID: z.string().optional(), GID: z.string().optional(), Mode: z.number().int().optional()}).optional(), SecretID: z.string().optional(), SecretName: z.string().optional()})).optional(), OomScoreAdj: z.number().int().optional(), Configs: z.array(z.object({File: z.object({Name: z.string().optional(), UID: z.string().optional(), GID: z.string().optional(), Mode: z.number().int().optional()}).optional(), Runtime: z.object({}).optional(), ConfigID: z.string().optional(), ConfigName: z.string().optional()})).optional(), Isolation: z.enum(["default", "process", "hyperv", ""]).optional(), Init: z.boolean().nullable().optional(), Sysctls: z.record(z.string(), z.string()).optional(), CapabilityAdd: z.array(z.string()).optional(), CapabilityDrop: z.array(z.string()).optional(), Ulimits: z.array(z.object({Name: z.string().optional(), Soft: z.number().int().optional(), Hard: z.number().int().optional()})).optional()}).optional(), NetworkAttachmentSpec: z.object({ContainerID: z.string().optional()}).optional(), Resources: z.object({Limits: limit.optional(), Reservations: resourceObject.optional()}).optional(), RestartPolicy: z.object({Condition: z.enum(["none", "on-failure", "any"]).optional(), Delay: z.number().int().optional(), MaxAttempts: z.number().int().optional(), Window: z.number().int().optional()}).optional(), Placement: z.object({Constraints: z.array(z.string()).optional(), Preferences: z.array(z.object({Spread: z.object({SpreadDescriptor: z.string().optional()}).optional()})).optional(), MaxReplicas: z.number().int().optional(), Platforms: z.array(platform).optional()}).optional(), ForceUpdate: z.number().int().optional(), Runtime: z.string().optional(), Networks: z.array(networkAttachmentConfig).optional(), LogDriver: z.object({Name: z.string().optional(), Options: z.record(z.string(), z.string()).optional()}).optional()});

export const taskState = z.enum(["new", "allocated", "pending", "assigned", "accepted", "preparing", "ready", "starting", "running", "complete", "shutdown", "failed", "rejected", "remove", "orphaned"]);

export const containerStatus = z.object({ContainerID: z.string().optional(), PID: z.number().int().optional(), ExitCode: z.number().int().optional()});

export const endpointPortConfig = z.object({Name: z.string().optional(), Protocol: z.enum(["tcp", "udp", "sctp"]).optional(), TargetPort: z.number().int().optional(), PublishedPort: z.number().int().optional(), PublishMode: z.enum(["ingress", "host"]).optional()});

export const portStatus = z.object({Ports: z.array(endpointPortConfig).optional()});

export const taskStatus = z.object({Timestamp: z.string().optional(), State: taskState.optional(), Message: z.string().optional(), Err: z.string().optional(), ContainerStatus: containerStatus.optional(), PortStatus: portStatus.optional()});

export const task = z.object({ID: z.string().optional(), Version: objectVersion.optional(), CreatedAt: z.string().optional(), UpdatedAt: z.string().optional(), Name: z.string().optional(), Labels: z.record(z.string(), z.string()).optional(), Spec: taskSpec.optional(), ServiceID: z.string().optional(), Slot: z.number().int().optional(), NodeID: z.string().optional(), AssignedGenericResources: genericResources.optional(), Status: taskStatus.optional(), DesiredState: taskState.optional(), JobIteration: objectVersion.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
};

/**
 * Inspect a task
 * 
 * GET /tasks/{id}
 */
export const getApiTasksId = async (args: GetApiTasksIdArgs, options: GetApiTasksIdOptions) => {
const {baseUrl, fetch: fetchFn = fetch} = options ?? {}

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

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