@browserbasehq/orca 3.0.9-alpha-4 → 3.1.0-patch.1

package/dist/esm/lib/v3/understudy/frameRegistry.js

@@ -0,0 +1,298 @@
+/** Minimal “shell” CDP frame used when we haven’t yet seen a real Frame from events. */
+function shellFrame(id) {
+    return {
+        id,
+        loaderId: "",
+        url: "",
+        domainAndRegistry: "",
+        securityOrigin: "",
+        mimeType: "text/html",
+        secureContextType: "InsecureScheme",
+        crossOriginIsolatedContextType: "NotIsolated",
+        gatedAPIFeatures: [],
+    };
+}
+export class FrameRegistry {
+    /** Owner target id (top-level target); informational only */
+    ownerTargetId;
+    /** Current main/root frame id (changes on root swaps) */
+    rootFrameId;
+    /** frameId → FrameInfo */
+    frames = new Map();
+    /** sessionId → Set<frameId> (inverse map for diagnostics/fast membership checks) */
+    framesBySession = new Map();
+    constructor(ownerTargetId, mainFrameId) {
+        this.ownerTargetId = ownerTargetId;
+        this.rootFrameId = mainFrameId;
+        this.ensureNode(mainFrameId);
+    }
+    // ---------------------- Mutators (called by Context/Page bridges) ----------------------
+    /**
+     * Record that a frame attached. If `parentId` is null and `frameId` differs from the current
+     * root, this is a root swap and we rename the root id.
+     *
+     * IMPORTANT: The emitter's `sessionId` is the **owner** for the new/attached frame.
+     */
+    onFrameAttached(frameId, parentId, sessionId) {
+        // Root swap (parentId === null for main frames).
+        if (!parentId && frameId !== this.rootFrameId) {
+            this.renameNodeId(this.rootFrameId, frameId);
+            this.rootFrameId = frameId;
+            // ownership moves to this session as well
+            this.setOwnerSessionIdInternal(frameId, sessionId);
+            return;
+        }
+        // Normal attach
+        this.ensureNode(frameId);
+        if (parentId)
+            this.ensureNode(parentId);
+        const info = this.frames.get(frameId);
+        info.parentId = parentId ?? null;
+        if (parentId) {
+            this.frames.get(parentId).children.add(frameId);
+        }
+        // Ownership: the session that emitted frameAttached owns this frame.
+        this.setOwnerSessionIdInternal(frameId, sessionId);
+    }
+    /**
+     * Record a navigation with the full CDP `Frame`. Also updates ownership based on the emitting
+     * session id. Handles root swap if the navigated frame is the new main (no parentId).
+     */
+    onFrameNavigated(frame, sessionId) {
+        this.ensureNode(frame.id);
+        const info = this.frames.get(frame.id);
+        info.lastSeen = frame;
+        // Ownership follows the session that reported the navigation
+        this.setOwnerSessionIdInternal(frame.id, sessionId);
+        // If this frame has no parent, it might be the (new) main/root
+        if (!("parentId" in frame) || !frame.parentId) {
+            if (frame.id !== this.rootFrameId) {
+                // carry ordinal semantics by renaming the root id
+                this.renameNodeId(this.rootFrameId, frame.id);
+                this.rootFrameId = frame.id;
+            }
+        }
+    }
+    onNavigatedWithinDocument(frameId, url, sessionId) {
+        this.ensureNode(frameId);
+        const info = this.frames.get(frameId);
+        const lastSeen = info.lastSeen ?? shellFrame(frameId);
+        info.lastSeen = { ...lastSeen, url };
+        this.setOwnerSessionIdInternal(frameId, sessionId);
+    }
+    /**
+     * Record that a frame detached. If `reason !== "swap"`, remove the subtree from the graph,
+     * and clean the inverse maps. For “swap” we keep the node to preserve continuity.
+     */
+    onFrameDetached(frameId, reason = "remove") {
+        if (reason === "swap")
+            return;
+        // Collect subtree starting from frameId.
+        const toRemove = [];
+        const collect = (fid) => {
+            toRemove.push(fid);
+            const kids = this.frames.get(fid)?.children ?? new Set();
+            for (const k of kids)
+                collect(k);
+        };
+        collect(frameId);
+        // Remove nodes, fix parents and inverse maps
+        for (const fid of toRemove) {
+            const info = this.frames.get(fid);
+            if (!info)
+                continue;
+            // unlink from parent
+            if (info.parentId) {
+                const p = this.frames.get(info.parentId);
+                p?.children.delete(fid);
+            }
+            // unlink inverse session map
+            if (info.ownerSessionId) {
+                const bag = this.framesBySession.get(info.ownerSessionId);
+                bag?.delete(fid);
+                if (bag && bag.size === 0)
+                    this.framesBySession.delete(info.ownerSessionId);
+            }
+            this.frames.delete(fid);
+        }
+        // Guard root if we removed it; assign a placeholder root if needed
+        if (!this.frames.has(this.rootFrameId)) {
+            // Choose an arbitrary remaining node as root
+            const iter = this.frames.keys().next();
+            if (!iter.done)
+                this.rootFrameId = iter.value;
+        }
+    }
+    /**
+     * An adopted OOPIF child session was created whose **main** frame id equals the parent iframe’s frameId.
+     * We mark the entire child subtree as owned by `childSessionId`.
+     * (Topology edges remain aligned by the parent session’s `frameAttached` events.)
+     */
+    adoptChildSession(childSessionId, childMainFrameId) {
+        // The child session will emit its own navigations/attachments; as a seed,
+        // mark the root frame as owned by the child session.
+        this.setOwnerSessionIdInternal(childMainFrameId, childSessionId);
+    }
+    /**
+     * Seed topology and ownership from an existing `Page.getFrameTree` snapshot, typically right after
+     * a session is attached. This is a best-effort: we record frames and set the provided `sessionId`
+     * as owner for the subtree **if** an owner isn't already set.
+     */
+    seedFromFrameTree(sessionId, frameTree) {
+        const walk = (tree, parent) => {
+            this.ensureNode(tree.frame.id);
+            // topology
+            this.frames.get(tree.frame.id).parentId = parent;
+            if (parent)
+                this.frames.get(parent).children.add(tree.frame.id);
+            // last-seen frame
+            this.frames.get(tree.frame.id).lastSeen = tree.frame;
+            // ownership (only if unknown)
+            if (!this.frames.get(tree.frame.id).ownerSessionId) {
+                this.setOwnerSessionIdInternal(tree.frame.id, sessionId);
+            }
+            for (const c of tree.childFrames ?? [])
+                walk(c, tree.frame.id);
+        };
+        walk(frameTree, null);
+    }
+    /**
+     * Set the backendNodeId of the `<iframe>` element for a child frame **as seen from its parent**.
+     * This is useful for building absolute XPath prefixes later (from the parent document).
+     */
+    setOwnerBackendNodeId(childFrameId, backendNodeId) {
+        this.ensureNode(childFrameId);
+        this.frames.get(childFrameId).ownerBackendNodeId = backendNodeId;
+    }
+    // ---------------------- Readers (consumed by Page/snapshot/locators) ----------------------
+    mainFrameId() {
+        return this.rootFrameId;
+    }
+    /**
+     * Return the owner session id for this frame. If unknown, returns `undefined`.
+     */
+    getOwnerSessionId(frameId) {
+        return this.frames.get(frameId)?.ownerSessionId;
+    }
+    /**
+     * Return the owner backendNodeId (iframe element) if recorded.
+     * This is in the **parent** document; pair it with `getParent`.
+     */
+    getOwnerBackendNodeId(frameId) {
+        return this.frames.get(frameId)?.ownerBackendNodeId;
+    }
+    /**
+     * Return the parent frame id, or null for root/unknown.
+     */
+    getParent(frameId) {
+        return this.frames.get(frameId)?.parentId ?? null;
+    }
+    /**
+     * List frame ids in root-first DFS order (same shape as CDP’s FrameTree traversal).
+     */
+    listAllFrames() {
+        const out = [];
+        const dfs = (fid) => {
+            out.push(fid);
+            const kids = this.frames.get(fid)?.children ?? new Set();
+            for (const k of kids)
+                dfs(k);
+        };
+        if (this.frames.has(this.rootFrameId))
+            dfs(this.rootFrameId);
+        return out;
+    }
+    /**
+     * Serialize to `Protocol.Page.FrameTree` starting at the given root id (typically mainFrameId()).
+     */
+    asProtocolFrameTree(rootId) {
+        const build = (fid) => {
+            const info = this.frames.get(fid);
+            const frame = info?.lastSeen ?? shellFrame(fid);
+            const kids = info?.children ?? new Set();
+            const childFrames = kids.size
+                ? [...kids].map((k) => build(k))
+                : undefined;
+            return childFrames ? { frame, childFrames } : { frame };
+        };
+        return build(rootId);
+    }
+    /**
+     * For diagnostics: return the current owner sessions for a frame id (0..n),
+     * usually 0 or 1, but helpful to see potential inconsistencies during wiring.
+     */
+    sessionsForFrame(frameId) {
+        const info = this.frames.get(frameId);
+        return info?.ownerSessionId ? [info.ownerSessionId] : [];
+    }
+    /**
+     * For diagnostics: return current frame set per session.
+     */
+    framesForSession(sessionId) {
+        return [...(this.framesBySession.get(sessionId) ?? new Set())];
+    }
+    // ---------------------- Internal helpers ----------------------
+    ensureNode(fid) {
+        if (this.frames.has(fid))
+            return;
+        this.frames.set(fid, {
+            parentId: null,
+            children: new Set(),
+            lastSeen: shellFrame(fid),
+            ownerSessionId: undefined,
+            ownerBackendNodeId: undefined,
+        });
+    }
+    renameNodeId(oldId, newId) {
+        if (oldId === newId)
+            return;
+        this.ensureNode(oldId);
+        const info = this.frames.get(oldId);
+        // Move info under new id
+        this.frames.delete(oldId);
+        this.frames.set(newId, { ...info });
+        // Fix parent’s children set
+        if (info.parentId) {
+            const p = this.frames.get(info.parentId);
+            if (p) {
+                p.children.delete(oldId);
+                p.children.add(newId);
+            }
+        }
+        // Fix children’s parent pointers
+        for (const c of info.children) {
+            const ci = this.frames.get(c);
+            if (ci)
+                ci.parentId = newId;
+        }
+        // Fix inverse map (session -> frames)
+        if (info.ownerSessionId) {
+            const bag = this.framesBySession.get(info.ownerSessionId);
+            if (bag) {
+                bag.delete(oldId);
+                bag.add(newId);
+            }
+        }
+        // If root moved, keep the root id updated is handled by caller
+    }
+    setOwnerSessionIdInternal(frameId, sessionId) {
+        this.ensureNode(frameId);
+        const info = this.frames.get(frameId);
+        // If the owner is unchanged, do nothing
+        if (info.ownerSessionId === sessionId)
+            return;
+        // Remove from previous owner bag
+        if (info.ownerSessionId) {
+            const prev = this.framesBySession.get(info.ownerSessionId);
+            prev?.delete(frameId);
+            if (prev && prev.size === 0)
+                this.framesBySession.delete(info.ownerSessionId);
+        }
+        // Set new owner and update bag
+        info.ownerSessionId = sessionId;
+        const bag = this.framesBySession.get(sessionId) ?? new Set();
+        bag.add(frameId);
+        this.framesBySession.set(sessionId, bag);
+    }
+}
+//# sourceMappingURL=frameRegistry.js.map

