source-map: Add mapping for the rule names to the names of the parse …

…functions
peggyjs · Jul 25, 2021 · a16f8d7 · a16f8d7
1 parent d2fabaf
commit a16f8d7
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/lib/compiler/passes/generate-js.js b/lib/compiler/passes/generate-js.js
@@ -127,7 +127,11 @@ function wrapInSourceNode(prefix, chunk, location, suffix, name) {
 // Generates parser JavaScript code.
 function generateJS(ast, options) {
   // These only indent non-empty lines to avoid trailing whitespace.
-  function indent2(code) { return code.replace(/^(.+)$/gm, "  $1"); }
+  function indent2(code) {
+    return code instanceof SourceNode
+      ? code
+      : code.replace(/^(.+)$/gm, "  $1");
+  }
 
   function l(i) { return "peg$c" + i; } // |literals[i]| of the abstract machine
   function r(i) { return "peg$r" + i; } // |classes[i]| of the abstract machine
@@ -578,7 +582,13 @@ function generateJS(ast, options) {
 
     const code = compile(rule.bytecode);
 
-    parts.push("function " + name(rule.name) + "() {");
+    parts.push(wrapInSourceNode(
+      "  function ",
+      name(rule.name),
+      rule.nameLocation,
+      "() {\n",
+      rule.name
+    ));
 
     if (options.trace) {
       parts.push("  var startPos = peg$currPos;");