import {z} from 'zod'

export type AtlasVisualizationMode = 'dynamic' | 'theoretical';

export type AtlasVisualizationMetadataRollbackRequest = {index_id: string, dates: Array<string>, visualization_mode: AtlasVisualizationMode};

export type PatchApiV0AoisAoiIdVisualizationsVisualizationIdMetadataArgs = {aoi_id: string, visualization_id: string, body: AtlasVisualizationMetadataRollbackRequest};

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

export const atlasAssetType = z.enum(["mosaic", "stac_item", "visualization"]);

export const atlasPolygon = z.object({type: z.string(), coordinates: z.array(z.array(z.array(z.number())))});

export const atlasAssetStatus = z.enum(["success", "failed", "started", "creating"]);

export const atlasAsset = z.object({aoi_id: z.string().optional(), area_in_sq_m: z.number().optional(), asset_type: atlasAssetType.optional(), band_list: z.array(z.string()).optional(), centroid: z.array(z.number()).optional(), created_at: z.string().optional(), date: z.string().optional(), description: z.string().optional(), geometry: atlasPolygon.optional(), id: z.string().optional(), name: z.string().optional(), path: z.string().optional(), progress: z.number().int().optional(), resolution: z.number().optional(), status: atlasAssetStatus.optional(), updated_at: z.string().optional(), visualization_id: z.string().optional()});

export const atlasExpressionType = z.enum(["composite", "index", "grayscale"]);

export const atlasLegendItem = z.object({area: z.number().optional(), area_percentage: z.number().optional(), color: z.string().optional(), label: z.string().optional(), max: z.number().optional(), min: z.number().optional()});

export const atlasVisualizationLegends = z.object({dynamic: z.array(atlasLegendItem).optional(), theoretical: z.array(atlasLegendItem).optional()});

export const atlasVisualizationMetadata = z.object({legend: z.record(z.string(), atlasVisualizationLegends).optional()});

export const atlasVisualizationStatus = z.enum(["creating", "success", "failed", "partial_success"]);

export const atlasVisualizationType = z.literal("custom");

export const atlasVisualizationDetailResponse = z.object({aoi_id: z.string().optional(), assets: z.array(atlasAsset).optional(), created_at: z.string().optional(), expression: z.string().optional(), expression_type: atlasExpressionType.optional(), id: z.string().optional(), index_id: z.string().optional(), index_name: z.string().optional(), metadata: atlasVisualizationMetadata.optional(), progress: z.number().int().optional(), rescale: z.array(z.number().int()).optional(), status: atlasVisualizationStatus.optional(), tenant_id: z.string().optional(), type: atlasVisualizationType.optional(), updated_at: z.string().optional(), user_id: z.string().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
};

/**
 * Rollback visualization metadata
 * 
 * PATCH /v0/aois/{aoi_id}/visualizations/{visualization_id}/metadata
 */
export const patchApiV0AoisAoiIdVisualizationsVisualizationIdMetadata = async (args: PatchApiV0AoisAoiIdVisualizationsVisualizationIdMetadataArgs, options?: PatchApiV0AoisAoiIdVisualizationsVisualizationIdMetadataOptions) => {
const {baseUrl = "https://api.pixxel.space", fetch: fetchFn = fetch, ApiKeyAuth} = options ?? {}

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

if (ApiKeyAuth !== undefined) headers.set("Authorization", ApiKeyAuth)

const url = `${baseUrl.replace(/\/+$/, '')}/v0/aois/${encodeURIComponent(String(args.aoi_id))}/visualizations/${encodeURIComponent(String(args.visualization_id))}/metadata${query.size > 0 ? `?${query}` : ''}`

const res = await fetchFn(url, {method: "PATCH", headers, body: JSON.stringify(args.body)})

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

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