package/dist/esm/lib/v3/understudy/frameRegistry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"frameRegistry.js","sourceRoot":"","sources":["../../../../../lib/v3/understudy/frameRegistry.ts"],"names":[],"mappings":"AA0CA,wFAAwF;AACxF,SAAS,UAAU,CAAC,EAAW;IAC7B,OAAO;QACL,EAAE;QACF,QAAQ,EAAE,EAAE;QACZ,GAAG,EAAE,EAAE;QACP,iBAAiB,EAAE,EAAE;QACrB,cAAc,EAAE,EAAE;QAClB,QAAQ,EAAE,WAAW;QACrB,iBAAiB,EAAE,gBAAgB;QACnC,8BAA8B,EAAE,aAAa;QAC7C,gBAAgB,EAAE,EAAE;KACE,CAAC;AAC3B,CAAC;AAED,MAAM,OAAO,aAAa;IACxB,6DAA6D;IAC5C,aAAa,CAAS;IAEvC,yDAAyD;IACjD,WAAW,CAAU;IAE7B,0BAA0B;IAClB,MAAM,GAAG,IAAI,GAAG,EAAsB,CAAC;IAE/C,oFAAoF;IAC5E,eAAe,GAAG,IAAI,GAAG,EAA2B,CAAC;IAE7D,YAAY,aAAqB,EAAE,WAAoB;QACrD,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;IAC/B,CAAC;IAED,0FAA0F;IAE1F;;;;;OAKG;IACH,eAAe,CACb,OAAgB,EAChB,QAAwB,EACxB,SAAoB;QAEpB,iDAAiD;QACjD,IAAI,CAAC,QAAQ,IAAI,OAAO,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;YAC9C,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YAC7C,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC;YAC3B,0CAA0C;YAC1C,IAAI,CAAC,yBAAyB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YACnD,OAAO;QACT,CAAC;QAED,gBAAgB;QAChB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACzB,IAAI,QAAQ;YAAE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAExC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAE,CAAC;QACvC,IAAI,CAAC,QAAQ,GAAG,QAAQ,IAAI,IAAI,CAAC;QAEjC,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACnD,CAAC;QAED,qEAAqE;QACrE,IAAI,CAAC,yBAAyB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IACrD,CAAC;IAED;;;OAGG;IACH,gBAAgB,CAAC,KAA0B,EAAE,SAAoB;QAC/D,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAE,CAAC;QACxC,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC;QAEtB,6DAA6D;QAC7D,IAAI,CAAC,yBAAyB,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QAEpD,+DAA+D;QAC/D,IAAI,CAAC,CAAC,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC9C,IAAI,KAAK,CAAC,EAAE,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;gBAClC,kDAAkD;gBAClD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;gBAC9C,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC,EAAE,CAAC;YAC9B,CAAC;QACH,CAAC;IACH,CAAC;IAED,yBAAyB,CACvB,OAAgB,EAChB,GAAW,EACX,SAAoB;QAEpB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAE,CAAC;QACvC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC;QACtD,IAAI,CAAC,QAAQ,GAAG,EAAE,GAAG,QAAQ,EAAE,GAAG,EAAE,CAAC;QACrC,IAAI,CAAC,yBAAyB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IACrD,CAAC;IAED;;;OAGG;IACH,eAAe,CACb,OAAgB,EAChB,SAAqC,QAAQ;QAE7C,IAAI,MAAM,KAAK,MAAM;YAAE,OAAO;QAE9B,yCAAyC;QACzC,MAAM,QAAQ,GAAc,EAAE,CAAC;QAC/B,MAAM,OAAO,GAAG,CAAC,GAAY,EAAE,EAAE;YAC/B,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,QAAQ,IAAI,IAAI,GAAG,EAAW,CAAC;YAClE,KAAK,MAAM,CAAC,IAAI,IAAI;gBAAE,OAAO,CAAC,CAAC,CAAC,CAAC;QACnC,CAAC,CAAC;QACF,OAAO,CAAC,OAAO,CAAC,CAAC;QAEjB,6CAA6C;QAC7C,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAClC,IAAI,CAAC,IAAI;gBAAE,SAAS;YAEpB,qBAAqB;YACrB,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACzC,CAAC,EAAE,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC1B,CAAC;YAED,6BAA6B;YAC7B,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBACxB,MAAM,GAAG,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;gBAC1D,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC;gBACjB,IAAI,GAAG,IAAI,GAAG,CAAC,IAAI,KAAK,CAAC;oBACvB,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACrD,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;QAED,mEAAmE;QACnE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;YACvC,6CAA6C;YAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;YACvC,IAAI,CAAC,IAAI,CAAC,IAAI;gBAAE,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC;QAChD,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,iBAAiB,CACf,cAAyB,EACzB,gBAAyB;QAEzB,0EAA0E;QAC1E,qDAAqD;QACrD,IAAI,CAAC,yBAAyB,CAAC,gBAAgB,EAAE,cAAc,CAAC,CAAC;IACnE,CAAC;IAED;;;;OAIG;IACH,iBAAiB,CACf,SAAoB,EACpB,SAAkC;QAElC,MAAM,IAAI,GAAG,CAAC,IAA6B,EAAE,MAAsB,EAAE,EAAE;YACrE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC/B,WAAW;YACX,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAE,CAAC,QAAQ,GAAG,MAAM,CAAC;YAClD,IAAI,MAAM;gBAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACjE,kBAAkB;YAClB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAE,CAAC,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC;YACtD,8BAA8B;YAC9B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAE,CAAC,cAAc,EAAE,CAAC;gBACpD,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;YAC3D,CAAC;YACD,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,WAAW,IAAI,EAAE;gBAAE,IAAI,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACjE,CAAC,CAAC;QACF,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IACxB,CAAC;IAED;;;OAGG;IACH,qBAAqB,CAAC,YAAqB,EAAE,aAAqB;QAChE,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,YAAY,CAAE,CAAC,kBAAkB,GAAG,aAAa,CAAC;IACpE,CAAC;IAED,6FAA6F;IAE7F,WAAW;QACT,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,OAAgB;QAChC,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,cAAc,CAAC;IAClD,CAAC;IAED;;;OAGG;IACH,qBAAqB,CAAC,OAAgB;QACpC,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,kBAAkB,CAAC;IACtD,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,OAAgB;QACxB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,QAAQ,IAAI,IAAI,CAAC;IACpD,CAAC;IAED;;OAEG;IACH,aAAa;QACX,MAAM,GAAG,GAAc,EAAE,CAAC;QAC1B,MAAM,GAAG,GAAG,CAAC,GAAY,EAAE,EAAE;YAC3B,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACd,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,QAAQ,IAAI,IAAI,GAAG,EAAW,CAAC;YAClE,KAAK,MAAM,CAAC,IAAI,IAAI;gBAAE,GAAG,CAAC,CAAC,CAAC,CAAC;QAC/B,CAAC,CAAC;QACF,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC;YAAE,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC7D,OAAO,GAAG,CAAC;IACb,CAAC;IAED;;OAEG;IACH,mBAAmB,CAAC,MAAe;QACjC,MAAM,KAAK,GAAG,CAAC,GAAY,EAA2B,EAAE;YACtD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAClC,MAAM,KAAK,GAAG,IAAI,EAAE,QAAQ,IAAI,UAAU,CAAC,GAAG,CAAC,CAAC;YAEhD,MAAM,IAAI,GAAG,IAAI,EAAE,QAAQ,IAAI,IAAI,GAAG,EAAW,CAAC;YAClD,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI;gBAC3B,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAChC,CAAC,CAAC,SAAS,CAAC;YAEd,OAAO,WAAW,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC;QAC1D,CAAC,CAAC;QAEF,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC;IACvB,CAAC;IAED;;;OAGG;IACH,gBAAgB,CAAC,OAAgB;QAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACtC,OAAO,IAAI,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3D,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,SAAoB;QACnC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC;IACjE,CAAC;IAED,iEAAiE;IAEzD,UAAU,CAAC,GAAY;QAC7B,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,OAAO;QACjC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE;YACnB,QAAQ,EAAE,IAAI;YACd,QAAQ,EAAE,IAAI,GAAG,EAAW;YAC5B,QAAQ,EAAE,UAAU,CAAC,GAAG,CAAC;YACzB,cAAc,EAAE,SAAS;YACzB,kBAAkB,EAAE,SAAS;SAC9B,CAAC,CAAC;IACL,CAAC;IAEO,YAAY,CAAC,KAAc,EAAE,KAAc;QACjD,IAAI,KAAK,KAAK,KAAK;YAAE,OAAO;QAC5B,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAEvB,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAE,CAAC;QAErC,yBAAyB;QACzB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;QAEpC,4BAA4B;QAC5B,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACzC,IAAI,CAAC,EAAE,CAAC;gBACN,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACzB,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;QAED,iCAAiC;QACjC,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC9B,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAC9B,IAAI,EAAE;gBAAE,EAAE,CAAC,QAAQ,GAAG,KAAK,CAAC;QAC9B,CAAC;QAED,sCAAsC;QACtC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,MAAM,GAAG,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC1D,IAAI,GAAG,EAAE,CAAC;gBACR,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAClB,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACjB,CAAC;QACH,CAAC;QAED,+DAA+D;IACjE,CAAC;IAEO,yBAAyB,CAC/B,OAAgB,EAChB,SAAoB;QAEpB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAE,CAAC;QAEvC,wCAAwC;QACxC,IAAI,IAAI,CAAC,cAAc,KAAK,SAAS;YAAE,OAAO;QAE9C,iCAAiC;QACjC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC3D,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;YACtB,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC;gBACzB,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACrD,CAAC;QAED,+BAA+B;QAC/B,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;QAChC,MAAM,GAAG,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,IAAI,GAAG,EAAW,CAAC;QACtE,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACjB,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IAC3C,CAAC;CACF","sourcesContent":["// lib/v3/understudy/frameRegistry.ts\nimport type { Protocol } from \"devtools-protocol\";\n\n/**\n * FrameRegistry\n *\n * Purpose:\n * A single, authoritative source of truth for **both**:\n *   1) Frame topology (parent/children, current main/root id, last-seen CDP `Frame`)\n *   2) Frame → Session ownership (which CDP session owns a given frameId)\n *   3) Optional iframe-owner metadata (backendNodeId of the <iframe> element in the parent doc)\n *\n *\n * Model:\n *  - This class is **CDP-agnostic**; it stores **sessionId strings** (not session objects).\n *  - Context bridges (wiring Target/Page events) must call the mutators below (onAttached,\n *    onNavigated, onDetached, adoptChildSession, seedFromFrameTree, setOwnerBackendNodeId).\n *  - Consumers ask read APIs (getOwnerSessionId, getParent, asProtocolFrameTree, listAll, …)\n *    and never probe ownership at run time.\n */\n\ntype FrameId = string;\ntype SessionId = string;\n\ntype FrameInfo = {\n  /** Parent frame id, or null for root */\n  parentId: FrameId | null;\n  /** Children frame ids (direct) */\n  children: Set<FrameId>;\n  /** Last-seen CDP Frame metadata for this id (may be a shell if never seen) */\n  lastSeen?: Protocol.Page.Frame;\n\n  /** Owning session id (CDP child session for OOPIF, top-level session for same-process) */\n  ownerSessionId?: SessionId;\n\n  /**\n   * The backendNodeId of the <iframe> element **in the parent document** that hosts this frame.\n   * Useful for building absolute XPath prefixes or DOM scoping in the parent session.\n   */\n  ownerBackendNodeId?: number;\n};\n\n/** Minimal “shell” CDP frame used when we haven’t yet seen a real Frame from events. */\nfunction shellFrame(id: FrameId): Protocol.Page.Frame {\n  return {\n    id,\n    loaderId: \"\",\n    url: \"\",\n    domainAndRegistry: \"\",\n    securityOrigin: \"\",\n    mimeType: \"text/html\",\n    secureContextType: \"InsecureScheme\",\n    crossOriginIsolatedContextType: \"NotIsolated\",\n    gatedAPIFeatures: [],\n  } as Protocol.Page.Frame;\n}\n\nexport class FrameRegistry {\n  /** Owner target id (top-level target); informational only */\n  private readonly ownerTargetId: string;\n\n  /** Current main/root frame id (changes on root swaps) */\n  private rootFrameId: FrameId;\n\n  /** frameId → FrameInfo */\n  private frames = new Map<FrameId, FrameInfo>();\n\n  /** sessionId → Set<frameId> (inverse map for diagnostics/fast membership checks) */\n  private framesBySession = new Map<SessionId, Set<FrameId>>();\n\n  constructor(ownerTargetId: string, mainFrameId: FrameId) {\n    this.ownerTargetId = ownerTargetId;\n    this.rootFrameId = mainFrameId;\n    this.ensureNode(mainFrameId);\n  }\n\n  // ---------------------- Mutators (called by Context/Page bridges) ----------------------\n\n  /**\n   * Record that a frame attached. If `parentId` is null and `frameId` differs from the current\n   * root, this is a root swap and we rename the root id.\n   *\n   * IMPORTANT: The emitter's `sessionId` is the **owner** for the new/attached frame.\n   */\n  onFrameAttached(\n    frameId: FrameId,\n    parentId: FrameId | null,\n    sessionId: SessionId,\n  ): void {\n    // Root swap (parentId === null for main frames).\n    if (!parentId && frameId !== this.rootFrameId) {\n      this.renameNodeId(this.rootFrameId, frameId);\n      this.rootFrameId = frameId;\n      // ownership moves to this session as well\n      this.setOwnerSessionIdInternal(frameId, sessionId);\n      return;\n    }\n\n    // Normal attach\n    this.ensureNode(frameId);\n    if (parentId) this.ensureNode(parentId);\n\n    const info = this.frames.get(frameId)!;\n    info.parentId = parentId ?? null;\n\n    if (parentId) {\n      this.frames.get(parentId)!.children.add(frameId);\n    }\n\n    // Ownership: the session that emitted frameAttached owns this frame.\n    this.setOwnerSessionIdInternal(frameId, sessionId);\n  }\n\n  /**\n   * Record a navigation with the full CDP `Frame`. Also updates ownership based on the emitting\n   * session id. Handles root swap if the navigated frame is the new main (no parentId).\n   */\n  onFrameNavigated(frame: Protocol.Page.Frame, sessionId: SessionId): void {\n    this.ensureNode(frame.id);\n    const info = this.frames.get(frame.id)!;\n    info.lastSeen = frame;\n\n    // Ownership follows the session that reported the navigation\n    this.setOwnerSessionIdInternal(frame.id, sessionId);\n\n    // If this frame has no parent, it might be the (new) main/root\n    if (!(\"parentId\" in frame) || !frame.parentId) {\n      if (frame.id !== this.rootFrameId) {\n        // carry ordinal semantics by renaming the root id\n        this.renameNodeId(this.rootFrameId, frame.id);\n        this.rootFrameId = frame.id;\n      }\n    }\n  }\n\n  onNavigatedWithinDocument(\n    frameId: FrameId,\n    url: string,\n    sessionId: SessionId,\n  ): void {\n    this.ensureNode(frameId);\n    const info = this.frames.get(frameId)!;\n    const lastSeen = info.lastSeen ?? shellFrame(frameId);\n    info.lastSeen = { ...lastSeen, url };\n    this.setOwnerSessionIdInternal(frameId, sessionId);\n  }\n\n  /**\n   * Record that a frame detached. If `reason !== \"swap\"`, remove the subtree from the graph,\n   * and clean the inverse maps. For “swap” we keep the node to preserve continuity.\n   */\n  onFrameDetached(\n    frameId: FrameId,\n    reason: \"remove\" | \"swap\" | string = \"remove\",\n  ): void {\n    if (reason === \"swap\") return;\n\n    // Collect subtree starting from frameId.\n    const toRemove: FrameId[] = [];\n    const collect = (fid: FrameId) => {\n      toRemove.push(fid);\n      const kids = this.frames.get(fid)?.children ?? new Set<FrameId>();\n      for (const k of kids) collect(k);\n    };\n    collect(frameId);\n\n    // Remove nodes, fix parents and inverse maps\n    for (const fid of toRemove) {\n      const info = this.frames.get(fid);\n      if (!info) continue;\n\n      // unlink from parent\n      if (info.parentId) {\n        const p = this.frames.get(info.parentId);\n        p?.children.delete(fid);\n      }\n\n      // unlink inverse session map\n      if (info.ownerSessionId) {\n        const bag = this.framesBySession.get(info.ownerSessionId);\n        bag?.delete(fid);\n        if (bag && bag.size === 0)\n          this.framesBySession.delete(info.ownerSessionId);\n      }\n\n      this.frames.delete(fid);\n    }\n\n    // Guard root if we removed it; assign a placeholder root if needed\n    if (!this.frames.has(this.rootFrameId)) {\n      // Choose an arbitrary remaining node as root\n      const iter = this.frames.keys().next();\n      if (!iter.done) this.rootFrameId = iter.value;\n    }\n  }\n\n  /**\n   * An adopted OOPIF child session was created whose **main** frame id equals the parent iframe’s frameId.\n   * We mark the entire child subtree as owned by `childSessionId`.\n   * (Topology edges remain aligned by the parent session’s `frameAttached` events.)\n   */\n  adoptChildSession(\n    childSessionId: SessionId,\n    childMainFrameId: FrameId,\n  ): void {\n    // The child session will emit its own navigations/attachments; as a seed,\n    // mark the root frame as owned by the child session.\n    this.setOwnerSessionIdInternal(childMainFrameId, childSessionId);\n  }\n\n  /**\n   * Seed topology and ownership from an existing `Page.getFrameTree` snapshot, typically right after\n   * a session is attached. This is a best-effort: we record frames and set the provided `sessionId`\n   * as owner for the subtree **if** an owner isn't already set.\n   */\n  seedFromFrameTree(\n    sessionId: SessionId,\n    frameTree: Protocol.Page.FrameTree,\n  ): void {\n    const walk = (tree: Protocol.Page.FrameTree, parent: FrameId | null) => {\n      this.ensureNode(tree.frame.id);\n      // topology\n      this.frames.get(tree.frame.id)!.parentId = parent;\n      if (parent) this.frames.get(parent)!.children.add(tree.frame.id);\n      // last-seen frame\n      this.frames.get(tree.frame.id)!.lastSeen = tree.frame;\n      // ownership (only if unknown)\n      if (!this.frames.get(tree.frame.id)!.ownerSessionId) {\n        this.setOwnerSessionIdInternal(tree.frame.id, sessionId);\n      }\n      for (const c of tree.childFrames ?? []) walk(c, tree.frame.id);\n    };\n    walk(frameTree, null);\n  }\n\n  /**\n   * Set the backendNodeId of the `<iframe>` element for a child frame **as seen from its parent**.\n   * This is useful for building absolute XPath prefixes later (from the parent document).\n   */\n  setOwnerBackendNodeId(childFrameId: FrameId, backendNodeId: number): void {\n    this.ensureNode(childFrameId);\n    this.frames.get(childFrameId)!.ownerBackendNodeId = backendNodeId;\n  }\n\n  // ---------------------- Readers (consumed by Page/snapshot/locators) ----------------------\n\n  mainFrameId(): FrameId {\n    return this.rootFrameId;\n  }\n\n  /**\n   * Return the owner session id for this frame. If unknown, returns `undefined`.\n   */\n  getOwnerSessionId(frameId: FrameId): SessionId | undefined {\n    return this.frames.get(frameId)?.ownerSessionId;\n  }\n\n  /**\n   * Return the owner backendNodeId (iframe element) if recorded.\n   * This is in the **parent** document; pair it with `getParent`.\n   */\n  getOwnerBackendNodeId(frameId: FrameId): number | undefined {\n    return this.frames.get(frameId)?.ownerBackendNodeId;\n  }\n\n  /**\n   * Return the parent frame id, or null for root/unknown.\n   */\n  getParent(frameId: FrameId): FrameId | null {\n    return this.frames.get(frameId)?.parentId ?? null;\n  }\n\n  /**\n   * List frame ids in root-first DFS order (same shape as CDP’s FrameTree traversal).\n   */\n  listAllFrames(): FrameId[] {\n    const out: FrameId[] = [];\n    const dfs = (fid: FrameId) => {\n      out.push(fid);\n      const kids = this.frames.get(fid)?.children ?? new Set<FrameId>();\n      for (const k of kids) dfs(k);\n    };\n    if (this.frames.has(this.rootFrameId)) dfs(this.rootFrameId);\n    return out;\n  }\n\n  /**\n   * Serialize to `Protocol.Page.FrameTree` starting at the given root id (typically mainFrameId()).\n   */\n  asProtocolFrameTree(rootId: FrameId): Protocol.Page.FrameTree {\n    const build = (fid: FrameId): Protocol.Page.FrameTree => {\n      const info = this.frames.get(fid);\n      const frame = info?.lastSeen ?? shellFrame(fid);\n\n      const kids = info?.children ?? new Set<FrameId>();\n      const childFrames = kids.size\n        ? [...kids].map((k) => build(k))\n        : undefined;\n\n      return childFrames ? { frame, childFrames } : { frame };\n    };\n\n    return build(rootId);\n  }\n\n  /**\n   * For diagnostics: return the current owner sessions for a frame id (0..n),\n   * usually 0 or 1, but helpful to see potential inconsistencies during wiring.\n   */\n  sessionsForFrame(frameId: FrameId): SessionId[] {\n    const info = this.frames.get(frameId);\n    return info?.ownerSessionId ? [info.ownerSessionId] : [];\n  }\n\n  /**\n   * For diagnostics: return current frame set per session.\n   */\n  framesForSession(sessionId: SessionId): FrameId[] {\n    return [...(this.framesBySession.get(sessionId) ?? new Set())];\n  }\n\n  // ---------------------- Internal helpers ----------------------\n\n  private ensureNode(fid: FrameId): void {\n    if (this.frames.has(fid)) return;\n    this.frames.set(fid, {\n      parentId: null,\n      children: new Set<FrameId>(),\n      lastSeen: shellFrame(fid),\n      ownerSessionId: undefined,\n      ownerBackendNodeId: undefined,\n    });\n  }\n\n  private renameNodeId(oldId: FrameId, newId: FrameId): void {\n    if (oldId === newId) return;\n    this.ensureNode(oldId);\n\n    const info = this.frames.get(oldId)!;\n\n    // Move info under new id\n    this.frames.delete(oldId);\n    this.frames.set(newId, { ...info });\n\n    // Fix parent’s children set\n    if (info.parentId) {\n      const p = this.frames.get(info.parentId);\n      if (p) {\n        p.children.delete(oldId);\n        p.children.add(newId);\n      }\n    }\n\n    // Fix children’s parent pointers\n    for (const c of info.children) {\n      const ci = this.frames.get(c);\n      if (ci) ci.parentId = newId;\n    }\n\n    // Fix inverse map (session -> frames)\n    if (info.ownerSessionId) {\n      const bag = this.framesBySession.get(info.ownerSessionId);\n      if (bag) {\n        bag.delete(oldId);\n        bag.add(newId);\n      }\n    }\n\n    // If root moved, keep the root id updated is handled by caller\n  }\n\n  private setOwnerSessionIdInternal(\n    frameId: FrameId,\n    sessionId: SessionId,\n  ): void {\n    this.ensureNode(frameId);\n    const info = this.frames.get(frameId)!;\n\n    // If the owner is unchanged, do nothing\n    if (info.ownerSessionId === sessionId) return;\n\n    // Remove from previous owner bag\n    if (info.ownerSessionId) {\n      const prev = this.framesBySession.get(info.ownerSessionId);\n      prev?.delete(frameId);\n      if (prev && prev.size === 0)\n        this.framesBySession.delete(info.ownerSessionId);\n    }\n\n    // Set new owner and update bag\n    info.ownerSessionId = sessionId;\n    const bag = this.framesBySession.get(sessionId) ?? new Set<FrameId>();\n    bag.add(frameId);\n    this.framesBySession.set(sessionId, bag);\n  }\n}\n"]}

