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, is_set?: boolean | undefined, set_id?: (string | null) | 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 OperationAnnotate = {id: 'annotate', parameters: unknown, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationAnnotateEdgesParameters = {attributes?: Array<string> | undefined};

export type OperationAnnotateEdges = {id: 'annotate_edges', parameters?: OperationAnnotateEdgesParameters | undefined, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationAnnotateNodesParameters = {attributes?: Array<string> | undefined};

export type OperationAnnotateNodes = {id: 'annotate_nodes', parameters?: OperationAnnotateNodesParameters | undefined, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationBind = {id: 'bind', parameters: unknown, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationCompleteResults = {id: 'complete_results', parameters: unknown, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationEnrichResultsParameters = {pvalue_threshold?: number | undefined, qnode_keys?: Array<string> | undefined};

export type OperationEnrichResults = {id: 'enrich_results', parameters?: OperationEnrichResultsParameters | undefined, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationFill = {id: 'fill', parameters?: ({allowlist?: Array<string> | undefined, qedge_keys?: Array<string> | undefined} | {denylist?: Array<string> | undefined, qedge_keys?: Array<string> | undefined}) | undefined, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationFilterKgraph = {id: 'filter_kgraph', parameters: unknown, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationFilterKgraphContinuousKedgeAttributeParameters = {edge_attribute: string, qedge_keys?: Array<string> | undefined, qnode_keys?: Array<string> | undefined, remove_above_or_below: 'above' | 'below', threshold: number};

export type OperationFilterKgraphContinuousKedgeAttribute = {id: 'filter_kgraph_continuous_kedge_attribute', parameters: OperationFilterKgraphContinuousKedgeAttributeParameters, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationFilterKgraphDiscreteKedgeAttributeParameters = {edge_attribute: string, qedge_keys?: Array<string> | undefined, qnode_keys?: Array<string> | undefined, remove_value: string};

export type OperationFilterKgraphDiscreteKedgeAttribute = {id: 'filter_kgraph_discrete_kedge_attribute', parameters: OperationFilterKgraphDiscreteKedgeAttributeParameters, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationFilterKgraphDiscreteKnodeAttributeParameters = {node_attribute: string, qnode_keys?: Array<string> | undefined, remove_value: string};

export type OperationFilterKgraphDiscreteKnodeAttribute = {id: 'filter_kgraph_discrete_knode_attribute', parameters: OperationFilterKgraphDiscreteKnodeAttributeParameters, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationFilterKgraphOrphans = {id: 'filter_kgraph_orphans', parameters: unknown, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationFilterKgraphPercentileParameters = {edge_attribute: string, qedge_keys?: Array<string> | undefined, qnode_keys?: Array<string> | undefined, remove_above_or_below?: ('above' | 'below') | undefined, threshold?: number | undefined};

export type OperationFilterKgraphPercentile = {id: 'filter_kgraph_percentile', parameters: OperationFilterKgraphPercentileParameters, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationFilterKgraphStdDevParameters = {edge_attribute: string, num_sigma?: number | undefined, plus_or_minus_std_dev?: ('plus' | 'minus') | undefined, qedge_keys?: Array<string> | undefined, qnode_keys?: Array<string> | undefined, remove_above_or_below?: ('above' | 'below') | undefined};

export type OperationFilterKgraphStdDev = {id: 'filter_kgraph_std_dev', parameters: OperationFilterKgraphStdDevParameters, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationFilterKgraphTopNParameters = {edge_attribute: string, keep_top_or_bottom?: ('top' | 'bottom') | undefined, max_edges?: number | undefined, qedge_keys?: Array<string> | undefined, qnode_keys?: Array<string> | undefined};

export type OperationFilterKgraphTopN = {id: 'filter_kgraph_top_n', parameters: OperationFilterKgraphTopNParameters, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationFilterResults = {id: 'filter_results', parameters: unknown, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationFilterResultsTopNParameters = {max_results: number};

export type OperationFilterResultsTopN = {id: 'filter_results_top_n', parameters: OperationFilterResultsTopNParameters, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationLookup = {id: 'lookup', parameters: unknown, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationOverlay = {id: 'overlay', parameters: unknown, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationOverlayComputeJaccardParameters = {end_node_keys: Array<string>, intermediate_node_key: string, virtual_relation_label: string};

export type OperationOverlayComputeJaccard = {id: 'overlay_compute_jaccard', parameters: OperationOverlayComputeJaccardParameters, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationOverlayComputeNgdParameters = {qnode_keys: Array<string>, virtual_relation_label: string};

export type OperationOverlayComputeNgd = {id: 'overlay_compute_ngd', parameters: OperationOverlayComputeNgdParameters, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationOverlayConnectKnodes = {id: 'overlay_connect_knodes', parameters: unknown, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationOverlayFisherExactTestParameters = {object_qnode_key: string, rel_edge_key?: string | undefined, subject_qnode_key: string, virtual_relation_label: string};

export type OperationOverlayFisherExactTest = {id: 'overlay_fisher_exact_test', parameters: OperationOverlayFisherExactTestParameters, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationRestate = {id: 'restate', parameters: unknown, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationScore = {id: 'score', parameters: unknown, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationSortResults = {id: 'sort_results', parameters: unknown, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationSortResultsEdgeAttributeParameters = {ascending_or_descending: 'ascending' | 'descending', edge_attribute: string, qedge_keys?: Array<string> | undefined};

export type OperationSortResultsEdgeAttribute = {id: 'sort_results_edge_attribute', parameters: OperationSortResultsEdgeAttributeParameters, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationSortResultsNodeAttributeParameters = {ascending_or_descending: 'ascending' | 'descending', node_attribute: string, qnode_keys?: Array<string> | undefined};

export type OperationSortResultsNodeAttribute = {id: 'sort_results_node_attribute', parameters: OperationSortResultsNodeAttributeParameters, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type OperationSortResultsScoreParameters = {ascending_or_descending: 'ascending' | 'descending'};

export type OperationSortResultsScore = {id: 'sort_results_score', parameters: OperationSortResultsScoreParameters, runner_parameters?: ({allowlist?: Array<string> | undefined} | {denylist?: Array<string> | undefined}) | undefined, unique?: boolean | undefined};

export type AsyncQuery = {callback: string, message: Message, log_level?: (LogLevel | null) | undefined, workflow?: (Array<OperationAnnotate | OperationAnnotateEdges | OperationAnnotateNodes | OperationBind | OperationCompleteResults | OperationEnrichResults | OperationFill | OperationFilterKgraph | OperationFilterKgraphContinuousKedgeAttribute | OperationFilterKgraphDiscreteKedgeAttribute | OperationFilterKgraphDiscreteKnodeAttribute | OperationFilterKgraphOrphans | OperationFilterKgraphPercentile | OperationFilterKgraphStdDev | OperationFilterKgraphTopN | OperationFilterResults | OperationFilterResultsTopN | OperationLookup | OperationOverlay | OperationOverlayComputeJaccard | OperationOverlayComputeNgd | OperationOverlayConnectKnodes | OperationOverlayFisherExactTest | OperationRestate | OperationScore | OperationSortResults | OperationSortResultsEdgeAttribute | OperationSortResultsNodeAttribute | OperationSortResultsScore> | null) | undefined, submitter?: (string | null) | undefined, bypass_cache?: boolean | undefined} | {[key: string]: unknown};

export type CreateApiAsyncqueryArgs = {body: AsyncQuery};

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

export const asyncQueryResponse = z.object({status: z.string().nullable().optional(), description: z.string().nullable().optional(), job_id: z.string()}).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 with a callback to receive the response
 * 
 * POST /asyncquery
 */
export const createApiAsyncquery = async (args: CreateApiAsyncqueryArgs, options?: CreateApiAsyncqueryOptions) => {
const {baseUrl = "https://arax.transltr.io/api/arax/v1.4", 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(/\/+$/, '')}/asyncquery${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(asyncQueryResponse, await res.json())
};