import {z} from 'zod'

export type LineageTopicsBatchReq = {group_by: 'source' | 'destination' | 'transform' | 'source_audit', entity_ids: Array<string>, page_size?: number | undefined, sort?: ('byteTotal' | 'latency' | 'speed' | 'lag' | 'name') | undefined, sort_dir?: ('asc' | 'desc') | undefined, topic_name?: (string | null) | undefined, filter?: (string | null) | undefined, snapshot?: (string | null) | undefined, include_metrics?: boolean | undefined, diff_bands?: (Array<string> | string | null) | undefined} | {[key: string]: unknown};

export type CreateApiDashboardLineageTopicsBatchArgs = {body: LineageTopicsBatchReq};

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

export const sourceSnapshotProgressRes = z.object({state: z.string(), snapshot_type: z.string().optional(), total_topic_count: z.number().int().optional(), remaining_topic_count: z.number().int().optional(), done_topic_count: z.number().int().optional(), cancelled_topic_count: z.number().int().optional(), failed_topic_count: z.number().int().optional(), total_table_count: z.number().int().optional(), remaining_table_count: z.number().int().optional(), rows_total: z.number().int().optional(), rows_scanned: z.number().int().optional(), progress_type: z.enum(["rows", "tables"]).optional(), percent_complete: z.number().optional(), eta_ms: z.number().int().optional(), start_timestamp: z.string().nullable().optional(), end_timestamp: z.string().nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const topicSourceRef = z.object({id: z.string(), name: z.string(), connector: z.string().optional(), snapshot_progress: sourceSnapshotProgressRes.nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const transformMetricsRes = z.object({latency: z.number().int().optional()}).and(z.record(z.string(), z.unknown()));

export const transformsLibraryEnum = z.enum(["fan_out", "sql_join", "enrich", "enrich_async", "map_filter", "toast_handling", "un_nesting", "rollup", "topic_router"]);

export const lagTrendRes = z.object({trend: z.enum(["down", "flat", "up", "none"]), lag: z.number().int(), previous_lag: z.number().int(), delta: z.number().int(), rate_per_min: z.number(), window_sec: z.number().int(), as_of: z.string(), eta_ms: z.number().int().nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const topicDestinationRef = z.object({id: z.string(), name: z.string(), connector: z.string().optional(), pipeline_id: z.string().optional(), pipeline_status: z.string().optional(), lag: z.number().int().nullable().optional(), lag_trend: lagTrendRes.nullable().optional(), latency_ms: z.number().int().nullable().optional(), qaCount: z.number().nullable().optional(), e2e_eta_ms: z.number().int().nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const topicTransformRef = z.object({id: z.string(), name: z.string(), status: z.string().optional(), metrics: transformMetricsRes.nullable().optional(), output_topics: z.array(z.string()).optional(), transform_type: transformsLibraryEnum.nullable().optional(), destinations: z.array(topicDestinationRef).optional()}).and(z.record(z.string(), z.unknown()));

export const lineageUpstreamTopicRef = z.object({name: z.string(), source: topicSourceRef}).and(z.record(z.string(), z.unknown()));

export const topicMetricsDetailRes = z.object({StreamingMilliSecondsBehindSource: z.number().int().optional(), SnapshotPercentageComplete: z.number().int().optional(), recordSendTotal: z.number().int().optional(), sourceRecordWriteTotal: z.number().int().optional(), snapshotPercentageComplete: z.number().int().optional(), recordsLag: z.number().int().optional(), byteTotal: z.number().int().optional(), sourceRecordPollRate: z.number().optional(), sourceQaCount: z.number().nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const topicSnapshotStatusRes = z.object({status: z.string(), start_timestamp: z.string().nullable().optional(), end_timestamp: z.string().nullable().optional(), duration_ms: z.number().int().nullable().optional(), rows_scanned: z.number().int().optional(), rows_total: z.number().int().optional(), trace: z.string().nullable().optional(), eta_ms: z.number().int().optional()}).and(z.record(z.string(), z.unknown()));

export const lineageTopicRes = z.object({id: z.string(), name: z.string(), producer_type: z.enum(["sources", "transforms"]).optional(), source: topicSourceRef, producer_transform: topicTransformRef.nullable().optional(), upstream_topics: z.array(lineageUpstreamTopicRef).optional(), metrics: topicMetricsDetailRes.nullable().optional(), snapshot_status: z.string().nullable().optional(), snapshot_details: z.array(topicSnapshotStatusRes).optional(), transforms: z.array(topicTransformRef).optional(), destinations: z.array(topicDestinationRef).optional()}).and(z.record(z.string(), z.unknown()));

export const embeddedTopicsRes = z.object({result: z.array(lineageTopicRes).optional(), total: z.number().int().optional(), page: z.number().int().optional(), has_next: z.boolean().optional()}).and(z.record(z.string(), z.unknown()));

export const lineageTopicsBatchRes = z.object({groups: z.record(z.string(), embeddedTopicsRes).optional()}).and(z.record(z.string(), z.unknown()));

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
};

/**
 * Batch fetch topics for multiple entities
 * 
 * POST /dashboard/lineage/topics/batch
 */
export const createApiDashboardLineageTopicsBatch = async (args: CreateApiDashboardLineageTopicsBatchArgs, options?: CreateApiDashboardLineageTopicsBatchOptions) => {
const {baseUrl = "https://api.streamkap.com", fetch: fetchFn = fetch, FronteggHTTPAuthentication} = options ?? {}

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

if (FronteggHTTPAuthentication !== undefined) headers.set('authorization', `Bearer ${FronteggHTTPAuthentication}`)

const url = `${baseUrl.replace(/\/+$/, '')}/dashboard/lineage/topics/batch${query.size > 0 ? `?${query}` : ''}`

const res = await fetchFn(url, {method: "POST", 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(lineageTopicsBatchRes, await res.json())
};