package/dist/esm/lib/v3/understudy/initScripts.d.ts

@@ -0,0 +1,2 @@
+import { InitScriptSource } from "../types/private/index.js";
+export declare function normalizeInitScriptSource<Arg>(script: InitScriptSource<Arg>, arg?: Arg, caller?: string): Promise<string>;

package/dist/esm/lib/v3/understudy/initScripts.js

@@ -0,0 +1,33 @@
+import { promises as fs } from "fs";
+import { StagehandInvalidArgumentError } from "../types/public/sdkErrors.js";
+const DEFAULT_CALLER = "context.addInitScript";
+function appendSourceURL(source, filePath) {
+    const sanitized = filePath.replace(/\n/g, "");
+    return `${source}\n//# sourceURL=${sanitized}`;
+}
+export async function normalizeInitScriptSource(script, arg, caller = DEFAULT_CALLER) {
+    if (typeof script === "function") {
+        const argString = Object.is(arg, undefined)
+            ? "undefined"
+            : JSON.stringify(arg);
+        return `(${script.toString()})(${argString})`;
+    }
+    if (!Object.is(arg, undefined)) {
+        throw new StagehandInvalidArgumentError(`${caller}: 'arg' is only supported when passing a function.`);
+    }
+    if (typeof script === "string") {
+        return script;
+    }
+    if (!script || typeof script !== "object") {
+        throw new StagehandInvalidArgumentError(`${caller}: provide a string, function, or an object with path/content.`);
+    }
+    if (typeof script.content === "string") {
+        return script.content;
+    }
+    if (typeof script.path === "string" && script.path.trim()) {
+        const raw = await fs.readFile(script.path, "utf8");
+        return appendSourceURL(raw, script.path);
+    }
+    throw new StagehandInvalidArgumentError(`${caller}: provide a string, function, or an object with path/content.`);
+}
+//# sourceMappingURL=initScripts.js.map

