assemblyscript-prettier/src/plugin.js

prettier for assemblyscript

import pluginTypescript from "prettier/plugins/typescript";
import { magic, preProcess } from "./replace.js";
import { builders } from "prettier/doc";

let as_estree = {};

async function initPrinter(jsPlugin) {
  let estree = jsPlugin.printers.estree;
  estree = typeof estree == "function" ? await estree() : estree;
  Object.assign(as_estree, {
    ...estree,
    printComment(commentPath, options) {
      let comment = commentPath.getValue().value;
      if (comment.startsWith(magic) && comment.endsWith(magic)) {
        let doc = [];
        if (commentPath.stack[commentPath.stack.length - 2] == 0) {
          doc.push(builders.hardline);
        }
        doc.push(comment.slice(magic.length, -magic.length));
        return doc;
      } else {
        return jsPlugin.printers.estree.printComment(commentPath, options);
      }
    },
  });
}

async function parse(text, options) {
  await initPrinter(options.plugins.find((plugin) => plugin.printers && plugin.printers.estree));
  let ast = pluginTypescript.parsers.typescript.parse(text, options);
  return ast;
}

export default {
  parsers: {
    typescript: {
      ...pluginTypescript.parsers.typescript,
      parse,
      astFormat: "as-estree",
      preprocess: preProcess,
    },
  },
  printers: { "as-estree": as_estree },
};