import type { ObjectNode, GraphNode } from "@ldo/rdf-utils";
import { namedNode } from "@rdfjs/data-model";
import { getSubjectProxyFromObject } from "./subjectProxy/isSubjectProxy";
import type { ObjectLike } from "./types";
import {
  _getUnderlyingDataset,
  _getUnderlyingMatch,
  _getUnderlyingNode,
  _proxyContext,
} from "./types";
import type { LdSet } from "./setProxy/ldSet/LdSet";
import { getNodeFromRawValue } from "./util/getNodeFromRaw";

/**
 * Returns the graph for which a defined triple is a member
 * @param subject A JsonldDatasetProxy that represents the subject
 * @param predicate The key on the JsonldDatasetProxy
 * @param object The direct object. This can be a JsonldDatasetProxy. This field
 * is optional.
 * @returns a list of graphs for which the triples are members
 */
export function graphOf<Subject extends ObjectLike, Key extends keyof Subject>(
  subject: Subject,
  predicate: Key,
  object?: NonNullable<Subject[Key]> extends LdSet<infer T> ? T : Subject[Key],
): GraphNode[] {
  const subjectProxy = getSubjectProxyFromObject(subject);
  const proxyContext = subjectProxy[_proxyContext];
  const subjectNode = subjectProxy[_getUnderlyingNode];
  const predicateNode = namedNode(
    proxyContext.contextUtil.keyToIri(
      predicate as string,
      proxyContext.getRdfType(subjectNode),
    ),
  );
  let objectNode: ObjectNode | undefined | null;
  if (object == null) {
    objectNode = null;
  } else {
    const datatype = proxyContext.contextUtil.getDataType(
      predicate as string,
      proxyContext.getRdfType(subjectNode),
    );
    objectNode = getNodeFromRawValue(object, proxyContext, datatype);
  }
  const quads = subjectProxy[_getUnderlyingDataset].match(
    subjectNode,
    predicateNode,
    objectNode,
  );
  return quads.toArray().map((quad): GraphNode => quad.graph as GraphNode);
}