fix debugDom parameter (#227)

Author: kamath

.changeset/olive-trains-grab.md

@@ -0,0 +1,5 @@
+---
+"@browserbasehq/stagehand": patch
+---
+
+Fix debugDom still showing chunks when set to false

lib/dom/debug.ts

@@ -22,6 +22,7 @@ function multiSelectorMapToSelectorMap(
 }
 
 function drawChunk(selectorMap: Record<number, string>) {
+  if (!window.showChunks) return;
   cleanupMarkers();
   Object.entries(selectorMap).forEach(([_index, selector]) => {
     const element = document.evaluate(

lib/dom/global.d.ts

@@ -2,6 +2,7 @@ export {};
 declare global {
   interface Window {
     chunkNumber: number;
+    showChunks?: boolean;
     processDom: (chunksSeen: Array<number>) => Promise<{
       outputString: string;
       selectorMap: Record<number, string[]>;

lib/index.ts

@@ -390,6 +390,9 @@ export class Stagehand {
     const originalGoto = this.page.goto.bind(this.page);
     this.page.goto = async (url: string, options?: any) => {
       const result = await originalGoto(url, options);
+      if (this.debugDom) {
+        await this.page.evaluate(() => (window.showChunks = this.debugDom));
+      }
       await this.page.waitForLoadState("domcontentloaded");
       await this._waitForSettledDom();
       return result;
@@ -456,6 +459,9 @@ export class Stagehand {
     const originalGoto = this.page.goto.bind(this.page);
     this.page.goto = async (url: string, options?: any) => {
       const result = await originalGoto(url, options);
+      if (this.debugDom) {
+        await this.page.evaluate(() => (window.showChunks = this.debugDom));
+      }
       await this.page.waitForLoadState("domcontentloaded");
       await this._waitForSettledDom();
       return result;