123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199
import type { Options } from "./options.ts";import { DataFactory, type IRI, type RDF, toRdf } from "./rdf.ts";import type { ExpandedProperty, ExpandedSchema } from "./schema/mod.ts";import { xsd } from "../namespaces/xsd.ts";import { rdf } from "../namespaces/rdf.ts";import { ldkit } from "../namespaces/ldkit.ts";
type DecodedNode = Record<string, unknown>;
type NodeId = RDF.NamedNode | RDF.BlankNode;
export const encode = (  node: DecodedNode,  schema: ExpandedSchema,  options: Options,  includeType = true,  variableInitCounter = 0,) => {  return Encoder.encode(    node,    schema,    options,    includeType,    variableInitCounter,  );};
export const encodeValue = (  value: unknown,  datatype: string,  df: DataFactory,) => {  if (datatype === ldkit.IRI) {    return df.namedNode(value as string);  }  return toRdf(value, {    datatype: df.namedNode(datatype),  });};
class Encoder {  private readonly options: Options;  private readonly includeType: boolean;
  private df: DataFactory = new DataFactory({    blankNodePrefix: "b",  });
  private variableCounter: number;
  private output: RDF.Quad[] = [];
  private constructor(    options: Options,    includeType: boolean,    variableInitCounter: number,  ) {    this.options = options;    this.includeType = includeType;    this.variableCounter = variableInitCounter;  }
  static encode(    node: DecodedNode,    schema: ExpandedSchema,    options: Options,    includeType: boolean,    variableInitCounter: number,  ) {    return new Encoder(options, includeType, variableInitCounter).encode(      node,      schema,    );  }
  encode(node: DecodedNode, schema: ExpandedSchema) {    const nodeId = this.getNodeId(node);    this.encodeNode(node, schema, nodeId);
    return this.output;  }
  push(    s: RDF.Quad_Subject,    p: RDF.Quad_Predicate,    o: RDF.Quad_Object,  ) {    this.output.push(this.df.quad(s, p, o));  }
  getNodeId(node: DecodedNode) {    return node.$id      ? this.df.namedNode(node.$id as string)      : this.df.blankNode();  }
  getNodeTypes(node: DecodedNode) {    if (Array.isArray(node.$type)) {      return node.$type;    }
    return node.$type ? [node.$type] : [];  }
  encodeNode(node: DecodedNode, schema: ExpandedSchema, nodeId: NodeId) {    if (this.includeType) {      this.encodeNodeType(node, schema["@type"], nodeId);    }
    Object.keys(schema).forEach((key) => {      if (key === "@type") {        return;      }      this.encodeNodeProperty(        node[key],        schema[key] as ExpandedProperty,        nodeId,      );    });  }
  encodeNodeType(node: DecodedNode, requiredTypes: IRI[], nodeId: NodeId) {    const finalTypes = new Set([...this.getNodeTypes(node), ...requiredTypes]);
    finalTypes.forEach((type) => {      this.push(nodeId, this.df.namedNode(rdf.type), this.df.namedNode(type));    });  }
  encodeNodeProperty(    value: unknown,    property: ExpandedProperty,    nodeId: NodeId,  ) {    if (value === undefined) {      return;    }
    const propertyId = this.df.namedNode(property["@id"]);
    if (value === null) {      // TODO      this.push(        nodeId,        propertyId,        this.df.variable(`v${this.variableCounter++}`),      );      return;    }
    if (property["@multilang"]) {      const multiValue = value as unknown as Record<string, unknown>;      Object.keys(multiValue).forEach((language) => {        const languageValue: string[] = Array.isArray(multiValue[language])          ? multiValue[language] as string[]          : [multiValue[language]] as string[];        languageValue.forEach((singleValue) => {          this.push(            nodeId,            propertyId,            this.df.literal(              singleValue,              language.length > 0 ? language : undefined,            ),          );        });      });      return;    }
    const values = Array.isArray(value) ? value : [value];
    values.forEach((val) => {      if (property["@schema"]) {        const subNodeId = this.getNodeId(val);        this.encodeNode(val, property["@schema"], subNodeId);        this.push(nodeId, propertyId, subNodeId);        return;      }
      const propertyType = property["@type"] ? property["@type"] : xsd.string;
      if (typeof val === "string" && this.options.language) {        if (propertyType === xsd.string || propertyType === rdf.langString) {          this.push(            nodeId,            propertyId,            this.df.literal(val, this.options.language),          );          return;        }      }
      const rdfValue = encodeValue(val, propertyType, this.df);      this.push(nodeId, propertyId, rdfValue);    });  }}