mirror of
https://github.com/Tencent/BrowserSkill.git
synced 2026-09-10 13:01:23 +08:00
refactor(observation): organize capture around document facts
This commit is contained in:
@@ -4,9 +4,9 @@ import { CAPTURE_SUPPRESS, type CaptureSuppressMessage } from "@/lib/capture-sup
|
|||||||
import { OVERLAY_HOST_MARKER_ATTR, OVERLAY_HOST_NAME } from "@/lib/overlay-bridge";
|
import { OVERLAY_HOST_MARKER_ATTR, OVERLAY_HOST_NAME } from "@/lib/overlay-bridge";
|
||||||
import { SessionManager } from "@/session-manager/manager";
|
import { SessionManager } from "@/session-manager/manager";
|
||||||
import type { CdpRunner } from "@/tools/shared";
|
import type { CdpRunner } from "@/tools/shared";
|
||||||
|
import type { BuildVomSceneOptions, VomFrameDocument } from "../observation";
|
||||||
import {
|
import {
|
||||||
buildFrameVomScene,
|
buildFrameVomScene,
|
||||||
buildVomScene,
|
|
||||||
type CdpAxNode,
|
type CdpAxNode,
|
||||||
captureVomObservation,
|
captureVomObservation,
|
||||||
handleGetHtml,
|
handleGetHtml,
|
||||||
@@ -17,7 +17,35 @@ import {
|
|||||||
type ScreenshotDeps,
|
type ScreenshotDeps,
|
||||||
stripDataUrlPrefix,
|
stripDataUrlPrefix,
|
||||||
} from "../observation";
|
} from "../observation";
|
||||||
import type { CapturedNode, CapturedViewModel } from "../vom/capture";
|
import type { CapturedNode, CapturedSceneInput } from "../vom/facts";
|
||||||
|
|
||||||
|
interface SceneFixture extends CapturedSceneInput {
|
||||||
|
documents?: VomFrameDocument[];
|
||||||
|
}
|
||||||
|
|
||||||
|
// Only supplies the common root fixture; child documents declare their own ownership.
|
||||||
|
function buildTestScene(
|
||||||
|
axNodes: CdpAxNode[],
|
||||||
|
captured: SceneFixture,
|
||||||
|
options: BuildVomSceneOptions = {},
|
||||||
|
) {
|
||||||
|
const rootFrameId = captured.rootFrameId ?? "root";
|
||||||
|
return buildFrameVomScene(
|
||||||
|
[
|
||||||
|
{
|
||||||
|
frameId: rootFrameId,
|
||||||
|
contextScopeId: rootFrameId,
|
||||||
|
target: { tabId: 7 },
|
||||||
|
url: options.pageUrl,
|
||||||
|
domNodes: captured.nodes,
|
||||||
|
axNodes,
|
||||||
|
},
|
||||||
|
...(captured.documents ?? []),
|
||||||
|
],
|
||||||
|
captured,
|
||||||
|
options,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
function fakeAgentWindow(ids: number[]) {
|
function fakeAgentWindow(ids: number[]) {
|
||||||
let i = 0;
|
let i = 0;
|
||||||
@@ -522,10 +550,10 @@ describe("handleScreenshot overlay suppression", () => {
|
|||||||
});
|
});
|
||||||
|
|
||||||
// ---------------------------------------------------------------------------
|
// ---------------------------------------------------------------------------
|
||||||
// buildVomScene
|
// Semantic scene fixtures
|
||||||
// ---------------------------------------------------------------------------
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
describe("buildVomScene", () => {
|
describe("buildFrameVomScene", () => {
|
||||||
it("merges multiple sibling and nested frames without mixing their targets", () => {
|
it("merges multiple sibling and nested frames without mixing their targets", () => {
|
||||||
const node = (
|
const node = (
|
||||||
backendNodeId: number,
|
backendNodeId: number,
|
||||||
@@ -563,25 +591,10 @@ describe("buildVomScene", () => {
|
|||||||
node(300, null, "nested", "body"),
|
node(300, null, "nested", "body"),
|
||||||
node(301, 300, "nested", "input", { placeholder: "Nested field" }),
|
node(301, 300, "nested", "input", { placeholder: "Nested field" }),
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 1200, height: 800 },
|
viewport: { width: 1200, height: 800 },
|
||||||
nodes: mainNodes,
|
nodes: mainNodes,
|
||||||
iframeNodes: new Map([
|
|
||||||
[10, firstNodes],
|
|
||||||
[20, secondNodes],
|
|
||||||
[210, nestedNodes],
|
|
||||||
]),
|
|
||||||
frameNodes: new Map([
|
|
||||||
["main", mainNodes],
|
|
||||||
["first", firstNodes],
|
|
||||||
["second", secondNodes],
|
|
||||||
["nested", nestedNodes],
|
|
||||||
]),
|
|
||||||
frameOwnerBackendNodeIds: new Map([
|
|
||||||
["first", 10],
|
|
||||||
["second", 20],
|
|
||||||
["nested", 210],
|
|
||||||
]),
|
|
||||||
rootFrameId: "main",
|
rootFrameId: "main",
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
};
|
};
|
||||||
@@ -680,11 +693,10 @@ describe("buildVomScene", () => {
|
|||||||
position: "static",
|
position: "static",
|
||||||
pointerEvents: "auto",
|
pointerEvents: "auto",
|
||||||
};
|
};
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 800, height: 600 },
|
viewport: { width: 800, height: 600 },
|
||||||
nodes: [],
|
nodes: [],
|
||||||
iframeNodes: new Map(),
|
|
||||||
frameNodes: new Map([["child", [childNode]]]),
|
|
||||||
rootFrameId: "main",
|
rootFrameId: "main",
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
};
|
};
|
||||||
@@ -746,14 +758,10 @@ describe("buildVomScene", () => {
|
|||||||
pointerEvents: "auto",
|
pointerEvents: "auto",
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 800, height: 600 },
|
viewport: { width: 800, height: 600 },
|
||||||
nodes: mainNodes,
|
nodes: mainNodes,
|
||||||
iframeNodes: new Map(),
|
|
||||||
frameNodes: new Map([
|
|
||||||
["main", mainNodes],
|
|
||||||
["child", []],
|
|
||||||
]),
|
|
||||||
rootFrameId: "main",
|
rootFrameId: "main",
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
};
|
};
|
||||||
@@ -829,9 +837,8 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 200,
|
backendDOMNodeId: 200,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -857,11 +864,11 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
const scene = buildVomScene(axNodes, captured);
|
const scene = buildTestScene(axNodes, captured);
|
||||||
expect(scene.viewport).toEqual({ width: 1000, height: 800 });
|
expect(scene.viewport).toEqual({ width: 1000, height: 800 });
|
||||||
expect(scene.nodes[0]).toEqual(
|
expect(scene.nodes[0]).toEqual(
|
||||||
expect.objectContaining({
|
expect.objectContaining({
|
||||||
id: 100,
|
backendNodeId: 100,
|
||||||
parentId: null,
|
parentId: null,
|
||||||
role: "RootWebArea",
|
role: "RootWebArea",
|
||||||
name: "Example",
|
name: "Example",
|
||||||
@@ -873,14 +880,14 @@ describe("buildVomScene", () => {
|
|||||||
);
|
);
|
||||||
expect(scene.nodes[1]).toEqual(
|
expect(scene.nodes[1]).toEqual(
|
||||||
expect.objectContaining({
|
expect.objectContaining({
|
||||||
id: 200,
|
backendNodeId: 200,
|
||||||
parentId: 100,
|
parentId: scene.nodes.find((node) => node.backendNodeId === 100)!.id,
|
||||||
role: "button",
|
role: "button",
|
||||||
name: "Submit",
|
name: "Submit",
|
||||||
tag: "button",
|
tag: "button",
|
||||||
rect: { x: 20, y: 20, w: 120, h: 40 },
|
rect: { x: 20, y: 20, w: 120, h: 40 },
|
||||||
domParentId: 100,
|
domParentId: scene.nodes.find((node) => node.backendNodeId === 100)!.id,
|
||||||
domAncestorIds: [100],
|
domAncestorIds: [scene.nodes.find((node) => node.backendNodeId === 100)!.id],
|
||||||
attrs: {},
|
attrs: {},
|
||||||
}),
|
}),
|
||||||
);
|
);
|
||||||
@@ -909,9 +916,8 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 300,
|
backendDOMNodeId: 300,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -947,10 +953,10 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
const scene = buildVomScene(axNodes, captured, { pageUrl: "https://app.example.test/home" });
|
const scene = buildTestScene(axNodes, captured, { pageUrl: "https://app.example.test/home" });
|
||||||
|
|
||||||
expect(scene.nodes.find((node) => node.id === 200)?.href).toBe("docs.example.org");
|
expect(scene.nodes.find((node) => node.backendNodeId === 200)?.href).toBe("docs.example.org");
|
||||||
expect(scene.nodes.find((node) => node.id === 300)?.href).toBeUndefined();
|
expect(scene.nodes.find((node) => node.backendNodeId === 300)?.href).toBeUndefined();
|
||||||
});
|
});
|
||||||
|
|
||||||
it("uses the nearest backend AX ancestor as parentId", () => {
|
it("uses the nearest backend AX ancestor as parentId", () => {
|
||||||
@@ -975,9 +981,8 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 30,
|
backendDOMNodeId: 30,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 400, height: 300 },
|
viewport: { width: 400, height: 300 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -1003,10 +1008,12 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
const scene = buildVomScene(axNodes, captured);
|
const scene = buildTestScene(axNodes, captured);
|
||||||
|
|
||||||
expect(scene.nodes.map((node) => node.id)).toEqual([10, 30]);
|
expect(scene.nodes.map((node) => node.backendNodeId)).toEqual([10, 30]);
|
||||||
expect(scene.nodes.find((node) => node.id === 30)?.parentId).toBe(10);
|
expect(scene.nodes.find((node) => node.backendNodeId === 30)?.parentId).toBe(
|
||||||
|
scene.nodes.find((node) => node.backendNodeId === 10)!.id,
|
||||||
|
);
|
||||||
});
|
});
|
||||||
|
|
||||||
it("maps iframe sub-document controls to VomNodes", () => {
|
it("maps iframe sub-document controls to VomNodes", () => {
|
||||||
@@ -1024,7 +1031,7 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 20,
|
backendDOMNodeId: 20,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -1048,10 +1055,15 @@ describe("buildVomScene", () => {
|
|||||||
pointerEvents: "auto",
|
pointerEvents: "auto",
|
||||||
},
|
},
|
||||||
],
|
],
|
||||||
iframeNodes: new Map([
|
documents: [
|
||||||
[
|
{
|
||||||
20,
|
frameId: "child",
|
||||||
[
|
contextScopeId: "child",
|
||||||
|
parentFrameId: "root",
|
||||||
|
ownerBackendNodeId: 20,
|
||||||
|
target: { tabId: 7 },
|
||||||
|
axNodes: [],
|
||||||
|
domNodes: [
|
||||||
{
|
{
|
||||||
backendNodeId: 101,
|
backendNodeId: 101,
|
||||||
parentBackendNodeId: null,
|
parentBackendNodeId: null,
|
||||||
@@ -1073,26 +1085,25 @@ describe("buildVomScene", () => {
|
|||||||
pointerEvents: "auto",
|
pointerEvents: "auto",
|
||||||
},
|
},
|
||||||
],
|
],
|
||||||
],
|
},
|
||||||
]),
|
],
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
};
|
};
|
||||||
|
|
||||||
const iframeControls = buildVomScene(axNodes, captured).nodes.filter((node) =>
|
const scene = buildTestScene(axNodes, captured);
|
||||||
[101, 102].includes(node.id),
|
const iframeControls = scene.nodes.filter((node) => [101, 102].includes(node.backendNodeId!));
|
||||||
);
|
|
||||||
|
|
||||||
expect(iframeControls).toEqual([
|
expect(iframeControls).toEqual([
|
||||||
expect.objectContaining({
|
expect.objectContaining({
|
||||||
id: 101,
|
backendNodeId: 101,
|
||||||
parentId: 20,
|
parentId: scene.nodes.find((node) => node.backendNodeId === 20)!.id,
|
||||||
role: "textbox",
|
role: "textbox",
|
||||||
name: "请输入手机号",
|
name: "请输入手机号",
|
||||||
sensitive: false,
|
sensitive: false,
|
||||||
}),
|
}),
|
||||||
expect.objectContaining({
|
expect.objectContaining({
|
||||||
id: 102,
|
backendNodeId: 102,
|
||||||
parentId: 20,
|
parentId: scene.nodes.find((node) => node.backendNodeId === 20)!.id,
|
||||||
role: "textbox",
|
role: "textbox",
|
||||||
name: "密码",
|
name: "密码",
|
||||||
sensitive: true,
|
sensitive: true,
|
||||||
@@ -1113,11 +1124,9 @@ describe("buildVomScene", () => {
|
|||||||
parentId: "1",
|
parentId: "1",
|
||||||
role: { type: "role", value: "Iframe" },
|
role: { type: "role", value: "Iframe" },
|
||||||
backendDOMNodeId: 20,
|
backendDOMNodeId: 20,
|
||||||
childIds: ["3"],
|
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
nodeId: "3",
|
nodeId: "3",
|
||||||
parentId: "2",
|
|
||||||
role: { type: "role", value: "textbox" },
|
role: { type: "role", value: "textbox" },
|
||||||
name: { type: "x", value: "输入密码" },
|
name: { type: "x", value: "输入密码" },
|
||||||
value: { value: "iframe-secret" },
|
value: { value: "iframe-secret" },
|
||||||
@@ -1125,7 +1134,7 @@ describe("buildVomScene", () => {
|
|||||||
properties: [{ name: "inputType", value: { value: "password" } }],
|
properties: [{ name: "inputType", value: { value: "password" } }],
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -1149,10 +1158,15 @@ describe("buildVomScene", () => {
|
|||||||
pointerEvents: "auto",
|
pointerEvents: "auto",
|
||||||
},
|
},
|
||||||
],
|
],
|
||||||
iframeNodes: new Map([
|
documents: [
|
||||||
[
|
{
|
||||||
20,
|
frameId: "child",
|
||||||
[
|
contextScopeId: "child",
|
||||||
|
parentFrameId: "root",
|
||||||
|
ownerBackendNodeId: 20,
|
||||||
|
target: { tabId: 7 },
|
||||||
|
axNodes: [axNodes[2]],
|
||||||
|
domNodes: [
|
||||||
{
|
{
|
||||||
backendNodeId: 102,
|
backendNodeId: 102,
|
||||||
parentBackendNodeId: null,
|
parentBackendNodeId: null,
|
||||||
@@ -1164,25 +1178,27 @@ describe("buildVomScene", () => {
|
|||||||
pointerEvents: "auto",
|
pointerEvents: "auto",
|
||||||
},
|
},
|
||||||
],
|
],
|
||||||
],
|
},
|
||||||
]),
|
],
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
};
|
};
|
||||||
|
|
||||||
const passwordNodes = buildVomScene(axNodes, captured).nodes.filter(
|
const passwordNodes = buildTestScene(axNodes.slice(0, 2), captured).nodes.filter(
|
||||||
(node) => node.role === "textbox" && node.name === "输入密码",
|
(node) => node.role === "textbox" && node.name === "输入密码",
|
||||||
);
|
);
|
||||||
|
|
||||||
expect(passwordNodes).toHaveLength(1);
|
expect(passwordNodes).toHaveLength(1);
|
||||||
expect(passwordNodes[0]).toEqual(expect.objectContaining({ id: 202, sensitive: true }));
|
expect(passwordNodes[0]).toEqual(
|
||||||
|
expect.objectContaining({ backendNodeId: 202, sensitive: true }),
|
||||||
|
);
|
||||||
expect(passwordNodes[0].value).toBeUndefined();
|
expect(passwordNodes[0].value).toBeUndefined();
|
||||||
const rendered = renderVom(buildVomScene(axNodes, captured)).text;
|
const rendered = renderVom(buildTestScene(axNodes.slice(0, 2), captured)).text;
|
||||||
expect(rendered).toContain('textbox "输入密码" [filled] ="•••"');
|
expect(rendered).toContain('textbox "输入密码" [filled] ="•••"');
|
||||||
expect(rendered).not.toContain("iframe-secret");
|
expect(rendered).not.toContain("iframe-secret");
|
||||||
});
|
});
|
||||||
|
|
||||||
it("keeps unnamed iframe controls but skips unnamed iframe links", () => {
|
it("keeps unnamed iframe controls but skips unnamed iframe links", () => {
|
||||||
const scene = buildVomScene(
|
const scene = buildTestScene(
|
||||||
[
|
[
|
||||||
{
|
{
|
||||||
nodeId: "1",
|
nodeId: "1",
|
||||||
@@ -1221,10 +1237,15 @@ describe("buildVomScene", () => {
|
|||||||
pointerEvents: "auto",
|
pointerEvents: "auto",
|
||||||
},
|
},
|
||||||
],
|
],
|
||||||
iframeNodes: new Map([
|
documents: [
|
||||||
[
|
{
|
||||||
20,
|
frameId: "child",
|
||||||
[
|
contextScopeId: "child",
|
||||||
|
parentFrameId: "root",
|
||||||
|
ownerBackendNodeId: 20,
|
||||||
|
target: { tabId: 7 },
|
||||||
|
axNodes: [],
|
||||||
|
domNodes: [
|
||||||
{
|
{
|
||||||
backendNodeId: 201,
|
backendNodeId: 201,
|
||||||
parentBackendNodeId: null,
|
parentBackendNodeId: null,
|
||||||
@@ -1246,25 +1267,25 @@ describe("buildVomScene", () => {
|
|||||||
pointerEvents: "auto",
|
pointerEvents: "auto",
|
||||||
},
|
},
|
||||||
],
|
],
|
||||||
],
|
},
|
||||||
]),
|
],
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
},
|
},
|
||||||
);
|
);
|
||||||
|
|
||||||
expect(scene.nodes.find((node) => node.id === 201)).toEqual(
|
expect(scene.nodes.find((node) => node.backendNodeId === 201)).toEqual(
|
||||||
expect.objectContaining({
|
expect.objectContaining({
|
||||||
id: 201,
|
backendNodeId: 201,
|
||||||
parentId: 20,
|
parentId: scene.nodes.find((node) => node.backendNodeId === 20)!.id,
|
||||||
role: "button",
|
role: "button",
|
||||||
}),
|
}),
|
||||||
);
|
);
|
||||||
expect(scene.nodes.find((node) => node.id === 201)).not.toHaveProperty("name");
|
expect(scene.nodes.find((node) => node.backendNodeId === 201)).not.toHaveProperty("name");
|
||||||
expect(scene.nodes.find((node) => node.id === 202)).toBeUndefined();
|
expect(scene.nodes.find((node) => node.backendNodeId === 202)).toBeUndefined();
|
||||||
});
|
});
|
||||||
|
|
||||||
it("preserves captured-only iframe anchors for iframe sub-document controls", () => {
|
it("preserves captured-only iframe anchors for iframe sub-document controls", () => {
|
||||||
const scene = buildVomScene(
|
const scene = buildTestScene(
|
||||||
[
|
[
|
||||||
{
|
{
|
||||||
nodeId: "1",
|
nodeId: "1",
|
||||||
@@ -1297,10 +1318,15 @@ describe("buildVomScene", () => {
|
|||||||
pointerEvents: "auto",
|
pointerEvents: "auto",
|
||||||
},
|
},
|
||||||
],
|
],
|
||||||
iframeNodes: new Map<number, CapturedNode[]>([
|
documents: [
|
||||||
[
|
{
|
||||||
20,
|
frameId: "child",
|
||||||
[
|
contextScopeId: "child",
|
||||||
|
parentFrameId: "root",
|
||||||
|
ownerBackendNodeId: 20,
|
||||||
|
target: { tabId: 7 },
|
||||||
|
axNodes: [],
|
||||||
|
domNodes: [
|
||||||
{
|
{
|
||||||
backendNodeId: 201,
|
backendNodeId: 201,
|
||||||
parentBackendNodeId: null,
|
parentBackendNodeId: null,
|
||||||
@@ -1322,16 +1348,16 @@ describe("buildVomScene", () => {
|
|||||||
pointerEvents: "auto",
|
pointerEvents: "auto",
|
||||||
},
|
},
|
||||||
],
|
],
|
||||||
],
|
},
|
||||||
]),
|
],
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
},
|
},
|
||||||
);
|
);
|
||||||
|
|
||||||
expect(scene.nodes.find((node) => node.id === 20)).toEqual(
|
expect(scene.nodes.find((node) => node.backendNodeId === 20)).toEqual(
|
||||||
expect.objectContaining({
|
expect.objectContaining({
|
||||||
id: 20,
|
backendNodeId: 20,
|
||||||
parentId: 10,
|
parentId: scene.nodes.find((node) => node.backendNodeId === 10)!.id,
|
||||||
role: "Iframe",
|
role: "Iframe",
|
||||||
name: "支付验证",
|
name: "支付验证",
|
||||||
}),
|
}),
|
||||||
@@ -1352,9 +1378,8 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 10,
|
backendDOMNodeId: 10,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 320, height: 240 },
|
viewport: { width: 320, height: 240 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -1375,9 +1400,9 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
expect(buildVomScene(axNodes, captured).nodes[0]).toEqual(
|
expect(buildTestScene(axNodes, captured).nodes[0]).toEqual(
|
||||||
expect.objectContaining({
|
expect.objectContaining({
|
||||||
id: 10,
|
backendNodeId: 10,
|
||||||
sensitive: true,
|
sensitive: true,
|
||||||
}),
|
}),
|
||||||
);
|
);
|
||||||
@@ -1392,9 +1417,8 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 10,
|
backendDOMNodeId: 10,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -1411,9 +1435,9 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
expect(buildVomScene(axNodes, captured).nodes[0]).toEqual(
|
expect(buildTestScene(axNodes, captured).nodes[0]).toEqual(
|
||||||
expect.objectContaining({
|
expect.objectContaining({
|
||||||
id: 10,
|
backendNodeId: 10,
|
||||||
role: "textbox",
|
role: "textbox",
|
||||||
name: "+86",
|
name: "+86",
|
||||||
placeholder: "输入手机号",
|
placeholder: "输入手机号",
|
||||||
@@ -1431,9 +1455,8 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 10,
|
backendDOMNodeId: 10,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -1450,8 +1473,13 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
expect(buildVomScene(axNodes, captured).nodes[0]).toEqual(
|
expect(buildTestScene(axNodes, captured).nodes[0]).toEqual(
|
||||||
expect.objectContaining({ id: 10, role: "textbox", name: "Email", value: "a@b.com" }),
|
expect.objectContaining({
|
||||||
|
backendNodeId: 10,
|
||||||
|
role: "textbox",
|
||||||
|
name: "Email",
|
||||||
|
value: "a@b.com",
|
||||||
|
}),
|
||||||
);
|
);
|
||||||
});
|
});
|
||||||
|
|
||||||
@@ -1470,9 +1498,8 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 20,
|
backendDOMNodeId: 20,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -1510,9 +1537,9 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
expect(buildVomScene(axNodes, captured).nodes.find((node) => node.id === 20)).toEqual(
|
expect(
|
||||||
expect.objectContaining({ role: "textbox", name: "验证码" }),
|
buildTestScene(axNodes, captured).nodes.find((node) => node.backendNodeId === 20),
|
||||||
);
|
).toEqual(expect.objectContaining({ role: "textbox", name: "验证码" }));
|
||||||
});
|
});
|
||||||
|
|
||||||
it("uses preceding AX static text as a form control label", () => {
|
it("uses preceding AX static text as a form control label", () => {
|
||||||
@@ -1536,9 +1563,8 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 20,
|
backendDOMNodeId: 20,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -1555,13 +1581,13 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
expect(buildVomScene(axNodes, captured).nodes.find((node) => node.id === 20)).toEqual(
|
expect(
|
||||||
expect.objectContaining({ role: "textbox", name: "验证码", inputState: "empty" }),
|
buildTestScene(axNodes, captured).nodes.find((node) => node.backendNodeId === 20),
|
||||||
);
|
).toEqual(expect.objectContaining({ role: "textbox", name: "验证码", inputState: "empty" }));
|
||||||
});
|
});
|
||||||
|
|
||||||
it("marks runtime default input values without treating them as ordinary filled input", () => {
|
it("marks runtime default input values without treating them as ordinary filled input", () => {
|
||||||
const scene = buildVomScene(
|
const scene = buildTestScene(
|
||||||
[
|
[
|
||||||
{
|
{
|
||||||
nodeId: "1",
|
nodeId: "1",
|
||||||
@@ -1571,7 +1597,6 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
{
|
{
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -1615,9 +1640,8 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 20,
|
backendDOMNodeId: 20,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -1667,9 +1691,14 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
const scene = buildVomScene(axNodes, captured);
|
const scene = buildTestScene(axNodes, captured);
|
||||||
expect(scene.nodes.find((n) => n.id === 20)).toEqual(
|
expect(scene.nodes.find((n) => n.backendNodeId === 20)).toEqual(
|
||||||
expect.objectContaining({ id: 20, role: "button", name: "close", cursor: "pointer" }),
|
expect.objectContaining({
|
||||||
|
backendNodeId: 20,
|
||||||
|
role: "button",
|
||||||
|
name: "close",
|
||||||
|
cursor: "pointer",
|
||||||
|
}),
|
||||||
);
|
);
|
||||||
const rendered = renderVom(scene);
|
const rendered = renderVom(scene);
|
||||||
expect(rendered.text).toContain('@e1 button "close"');
|
expect(rendered.text).toContain('@e1 button "close"');
|
||||||
@@ -1703,9 +1732,8 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 30,
|
backendDOMNodeId: 30,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -1744,9 +1772,9 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
const scene = buildVomScene(axNodes, captured);
|
const scene = buildTestScene(axNodes, captured);
|
||||||
expect(scene.nodes.find((n) => n.id === 20)).toBeUndefined();
|
expect(scene.nodes.find((n) => n.backendNodeId === 20)).toBeUndefined();
|
||||||
expect(scene.nodes.find((n) => n.id === 30)?.role).toBe("link");
|
expect(scene.nodes.find((n) => n.backendNodeId === 30)?.role).toBe("link");
|
||||||
});
|
});
|
||||||
|
|
||||||
it("promotes only the outermost clickable in a nested pointer chain", () => {
|
it("promotes only the outermost clickable in a nested pointer chain", () => {
|
||||||
@@ -1771,9 +1799,8 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 30,
|
backendDOMNodeId: 30,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -1812,11 +1839,15 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
const scene = buildVomScene(axNodes, captured);
|
const scene = buildTestScene(axNodes, captured);
|
||||||
expect(scene.nodes.find((n) => n.id === 20)).toEqual(
|
expect(scene.nodes.find((n) => n.backendNodeId === 20)).toEqual(
|
||||||
expect.objectContaining({ id: 20, role: "button", attrs: { "aria-label": "收藏" } }),
|
expect.objectContaining({
|
||||||
|
backendNodeId: 20,
|
||||||
|
role: "button",
|
||||||
|
attrs: { "aria-label": "收藏" },
|
||||||
|
}),
|
||||||
);
|
);
|
||||||
expect(scene.nodes.find((n) => n.id === 30)).toBeUndefined();
|
expect(scene.nodes.find((n) => n.backendNodeId === 30)).toBeUndefined();
|
||||||
const rendered = renderVom(scene);
|
const rendered = renderVom(scene);
|
||||||
expect(rendered.text).toContain('@e1 button "收藏"');
|
expect(rendered.text).toContain('@e1 button "收藏"');
|
||||||
expect(rendered.refs.map(({ ref, backendNodeId }) => ({ ref, backendNodeId }))).toEqual([
|
expect(rendered.refs.map(({ ref, backendNodeId }) => ({ ref, backendNodeId }))).toEqual([
|
||||||
@@ -1840,9 +1871,8 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 20,
|
backendDOMNodeId: 20,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -1890,10 +1920,10 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
const scene = buildVomScene(axNodes, captured);
|
const scene = buildTestScene(axNodes, captured);
|
||||||
expect(scene.activeScopeBlocks).toEqual([
|
expect(scene.activeScopeBlocks).toEqual([
|
||||||
{
|
{
|
||||||
triggerId: 20,
|
triggerId: scene.nodes.find((node) => node.backendNodeId === 20)!.id,
|
||||||
label: "Reviews (12)",
|
label: "Reviews (12)",
|
||||||
lines: ["Jane - ear cups are small", "Bob - great sound"],
|
lines: ["Jane - ear cups are small", "Bob - great sound"],
|
||||||
},
|
},
|
||||||
@@ -1917,11 +1947,10 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 20,
|
backendDOMNodeId: 20,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const scene = buildVomScene(
|
const scene = buildTestScene(
|
||||||
axNodes,
|
axNodes,
|
||||||
{
|
{
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -1954,7 +1983,11 @@ describe("buildVomScene", () => {
|
|||||||
);
|
);
|
||||||
|
|
||||||
expect(scene.surfaces).toEqual([
|
expect(scene.surfaces).toEqual([
|
||||||
{ triggerId: 20, triggerAction: "hover", subItems: ["Shoes", "Bags"] },
|
{
|
||||||
|
triggerId: scene.nodes.find((node) => node.backendNodeId === 20)!.id,
|
||||||
|
triggerAction: "hover",
|
||||||
|
subItems: ["Shoes", "Bags"],
|
||||||
|
},
|
||||||
]);
|
]);
|
||||||
expect(renderVom(scene).text).toContain('@e1 button "Products" [hover first: Shoes | Bags]');
|
expect(renderVom(scene).text).toContain('@e1 button "Products" [hover first: Shoes | Bags]');
|
||||||
});
|
});
|
||||||
@@ -1975,11 +2008,10 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 21,
|
backendDOMNodeId: 21,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const scene = buildVomScene(
|
const scene = buildTestScene(
|
||||||
axNodes,
|
axNodes,
|
||||||
{
|
{
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -2040,11 +2072,10 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 21,
|
backendDOMNodeId: 21,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const scene = buildVomScene(
|
const scene = buildTestScene(
|
||||||
axNodes,
|
axNodes,
|
||||||
{
|
{
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -2089,7 +2120,11 @@ describe("buildVomScene", () => {
|
|||||||
);
|
);
|
||||||
|
|
||||||
expect(scene.surfaces).toEqual([
|
expect(scene.surfaces).toEqual([
|
||||||
{ triggerId: 21, triggerAction: "hover", subItems: ["My profile"] },
|
{
|
||||||
|
triggerId: scene.nodes.find((node) => node.backendNodeId === 21)!.id,
|
||||||
|
triggerAction: "hover",
|
||||||
|
subItems: ["My profile"],
|
||||||
|
},
|
||||||
]);
|
]);
|
||||||
});
|
});
|
||||||
|
|
||||||
@@ -2109,11 +2144,10 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 21,
|
backendDOMNodeId: 21,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const scene = buildVomScene(
|
const scene = buildTestScene(
|
||||||
axNodes,
|
axNodes,
|
||||||
{
|
{
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -2164,7 +2198,11 @@ describe("buildVomScene", () => {
|
|||||||
);
|
);
|
||||||
|
|
||||||
expect(scene.surfaces).toEqual([
|
expect(scene.surfaces).toEqual([
|
||||||
{ triggerId: 20, triggerAction: "hover", subItems: ["My profile", "Sign out"] },
|
{
|
||||||
|
triggerId: scene.nodes.find((node) => node.backendNodeId === 20)!.id,
|
||||||
|
triggerAction: "hover",
|
||||||
|
subItems: ["My profile", "Sign out"],
|
||||||
|
},
|
||||||
]);
|
]);
|
||||||
expect(renderVom(scene).text).toContain(
|
expect(renderVom(scene).text).toContain(
|
||||||
'@e1 button "image" [hover first: My profile | Sign out]',
|
'@e1 button "image" [hover first: My profile | Sign out]',
|
||||||
@@ -2187,11 +2225,10 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 42,
|
backendDOMNodeId: 42,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const scene = buildVomScene(
|
const scene = buildTestScene(
|
||||||
axNodes,
|
axNodes,
|
||||||
{
|
{
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -2261,9 +2298,8 @@ describe("buildVomScene", () => {
|
|||||||
properties: [{ name: "inputType", value: { value: "credit-card" } }],
|
properties: [{ name: "inputType", value: { value: "credit-card" } }],
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const scene = buildVomScene(axNodes, {
|
const scene = buildTestScene(axNodes, {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -2310,7 +2346,7 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
});
|
});
|
||||||
|
|
||||||
expect(scene.nodes.find((node) => node.id === 20)).toEqual(
|
expect(scene.nodes.find((node) => node.backendNodeId === 20)).toEqual(
|
||||||
expect.objectContaining({
|
expect.objectContaining({
|
||||||
name: "Products [expanded]",
|
name: "Products [expanded]",
|
||||||
attrs: expect.objectContaining({
|
attrs: expect.objectContaining({
|
||||||
@@ -2320,13 +2356,17 @@ describe("buildVomScene", () => {
|
|||||||
}),
|
}),
|
||||||
);
|
);
|
||||||
expect(scene.activeScopeBlocks).toEqual([
|
expect(scene.activeScopeBlocks).toEqual([
|
||||||
{ triggerId: 20, label: "Products [expanded]", lines: ["Shoes Bags"] },
|
{
|
||||||
|
triggerId: scene.nodes.find((node) => node.backendNodeId === 20)!.id,
|
||||||
|
label: "Products [expanded]",
|
||||||
|
lines: ["Shoes Bags"],
|
||||||
|
},
|
||||||
]);
|
]);
|
||||||
expect(scene.nodes.find((node) => node.id === 30)?.sensitive).toBe(true);
|
expect(scene.nodes.find((node) => node.backendNodeId === 30)?.sensitive).toBe(true);
|
||||||
});
|
});
|
||||||
|
|
||||||
it("aggregates AX virtual text into unnamed structural nodes", () => {
|
it("aggregates AX virtual text into unnamed structural nodes", () => {
|
||||||
const scene = buildVomScene(
|
const scene = buildTestScene(
|
||||||
[
|
[
|
||||||
{
|
{
|
||||||
nodeId: "1",
|
nodeId: "1",
|
||||||
@@ -2350,7 +2390,6 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
{
|
{
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -2377,13 +2416,12 @@ describe("buildVomScene", () => {
|
|||||||
},
|
},
|
||||||
);
|
);
|
||||||
|
|
||||||
expect(scene.nodes.find((node) => node.id === 20)?.name).toBe("Inline only text");
|
expect(scene.nodes.find((node) => node.backendNodeId === 20)?.name).toBe("Inline only text");
|
||||||
});
|
});
|
||||||
|
|
||||||
it("marks captured dialog elements as modal without AX role or aria-modal", () => {
|
it("marks captured dialog elements as modal without AX role or aria-modal", () => {
|
||||||
const scene = buildVomScene([], {
|
const scene = buildTestScene([], {
|
||||||
viewport: { width: 640, height: 480 },
|
viewport: { width: 640, height: 480 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set(),
|
excludedBackendNodeIds: new Set(),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -2401,7 +2439,7 @@ describe("buildVomScene", () => {
|
|||||||
|
|
||||||
expect(scene.nodes[0]).toEqual(
|
expect(scene.nodes[0]).toEqual(
|
||||||
expect.objectContaining({
|
expect.objectContaining({
|
||||||
id: 50,
|
backendNodeId: 50,
|
||||||
modal: true,
|
modal: true,
|
||||||
}),
|
}),
|
||||||
);
|
);
|
||||||
@@ -2430,9 +2468,8 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 202,
|
backendDOMNodeId: 202,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const captured: CapturedViewModel = {
|
const captured: SceneFixture = {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set([200, 201, 202]),
|
excludedBackendNodeIds: new Set([200, 201, 202]),
|
||||||
nodes: [
|
nodes: [
|
||||||
{
|
{
|
||||||
@@ -2448,9 +2485,9 @@ describe("buildVomScene", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
const scene = buildVomScene(axNodes, captured);
|
const scene = buildTestScene(axNodes, captured);
|
||||||
expect(scene.nodes.find((n) => n.id === 202)).toBeUndefined();
|
expect(scene.nodes.find((n) => n.backendNodeId === 202)).toBeUndefined();
|
||||||
expect(scene.nodes.find((n) => n.id === 2)).toEqual(
|
expect(scene.nodes.find((n) => n.backendNodeId === 2)).toEqual(
|
||||||
expect.objectContaining({ role: "button", name: "Real Button" }),
|
expect.objectContaining({ role: "button", name: "Real Button" }),
|
||||||
);
|
);
|
||||||
});
|
});
|
||||||
@@ -2478,13 +2515,12 @@ describe("buildVomScene", () => {
|
|||||||
backendDOMNodeId: 203,
|
backendDOMNodeId: 203,
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
const scene = buildVomScene(axNodes, {
|
const scene = buildTestScene(axNodes, {
|
||||||
viewport: { width: 1000, height: 800 },
|
viewport: { width: 1000, height: 800 },
|
||||||
iframeNodes: new Map(),
|
|
||||||
excludedBackendNodeIds: new Set([200]),
|
excludedBackendNodeIds: new Set([200]),
|
||||||
nodes: [],
|
nodes: [],
|
||||||
});
|
});
|
||||||
expect(scene.nodes.find((n) => n.id === 203)).toBeUndefined();
|
expect(scene.nodes.find((n) => n.backendNodeId === 203)).toBeUndefined();
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
@@ -2589,6 +2625,7 @@ describe("handleSnapshot", () => {
|
|||||||
strings,
|
strings,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0],
|
parentIndex: [-1, 0],
|
||||||
nodeName: [i("body"), i("button")],
|
nodeName: [i("body"), i("button")],
|
||||||
@@ -2680,6 +2717,7 @@ describe("handleSnapshot", () => {
|
|||||||
strings,
|
strings,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0],
|
parentIndex: [-1, 0],
|
||||||
nodeName: [i("body"), i("button")],
|
nodeName: [i("body"), i("button")],
|
||||||
@@ -2774,6 +2812,7 @@ describe("handleSnapshot", () => {
|
|||||||
strings,
|
strings,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0],
|
parentIndex: [-1, 0],
|
||||||
nodeName: [i("body"), i("button")],
|
nodeName: [i("body"), i("button")],
|
||||||
@@ -3038,6 +3077,7 @@ describe("handleSnapshot", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1, 2],
|
parentIndex: [-1, 0, 1, 2],
|
||||||
nodeName: [i("html"), i("body"), i("div"), i("input")],
|
nodeName: [i("html"), i("body"), i("div"), i("input")],
|
||||||
@@ -3095,6 +3135,7 @@ describe("handleSnapshot", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1, 1, 1, 1],
|
parentIndex: [-1, 0, 1, 1, 1, 1],
|
||||||
nodeName: [i("html"), i("body"), i("input"), i("input"), i("input"), i("input")],
|
nodeName: [i("html"), i("body"), i("input"), i("input"), i("input"), i("input")],
|
||||||
@@ -3240,12 +3281,17 @@ describe("handleSnapshot", () => {
|
|||||||
if (method === "Runtime.releaseObject") return {};
|
if (method === "Runtime.releaseObject") return {};
|
||||||
}
|
}
|
||||||
if (method === "DOMSnapshot.enable" || method === "Accessibility.enable") return {};
|
if (method === "DOMSnapshot.enable" || method === "Accessibility.enable") return {};
|
||||||
if (method === "DOMSnapshot.captureSnapshot") return snapshot;
|
if (method === "DOMSnapshot.captureSnapshot")
|
||||||
|
return { ...snapshot, documents: [snapshot.documents[0]] };
|
||||||
if (method === "Accessibility.getFullAXTree") return { nodes: mainAx };
|
if (method === "Accessibility.getFullAXTree") return { nodes: mainAx };
|
||||||
throw new Error(`unexpected root CDP method ${method}`);
|
throw new Error(`unexpected root CDP method ${method}`);
|
||||||
});
|
});
|
||||||
const sendToTarget = vi.fn(async (_target, method: string) => {
|
const sendToTarget = vi.fn(async (_target, method: string) => {
|
||||||
if (method === "Accessibility.enable") return {};
|
if (method === "Page.getLayoutMetrics" && ownerGeometry !== "unavailable")
|
||||||
|
return { cssLayoutViewport: { clientWidth: 400, clientHeight: 300 } };
|
||||||
|
if (method === "Accessibility.enable" || method === "DOMSnapshot.enable") return {};
|
||||||
|
if (method === "DOMSnapshot.captureSnapshot")
|
||||||
|
return { ...snapshot, documents: [snapshot.documents[1]] };
|
||||||
if (method === "Accessibility.getFullAXTree") return { nodes: childAx };
|
if (method === "Accessibility.getFullAXTree") return { nodes: childAx };
|
||||||
throw new Error(`unexpected child CDP method ${method}`);
|
throw new Error(`unexpected child CDP method ${method}`);
|
||||||
});
|
});
|
||||||
@@ -3346,6 +3392,7 @@ describe("handleSnapshot", () => {
|
|||||||
strings,
|
strings,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1, 1, 3],
|
parentIndex: [-1, 0, 1, 1, 3],
|
||||||
nodeName: [index("html"), index("body"), index("button"), index("div"), index("#text")],
|
nodeName: [index("html"), index("body"), index("button"), index("div"), index("#text")],
|
||||||
@@ -3670,6 +3717,7 @@ describe("handleSnapshot", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
scrollOffsetX: 0,
|
scrollOffsetX: 0,
|
||||||
scrollOffsetY: 0,
|
scrollOffsetY: 0,
|
||||||
nodes: {
|
nodes: {
|
||||||
@@ -3696,6 +3744,7 @@ describe("handleSnapshot", () => {
|
|||||||
},
|
},
|
||||||
// documents[1]: cross-origin login form
|
// documents[1]: cross-origin login form
|
||||||
{
|
{
|
||||||
|
frameId: "child",
|
||||||
scrollOffsetX: 0,
|
scrollOffsetX: 0,
|
||||||
scrollOffsetY: 0,
|
scrollOffsetY: 0,
|
||||||
nodes: {
|
nodes: {
|
||||||
@@ -3874,6 +3923,7 @@ describe("handleSnapshot", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1, 0, 3, 4, 5],
|
parentIndex: [-1, 0, 1, 0, 3, 4, 5],
|
||||||
nodeName: [
|
nodeName: [
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
|
import type { CapturedSceneInput } from "./vom/facts";
|
||||||
// Observation handlers — `tool.snapshot`, `tool.get_html`, `tool.screenshot`,
|
// Observation handlers — `tool.snapshot`, `tool.get_html`, `tool.screenshot`,
|
||||||
// and semantic `tool.observe` (design §7). Each handler resolves the target
|
// and semantic `tool.observe` (design §7). Each handler resolves the target
|
||||||
// tab (defaulting to the Agent Window's active tab when omitted) and
|
// tab (defaulting to the Agent Window's active tab when omitted) and
|
||||||
@@ -35,7 +36,6 @@ import { attachDialogs, markDialogCursor } from "./dialogs";
|
|||||||
import { rpcError } from "./errors";
|
import { rpcError } from "./errors";
|
||||||
import { resolveNodeGeometry } from "./frame-geometry";
|
import { resolveNodeGeometry } from "./frame-geometry";
|
||||||
import { screenshotPageRect } from "./geometry/coordinate-types";
|
import { screenshotPageRect } from "./geometry/coordinate-types";
|
||||||
import { cssViewport, GeometryContext } from "./geometry/frame-context";
|
|
||||||
import {
|
import {
|
||||||
type ChromeTabsApi,
|
type ChromeTabsApi,
|
||||||
enforceToolTargetScope,
|
enforceToolTargetScope,
|
||||||
@@ -49,15 +49,9 @@ import {
|
|||||||
type ToolEffect,
|
type ToolEffect,
|
||||||
} from "./shared";
|
} from "./shared";
|
||||||
import { resolveSnapshotRef } from "./snapshot-ref";
|
import { resolveSnapshotRef } from "./snapshot-ref";
|
||||||
import {
|
import { type CapturedNode, type CapturedSurfaceProbe, probeHoverSurfaces } from "./vom/capture";
|
||||||
type CapturedNode,
|
import { captureObservationFacts, semanticCapture } from "./vom/capture-coordinator";
|
||||||
type CapturedSurfaceProbe,
|
import type { FrameDocument as CapturedFrameDocument } from "./vom/frame-document";
|
||||||
type CapturedViewModel,
|
|
||||||
captureViewModel,
|
|
||||||
collectOverlayExcludedBackendIds,
|
|
||||||
probeHoverSurfaces,
|
|
||||||
} from "./vom/capture";
|
|
||||||
import { type CapturedFrameDocument, captureFrameData } from "./vom/frame-capture";
|
|
||||||
import { withOverlayBypass } from "./vom/hover-perception";
|
import { withOverlayBypass } from "./vom/hover-perception";
|
||||||
import { probeTooltipNames } from "./vom/name-enrichment";
|
import { probeTooltipNames } from "./vom/name-enrichment";
|
||||||
import {
|
import {
|
||||||
@@ -531,7 +525,7 @@ function buildActiveScopeBlocks(nodes: VomNode[], signals: VomNodeDomSignals): A
|
|||||||
|
|
||||||
function buildConditionalSurfaces(
|
function buildConditionalSurfaces(
|
||||||
nodes: VomNode[],
|
nodes: VomNode[],
|
||||||
captured: CapturedViewModel,
|
captured: CapturedSceneInput,
|
||||||
probes: CapturedSurfaceProbe[],
|
probes: CapturedSurfaceProbe[],
|
||||||
): CondSurface[] {
|
): CondSurface[] {
|
||||||
if (probes.length === 0) return [];
|
if (probes.length === 0) return [];
|
||||||
@@ -661,161 +655,9 @@ export interface BuildVomSceneOptions {
|
|||||||
|
|
||||||
export type VomFrameDocument = CapturedFrameDocument<CdpAxNode>;
|
export type VomFrameDocument = CapturedFrameDocument<CdpAxNode>;
|
||||||
|
|
||||||
function legacyFrameDocuments(
|
|
||||||
axNodes: CdpAxNode[],
|
|
||||||
captured: CapturedViewModel,
|
|
||||||
pageUrl?: string,
|
|
||||||
rootTarget: VomFrameDocument["target"] = { tabId: 0 },
|
|
||||||
): VomFrameDocument[] {
|
|
||||||
const rootFrameId = captured.rootFrameId ?? "root";
|
|
||||||
const documents: VomFrameDocument[] = [
|
|
||||||
{
|
|
||||||
frameId: rootFrameId,
|
|
||||||
contextScopeId: rootFrameId,
|
|
||||||
target: rootTarget,
|
|
||||||
...(pageUrl ? { url: pageUrl } : {}),
|
|
||||||
axNodes: [],
|
|
||||||
domNodes: captured.nodes.map((node) => ({ ...node, frameId: node.frameId ?? rootFrameId })),
|
|
||||||
},
|
|
||||||
];
|
|
||||||
const pending = [...captured.iframeNodes.entries()];
|
|
||||||
let progress = true;
|
|
||||||
let nextSyntheticFrame = 1;
|
|
||||||
while (pending.length > 0 && progress) {
|
|
||||||
progress = false;
|
|
||||||
for (let index = pending.length - 1; index >= 0; index -= 1) {
|
|
||||||
const [ownerBackendNodeId, domNodes] = pending[index];
|
|
||||||
const parent = documents.find((document) =>
|
|
||||||
document.domNodes.some((node) => node.backendNodeId === ownerBackendNodeId),
|
|
||||||
);
|
|
||||||
if (!parent) continue;
|
|
||||||
const frameId =
|
|
||||||
domNodes.find((node) => node.frameId)?.frameId ?? `legacy-frame-${nextSyntheticFrame++}`;
|
|
||||||
documents.push({
|
|
||||||
frameId,
|
|
||||||
parentFrameId: parent.frameId,
|
|
||||||
ownerBackendNodeId,
|
|
||||||
contextScopeId: frameId,
|
|
||||||
target: parent.target,
|
|
||||||
axNodes: [],
|
|
||||||
domNodes: domNodes.map((node) => ({ ...node, frameId: node.frameId ?? frameId })),
|
|
||||||
});
|
|
||||||
pending.splice(index, 1);
|
|
||||||
progress = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const documentByFrameId = new Map(documents.map((document) => [document.frameId, document]));
|
|
||||||
const backendFrame = new Map<number, string>();
|
|
||||||
const childFrameByOwner = new Map<number, string>();
|
|
||||||
for (const document of documents) {
|
|
||||||
if (document.ownerBackendNodeId !== undefined) {
|
|
||||||
childFrameByOwner.set(document.ownerBackendNodeId, document.frameId);
|
|
||||||
}
|
|
||||||
for (const node of document.domNodes) backendFrame.set(node.backendNodeId, document.frameId);
|
|
||||||
}
|
|
||||||
const axById = new Map(axNodes.map((node) => [node.nodeId, node]));
|
|
||||||
const ownership = new Map<string, string>();
|
|
||||||
const resolving = new Set<string>();
|
|
||||||
const frameForAx = (node: CdpAxNode): string => {
|
|
||||||
const cached = ownership.get(node.nodeId);
|
|
||||||
if (cached) return cached;
|
|
||||||
let frameId: string | undefined;
|
|
||||||
if (node.frameId && documentByFrameId.has(node.frameId)) frameId = node.frameId;
|
|
||||||
if (!frameId && typeof node.backendDOMNodeId === "number") {
|
|
||||||
frameId = backendFrame.get(node.backendDOMNodeId);
|
|
||||||
}
|
|
||||||
if (!frameId && node.parentId && !resolving.has(node.nodeId)) {
|
|
||||||
const parent = axById.get(node.parentId);
|
|
||||||
if (parent) {
|
|
||||||
frameId =
|
|
||||||
typeof parent.backendDOMNodeId === "number"
|
|
||||||
? childFrameByOwner.get(parent.backendDOMNodeId)
|
|
||||||
: undefined;
|
|
||||||
if (!frameId) {
|
|
||||||
resolving.add(node.nodeId);
|
|
||||||
frameId = frameForAx(parent);
|
|
||||||
resolving.delete(node.nodeId);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
frameId ??= rootFrameId;
|
|
||||||
ownership.set(node.nodeId, frameId);
|
|
||||||
return frameId;
|
|
||||||
};
|
|
||||||
for (const node of axNodes) frameForAx(node);
|
|
||||||
for (const node of axNodes) {
|
|
||||||
const frameId = ownership.get(node.nodeId) ?? rootFrameId;
|
|
||||||
const document = documentByFrameId.get(frameId) ?? documents[0];
|
|
||||||
document.axNodes.push({
|
|
||||||
...node,
|
|
||||||
frameId,
|
|
||||||
...(node.parentId && ownership.get(node.parentId) === frameId
|
|
||||||
? { parentId: node.parentId }
|
|
||||||
: { parentId: undefined }),
|
|
||||||
...(node.childIds
|
|
||||||
? { childIds: node.childIds.filter((childId) => ownership.get(childId) === frameId) }
|
|
||||||
: {}),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
return documents;
|
|
||||||
}
|
|
||||||
|
|
||||||
function withLegacyBackendIds(scene: VomScene): VomScene {
|
|
||||||
const used = new Set<number>();
|
|
||||||
const idMap = new Map<number, number>();
|
|
||||||
let nextVirtualId = -1;
|
|
||||||
for (const node of scene.nodes) {
|
|
||||||
const preferred = node.backendNodeId;
|
|
||||||
const id = preferred !== undefined && !used.has(preferred) ? preferred : nextVirtualId--;
|
|
||||||
used.add(id);
|
|
||||||
idMap.set(node.id, id);
|
|
||||||
}
|
|
||||||
return {
|
|
||||||
...scene,
|
|
||||||
nodes: scene.nodes.map((node) => ({
|
|
||||||
...node,
|
|
||||||
id: idMap.get(node.id) as number,
|
|
||||||
parentId: node.parentId === null ? null : (idMap.get(node.parentId) ?? null),
|
|
||||||
...(node.domParentId !== undefined
|
|
||||||
? { domParentId: node.domParentId === null ? null : (idMap.get(node.domParentId) ?? null) }
|
|
||||||
: {}),
|
|
||||||
...(node.domAncestorIds
|
|
||||||
? { domAncestorIds: node.domAncestorIds.flatMap((id) => idMap.get(id) ?? []) }
|
|
||||||
: {}),
|
|
||||||
})),
|
|
||||||
...(scene.surfaces
|
|
||||||
? {
|
|
||||||
surfaces: scene.surfaces.map((surface) => ({
|
|
||||||
...surface,
|
|
||||||
triggerId: idMap.get(surface.triggerId) ?? surface.triggerId,
|
|
||||||
})),
|
|
||||||
}
|
|
||||||
: {}),
|
|
||||||
...(scene.activeScopeBlocks
|
|
||||||
? {
|
|
||||||
activeScopeBlocks: scene.activeScopeBlocks.map((block) => ({
|
|
||||||
...block,
|
|
||||||
triggerId: idMap.get(block.triggerId) ?? block.triggerId,
|
|
||||||
})),
|
|
||||||
}
|
|
||||||
: {}),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
export function buildVomScene(
|
|
||||||
axNodes: CdpAxNode[],
|
|
||||||
captured: CapturedViewModel,
|
|
||||||
options: BuildVomSceneOptions = {},
|
|
||||||
): VomScene {
|
|
||||||
return withLegacyBackendIds(
|
|
||||||
buildFrameVomScene(legacyFrameDocuments(axNodes, captured, options.pageUrl), captured, options),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
export function buildFrameVomScene(
|
export function buildFrameVomScene(
|
||||||
documents: VomFrameDocument[],
|
documents: VomFrameDocument[],
|
||||||
captured: CapturedViewModel,
|
captured: CapturedSceneInput,
|
||||||
options: BuildVomSceneOptions = {},
|
options: BuildVomSceneOptions = {},
|
||||||
): VomScene {
|
): VomScene {
|
||||||
const scene = buildSemanticVomScene({
|
const scene = buildSemanticVomScene({
|
||||||
@@ -831,7 +673,7 @@ export function buildFrameVomScene(
|
|||||||
function attachCapturedSceneAnnotations(
|
function attachCapturedSceneAnnotations(
|
||||||
scene: VomScene,
|
scene: VomScene,
|
||||||
documents: VomFrameDocument[],
|
documents: VomFrameDocument[],
|
||||||
captured: CapturedViewModel,
|
captured: CapturedSceneInput,
|
||||||
surfaceProbes: CapturedSurfaceProbe[],
|
surfaceProbes: CapturedSurfaceProbe[],
|
||||||
): VomScene {
|
): VomScene {
|
||||||
const rootDocument = documents.find((document) => document.frameId === scene.rootFrameId);
|
const rootDocument = documents.find((document) => document.frameId === scene.rootFrameId);
|
||||||
@@ -981,45 +823,6 @@ export async function handleGetHtml(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
function emptyCapturedViewModel(viewport = { width: 0, height: 0 }): CapturedViewModel {
|
|
||||||
return { viewport, nodes: [], iframeNodes: new Map(), excludedBackendNodeIds: new Set() };
|
|
||||||
}
|
|
||||||
|
|
||||||
async function fallbackCapturedViewModel(
|
|
||||||
cdp: CdpRunner,
|
|
||||||
tabId: number,
|
|
||||||
geometry: GeometryContext,
|
|
||||||
signal?: AbortSignal,
|
|
||||||
): Promise<CapturedViewModel> {
|
|
||||||
throwIfAborted(signal, "observation");
|
|
||||||
let viewport = { width: 0, height: 0 };
|
|
||||||
try {
|
|
||||||
const metrics = await geometry.layoutMetrics({ tabId });
|
|
||||||
throwIfAborted(signal, "observation");
|
|
||||||
const source = cssViewport(metrics);
|
|
||||||
viewport = { width: source.width, height: source.height };
|
|
||||||
} catch (error) {
|
|
||||||
if (isAbortError(error)) throw error;
|
|
||||||
}
|
|
||||||
const excludedBackendNodeIds = await collectOverlayExcludedBackendIds(cdp, tabId, signal);
|
|
||||||
throwIfAborted(signal, "observation");
|
|
||||||
return { ...emptyCapturedViewModel(viewport), excludedBackendNodeIds };
|
|
||||||
}
|
|
||||||
|
|
||||||
async function captureForVom(
|
|
||||||
cdp: CdpRunner,
|
|
||||||
tabId: number,
|
|
||||||
options: CaptureVomObservationOptions,
|
|
||||||
geometry: GeometryContext,
|
|
||||||
): Promise<CapturedViewModel> {
|
|
||||||
try {
|
|
||||||
return await captureViewModel(cdp, tabId, { signal: options.signal, geometry });
|
|
||||||
} catch (error) {
|
|
||||||
if (isAbortError(error)) throw error;
|
|
||||||
return fallbackCapturedViewModel(cdp, tabId, geometry, options.signal);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface HoverProbeOutcome {
|
export interface HoverProbeOutcome {
|
||||||
/** Whether any active hover was dispatched during this observation. */
|
/** Whether any active hover was dispatched during this observation. */
|
||||||
performed: boolean;
|
performed: boolean;
|
||||||
@@ -1052,7 +855,7 @@ const NO_HOVER_PROBES: HoverProbeOutcome = {
|
|||||||
async function runHoverProbes(
|
async function runHoverProbes(
|
||||||
cdp: CdpRunner,
|
cdp: CdpRunner,
|
||||||
tabId: number,
|
tabId: number,
|
||||||
captured: CapturedViewModel,
|
captured: CapturedSceneInput,
|
||||||
documents: VomFrameDocument[],
|
documents: VomFrameDocument[],
|
||||||
staticSemantics: ReturnType<typeof resolveSemanticGraph>,
|
staticSemantics: ReturnType<typeof resolveSemanticGraph>,
|
||||||
options: CaptureVomObservationOptions,
|
options: CaptureVomObservationOptions,
|
||||||
@@ -1091,21 +894,9 @@ export async function captureVomObservation(
|
|||||||
throwIfAborted(options.signal, "observation");
|
throwIfAborted(options.signal, "observation");
|
||||||
await cdp.ensureAttachedToUrl?.(tabId, url);
|
await cdp.ensureAttachedToUrl?.(tabId, url);
|
||||||
throwIfAborted(options.signal, "observation");
|
throwIfAborted(options.signal, "observation");
|
||||||
const geometry = new GeometryContext(cdp, tabId, undefined, options.signal);
|
const facts = await captureObservationFacts<CdpAxNode>(cdp, tabId, options.signal);
|
||||||
const captured = await captureForVom(cdp, tabId, options, geometry);
|
const { captured, documents: normalizedDocuments } = semanticCapture(facts);
|
||||||
throwIfAborted(options.signal, "observation");
|
throwIfAborted(options.signal, "observation");
|
||||||
const documents = await captureFrameData<CdpAxNode>(
|
|
||||||
cdp,
|
|
||||||
tabId,
|
|
||||||
captured,
|
|
||||||
options.signal,
|
|
||||||
geometry,
|
|
||||||
);
|
|
||||||
throwIfAborted(options.signal, "observation");
|
|
||||||
const normalizedDocuments =
|
|
||||||
documents.length === 1 && captured.iframeNodes.size > 0
|
|
||||||
? legacyFrameDocuments(documents[0].axNodes, captured, url, documents[0].target)
|
|
||||||
: documents;
|
|
||||||
const semanticGraph = buildSemanticGraph({
|
const semanticGraph = buildSemanticGraph({
|
||||||
documents: normalizedDocuments,
|
documents: normalizedDocuments,
|
||||||
viewport: captured.viewport,
|
viewport: captured.viewport,
|
||||||
@@ -1135,7 +926,7 @@ export async function captureVomObservation(
|
|||||||
);
|
);
|
||||||
// Reserve space using the renderer's character-based token estimate. Like VOM
|
// Reserve space using the renderer's character-based token estimate. Like VOM
|
||||||
// headers, this integrity notice remains visible even under a tiny token budget.
|
// headers, this integrity notice remains visible even under a tiny token budget.
|
||||||
const geometryNotice = captured.frameGeometryIssues?.length
|
const geometryNotice = facts.issues.some((issue) => issue.stage === "geometry")
|
||||||
? "@warning iframe geometry incomplete: some frame content has no top-level coordinates.\n"
|
? "@warning iframe geometry incomplete: some frame content has no top-level coordinates.\n"
|
||||||
: "";
|
: "";
|
||||||
const rendered = renderVom(decoratedScene, {
|
const rendered = renderVom(decoratedScene, {
|
||||||
|
|||||||
@@ -0,0 +1,619 @@
|
|||||||
|
import { describe, expect, it, vi } from "vitest";
|
||||||
|
import type { CdpFrame, CdpTarget } from "@/browser-driver/frame-graph";
|
||||||
|
import { cdpTargetKey } from "@/browser-driver/frame-graph";
|
||||||
|
import { OVERLAY_HOST_MARKER_ATTR } from "@/lib/overlay-bridge";
|
||||||
|
import type { CdpRunner } from "../../shared";
|
||||||
|
import { captureObservationFacts, semanticCapture } from "../capture-coordinator";
|
||||||
|
import { buildSemanticGraph } from "../semantic-graph/build";
|
||||||
|
import { REQUESTED_STYLES } from "../snapshot";
|
||||||
|
|
||||||
|
function fixture(
|
||||||
|
options: { frames?: CdpFrame[]; fail?: string; omitDocument?: string; overlay?: string } = {},
|
||||||
|
) {
|
||||||
|
const frames = options.frames ?? [
|
||||||
|
{ frameId: "main", target: { tabId: 4 } },
|
||||||
|
{
|
||||||
|
frameId: "same",
|
||||||
|
parentFrameId: "main",
|
||||||
|
ownerBackendNodeId: 3,
|
||||||
|
target: { tabId: 4 },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
frameId: "nested",
|
||||||
|
parentFrameId: "same",
|
||||||
|
ownerBackendNodeId: 13,
|
||||||
|
target: { tabId: 4 },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
frameId: "remote",
|
||||||
|
parentFrameId: "main",
|
||||||
|
ownerBackendNodeId: 4,
|
||||||
|
target: { tabId: 4, sessionId: "remote" },
|
||||||
|
},
|
||||||
|
];
|
||||||
|
const elements = new Map(
|
||||||
|
frames.map((frame, i) => [frame.frameId, frame.target.sessionId ? 1 : i * 10 + 1]),
|
||||||
|
);
|
||||||
|
const logs: Array<{ target: CdpTarget; method: string; params: Record<string, unknown> }> = [];
|
||||||
|
const send = vi.fn(
|
||||||
|
async <T>(target: CdpTarget, method: string, params: object = {}): Promise<T> => {
|
||||||
|
logs.push({ target, method, params: params as Record<string, unknown> });
|
||||||
|
const args = params as Record<string, string | number>;
|
||||||
|
if (options.fail === `${target.sessionId ?? "main"}:${method}`)
|
||||||
|
throw new Error("fixture failure");
|
||||||
|
let result: unknown = {};
|
||||||
|
if (method === "Page.getLayoutMetrics")
|
||||||
|
result = { cssLayoutViewport: { clientWidth: 1000, clientHeight: 800 } };
|
||||||
|
if (method === "DOMSnapshot.captureSnapshot") {
|
||||||
|
expect((params as { computedStyles: unknown }).computedStyles).toEqual(REQUESTED_STYLES);
|
||||||
|
result = {
|
||||||
|
strings: [
|
||||||
|
"#document",
|
||||||
|
"html",
|
||||||
|
"button",
|
||||||
|
OVERLAY_HOST_MARKER_ATTR,
|
||||||
|
"",
|
||||||
|
"visible",
|
||||||
|
"1",
|
||||||
|
"static",
|
||||||
|
"auto",
|
||||||
|
],
|
||||||
|
documents: frames
|
||||||
|
.filter(
|
||||||
|
(frame) =>
|
||||||
|
cdpTargetKey(frame.target) === cdpTargetKey(target) &&
|
||||||
|
frame.frameId !== options.omitDocument,
|
||||||
|
)
|
||||||
|
.map((frame) => {
|
||||||
|
const element = elements.get(frame.frameId)!;
|
||||||
|
return {
|
||||||
|
frameId: frame.frameId,
|
||||||
|
nodes: {
|
||||||
|
backendNodeId: [element - 1, element + 10000, element, element + 1],
|
||||||
|
nodeName: [0, 1, 1, 2],
|
||||||
|
nodeType: [9, 10, 1, 1],
|
||||||
|
parentIndex: [-1, 0, 0, 2],
|
||||||
|
attributes: [[], [], [], frame.frameId === options.overlay ? [3, 4] : []],
|
||||||
|
},
|
||||||
|
layout: {
|
||||||
|
nodeIndex: [2, 3],
|
||||||
|
bounds: [
|
||||||
|
[0, 0, 1000, 800],
|
||||||
|
[10, 20, 100, 40],
|
||||||
|
],
|
||||||
|
styles: [
|
||||||
|
[7, 8, 8, 5, 6],
|
||||||
|
[7, 8, 8, 5, 6],
|
||||||
|
],
|
||||||
|
},
|
||||||
|
};
|
||||||
|
}),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if (method === "Accessibility.getFullAXTree") {
|
||||||
|
const frameId = String(args.frameId);
|
||||||
|
result = {
|
||||||
|
nodes: [
|
||||||
|
{
|
||||||
|
nodeId: `${frameId}-button`,
|
||||||
|
frameId,
|
||||||
|
backendDOMNodeId: elements.get(frameId)! + 1,
|
||||||
|
role: { value: "button" },
|
||||||
|
name: { value: frameId },
|
||||||
|
},
|
||||||
|
],
|
||||||
|
};
|
||||||
|
}
|
||||||
|
return result as T;
|
||||||
|
},
|
||||||
|
);
|
||||||
|
const cdp: CdpRunner = {
|
||||||
|
getFrameGraph: async () => ({ rootFrameId: frames[0].frameId, frames }),
|
||||||
|
send: (tabId, method, params) =>
|
||||||
|
(send as NonNullable<CdpRunner["sendToTarget"]>)({ tabId }, method, params),
|
||||||
|
sendToTarget: send as NonNullable<CdpRunner["sendToTarget"]>,
|
||||||
|
};
|
||||||
|
return { cdp, logs, elements };
|
||||||
|
}
|
||||||
|
|
||||||
|
describe("captureObservationFacts", () => {
|
||||||
|
it("collects each target once and scopes equal backend IDs", async () => {
|
||||||
|
const { cdp, logs } = fixture();
|
||||||
|
const facts = await captureObservationFacts(cdp, 4);
|
||||||
|
expect(logs.filter((call) => call.method === "DOMSnapshot.captureSnapshot")).toHaveLength(2);
|
||||||
|
expect(logs.filter((call) => call.method === "Accessibility.enable")).toHaveLength(2);
|
||||||
|
expect(logs.filter((call) => call.method === "Accessibility.getFullAXTree")).toHaveLength(4);
|
||||||
|
const main = facts.documents.find((doc) => doc.frame.frameId === "main")!;
|
||||||
|
const remote = facts.documents.find((doc) => doc.frame.frameId === "remote")!;
|
||||||
|
expect(main.index.nodes.get(2)).not.toBe(remote.index.nodes.get(2));
|
||||||
|
expect(main.index.children.get(1)).toEqual([2]);
|
||||||
|
expect(main.index.nodes.get(10001)?.nodeType).toBe(10);
|
||||||
|
expect(remote.axNodes[0].frameId).toBe("remote");
|
||||||
|
expect(main.index.nodes.get(2)).toBe(main.domNodes.find((node) => node.backendNodeId === 2));
|
||||||
|
expect(facts.finishedAt).toBeGreaterThanOrEqual(facts.startedAt);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("does not retry failed or missing documents and retains valid AX-only semantics", async () => {
|
||||||
|
const { cdp, logs } = fixture({
|
||||||
|
fail: "remote:DOMSnapshot.captureSnapshot",
|
||||||
|
omitDocument: "same",
|
||||||
|
});
|
||||||
|
const facts = await captureObservationFacts(cdp, 4);
|
||||||
|
const remote = facts.documents.find((doc) => doc.frame.frameId === "remote")!;
|
||||||
|
expect(remote.domNodes).toHaveLength(0);
|
||||||
|
expect(remote.axNodes).toHaveLength(1);
|
||||||
|
expect(logs.filter((call) => call.method === "DOMSnapshot.captureSnapshot")).toHaveLength(2);
|
||||||
|
expect(facts.issues).toContainEqual(
|
||||||
|
expect.objectContaining({ frameId: "same", stage: "dom", reason: "capture-unavailable" }),
|
||||||
|
);
|
||||||
|
expect(facts.issues).toContainEqual(
|
||||||
|
expect.objectContaining({ target: { tabId: 4, sessionId: "remote" }, stage: "dom" }),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("excludes overlay AX in its own document without excluding equal IDs elsewhere", async () => {
|
||||||
|
const facts = await captureObservationFacts(fixture({ overlay: "remote" }).cdp, 4);
|
||||||
|
const { documents } = semanticCapture(facts);
|
||||||
|
const graph = buildSemanticGraph({
|
||||||
|
documents,
|
||||||
|
viewport: facts.viewport,
|
||||||
|
rootFrameId: facts.rootFrameId,
|
||||||
|
});
|
||||||
|
expect(
|
||||||
|
[...graph.nodes.values()].find(
|
||||||
|
(node) => node.frameId === "remote" && node.backendNodeId === 2,
|
||||||
|
)?.excluded,
|
||||||
|
).toBe(true);
|
||||||
|
expect(
|
||||||
|
[...graph.nodes.values()].find((node) => node.frameId === "main" && node.backendNodeId === 2)
|
||||||
|
?.excluded,
|
||||||
|
).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("bounds collection across many targets and stops scheduling after cancellation", async () => {
|
||||||
|
const frames = Array.from({ length: 12 }, (_, i) => ({
|
||||||
|
frameId: `f${i}`,
|
||||||
|
target: { tabId: 4, ...(i ? { sessionId: `s${i}` } : {}) },
|
||||||
|
}));
|
||||||
|
const { cdp } = fixture({ frames });
|
||||||
|
let active = 0,
|
||||||
|
peak = 0;
|
||||||
|
const original = cdp.sendToTarget!;
|
||||||
|
const send: NonNullable<CdpRunner["sendToTarget"]> = async (target, method, params) => {
|
||||||
|
active++;
|
||||||
|
peak = Math.max(peak, active);
|
||||||
|
try {
|
||||||
|
await new Promise((resolve) => setTimeout(resolve, 1));
|
||||||
|
return await original(target, method, params);
|
||||||
|
} finally {
|
||||||
|
active--;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
cdp.sendToTarget = send;
|
||||||
|
cdp.send = (tabId, method, params) => send({ tabId }, method, params);
|
||||||
|
await captureObservationFacts(cdp, 4);
|
||||||
|
// This fixture has no frame edges, so no concurrent owner requests.
|
||||||
|
expect(peak).toBeLessThanOrEqual(4);
|
||||||
|
expect(peak).toBeGreaterThan(1);
|
||||||
|
const controller = new AbortController();
|
||||||
|
controller.abort();
|
||||||
|
await expect(captureObservationFacts(cdp, 4, controller.signal)).rejects.toMatchObject({
|
||||||
|
name: "AbortError",
|
||||||
|
});
|
||||||
|
expect(active).toBe(0);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
function childSnapshot(frameId: string, backendNodeId: number) {
|
||||||
|
const strings = [frameId, "body", "button", "static", "auto", "pointer"];
|
||||||
|
return {
|
||||||
|
strings,
|
||||||
|
documents: [
|
||||||
|
{
|
||||||
|
frameId,
|
||||||
|
nodes: {
|
||||||
|
parentIndex: [-1, 0],
|
||||||
|
nodeName: [1, 2],
|
||||||
|
backendNodeId: [backendNodeId - 1, backendNodeId],
|
||||||
|
attributes: [[], []],
|
||||||
|
},
|
||||||
|
layout: {
|
||||||
|
nodeIndex: [0, 1],
|
||||||
|
styles: [
|
||||||
|
[3, 4, 4],
|
||||||
|
[3, 4, 5],
|
||||||
|
],
|
||||||
|
bounds: [
|
||||||
|
[0, 0, 300, 200],
|
||||||
|
[10, 20, 100, 40],
|
||||||
|
],
|
||||||
|
paintOrders: [0, 1],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
],
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
describe("OOPIF capture", () => {
|
||||||
|
it("captures and positions multiple OOPIF documents missing from the root snapshot", async () => {
|
||||||
|
const sendToTarget = vi.fn(async (target, method) => {
|
||||||
|
if (method === "Page.getLayoutMetrics") {
|
||||||
|
return { cssLayoutViewport: { clientWidth: 300, clientHeight: 200, pageX: 0, pageY: 0 } };
|
||||||
|
}
|
||||||
|
if (method === "DOMSnapshot.enable" || method === "Accessibility.enable") return {};
|
||||||
|
if (method === "DOMSnapshot.captureSnapshot") {
|
||||||
|
return target.sessionId === "left-session"
|
||||||
|
? childSnapshot("left", 101)
|
||||||
|
: childSnapshot("right", 201);
|
||||||
|
}
|
||||||
|
if (method === "Accessibility.getFullAXTree") return { nodes: [] };
|
||||||
|
throw new Error(`unexpected ${method}`);
|
||||||
|
});
|
||||||
|
const cdp: CdpRunner = {
|
||||||
|
send: vi.fn(async (_tabId, method, params) => {
|
||||||
|
if (method === "Accessibility.enable" || method === "DOMSnapshot.enable") return {};
|
||||||
|
if (method === "DOMSnapshot.captureSnapshot") return childSnapshot("main", 1);
|
||||||
|
if (method === "Accessibility.getFullAXTree") return { nodes: [] };
|
||||||
|
if (method === "DOM.getBoxModel") {
|
||||||
|
const backendNodeId = (params as { backendNodeId?: number })?.backendNodeId;
|
||||||
|
const x = backendNodeId === 10 ? 50 : 500;
|
||||||
|
return { model: { content: [x, 100, x + 300, 100, x + 300, 300, x, 300] } };
|
||||||
|
}
|
||||||
|
if (method === "Page.getLayoutMetrics") {
|
||||||
|
return { cssLayoutViewport: { clientWidth: 1000, clientHeight: 800 } };
|
||||||
|
}
|
||||||
|
throw new Error(`unexpected root ${method}`);
|
||||||
|
}) as CdpRunner["send"],
|
||||||
|
sendToTarget: sendToTarget as unknown as NonNullable<CdpRunner["sendToTarget"]>,
|
||||||
|
getFrameGraph: vi.fn(async () => ({
|
||||||
|
rootFrameId: "main",
|
||||||
|
frames: [
|
||||||
|
{ frameId: "main", target: { tabId: 4 } },
|
||||||
|
{
|
||||||
|
frameId: "left",
|
||||||
|
parentFrameId: "main",
|
||||||
|
ownerBackendNodeId: 10,
|
||||||
|
target: { tabId: 4, sessionId: "left-session" },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
frameId: "right",
|
||||||
|
parentFrameId: "main",
|
||||||
|
ownerBackendNodeId: 20,
|
||||||
|
target: { tabId: 4, sessionId: "right-session" },
|
||||||
|
},
|
||||||
|
],
|
||||||
|
})),
|
||||||
|
};
|
||||||
|
|
||||||
|
const { documents: trees } = semanticCapture(await captureObservationFacts(cdp, 4));
|
||||||
|
|
||||||
|
expect(trees.map((tree) => tree.frameId)).toEqual(["main", "left", "right"]);
|
||||||
|
expect(
|
||||||
|
trees
|
||||||
|
.find((doc) => doc.frameId === "left")
|
||||||
|
?.domNodes?.find((node) => node.backendNodeId === 101)?.rect,
|
||||||
|
).toEqual({ x: 60, y: 120, w: 100, h: 40 });
|
||||||
|
expect(
|
||||||
|
trees
|
||||||
|
.find((doc) => doc.frameId === "right")
|
||||||
|
?.domNodes?.find((node) => node.backendNodeId === 201)?.rect,
|
||||||
|
).toEqual({ x: 510, y: 120, w: 100, h: 40 });
|
||||||
|
expect(
|
||||||
|
new Map(
|
||||||
|
trees
|
||||||
|
.filter((doc) => doc.ownerBackendNodeId !== undefined)
|
||||||
|
.map((doc) => [doc.frameId, doc.ownerBackendNodeId]),
|
||||||
|
),
|
||||||
|
).toEqual(
|
||||||
|
new Map([
|
||||||
|
["left", 10],
|
||||||
|
["right", 20],
|
||||||
|
]),
|
||||||
|
);
|
||||||
|
expect(
|
||||||
|
new Map(
|
||||||
|
trees.filter((doc) => doc.parentFrameId).map((doc) => [doc.frameId, doc.parentFrameId]),
|
||||||
|
),
|
||||||
|
).toEqual(
|
||||||
|
new Map([
|
||||||
|
["left", "main"],
|
||||||
|
["right", "main"],
|
||||||
|
]),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("retains nested owner failures when merging a captured OOPIF", async () => {
|
||||||
|
const child = childSnapshot("child", 101);
|
||||||
|
const nested = childSnapshot("nested", 201);
|
||||||
|
const document = child.documents[0];
|
||||||
|
const snapshot = {
|
||||||
|
strings: child.strings,
|
||||||
|
documents: [
|
||||||
|
{
|
||||||
|
...document,
|
||||||
|
nodes: { ...document.nodes, contentDocumentIndex: { index: [1], value: [1] } },
|
||||||
|
},
|
||||||
|
nested.documents[0],
|
||||||
|
],
|
||||||
|
};
|
||||||
|
const reply = async (_target: unknown, method: string) => {
|
||||||
|
if (method === "Page.getLayoutMetrics")
|
||||||
|
return { cssLayoutViewport: { clientWidth: 300, clientHeight: 200 } };
|
||||||
|
if (method === "DOMSnapshot.captureSnapshot") return snapshot;
|
||||||
|
if (method === "DOM.getBoxModel")
|
||||||
|
return { model: { content: [50, 100, 350, 100, 350, 300, 50, 300] } };
|
||||||
|
if (method === "DOM.resolveNode") throw new Error("nested owner replaced");
|
||||||
|
return {};
|
||||||
|
};
|
||||||
|
const cdp: CdpRunner = {
|
||||||
|
send: vi.fn(reply) as CdpRunner["send"],
|
||||||
|
sendToTarget: vi.fn(reply) as NonNullable<CdpRunner["sendToTarget"]>,
|
||||||
|
getFrameGraph: vi.fn(async () => ({
|
||||||
|
rootFrameId: "main",
|
||||||
|
frames: [
|
||||||
|
{ frameId: "main", target: { tabId: 4 } },
|
||||||
|
{
|
||||||
|
frameId: "nested",
|
||||||
|
parentFrameId: "child",
|
||||||
|
ownerBackendNodeId: 101,
|
||||||
|
target: { tabId: 4, sessionId: "child-session" },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
frameId: "child",
|
||||||
|
parentFrameId: "main",
|
||||||
|
ownerBackendNodeId: 10,
|
||||||
|
target: { tabId: 4, sessionId: "child-session" },
|
||||||
|
},
|
||||||
|
],
|
||||||
|
})),
|
||||||
|
};
|
||||||
|
const captured = await captureObservationFacts(cdp, 4);
|
||||||
|
expect(
|
||||||
|
captured.issues
|
||||||
|
.filter((issue) => issue.projectionIssue)
|
||||||
|
.map((issue) => issue.projectionIssue),
|
||||||
|
).toEqual([
|
||||||
|
{
|
||||||
|
status: "unavailable",
|
||||||
|
source: { target: { tabId: 4, sessionId: "child-session" }, frameId: "nested" },
|
||||||
|
ownerBackendNodeId: 101,
|
||||||
|
},
|
||||||
|
]);
|
||||||
|
expect(
|
||||||
|
captured.documents.find((doc) => doc.frame.frameId === "nested")?.domNodes[1],
|
||||||
|
).toMatchObject({
|
||||||
|
rect: null,
|
||||||
|
localRect: { x: 10, y: 20, w: 100, h: 40 },
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it("keeps OOPIF semantics when viewport projection is unavailable", async () => {
|
||||||
|
const cdp: CdpRunner = {
|
||||||
|
send: vi.fn(async (_tabId, method) => {
|
||||||
|
if (method === "Accessibility.enable" || method === "DOMSnapshot.enable") return {};
|
||||||
|
if (method === "DOMSnapshot.captureSnapshot") return childSnapshot("main", 1);
|
||||||
|
if (method === "Accessibility.getFullAXTree") return { nodes: [] };
|
||||||
|
if (method === "DOM.getBoxModel") throw new Error("owner geometry unavailable");
|
||||||
|
if (method === "Page.getLayoutMetrics") {
|
||||||
|
return { cssLayoutViewport: { clientWidth: 1000, clientHeight: 800 } };
|
||||||
|
}
|
||||||
|
throw new Error(`unexpected root ${method}`);
|
||||||
|
}) as CdpRunner["send"],
|
||||||
|
sendToTarget: vi.fn(async (_target, method) => {
|
||||||
|
if (method === "Page.getLayoutMetrics") throw new Error("viewport unavailable");
|
||||||
|
if (method === "DOMSnapshot.enable" || method === "Accessibility.enable") return {};
|
||||||
|
if (method === "DOMSnapshot.captureSnapshot") return childSnapshot("child", 101);
|
||||||
|
if (method === "Accessibility.getFullAXTree") {
|
||||||
|
return {
|
||||||
|
nodes: [
|
||||||
|
{
|
||||||
|
nodeId: "button",
|
||||||
|
backendDOMNodeId: 101,
|
||||||
|
role: { type: "role", value: "button" },
|
||||||
|
name: { type: "computedString", value: "Continue" },
|
||||||
|
},
|
||||||
|
],
|
||||||
|
};
|
||||||
|
}
|
||||||
|
throw new Error(`unexpected ${method}`);
|
||||||
|
}) as unknown as NonNullable<CdpRunner["sendToTarget"]>,
|
||||||
|
getFrameGraph: vi.fn(async () => ({
|
||||||
|
rootFrameId: "main",
|
||||||
|
frames: [
|
||||||
|
{ frameId: "main", target: { tabId: 4 } },
|
||||||
|
{
|
||||||
|
frameId: "child",
|
||||||
|
parentFrameId: "main",
|
||||||
|
ownerBackendNodeId: 10,
|
||||||
|
target: { tabId: 4, sessionId: "child-session" },
|
||||||
|
},
|
||||||
|
],
|
||||||
|
})),
|
||||||
|
};
|
||||||
|
|
||||||
|
const { documents } = semanticCapture(await captureObservationFacts(cdp, 4));
|
||||||
|
const childDocument = documents.find((document) => document.frameId === "child");
|
||||||
|
|
||||||
|
expect(childDocument?.axNodes).toEqual([
|
||||||
|
expect.objectContaining({ backendDOMNodeId: 101, frameId: "child" }),
|
||||||
|
]);
|
||||||
|
expect(childDocument?.domNodes.find((node) => node.backendNodeId === 101)).toEqual(
|
||||||
|
expect.objectContaining({ rect: null, localRect: { x: 10, y: 20, w: 100, h: 40 } }),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
// Six sibling frames and one nested frame exercise scheduling through the real capture entry.
|
||||||
|
function siblingCaptureFixture(
|
||||||
|
beforeReply: (method: string, params: Record<string, unknown>) => Promise<void> = async () => {},
|
||||||
|
) {
|
||||||
|
const document = (id: number, owners: number[], childIndexes: number[]) => ({
|
||||||
|
frameId: `frame-${id}`,
|
||||||
|
nodes: {
|
||||||
|
parentIndex: [-1, ...owners.map(() => 0)],
|
||||||
|
nodeName: [0, ...owners.map(() => 1)],
|
||||||
|
backendNodeId: [1000 + id, ...owners],
|
||||||
|
attributes: [[], ...owners.map(() => [])],
|
||||||
|
contentDocumentIndex: { index: owners.map((_, i) => i + 1), value: childIndexes },
|
||||||
|
},
|
||||||
|
layout: {
|
||||||
|
nodeIndex: [0, ...owners.map((_, i) => i + 1)],
|
||||||
|
bounds: [[0, 0, 200, 100], ...owners.map(() => [0, 0, 200, 100])],
|
||||||
|
},
|
||||||
|
});
|
||||||
|
const snapshot = {
|
||||||
|
strings: ["body", "iframe"],
|
||||||
|
documents: [
|
||||||
|
document(0, [100, 101, 102, 103, 104, 105], [1, 2, 3, 4, 5, 6]),
|
||||||
|
document(1, [200], [7]),
|
||||||
|
...Array.from({ length: 6 }, (_, i) => document(i + 2, [], [])),
|
||||||
|
],
|
||||||
|
};
|
||||||
|
let active = 0;
|
||||||
|
let peak = 0;
|
||||||
|
const send = vi.fn(async (_tabId: number, method: string, params: object = {}) => {
|
||||||
|
const args = params as Record<string, unknown>;
|
||||||
|
active++;
|
||||||
|
peak = Math.max(peak, active);
|
||||||
|
try {
|
||||||
|
await beforeReply(method, args);
|
||||||
|
if (method === "DOMSnapshot.captureSnapshot") return snapshot;
|
||||||
|
if (method === "Page.getLayoutMetrics")
|
||||||
|
return { cssLayoutViewport: { clientWidth: 1000, clientHeight: 800 } };
|
||||||
|
if (method === "DOM.getBoxModel")
|
||||||
|
return { model: { content: [0, 0, 200, 0, 200, 100, 0, 100] } };
|
||||||
|
if (method === "DOM.resolveNode") return { object: { objectId: String(args.backendNodeId) } };
|
||||||
|
if (method === "Runtime.callFunctionOn")
|
||||||
|
return { result: { value: { width: 200, height: 100 } } };
|
||||||
|
return {};
|
||||||
|
} finally {
|
||||||
|
active--;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return {
|
||||||
|
cdp: { send: send as CdpRunner["send"] },
|
||||||
|
send,
|
||||||
|
peak: () => peak,
|
||||||
|
active: () => active,
|
||||||
|
measured: () =>
|
||||||
|
send.mock.calls
|
||||||
|
.filter(([, method]) => method === "DOM.getBoxModel")
|
||||||
|
.map(([, , params]) => (params as { backendNodeId: number }).backendNodeId),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
describe("sibling frame measurement scheduling", () => {
|
||||||
|
it("bounds concurrent reads, fills free slots and preserves breadth-first output despite reordered replies", async () => {
|
||||||
|
const pending = new Map<number, () => void>();
|
||||||
|
const fixture = siblingCaptureFixture(async (method, params) => {
|
||||||
|
if (method === "DOM.getBoxModel" && Number(params.backendNodeId) < 200)
|
||||||
|
await new Promise<void>((resolve) => pending.set(Number(params.backendNodeId), resolve));
|
||||||
|
});
|
||||||
|
const capture = captureObservationFacts(fixture.cdp, 4);
|
||||||
|
await vi.waitFor(() => expect(pending.size).toBe(4));
|
||||||
|
expect(fixture.measured()).toEqual([100, 101, 102, 103]);
|
||||||
|
pending.get(103)!();
|
||||||
|
await vi.waitFor(() => expect(pending.has(104)).toBe(true));
|
||||||
|
pending.get(104)!();
|
||||||
|
await vi.waitFor(() => expect(pending.has(105)).toBe(true));
|
||||||
|
expect(fixture.measured()).not.toContain(200);
|
||||||
|
for (const id of [105, 102, 101, 100]) pending.get(id)!();
|
||||||
|
const captured = await capture;
|
||||||
|
expect(
|
||||||
|
captured.issues
|
||||||
|
.filter((issue) => issue.stage === "geometry")
|
||||||
|
.map((issue) => issue.projectionIssue),
|
||||||
|
).toEqual([]);
|
||||||
|
expect(fixture.peak()).toBe(4);
|
||||||
|
expect(fixture.active()).toBe(0);
|
||||||
|
expect(fixture.measured()).toEqual([100, 101, 102, 103, 104, 105, 200]);
|
||||||
|
expect(captured.documents.map((doc) => doc.frame.frameId)).toEqual(
|
||||||
|
Array.from({ length: 8 }, (_, i) => `frame-${i}`),
|
||||||
|
);
|
||||||
|
expect(
|
||||||
|
captured.documents.find((doc) => doc.frame.frameId === "frame-7")?.domNodes[0].rect,
|
||||||
|
).toEqual({ x: 0, y: 0, w: 200, h: 100 });
|
||||||
|
for (const method of [
|
||||||
|
"DOM.getBoxModel",
|
||||||
|
"DOM.resolveNode",
|
||||||
|
"Runtime.callFunctionOn",
|
||||||
|
"Runtime.releaseObject",
|
||||||
|
])
|
||||||
|
expect(fixture.send.mock.calls.filter(([, name]) => name === method)).toHaveLength(7);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("does not measure descendants of a failed owner and retains sibling geometry and local nodes", async () => {
|
||||||
|
const fixture = siblingCaptureFixture(async (method, params) => {
|
||||||
|
if (method === "DOM.resolveNode" && params.backendNodeId === 100)
|
||||||
|
throw new Error("owner replaced");
|
||||||
|
});
|
||||||
|
const captured = await captureObservationFacts(fixture.cdp, 4);
|
||||||
|
expect(
|
||||||
|
captured.issues
|
||||||
|
.filter((issue) => issue.stage === "geometry")
|
||||||
|
.map((issue) => issue.projectionIssue),
|
||||||
|
).toEqual([
|
||||||
|
{
|
||||||
|
status: "unavailable",
|
||||||
|
source: { target: { tabId: 4 }, frameId: "frame-1" },
|
||||||
|
ownerBackendNodeId: 100,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
status: "blocked",
|
||||||
|
source: { target: { tabId: 4 }, frameId: "frame-7" },
|
||||||
|
cause: captured.issues.find((issue) => issue.frameId === "frame-1")?.projectionIssue,
|
||||||
|
},
|
||||||
|
]);
|
||||||
|
expect(fixture.measured()).not.toContain(200);
|
||||||
|
expect(
|
||||||
|
captured.documents.find((doc) => doc.frame.frameId === "frame-7")?.domNodes[0],
|
||||||
|
).toMatchObject({
|
||||||
|
tag: "body",
|
||||||
|
rect: null,
|
||||||
|
localRect: { x: 0, y: 0, w: 200, h: 100 },
|
||||||
|
});
|
||||||
|
expect(
|
||||||
|
captured.documents.find((doc) => doc.frame.frameId === "frame-2")?.domNodes[0].rect,
|
||||||
|
).toEqual({ x: 0, y: 0, w: 200, h: 100 });
|
||||||
|
expect(captured.documents.map((doc) => doc.frame.frameId)).toEqual(
|
||||||
|
Array.from({ length: 8 }, (_, i) => `frame-${i}`),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("stops scheduling on cancellation and waits for resolved objects to be released", async () => {
|
||||||
|
const controller = new AbortController();
|
||||||
|
const resolutions = new Map<string, () => void>();
|
||||||
|
const releases = new Map<string, () => void>();
|
||||||
|
const fixture = siblingCaptureFixture(async (method, params) => {
|
||||||
|
if (method === "DOM.resolveNode")
|
||||||
|
await new Promise<void>((resolve) =>
|
||||||
|
resolutions.set(String(params.backendNodeId), resolve),
|
||||||
|
);
|
||||||
|
if (method === "Runtime.releaseObject")
|
||||||
|
await new Promise<void>((resolve) => releases.set(String(params.objectId), resolve));
|
||||||
|
});
|
||||||
|
let settled = false;
|
||||||
|
const capture = captureObservationFacts(fixture.cdp, 4, controller.signal);
|
||||||
|
const rejected = expect(capture).rejects.toMatchObject({ name: "AbortError" });
|
||||||
|
void capture.then(
|
||||||
|
() => {
|
||||||
|
settled = true;
|
||||||
|
},
|
||||||
|
() => {
|
||||||
|
settled = true;
|
||||||
|
},
|
||||||
|
);
|
||||||
|
await vi.waitFor(() => expect(resolutions.size).toBe(4));
|
||||||
|
controller.abort();
|
||||||
|
for (const resolve of resolutions.values()) resolve();
|
||||||
|
await vi.waitFor(() => expect(releases.size).toBe(4));
|
||||||
|
expect(settled).toBe(false);
|
||||||
|
expect(fixture.measured()).toEqual([100, 101, 102, 103]);
|
||||||
|
for (const release of releases.values()) release();
|
||||||
|
await rejected;
|
||||||
|
expect(fixture.active()).toBe(0);
|
||||||
|
expect(fixture.send.mock.calls.some(([, method]) => method === "Runtime.callFunctionOn")).toBe(
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
});
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
import { describe, expect, it, vi } from "vitest";
|
import { describe, expect, it, vi } from "vitest";
|
||||||
import { OVERLAY_HOST_MARKER_ATTR, OVERLAY_HOST_NAME } from "../../../lib/overlay-bridge";
|
import { OVERLAY_HOST_MARKER_ATTR, OVERLAY_HOST_NAME } from "../../../lib/overlay-bridge";
|
||||||
import type { CdpRunner } from "../../shared";
|
import { collectOverlayExcludedBackendIds, probeHoverSurfaces } from "../capture";
|
||||||
import { captureViewModel, collectOverlayExcludedBackendIds, probeHoverSurfaces } from "../capture";
|
import { captureObservationFacts, semanticCapture } from "../capture-coordinator";
|
||||||
|
|
||||||
// Minimal but format-accurate captureSnapshot reply: a body with one
|
// Minimal but format-accurate captureSnapshot reply: a body with one
|
||||||
// fixed full-screen overlay div carrying a password input.
|
// fixed full-screen overlay div carrying a password input.
|
||||||
@@ -27,6 +27,7 @@ function fakeSnapshotReply() {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1, 2],
|
parentIndex: [-1, 0, 1, 2],
|
||||||
nodeType: [1, 1, 1, 1],
|
nodeType: [1, 1, 1, 1],
|
||||||
@@ -74,6 +75,7 @@ function hoverTriggerSnapshotReply() {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1],
|
parentIndex: [-1, 0, 1],
|
||||||
nodeType: [1, 1, 1],
|
nodeType: [1, 1, 1],
|
||||||
@@ -118,6 +120,7 @@ function twoHoverTriggerSnapshotReply() {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1, 1],
|
parentIndex: [-1, 0, 1, 1],
|
||||||
nodeType: [1, 1, 1, 1],
|
nodeType: [1, 1, 1, 1],
|
||||||
@@ -171,6 +174,7 @@ function nestedHoverTriggerSnapshotReply() {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1, 2, 3],
|
parentIndex: [-1, 0, 1, 2, 3],
|
||||||
nodeType: [1, 1, 1, 1, 1],
|
nodeType: [1, 1, 1, 1, 1],
|
||||||
@@ -244,13 +248,11 @@ function makeCdp(snapshot: unknown) {
|
|||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
describe("captureViewModel", () => {
|
describe("snapshot capture and hover", () => {
|
||||||
it("parses nodes, attrs, rects, paint order and styles", async () => {
|
it("parses nodes, attrs, rects, paint order and styles", async () => {
|
||||||
const { nodes, viewport, iframeNodes } = await captureViewModel(
|
const { nodes, viewport } = semanticCapture(
|
||||||
makeCdp(fakeSnapshotReply()),
|
await captureObservationFacts(makeCdp(fakeSnapshotReply()), 4),
|
||||||
4,
|
).captured;
|
||||||
);
|
|
||||||
expect(iframeNodes.size).toBe(0);
|
|
||||||
expect(viewport).toEqual({ width: 1000, height: 800 });
|
expect(viewport).toEqual({ width: 1000, height: 800 });
|
||||||
const div = nodes.find((n) => n.backendNodeId === 12);
|
const div = nodes.find((n) => n.backendNodeId === 12);
|
||||||
expect(div).toMatchObject({ tag: "div", position: "fixed", parentBackendNodeId: 11 });
|
expect(div).toMatchObject({ tag: "div", position: "fixed", parentBackendNodeId: 11 });
|
||||||
@@ -262,35 +264,11 @@ describe("captureViewModel", () => {
|
|||||||
|
|
||||||
it("never hovers the page while capturing", async () => {
|
it("never hovers the page while capturing", async () => {
|
||||||
const cdp = makeCdp(fakeSnapshotReply());
|
const cdp = makeCdp(fakeSnapshotReply());
|
||||||
await captureViewModel(cdp, 4);
|
semanticCapture(await captureObservationFacts(cdp, 4)).captured;
|
||||||
|
|
||||||
expect(cdp.send).not.toHaveBeenCalledWith(4, "Input.dispatchMouseEvent", expect.anything());
|
expect(cdp.send).not.toHaveBeenCalledWith(4, "Input.dispatchMouseEvent", expect.anything());
|
||||||
});
|
});
|
||||||
|
|
||||||
it("batch-enriches form controls without per-node object resolution", async () => {
|
|
||||||
const cdp = makeCdp(fakeSnapshotReply());
|
|
||||||
const result = await captureViewModel(cdp, 4);
|
|
||||||
const input = result.nodes.find((node) => node.backendNodeId === 13);
|
|
||||||
|
|
||||||
expect(input).toMatchObject({
|
|
||||||
formState: "filled",
|
|
||||||
formPlaceholder: "secret",
|
|
||||||
attrs: { type: "password" },
|
|
||||||
});
|
|
||||||
expect(input?.formValue).toBeUndefined();
|
|
||||||
expect(input?.formDefaultValue).toBeUndefined();
|
|
||||||
expect(cdp.send).toHaveBeenCalledWith(
|
|
||||||
4,
|
|
||||||
"Runtime.evaluate",
|
|
||||||
expect.objectContaining({
|
|
||||||
serializationOptions: expect.objectContaining({ serialization: "deep" }),
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
expect(cdp.send).toHaveBeenCalledWith(4, "Runtime.releaseObjectGroup", expect.anything());
|
|
||||||
expect(cdp.send).not.toHaveBeenCalledWith(4, "DOM.resolveNode", expect.anything());
|
|
||||||
expect(cdp.send).not.toHaveBeenCalledWith(4, "Runtime.callFunctionOn", expect.anything());
|
|
||||||
});
|
|
||||||
|
|
||||||
it("runs hover surface probes when explicitly enabled", async () => {
|
it("runs hover surface probes when explicitly enabled", async () => {
|
||||||
let hoverStateCalls = 0;
|
let hoverStateCalls = 0;
|
||||||
const cdp = {
|
const cdp = {
|
||||||
@@ -330,7 +308,7 @@ describe("captureViewModel", () => {
|
|||||||
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
||||||
};
|
};
|
||||||
|
|
||||||
const captured = await captureViewModel(cdp, 4);
|
const captured = semanticCapture(await captureObservationFacts(cdp, 4)).captured;
|
||||||
const surfaceProbes = await probeHoverSurfaces(cdp, 4, captured.nodes);
|
const surfaceProbes = await probeHoverSurfaces(cdp, 4, captured.nodes);
|
||||||
|
|
||||||
expect(surfaceProbes).toEqual([
|
expect(surfaceProbes).toEqual([
|
||||||
@@ -390,7 +368,7 @@ describe("captureViewModel", () => {
|
|||||||
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
||||||
};
|
};
|
||||||
|
|
||||||
const captured = await captureViewModel(cdp, 4);
|
const captured = semanticCapture(await captureObservationFacts(cdp, 4)).captured;
|
||||||
const surfaceProbes = await probeHoverSurfaces(cdp, 4, captured.nodes);
|
const surfaceProbes = await probeHoverSurfaces(cdp, 4, captured.nodes);
|
||||||
|
|
||||||
expect(surfaceProbes).toEqual([
|
expect(surfaceProbes).toEqual([
|
||||||
@@ -435,7 +413,7 @@ describe("captureViewModel", () => {
|
|||||||
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
||||||
};
|
};
|
||||||
|
|
||||||
const captured = await captureViewModel(cdp, 4);
|
const captured = semanticCapture(await captureObservationFacts(cdp, 4)).captured;
|
||||||
await probeHoverSurfaces(cdp, 4, captured.nodes);
|
await probeHoverSurfaces(cdp, 4, captured.nodes);
|
||||||
|
|
||||||
expect(hoverMoves).toBe(1);
|
expect(hoverMoves).toBe(1);
|
||||||
@@ -467,6 +445,7 @@ describe("captureViewModel", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1, 0, 3, 4],
|
parentIndex: [-1, 0, 1, 0, 3, 4],
|
||||||
nodeName: [i("html"), i("body"), i("div"), i(OVERLAY_HOST_NAME), i("div"), i("button")],
|
nodeName: [i("html"), i("body"), i("div"), i(OVERLAY_HOST_NAME), i("div"), i("button")],
|
||||||
@@ -497,7 +476,7 @@ describe("captureViewModel", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
const { nodes } = await captureViewModel(makeCdp(snapshot), 4);
|
const { nodes } = semanticCapture(await captureObservationFacts(makeCdp(snapshot), 4)).captured;
|
||||||
const ids = nodes.map((n) => n.backendNodeId);
|
const ids = nodes.map((n) => n.backendNodeId);
|
||||||
// Real page nodes survive.
|
// Real page nodes survive.
|
||||||
expect(ids).toContain(12);
|
expect(ids).toContain(12);
|
||||||
@@ -528,6 +507,7 @@ describe("captureViewModel", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1, 0, 3, 4],
|
parentIndex: [-1, 0, 1, 0, 3, 4],
|
||||||
nodeName: [i("html"), i("body"), i("div"), i(OVERLAY_HOST_NAME), i("div"), i("button")],
|
nodeName: [i("html"), i("body"), i("div"), i(OVERLAY_HOST_NAME), i("div"), i("button")],
|
||||||
@@ -558,7 +538,9 @@ describe("captureViewModel", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
const { excludedBackendNodeIds } = await captureViewModel(makeCdp(snapshot), 4);
|
const { excludedBackendNodeIds } = semanticCapture(
|
||||||
|
await captureObservationFacts(makeCdp(snapshot), 4),
|
||||||
|
).captured;
|
||||||
expect(excludedBackendNodeIds).toEqual(new Set([13, 14, 15]));
|
expect(excludedBackendNodeIds).toEqual(new Set([13, 14, 15]));
|
||||||
});
|
});
|
||||||
|
|
||||||
@@ -579,6 +561,7 @@ describe("captureViewModel", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1],
|
parentIndex: [-1, 0, 1],
|
||||||
nodeName: [i("html"), i("body"), i(OVERLAY_HOST_NAME)],
|
nodeName: [i("html"), i("body"), i(OVERLAY_HOST_NAME)],
|
||||||
@@ -601,7 +584,9 @@ describe("captureViewModel", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
const { nodes, excludedBackendNodeIds } = await captureViewModel(makeCdp(snapshot), 4);
|
const { nodes, excludedBackendNodeIds } = semanticCapture(
|
||||||
|
await captureObservationFacts(makeCdp(snapshot), 4),
|
||||||
|
).captured;
|
||||||
expect(nodes.map((n) => n.backendNodeId)).not.toContain(13);
|
expect(nodes.map((n) => n.backendNodeId)).not.toContain(13);
|
||||||
expect(excludedBackendNodeIds).toEqual(new Set([13]));
|
expect(excludedBackendNodeIds).toEqual(new Set([13]));
|
||||||
});
|
});
|
||||||
@@ -624,6 +609,7 @@ describe("captureViewModel", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1],
|
parentIndex: [-1, 0, 1],
|
||||||
nodeName: [i("html"), i("body"), i("div")],
|
nodeName: [i("html"), i("body"), i("div")],
|
||||||
@@ -646,7 +632,9 @@ describe("captureViewModel", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
const { nodes, excludedBackendNodeIds } = await captureViewModel(makeCdp(snapshot), 4);
|
const { nodes, excludedBackendNodeIds } = semanticCapture(
|
||||||
|
await captureObservationFacts(makeCdp(snapshot), 4),
|
||||||
|
).captured;
|
||||||
expect(nodes.map((n) => n.backendNodeId)).not.toContain(13);
|
expect(nodes.map((n) => n.backendNodeId)).not.toContain(13);
|
||||||
expect(excludedBackendNodeIds).toEqual(new Set([13]));
|
expect(excludedBackendNodeIds).toEqual(new Set([13]));
|
||||||
});
|
});
|
||||||
@@ -669,6 +657,7 @@ describe("captureViewModel", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1],
|
parentIndex: [-1, 0, 1],
|
||||||
nodeName: [i("html"), i("body"), i("div")],
|
nodeName: [i("html"), i("body"), i("div")],
|
||||||
@@ -691,7 +680,7 @@ describe("captureViewModel", () => {
|
|||||||
},
|
},
|
||||||
],
|
],
|
||||||
};
|
};
|
||||||
const { nodes } = await captureViewModel(makeCdp(snapshot), 4);
|
const { nodes } = semanticCapture(await captureObservationFacts(makeCdp(snapshot), 4)).captured;
|
||||||
expect(nodes.find((n) => n.backendNodeId === 12)?.cursor).toBe("pointer");
|
expect(nodes.find((n) => n.backendNodeId === 12)?.cursor).toBe("pointer");
|
||||||
expect(nodes.find((n) => n.backendNodeId === 11)?.cursor).toBe("auto");
|
expect(nodes.find((n) => n.backendNodeId === 11)?.cursor).toBe("auto");
|
||||||
});
|
});
|
||||||
@@ -703,6 +692,7 @@ describe("captureViewModel", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1, 1, 1],
|
parentIndex: [-1, 0, 1, 1, 1],
|
||||||
nodeName: [i("html"), i("body"), i("button"), i("button"), i("button")],
|
nodeName: [i("html"), i("body"), i("button"), i("button"), i("button")],
|
||||||
@@ -729,7 +719,7 @@ describe("captureViewModel", () => {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
const { nodes } = await captureViewModel(makeCdp(snapshot), 4);
|
const { nodes } = semanticCapture(await captureObservationFacts(makeCdp(snapshot), 4)).captured;
|
||||||
|
|
||||||
expect(nodes.find((node) => node.backendNodeId === 12)?.rendered).toBe(true);
|
expect(nodes.find((node) => node.backendNodeId === 12)?.rendered).toBe(true);
|
||||||
expect(nodes.find((node) => node.backendNodeId === 13)?.rendered).toBe(false);
|
expect(nodes.find((node) => node.backendNodeId === 13)?.rendered).toBe(false);
|
||||||
@@ -749,7 +739,7 @@ describe("captureViewModel", () => {
|
|||||||
throw new Error(method);
|
throw new Error(method);
|
||||||
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
||||||
};
|
};
|
||||||
const { nodes } = await captureViewModel(cdp, 4);
|
const { nodes } = semanticCapture(await captureObservationFacts(cdp, 4)).captured;
|
||||||
const div = nodes.find((n) => n.backendNodeId === 12);
|
const div = nodes.find((n) => n.backendNodeId === 12);
|
||||||
expect(div?.localRect?.y).toBe(-200);
|
expect(div?.localRect?.y).toBe(-200);
|
||||||
expect(div?.rect).toMatchObject({ y: 0, h: 600 });
|
expect(div?.rect).toMatchObject({ y: 0, h: 600 });
|
||||||
@@ -762,6 +752,7 @@ describe("captureViewModel", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1],
|
parentIndex: [-1, 0, 1],
|
||||||
nodeName: [i("html"), i("body"), i("div")],
|
nodeName: [i("html"), i("body"), i("div")],
|
||||||
@@ -796,7 +787,7 @@ describe("captureViewModel", () => {
|
|||||||
throw new Error(method);
|
throw new Error(method);
|
||||||
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
||||||
};
|
};
|
||||||
const { nodes } = await captureViewModel(cdp, 4);
|
const { nodes } = semanticCapture(await captureObservationFacts(cdp, 4)).captured;
|
||||||
expect(nodes.find((n) => n.backendNodeId === 12)?.rect).toEqual({
|
expect(nodes.find((n) => n.backendNodeId === 12)?.rect).toEqual({
|
||||||
x: 0,
|
x: 0,
|
||||||
y: 0,
|
y: 0,
|
||||||
@@ -832,6 +823,7 @@ describe("captureViewModel", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
scrollOffsetX: 0,
|
scrollOffsetX: 0,
|
||||||
scrollOffsetY: 0,
|
scrollOffsetY: 0,
|
||||||
nodes: {
|
nodes: {
|
||||||
@@ -870,7 +862,7 @@ describe("captureViewModel", () => {
|
|||||||
throw new Error(method);
|
throw new Error(method);
|
||||||
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
||||||
};
|
};
|
||||||
const { nodes } = await captureViewModel(cdp, 4);
|
const { nodes } = semanticCapture(await captureObservationFacts(cdp, 4)).captured;
|
||||||
const btn = nodes.find((n) => n.backendNodeId === 12);
|
const btn = nodes.find((n) => n.backendNodeId === 12);
|
||||||
expect(btn?.textContent).toBe("登录");
|
expect(btn?.textContent).toBe("登录");
|
||||||
const link = nodes.find((n) => n.backendNodeId === 14);
|
const link = nodes.find((n) => n.backendNodeId === 14);
|
||||||
@@ -880,7 +872,7 @@ describe("captureViewModel", () => {
|
|||||||
expect(body?.textContent).toBeUndefined();
|
expect(body?.textContent).toBeUndefined();
|
||||||
});
|
});
|
||||||
|
|
||||||
it("parses iframe sub-documents and returns iframeNodes keyed by iframe backendNodeId", async () => {
|
it("retains explicit iframe document ownership and projected nodes", async () => {
|
||||||
const S = [
|
const S = [
|
||||||
"html",
|
"html",
|
||||||
"body",
|
"body",
|
||||||
@@ -966,30 +958,21 @@ describe("captureViewModel", () => {
|
|||||||
throw new Error(method);
|
throw new Error(method);
|
||||||
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
||||||
};
|
};
|
||||||
const {
|
const facts = await captureObservationFacts(cdp, 4);
|
||||||
nodes,
|
const root = facts.documents.find((doc) => doc.frame.frameId === "main-frame")!;
|
||||||
iframeNodes,
|
const child = facts.documents.find((doc) => doc.frame.frameId === "child-frame")!;
|
||||||
frameNodes,
|
expect(facts.documents).toHaveLength(2);
|
||||||
frameOwnerBackendNodeIds,
|
expect(root.domNodes.find((n) => n.backendNodeId === 13)?.tag).toBe("iframe");
|
||||||
frameParentIds,
|
const subNodes = child.domNodes;
|
||||||
rootFrameId,
|
|
||||||
} = await captureViewModel(cdp, 4);
|
|
||||||
// Main frame has 4 nodes; iframe is included
|
|
||||||
expect(nodes.find((n) => n.backendNodeId === 13)?.tag).toBe("iframe");
|
|
||||||
// iframeNodes keyed by the <iframe> element's backendNodeId=13
|
|
||||||
expect(iframeNodes.size).toBe(1);
|
|
||||||
expect(iframeNodes.has(13)).toBe(true);
|
|
||||||
const subNodes = iframeNodes.get(13)!;
|
|
||||||
const input = subNodes.find((n) => n.backendNodeId === 101);
|
const input = subNodes.find((n) => n.backendNodeId === 101);
|
||||||
expect(input?.tag).toBe("input");
|
expect(input?.tag).toBe("input");
|
||||||
expect(input?.attrs.type).toBe("text");
|
expect(input?.attrs.type).toBe("text");
|
||||||
expect(input?.ownerFrameBackendNodeId).toBe(13);
|
expect(input?.ownerFrameBackendNodeId).toBe(13);
|
||||||
expect(input?.localRect).toEqual({ x: 0, y: 0, w: 200, h: 40 });
|
expect(input?.localRect).toEqual({ x: 0, y: 0, w: 200, h: 40 });
|
||||||
expect(input?.rect).toEqual({ x: 100, y: 300, w: 200, h: 40 });
|
expect(input?.rect).toEqual({ x: 100, y: 300, w: 200, h: 40 });
|
||||||
expect(rootFrameId).toBe("main-frame");
|
expect(facts.rootFrameId).toBe("main-frame");
|
||||||
expect(frameNodes?.get("child-frame")).toBe(subNodes);
|
expect(child.frame.ownerBackendNodeId).toBe(13);
|
||||||
expect(frameOwnerBackendNodeIds?.get("child-frame")).toBe(13);
|
expect(child.frame.parentFrameId).toBe("main-frame");
|
||||||
expect(frameParentIds?.get("child-frame")).toBe("main-frame");
|
|
||||||
});
|
});
|
||||||
|
|
||||||
it("projects iframe documents independently of owner snapshot geometry", async () => {
|
it("projects iframe documents independently of owner snapshot geometry", async () => {
|
||||||
@@ -1010,6 +993,7 @@ describe("captureViewModel", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
scrollOffsetX: 0,
|
scrollOffsetX: 0,
|
||||||
scrollOffsetY: 0,
|
scrollOffsetY: 0,
|
||||||
nodes: {
|
nodes: {
|
||||||
@@ -1027,6 +1011,7 @@ describe("captureViewModel", () => {
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
|
frameId: "child",
|
||||||
scrollOffsetX: 0,
|
scrollOffsetX: 0,
|
||||||
scrollOffsetY: 0,
|
scrollOffsetY: 0,
|
||||||
nodes: {
|
nodes: {
|
||||||
@@ -1063,8 +1048,10 @@ describe("captureViewModel", () => {
|
|||||||
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
||||||
};
|
};
|
||||||
|
|
||||||
const { iframeNodes } = await captureViewModel(cdp, 4);
|
const { documents } = await captureObservationFacts(cdp, 4);
|
||||||
const input = iframeNodes.get(13)?.find((n) => n.backendNodeId === 101);
|
const input = documents
|
||||||
|
.find((doc) => doc.frame.frameId === "child")
|
||||||
|
?.domNodes?.find((n) => n.backendNodeId === 101);
|
||||||
|
|
||||||
expect(input?.localRect).toEqual({ x: 20, y: 20, w: 120, h: 60 });
|
expect(input?.localRect).toEqual({ x: 20, y: 20, w: 120, h: 60 });
|
||||||
expect(input?.rect).toEqual({ x: 120, y: 320, w: 120, h: 60 });
|
expect(input?.rect).toEqual({ x: 120, y: 320, w: 120, h: 60 });
|
||||||
@@ -1088,6 +1075,7 @@ describe("captureViewModel", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1],
|
parentIndex: [-1, 0, 1],
|
||||||
nodeName: [i("html"), i("body"), i("iframe")],
|
nodeName: [i("html"), i("body"), i("iframe")],
|
||||||
@@ -1109,6 +1097,7 @@ describe("captureViewModel", () => {
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
|
frameId: "child",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1],
|
parentIndex: [-1, 0, 1],
|
||||||
nodeName: [i("html"), i("body"), i("iframe")],
|
nodeName: [i("html"), i("body"), i("iframe")],
|
||||||
@@ -1130,6 +1119,7 @@ describe("captureViewModel", () => {
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
|
frameId: "nested",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0],
|
parentIndex: [-1, 0],
|
||||||
nodeName: [i("body"), i("input")],
|
nodeName: [i("body"), i("input")],
|
||||||
@@ -1185,9 +1175,13 @@ describe("captureViewModel", () => {
|
|||||||
) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
||||||
};
|
};
|
||||||
|
|
||||||
const { iframeNodes } = await captureViewModel(cdp, 4);
|
const { documents } = await captureObservationFacts(cdp, 4);
|
||||||
const nestedIframe = iframeNodes.get(13)?.find((n) => n.backendNodeId === 23);
|
const nestedIframe = documents
|
||||||
const input = iframeNodes.get(23)?.find((n) => n.backendNodeId === 31);
|
.find((doc) => doc.frame.frameId === "child")
|
||||||
|
?.domNodes?.find((n) => n.backendNodeId === 23);
|
||||||
|
const input = documents
|
||||||
|
.find((doc) => doc.frame.frameId === "nested")
|
||||||
|
?.domNodes?.find((n) => n.backendNodeId === 31);
|
||||||
|
|
||||||
expect(nestedIframe?.localRect).toEqual({ x: 5, y: 6, w: 100, h: 80 });
|
expect(nestedIframe?.localRect).toEqual({ x: 5, y: 6, w: 100, h: 80 });
|
||||||
expect(nestedIframe?.rect).toEqual({ x: 15, y: 26, w: 100, h: 80 });
|
expect(nestedIframe?.rect).toEqual({ x: 15, y: 26, w: 100, h: 80 });
|
||||||
@@ -1203,6 +1197,7 @@ describe("captureViewModel", () => {
|
|||||||
strings: S,
|
strings: S,
|
||||||
documents: [
|
documents: [
|
||||||
{
|
{
|
||||||
|
frameId: "root",
|
||||||
nodes: {
|
nodes: {
|
||||||
parentIndex: [-1, 0, 1],
|
parentIndex: [-1, 0, 1],
|
||||||
nodeName: [i("html"), i("body"), i("div")],
|
nodeName: [i("html"), i("body"), i("div")],
|
||||||
@@ -1237,7 +1232,7 @@ describe("captureViewModel", () => {
|
|||||||
throw new Error(method);
|
throw new Error(method);
|
||||||
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
}) as unknown as <T>(tabId: number, method: string, params?: object) => Promise<T>,
|
||||||
};
|
};
|
||||||
const { nodes } = await captureViewModel(cdp, 4);
|
const { nodes } = semanticCapture(await captureObservationFacts(cdp, 4)).captured;
|
||||||
expect(nodes.find((n) => n.backendNodeId === 12)?.rect?.y).toBe(400 - 100);
|
expect(nodes.find((n) => n.backendNodeId === 12)?.rect?.y).toBe(400 - 100);
|
||||||
});
|
});
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,156 @@
|
|||||||
|
import { describe, expect, it } from "vitest";
|
||||||
|
import { OVERLAY_HOST_MARKER_ATTR } from "@/lib/overlay-bridge";
|
||||||
|
import { buildDocumentIndex, type DecodedNode } from "../facts";
|
||||||
|
import { decodeDocument, REQUESTED_STYLES } from "../snapshot";
|
||||||
|
|
||||||
|
function node(id: number, parent: number | null): DecodedNode {
|
||||||
|
return {
|
||||||
|
backendNodeId: id,
|
||||||
|
parentBackendNodeId: parent,
|
||||||
|
tag: "div",
|
||||||
|
attrs: {},
|
||||||
|
paintOrder: 0,
|
||||||
|
position: "static",
|
||||||
|
pointerEvents: "auto",
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
describe("document facts", () => {
|
||||||
|
it.each([
|
||||||
|
1_000, 10_000, 100_000,
|
||||||
|
])("indexes deep and wide inputs of %i without copying ancestors", async (count) => {
|
||||||
|
for (const wide of [false, true]) {
|
||||||
|
let reads = 0;
|
||||||
|
const input = Array.from({ length: count }, (_, i) => {
|
||||||
|
const item = node(i, i ? (wide ? 0 : i - 1) : null);
|
||||||
|
Object.defineProperty(item, "parentBackendNodeId", {
|
||||||
|
get: () => {
|
||||||
|
reads++;
|
||||||
|
return i ? (wide ? 0 : i - 1) : null;
|
||||||
|
},
|
||||||
|
});
|
||||||
|
return item;
|
||||||
|
}).reverse(); // parent need not precede child
|
||||||
|
const index = await buildDocumentIndex(input);
|
||||||
|
expect(index.nodes.size).toBe(count);
|
||||||
|
expect(index.ancestry.get(count - 1)).toEqual({ complete: true, overlay: false });
|
||||||
|
expect(index.children.get(0)?.length).toBe(wide ? count - 1 : 1);
|
||||||
|
expect(reads).toBeLessThanOrEqual(count * 8);
|
||||||
|
expect(index.nodes.get(count - 1)).toBe(input[0]);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
it("marks cycles and orphans incomplete and propagates overlay through shadow roots", async () => {
|
||||||
|
const host = { ...node(1, null), attrs: { [OVERLAY_HOST_MARKER_ATTR]: "" } };
|
||||||
|
const input = [
|
||||||
|
node(3, 2),
|
||||||
|
{ ...node(2, 1), tag: "#document-fragment" },
|
||||||
|
host,
|
||||||
|
node(4, 99),
|
||||||
|
node(5, 6),
|
||||||
|
node(6, 5),
|
||||||
|
node(7, 6),
|
||||||
|
];
|
||||||
|
const index = await buildDocumentIndex(input);
|
||||||
|
expect([...index.excludedBackendNodeIds].sort()).toEqual([1, 2, 3]);
|
||||||
|
for (const id of [4, 5, 6, 7]) expect(index.ancestry.get(id)?.complete).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("does not propagate a descendant overlay backwards into a cycle", async () => {
|
||||||
|
const overlay = { ...node(3, 1), attrs: { [OVERLAY_HOST_MARKER_ATTR]: "" } };
|
||||||
|
const index = await buildDocumentIndex([overlay, node(1, 2), node(2, 1)]);
|
||||||
|
expect([...index.excludedBackendNodeIds]).toEqual([3]);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("does not treat a missing snapshot parent as a complete root", async () => {
|
||||||
|
const { nodes } = await decodeDocument(
|
||||||
|
{ nodes: { backendNodeId: [1], nodeName: [0], parentIndex: [99] } },
|
||||||
|
["div"],
|
||||||
|
);
|
||||||
|
const index = await buildDocumentIndex(nodes);
|
||||||
|
expect(index.ancestry.get(1)?.complete).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("retains raw box units and independent style/interaction evidence", async () => {
|
||||||
|
const strings = [
|
||||||
|
"html",
|
||||||
|
"canvas",
|
||||||
|
"aria-hidden",
|
||||||
|
"true",
|
||||||
|
"inert",
|
||||||
|
"",
|
||||||
|
"hidden",
|
||||||
|
"visible",
|
||||||
|
"0",
|
||||||
|
"auto",
|
||||||
|
"scroll",
|
||||||
|
"matrix(2,0,0,2,0,0)",
|
||||||
|
];
|
||||||
|
const styles = REQUESTED_STYLES.map((style) =>
|
||||||
|
style === "visibility"
|
||||||
|
? 7
|
||||||
|
: style === "opacity"
|
||||||
|
? 8
|
||||||
|
: style === "overflow-x"
|
||||||
|
? 10
|
||||||
|
: style === "transform"
|
||||||
|
? 11
|
||||||
|
: 9,
|
||||||
|
);
|
||||||
|
const { nodes } = await decodeDocument(
|
||||||
|
{
|
||||||
|
nodes: {
|
||||||
|
backendNodeId: [1, 2],
|
||||||
|
parentIndex: [-1, 0],
|
||||||
|
nodeName: [0, 1],
|
||||||
|
attributes: [[2, 3, 4, 5], []],
|
||||||
|
},
|
||||||
|
layout: {
|
||||||
|
nodeIndex: [0, 1],
|
||||||
|
bounds: [
|
||||||
|
[0, 0, 200, 100],
|
||||||
|
[10, 20, 120, 40],
|
||||||
|
],
|
||||||
|
clientRects: [
|
||||||
|
[5, 5, 90, 40],
|
||||||
|
[0, 0, 120, 40],
|
||||||
|
],
|
||||||
|
styles: [styles, styles],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
strings,
|
||||||
|
);
|
||||||
|
expect(nodes[0].attrs).toEqual({ "aria-hidden": "true", inert: "" });
|
||||||
|
expect(nodes[0].layout).toMatchObject({
|
||||||
|
boundsSpace: "snapshot-document-css",
|
||||||
|
clientSpace: "unscaled-client-offset-css",
|
||||||
|
clientRect: [5, 5, 90, 40],
|
||||||
|
styles: {
|
||||||
|
visibility: "visible",
|
||||||
|
opacity: "0",
|
||||||
|
"overflow-x": "scroll",
|
||||||
|
transform: "matrix(2,0,0,2,0,0)",
|
||||||
|
},
|
||||||
|
});
|
||||||
|
expect(nodes[0]).not.toHaveProperty("rect");
|
||||||
|
expect(nodes[0]).not.toHaveProperty("excluded");
|
||||||
|
});
|
||||||
|
|
||||||
|
it("yields during large inputs so cancellation interrupts indexing", async () => {
|
||||||
|
const controller = new AbortController();
|
||||||
|
const input = Array.from({ length: 100_000 }, (_, i) => node(i, i ? i - 1 : null));
|
||||||
|
let reads = 0;
|
||||||
|
for (const item of input)
|
||||||
|
Object.defineProperty(item, "attrs", {
|
||||||
|
get: () => {
|
||||||
|
reads++;
|
||||||
|
if (reads === 100) controller.abort();
|
||||||
|
return {};
|
||||||
|
},
|
||||||
|
});
|
||||||
|
await expect(buildDocumentIndex(input, controller.signal)).rejects.toMatchObject({
|
||||||
|
name: "AbortError",
|
||||||
|
});
|
||||||
|
expect(reads).toBeLessThanOrEqual(356);
|
||||||
|
});
|
||||||
|
});
|
||||||
@@ -1,6 +1,6 @@
|
|||||||
import { describe, expect, it, vi } from "vitest";
|
import { describe, expect, it, vi } from "vitest";
|
||||||
import { type CdpRunner, cdpRunnerForTarget } from "../../shared";
|
import { type CdpRunner, cdpRunnerForTarget } from "../../shared";
|
||||||
import { captureViewModel } from "../capture";
|
import { captureObservationFacts } from "../capture-coordinator";
|
||||||
|
|
||||||
interface Control {
|
interface Control {
|
||||||
id: number;
|
id: number;
|
||||||
@@ -106,17 +106,19 @@ describe("form state identity", () => {
|
|||||||
[11, state("alice@example.com")],
|
[11, state("alice@example.com")],
|
||||||
]),
|
]),
|
||||||
);
|
);
|
||||||
const captured = await captureViewModel(cdp, 7);
|
const captured = await captureObservationFacts(cdp, 7);
|
||||||
expect(captured.nodes.find((n) => n.backendNodeId === 10)).toMatchObject({
|
expect(captured.documents[0].domNodes.find((n) => n.backendNodeId === 10)).toMatchObject({
|
||||||
formValue: "",
|
formValue: "",
|
||||||
formState: "empty",
|
formState: "empty",
|
||||||
formPlaceholder: "snapshot placeholder",
|
formPlaceholder: "snapshot placeholder",
|
||||||
});
|
});
|
||||||
expect(captured.nodes.find((n) => n.backendNodeId === 11)).toMatchObject({
|
expect(captured.documents[0].domNodes.find((n) => n.backendNodeId === 11)).toMatchObject({
|
||||||
formValue: "alice@example.com",
|
formValue: "alice@example.com",
|
||||||
});
|
});
|
||||||
expect(captured.nodes.find((n) => n.backendNodeId === 12)).toMatchObject({ formValue: "上海" });
|
expect(captured.documents[0].domNodes.find((n) => n.backendNodeId === 12)).toMatchObject({
|
||||||
expect(captured.nodes.some((n) => n.backendNodeId === 99)).toBe(false);
|
formValue: "上海",
|
||||||
|
});
|
||||||
|
expect(captured.documents[0].domNodes.some((n) => n.backendNodeId === 99)).toBe(false);
|
||||||
});
|
});
|
||||||
|
|
||||||
it("does not shift state between frames when one frame is absent from the runtime batch", async () => {
|
it("does not shift state between frames when one frame is absent from the runtime batch", async () => {
|
||||||
@@ -133,8 +135,10 @@ describe("form state identity", () => {
|
|||||||
[21, state("second frame")],
|
[21, state("second frame")],
|
||||||
]),
|
]),
|
||||||
);
|
);
|
||||||
const captured = await captureViewModel(cdp, 7);
|
const captured = await captureObservationFacts(cdp, 7);
|
||||||
const controls = [...captured.iframeNodes.values()].flat();
|
const controls = captured.documents
|
||||||
|
.filter((doc) => doc.frame.parentFrameId)
|
||||||
|
.flatMap((doc) => doc.domNodes);
|
||||||
expect(controls.find((n) => n.backendNodeId === 11)?.formValue).toBe("inaccessible frame");
|
expect(controls.find((n) => n.backendNodeId === 11)?.formValue).toBe("inaccessible frame");
|
||||||
expect(controls.find((n) => n.backendNodeId === 21)?.formValue).toBe("second frame");
|
expect(controls.find((n) => n.backendNodeId === 21)?.formValue).toBe("second frame");
|
||||||
expect(controls.find((n) => n.backendNodeId === 31)?.formValue).toBe("third frame");
|
expect(controls.find((n) => n.backendNodeId === 31)?.formValue).toBe("third frame");
|
||||||
@@ -151,8 +155,13 @@ describe("form state identity", () => {
|
|||||||
});
|
});
|
||||||
const cdp = { send, sendToTarget } as unknown as CdpRunner;
|
const cdp = { send, sendToTarget } as unknown as CdpRunner;
|
||||||
for (const sessionId of ["left-frame", "right-frame"]) {
|
for (const sessionId of ["left-frame", "right-frame"]) {
|
||||||
const captured = await captureViewModel(cdpRunnerForTarget(cdp, { tabId: 7, sessionId }), 7);
|
const captured = await captureObservationFacts(
|
||||||
expect(captured.nodes.find((n) => n.backendNodeId === 11)?.formValue).toBe(sessionId);
|
cdpRunnerForTarget(cdp, { tabId: 7, sessionId }),
|
||||||
|
7,
|
||||||
|
);
|
||||||
|
expect(captured.documents[0].domNodes.find((n) => n.backendNodeId === 11)?.formValue).toBe(
|
||||||
|
sessionId,
|
||||||
|
);
|
||||||
expect(sendToTarget).toHaveBeenCalledWith(
|
expect(sendToTarget).toHaveBeenCalledWith(
|
||||||
{ tabId: 7, sessionId },
|
{ tabId: 7, sessionId },
|
||||||
"Runtime.releaseObjectGroup",
|
"Runtime.releaseObjectGroup",
|
||||||
@@ -178,8 +187,8 @@ describe("form state identity", () => {
|
|||||||
result.result.deepSerializedValue.value[0].value[1].value = "not json";
|
result.result.deepSerializedValue.value[0].value[1].value = "not json";
|
||||||
return result;
|
return result;
|
||||||
});
|
});
|
||||||
const captured = await captureViewModel(cdp, 7);
|
const captured = await captureObservationFacts(cdp, 7);
|
||||||
expect(captured.nodes.find((n) => n.backendNodeId === 11)).toMatchObject({
|
expect(captured.documents[0].domNodes.find((n) => n.backendNodeId === 11)).toMatchObject({
|
||||||
formValue: "snapshot value",
|
formValue: "snapshot value",
|
||||||
formPlaceholder: "snapshot placeholder",
|
formPlaceholder: "snapshot placeholder",
|
||||||
});
|
});
|
||||||
@@ -192,7 +201,7 @@ describe("form state identity", () => {
|
|||||||
controller.abort();
|
controller.abort();
|
||||||
return reply([[11, state("new")]]);
|
return reply([[11, state("new")]]);
|
||||||
});
|
});
|
||||||
await expect(captureViewModel(cdp, 7, { signal: controller.signal })).rejects.toMatchObject({
|
await expect(captureObservationFacts(cdp, 7, controller.signal)).rejects.toMatchObject({
|
||||||
name: "AbortError",
|
name: "AbortError",
|
||||||
});
|
});
|
||||||
expect(send).toHaveBeenCalledWith(7, "Runtime.releaseObjectGroup", expect.anything());
|
expect(send).toHaveBeenCalledWith(7, "Runtime.releaseObjectGroup", expect.anything());
|
||||||
@@ -216,9 +225,13 @@ describe("form state identity", () => {
|
|||||||
return result;
|
return result;
|
||||||
},
|
},
|
||||||
);
|
);
|
||||||
const captured = await captureViewModel(cdp, 7);
|
const captured = await captureObservationFacts(cdp, 7);
|
||||||
expect(captured.nodes.find((n) => n.backendNodeId === 11)?.formValue).toBe("snapshot value");
|
expect(captured.documents[0].domNodes.find((n) => n.backendNodeId === 11)?.formValue).toBe(
|
||||||
expect(captured.nodes.find((n) => n.backendNodeId === 12)?.formValue).toBe("live value");
|
"snapshot value",
|
||||||
|
);
|
||||||
|
expect(captured.documents[0].domNodes.find((n) => n.backendNodeId === 12)?.formValue).toBe(
|
||||||
|
"live value",
|
||||||
|
);
|
||||||
});
|
});
|
||||||
|
|
||||||
it.each([
|
it.each([
|
||||||
@@ -228,8 +241,8 @@ describe("form state identity", () => {
|
|||||||
const { cdp } = fakeCdp([[{ id: 11, value: "secret", type }]], () =>
|
const { cdp } = fakeCdp([[{ id: 11, value: "secret", type }]], () =>
|
||||||
reply([[11, { ...state("secret"), sensitive: true }]]),
|
reply([[11, { ...state("secret"), sensitive: true }]]),
|
||||||
);
|
);
|
||||||
const captured = await captureViewModel(cdp, 7);
|
const captured = await captureObservationFacts(cdp, 7);
|
||||||
const control = captured.nodes.find((n) => n.backendNodeId === 11);
|
const control = captured.documents[0].domNodes.find((n) => n.backendNodeId === 11);
|
||||||
expect(control?.formState).toBe("filled");
|
expect(control?.formState).toBe("filled");
|
||||||
expect(control?.formValue).toBeUndefined();
|
expect(control?.formValue).toBeUndefined();
|
||||||
expect(control?.formDefaultValue).toBeUndefined();
|
expect(control?.formDefaultValue).toBeUndefined();
|
||||||
@@ -257,9 +270,13 @@ describe("form state identity", () => {
|
|||||||
expect(entries).toHaveLength(250);
|
expect(entries).toHaveLength(250);
|
||||||
return reply(entries.map(([element, json]) => [element.backendNodeId, JSON.parse(json)]));
|
return reply(entries.map(([element, json]) => [element.backendNodeId, JSON.parse(json)]));
|
||||||
});
|
});
|
||||||
const captured = await captureViewModel(cdp, 7);
|
const captured = await captureObservationFacts(cdp, 7);
|
||||||
expect(captured.nodes.find((n) => n.backendNodeId === 250)?.formValue).toBe("live 250");
|
expect(captured.documents[0].domNodes.find((n) => n.backendNodeId === 250)?.formValue).toBe(
|
||||||
expect(captured.nodes.find((n) => n.backendNodeId === 251)?.formValue).toBe("old");
|
"live 250",
|
||||||
|
);
|
||||||
|
expect(captured.documents[0].domNodes.find((n) => n.backendNodeId === 251)?.formValue).toBe(
|
||||||
|
"old",
|
||||||
|
);
|
||||||
expect(send.mock.calls.filter(([, method]) => method === "Runtime.evaluate")).toHaveLength(1);
|
expect(send.mock.calls.filter(([, method]) => method === "Runtime.evaluate")).toHaveLength(1);
|
||||||
expect(
|
expect(
|
||||||
send.mock.calls.filter(([, method]) => method === "Runtime.releaseObjectGroup"),
|
send.mock.calls.filter(([, method]) => method === "Runtime.releaseObjectGroup"),
|
||||||
|
|||||||
@@ -1,260 +0,0 @@
|
|||||||
import { describe, expect, it, vi } from "vitest";
|
|
||||||
import type { CdpRunner } from "../../shared";
|
|
||||||
import type { CapturedNode, CapturedViewModel } from "../capture";
|
|
||||||
import { captureFrameData } from "../frame-capture";
|
|
||||||
|
|
||||||
function ownerNode(backendNodeId: number, x: number): CapturedNode {
|
|
||||||
return {
|
|
||||||
backendNodeId,
|
|
||||||
parentBackendNodeId: 1,
|
|
||||||
frameId: "main",
|
|
||||||
ownerFrameBackendNodeId: null,
|
|
||||||
tag: "iframe",
|
|
||||||
attrs: {},
|
|
||||||
rect: { x, y: 100, w: 300, h: 200 },
|
|
||||||
localRect: { x, y: 100, w: 300, h: 200 },
|
|
||||||
paintOrder: 1,
|
|
||||||
position: "static",
|
|
||||||
pointerEvents: "auto",
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
function childSnapshot(frameId: string, backendNodeId: number) {
|
|
||||||
const strings = [frameId, "body", "button", "static", "auto", "pointer"];
|
|
||||||
return {
|
|
||||||
strings,
|
|
||||||
documents: [
|
|
||||||
{
|
|
||||||
frameId,
|
|
||||||
nodes: {
|
|
||||||
parentIndex: [-1, 0],
|
|
||||||
nodeName: [1, 2],
|
|
||||||
backendNodeId: [backendNodeId - 1, backendNodeId],
|
|
||||||
attributes: [[], []],
|
|
||||||
},
|
|
||||||
layout: {
|
|
||||||
nodeIndex: [0, 1],
|
|
||||||
styles: [
|
|
||||||
[3, 4, 4],
|
|
||||||
[3, 4, 5],
|
|
||||||
],
|
|
||||||
bounds: [
|
|
||||||
[0, 0, 300, 200],
|
|
||||||
[10, 20, 100, 40],
|
|
||||||
],
|
|
||||||
paintOrders: [0, 1],
|
|
||||||
},
|
|
||||||
},
|
|
||||||
],
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
describe("captureFrameData", () => {
|
|
||||||
it("retains nested owner failures when merging a captured OOPIF", async () => {
|
|
||||||
const owner = ownerNode(10, 50);
|
|
||||||
const captured: CapturedViewModel = {
|
|
||||||
nodes: [owner],
|
|
||||||
viewport: { width: 1000, height: 800 },
|
|
||||||
iframeNodes: new Map(),
|
|
||||||
frameNodes: new Map([["main", [owner]]]),
|
|
||||||
rootFrameId: "main",
|
|
||||||
excludedBackendNodeIds: new Set(),
|
|
||||||
};
|
|
||||||
const child = childSnapshot("child", 101);
|
|
||||||
const nested = childSnapshot("nested", 201);
|
|
||||||
const document = child.documents[0];
|
|
||||||
const snapshot = {
|
|
||||||
strings: child.strings,
|
|
||||||
documents: [
|
|
||||||
{
|
|
||||||
...document,
|
|
||||||
nodes: { ...document.nodes, contentDocumentIndex: { index: [1], value: [1] } },
|
|
||||||
},
|
|
||||||
nested.documents[0],
|
|
||||||
],
|
|
||||||
};
|
|
||||||
const reply = async (_target: unknown, method: string) => {
|
|
||||||
if (method === "Page.getLayoutMetrics")
|
|
||||||
return { cssLayoutViewport: { clientWidth: 300, clientHeight: 200 } };
|
|
||||||
if (method === "DOMSnapshot.captureSnapshot") return snapshot;
|
|
||||||
if (method === "DOM.getBoxModel")
|
|
||||||
return { model: { content: [50, 100, 350, 100, 350, 300, 50, 300] } };
|
|
||||||
if (method === "DOM.resolveNode") throw new Error("nested owner replaced");
|
|
||||||
return {};
|
|
||||||
};
|
|
||||||
const cdp: CdpRunner = {
|
|
||||||
send: vi.fn(reply) as CdpRunner["send"],
|
|
||||||
sendToTarget: vi.fn(reply) as NonNullable<CdpRunner["sendToTarget"]>,
|
|
||||||
getFrameGraph: vi.fn(async () => ({
|
|
||||||
rootFrameId: "main",
|
|
||||||
frames: [
|
|
||||||
{ frameId: "main", target: { tabId: 4 } },
|
|
||||||
{
|
|
||||||
frameId: "child",
|
|
||||||
parentFrameId: "main",
|
|
||||||
ownerBackendNodeId: 10,
|
|
||||||
target: { tabId: 4, sessionId: "child-session" },
|
|
||||||
},
|
|
||||||
],
|
|
||||||
})),
|
|
||||||
};
|
|
||||||
await captureFrameData(cdp, 4, captured);
|
|
||||||
expect(captured.frameGeometryIssues).toEqual([
|
|
||||||
{
|
|
||||||
status: "unavailable",
|
|
||||||
source: { target: { tabId: 4, sessionId: "child-session" }, frameId: "nested" },
|
|
||||||
ownerBackendNodeId: 101,
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
expect(captured.frameNodes?.get("nested")?.[1]).toMatchObject({
|
|
||||||
rect: null,
|
|
||||||
localRect: { x: 10, y: 20, w: 100, h: 40 },
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it("captures and positions multiple OOPIF documents missing from the root snapshot", async () => {
|
|
||||||
const leftOwner = ownerNode(10, 50);
|
|
||||||
const rightOwner = ownerNode(20, 500);
|
|
||||||
const captured: CapturedViewModel = {
|
|
||||||
nodes: [leftOwner, rightOwner],
|
|
||||||
viewport: { width: 1000, height: 800 },
|
|
||||||
iframeNodes: new Map(),
|
|
||||||
frameNodes: new Map([["main", [leftOwner, rightOwner]]]),
|
|
||||||
frameOwnerBackendNodeIds: new Map(),
|
|
||||||
rootFrameId: "main",
|
|
||||||
excludedBackendNodeIds: new Set(),
|
|
||||||
};
|
|
||||||
const sendToTarget = vi.fn(async (target, method) => {
|
|
||||||
if (method === "Page.getLayoutMetrics") {
|
|
||||||
return { cssLayoutViewport: { clientWidth: 300, clientHeight: 200, pageX: 0, pageY: 0 } };
|
|
||||||
}
|
|
||||||
if (method === "DOMSnapshot.enable" || method === "Accessibility.enable") return {};
|
|
||||||
if (method === "DOMSnapshot.captureSnapshot") {
|
|
||||||
return target.sessionId === "left-session"
|
|
||||||
? childSnapshot("left", 101)
|
|
||||||
: childSnapshot("right", 201);
|
|
||||||
}
|
|
||||||
if (method === "Accessibility.getFullAXTree") return { nodes: [] };
|
|
||||||
throw new Error(`unexpected ${method}`);
|
|
||||||
});
|
|
||||||
const cdp: CdpRunner = {
|
|
||||||
send: vi.fn(async (_tabId, method, params) => {
|
|
||||||
if (method === "Accessibility.enable") return {};
|
|
||||||
if (method === "Accessibility.getFullAXTree") return { nodes: [] };
|
|
||||||
if (method === "DOM.getBoxModel") {
|
|
||||||
const backendNodeId = (params as { backendNodeId?: number })?.backendNodeId;
|
|
||||||
const x = backendNodeId === 10 ? 50 : 500;
|
|
||||||
return { model: { content: [x, 100, x + 300, 100, x + 300, 300, x, 300] } };
|
|
||||||
}
|
|
||||||
if (method === "Page.getLayoutMetrics") {
|
|
||||||
return { cssLayoutViewport: { clientWidth: 1000, clientHeight: 800 } };
|
|
||||||
}
|
|
||||||
throw new Error(`unexpected root ${method}`);
|
|
||||||
}) as CdpRunner["send"],
|
|
||||||
sendToTarget: sendToTarget as unknown as NonNullable<CdpRunner["sendToTarget"]>,
|
|
||||||
getFrameGraph: vi.fn(async () => ({
|
|
||||||
rootFrameId: "main",
|
|
||||||
frames: [
|
|
||||||
{ frameId: "main", target: { tabId: 4 } },
|
|
||||||
{
|
|
||||||
frameId: "left",
|
|
||||||
parentFrameId: "main",
|
|
||||||
ownerBackendNodeId: 10,
|
|
||||||
target: { tabId: 4, sessionId: "left-session" },
|
|
||||||
},
|
|
||||||
{
|
|
||||||
frameId: "right",
|
|
||||||
parentFrameId: "main",
|
|
||||||
ownerBackendNodeId: 20,
|
|
||||||
target: { tabId: 4, sessionId: "right-session" },
|
|
||||||
},
|
|
||||||
],
|
|
||||||
})),
|
|
||||||
};
|
|
||||||
|
|
||||||
const trees = await captureFrameData(cdp, 4, captured);
|
|
||||||
|
|
||||||
expect(trees.map((tree) => tree.frameId)).toEqual(["main", "left", "right"]);
|
|
||||||
expect(
|
|
||||||
captured.frameNodes?.get("left")?.find((node) => node.backendNodeId === 101)?.rect,
|
|
||||||
).toEqual({ x: 60, y: 120, w: 100, h: 40 });
|
|
||||||
expect(
|
|
||||||
captured.frameNodes?.get("right")?.find((node) => node.backendNodeId === 201)?.rect,
|
|
||||||
).toEqual({ x: 510, y: 120, w: 100, h: 40 });
|
|
||||||
expect(captured.frameOwnerBackendNodeIds).toEqual(
|
|
||||||
new Map([
|
|
||||||
["left", 10],
|
|
||||||
["right", 20],
|
|
||||||
]),
|
|
||||||
);
|
|
||||||
expect(captured.frameParentIds).toEqual(
|
|
||||||
new Map([
|
|
||||||
["left", "main"],
|
|
||||||
["right", "main"],
|
|
||||||
]),
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("keeps OOPIF semantics when viewport projection is unavailable", async () => {
|
|
||||||
const owner = { ...ownerNode(10, 50), rect: null, localRect: null };
|
|
||||||
const captured: CapturedViewModel = {
|
|
||||||
nodes: [owner],
|
|
||||||
viewport: { width: 1000, height: 800 },
|
|
||||||
iframeNodes: new Map(),
|
|
||||||
frameNodes: new Map([["main", [owner]]]),
|
|
||||||
rootFrameId: "main",
|
|
||||||
excludedBackendNodeIds: new Set(),
|
|
||||||
};
|
|
||||||
const cdp: CdpRunner = {
|
|
||||||
send: vi.fn(async (_tabId, method) => {
|
|
||||||
if (method === "Accessibility.enable") return {};
|
|
||||||
if (method === "Accessibility.getFullAXTree") return { nodes: [] };
|
|
||||||
if (method === "DOM.getBoxModel") throw new Error("owner geometry unavailable");
|
|
||||||
if (method === "Page.getLayoutMetrics") {
|
|
||||||
return { cssLayoutViewport: { clientWidth: 1000, clientHeight: 800 } };
|
|
||||||
}
|
|
||||||
throw new Error(`unexpected root ${method}`);
|
|
||||||
}) as CdpRunner["send"],
|
|
||||||
sendToTarget: vi.fn(async (_target, method) => {
|
|
||||||
if (method === "Page.getLayoutMetrics") throw new Error("viewport unavailable");
|
|
||||||
if (method === "DOMSnapshot.enable" || method === "Accessibility.enable") return {};
|
|
||||||
if (method === "DOMSnapshot.captureSnapshot") return childSnapshot("child", 101);
|
|
||||||
if (method === "Accessibility.getFullAXTree") {
|
|
||||||
return {
|
|
||||||
nodes: [
|
|
||||||
{
|
|
||||||
nodeId: "button",
|
|
||||||
backendDOMNodeId: 101,
|
|
||||||
role: { type: "role", value: "button" },
|
|
||||||
name: { type: "computedString", value: "Continue" },
|
|
||||||
},
|
|
||||||
],
|
|
||||||
};
|
|
||||||
}
|
|
||||||
throw new Error(`unexpected ${method}`);
|
|
||||||
}) as unknown as NonNullable<CdpRunner["sendToTarget"]>,
|
|
||||||
getFrameGraph: vi.fn(async () => ({
|
|
||||||
rootFrameId: "main",
|
|
||||||
frames: [
|
|
||||||
{ frameId: "main", target: { tabId: 4 } },
|
|
||||||
{
|
|
||||||
frameId: "child",
|
|
||||||
parentFrameId: "main",
|
|
||||||
ownerBackendNodeId: 10,
|
|
||||||
target: { tabId: 4, sessionId: "child-session" },
|
|
||||||
},
|
|
||||||
],
|
|
||||||
})),
|
|
||||||
};
|
|
||||||
|
|
||||||
const documents = await captureFrameData(cdp, 4, captured);
|
|
||||||
const childDocument = documents.find((document) => document.frameId === "child");
|
|
||||||
|
|
||||||
expect(childDocument?.axNodes).toEqual([
|
|
||||||
expect.objectContaining({ backendDOMNodeId: 101, frameId: "child" }),
|
|
||||||
]);
|
|
||||||
expect(childDocument?.domNodes.find((node) => node.backendNodeId === 101)).toEqual(
|
|
||||||
expect.objectContaining({ rect: null, localRect: { x: 10, y: 20, w: 100, h: 40 } }),
|
|
||||||
);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,7 +1,12 @@
|
|||||||
import { describe, expect, it } from "vitest";
|
import { describe, expect, it, vi } from "vitest";
|
||||||
import type { CdpFrameGraph } from "@/browser-driver/frame-graph";
|
import type { CdpFrameGraph } from "@/browser-driver/frame-graph";
|
||||||
import type { CapturedNode, CapturedViewModel } from "../capture";
|
import type { CapturedNode } from "../capture";
|
||||||
import { buildFrameDocuments, type FrameAxBatch, type FrameOwnedAxNode } from "../frame-document";
|
import {
|
||||||
|
buildFrameDocuments,
|
||||||
|
type FrameAxBatch,
|
||||||
|
type FrameDomInput,
|
||||||
|
type FrameOwnedAxNode,
|
||||||
|
} from "../frame-document";
|
||||||
|
|
||||||
function domNode(frameId: string, backendNodeId: number): CapturedNode {
|
function domNode(frameId: string, backendNodeId: number): CapturedNode {
|
||||||
return {
|
return {
|
||||||
@@ -17,19 +22,16 @@ function domNode(frameId: string, backendNodeId: number): CapturedNode {
|
|||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
function captured(frameNodes: Map<string, CapturedNode[]>): CapturedViewModel {
|
function captured(frameNodes: Map<string, CapturedNode[]>): FrameDomInput {
|
||||||
return {
|
return {
|
||||||
nodes: frameNodes.get("main") ?? [],
|
nodes: frameNodes.get("main") ?? [],
|
||||||
viewport: { width: 800, height: 600 },
|
|
||||||
iframeNodes: new Map(),
|
|
||||||
frameNodes,
|
frameNodes,
|
||||||
rootFrameId: "main",
|
rootFrameId: "main",
|
||||||
excludedBackendNodeIds: new Set(),
|
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
describe("buildFrameDocuments", () => {
|
describe("buildFrameDocuments", () => {
|
||||||
it("partitions mixed AX results by node ownership and cuts cross-frame parent edges", () => {
|
it("partitions mixed AX results by node ownership and cuts cross-frame parent edges", async () => {
|
||||||
const graph: CdpFrameGraph = {
|
const graph: CdpFrameGraph = {
|
||||||
rootFrameId: "main",
|
rootFrameId: "main",
|
||||||
frames: [
|
frames: [
|
||||||
@@ -54,7 +56,7 @@ describe("buildFrameDocuments", () => {
|
|||||||
},
|
},
|
||||||
];
|
];
|
||||||
|
|
||||||
const documents = buildFrameDocuments(
|
const documents = await buildFrameDocuments(
|
||||||
graph,
|
graph,
|
||||||
batches,
|
batches,
|
||||||
captured(
|
captured(
|
||||||
@@ -72,7 +74,7 @@ describe("buildFrameDocuments", () => {
|
|||||||
expect(childNodes.find((node) => node.nodeId === "virtual")?.frameId).toBe("child");
|
expect(childNodes.find((node) => node.nodeId === "virtual")?.frameId).toBe("child");
|
||||||
});
|
});
|
||||||
|
|
||||||
it("deduplicates overlapping same-target AX results using the strongest ownership", () => {
|
it("deduplicates overlapping same-target AX results using the strongest ownership", async () => {
|
||||||
const graph: CdpFrameGraph = {
|
const graph: CdpFrameGraph = {
|
||||||
rootFrameId: "main",
|
rootFrameId: "main",
|
||||||
frames: [
|
frames: [
|
||||||
@@ -92,7 +94,7 @@ describe("buildFrameDocuments", () => {
|
|||||||
},
|
},
|
||||||
];
|
];
|
||||||
|
|
||||||
const documents = buildFrameDocuments(
|
const documents = await buildFrameDocuments(
|
||||||
graph,
|
graph,
|
||||||
batches,
|
batches,
|
||||||
captured(
|
captured(
|
||||||
@@ -107,7 +109,7 @@ describe("buildFrameDocuments", () => {
|
|||||||
expect(documents.find((document) => document.frameId === "child")?.axNodes).toHaveLength(1);
|
expect(documents.find((document) => document.frameId === "child")?.axNodes).toHaveLength(1);
|
||||||
});
|
});
|
||||||
|
|
||||||
it("uses target-scoped backend ownership when OOPIF backend ids collide", () => {
|
it("uses target-scoped backend ownership when OOPIF backend ids collide", async () => {
|
||||||
const graph: CdpFrameGraph = {
|
const graph: CdpFrameGraph = {
|
||||||
rootFrameId: "main",
|
rootFrameId: "main",
|
||||||
frames: [
|
frames: [
|
||||||
@@ -129,7 +131,7 @@ describe("buildFrameDocuments", () => {
|
|||||||
nodes: [{ nodeId: `${frame.frameId}-node`, backendDOMNodeId: 7 }],
|
nodes: [{ nodeId: `${frame.frameId}-node`, backendDOMNodeId: 7 }],
|
||||||
}));
|
}));
|
||||||
|
|
||||||
const documents = buildFrameDocuments(
|
const documents = await buildFrameDocuments(
|
||||||
graph,
|
graph,
|
||||||
batches,
|
batches,
|
||||||
captured(
|
captured(
|
||||||
@@ -149,7 +151,7 @@ describe("buildFrameDocuments", () => {
|
|||||||
);
|
);
|
||||||
});
|
});
|
||||||
|
|
||||||
it("uses captured frame parent identity instead of guessing from backend ids", () => {
|
it("uses captured frame parent identity instead of guessing from backend ids", async () => {
|
||||||
const frameNodes = new Map([
|
const frameNodes = new Map([
|
||||||
["main", [domNode("main", 1)]],
|
["main", [domNode("main", 1)]],
|
||||||
["left", [domNode("left", 7)]],
|
["left", [domNode("left", 7)]],
|
||||||
@@ -164,11 +166,42 @@ describe("buildFrameDocuments", () => {
|
|||||||
nodes: [],
|
nodes: [],
|
||||||
}));
|
}));
|
||||||
|
|
||||||
const documents = buildFrameDocuments(null, frames, capture);
|
const documents = await buildFrameDocuments(null, frames, capture);
|
||||||
|
|
||||||
expect(documents.find((document) => document.frameId === "child")).toMatchObject({
|
expect(documents.find((document) => document.frameId === "child")).toMatchObject({
|
||||||
parentFrameId: "left",
|
parentFrameId: "left",
|
||||||
ownerBackendNodeId: 7,
|
ownerBackendNodeId: 7,
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
it("rejects explicit AX frame ownership in a different target", async () => {
|
||||||
|
const frames = [
|
||||||
|
{ frameId: "main", target: { tabId: 4 } },
|
||||||
|
{ frameId: "remote", target: { tabId: 4, sessionId: "remote" } },
|
||||||
|
];
|
||||||
|
const unresolved = vi.fn();
|
||||||
|
const documents = await buildFrameDocuments(
|
||||||
|
{ rootFrameId: "main", frames },
|
||||||
|
[{ frame: frames[0], nodes: [{ nodeId: "wrong", frameId: "remote" }] }],
|
||||||
|
captured(new Map()),
|
||||||
|
undefined,
|
||||||
|
unresolved,
|
||||||
|
);
|
||||||
|
expect(documents.every((doc) => doc.axNodes.length === 0)).toBe(true);
|
||||||
|
expect(unresolved).toHaveBeenCalledOnce();
|
||||||
|
});
|
||||||
|
|
||||||
|
it("resolves a reverse-ordered deep AX parent chain without recursion", async () => {
|
||||||
|
const frame = { frameId: "main", target: { tabId: 4 } };
|
||||||
|
const nodes: FrameOwnedAxNode[] = Array.from({ length: 10000 }, (_, i) => ({
|
||||||
|
nodeId: String(i),
|
||||||
|
...(i ? { parentId: String(i - 1) } : { frameId: "main" }),
|
||||||
|
})).reverse();
|
||||||
|
const documents = await buildFrameDocuments(
|
||||||
|
{ rootFrameId: "main", frames: [frame] },
|
||||||
|
[{ frame, nodes }],
|
||||||
|
captured(new Map()),
|
||||||
|
);
|
||||||
|
expect(documents[0].axNodes).toHaveLength(10000);
|
||||||
|
expect(documents[0].axNodes[0].frameId).toBe("main");
|
||||||
|
});
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -0,0 +1,268 @@
|
|||||||
|
import {
|
||||||
|
type CdpFrame,
|
||||||
|
type CdpFrameGraph,
|
||||||
|
type CdpTarget,
|
||||||
|
cdpTargetKey,
|
||||||
|
} from "@/browser-driver/frame-graph";
|
||||||
|
import { cssViewport, GeometryContext } from "../geometry/frame-context";
|
||||||
|
import { type CdpRunner, cdpRunnerForTarget } from "../shared";
|
||||||
|
import { collectOverlayExcludedBackendIds } from "./capture";
|
||||||
|
import type { CapturedSceneInput } from "./facts";
|
||||||
|
import {
|
||||||
|
buildDocumentIndex,
|
||||||
|
type CaptureIssue,
|
||||||
|
isCaptureAbort,
|
||||||
|
type ObservationFacts,
|
||||||
|
throwCaptureAborted,
|
||||||
|
} from "./facts";
|
||||||
|
import { enrichFormControlStates } from "./form-capture";
|
||||||
|
import {
|
||||||
|
buildFrameDocuments,
|
||||||
|
type FrameAxBatch,
|
||||||
|
type FrameDocument,
|
||||||
|
type FrameOwnedAxNode,
|
||||||
|
} from "./frame-document";
|
||||||
|
import { type NormalizedDocument, normalizeSnapshot } from "./normalize";
|
||||||
|
import { REQUESTED_STYLES, type SnapshotReply, snapshotFrameId } from "./snapshot";
|
||||||
|
|
||||||
|
interface TargetBatch<T extends FrameOwnedAxNode> {
|
||||||
|
target: CdpTarget;
|
||||||
|
frames: CdpFrame[];
|
||||||
|
documents: NormalizedDocument[];
|
||||||
|
ax: FrameAxBatch<T>[];
|
||||||
|
fallbackExcluded: Set<number>;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** A fixed worker pool scoped to one capture; each worker issues at most one
|
||||||
|
* collection request at a time. Geometry has its own measurement-phase bound. */
|
||||||
|
async function collectTargets<T>(
|
||||||
|
items: readonly T[],
|
||||||
|
collect: (item: T) => Promise<void>,
|
||||||
|
signal?: AbortSignal,
|
||||||
|
): Promise<void> {
|
||||||
|
let cursor = 0;
|
||||||
|
await Promise.all(
|
||||||
|
Array.from({ length: Math.min(4, items.length) }, async () => {
|
||||||
|
while (cursor < items.length) {
|
||||||
|
throwCaptureAborted(signal);
|
||||||
|
const item = items[cursor++];
|
||||||
|
await collect(item);
|
||||||
|
}
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
/** When topology discovery is unavailable, snapshot documents still carry
|
||||||
|
* explicit target/frame provenance. Only explicit contentDocumentIndex edges
|
||||||
|
* establish ancestry; a foreign frame is never renamed to fit a request. */
|
||||||
|
function snapshotFrames(snapshot: SnapshotReply, target: CdpTarget): CdpFrame[] {
|
||||||
|
const raw = snapshot.documents ?? [];
|
||||||
|
const strings = snapshot.strings ?? [];
|
||||||
|
const frames = raw.map((doc) => {
|
||||||
|
const frameId = snapshotFrameId(doc, strings);
|
||||||
|
return frameId ? ({ frameId, target } as CdpFrame) : undefined;
|
||||||
|
});
|
||||||
|
for (let i = 0; i < raw.length; i++) {
|
||||||
|
const edges = raw[i].nodes?.contentDocumentIndex;
|
||||||
|
const parent = frames[i];
|
||||||
|
if (!parent || !edges) continue;
|
||||||
|
for (let e = 0; e < edges.index.length; e++) {
|
||||||
|
const child = frames[edges.value[e]];
|
||||||
|
const owner = raw[i].nodes?.backendNodeId?.[edges.index[e]];
|
||||||
|
if (child && owner !== undefined && child !== parent) {
|
||||||
|
child.parentFrameId = parent.frameId;
|
||||||
|
child.ownerBackendNodeId = owner;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return frames.filter((frame): frame is CdpFrame => !!frame);
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function captureObservationFacts<T extends FrameOwnedAxNode>(
|
||||||
|
cdp: CdpRunner,
|
||||||
|
tabId: number,
|
||||||
|
signal?: AbortSignal,
|
||||||
|
): Promise<ObservationFacts<T>> {
|
||||||
|
const startedAt = Date.now();
|
||||||
|
throwCaptureAborted(signal);
|
||||||
|
let graph: CdpFrameGraph | undefined;
|
||||||
|
try {
|
||||||
|
graph = await cdp.getFrameGraph?.(tabId);
|
||||||
|
} catch (error) {
|
||||||
|
if (isCaptureAbort(error)) throw error;
|
||||||
|
}
|
||||||
|
const geometry = new GeometryContext(cdp, tabId, graph, signal);
|
||||||
|
const issues: CaptureIssue[] = [];
|
||||||
|
const groups = new Map<string, TargetBatch<T>>();
|
||||||
|
for (const frame of graph?.frames ?? []) {
|
||||||
|
const key = cdpTargetKey(frame.target);
|
||||||
|
let group = groups.get(key);
|
||||||
|
if (!group) {
|
||||||
|
group = {
|
||||||
|
target: frame.target,
|
||||||
|
frames: [],
|
||||||
|
documents: [],
|
||||||
|
ax: [],
|
||||||
|
fallbackExcluded: new Set(),
|
||||||
|
};
|
||||||
|
groups.set(key, group);
|
||||||
|
}
|
||||||
|
group.frames.push(frame);
|
||||||
|
}
|
||||||
|
if (!groups.size)
|
||||||
|
groups.set(cdpTargetKey({ tabId }), {
|
||||||
|
target: { tabId },
|
||||||
|
frames: [],
|
||||||
|
documents: [],
|
||||||
|
ax: [],
|
||||||
|
fallbackExcluded: new Set(),
|
||||||
|
});
|
||||||
|
const batches = [...groups.values()];
|
||||||
|
const unavailable = (batch: TargetBatch<T>, stage: CaptureIssue["stage"], frameId?: string) =>
|
||||||
|
issues.push({ target: batch.target, frameId, stage, reason: "capture-unavailable" });
|
||||||
|
let firstFailure: unknown;
|
||||||
|
await collectTargets(
|
||||||
|
batches,
|
||||||
|
async (batch) => {
|
||||||
|
const scoped = cdpRunnerForTarget(cdp, batch.target);
|
||||||
|
try {
|
||||||
|
throwCaptureAborted(signal);
|
||||||
|
await scoped.send(tabId, "DOMSnapshot.enable", {});
|
||||||
|
throwCaptureAborted(signal);
|
||||||
|
const snapshot = await scoped.send<SnapshotReply>(tabId, "DOMSnapshot.captureSnapshot", {
|
||||||
|
computedStyles: REQUESTED_STYLES,
|
||||||
|
includePaintOrder: true,
|
||||||
|
includeDOMRects: true,
|
||||||
|
});
|
||||||
|
throwCaptureAborted(signal);
|
||||||
|
if (!graph) batch.frames = snapshotFrames(snapshot, batch.target);
|
||||||
|
batch.documents = await normalizeSnapshot(
|
||||||
|
snapshot,
|
||||||
|
batch.target,
|
||||||
|
batch.frames,
|
||||||
|
geometry,
|
||||||
|
issues,
|
||||||
|
signal,
|
||||||
|
);
|
||||||
|
if (
|
||||||
|
!(await enrichFormControlStates(
|
||||||
|
scoped,
|
||||||
|
tabId,
|
||||||
|
batch.documents.map((doc) => doc.domNodes),
|
||||||
|
signal,
|
||||||
|
))
|
||||||
|
)
|
||||||
|
unavailable(batch, "forms");
|
||||||
|
const present = new Set(batch.documents.map((doc) => doc.frame.frameId));
|
||||||
|
for (const frame of batch.frames)
|
||||||
|
if (!present.has(frame.frameId)) unavailable(batch, "dom", frame.frameId);
|
||||||
|
} catch (error) {
|
||||||
|
if (isCaptureAbort(error)) throw error;
|
||||||
|
firstFailure ??= error;
|
||||||
|
unavailable(batch, "dom");
|
||||||
|
batch.fallbackExcluded = await collectOverlayExcludedBackendIds(scoped, tabId, signal);
|
||||||
|
}
|
||||||
|
if (!batch.frames.length && !graph)
|
||||||
|
batch.frames = [{ frameId: "root", target: batch.target }];
|
||||||
|
try {
|
||||||
|
throwCaptureAborted(signal);
|
||||||
|
await scoped.send(tabId, "Accessibility.enable", {});
|
||||||
|
} catch (error) {
|
||||||
|
if (isCaptureAbort(error)) throw error;
|
||||||
|
firstFailure ??= error;
|
||||||
|
unavailable(batch, "ax");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
for (const frame of batch.frames) {
|
||||||
|
try {
|
||||||
|
throwCaptureAborted(signal);
|
||||||
|
const result = await scoped.send<{ nodes?: T[] }>(
|
||||||
|
tabId,
|
||||||
|
"Accessibility.getFullAXTree",
|
||||||
|
frame.frameId === "root" ? {} : { frameId: frame.frameId },
|
||||||
|
);
|
||||||
|
batch.ax.push({ frame, nodes: result.nodes ?? [] });
|
||||||
|
} catch (error) {
|
||||||
|
if (isCaptureAbort(error)) throw error;
|
||||||
|
firstFailure ??= error;
|
||||||
|
unavailable(batch, "ax", frame.frameId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
signal,
|
||||||
|
);
|
||||||
|
|
||||||
|
const frames = batches.flatMap((batch) => batch.frames);
|
||||||
|
const rootFrameId = graph?.rootFrameId ?? frames[0]?.frameId ?? "root";
|
||||||
|
const decoded = new Map(
|
||||||
|
batches.flatMap((batch) => batch.documents).map((doc) => [doc.frame.frameId, doc]),
|
||||||
|
);
|
||||||
|
const ax = batches.flatMap((batch) => batch.ax);
|
||||||
|
const documents = await buildFrameDocuments(
|
||||||
|
{ rootFrameId, frames },
|
||||||
|
ax,
|
||||||
|
{
|
||||||
|
nodes: [],
|
||||||
|
rootFrameId,
|
||||||
|
frameNodes: new Map([...decoded].map(([id, doc]) => [id, doc.domNodes])),
|
||||||
|
},
|
||||||
|
signal,
|
||||||
|
(frame) =>
|
||||||
|
issues.push({
|
||||||
|
target: frame.target,
|
||||||
|
frameId: frame.frameId,
|
||||||
|
stage: "ownership",
|
||||||
|
reason: "frame-ownership-unresolved",
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
let viewport = { width: 0, height: 0 };
|
||||||
|
try {
|
||||||
|
const measured = cssViewport(await geometry.layoutMetrics({ tabId }));
|
||||||
|
viewport = { width: measured.width, height: measured.height };
|
||||||
|
} catch (error) {
|
||||||
|
if (isCaptureAbort(error)) throw error;
|
||||||
|
}
|
||||||
|
const facts: ObservationFacts<T>["documents"][number][] = [];
|
||||||
|
for (const document of documents) {
|
||||||
|
const doc = decoded.get(document.frameId);
|
||||||
|
const index = doc?.index ?? (await buildDocumentIndex([], signal));
|
||||||
|
const fallback = groups.get(cdpTargetKey(document.target))?.fallbackExcluded;
|
||||||
|
const { domNodes, axNodes, contextScopeId: _scope, ...frame } = document;
|
||||||
|
facts.push({
|
||||||
|
frame,
|
||||||
|
index: fallback?.size ? { ...index, excludedBackendNodeIds: fallback } : index,
|
||||||
|
domNodes,
|
||||||
|
axNodes,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
throwCaptureAborted(signal);
|
||||||
|
if (firstFailure && facts.every((doc) => !doc.domNodes.length && !doc.axNodes.length))
|
||||||
|
throw firstFailure;
|
||||||
|
return { rootFrameId, viewport, documents: facts, issues, startedAt, finishedAt: Date.now() };
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Existing semantic consumers get a narrow view; no raw snapshot
|
||||||
|
* data can leak through recording's allowlisted output DTO. */
|
||||||
|
export function semanticCapture<T extends FrameOwnedAxNode>(
|
||||||
|
facts: ObservationFacts<T>,
|
||||||
|
): { captured: CapturedSceneInput; documents: FrameDocument<T>[] } {
|
||||||
|
const documents = facts.documents.map((doc) => ({
|
||||||
|
...doc.frame,
|
||||||
|
contextScopeId: doc.frame.frameId,
|
||||||
|
domNodes: doc.domNodes.filter(
|
||||||
|
(node) => !doc.index.excludedBackendNodeIds.has(node.backendNodeId),
|
||||||
|
),
|
||||||
|
axNodes: doc.axNodes,
|
||||||
|
excludedBackendNodeIds: doc.index.excludedBackendNodeIds,
|
||||||
|
}));
|
||||||
|
const root = documents.find((doc) => doc.frameId === facts.rootFrameId);
|
||||||
|
return {
|
||||||
|
documents,
|
||||||
|
captured: {
|
||||||
|
nodes: root?.domNodes ?? [],
|
||||||
|
viewport: facts.viewport,
|
||||||
|
rootFrameId: facts.rootFrameId,
|
||||||
|
excludedBackendNodeIds: root?.excludedBackendNodeIds ?? new Set(),
|
||||||
|
},
|
||||||
|
};
|
||||||
|
}
|
||||||
@@ -1,333 +1,21 @@
|
|||||||
// CDP capture adapter: DOMSnapshot.captureSnapshot + Page.getLayoutMetrics
|
// Browser-side hover probes and overlay exclusion fallback. Static collection
|
||||||
// → CapturedNode[] + Viewport. This is the ONLY VOM module that touches
|
// lives in capture-coordinator; snapshot decoding has no browser dependency.
|
||||||
// raw CDP. The captureSnapshot reply is columnar (parallel arrays + a
|
|
||||||
// shared string table); requested computed styles are decoded through
|
|
||||||
// STYLE_COL so capture and visibility policy share one explicit contract.
|
|
||||||
|
|
||||||
import type { Rect, Viewport } from "@browser-skill/vom";
|
|
||||||
import type { CdpTarget } from "@/browser-driver/frame-graph";
|
|
||||||
import { evaluateHoverTrigger } from "@/lib/hover-trigger-policy";
|
import { evaluateHoverTrigger } from "@/lib/hover-trigger-policy";
|
||||||
import { isOverlayHostNode, OVERLAY_HOST_SELECTOR } from "../../lib/overlay-bridge";
|
import { OVERLAY_HOST_SELECTOR } from "../../lib/overlay-bridge";
|
||||||
import {
|
|
||||||
type FrameProjectionIssue,
|
|
||||||
type FrameProjectionState,
|
|
||||||
projectSnapshotRect,
|
|
||||||
snapshotViewportRect,
|
|
||||||
} from "../geometry/coordinate-types";
|
|
||||||
import { cssViewport, GeometryContext, type LayoutMetrics } from "../geometry/frame-context";
|
|
||||||
import type { CdpRunner } from "../shared";
|
import type { CdpRunner } from "../shared";
|
||||||
|
import type { CapturedNode, CapturedSurfaceProbe } from "./facts";
|
||||||
|
|
||||||
|
export type {
|
||||||
|
CapturedNode,
|
||||||
|
CapturedSurfaceProbe,
|
||||||
|
} from "./facts";
|
||||||
|
|
||||||
|
import { isCaptureAbort as isAbortError, throwCaptureAborted as throwIfAborted } from "./facts";
|
||||||
import { clearHover, ProbeBudget, waitForHover } from "./hover-perception";
|
import { clearHover, ProbeBudget, waitForHover } from "./hover-perception";
|
||||||
|
|
||||||
const REQUESTED_STYLES = ["position", "pointer-events", "cursor", "visibility", "opacity"] as const;
|
|
||||||
const STYLE_COL = Object.fromEntries(
|
|
||||||
REQUESTED_STYLES.map((name, index) => [name, index]),
|
|
||||||
) as Record<(typeof REQUESTED_STYLES)[number], number>;
|
|
||||||
|
|
||||||
export interface CapturedNode {
|
|
||||||
backendNodeId: number;
|
|
||||||
parentBackendNodeId: number | null;
|
|
||||||
frameId?: string;
|
|
||||||
/** Owning iframe backend node id; `null` for the top-level document. */
|
|
||||||
ownerFrameBackendNodeId?: number | null;
|
|
||||||
tag: string;
|
|
||||||
attrs: Record<string, string>;
|
|
||||||
/** Top-level viewport-relative CSS px, clipped to the owning frame viewport. */
|
|
||||||
rect: Rect | null;
|
|
||||||
/** Frame-local viewport-relative CSS px before top-level projection. */
|
|
||||||
localRect?: Rect | null;
|
|
||||||
paintOrder: number;
|
|
||||||
position: string;
|
|
||||||
pointerEvents: string;
|
|
||||||
/**
|
|
||||||
* computed `cursor`. `cursor: pointer` is the strongest CDP-free signal
|
|
||||||
* that a non-semantic element (a `<div>`/`<span>` with a click handler)
|
|
||||||
* is actually an interactive control — used by the adapter to surface
|
|
||||||
* custom buttons/checkboxes the AX tree drops as `generic`. Optional like
|
|
||||||
* `textContent`: the live parser always sets it, hand-built fixtures may not.
|
|
||||||
*/
|
|
||||||
cursor?: string;
|
|
||||||
/**
|
|
||||||
* Whether the live DOM snapshot provides a painted, non-hidden box for this
|
|
||||||
* node. Semantic resolution uses this only for DOM fallback nodes; AX-backed
|
|
||||||
* nodes remain authoritative even when they are outside the viewport.
|
|
||||||
*/
|
|
||||||
rendered?: boolean;
|
|
||||||
textContent?: string;
|
|
||||||
formState?: "empty" | "filled" | "default";
|
|
||||||
formValue?: string;
|
|
||||||
formDefaultValue?: string;
|
|
||||||
formPlaceholder?: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
export type CapturedIframeNodes = Map<number, CapturedNode[]>;
|
|
||||||
|
|
||||||
export interface CapturedSurfaceProbe {
|
|
||||||
triggerBackendNodeId: number;
|
|
||||||
triggerPoint?: { x: number; y: number };
|
|
||||||
triggerAction: "hover" | "focus" | string;
|
|
||||||
subItems: string[];
|
|
||||||
confidence?: "high" | "medium" | "low";
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface CapturedViewModel {
|
|
||||||
nodes: CapturedNode[];
|
|
||||||
viewport: Viewport;
|
|
||||||
iframeNodes: CapturedIframeNodes;
|
|
||||||
frameNodes?: Map<string, CapturedNode[]>;
|
|
||||||
frameOwnerBackendNodeIds?: Map<string, number>;
|
|
||||||
/** Explicit DOMSnapshot frame ancestry; never inferred from backend node ids. */
|
|
||||||
frameParentIds?: Map<string, string>;
|
|
||||||
rootFrameId?: string;
|
|
||||||
/** Failed boundaries and descendants blocked by them; clipping is not a failure. */
|
|
||||||
frameGeometryIssues?: FrameProjectionIssue[];
|
|
||||||
/** Backend node ids belonging to the agent overlay host + its shadow subtree. */
|
|
||||||
excludedBackendNodeIds: Set<number>;
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface CaptureViewModelOptions {
|
|
||||||
signal?: AbortSignal;
|
|
||||||
geometry?: GeometryContext;
|
|
||||||
target?: CdpTarget;
|
|
||||||
}
|
|
||||||
|
|
||||||
function captureAbortError(): Error {
|
|
||||||
const error = new Error("observation aborted");
|
|
||||||
error.name = "AbortError";
|
|
||||||
return error;
|
|
||||||
}
|
|
||||||
|
|
||||||
function isAbortError(error: unknown): boolean {
|
|
||||||
return (
|
|
||||||
typeof error === "object" &&
|
|
||||||
error !== null &&
|
|
||||||
(error as { name?: string }).name === "AbortError"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
function throwIfAborted(signal: AbortSignal | undefined): void {
|
|
||||||
if (signal?.aborted) throw captureAbortError();
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Sparse array format Chrome uses for infrequently-set per-node fields. */
|
|
||||||
interface SparseArray {
|
|
||||||
index: number[];
|
|
||||||
value: number[];
|
|
||||||
}
|
|
||||||
|
|
||||||
interface RareBooleanData {
|
|
||||||
index: number[];
|
|
||||||
}
|
|
||||||
|
|
||||||
interface SnapshotDocument {
|
|
||||||
frameId?: string | number;
|
|
||||||
scrollOffsetX?: number;
|
|
||||||
scrollOffsetY?: number;
|
|
||||||
nodes?: {
|
|
||||||
parentIndex?: number[];
|
|
||||||
nodeName?: number[];
|
|
||||||
backendNodeId?: number[];
|
|
||||||
attributes?: number[][];
|
|
||||||
/**
|
|
||||||
* Per-node text value (index into strings), set for `#text` / CDATA nodes.
|
|
||||||
* Element nodes carry -1. Same length as `backendNodeId`.
|
|
||||||
*/
|
|
||||||
nodeValue?: number[];
|
|
||||||
/** Maps node array index → index into `documents[]` for frame content. */
|
|
||||||
contentDocumentIndex?: SparseArray;
|
|
||||||
inputValue?: SparseArray;
|
|
||||||
textValue?: SparseArray;
|
|
||||||
inputChecked?: RareBooleanData;
|
|
||||||
optionSelected?: RareBooleanData;
|
|
||||||
};
|
|
||||||
layout?: {
|
|
||||||
nodeIndex?: number[];
|
|
||||||
styles?: number[][];
|
|
||||||
bounds?: number[][];
|
|
||||||
paintOrders?: number[];
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
function snapshotFrameId(document: SnapshotDocument, strings: string[]): string | undefined {
|
|
||||||
if (typeof document.frameId === "string") return document.frameId || undefined;
|
|
||||||
if (typeof document.frameId === "number") return str(strings, document.frameId) || undefined;
|
|
||||||
return undefined;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface SnapshotReply {
|
|
||||||
strings?: string[];
|
|
||||||
documents?: SnapshotDocument[];
|
|
||||||
}
|
|
||||||
|
|
||||||
function isFormControlTag(tag: string): boolean {
|
|
||||||
return tag === "input" || tag === "textarea" || tag === "select";
|
|
||||||
}
|
|
||||||
|
|
||||||
function isSensitiveFormControl(node: Pick<CapturedNode, "tag" | "attrs">): boolean {
|
|
||||||
return node.tag === "input" && (node.attrs.type ?? "").toLowerCase() === "password";
|
|
||||||
}
|
|
||||||
|
|
||||||
function snapshotFormState(
|
|
||||||
value: string | undefined,
|
|
||||||
defaultValue: string,
|
|
||||||
hasDefaultValue: boolean,
|
|
||||||
sensitive: boolean,
|
|
||||||
): CapturedNode["formState"] {
|
|
||||||
if (sensitive) {
|
|
||||||
if (value === undefined && !hasDefaultValue) return undefined;
|
|
||||||
return (value ?? defaultValue) === "" ? "empty" : "filled";
|
|
||||||
}
|
|
||||||
if (value === undefined) return undefined;
|
|
||||||
if (value === "") return "empty";
|
|
||||||
return value === defaultValue ? "default" : "filled";
|
|
||||||
}
|
|
||||||
|
|
||||||
const MAX_FORM_ENRICH_CONTROLS = 250;
|
|
||||||
|
|
||||||
interface CapturedFormState {
|
|
||||||
value?: string;
|
|
||||||
defaultValue?: string;
|
|
||||||
placeholder?: string;
|
|
||||||
state?: "empty" | "filled" | "default";
|
|
||||||
sensitive?: boolean;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface DeepSerializedValue {
|
|
||||||
type: string;
|
|
||||||
value?: unknown;
|
|
||||||
}
|
|
||||||
|
|
||||||
function formStateBatchExpression(maxControls: number): string {
|
|
||||||
return `(() => {
|
|
||||||
const maxControls = ${JSON.stringify(maxControls)};
|
|
||||||
let remaining = maxControls;
|
|
||||||
const controlSelector = "input,textarea,select";
|
|
||||||
const controlState = (el) => {
|
|
||||||
const tag = el.tagName.toLowerCase();
|
|
||||||
const type = tag === "input" ? String(el.type || "text").toLowerCase() : tag;
|
|
||||||
const sensitive = type === "password" || type === "credit-card";
|
|
||||||
const rawValue = typeof el.value === "string" ? el.value : "";
|
|
||||||
const defaultValue = typeof el.defaultValue === "string" ? el.defaultValue : "";
|
|
||||||
const placeholder = typeof el.placeholder === "string" ? el.placeholder : "";
|
|
||||||
const state = rawValue === "" ? "empty" : rawValue === defaultValue ? "default" : "filled";
|
|
||||||
return {
|
|
||||||
state,
|
|
||||||
sensitive,
|
|
||||||
placeholder,
|
|
||||||
...(sensitive ? {} : { value: rawValue, defaultValue }),
|
|
||||||
};
|
|
||||||
};
|
|
||||||
const controls = [];
|
|
||||||
const collect = (doc) => {
|
|
||||||
for (const el of Array.from(doc.querySelectorAll(controlSelector))) {
|
|
||||||
if (remaining <= 0) break;
|
|
||||||
// Deep serialization supplies the node's backend id. Keep the
|
|
||||||
// state as JSON so decoding needs no general-purpose V8 deserializer.
|
|
||||||
controls.push([el, JSON.stringify(controlState(el))]);
|
|
||||||
remaining -= 1;
|
|
||||||
}
|
|
||||||
for (const frame of Array.from(doc.querySelectorAll("iframe"))) {
|
|
||||||
if (remaining <= 0) break;
|
|
||||||
let childDoc = null;
|
|
||||||
try { childDoc = frame.contentDocument; } catch { childDoc = null; }
|
|
||||||
if (childDoc) collect(childDoc);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
collect(document);
|
|
||||||
return controls;
|
|
||||||
})()`;
|
|
||||||
}
|
|
||||||
|
|
||||||
function formStatesByBackendId(result: RuntimeEvaluateReply): Map<number, CapturedFormState> {
|
|
||||||
const states = new Map<number, CapturedFormState>();
|
|
||||||
const serialized = result.result?.deepSerializedValue;
|
|
||||||
if (serialized?.type !== "array" || !Array.isArray(serialized.value)) return states;
|
|
||||||
for (const entry of serialized.value as DeepSerializedValue[]) {
|
|
||||||
if (entry?.type !== "array" || !Array.isArray(entry.value)) continue;
|
|
||||||
const [element, json] = entry.value as DeepSerializedValue[];
|
|
||||||
if (element?.type !== "node" || json?.type !== "string" || typeof json.value !== "string") {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
const backendNodeId = (element.value as { backendNodeId?: number } | undefined)?.backendNodeId;
|
|
||||||
if (typeof backendNodeId !== "number" || !Number.isSafeInteger(backendNodeId)) continue;
|
|
||||||
try {
|
|
||||||
const state = JSON.parse(json.value) as CapturedFormState | null;
|
|
||||||
if (
|
|
||||||
!state ||
|
|
||||||
!["empty", "filled", "default"].includes(state.state ?? "") ||
|
|
||||||
typeof state.sensitive !== "boolean" ||
|
|
||||||
typeof state.placeholder !== "string" ||
|
|
||||||
(!state.sensitive &&
|
|
||||||
(typeof state.value !== "string" || typeof state.defaultValue !== "string"))
|
|
||||||
) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
states.set(backendNodeId, state);
|
|
||||||
} catch {
|
|
||||||
// A malformed entry must not overwrite the snapshot's own state.
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return states;
|
|
||||||
}
|
|
||||||
|
|
||||||
function applyFormStates(nodes: CapturedNode[], states: Map<number, CapturedFormState>): void {
|
|
||||||
for (const node of nodes) {
|
|
||||||
if (!isFormControlTag(node.tag)) continue;
|
|
||||||
const state = states.get(node.backendNodeId);
|
|
||||||
if (!state) continue;
|
|
||||||
node.formState = state.state;
|
|
||||||
node.formPlaceholder = state.placeholder ?? "";
|
|
||||||
if (state.sensitive || isSensitiveFormControl(node)) {
|
|
||||||
delete node.formValue;
|
|
||||||
delete node.formDefaultValue;
|
|
||||||
delete node.attrs.value;
|
|
||||||
} else {
|
|
||||||
node.formDefaultValue = state.defaultValue ?? "";
|
|
||||||
if (state.value !== undefined) node.formValue = state.value;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function enrichFormControlStates(
|
|
||||||
cdp: CdpRunner,
|
|
||||||
tabId: number,
|
|
||||||
frameNodeGroups: CapturedNode[][],
|
|
||||||
signal?: AbortSignal,
|
|
||||||
): Promise<void> {
|
|
||||||
const hasControls = frameNodeGroups.some((nodes) =>
|
|
||||||
nodes.some((node) => isFormControlTag(node.tag)),
|
|
||||||
);
|
|
||||||
if (!hasControls) return;
|
|
||||||
const objectGroup = `bsk-vom-forms-${crypto.randomUUID()}`;
|
|
||||||
try {
|
|
||||||
throwIfAborted(signal);
|
|
||||||
const result = await cdp.send<RuntimeEvaluateReply>(tabId, "Runtime.evaluate", {
|
|
||||||
expression: formStateBatchExpression(MAX_FORM_ENRICH_CONTROLS),
|
|
||||||
objectGroup,
|
|
||||||
serializationOptions: {
|
|
||||||
serialization: "deep",
|
|
||||||
maxDepth: 3,
|
|
||||||
additionalParameters: { maxNodeDepth: 0, includeShadowTree: "none" },
|
|
||||||
},
|
|
||||||
});
|
|
||||||
throwIfAborted(signal);
|
|
||||||
// Backend ids are scoped to this capture's CDP target. OOPIF targets
|
|
||||||
// use their own captureViewModel call and never share this lookup.
|
|
||||||
const states = formStatesByBackendId(result);
|
|
||||||
for (const nodes of frameNodeGroups) {
|
|
||||||
applyFormStates(nodes, states);
|
|
||||||
}
|
|
||||||
} catch (error) {
|
|
||||||
if (isAbortError(error)) throw error;
|
|
||||||
// Best-effort enrichment. DOMSnapshot/AX data still carries the nodes.
|
|
||||||
} finally {
|
|
||||||
await cdp.send(tabId, "Runtime.releaseObjectGroup", { objectGroup }).catch(() => undefined);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
interface RuntimeEvaluateReply {
|
interface RuntimeEvaluateReply {
|
||||||
result?: {
|
result?: {
|
||||||
value?: unknown;
|
value?: unknown;
|
||||||
deepSerializedValue?: DeepSerializedValue;
|
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -346,36 +34,6 @@ interface CdpDomNode {
|
|||||||
shadowRoots?: CdpDomNode[];
|
shadowRoots?: CdpDomNode[];
|
||||||
}
|
}
|
||||||
|
|
||||||
interface ParseDocumentResult {
|
|
||||||
nodes: CapturedNode[];
|
|
||||||
excludedBackendNodeIds: Set<number>;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface FrameContext {
|
|
||||||
frameId?: string;
|
|
||||||
ownerFrameBackendNodeId: number | null;
|
|
||||||
projection: FrameProjectionState;
|
|
||||||
target: CdpTarget;
|
|
||||||
scrollX: number;
|
|
||||||
scrollY: number;
|
|
||||||
}
|
|
||||||
|
|
||||||
function str(strings: string[], idx: number | undefined): string {
|
|
||||||
if (idx === undefined || idx < 0) return "";
|
|
||||||
return strings[idx] ?? "";
|
|
||||||
}
|
|
||||||
|
|
||||||
function sparseIndexMap(sparse: SparseArray | undefined): Map<number, number> {
|
|
||||||
const out = new Map<number, number>();
|
|
||||||
if (!sparse?.index || !sparse.value) return out;
|
|
||||||
for (let i = 0; i < sparse.index.length; i++) {
|
|
||||||
const nodeIndex = sparse.index[i];
|
|
||||||
const docIndex = sparse.value[i];
|
|
||||||
if (nodeIndex !== undefined && docIndex !== undefined) out.set(nodeIndex, docIndex);
|
|
||||||
}
|
|
||||||
return out;
|
|
||||||
}
|
|
||||||
|
|
||||||
function collectBackendIdsFromDomNode(node: CdpDomNode | undefined, out: Set<number>): void {
|
function collectBackendIdsFromDomNode(node: CdpDomNode | undefined, out: Set<number>): void {
|
||||||
if (!node) return;
|
if (!node) return;
|
||||||
if (typeof node.backendNodeId === "number") {
|
if (typeof node.backendNodeId === "number") {
|
||||||
|
|||||||
@@ -0,0 +1,210 @@
|
|||||||
|
import type { Rect, Viewport } from "@browser-skill/vom";
|
||||||
|
import type { CdpFrame, CdpTarget } from "@/browser-driver/frame-graph";
|
||||||
|
import { isOverlayHostNode } from "@/lib/overlay-bridge";
|
||||||
|
import type { FrameProjectionIssue } from "../geometry/coordinate-types";
|
||||||
|
import type { FrameOwnedAxNode } from "./frame-document";
|
||||||
|
|
||||||
|
export interface CapturedNode {
|
||||||
|
backendNodeId: number;
|
||||||
|
parentBackendNodeId: number | null;
|
||||||
|
frameId?: string;
|
||||||
|
/** Owning iframe backend node id; `null` for the top-level document. */
|
||||||
|
ownerFrameBackendNodeId?: number | null;
|
||||||
|
tag: string;
|
||||||
|
attrs: Record<string, string>;
|
||||||
|
/** Top-level viewport-relative CSS px, clipped to the owning frame viewport. */
|
||||||
|
rect: Rect | null;
|
||||||
|
/** Frame-local viewport-relative CSS px before top-level projection. */
|
||||||
|
localRect?: Rect | null;
|
||||||
|
paintOrder: number;
|
||||||
|
position: string;
|
||||||
|
pointerEvents: string;
|
||||||
|
/**
|
||||||
|
* computed `cursor`. `cursor: pointer` is the strongest CDP-free signal
|
||||||
|
* that a non-semantic element (a `<div>`/`<span>` with a click handler)
|
||||||
|
* is actually an interactive control — used by the adapter to surface
|
||||||
|
* custom buttons/checkboxes the AX tree drops as `generic`. Optional like
|
||||||
|
* `textContent`: the live parser always sets it, hand-built fixtures may not.
|
||||||
|
*/
|
||||||
|
cursor?: string;
|
||||||
|
/**
|
||||||
|
* Whether the live DOM snapshot provides a painted, non-hidden box for this
|
||||||
|
* node. Semantic resolution uses this only for DOM fallback nodes; AX-backed
|
||||||
|
* nodes remain authoritative even when they are outside the viewport.
|
||||||
|
*/
|
||||||
|
rendered?: boolean;
|
||||||
|
textContent?: string;
|
||||||
|
formState?: "empty" | "filled" | "default";
|
||||||
|
formValue?: string;
|
||||||
|
formDefaultValue?: string;
|
||||||
|
formPlaceholder?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface CapturedSurfaceProbe {
|
||||||
|
triggerBackendNodeId: number;
|
||||||
|
triggerPoint?: { x: number; y: number };
|
||||||
|
triggerAction: "hover" | "focus" | string;
|
||||||
|
subItems: string[];
|
||||||
|
confidence?: "high" | "medium" | "low";
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Narrow input shared by the existing semantic scene/hover consumers. */
|
||||||
|
export interface CapturedSceneInput {
|
||||||
|
nodes: CapturedNode[];
|
||||||
|
viewport: Viewport;
|
||||||
|
rootFrameId?: string;
|
||||||
|
excludedBackendNodeIds: ReadonlySet<number>;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Snapshot client/offset rectangles retain their protocol units. In particular,
|
||||||
|
* transformed bounds cannot be combined with unscaled client offsets. */
|
||||||
|
export interface SnapshotLayout {
|
||||||
|
readonly boundsSpace: "snapshot-document-css";
|
||||||
|
readonly clientSpace: "unscaled-client-offset-css";
|
||||||
|
bounds?: number[];
|
||||||
|
clientRect?: number[];
|
||||||
|
offsetRect?: number[];
|
||||||
|
scrollRect?: number[];
|
||||||
|
styles: Readonly<Record<string, string>>;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface DecodedNode extends Omit<CapturedNode, "rect" | "localRect" | "rendered"> {
|
||||||
|
nodeType?: number;
|
||||||
|
parentMissing?: boolean;
|
||||||
|
layout?: SnapshotLayout;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface AncestorState {
|
||||||
|
complete: boolean;
|
||||||
|
overlay: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface NodeFacts extends CapturedNode {
|
||||||
|
nodeType?: number;
|
||||||
|
layout?: SnapshotLayout;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface DocumentIndex<T extends DecodedNode = NodeFacts> {
|
||||||
|
readonly nodes: ReadonlyMap<number, T>;
|
||||||
|
readonly children: ReadonlyMap<number, readonly number[]>;
|
||||||
|
readonly ancestry: ReadonlyMap<number, AncestorState>;
|
||||||
|
readonly excludedBackendNodeIds: ReadonlySet<number>;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface DecodedDocument {
|
||||||
|
nodes: DecodedNode[];
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface CaptureIssue {
|
||||||
|
/** Owner failure or inherited blockage; absent for other geometry failures. */
|
||||||
|
projectionIssue?: FrameProjectionIssue;
|
||||||
|
target: CdpTarget;
|
||||||
|
frameId?: string;
|
||||||
|
stage: "dom" | "ax" | "ownership" | "geometry" | "forms";
|
||||||
|
reason: "capture-unavailable" | "frame-ownership-unresolved" | "geometry-unavailable";
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface DocumentFacts<T extends FrameOwnedAxNode> {
|
||||||
|
readonly frame: CdpFrame;
|
||||||
|
readonly index: DocumentIndex;
|
||||||
|
readonly domNodes: CapturedNode[];
|
||||||
|
readonly axNodes: T[];
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface ObservationFacts<T extends FrameOwnedAxNode> {
|
||||||
|
readonly rootFrameId: string;
|
||||||
|
readonly viewport: Viewport;
|
||||||
|
readonly documents: readonly DocumentFacts<T>[];
|
||||||
|
readonly issues: readonly CaptureIssue[];
|
||||||
|
readonly startedAt: number;
|
||||||
|
readonly finishedAt: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function throwCaptureAborted(signal?: AbortSignal): void {
|
||||||
|
if (signal?.aborted) throw new DOMException("observation aborted", "AbortError");
|
||||||
|
}
|
||||||
|
|
||||||
|
export function isCaptureAbort(error: unknown): boolean {
|
||||||
|
return error instanceof Error && error.name === "AbortError";
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Called only at bounded block boundaries, not once per node. */
|
||||||
|
export async function captureCheckpoint(signal?: AbortSignal): Promise<void> {
|
||||||
|
throwCaptureAborted(signal);
|
||||||
|
await new Promise<void>((resolve) => {
|
||||||
|
const channel = new MessageChannel();
|
||||||
|
channel.port1.onmessage = () => {
|
||||||
|
channel.port1.close();
|
||||||
|
channel.port2.close();
|
||||||
|
resolve();
|
||||||
|
};
|
||||||
|
channel.port2.postMessage(null);
|
||||||
|
});
|
||||||
|
throwCaptureAborted(signal);
|
||||||
|
}
|
||||||
|
|
||||||
|
/** All snapshot nodes participate, including document and shadow roots. This
|
||||||
|
* preserves ancestry evidence even when the semantic adapter omits those nodes. */
|
||||||
|
export async function buildDocumentIndex<T extends DecodedNode>(
|
||||||
|
input: readonly T[],
|
||||||
|
signal?: AbortSignal,
|
||||||
|
): Promise<DocumentIndex<T>> {
|
||||||
|
const nodes = new Map<number, T>();
|
||||||
|
const children = new Map<number, number[]>();
|
||||||
|
const ancestry = new Map<number, AncestorState>();
|
||||||
|
const excludedBackendNodeIds = new Set<number>();
|
||||||
|
for (let i = 0; i < input.length; i++) {
|
||||||
|
if (i % 256 === 0) await captureCheckpoint(signal);
|
||||||
|
const node = input[i];
|
||||||
|
nodes.set(node.backendNodeId, node);
|
||||||
|
if (node.parentBackendNodeId !== null) {
|
||||||
|
const siblings = children.get(node.parentBackendNodeId);
|
||||||
|
if (siblings) siblings.push(node.backendNodeId);
|
||||||
|
else children.set(node.parentBackendNodeId, [node.backendNodeId]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let work = 0;
|
||||||
|
for (const node of input) {
|
||||||
|
if (work++ % 256 === 0) await captureCheckpoint(signal);
|
||||||
|
if (ancestry.has(node.backendNodeId)) continue;
|
||||||
|
const path: DecodedNode[] = [];
|
||||||
|
const visiting = new Set<number>();
|
||||||
|
let current: DecodedNode | undefined = node;
|
||||||
|
let state: AncestorState = { complete: true, overlay: false };
|
||||||
|
while (current && !ancestry.has(current.backendNodeId)) {
|
||||||
|
if (work++ % 256 === 0) await captureCheckpoint(signal);
|
||||||
|
if (visiting.has(current.backendNodeId)) {
|
||||||
|
// No member of a malformed cycle supplies complete ancestry evidence.
|
||||||
|
state = {
|
||||||
|
complete: false,
|
||||||
|
overlay: path
|
||||||
|
.slice(path.findIndex((item) => item.backendNodeId === current!.backendNodeId))
|
||||||
|
.some((item) => isOverlayHostNode(item.tag, Object.keys(item.attrs))),
|
||||||
|
};
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
visiting.add(current.backendNodeId);
|
||||||
|
path.push(current);
|
||||||
|
if (current.parentMissing) state = { complete: false, overlay: false };
|
||||||
|
if (current.parentBackendNodeId === null) {
|
||||||
|
current = undefined;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
current = nodes.get(current.parentBackendNodeId);
|
||||||
|
if (!current) state = { complete: false, overlay: false };
|
||||||
|
}
|
||||||
|
if (current && ancestry.has(current.backendNodeId))
|
||||||
|
state = ancestry.get(current.backendNodeId)!;
|
||||||
|
for (let i = path.length - 1; i >= 0; i--) {
|
||||||
|
if (work++ % 256 === 0) await captureCheckpoint(signal);
|
||||||
|
const item = path[i];
|
||||||
|
state = {
|
||||||
|
complete: state.complete,
|
||||||
|
overlay: state.overlay || isOverlayHostNode(item.tag, Object.keys(item.attrs)),
|
||||||
|
};
|
||||||
|
ancestry.set(item.backendNodeId, state);
|
||||||
|
if (state.overlay) excludedBackendNodeIds.add(item.backendNodeId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return { nodes, children, ancestry, excludedBackendNodeIds };
|
||||||
|
}
|
||||||
@@ -0,0 +1,154 @@
|
|||||||
|
import type { CdpRunner } from "../shared";
|
||||||
|
import type { CapturedNode } from "./facts";
|
||||||
|
import { isCaptureAbort as isAbortError, throwCaptureAborted as throwIfAborted } from "./facts";
|
||||||
|
|
||||||
|
interface RuntimeEvaluateReply {
|
||||||
|
result?: { deepSerializedValue?: DeepSerializedValue };
|
||||||
|
}
|
||||||
|
const MAX_FORM_ENRICH_CONTROLS = 250;
|
||||||
|
|
||||||
|
interface CapturedFormState {
|
||||||
|
value?: string;
|
||||||
|
defaultValue?: string;
|
||||||
|
placeholder?: string;
|
||||||
|
state?: "empty" | "filled" | "default";
|
||||||
|
sensitive?: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface DeepSerializedValue {
|
||||||
|
type: string;
|
||||||
|
value?: unknown;
|
||||||
|
}
|
||||||
|
|
||||||
|
function formStateBatchExpression(maxControls: number): string {
|
||||||
|
return `(() => {
|
||||||
|
const maxControls = ${JSON.stringify(maxControls)};
|
||||||
|
let remaining = maxControls;
|
||||||
|
const controlSelector = "input,textarea,select";
|
||||||
|
const controlState = (el) => {
|
||||||
|
const tag = el.tagName.toLowerCase();
|
||||||
|
const type = tag === "input" ? String(el.type || "text").toLowerCase() : tag;
|
||||||
|
const sensitive = type === "password" || type === "credit-card";
|
||||||
|
const rawValue = typeof el.value === "string" ? el.value : "";
|
||||||
|
const defaultValue = typeof el.defaultValue === "string" ? el.defaultValue : "";
|
||||||
|
const placeholder = typeof el.placeholder === "string" ? el.placeholder : "";
|
||||||
|
const state = rawValue === "" ? "empty" : rawValue === defaultValue ? "default" : "filled";
|
||||||
|
return {
|
||||||
|
state,
|
||||||
|
sensitive,
|
||||||
|
placeholder,
|
||||||
|
...(sensitive ? {} : { value: rawValue, defaultValue }),
|
||||||
|
};
|
||||||
|
};
|
||||||
|
const controls = [];
|
||||||
|
const collect = (doc) => {
|
||||||
|
for (const el of Array.from(doc.querySelectorAll(controlSelector))) {
|
||||||
|
if (remaining <= 0) break;
|
||||||
|
// Deep serialization supplies the node's backend id. Keep the
|
||||||
|
// state as JSON so decoding needs no general-purpose V8 deserializer.
|
||||||
|
controls.push([el, JSON.stringify(controlState(el))]);
|
||||||
|
remaining -= 1;
|
||||||
|
}
|
||||||
|
for (const frame of Array.from(doc.querySelectorAll("iframe"))) {
|
||||||
|
if (remaining <= 0) break;
|
||||||
|
let childDoc = null;
|
||||||
|
try { childDoc = frame.contentDocument; } catch { childDoc = null; }
|
||||||
|
if (childDoc) collect(childDoc);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
collect(document);
|
||||||
|
return controls;
|
||||||
|
})()`;
|
||||||
|
}
|
||||||
|
|
||||||
|
function formStatesByBackendId(result: RuntimeEvaluateReply): Map<number, CapturedFormState> {
|
||||||
|
const states = new Map<number, CapturedFormState>();
|
||||||
|
const serialized = result.result?.deepSerializedValue;
|
||||||
|
if (serialized?.type !== "array" || !Array.isArray(serialized.value)) return states;
|
||||||
|
for (const entry of serialized.value as DeepSerializedValue[]) {
|
||||||
|
if (entry?.type !== "array" || !Array.isArray(entry.value)) continue;
|
||||||
|
const [element, json] = entry.value as DeepSerializedValue[];
|
||||||
|
if (element?.type !== "node" || json?.type !== "string" || typeof json.value !== "string") {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
const backendNodeId = (element.value as { backendNodeId?: number } | undefined)?.backendNodeId;
|
||||||
|
if (typeof backendNodeId !== "number" || !Number.isSafeInteger(backendNodeId)) continue;
|
||||||
|
try {
|
||||||
|
const state = JSON.parse(json.value) as CapturedFormState | null;
|
||||||
|
if (
|
||||||
|
!state ||
|
||||||
|
!["empty", "filled", "default"].includes(state.state ?? "") ||
|
||||||
|
typeof state.sensitive !== "boolean" ||
|
||||||
|
typeof state.placeholder !== "string" ||
|
||||||
|
(!state.sensitive &&
|
||||||
|
(typeof state.value !== "string" || typeof state.defaultValue !== "string"))
|
||||||
|
) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
states.set(backendNodeId, state);
|
||||||
|
} catch {
|
||||||
|
// A malformed entry must not overwrite the snapshot's own state.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return states;
|
||||||
|
}
|
||||||
|
|
||||||
|
function applyFormStates(nodes: CapturedNode[], states: Map<number, CapturedFormState>): void {
|
||||||
|
for (const node of nodes) {
|
||||||
|
if (!["input", "textarea", "select"].includes(node.tag)) continue;
|
||||||
|
const state = states.get(node.backendNodeId);
|
||||||
|
if (!state) continue;
|
||||||
|
node.formState = state.state;
|
||||||
|
node.formPlaceholder = state.placeholder ?? "";
|
||||||
|
if (
|
||||||
|
state.sensitive ||
|
||||||
|
(node.tag === "input" && (node.attrs.type ?? "").toLowerCase() === "password")
|
||||||
|
) {
|
||||||
|
delete node.formValue;
|
||||||
|
delete node.formDefaultValue;
|
||||||
|
delete node.attrs.value;
|
||||||
|
} else {
|
||||||
|
node.formDefaultValue = state.defaultValue ?? "";
|
||||||
|
if (state.value !== undefined) node.formValue = state.value;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function enrichFormControlStates(
|
||||||
|
cdp: CdpRunner,
|
||||||
|
tabId: number,
|
||||||
|
frameNodeGroups: CapturedNode[][],
|
||||||
|
signal?: AbortSignal,
|
||||||
|
): Promise<boolean> {
|
||||||
|
const hasControls = frameNodeGroups.some((nodes) =>
|
||||||
|
nodes.some((node) => ["input", "textarea", "select"].includes(node.tag)),
|
||||||
|
);
|
||||||
|
if (!hasControls) return true;
|
||||||
|
const objectGroup = `bsk-vom-forms-${crypto.randomUUID()}`;
|
||||||
|
try {
|
||||||
|
throwIfAborted(signal);
|
||||||
|
const result = await cdp.send<RuntimeEvaluateReply>(tabId, "Runtime.evaluate", {
|
||||||
|
expression: formStateBatchExpression(MAX_FORM_ENRICH_CONTROLS),
|
||||||
|
objectGroup,
|
||||||
|
serializationOptions: {
|
||||||
|
serialization: "deep",
|
||||||
|
maxDepth: 3,
|
||||||
|
additionalParameters: { maxNodeDepth: 0, includeShadowTree: "none" },
|
||||||
|
},
|
||||||
|
});
|
||||||
|
throwIfAborted(signal);
|
||||||
|
// Backend ids are scoped to this capture's CDP target. OOPIF targets
|
||||||
|
// have separate enrichment calls and never share this lookup.
|
||||||
|
const states = formStatesByBackendId(result);
|
||||||
|
for (const nodes of frameNodeGroups) {
|
||||||
|
applyFormStates(nodes, states);
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
} catch (error) {
|
||||||
|
if (isAbortError(error)) throw error;
|
||||||
|
// Best-effort enrichment. DOMSnapshot/AX data still carries the nodes.
|
||||||
|
return false;
|
||||||
|
} finally {
|
||||||
|
await cdp.send(tabId, "Runtime.releaseObjectGroup", { objectGroup }).catch(() => undefined);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,178 +0,0 @@
|
|||||||
import type { CdpFrame, CdpFrameGraph } from "@/browser-driver/frame-graph";
|
|
||||||
import { type GeometryProjection, projectRectToViewport } from "../geometry";
|
|
||||||
import { GeometryContext } from "../geometry/frame-context";
|
|
||||||
import { type CdpRunner, cdpRunnerForTarget, sendToCdpTarget } from "../shared";
|
|
||||||
import { type CapturedNode, type CapturedViewModel, captureViewModel } from "./capture";
|
|
||||||
import {
|
|
||||||
buildFrameDocuments,
|
|
||||||
type FrameAxBatch,
|
|
||||||
type FrameDocument,
|
|
||||||
type FrameOwnedAxNode,
|
|
||||||
} from "./frame-document";
|
|
||||||
|
|
||||||
export type FrameAxNode = FrameOwnedAxNode;
|
|
||||||
export type CapturedFrameDocument<T extends FrameAxNode> = FrameDocument<T>;
|
|
||||||
|
|
||||||
function throwIfAborted(signal?: AbortSignal): void {
|
|
||||||
if (signal?.aborted) throw new DOMException("observation aborted", "AbortError");
|
|
||||||
}
|
|
||||||
|
|
||||||
function isAbortError(err: unknown): boolean {
|
|
||||||
return err instanceof Error && err.name === "AbortError";
|
|
||||||
}
|
|
||||||
|
|
||||||
function transformFrameNodes(
|
|
||||||
nodes: CapturedNode[],
|
|
||||||
projection: GeometryProjection | null,
|
|
||||||
): CapturedNode[] {
|
|
||||||
return nodes.map((node) => ({
|
|
||||||
...node,
|
|
||||||
rect: projection
|
|
||||||
? (() => {
|
|
||||||
const bounds = projectRectToViewport(node.rect, projection);
|
|
||||||
return bounds ? { x: bounds.x, y: bounds.y, w: bounds.width, h: bounds.height } : null;
|
|
||||||
})()
|
|
||||||
: null,
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
|
|
||||||
async function captureMissingFrameDocuments(
|
|
||||||
cdp: CdpRunner,
|
|
||||||
tabId: number,
|
|
||||||
graph: CdpFrameGraph,
|
|
||||||
captured: CapturedViewModel,
|
|
||||||
geometry: GeometryContext,
|
|
||||||
signal?: AbortSignal,
|
|
||||||
): Promise<void> {
|
|
||||||
if (!captured.frameNodes) captured.frameNodes = new Map();
|
|
||||||
if (!captured.frameOwnerBackendNodeIds) captured.frameOwnerBackendNodeIds = new Map();
|
|
||||||
if (!captured.frameParentIds) captured.frameParentIds = new Map();
|
|
||||||
|
|
||||||
for (const frame of graph.frames) {
|
|
||||||
if (!frame.target.sessionId || captured.frameNodes.has(frame.frameId)) continue;
|
|
||||||
try {
|
|
||||||
const child = await captureViewModel(cdpRunnerForTarget(cdp, frame.target), tabId, {
|
|
||||||
signal,
|
|
||||||
geometry,
|
|
||||||
target: frame.target,
|
|
||||||
});
|
|
||||||
if (child.frameGeometryIssues?.length) {
|
|
||||||
(captured.frameGeometryIssues ??= []).push(...child.frameGeometryIssues);
|
|
||||||
}
|
|
||||||
let projection: GeometryProjection | null = null;
|
|
||||||
try {
|
|
||||||
projection = await geometry.targetProjection(frame.frameId);
|
|
||||||
} catch (err) {
|
|
||||||
if (isAbortError(err)) throw err;
|
|
||||||
console.debug("[bsk observation] child frame geometry projection failed", {
|
|
||||||
frameId: frame.frameId,
|
|
||||||
err,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
const rootFrameId = child.rootFrameId ?? frame.frameId;
|
|
||||||
const childFrames = child.frameNodes ?? new Map<string, CapturedNode[]>();
|
|
||||||
if (!childFrames.has(rootFrameId)) childFrames.set(rootFrameId, child.nodes);
|
|
||||||
if (rootFrameId !== frame.frameId && !childFrames.has(frame.frameId)) {
|
|
||||||
childFrames.set(
|
|
||||||
frame.frameId,
|
|
||||||
child.nodes.map((node) => ({ ...node, frameId: frame.frameId })),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
for (const [childFrameId, nodes] of childFrames) {
|
|
||||||
captured.frameNodes.set(childFrameId, transformFrameNodes(nodes, projection));
|
|
||||||
}
|
|
||||||
if (frame.ownerBackendNodeId !== undefined) {
|
|
||||||
captured.iframeNodes.set(
|
|
||||||
frame.ownerBackendNodeId,
|
|
||||||
captured.frameNodes.get(frame.frameId) ?? [],
|
|
||||||
);
|
|
||||||
captured.frameOwnerBackendNodeIds.set(frame.frameId, frame.ownerBackendNodeId);
|
|
||||||
}
|
|
||||||
if (frame.parentFrameId) captured.frameParentIds.set(frame.frameId, frame.parentFrameId);
|
|
||||||
for (const [childFrameId, ownerBackendNodeId] of child.frameOwnerBackendNodeIds ?? []) {
|
|
||||||
captured.frameOwnerBackendNodeIds.set(childFrameId, ownerBackendNodeId);
|
|
||||||
}
|
|
||||||
for (const [childFrameId, parentFrameId] of child.frameParentIds ?? []) {
|
|
||||||
captured.frameParentIds.set(childFrameId, parentFrameId);
|
|
||||||
}
|
|
||||||
} catch (err) {
|
|
||||||
if (isAbortError(err)) throw err;
|
|
||||||
console.debug("[bsk observation] child frame DOM capture failed", {
|
|
||||||
frameId: frame.frameId,
|
|
||||||
err,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function captureAxTrees<T extends FrameAxNode>(
|
|
||||||
cdp: CdpRunner,
|
|
||||||
tabId: number,
|
|
||||||
discoveredFrames: CdpFrame[],
|
|
||||||
captured: CapturedViewModel,
|
|
||||||
signal?: AbortSignal,
|
|
||||||
): Promise<FrameAxBatch<T>[]> {
|
|
||||||
let frames = [...discoveredFrames];
|
|
||||||
const capturedFrameIds = [...(captured.frameNodes?.keys() ?? [])];
|
|
||||||
if (frames.length === 0) {
|
|
||||||
const rootFrameId = captured.rootFrameId ?? capturedFrameIds[0] ?? "root";
|
|
||||||
frames = [{ frameId: rootFrameId, target: { tabId } }];
|
|
||||||
}
|
|
||||||
for (const frameId of capturedFrameIds) {
|
|
||||||
if (!frames.some((frame) => frame.frameId === frameId)) {
|
|
||||||
frames.push({ frameId, target: { tabId } });
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let firstFailure: unknown;
|
|
||||||
const trees = await Promise.all(
|
|
||||||
frames.map(async (frame): Promise<FrameAxBatch<T>> => {
|
|
||||||
try {
|
|
||||||
throwIfAborted(signal);
|
|
||||||
await sendToCdpTarget(cdp, frame.target, "Accessibility.enable", {});
|
|
||||||
const result = await sendToCdpTarget<{ nodes?: T[] }>(
|
|
||||||
cdp,
|
|
||||||
frame.target,
|
|
||||||
"Accessibility.getFullAXTree",
|
|
||||||
frame.frameId === "root" ? {} : { frameId: frame.frameId },
|
|
||||||
);
|
|
||||||
throwIfAborted(signal);
|
|
||||||
return {
|
|
||||||
frame,
|
|
||||||
nodes: result.nodes ?? [],
|
|
||||||
};
|
|
||||||
} catch (err) {
|
|
||||||
if (isAbortError(err)) throw err;
|
|
||||||
firstFailure ??= err;
|
|
||||||
console.debug("[bsk observation] frame AX capture failed", {
|
|
||||||
frameId: frame.frameId,
|
|
||||||
err,
|
|
||||||
});
|
|
||||||
return { frame, nodes: [] };
|
|
||||||
}
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
const capturedNodeCount = [...captured.iframeNodes.values()].reduce(
|
|
||||||
(count, nodes) => count + nodes.length,
|
|
||||||
captured.nodes.length,
|
|
||||||
);
|
|
||||||
if (firstFailure && capturedNodeCount === 0 && trees.every((tree) => tree.nodes.length === 0)) {
|
|
||||||
throw firstFailure;
|
|
||||||
}
|
|
||||||
return trees;
|
|
||||||
}
|
|
||||||
|
|
||||||
export async function captureFrameData<T extends FrameAxNode>(
|
|
||||||
cdp: CdpRunner,
|
|
||||||
tabId: number,
|
|
||||||
captured: CapturedViewModel,
|
|
||||||
signal?: AbortSignal,
|
|
||||||
geometry = new GeometryContext(cdp, tabId, undefined, signal),
|
|
||||||
): Promise<CapturedFrameDocument<T>[]> {
|
|
||||||
const graph = await geometry.graph();
|
|
||||||
const frames = graph?.frames ?? [];
|
|
||||||
if (graph) await captureMissingFrameDocuments(cdp, tabId, graph, captured, geometry, signal);
|
|
||||||
throwIfAborted(signal);
|
|
||||||
const batches = await captureAxTrees<T>(cdp, tabId, frames, captured, signal);
|
|
||||||
return buildFrameDocuments(graph, batches, captured);
|
|
||||||
}
|
|
||||||
@@ -4,7 +4,8 @@ import {
|
|||||||
type CdpTarget,
|
type CdpTarget,
|
||||||
cdpTargetKey,
|
cdpTargetKey,
|
||||||
} from "@/browser-driver/frame-graph";
|
} from "@/browser-driver/frame-graph";
|
||||||
import type { CapturedNode, CapturedViewModel } from "./capture";
|
import type { CapturedNode } from "./facts";
|
||||||
|
import { captureCheckpoint } from "./facts";
|
||||||
|
|
||||||
export interface FrameOwnedAxNode {
|
export interface FrameOwnedAxNode {
|
||||||
nodeId: string;
|
nodeId: string;
|
||||||
@@ -23,6 +24,16 @@ export interface FrameDocument<T extends FrameOwnedAxNode> extends CdpFrame {
|
|||||||
contextScopeId: string;
|
contextScopeId: string;
|
||||||
axNodes: T[];
|
axNodes: T[];
|
||||||
domNodes: CapturedNode[];
|
domNodes: CapturedNode[];
|
||||||
|
excludedBackendNodeIds?: ReadonlySet<number>;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Explicit snapshot document data used to join DOM and AX ownership. */
|
||||||
|
export interface FrameDomInput {
|
||||||
|
nodes: CapturedNode[];
|
||||||
|
rootFrameId?: string;
|
||||||
|
frameNodes?: ReadonlyMap<string, CapturedNode[]>;
|
||||||
|
frameOwnerBackendNodeIds?: ReadonlyMap<string, number>;
|
||||||
|
frameParentIds?: ReadonlyMap<string, string>;
|
||||||
}
|
}
|
||||||
|
|
||||||
interface Ownership {
|
interface Ownership {
|
||||||
@@ -47,17 +58,13 @@ function targetBackendKey(target: CdpTarget, backendNodeId: number): string {
|
|||||||
function frameList<T extends FrameOwnedAxNode>(
|
function frameList<T extends FrameOwnedAxNode>(
|
||||||
graph: CdpFrameGraph | null,
|
graph: CdpFrameGraph | null,
|
||||||
batches: FrameAxBatch<T>[],
|
batches: FrameAxBatch<T>[],
|
||||||
captured: CapturedViewModel,
|
captured: FrameDomInput,
|
||||||
): CdpFrame[] {
|
): CdpFrame[] {
|
||||||
const frames = new Map<string, CdpFrame>();
|
const frames = new Map<string, CdpFrame>();
|
||||||
for (const frame of graph?.frames ?? []) frames.set(frame.frameId, frame);
|
for (const frame of graph?.frames ?? []) frames.set(frame.frameId, frame);
|
||||||
for (const batch of batches) {
|
for (const batch of batches) {
|
||||||
if (!frames.has(batch.frame.frameId)) frames.set(batch.frame.frameId, batch.frame);
|
if (!frames.has(batch.frame.frameId)) frames.set(batch.frame.frameId, batch.frame);
|
||||||
}
|
}
|
||||||
for (const frameId of captured.frameNodes?.keys() ?? []) {
|
|
||||||
if (!frames.has(frameId))
|
|
||||||
frames.set(frameId, { frameId, target: { tabId: batches[0]?.frame.target.tabId ?? 0 } });
|
|
||||||
}
|
|
||||||
return [...frames.values()].map((frame) => {
|
return [...frames.values()].map((frame) => {
|
||||||
const ownerBackendNodeId =
|
const ownerBackendNodeId =
|
||||||
frame.ownerBackendNodeId ?? captured.frameOwnerBackendNodeIds?.get(frame.frameId);
|
frame.ownerBackendNodeId ?? captured.frameOwnerBackendNodeIds?.get(frame.frameId);
|
||||||
@@ -70,11 +77,14 @@ function frameList<T extends FrameOwnedAxNode>(
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
export function buildFrameDocuments<T extends FrameOwnedAxNode>(
|
export async function buildFrameDocuments<T extends FrameOwnedAxNode>(
|
||||||
graph: CdpFrameGraph | null,
|
graph: CdpFrameGraph | null,
|
||||||
batches: FrameAxBatch<T>[],
|
batches: FrameAxBatch<T>[],
|
||||||
captured: CapturedViewModel,
|
captured: FrameDomInput,
|
||||||
): FrameDocument<T>[] {
|
signal?: AbortSignal,
|
||||||
|
unresolved?: (frame: CdpFrame) => void,
|
||||||
|
): Promise<FrameDocument<T>[]> {
|
||||||
|
let work = 0;
|
||||||
const frames = frameList(graph, batches, captured);
|
const frames = frameList(graph, batches, captured);
|
||||||
const frameById = new Map(frames.map((frame) => [frame.frameId, frame]));
|
const frameById = new Map(frames.map((frame) => [frame.frameId, frame]));
|
||||||
const rootFrameId = graph?.rootFrameId ?? captured.rootFrameId ?? frames[0]?.frameId;
|
const rootFrameId = graph?.rootFrameId ?? captured.rootFrameId ?? frames[0]?.frameId;
|
||||||
@@ -83,54 +93,74 @@ export function buildFrameDocuments<T extends FrameOwnedAxNode>(
|
|||||||
const backendOwner = new Map<string, string>();
|
const backendOwner = new Map<string, string>();
|
||||||
for (const frame of frames) {
|
for (const frame of frames) {
|
||||||
for (const node of domNodesForFrame(frame.frameId)) {
|
for (const node of domNodesForFrame(frame.frameId)) {
|
||||||
|
if (work++ % 256 === 0) await captureCheckpoint(signal);
|
||||||
backendOwner.set(targetBackendKey(frame.target, node.backendNodeId), frame.frameId);
|
backendOwner.set(targetBackendKey(frame.target, node.backendNodeId), frame.frameId);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const candidates = new Map<string, OwnedCandidate<T>>();
|
const candidates = new Map<string, OwnedCandidate<T>>();
|
||||||
for (const batch of batches) {
|
for (const batch of batches) {
|
||||||
const nodeById = new Map(batch.nodes.map((node) => [node.nodeId, node]));
|
const nodeById = new Map<string, T>();
|
||||||
|
for (const node of batch.nodes) {
|
||||||
|
if (work++ % 256 === 0) await captureCheckpoint(signal);
|
||||||
|
nodeById.set(node.nodeId, node);
|
||||||
|
}
|
||||||
|
let reported = false;
|
||||||
const ownershipByNodeId = new Map<string, Ownership>();
|
const ownershipByNodeId = new Map<string, Ownership>();
|
||||||
const resolving = new Set<string>();
|
const resolveOwnership = async (node: T): Promise<Ownership> => {
|
||||||
const resolveOwnership = (node: T): Ownership => {
|
const path: T[] = [];
|
||||||
const cached = ownershipByNodeId.get(node.nodeId);
|
const visiting = new Set<string>();
|
||||||
if (cached) return cached;
|
let current: T | undefined = node;
|
||||||
if (node.frameId && frameById.has(node.frameId)) {
|
let ownership: Ownership = { frameId: batch.frame.frameId, strength: 1 };
|
||||||
const ownership = { frameId: node.frameId, strength: 4 };
|
while (current) {
|
||||||
ownershipByNodeId.set(node.nodeId, ownership);
|
if (work++ % 256 === 0) await captureCheckpoint(signal);
|
||||||
return ownership;
|
const cached = ownershipByNodeId.get(current.nodeId);
|
||||||
}
|
if (cached) {
|
||||||
if (typeof node.backendDOMNodeId === "number") {
|
ownership = cached;
|
||||||
const frameId = backendOwner.get(
|
break;
|
||||||
targetBackendKey(batch.frame.target, node.backendDOMNodeId),
|
}
|
||||||
);
|
if (visiting.has(current.nodeId)) break;
|
||||||
|
visiting.add(current.nodeId);
|
||||||
|
if (current.frameId) {
|
||||||
|
const frame = frameById.get(current.frameId);
|
||||||
|
// An explicit contradiction must not fall through to a guessed owner.
|
||||||
|
if (!frame || cdpTargetKey(frame.target) !== cdpTargetKey(batch.frame.target)) {
|
||||||
|
if (!reported) {
|
||||||
|
unresolved?.(batch.frame);
|
||||||
|
reported = true;
|
||||||
|
}
|
||||||
|
ownership = { frameId: "", strength: 0 };
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
ownership = { frameId: frame.frameId, strength: 4 };
|
||||||
|
ownershipByNodeId.set(current.nodeId, ownership);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
const frameId =
|
||||||
|
typeof current.backendDOMNodeId === "number"
|
||||||
|
? backendOwner.get(targetBackendKey(batch.frame.target, current.backendDOMNodeId))
|
||||||
|
: undefined;
|
||||||
if (frameId) {
|
if (frameId) {
|
||||||
const ownership = { frameId, strength: 3 };
|
ownership = { frameId, strength: 3 };
|
||||||
ownershipByNodeId.set(node.nodeId, ownership);
|
ownershipByNodeId.set(current.nodeId, ownership);
|
||||||
return ownership;
|
break;
|
||||||
}
|
}
|
||||||
|
path.push(current);
|
||||||
|
current = current.parentId ? nodeById.get(current.parentId) : undefined;
|
||||||
}
|
}
|
||||||
if (node.parentId && !resolving.has(node.nodeId)) {
|
for (let i = path.length - 1; i >= 0; i--) {
|
||||||
const parent = nodeById.get(node.parentId);
|
if (work++ % 256 === 0) await captureCheckpoint(signal);
|
||||||
if (parent) {
|
ownership = { frameId: ownership.frameId, strength: Math.min(2, ownership.strength) };
|
||||||
resolving.add(node.nodeId);
|
ownershipByNodeId.set(path[i].nodeId, ownership);
|
||||||
const inherited = resolveOwnership(parent);
|
|
||||||
resolving.delete(node.nodeId);
|
|
||||||
const ownership = {
|
|
||||||
frameId: inherited.frameId,
|
|
||||||
strength: Math.min(2, inherited.strength),
|
|
||||||
};
|
|
||||||
ownershipByNodeId.set(node.nodeId, ownership);
|
|
||||||
return ownership;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
const ownership = { frameId: batch.frame.frameId, strength: 1 };
|
if (!ownershipByNodeId.has(node.nodeId)) ownershipByNodeId.set(node.nodeId, ownership);
|
||||||
ownershipByNodeId.set(node.nodeId, ownership);
|
|
||||||
return ownership;
|
return ownership;
|
||||||
};
|
};
|
||||||
|
|
||||||
for (const node of batch.nodes) {
|
for (const node of batch.nodes) {
|
||||||
const ownership = resolveOwnership(node);
|
if (work++ % 256 === 0) await captureCheckpoint(signal);
|
||||||
|
const ownership = await resolveOwnership(node);
|
||||||
|
if (!ownership.frameId) continue;
|
||||||
const key = targetNodeKey(batch.frame.target, node.nodeId);
|
const key = targetNodeKey(batch.frame.target, node.nodeId);
|
||||||
const existing = candidates.get(key);
|
const existing = candidates.get(key);
|
||||||
if (!existing || ownership.strength > existing.ownership.strength) {
|
if (!existing || ownership.strength > existing.ownership.strength) {
|
||||||
@@ -144,6 +174,7 @@ export function buildFrameDocuments<T extends FrameOwnedAxNode>(
|
|||||||
);
|
);
|
||||||
const axNodesByFrame = new Map<string, T[]>();
|
const axNodesByFrame = new Map<string, T[]>();
|
||||||
for (const candidate of candidates.values()) {
|
for (const candidate of candidates.values()) {
|
||||||
|
if (work++ % 256 === 0) await captureCheckpoint(signal);
|
||||||
const { node, target, ownership } = candidate;
|
const { node, target, ownership } = candidate;
|
||||||
const parentFrameId = node.parentId
|
const parentFrameId = node.parentId
|
||||||
? ownershipByTargetNode.get(targetNodeKey(target, node.parentId))
|
? ownershipByTargetNode.get(targetNodeKey(target, node.parentId))
|
||||||
|
|||||||
@@ -0,0 +1,217 @@
|
|||||||
|
import type { Viewport } from "@browser-skill/vom";
|
||||||
|
import type { CdpFrame, CdpTarget } from "@/browser-driver/frame-graph";
|
||||||
|
import { type GeometryProjection, projectRectToViewport } from "../geometry";
|
||||||
|
import {
|
||||||
|
type FrameProjectionState,
|
||||||
|
projectSnapshotRect,
|
||||||
|
snapshotViewportRect,
|
||||||
|
} from "../geometry/coordinate-types";
|
||||||
|
import { cssViewport, GeometryContext } from "../geometry/frame-context";
|
||||||
|
import type { CapturedNode } from "./facts";
|
||||||
|
import {
|
||||||
|
buildDocumentIndex,
|
||||||
|
type CaptureIssue,
|
||||||
|
captureCheckpoint,
|
||||||
|
type DocumentIndex,
|
||||||
|
isCaptureAbort,
|
||||||
|
type NodeFacts,
|
||||||
|
throwCaptureAborted,
|
||||||
|
} from "./facts";
|
||||||
|
import {
|
||||||
|
decodeDocument,
|
||||||
|
type SnapshotDocument,
|
||||||
|
type SnapshotReply,
|
||||||
|
snapshotFrameId,
|
||||||
|
} from "./snapshot";
|
||||||
|
|
||||||
|
export interface NormalizedDocument {
|
||||||
|
index: DocumentIndex;
|
||||||
|
frame: CdpFrame;
|
||||||
|
domNodes: CapturedNode[];
|
||||||
|
}
|
||||||
|
|
||||||
|
/** One target's snapshot. Source ownership is checked before interpreting any
|
||||||
|
* coordinates. Snapshot and live quad coordinates are deliberately not conflated. */
|
||||||
|
export async function normalizeSnapshot(
|
||||||
|
snapshot: SnapshotReply,
|
||||||
|
target: CdpTarget,
|
||||||
|
frames: readonly CdpFrame[],
|
||||||
|
geometry: GeometryContext,
|
||||||
|
issues: CaptureIssue[],
|
||||||
|
signal?: AbortSignal,
|
||||||
|
): Promise<NormalizedDocument[]> {
|
||||||
|
const strings = snapshot.strings ?? [];
|
||||||
|
const raw = snapshot.documents ?? [];
|
||||||
|
const frameById = new Map(frames.map((frame) => [frame.frameId, frame]));
|
||||||
|
const sources = new Map<string, SnapshotDocument>();
|
||||||
|
for (const doc of raw) {
|
||||||
|
const id = snapshotFrameId(doc, strings);
|
||||||
|
if (!id || !frameById.has(id) || sources.has(id)) {
|
||||||
|
issues.push({
|
||||||
|
target,
|
||||||
|
frameId: id,
|
||||||
|
stage: "ownership",
|
||||||
|
reason: "frame-ownership-unresolved",
|
||||||
|
});
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
sources.set(id, doc);
|
||||||
|
}
|
||||||
|
let viewport: Viewport = { width: 0, height: 0 };
|
||||||
|
let scrollX = 0,
|
||||||
|
scrollY = 0;
|
||||||
|
try {
|
||||||
|
const measured = cssViewport(await geometry.layoutMetrics(target));
|
||||||
|
viewport = { width: measured.width, height: measured.height };
|
||||||
|
scrollX = measured.scrollX;
|
||||||
|
scrollY = measured.scrollY;
|
||||||
|
} catch (error) {
|
||||||
|
if (isCaptureAbort(error)) throw error;
|
||||||
|
}
|
||||||
|
|
||||||
|
const children = new Map<string, CdpFrame[]>();
|
||||||
|
const pending: CdpFrame[] = [];
|
||||||
|
for (const frame of frames) {
|
||||||
|
if (!frame.parentFrameId || !frameById.has(frame.parentFrameId)) pending.push(frame);
|
||||||
|
else {
|
||||||
|
const siblings = children.get(frame.parentFrameId);
|
||||||
|
if (siblings) siblings.push(frame);
|
||||||
|
else children.set(frame.parentFrameId, [frame]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
const projections = new Map<string, FrameProjectionState | null>();
|
||||||
|
const result: NormalizedDocument[] = [];
|
||||||
|
let targetProjection: GeometryProjection | null = null;
|
||||||
|
const targetRoot = pending[0];
|
||||||
|
if (target.sessionId && targetRoot) {
|
||||||
|
try {
|
||||||
|
targetProjection = await geometry.targetProjection(targetRoot.frameId);
|
||||||
|
} catch (error) {
|
||||||
|
if (isCaptureAbort(error)) throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for (let cursor = 0; cursor < pending.length; cursor++) {
|
||||||
|
if (cursor % 256 === 0) await captureCheckpoint(signal);
|
||||||
|
const frame = pending[cursor];
|
||||||
|
for (const child of children.get(frame.frameId) ?? []) pending.push(child);
|
||||||
|
const doc = sources.get(frame.frameId);
|
||||||
|
if (!doc) continue;
|
||||||
|
const source = { target, frameId: frame.frameId };
|
||||||
|
const parent = frame.parentFrameId ? frameById.get(frame.parentFrameId) : undefined;
|
||||||
|
const state: FrameProjectionState | null = !parent
|
||||||
|
? {
|
||||||
|
status: "available",
|
||||||
|
projection: { source, geometry: { sourceClips: [], edges: [], topViewport: viewport } },
|
||||||
|
}
|
||||||
|
: (projections.get(frame.frameId) ?? null);
|
||||||
|
projections.set(frame.frameId, state);
|
||||||
|
const projection = state?.status === "available" ? state.projection : null;
|
||||||
|
if (!projection || (target.sessionId && !targetProjection))
|
||||||
|
issues.push({
|
||||||
|
target,
|
||||||
|
frameId: frame.frameId,
|
||||||
|
stage: "geometry",
|
||||||
|
reason: "geometry-unavailable",
|
||||||
|
...(state && state.status !== "available" ? { projectionIssue: state } : {}),
|
||||||
|
});
|
||||||
|
|
||||||
|
// Only direct siblings share this pool. Keep normalization in its existing
|
||||||
|
// breadth-first order, independent of owner read completion order.
|
||||||
|
const siblings = (children.get(frame.frameId) ?? []).filter((child) =>
|
||||||
|
sources.has(child.frameId),
|
||||||
|
);
|
||||||
|
const edge = projection?.geometry.edges[0];
|
||||||
|
const clips = edge
|
||||||
|
? [edge.destinationQuad, ...(edge.destinationClips ?? [])]
|
||||||
|
: (projection?.geometry.sourceClips ?? []);
|
||||||
|
let next = 0;
|
||||||
|
let failure: unknown;
|
||||||
|
await Promise.all(
|
||||||
|
Array.from({ length: Math.min(4, siblings.length) }, async () => {
|
||||||
|
while (next < siblings.length && !signal?.aborted && !failure) {
|
||||||
|
const child = siblings[next++];
|
||||||
|
const childSource = { target, frameId: child.frameId };
|
||||||
|
if (!projection) {
|
||||||
|
projections.set(
|
||||||
|
child.frameId,
|
||||||
|
state && state.status !== "available"
|
||||||
|
? {
|
||||||
|
status: "blocked",
|
||||||
|
source: childSource,
|
||||||
|
cause: state.status === "blocked" ? state.cause : state,
|
||||||
|
}
|
||||||
|
: null,
|
||||||
|
);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if (child.ownerBackendNodeId === undefined) {
|
||||||
|
projections.set(child.frameId, null);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
projections.set(
|
||||||
|
child.frameId,
|
||||||
|
await geometry.snapshotProjection(
|
||||||
|
childSource,
|
||||||
|
child.ownerBackendNodeId,
|
||||||
|
clips,
|
||||||
|
viewport,
|
||||||
|
),
|
||||||
|
);
|
||||||
|
} catch (error) {
|
||||||
|
failure ??= error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
// Join active reads and their cleanup before returning cancellation.
|
||||||
|
if (failure) throw failure;
|
||||||
|
throwCaptureAborted(signal);
|
||||||
|
const decoded = await decodeDocument(doc, strings, signal);
|
||||||
|
const nodes: NodeFacts[] = [];
|
||||||
|
for (let i = 0; i < decoded.nodes.length; i++) {
|
||||||
|
if (i % 256 === 0) await captureCheckpoint(signal);
|
||||||
|
const node = decoded.nodes[i];
|
||||||
|
const input = snapshotViewportRect(node.layout?.bounds ?? [], source, {
|
||||||
|
x: doc.scrollOffsetX ?? (parent ? 0 : scrollX),
|
||||||
|
y: doc.scrollOffsetY ?? (parent ? 0 : scrollY),
|
||||||
|
});
|
||||||
|
const local = input?.rect;
|
||||||
|
let rect = input && projection ? projectSnapshotRect(input, projection) : null;
|
||||||
|
if (rect && target.sessionId)
|
||||||
|
rect = targetProjection
|
||||||
|
? projectRectToViewport(
|
||||||
|
{ x: rect.x, y: rect.y, w: rect.width, h: rect.height },
|
||||||
|
targetProjection,
|
||||||
|
)
|
||||||
|
: null;
|
||||||
|
const styles = node.layout?.styles;
|
||||||
|
const visibility = styles?.visibility || "visible";
|
||||||
|
const opacity = styles?.opacity || "1";
|
||||||
|
nodes.push({
|
||||||
|
...node,
|
||||||
|
frameId: frame.frameId,
|
||||||
|
ownerFrameBackendNodeId: frame.ownerBackendNodeId ?? null,
|
||||||
|
localRect: local ? { x: local.x, y: local.y, w: local.width, h: local.height } : null,
|
||||||
|
rect: rect ? { x: rect.x, y: rect.y, w: rect.width, h: rect.height } : null,
|
||||||
|
rendered:
|
||||||
|
!!local &&
|
||||||
|
visibility !== "hidden" &&
|
||||||
|
visibility !== "collapse" &&
|
||||||
|
(Number.parseFloat(opacity) || 0) > 0,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
const index = await buildDocumentIndex(nodes, signal);
|
||||||
|
const domNodes = nodes.filter((node) => !node.tag.startsWith("#"));
|
||||||
|
result.push({
|
||||||
|
index,
|
||||||
|
frame,
|
||||||
|
domNodes,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
for (const frameId of sources.keys()) {
|
||||||
|
if (!projections.has(frameId))
|
||||||
|
issues.push({ target, frameId, stage: "ownership", reason: "frame-ownership-unresolved" });
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
}
|
||||||
@@ -2,7 +2,10 @@ import type { Rect, RenderedRef, VomResult } from "@browser-skill/vom";
|
|||||||
import type { CdpTarget } from "@/browser-driver/frame-graph";
|
import type { CdpTarget } from "@/browser-driver/frame-graph";
|
||||||
import { readRecordingDocumentIdentity } from "@/shared/recording-document-identity";
|
import { readRecordingDocumentIdentity } from "@/shared/recording-document-identity";
|
||||||
import type { CapturedSurfaceProbe } from "./capture";
|
import type { CapturedSurfaceProbe } from "./capture";
|
||||||
import type { CapturedFrameDocument, FrameAxNode } from "./frame-capture";
|
import type {
|
||||||
|
FrameDocument as CapturedFrameDocument,
|
||||||
|
FrameOwnedAxNode as FrameAxNode,
|
||||||
|
} from "./frame-document";
|
||||||
|
|
||||||
/** Frame identity needed to resolve an `@eN` that lives in an iframe. */
|
/** Frame identity needed to resolve an `@eN` that lives in an iframe. */
|
||||||
export interface CaptureVomFrame {
|
export interface CaptureVomFrame {
|
||||||
|
|||||||
@@ -61,8 +61,9 @@ export function buildSemanticGraph(input: BuildSemanticGraphInput): SemanticGrap
|
|||||||
dom,
|
dom,
|
||||||
sourceOrder: index,
|
sourceOrder: index,
|
||||||
excluded:
|
excluded:
|
||||||
document.frameId === rootFrameId &&
|
document.excludedBackendNodeIds?.has(dom.backendNodeId) === true ||
|
||||||
input.excludedBackendNodeIds?.has(dom.backendNodeId) === true,
|
(document.frameId === rootFrameId &&
|
||||||
|
input.excludedBackendNodeIds?.has(dom.backendNodeId) === true),
|
||||||
};
|
};
|
||||||
nodes.set(id, node);
|
nodes.set(id, node);
|
||||||
frameNodeIds.push(id);
|
frameNodeIds.push(id);
|
||||||
@@ -90,9 +91,10 @@ export function buildSemanticGraph(input: BuildSemanticGraphInput): SemanticGrap
|
|||||||
ax,
|
ax,
|
||||||
sourceOrder: document.domNodes.length + index,
|
sourceOrder: document.domNodes.length + index,
|
||||||
excluded:
|
excluded:
|
||||||
document.frameId === rootFrameId &&
|
|
||||||
typeof ax.backendDOMNodeId === "number" &&
|
typeof ax.backendDOMNodeId === "number" &&
|
||||||
input.excludedBackendNodeIds?.has(ax.backendDOMNodeId) === true,
|
(document.excludedBackendNodeIds?.has(ax.backendDOMNodeId) === true ||
|
||||||
|
(document.frameId === rootFrameId &&
|
||||||
|
input.excludedBackendNodeIds?.has(ax.backendDOMNodeId) === true)),
|
||||||
});
|
});
|
||||||
frameNodeIds.push(id);
|
frameNodeIds.push(id);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,230 @@
|
|||||||
|
import type { CapturedNode } from "./facts";
|
||||||
|
import { captureCheckpoint, type DecodedDocument, type DecodedNode } from "./facts";
|
||||||
|
export const REQUESTED_STYLES = [
|
||||||
|
"position",
|
||||||
|
"pointer-events",
|
||||||
|
"cursor",
|
||||||
|
"visibility",
|
||||||
|
"opacity",
|
||||||
|
"display",
|
||||||
|
"overflow-x",
|
||||||
|
"overflow-y",
|
||||||
|
"transform",
|
||||||
|
"zoom",
|
||||||
|
"clip-path",
|
||||||
|
"mask-image",
|
||||||
|
] as const;
|
||||||
|
const STYLE_COL = Object.fromEntries(
|
||||||
|
REQUESTED_STYLES.map((name, index) => [name, index]),
|
||||||
|
) as Record<(typeof REQUESTED_STYLES)[number], number>;
|
||||||
|
/** Sparse array format Chrome uses for infrequently-set per-node fields. */
|
||||||
|
interface SparseArray {
|
||||||
|
index: number[];
|
||||||
|
value: number[];
|
||||||
|
}
|
||||||
|
|
||||||
|
interface RareBooleanData {
|
||||||
|
index: number[];
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface SnapshotDocument {
|
||||||
|
frameId?: string | number;
|
||||||
|
scrollOffsetX?: number;
|
||||||
|
scrollOffsetY?: number;
|
||||||
|
nodes?: {
|
||||||
|
parentIndex?: number[];
|
||||||
|
nodeName?: number[];
|
||||||
|
nodeType?: number[];
|
||||||
|
backendNodeId?: number[];
|
||||||
|
attributes?: number[][];
|
||||||
|
/**
|
||||||
|
* Per-node text value (index into strings), set for `#text` / CDATA nodes.
|
||||||
|
* Element nodes carry -1. Same length as `backendNodeId`.
|
||||||
|
*/
|
||||||
|
nodeValue?: number[];
|
||||||
|
/** Maps node array index → index into `documents[]` for frame content. */
|
||||||
|
contentDocumentIndex?: SparseArray;
|
||||||
|
inputValue?: SparseArray;
|
||||||
|
textValue?: SparseArray;
|
||||||
|
inputChecked?: RareBooleanData;
|
||||||
|
optionSelected?: RareBooleanData;
|
||||||
|
};
|
||||||
|
layout?: {
|
||||||
|
nodeIndex?: number[];
|
||||||
|
styles?: number[][];
|
||||||
|
bounds?: number[][];
|
||||||
|
paintOrders?: number[];
|
||||||
|
clientRects?: number[][];
|
||||||
|
offsetRects?: number[][];
|
||||||
|
scrollRects?: number[][];
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
export function snapshotFrameId(document: SnapshotDocument, strings: string[]): string | undefined {
|
||||||
|
if (typeof document.frameId === "string") return document.frameId || undefined;
|
||||||
|
if (typeof document.frameId === "number") return str(strings, document.frameId) || undefined;
|
||||||
|
return undefined;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface SnapshotReply {
|
||||||
|
strings?: string[];
|
||||||
|
documents?: SnapshotDocument[];
|
||||||
|
}
|
||||||
|
|
||||||
|
function isSensitiveFormControl(node: Pick<CapturedNode, "tag" | "attrs">): boolean {
|
||||||
|
return node.tag === "input" && (node.attrs.type ?? "").toLowerCase() === "password";
|
||||||
|
}
|
||||||
|
|
||||||
|
function snapshotFormState(
|
||||||
|
value: string | undefined,
|
||||||
|
defaultValue: string,
|
||||||
|
hasDefaultValue: boolean,
|
||||||
|
sensitive: boolean,
|
||||||
|
): CapturedNode["formState"] {
|
||||||
|
if (sensitive) {
|
||||||
|
if (value === undefined && !hasDefaultValue) return undefined;
|
||||||
|
return (value ?? defaultValue) === "" ? "empty" : "filled";
|
||||||
|
}
|
||||||
|
if (value === undefined) return undefined;
|
||||||
|
if (value === "") return "empty";
|
||||||
|
return value === defaultValue ? "default" : "filled";
|
||||||
|
}
|
||||||
|
|
||||||
|
function str(strings: string[], idx: number | undefined): string {
|
||||||
|
if (idx === undefined || idx < 0) return "";
|
||||||
|
return strings[idx] ?? "";
|
||||||
|
}
|
||||||
|
|
||||||
|
function sparseIndexMap(sparse: SparseArray | undefined): Map<number, number> {
|
||||||
|
const out = new Map<number, number>();
|
||||||
|
if (!sparse?.index || !sparse.value) return out;
|
||||||
|
for (let i = 0; i < sparse.index.length; i++) {
|
||||||
|
const nodeIndex = sparse.index[i];
|
||||||
|
const docIndex = sparse.value[i];
|
||||||
|
if (nodeIndex !== undefined && docIndex !== undefined) out.set(nodeIndex, docIndex);
|
||||||
|
}
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function decodeDocument(
|
||||||
|
doc: SnapshotDocument,
|
||||||
|
strings: string[],
|
||||||
|
signal?: AbortSignal,
|
||||||
|
): Promise<DecodedDocument> {
|
||||||
|
const dn = doc.nodes;
|
||||||
|
const dl = doc.layout;
|
||||||
|
if (!dn?.backendNodeId) {
|
||||||
|
return { nodes: [] };
|
||||||
|
}
|
||||||
|
|
||||||
|
const count = dn.backendNodeId.length;
|
||||||
|
const layoutByNode = new Map<number, number>();
|
||||||
|
for (let i = 0; i < (dl?.nodeIndex?.length ?? 0); i++) {
|
||||||
|
if (i % 256 === 0) await captureCheckpoint(signal);
|
||||||
|
layoutByNode.set(dl!.nodeIndex![i], i);
|
||||||
|
}
|
||||||
|
|
||||||
|
const inputValues = sparseIndexMap(dn.inputValue);
|
||||||
|
const textValues = sparseIndexMap(dn.textValue);
|
||||||
|
const checkedInputs = new Set(dn.inputChecked?.index ?? []);
|
||||||
|
const selectedOptions = new Set(dn.optionSelected?.index ?? []);
|
||||||
|
|
||||||
|
// Collect visible text from #text child nodes so element CapturedNodes
|
||||||
|
// carry a textContent value usable as a button/link label fallback.
|
||||||
|
// nodeValue is a parallel array: string index for text nodes, -1 otherwise.
|
||||||
|
const nodeTextContent = new Map<number, string[]>();
|
||||||
|
if (dn.nodeValue) {
|
||||||
|
for (let n = 0; n < count; n++) {
|
||||||
|
if (n % 256 === 0) await captureCheckpoint(signal);
|
||||||
|
const nvIdx = dn.nodeValue[n] ?? -1;
|
||||||
|
if (nvIdx < 0) continue;
|
||||||
|
const text = str(strings, nvIdx).trim();
|
||||||
|
if (!text) continue;
|
||||||
|
const parentIdx = dn.parentIndex?.[n] ?? -1;
|
||||||
|
if (parentIdx >= 0) {
|
||||||
|
const existing = nodeTextContent.get(parentIdx);
|
||||||
|
if (existing) existing.push(text);
|
||||||
|
else nodeTextContent.set(parentIdx, [text]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Decode fields without assigning a coordinate projection or semantic policy.
|
||||||
|
const nodes: DecodedNode[] = [];
|
||||||
|
for (let n = 0; n < count; n++) {
|
||||||
|
if (n % 256 === 0) await captureCheckpoint(signal);
|
||||||
|
const backendNodeId = dn.backendNodeId[n];
|
||||||
|
const parentIdx = dn.parentIndex?.[n] ?? -1;
|
||||||
|
const parentBackendNodeId = parentIdx >= 0 ? (dn.backendNodeId[parentIdx] ?? null) : null;
|
||||||
|
const tag = str(strings, dn.nodeName?.[n]).toLowerCase();
|
||||||
|
|
||||||
|
const attrs: Record<string, string> = {};
|
||||||
|
const pairs = dn.attributes?.[n] ?? [];
|
||||||
|
for (let a = 0; a + 1 < pairs.length; a += 2) {
|
||||||
|
attrs[str(strings, pairs[a]).toLowerCase()] = str(strings, pairs[a + 1]);
|
||||||
|
}
|
||||||
|
|
||||||
|
const li = layoutByNode.get(n);
|
||||||
|
const styleRow = li === undefined ? [] : (dl?.styles?.[li] ?? []);
|
||||||
|
const styles: Record<string, string> = {};
|
||||||
|
for (const name of REQUESTED_STYLES) styles[name] = str(strings, styleRow[STYLE_COL[name]]);
|
||||||
|
const layout =
|
||||||
|
li === undefined
|
||||||
|
? undefined
|
||||||
|
: {
|
||||||
|
boundsSpace: "snapshot-document-css" as const,
|
||||||
|
clientSpace: "unscaled-client-offset-css" as const,
|
||||||
|
bounds: dl?.bounds?.[li],
|
||||||
|
clientRect: dl?.clientRects?.[li],
|
||||||
|
offsetRect: dl?.offsetRects?.[li],
|
||||||
|
scrollRect: dl?.scrollRects?.[li],
|
||||||
|
styles,
|
||||||
|
};
|
||||||
|
|
||||||
|
const textContent = nodeTextContent.get(n)?.join(" ");
|
||||||
|
|
||||||
|
const rawFormValueIndex = tag === "textarea" ? textValues.get(n) : inputValues.get(n);
|
||||||
|
const rawFormValue =
|
||||||
|
rawFormValueIndex !== undefined ? str(strings, rawFormValueIndex) : undefined;
|
||||||
|
const sensitive = isSensitiveFormControl({ tag, attrs });
|
||||||
|
const formDefaultValue = attrs.value ?? "";
|
||||||
|
const formValue = sensitive ? undefined : rawFormValue;
|
||||||
|
const formState = snapshotFormState(
|
||||||
|
rawFormValue,
|
||||||
|
formDefaultValue,
|
||||||
|
Object.prototype.hasOwnProperty.call(attrs, "value"),
|
||||||
|
sensitive,
|
||||||
|
);
|
||||||
|
if (sensitive) delete attrs.value;
|
||||||
|
|
||||||
|
nodes.push({
|
||||||
|
backendNodeId,
|
||||||
|
nodeType: dn.nodeType?.[n],
|
||||||
|
parentBackendNodeId,
|
||||||
|
...(parentIdx >= 0 && dn.backendNodeId[parentIdx] === undefined
|
||||||
|
? { parentMissing: true }
|
||||||
|
: {}),
|
||||||
|
|
||||||
|
tag,
|
||||||
|
attrs,
|
||||||
|
layout,
|
||||||
|
paintOrder: li === undefined ? 0 : (dl?.paintOrders?.[li] ?? 0),
|
||||||
|
position: styles.position || "static",
|
||||||
|
pointerEvents: styles["pointer-events"] || "auto",
|
||||||
|
cursor: styles.cursor || "auto",
|
||||||
|
|
||||||
|
textContent,
|
||||||
|
...(formValue !== undefined ? { formValue } : {}),
|
||||||
|
...(tag === "input" || tag === "textarea"
|
||||||
|
? {
|
||||||
|
formPlaceholder: attrs.placeholder ?? "",
|
||||||
|
...(!sensitive ? { formDefaultValue } : {}),
|
||||||
|
...(formState ? { formState } : {}),
|
||||||
|
}
|
||||||
|
: {}),
|
||||||
|
...(checkedInputs.has(n) ? { formValue: "true", formState: "filled" } : {}),
|
||||||
|
...(selectedOptions.has(n) ? { formValue: attrs.value ?? textContent ?? "" } : {}),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
return { nodes };
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user