Implement bulder setRangeCallSite

Author: szuend

src/builder/builder.test.ts

@@ -288,6 +288,26 @@ describe("ScopeInfoBuilder", () => {
     });
   });
 
+  describe("setRangeCallSite", () => {
+    it("sets the callSite", () => {
+      const info = builder.startRange(0, 0).setRangeCallSite({
+        line: 10,
+        column: 20,
+        sourceIndex: 0,
+      }).endRange(0, 10).build();
+
+      assertEquals(info.ranges[0].callSite, {
+        line: 10,
+        column: 20,
+        sourceIndex: 0,
+      });
+    });
+
+    it("does nothing when no range is on the stack", () => {
+      builder.setRangeCallSite({ line: 10, column: 20, sourceIndex: 0 });
+    });
+  });
+
   describe("endRange", () => {
     it("does nothing when no range is open", () => {
       builder.endRange(0, 20);

src/builder/builder.ts

@@ -200,6 +200,13 @@ export class ScopeInfoBuilder {
     return this;
   }
 
+  setRangeCallSite(callSite: OriginalPosition): this {
+    const range = this.#rangeStack.at(-1);
+    if (range) range.callSite = callSite;
+
+    return this;
+  }
+
   endRange(line: number, column: number): this {
     const range = this.#rangeStack.pop();
     if (!range) return this;

src/builder/safe_builder.test.ts

@@ -298,6 +298,30 @@ describe("SafeScopeInfoBuilder", () => {
     });
   });
 
+  describe("setRangeCallSite", () => {
+    it("throws when no range is on open", () => {
+      assertThrows(() =>
+        builder.setRangeCallSite({
+          line: 10,
+          column: 20,
+          sourceIndex: 0,
+        })
+      );
+    });
+
+    it("throws while building a scope", () => {
+      builder.startScope(0, 0);
+
+      assertThrows(() =>
+        builder.setRangeCallSite({
+          line: 10,
+          column: 20,
+          sourceIndex: 0,
+        })
+      );
+    });
+  });
+
   describe("endRange", () => {
     it("throws when the range stack is empty", () => {
       assertThrows(() => builder.endRange(5, 0));

src/builder/safe_builder.ts

@@ -2,7 +2,12 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-import type { Binding, OriginalPosition, OriginalScope, ScopeInfo } from "../scopes.d.ts";
+import type {
+  Binding,
+  OriginalPosition,
+  OriginalScope,
+  ScopeInfo,
+} from "../scopes.d.ts";
 import { comparePositions } from "../util.ts";
 import { ScopeInfoBuilder, type ScopeKey } from "./builder.ts";
 
@@ -241,6 +246,14 @@ export class SafeScopeInfoBuilder extends ScopeInfoBuilder {
     return this;
   }
 
+  override setRangeCallSite(callSite: OriginalPosition): this {
+    this.#verifyEmptyScopeStack("setRangeCallSite");
+    this.#verifyRangePresent("setRangeCallSite");
+
+    super.setRangeCallSite(callSite);
+    return this;
+  }
+
   override endRange(line: number, column: number): this {
     this.#verifyEmptyScopeStack("endRange");