实现一个 Code Pen：（三）10 行代码实现代码格式化

sxkk20081年前 (2023-11-23)知识分享96

在上文中，我们使用 monaco-editor 结合 Next.js，打造了编辑器的功能，在本文中，我们将继续优化 monaco-editor，使它拥有代码格式化的功能。

prettier 在浏览器使用

关于代码格式化，被人熟悉的是 prettier，在前端工程中，为了保证团队成员提交代码的格式一致，会先安装 prettier 和 husky，使用 Git hooks 函数，在代码提交之前把代码格式化，此时的 prettier 是 nodejs 版本，是一个可执行的 cli 工具，当然 prettier 也有 Browser 版本，也就是 prettier/standalone，现代浏览器都支持 ES modules，通过下面这几行代码就可以实现浏览器端代码格式化了。

<script type="module">
  import prettier from 'https://unpkg.com/prettier@2.6.2/esm/standalone.mjs'
  import parserBabel from 'https://unpkg.com/prettier@2.6.2/esm/parser-babel.mjs'
  import parserHtml from 'https://unpkg.com/prettier@2.6.2/esm/parser-html.mjs'

  function formatCode(code) {
    return prettier.format(code, {
      parser: 'babel',
      plugins: [parserBabel, parserHtml],
    })
  }

  console.log(formatCode('const html=/* HTML */ ` 
`'))
  // Output: const html = /* HTML */ `
`;
script>

prettier 使用方法的核心就是调用不同的 parser，去解析不同的文本，在我当前的开发的 Code Pen 场景中，使用到了以下几个 parser:

babel: 处理 js
html: 处理 html
postcss: 用来处理 css, less, scss
typescript: 处理 ts

除了 ES modules 方式， Prettier 浏览器版本，还支持 amd, commonjs 的用法，使用非常方便。详情用法可以查看官方文档。

集成到 monaco-editor

monaco-editor 本身也提供了格式化的命令，可以通过右键菜单或者快捷键⇧ + ⌥ + F来对代码进行格式化，目前自带的格式化工具不如 Prettier 的标准，因此我们可以覆盖原先的格式化指令，主要通过monaco.languages.registerDocumentFormattingEditProvider 来实现。

import * as monaco from "monaco-editor/esm/vs/editor/editor.api";

const modelToPaser={
  html
  css,
  less,
  sass,
  typescript,
  javascript:'babel'
}
export function registerDocumentFormattingEditProviders() {
  const disposables = [];

  const formattingEditProvider = {
    async provideDocumentFormattingEdits(model, _options, _token) {
      const pretty=formatCode( model.getValue(), modelToPaser[model.getLanguageId()])
      if (canceled || error) return [];
      return [
        {
          range: model.getFullModelRange(),
          text: pretty,
        },
      ];
    },
  };

 ['html','css','less','scss','javascript','typescript'].forEach((id)=>{
  disposables.push(
    monaco.languages.registerDocumentFormattingEditProvider(
      id,
      formattingEditProvider
    )
  );
 })

  return {
    dispose() {
      disposables.forEach((disposable) => disposable.dispose());
    },
  };
}

上述代码中通过 model.getValue() 获得当前编辑器中的文本，通过 model.getLanguageId() 获得当前编辑器的编程语言，每一种语言都有不同的解析器，需要与Prettier的 paser 对应，比如：JavaScript 语言对应的就是babel paser。

至此，整个 Prettier 的流程便已完成，为了提高解析性能，可以将格式化的代码放入一个 web worker 中，完整的 web worker 代码如下：

import prettier from 'prettier/standalone'

const options = {
  html: async () => ({
    parser: 'html',
    plugins: [await import('prettier/parser-html')],
    printWidth: 100,
  }),
  typescript: async () => ({
    parser: 'typescript',
    plugins: [await import('prettier/parser-typescript')],
    printWidth: 100,
  }),
  css: async () => ({
    parser: 'css',
    plugins: [await import('prettier/parser-postcss')],
    printWidth: 100,
  }),
  less: async () => ({
    parser: 'less',
    plugins: [await import('prettier/parser-postcss')],
    printWidth: 100,
  }),
  scss: async () => ({
    parser: 'scss',
    plugins: [await import('prettier/parser-postcss')],
    printWidth: 100,
  }),
  javascript: async () => ({
    parser: 'babel',
    plugins: [await import('prettier/parser-babel')],
    printWidth: 100,
    semi: false,
    singleQuote: true,
  }),
}

let current

addEventListener('message', async (event) => {
  if (event.data._current) {
    current = event.data._current
    return
  }

  function respond(data) {
    setTimeout(() => {
      if (event.data._id === current) {
        postMessage({ _id: event.data._id, ...data })
      } else {
        postMessage({ _id: event.data._id, canceled: true })
      }
    }, 0)
  }

  const opts = await options[event.data.language]()

  try {
    respond({
      pretty: prettier.format(event.data.text, opts),
    })
  } catch (error) {
    respond({ error })
  }
})

覆盖快捷键

相比于 cmd + s 时，执行自定义的函数，不如直接覆盖掉自带的格式化指令，在 cmd + s 时直接执行指令来完成格式化来的优雅。执行上面的代码就已经覆盖格式化的指令，接下来，只需要绑定快捷键就可以了。

function setupKeybindings(editor) {
  let formatCommandId = 'editor.action.formatDocument'
  const { handler, when } = CommandsRegistry.getCommand(formatCommandId)
  editor._standaloneKeybindingService.addDynamicKeybinding(
    formatCommandId,
    monaco.KeyMod.CtrlCmd | monaco.KeyCode.KeyS,
    handler,
    when
  )
}

通过 CommandsRegistry.getCommand(formatCommandId) 获得 action 的方法，在通过 _standaloneKeybindingService.addDynamicKeybinding 绑定快捷键。然后在 react 组件初始化的时候绑定快捷键就可以了

useEffect(() => {
  if (divEl.current) {
    editor.current = monaco.editor.create(divEl.current, {
      minimap: { enabled: false },
      theme: 'vs-dark',
    })
  }

  setupKeybindings(editor.current)

  return () => {
    editor.current.dispose()
  }
}, [])