bolt/src/emitter.ts

import { Syntax, SyntaxKind, kindToString } from "./ast"

export class Emitter {

  emit(node: Syntax) {

    let out = '';

    switch (node.kind) {

      case SyntaxKind.JSReferenceExpression:
        out += node.name;
        break;

      case SyntaxKind.JSConstantExpression:
        if (typeof node.value === 'string') {
          out += '"' + node.value + '"';
        } else if (typeof node.value === 'bigint') {
          out += node.value.toString();
        } else {
          throw new Error(`Could not emit the value of a specific JSConstantExpression.`);
        }
        break;

      case SyntaxKind.JSFunctionDeclaration:
        out += 'function ' + node.name.text + '(';
        //out += node.params.map(p => this.emit(p)).join(', ');
        out += ') {\n';
        out += '}\n\n'
        break;

      case SyntaxKind.JSCallExpression:
        out += this.emit(node.operator) + '(';
        out += node.operands.map(op => this.emit(op)).join(', ');
        out += ')'
        break;

      case SyntaxKind.JSSourceFile:
        for (const element of node.elements) {
          out += this.emit(element);
        }
        break;

      default:
        throw new Error(`Could not emit source code for ${kindToString(node.kind)}`)

    }

    return out;

  }
}

/**
 * A wrapper around `Emitter` for quick emission of AST nodes with sane defaults.
 */
export function emit(node: Syntax) {
  const emitter = new Emitter();
  return emitter.emit(node);
}