package/dist/esm/lib/v3/understudy/initScripts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"initScripts.js","sourceRoot":"","sources":["../../../../../lib/v3/understudy/initScripts.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,IAAI,CAAC;AAEpC,OAAO,EAAE,6BAA6B,EAAE,MAAM,8BAA8B,CAAC;AAE7E,MAAM,cAAc,GAAG,uBAAuB,CAAC;AAE/C,SAAS,eAAe,CAAC,MAAc,EAAE,QAAgB;IACvD,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAC9C,OAAO,GAAG,MAAM,mBAAmB,SAAS,EAAE,CAAC;AACjD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,MAA6B,EAC7B,GAAS,EACT,SAAiB,cAAc;IAE/B,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;QACjC,MAAM,SAAS,GAAG,MAAM,CAAC,EAAE,CAAC,GAAG,EAAE,SAAS,CAAC;YACzC,CAAC,CAAC,WAAW;YACb,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QACxB,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,KAAK,SAAS,GAAG,CAAC;IAChD,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,CAAC;QAC/B,MAAM,IAAI,6BAA6B,CACrC,GAAG,MAAM,oDAAoD,CAC9D,CAAC;IACJ,CAAC;IAED,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,6BAA6B,CACrC,GAAG,MAAM,+DAA+D,CACzE,CAAC;IACJ,CAAC;IAED,IAAI,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;QACvC,OAAO,MAAM,CAAC,OAAO,CAAC;IACxB,CAAC;IAED,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;QAC1D,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACnD,OAAO,eAAe,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,IAAI,6BAA6B,CACrC,GAAG,MAAM,+DAA+D,CACzE,CAAC;AACJ,CAAC","sourcesContent":["import { promises as fs } from \"fs\";\nimport { InitScriptSource } from \"../types/private/index.js\";\nimport { StagehandInvalidArgumentError } from \"../types/public/sdkErrors.js\";\n\nconst DEFAULT_CALLER = \"context.addInitScript\";\n\nfunction appendSourceURL(source: string, filePath: string): string {\n  const sanitized = filePath.replace(/\\n/g, \"\");\n  return `${source}\\n//# sourceURL=${sanitized}`;\n}\n\nexport async function normalizeInitScriptSource<Arg>(\n  script: InitScriptSource<Arg>,\n  arg?: Arg,\n  caller: string = DEFAULT_CALLER,\n): Promise<string> {\n  if (typeof script === \"function\") {\n    const argString = Object.is(arg, undefined)\n      ? \"undefined\"\n      : JSON.stringify(arg);\n    return `(${script.toString()})(${argString})`;\n  }\n\n  if (!Object.is(arg, undefined)) {\n    throw new StagehandInvalidArgumentError(\n      `${caller}: 'arg' is only supported when passing a function.`,\n    );\n  }\n\n  if (typeof script === \"string\") {\n    return script;\n  }\n\n  if (!script || typeof script !== \"object\") {\n    throw new StagehandInvalidArgumentError(\n      `${caller}: provide a string, function, or an object with path/content.`,\n    );\n  }\n\n  if (typeof script.content === \"string\") {\n    return script.content;\n  }\n\n  if (typeof script.path === \"string\" && script.path.trim()) {\n    const raw = await fs.readFile(script.path, \"utf8\");\n    return appendSourceURL(raw, script.path);\n  }\n\n  throw new StagehandInvalidArgumentError(\n    `${caller}: provide a string, function, or an object with path/content.`,\n  );\n}\n"]}

