import {unified} from "unified";
import {toHtml} from "hast-util-to-html";
import rehypeParse from "rehype-parse";

const CALLOUT = /__(.*?)__/g;

export default function (code) {
  const html = toHtml(code);
  const result = html.replace(CALLOUT, (_, text) => `<span class="highlight-word">${text}</span>`);
  const hast = unified().use(rehypeParse, {emitParseErrors: true, fragment: true}).parse(result);

  return hast.children;
}