html_renderer.js

const SPAN_CLOSE = '</span>';

import {escapeHTML} from './utils';

export default class HTMLRenderer {
  constructor(tree, options) {
    this.buffer = "";
    this.classPrefix = options.classPrefix;
    tree.walk(this);
  }

  // renderer API

  addText(text) {
    this.buffer += escapeHTML(text)
  }

  openNode(node) {
    let className = node.kind;
    if (!node.kind) return;

    if (!node.sublanguage)
      className = `${this.classPrefix}${className}`;
    this.span(className);
  }

  closeNode(node) {
    if (!node.kind) return;

    this.buffer += SPAN_CLOSE;
  }

  // helpers

  span(className) {
    this.buffer += `<span class="${className}">`
  }

  value() {
    return this.buffer;
  }
}