From e2337dce26e83d3604fb7a9be7e3e58311fec2b5 Mon Sep 17 00:00:00 2001
From: fisker Cheung <lionkay@gmail.com>
Date: Fri, 9 Sep 2022 11:25:30 +0800
Subject: [PATCH] Glimmer: Combine plugins to reduce traverse (#13446)

---
 src/language-handlebars/parser-glimmer.js | 58 +++++++++++------------
 1 file changed, 29 insertions(+), 29 deletions(-)

diff --git a/src/language-handlebars/parser-glimmer.js b/src/language-handlebars/parser-glimmer.js
index a9a23d881..ba7a151dc 100644
--- a/src/language-handlebars/parser-glimmer.js
+++ b/src/language-handlebars/parser-glimmer.js
@@ -6,42 +6,42 @@ import { locStart, locEnd } from "./loc.js";
  * glimmer parser will produce an AST missing a backslash
  * so here we add it back
  * */
-function addBackslash(/* options*/) {
-  return {
-    name: "addBackslash",
-    visitor: {
-      All(node) {
-        const childrenOrBody = node.children ?? node.body;
-        if (childrenOrBody) {
-          for (let i = 0; i < childrenOrBody.length - 1; i++) {
-            if (
-              childrenOrBody[i].type === "TextNode" &&
-              childrenOrBody[i + 1].type === "MustacheStatement"
-            ) {
-              childrenOrBody[i].chars = childrenOrBody[i].chars.replace(
-                /\\$/,
-                "\\\\"
-              );
-            }
-          }
-        }
-      },
-    },
-  };
+function addBackslash(node) {
+  const childrenOrBody = node.children ?? node.body;
+  if (childrenOrBody) {
+    for (let i = 0; i < childrenOrBody.length - 1; i++) {
+      if (
+        childrenOrBody[i].type === "TextNode" &&
+        childrenOrBody[i + 1].type === "MustacheStatement"
+      ) {
+        childrenOrBody[i].chars = childrenOrBody[i].chars.replace(
+          /\\$/,
+          "\\\\"
+        );
+      }
+    }
+  }
 }
 
-// Add `loc.{start,end}.offset`
-function addOffset(text) {
+// Combine plugins to reduce traverse https://github.com/glimmerjs/glimmer-vm/blob/cdfb8f93d7ff0b504c8e9eab293f656a9b942025/packages/%40glimmer/syntax/lib/parser/tokenizer-event-handlers.ts#L442-L451
+function createPlugin(text) {
   const lines = new LinesAndColumns(text);
   const calculateOffset = ({ line, column }) =>
     lines.indexForLocation({ line: line - 1, column });
+
+  // Add `loc.{start,end}.offset`
+  const addOffset = (node) => {
+    const { start, end } = node.loc;
+    start.offset = calculateOffset(start);
+    end.offset = calculateOffset(end);
+  };
+
   return (/* options*/) => ({
-    name: "addOffset",
+    name: "prettierParsePlugin",
     visitor: {
       All(node) {
-        const { start, end } = node.loc;
-        start.offset = calculateOffset(start);
-        end.offset = calculateOffset(end);
+        addOffset(node);
+        addBackslash(node);
       },
     },
   });
@@ -61,7 +61,7 @@ async function parse(text /*, options */) {
   try {
     ast = glimmer(text, {
       mode: "codemod",
-      plugins: { ast: [addBackslash, addOffset(text)] },
+      plugins: { ast: [createPlugin(text)] },
     });
   } catch (error) {
     const location = getErrorLocation(error);