import {z} from 'zod'

export type CURIE = string;

export type Attribute = {attribute_type_id: string, original_attribute_name?: (string | null) | undefined, value: unknown, value_type_id?: (CURIE | null) | undefined, attribute_source?: (string | null) | undefined, value_url?: (string | null) | undefined, description?: (string | null) | undefined, attributes?: (Array<Attribute> | null) | undefined};

export type NodeBinding = {id: CURIE, query_id?: CURIE | undefined, attributes?: (Array<Attribute> | null) | undefined} | {[key: string]: unknown};

export type EdgeBinding = {id: string, attributes?: (Array<Attribute> | null) | undefined} | {[key: string]: unknown};

export type Analysis = {resource_id: string, score?: (number | null) | undefined, edge_bindings: {[key: string]: Array<EdgeBinding>}, support_graphs?: (Array<string> | null) | undefined, scoring_method?: (string | null) | undefined, attributes?: (Array<Attribute> | null) | undefined} | {[key: string]: unknown};

export type Result = {node_bindings: {[key: string]: Array<NodeBinding>}, analyses: Array<Analysis>, id?: (string | null) | undefined, description?: (string | null) | undefined, essence?: (string | null) | undefined, essence_category?: (string | null) | undefined, row_data?: (Array<string | number> | null) | undefined, score?: (number | null) | undefined, score_name?: (string | null) | undefined, score_direction?: (string | null) | undefined, confidence?: (number | null) | undefined, result_group?: (number | null) | undefined, result_group_similarity_score?: (number | null) | undefined, resource_id?: (string | null) | undefined} | {[key: string]: unknown};

export type BiolinkEntity = string;

export type AttributeConstraint = {id: CURIE, name: string, not?: boolean | undefined, operator: '==' | '>' | '<' | 'matches' | '===', value: unknown, unit_id?: (string | null) | undefined, unit_name?: (string | null) | undefined};

export type QNode = {ids?: (Array<CURIE> | null) | undefined, categories?: (Array<BiolinkEntity> | null) | undefined, set_id?: (string | null) | undefined, is_set?: boolean | undefined, set_interpretation?: ('BATCH' | 'ALL' | 'MANY' | null) | undefined, constraints?: Array<AttributeConstraint> | undefined, option_group_id?: (string | null) | undefined} | {[key: string]: unknown};

export type BiolinkPredicate = string;

export type Qualifier = {qualifier_type_id: string, qualifier_value: string};

export type QualifierConstraint = {qualifier_set: Array<Qualifier>};

export type QEdge = {knowledge_type?: (string | null) | undefined, predicates?: (Array<BiolinkPredicate> | null) | undefined, subject: string, object: string, attribute_constraints?: Array<AttributeConstraint> | undefined, qualifier_constraints?: Array<QualifierConstraint> | undefined, exclude?: boolean | undefined, option_group_id?: (string | null) | undefined} | {[key: string]: unknown};

export type QueryGraph = {nodes: {[key: string]: QNode}, edges: {[key: string]: QEdge}} | {[key: string]: unknown};

export type Node = {name?: (string | null) | undefined, categories?: (Array<BiolinkEntity> | null) | undefined, attributes?: (Array<Attribute> | null) | undefined};

export type ResourceRoleEnum = 'primary_knowledge_source' | 'aggregator_knowledge_source' | 'supporting_data_source';

export type RetrievalSource = {resource_id: string, resource_role: ResourceRoleEnum, upstream_resource_ids?: Array<CURIE> | undefined, source_record_urls?: Array<string> | undefined} | {[key: string]: unknown};

export type Edge = {predicate: BiolinkPredicate, subject: CURIE, object: CURIE, attributes?: (Array<Attribute> | null) | undefined, qualifiers?: (Array<Qualifier> | null) | undefined, sources: Array<RetrievalSource>};

export type KnowledgeGraph = {nodes: {[key: string]: Node}, edges: {[key: string]: Edge}} | {[key: string]: unknown};

export type AuxiliaryGraph = {edges: Array<string>, attributes?: (Array<Attribute> | null) | undefined} | {[key: string]: unknown};

export type Message = {results?: (Array<Result> | null) | undefined, query_graph?: (QueryGraph | null) | undefined, knowledge_graph?: (KnowledgeGraph | null) | undefined, auxiliary_graphs?: {[key: string]: AuxiliaryGraph} | undefined};

export type LogLevel = 'ERROR' | 'WARNING' | 'INFO' | 'DEBUG';

export type OperationLookup = {id: 'lookup', parameters: unknown};

export type Operations = {message_uris?: Array<string> | undefined, messages?: Array<Message> | undefined, actions?: Array<string> | undefined, options?: {[key: string]: unknown} | undefined} | {[key: string]: unknown} | null;

export type Query = {message: Message, log_level?: (LogLevel | null) | undefined, workflow?: Array<OperationLookup> | undefined, submitter?: (string | null) | undefined, bypass_cache?: boolean | undefined, stream_progress?: boolean | undefined, enforce_edge_directionality?: boolean | undefined, return_minimal_metadata?: boolean | undefined, max_results?: number | undefined, page_size?: number | undefined, page_number?: number | undefined, operations?: (Operations | null) | undefined} | {[key: string]: unknown};

export type CreateApiQueryArgs = {body: Query};

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

export const cURIE = z.string();

export const attribute: z.ZodType<Attribute> = z.object({attribute_type_id: z.string(), original_attribute_name: z.string().nullable().optional(), value: z.unknown(), value_type_id: cURIE.nullable().optional(), attribute_source: z.string().nullable().optional(), value_url: z.string().nullable().optional(), description: z.string().nullable().optional(), attributes: z.array(z.lazy(() => attribute)).nullable().optional()});

