refactor: remove addOrigRange

Author: fisker

src/parse.ts

@@ -6,14 +6,13 @@ import { removeCstBlankLine } from "./preprocess.js";
 import Context from "./transforms/context.js";
 import { transformError } from "./transforms/error.js";
 import { type Root } from "./types.js";
-import { addOrigRange } from "./utils/add-orig-range.js";
 import { removeFakeNodes } from "./utils/remove-fake-nodes.js";
 import { updatePositions } from "./utils/update-positions.js";
 
 export function parse(text: string): Root {
   const cst = parseCST(text);
-
-  addOrigRange(cst);
+  const context = new Context(cst, text);
+  context.setOrigRanges();
 
   const documents = cst.map(cstDocument =>
     new Document({
@@ -22,8 +21,6 @@ export function parse(text: string): Root {
     }).parse(cstDocument),
   );
 
-  const context = new Context(cst, text);
-
   for (const document of documents) {
     for (const error of document.errors) {
       if (
@@ -39,7 +36,7 @@ export function parse(text: string): Root {
   documents.forEach(document => removeCstBlankLine(document.cstNode!));
 
   const root = createRoot(
-    context.transformRange({ origStart: 0, origEnd: context.text.length }),
+    context.transformRange({ origStart: 0, origEnd: text.length }),
     documents.map(document => context.transformNode(document)),
     context.comments,
   );

src/transforms/context.ts

@@ -37,6 +37,16 @@ class Context {
     this.#cst = cst;
   }
 
+  setOrigRanges() {
+    if (this.#cst.setOrigRanges()) {
+      return;
+    }
+
+    for (const document of this.#cst) {
+      document.setOrigRanges([], 0);
+    }
+  }
+
   #getRangePosition(range: Range): { start: Point; end: Point } {
     if (!rangeAsLinePosGetter) {
       const [document] = this.#cst;