mermaid/packages/mermaid-example-diagram/src/mermaidUtils.ts

const warning = () => null;

export type LogLevel = 'trace' | 'debug' | 'info' | 'warn' | 'error' | 'fatal';

export const LEVELS: Record<LogLevel, number> = {
  trace: 0,
  debug: 1,
  info: 2,
  warn: 3,
  error: 4,
  fatal: 5,
};

export const log: Record<keyof typeof LEVELS, typeof console.log> = {
  trace: warning,
  debug: warning,
  info: warning,
  warn: warning,
  error: warning,
  fatal: warning,
};
export let setLogLevel: (level: keyof typeof LEVELS | number | string) => void;
export let getConfig: () => object;
export let sanitizeText: (str: string) => string;
/**
 * Placeholder for the real function that will be injected by mermaid.
 */
// eslint-disable @typescript-eslint/no-explicit-any
export let setupGraphViewbox: (
  graph: any,
  svgElem: any,
  padding: any,
  useMaxWidth: boolean
) => void;

/**
 * Function called by mermaid that injects utility functions that help the diagram to be a good citizen.
 *
 * @param _log - log from mermaid/src/diagramAPI.ts
 * @param _setLogLevel - setLogLevel from mermaid/src/diagramAPI.ts
 * @param _getConfig - getConfig from mermaid/src/diagramAPI.ts
 * @param _sanitizeText - sanitizeText from mermaid/src/diagramAPI.ts
 * @param _setupGraphViewbox - setupGraphViewbox from mermaid/src/diagramAPI.ts
 */
export const injectUtils = (
  _log: Record<keyof typeof LEVELS, typeof console.log>,
  _setLogLevel: typeof setLogLevel,
  _getConfig: typeof getConfig,
  _sanitizeText: typeof sanitizeText,
  _setupGraphViewbox: typeof setupGraphViewbox
) => {
  _log.debug('Mermaid utils injected into example-diagram');
  log.trace = _log.trace;
  log.debug = _log.debug;
  log.info = _log.info;
  log.warn = _log.warn;
  log.error = _log.error;
  log.fatal = _log.fatal;
  setLogLevel = _setLogLevel;
  getConfig = _getConfig;
  sanitizeText = _sanitizeText;
  setupGraphViewbox = _setupGraphViewbox;
};