export const nodeBinding = z.object({id: cURIE, query_id: cURIE.optional(), attributes: z.array(attribute).nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const edgeBinding = z.object({id: z.string(), attributes: z.array(attribute).nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const analysis = z.object({resource_id: z.string(), score: z.number().nullable().optional(), edge_bindings: z.record(z.string(), z.array(edgeBinding)), support_graphs: z.array(z.string()).nullable().optional(), scoring_method: z.string().nullable().optional(), attributes: z.array(attribute).nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const result = z.object({node_bindings: z.record(z.string(), z.array(nodeBinding)), analyses: z.array(analysis), id: z.string().nullable().optional(), description: z.string().nullable().optional(), essence: z.string().nullable().optional(), essence_category: z.string().nullable().optional(), row_data: z.array(z.union([z.string(), z.number()])).nullable().optional(), score: z.number().nullable().optional(), score_name: z.string().nullable().optional(), score_direction: z.string().nullable().optional(), confidence: z.number().nullable().optional(), result_group: z.number().int().nullable().optional(), result_group_similarity_score: z.number().nullable().optional(), resource_id: z.string().nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const biolinkEntity = z.string();

export const attributeConstraint = z.object({id: cURIE, name: z.string(), not: z.boolean().optional(), operator: z.enum(["==", ">", "<", "matches", "==="]), value: z.unknown(), unit_id: z.string().nullable().optional(), unit_name: z.string().nullable().optional()});

export const qNode = z.object({ids: z.array(cURIE).nullable().optional(), categories: z.array(biolinkEntity).nullable().optional(), set_id: z.string().nullable().optional(), is_set: z.boolean().optional(), set_interpretation: z.enum(["BATCH", "ALL", "MANY"]).nullable().optional(), constraints: z.array(attributeConstraint).optional(), option_group_id: z.string().nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const biolinkPredicate = z.string();

export const qualifier = z.object({qualifier_type_id: z.string(), qualifier_value: z.string()});

export const qualifierConstraint = z.object({qualifier_set: z.array(qualifier)});

export const qEdge = z.object({knowledge_type: z.string().nullable().optional(), predicates: z.array(biolinkPredicate).nullable().optional(), subject: z.string(), object: z.string(), attribute_constraints: z.array(attributeConstraint).optional(), qualifier_constraints: z.array(qualifierConstraint).optional(), exclude: z.boolean().optional(), option_group_id: z.string().nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const queryGraph = z.object({nodes: z.record(z.string(), qNode), edges: z.record(z.string(), qEdge)}).and(z.record(z.string(), z.unknown()));

export const node = z.object({name: z.string().nullable().optional(), categories: z.array(biolinkEntity).nullable().optional(), attributes: z.array(attribute).nullable().optional()});

export const resourceRoleEnum = z.enum(["primary_knowledge_source", "aggregator_knowledge_source", "supporting_data_source"]);

export const retrievalSource = z.object({resource_id: z.string(), resource_role: resourceRoleEnum, upstream_resource_ids: z.array(cURIE).optional(), source_record_urls: z.array(z.string()).optional()}).and(z.record(z.string(), z.unknown()));

export const edge = z.object({predicate: biolinkPredicate, subject: cURIE, object: cURIE, attributes: z.array(attribute).nullable().optional(), qualifiers: z.array(qualifier).nullable().optional(), sources: z.array(retrievalSource)});

export const knowledgeGraph = z.object({nodes: z.record(z.string(), node), edges: z.record(z.string(), edge)}).and(z.record(z.string(), z.unknown()));

export const auxiliaryGraph = z.object({edges: z.array(z.string()), attributes: z.array(attribute).nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const message = z.object({results: z.array(result).nullable().optional(), query_graph: queryGraph.nullable().optional(), knowledge_graph: knowledgeGraph.nullable().optional(), auxiliary_graphs: z.record(z.string(), auxiliaryGraph).optional()});

export const logLevel = z.enum(["ERROR", "WARNING", "INFO", "DEBUG"]);

export const logEntry = z.object({timestamp: z.string(), level: logLevel.nullable().optional(), code: z.string().nullable().optional(), message: z.string()}).and(z.record(z.string(), z.unknown()));

export const operationLookup = z.object({id: z.literal("lookup"), parameters: z.unknown()});

export const operations = z.object({message_uris: z.array(z.string()).optional(), messages: z.array(message).optional(), actions: z.array(z.string()).optional(), options: z.record(z.string(), z.unknown()).optional()}).and(z.record(z.string(), z.unknown())).nullable();

export const response = z.object({message: message, status: z.string().nullable().optional(), description: z.string().nullable().optional(), logs: z.array(logEntry).nullable().optional(), workflow: z.array(operationLookup).optional(), submitter: z.string().nullable().optional(), operations: operations.optional(), job_id: z.string().nullable().optional(), resource_id: z.string().nullable().optional(), tool_version: z.string().optional(), schema_version: z.string().optional(), biolink_version: z.string().optional(), datetime: z.string().optional(), table_column_names: z.array(z.string()).optional(), original_question: z.string().optional(), restated_question: z.string().optional(), query_options: z.object({}).optional(), context: z.string().optional(), type: z.string().optional(), id: z.string().optional(), validation_result: z.object({}).optional(), total_results_count: z.number().int().nullable().optional(), info: z.string().nullable().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
};

/**
 * Initiate a query and wait to receive a Response
 * 
 * POST /query
 */
export const createApiQuery = async (args: CreateApiQueryArgs, options?: CreateApiQueryOptions) => {
const {baseUrl = "https://kg2cploverdb.transltr.io", fetch: fetchFn = fetch} = options ?? {}

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

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