mermaid/packages/mermaid/src/diagrams/block/blockDB.ts

// import type { BlockDB } from './blockTypes.js';
import type { DiagramDB } from '../../diagram-api/types.js';
import { BlockConfig } from './blockTypes.js';

import * as configApi from '../../config.js';
// import common from '../common/common.js';
import {
  // setAccTitle,
  // getAccTitle,
  // getAccDescription,
  // setAccDescription,
  // setDiagramTitle,
  // getDiagramTitle,
  clear as commonClear,
} from '../../commonDb.js';

// export type TBlockColumnsDefaultValue = 'H'; // Do we support something else, like 'auto' | 0?

// TODO: Convert to generic TreeNode type? Convert to class?
export interface Block {
  ID?: string;
  label?: string;
  parent?: Block;
  children?: Block[];
  columns?: number; // | TBlockColumnsDefaultValue;
}

export interface Link {
  source: Block;
  target: Block;
}

let rootBlocks: Block[] = [];
let blocks: Block[] = [];
let links: Link[] = [];

const clear = (): void => {
  blocks = [];
  commonClear();
};

type IAddBlock = (block: Block) => Block;
const addBlock: IAddBlock = (block: Block, parent?: Block): Block => {
  if(parent) {
    parent.children ??= [];
    parent.children.push(block);
  } else {
    rootBlocks.push(block);
  }
  blocks.push(block);
  return block;
};

type IAddLink = (link: Link) => Link;
const addLink: IAddLink = (link: Link): Link => {
  links.push(link);
  return link;
};

type IGetBlocks = () => Block[];
const getBlocks:IGetBlocks = () => blocks;

type IGetLinks = () => Link[];
const getLinks:IGetLinks = () => links;

type IGetLogger = () => Console;
const getLogger:IGetLogger = () => console;

export interface BlockDB extends DiagramDB {
  clear: () => void;
  getConfig: () => BlockConfig | undefined;
  addBlock: IAddBlock;
  addLink: IAddLink;
  getLogger: IGetLogger;
  getBlocks: IGetBlocks;
  getLinks: IGetLinks;
}

const db: BlockDB = {
  getConfig: () => configApi.getConfig().block,
  addBlock: addBlock,
  addLink: addLink,
  getLogger, // TODO: remove
  getBlocks,
  getLinks,
  // getAccTitle,
  // setAccTitle,
  // getAccDescription,
  // setAccDescription,
  // getDiagramTitle,
  // setDiagramTitle,
  clear,
};

export default db;