import {z} from 'zod'

export type GetApiServicesArgs = {filters?: string | undefined, status?: boolean | undefined};

export type GetApiServicesOptions = {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 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 endpointSpec = z.object({Mode: z.enum(["vip", "dnsrr"]).optional(), Ports: z.array(endpointPortConfig).optional()});

export const serviceSpec = z.object({Name: z.string().optional(), Labels: z.record(z.string(), z.string()).optional(), TaskTemplate: taskSpec.optional(), Mode: z.object({Replicated: z.object({Replicas: z.number().int().optional()}).optional(), Global: z.object({}).optional(), ReplicatedJob: z.object({MaxConcurrent: z.number().int().optional(), TotalCompletions: z.number().int().optional()}).optional(), GlobalJob: z.object({}).optional()}).optional(), UpdateConfig: z.object({Parallelism: z.number().int().optional(), Delay: z.number().int().optional(), FailureAction: z.enum(["continue", "pause", "rollback"]).optional(), Monitor: z.number().int().optional(), MaxFailureRatio: z.number().optional(), Order: z.enum(["stop-first", "start-first"]).optional()}).optional(), RollbackConfig: z.object({Parallelism: z.number().int().optional(), Delay: z.number().int().optional(), FailureAction: z.enum(["continue", "pause"]).optional(), Monitor: z.number().int().optional(), MaxFailureRatio: z.number().optional(), Order: z.enum(["stop-first", "start-first"]).optional()}).optional(), Networks: z.array(networkAttachmentConfig).optional(), EndpointSpec: endpointSpec.optional()});

export const service = z.object({ID: z.string().optional(), Version: objectVersion.optional(), CreatedAt: z.string().optional(), UpdatedAt: z.string().optional(), Spec: serviceSpec.optional(), Endpoint: z.object({Spec: endpointSpec.optional(), Ports: z.array(endpointPortConfig).optional(), VirtualIPs: z.array(z.object({NetworkID: z.string().optional(), Addr: z.string().optional()})).optional()}).optional(), UpdateStatus: z.object({State: z.enum(["updating", "paused", "completed"]).optional(), StartedAt: z.string().optional(), CompletedAt: z.string().optional(), Message: z.string().optional()}).optional(), ServiceStatus: z.object({RunningTasks: z.number().int().optional(), DesiredTasks: z.number().int().optional(), CompletedTasks: z.number().int().optional()}).optional(), JobStatus: z.object({JobIteration: objectVersion.optional(), LastExecution: z.string().optional()}).optional()});

export const getApiServicesResponse = z.array(service);

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 services
 * 
 * GET /services
 */
export const getApiServices = async (args: GetApiServicesArgs, options: GetApiServicesOptions) => {
const {baseUrl, fetch: fetchFn = fetch} = options ?? {}

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

if (args.filters != null) query.append("filters", String(args.filters))
if (args.status != null) query.append("status", String(args.status))

const url = `${baseUrl.replace(/\/+$/, '')}/services${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(getApiServicesResponse, await res.json())
};