package/dist/esm/lib/v3/understudy/lifecycleWatcher.d.ts

@@ -0,0 +1,64 @@
+import type { LoadState } from "../types/public/page.js";
+import type { CDPSessionLike } from "./cdp.js";
+import type { NetworkManager } from "./networkManager.js";
+import type { Page } from "./page.js";
+/**
+ * Coordinates page lifecycle waits (load/domcontentloaded/networkidle) while
+ * following main-frame swaps and navigation aborts. Each navigation spawns a
+ * one-off watcher that listens for relevant CDP events and resolves or rejects
+ * depending on the requested `waitUntil` state.
+ */
+/**
+ * Small utility that mirrors Playwright's lifecycle watcher semantics. Bridges
+ * main-frame lifecycle events with the NetworkManager's idle signal so callers
+ * can await `load`, `domcontentloaded`, or `networkidle` with a single promise.
+ */
+export declare class LifecycleWatcher {
+    private readonly page;
+    private readonly mainSession;
+    private readonly networkManager;
+    private readonly waitUntil;
+    private readonly timeoutMs;
+    private readonly startTime;
+    private readonly navigationCommandId;
+    private currentLoaderId;
+    private idleStartTime;
+    private cleanupCallbacks;
+    private idleHandle;
+    private abortReject;
+    private abortPromise;
+    private abortError;
+    private disposed;
+    private expectedLoaderId;
+    private initialLoaderId;
+    private pendingFollowupNavigation;
+    /**
+     * Create a watcher; callers should subsequently invoke {@link wait}.
+     */
+    constructor(params: {
+        page: Page;
+        mainSession: CDPSessionLike;
+        networkManager: NetworkManager;
+        waitUntil: LoadState;
+        timeoutMs: number;
+        navigationCommandId: number;
+    });
+    /** Hint the watcher with the loader id returned by Page.navigate. */
+    setExpectedLoaderId(loaderId: string | undefined): void;
+    /** Wait for the requested lifecycle state or throw on timeout/abort. */
+    wait(): Promise<void>;
+    /** Cancel any outstanding network-idle waits and remove event listeners. */
+    dispose(): void;
+    /** Subscribe to main-frame events to detect abort conditions. */
+    private installSessionListeners;
+    /** Compute remaining time until the shared deadline elapses. */
+    private timeRemaining;
+    /** Await an operation but abort early if navigation replacement fires. */
+    private awaitWithAbort;
+    /** Mark the watcher as aborted and reject any pending waiters. */
+    private triggerAbort;
+    private waitForNetworkIdle;
+    private shouldRestartAfterFollowup;
+    private adoptNewMainLoader;
+    private buildIdleFilter;
+}