transform.mjs

import { getValue, parse } from 'markdown-code-block-meta';

import { create } from './ast.mjs';
import { fetchData, mime, toDataURL } from './utils.mjs';

/* eslint-disable no-param-reassign */

function removeXML(string) {
  return string.replace(/<\?xml.+\?>/, '');
}

const modes = {
  'img-base64': ({ node, diagramType, data, alt }) => {
    node.type = 'paragraph';
    node.children = [
      {
        type: 'image',
        _meta: { kroki: true, type: diagramType },
        alt: alt || diagramType,
        url: toDataURL(data),
      },
    ];
  },
  'object-base64': ({ target, node, diagramType, data, alt }) => {
    Object.assign(
      node,
      create(target, {
        tag: 'object',
        child: 'Load SVG fail...',
        attr: {
          type: mime,
          class: 'kroki-object',
          'data-type': diagramType,
          title: alt || diagramType,
          data: toDataURL(data),
        },
      }),
    );
  },
  'img-html-base64': ({ target, node, diagramType, data, alt }) => {
    node.type = 'paragraph';
    node.children = [
      create(target, {
        tag: 'img',
        attr: {
          class: 'kroki-image',
          alt: alt || diagramType,
          'data-type': diagramType,
          src: toDataURL(data),
        },
      }),
    ];
  },
  'inline-svg': ({ target, node, diagramType, data, alt }) => {
    Object.assign(
      node,
      create(target, {
        tag: 'p',
        attr: {
          class: 'kroki-inline-svg',
          'data-type': diagramType,
          'data-alt': alt || diagramType,
        },
        child: removeXML(data.toString()),
      }),
    );
  },
};

export const outputType = Object.keys(modes);

export async function transform({ node, server, headers, output, target }) {
  const { meta, value, lang } = node;

  const object = parse(meta);

  const alt = getValue(object.get('alt'));
  const type = getValue(object.get('type'));

  const diagramType = lang === 'kroki' ? type : lang;

  const data = await fetchData({
    server,
    headers,
    type: diagramType,
    value,
  });

  delete node.lang;
  delete node.value;
  delete node.meta;
  delete node.type;

  modes[output]({ node, diagramType, data, alt, target });
}