import {z} from 'zod'

export type GetApiMarbleV1OperationsOperationIdArgs = {operation_id: string};

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

export const operationCostLineItem = z.object({credits: z.number().int(), name: z.string()});

export const operationCost = z.object({line_items: z.array(operationCostLineItem).optional(), total_credits: z.number().int()});

export const operationError = z.object({code: z.number().int().nullable().optional(), message: z.string().nullable().optional()});

export const imageryAssets = z.object({pano_url: z.string().nullable().optional()});

export const meshAssets = z.object({collider_mesh_url: z.string().nullable().optional(), full_res_mesh_url: z.string().nullable().optional(), hq_mesh_url: z.string().nullable().optional()});

export const worldSemanticsMetadata = z.object({ground_plane_offset: z.number().nullable().optional(), metric_scale_factor: z.number().nullable().optional()});

export const splatAssets = z.object({semantics_metadata: worldSemanticsMetadata.nullable().optional(), spz_urls: z.record(z.string(), z.string()).nullable().optional()});

export const worldAssets = z.object({caption: z.string().nullable().optional(), imagery: imageryAssets.nullable().optional(), mesh: meshAssets.nullable().optional(), splats: splatAssets.nullable().optional(), thumbnail_url: z.string().nullable().optional()});

export const permission = z.object({allow_id_access: z.boolean().optional(), allowed_readers: z.array(z.string()).optional(), allowed_writers: z.array(z.string()).optional(), public: z.boolean().optional()});

export const worldTextPromptOutput = z.object({text_prompt: z.string().nullable().optional(), type: z.literal("text").optional()});

export const content = z.object({data_base64: z.string().nullable().optional(), extension: z.string().nullable().optional(), uri: z.string().nullable().optional()});

export const prompt = z.object({image_prompt: content, is_pano: z.boolean().optional(), text_prompt: z.string().nullable().optional(), type: z.literal("image").optional()});

export const sphericallyLocatedContentOutput = z.object({azimuth: z.number().nullable().optional(), data_base64: z.string().nullable().optional(), extension: z.string().nullable().optional(), uri: z.string().nullable().optional()});

export const multiImagePromptOutput = z.object({multi_image_prompt: z.array(sphericallyLocatedContentOutput), reconstruct_images: z.boolean().optional(), text_prompt: z.string().nullable().optional(), type: z.literal("multi-image").optional()});

export const videoPromptOutput = z.object({text_prompt: z.string().nullable().optional(), type: z.literal("video").optional(), video_prompt: content});

export const depthPanoPrompt = z.object({depth_pano_image: content, text_prompt: z.string().nullable().optional(), type: z.literal("depth-pano").optional(), z_max: z.number().nullable().optional(), z_min: z.number().nullable().optional()});

export const inpaintPanoPrompt = z.object({pano_image: content, pano_mask: content, text_prompt: z.string().nullable().optional(), type: z.literal("inpaint-pano").optional()});

export const world = z.object({assets: worldAssets.nullable().optional(), created_at: z.string().nullable().optional(), display_name: z.string(), model: z.string().nullable().optional(), permission: permission.nullable().optional(), tags: z.array(z.string()).nullable().optional(), updated_at: z.string().nullable().optional(), world_id: z.string(), world_marble_url: z.string(), world_prompt: z.union([worldTextPromptOutput, prompt, multiImagePromptOutput, videoPromptOutput, depthPanoPrompt, inpaintPanoPrompt]).nullable().optional()});

export const panoDepthToRgbResult = z.object({pano_url: z.string().nullable().optional()});

export const exportWorldResult = z.object({asset_type: z.enum(["splats", "mesh"]), format: z.enum(["ply", "glb"]), mesh_variant: z.enum(["textured", "vertex_colored"]).nullable().optional(), resolution: z.string().nullable().optional(), url: z.string()});

export const getOperationResponseUnionWorldPanoDepthToRgbResultExportWorldResult = z.object({cost: operationCost.nullable().optional(), created_at: z.string().nullable().optional(), done: z.boolean(), error: operationError.nullable().optional(), expires_at: z.string().nullable().optional(), metadata: z.record(z.string(), z.unknown()).nullable().optional(), operation_id: z.string(), response: z.union([world, panoDepthToRgbResult, exportWorldResult]).nullable().optional(), updated_at: z.string().nullable().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 Operation
 * 
 * GET /marble/v1/operations/{operation_id}
 */
export const getApiMarbleV1OperationsOperationId = async (args: GetApiMarbleV1OperationsOperationIdArgs, options?: GetApiMarbleV1OperationsOperationIdOptions) => {
const {baseUrl = "https://api.worldlabs.ai", fetch: fetchFn = fetch, ApiKeyAuth} = options ?? {}

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

if (ApiKeyAuth !== undefined) headers.set("WLT-Api-Key", ApiKeyAuth)

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