Add isStackFrame for ranges to builder

Author: szuend

src/builder/builder.test.ts

@@ -142,6 +142,15 @@ describe("ScopeInfoBuilder", () => {
       assertStrictEquals(info.scopes[0], info.ranges[0].originalScope);
       assertStrictEquals(info.ranges[0].originalScope, scope);
     });
+
+    it("can set isStackFrame via option", () => {
+      const info = builder.startRange(0, 0, { isStackFrame: true }).endRange(
+        10,
+        0,
+      ).build();
+
+      assertStrictEquals(info.ranges[0]?.isStackFrame, true);
+    });
   });
 
   describe("setRangeDefinitionScope", () => {
@@ -162,6 +171,18 @@ describe("ScopeInfoBuilder", () => {
     });
   });
 
+  describe("setRangeStackFrame", () => {
+    it("sets the isStackFrame flag", () => {
+      const info = builder.startRange(0, 0).setRangeStackFrame(true).endRange(
+        10,
+        0,
+      )
+        .build();
+
+      assertStrictEquals(info.ranges[0]?.isStackFrame, true);
+    });
+  });
+
   describe("endRange", () => {
     it("does nothing when no range is open", () => {
       builder.endRange(0, 20);

src/builder/builder.ts

@@ -113,12 +113,12 @@ export class ScopeInfoBuilder {
   startRange(
     line: number,
     column: number,
-    options?: { scope?: number | OriginalScope },
+    options?: { scope?: number | OriginalScope; isStackFrame?: boolean },
   ): this {
     const range: GeneratedRange = {
       start: { line, column },
       end: { line, column },
-      isStackFrame: false,
+      isStackFrame: Boolean(options?.isStackFrame),
       isHidden: false,
       values: [],
       children: [],
@@ -152,6 +152,13 @@ export class ScopeInfoBuilder {
     return this;
   }
 
+  setRangeStackFrame(isStackFrame: boolean): this {
+    const range = this.#rangeStack.at(-1);
+    if (range) range.isStackFrame = isStackFrame;
+
+    return this;
+  }
+
   endRange(line: number, column: number): this {
     const range = this.#rangeStack.pop();
     if (!range) return this;

src/builder/safe_builder.test.ts

@@ -189,6 +189,18 @@ describe("SafeScopeInfoBuilder", () => {
     });
   });
 
+  describe("setRangeStackFrame", () => {
+    it("throws when no range is on open", () => {
+      assertThrows(() => builder.setRangeStackFrame(true));
+    });
+
+    it("throws while building a scope", () => {
+      builder.startScope(0, 0);
+
+      assertThrows(() => builder.setRangeStackFrame(true));
+    });
+  });
+
   describe("endRange", () => {
     it("throws when the range stack is empty", () => {
       assertThrows(() => builder.endRange(5, 0));

src/builder/safe_builder.ts

@@ -162,6 +162,14 @@ export class SafeScopeInfoBuilder extends ScopeInfoBuilder {
     return this;
   }
 
+  override setRangeStackFrame(isStackFrame: boolean): this {
+    this.#verifyEmptyScopeStack("setRangeStackFrame");
+    this.#verifyRangePresent("setRangeStackFrame");
+
+    super.setRangeStackFrame(isStackFrame);
+    return this;
+  }
+
   override endRange(line: number, column: number): this {
     this.#verifyEmptyScopeStack("endRange");