prettierx/src/language-markdown/pragma.js

prettierX - a less opinionated fork of the Prettier code formatter

"use strict";

const parseFrontMatter = require("../utils/front-matter/parse");

const pragmas = ["format", "prettier"];

function startWithPragma(text) {
  const pragma = `@(${pragmas.join("|")})`;
  const regex = new RegExp(
    [
      `<!--\\s*${pragma}\\s*-->`,
      `<!--.*\r?\n[\\s\\S]*(^|\n)[^\\S\n]*${pragma}[^\\S\n]*($|\n)[\\s\\S]*\n.*-->`,
    ].join("|"),
    "m"
  );
  const matched = text.match(regex);
  return matched && matched.index === 0;
}

module.exports = {
  startWithPragma,
  hasPragma: (text) =>
    startWithPragma(parseFrontMatter(text).content.trimStart()),
  insertPragma: (text) => {
    const extracted = parseFrontMatter(text);
    const pragma = `<!-- @${pragmas[0]} -->`;
    return extracted.frontMatter
      ? `${extracted.frontMatter.raw}\n\n${pragma}\n\n${extracted.content}`
      : `${pragma}\n\n${extracted.content}`;
  },
};