excel-builder-vanilla/src/Excel/RelationshipManager.ts

An easy way of building Excel files with javascript

import { uniqueId } from '../utilities/uniqueId.js';
import { Paths } from './Paths.js';
import { Util } from './Util.js';

type Relation = {
  [id: string]: {
    id: string;
    schema: string;
    object: any;
    data?: {
      id: number;
      schema: string;
      object: any;
    };
  };
};

/**
 * @module Excel/RelationshipManager
 */
export class RelationshipManager {
  relations: Relation = {};
  lastId = 1;

  constructor() {
    uniqueId('rId'); // priming
  }

  importData(data: { relations: Relation; lastId: number }) {
    this.relations = data.relations;
    this.lastId = data.lastId;
  }

  exportData() {
    return {
      relations: this.relations,
      lastId: this.lastId,
    };
  }

  addRelation(object: { id: string }, type: keyof typeof Util.schemas) {
    this.relations[object.id] = {
      id: uniqueId('rId'),
      schema: Util.schemas[type],
      object,
    };
    return this.relations[object.id].id;
  }

  getRelationshipId(object: { id: string }) {
    return this.relations[object.id] ? this.relations[object.id].id : null;
  }

  toXML() {
    const doc = Util.createXmlDoc(Util.schemas.relationshipPackage, 'Relationships');
    const relationships = doc.documentElement;

    for (const [id, data] of Object.entries(this.relations)) {
      const relationship = Util.createElement(doc, 'Relationship', [
        ['Id', data.id],
        ['Type', data.schema],
        ['Target', data.object.target || Paths[id]],
      ]);
      if (data.object.targetMode) {
        relationship.setAttribute('TargetMode', data.object.targetMode);
      }
      relationships.appendChild(relationship);
    }
    return doc;
  }
}