import {z} from 'zod'

export type LineageGroupSortEnum = 'name' | 'topic_count' | 'max_latency' | 'total_speed' | 'error_count';

export type GetApiDashboardLineageArgs = {group_by?: ('source' | 'destination' | 'transform' | 'source_audit') | undefined, search?: (string | null) | undefined, sort?: (LineageGroupSortEnum | null) | undefined, sort_dir?: ('asc' | 'desc') | undefined, page?: number | undefined, page_size?: number | undefined, include_metrics?: boolean | undefined, topic_name?: (string | null) | undefined, source_ids?: (string | null) | undefined, destination_ids?: (string | null) | undefined, transform_ids?: (string | null) | undefined, include_topics?: boolean | undefined, topics_page_size?: number | undefined, topics_sort?: ('byteTotal' | 'latency' | 'speed' | 'lag' | 'name') | undefined, topics_sort_dir?: ('asc' | 'desc') | undefined, diff_bands?: (string | null) | undefined, latency?: (string | null) | undefined, lag?: (string | null) | undefined, snapshot?: (string | null) | undefined};

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

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 lineageGroupMetricsRes = z.object({max_latency_ms: z.number().int().optional(), total_speed_rps: z.number().optional(), total_lag: z.number().int().optional(), lag_trend: lagTrendRes.nullable().optional(), max_dest_latency_ms: z.number().int().optional(), topic_count: z.number().int().optional(), error_count: z.number().int().optional()}).and(z.record(z.string(), z.unknown()));

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 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 lineageGroupRes = z.object({id: z.string(), entity_id: z.string(), name: z.string(), type: z.string(), connector: z.string().optional(), status: z.string().optional(), topic_count: z.number().int().optional(), total_topic_count: z.number().int().optional(), metrics: lineageGroupMetricsRes.nullable().optional(), snapshot_progress: sourceSnapshotProgressRes.nullable().optional(), last_audit_at: z.string().nullable().optional(), topics: embeddedTopicsRes.nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const dashboardLineageRes = 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(lineageGroupRes).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 groups
 * 
 * GET /dashboard/lineage
 */
export const getApiDashboardLineage = async (args: GetApiDashboardLineageArgs, options?: GetApiDashboardLineageOptions) => {
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.group_by != null) query.append("group_by", String(args.group_by))
if (args.search != null) query.append("search", String(args.search))
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.page != null) query.append("page", String(args.page))
if (args.page_size != null) query.append("page_size", String(args.page_size))
if (args.include_metrics != null) query.append("include_metrics", String(args.include_metrics))
if (args.topic_name != null) query.append("topic_name", String(args.topic_name))
if (args.source_ids != null) query.append("source_ids", String(args.source_ids))
if (args.destination_ids != null) query.append("destination_ids", String(args.destination_ids))
if (args.transform_ids != null) query.append("transform_ids", String(args.transform_ids))
if (args.include_topics != null) query.append("include_topics", String(args.include_topics))
if (args.topics_page_size != null) query.append("topics_page_size", String(args.topics_page_size))
if (args.topics_sort != null) query.append("topics_sort", String(args.topics_sort))
if (args.topics_sort_dir != null) query.append("topics_sort_dir", String(args.topics_sort_dir))
if (args.diff_bands != null) query.append("diff_bands", String(args.diff_bands))
if (args.latency != null) query.append("latency", String(args.latency))
if (args.lag != null) query.append("lag", String(args.lag))
if (args.snapshot != null) query.append("snapshot", String(args.snapshot))
if (FronteggHTTPAuthentication !== undefined) headers.set('authorization', `Bearer ${FronteggHTTPAuthentication}`)

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