import {z} from 'zod'

export type GetApiCollectionsCollectionNameOptimizationsArgs = {collection_name: string, with?: string | undefined, completed_limit?: number | undefined};

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

export const hardwareUsage = z.object({cpu: z.number().int().gte(0), payload_io_read: z.number().int().gte(0), payload_io_write: z.number().int().gte(0), payload_index_io_read: z.number().int().gte(0), payload_index_io_write: z.number().int().gte(0), vector_io_read: z.number().int().gte(0), vector_io_write: z.number().int().gte(0)});

export const modelUsage = z.object({tokens: z.number().int().gte(0)});

export const inferenceUsage = z.object({models: z.record(z.string(), modelUsage)});

export const usage = z.object({hardware: z.union([hardwareUsage, z.unknown()]).optional(), inference: z.union([inferenceUsage, z.unknown()]).optional()});

export const optimizationsSummary = z.object({queued_optimizations: z.number().int().gte(0), queued_segments: z.number().int().gte(0), queued_points: z.number().int().gte(0), idle_segments: z.number().int().gte(0)});

export const trackerStatus = z.union([z.enum(["optimizing", "done"]), z.object({cancelled: z.string()}), z.object({error: z.string()})]);

export const optimizationSegmentInfo = z.object({uuid: z.string(), points_count: z.number().int().gte(0)});

export const progressTree: z.ZodType<ProgressTree> = z.object({name: z.string(), started_at: z.string().nullable().optional(), finished_at: z.string().nullable().optional(), duration_sec: z.number().nullable().optional(), done: z.number().int().gte(0).nullable().optional(), total: z.number().int().gte(0).nullable().optional(), children: z.array(z.lazy(() => progressTree)).optional()});

export const optimization = z.object({uuid: z.string(), optimizer: z.string(), status: trackerStatus, segments: z.array(optimizationSegmentInfo), progress: progressTree});

export const pendingOptimization = z.object({optimizer: z.string(), segments: z.array(optimizationSegmentInfo)});

export const optimizationsResponse = z.object({summary: optimizationsSummary, running: z.array(optimization), queued: z.array(pendingOptimization).nullable().optional(), completed: z.array(optimization).nullable().optional(), idle_segments: z.array(optimizationSegmentInfo).nullable().optional()});

export const getApiCollectionsCollectionNameOptimizationsResponse = z.object({usage: z.union([usage, z.unknown()]).optional(), time: z.number().optional(), status: z.string().optional(), result: optimizationsResponse.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
};

/**
 * Get optimization progress
 * 
 * GET /collections/{collection_name}/optimizations
 * 
 * Also accepts auth: bearerAuth
 */
export const getApiCollectionsCollectionNameOptimizations = async (args: GetApiCollectionsCollectionNameOptimizationsArgs, options?: GetApiCollectionsCollectionNameOptimizationsOptions) => {
const {baseUrl = "http://localhost:6333", fetch: fetchFn = fetch, apiKey} = options ?? {}

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

if (args.with != null) query.append("with", String(args.with))
if (args.completed_limit != null) query.append("completed_limit", String(args.completed_limit))
if (apiKey !== undefined) headers.set("api-key", apiKey)

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