import {z} from 'zod'

export type GetApiDashboardLineageTopicsArgs = {source_id?: (string | null) | undefined, destination_id?: (string | null) | undefined, transform_id?: (string | null) | undefined, topic_name?: (string | null) | undefined, topic_id?: (string | null) | undefined, filter?: (string | null) | undefined, include_dlq?: boolean | undefined, snapshot?: (string | null) | undefined, page?: number | undefined, page_size?: number | undefined, sort?: ('byteTotal' | 'latency' | 'speed' | 'lag' | 'name') | undefined, sort_dir?: ('asc' | 'desc') | undefined, include_metrics?: boolean | undefined};

export type GetApiDashboardLineageTopicsOptions = {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 dashboardLineageTopicsRes = z.object({page: z.number().int().optional(), page_size: z.number().int().optional(), total: z.number().int().nullable().optional(), has_next: z.boolean().nullable().optional(), result: z.array(lineageTopicRes).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
};

/**
 * Get dashboard lineage topics
 * 
 * GET /dashboard/lineage/topics
 */
export const getApiDashboardLineageTopics = async (args: GetApiDashboardLineageTopicsArgs, options?: GetApiDashboardLineageTopicsOptions) => {
const {baseUrl = "https://api.streamkap.com", fetch: fetchFn = fetch, FronteggHTTPAuthentication} = options ?? {}

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

if (args.source_id != null) query.append("source_id", String(args.source_id))
if (args.destination_id != null) query.append("destination_id", String(args.destination_id))
if (args.transform_id != null) query.append("transform_id", String(args.transform_id))
if (args.topic_name != null) query.append("topic_name", String(args.topic_name))
if (args.topic_id != null) query.append("topic_id", String(args.topic_id))
if (args.filter != null) query.append("filter", String(args.filter))
if (args.include_dlq != null) query.append("include_dlq", String(args.include_dlq))
if (args.snapshot != null) query.append("snapshot", String(args.snapshot))
if (args.page != null) query.append("page", String(args.page))
if (args.page_size != null) query.append("page_size", String(args.page_size))
if (args.sort != null) query.append("sort", String(args.sort))
if (args.sort_dir != null) query.append("sort_dir", String(args.sort_dir))
if (args.include_metrics != null) query.append("include_metrics", String(args.include_metrics))
if (FronteggHTTPAuthentication !== undefined) headers.set('authorization', `Bearer ${FronteggHTTPAuthentication}`)

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