testplane 8.32.2 → 8.32.3

package/build/src/browser/cdp/types.d.ts

@@ -6,7 +6,14 @@ export type CDPRequestId = number;
 export type CDPTargetId = string;
 export type CDPSessionId = string;
 export type CDPBrowserContextId = string;
-export type RuntimeScriptId = string;
+export type CDPRuntimeScriptId = string;
+export type CDPCallFrameId = string;
+export type CDPExecutionContextId = number;
+export type CDPStyleSheetId = string;
+export type CDPFrameId = string;
+export type CDPDOMBackendNodeId = number;
+export type CDPStyleSheetOrigin = "injected" | "user-agent" | "inspector" | "regular";
+export type CDPDebuggerPausedReason = "ambiguous" | "assert" | "CSPViolation" | "debugCommand" | "DOM" | "EventListener" | "exception" | "instrumentation" | "OOM" | "other" | "promiseRejection" | "XHR" | "step";
 type TargetType = "tab" | "page" | "iframe" | "worker" | "shared_worker" | "service_worker" | "worklet" | "shared_storage_worklet" | "browser" | "webview" | "other" | "auction_worklet" | "assistive_technology";
 export interface CDPRequest<T extends {
     [key in keyof T]: unknown;
@@ -61,7 +68,7 @@ interface FunctionCoverage {
 }
 export interface CDPScriptCoverage {
     /** JavaScript script id. */
-    scriptId: RuntimeScriptId;
+    scriptId: CDPRuntimeScriptId;
     /** JavaScript script name or url. */
     url: string;
     /** Functions contained in the script that has coverage data. */
@@ -69,7 +76,7 @@ export interface CDPScriptCoverage {
 }
 export interface CDPDebuggerLocation {
     /** Script identifier as reported in the Debugger.scriptParsed. */
-    scriptId: RuntimeScriptId;
+    scriptId: CDPRuntimeScriptId;
     /** Line number in the script (0-based). */
     lineNumber: number;
     /** Column number in the script (0-based). */
@@ -79,7 +86,7 @@ interface RuntimeCallFrame {
     /** JavaScript function name. */
     functionName: string;
     /** JavaScript script id. */
-    scriptId: RuntimeScriptId;
+    scriptId: CDPRuntimeScriptId;
     /** JavaScript script name or url. */
     url: string;
     /** JavaScript script line number (0-based). */
@@ -121,4 +128,76 @@ export interface CDPProfile {
      */
     timeDeltas: number[];
 }
+interface DebuggerLocation {
+    scriptId: CDPRuntimeScriptId;
+    /** Line number in the script (0-based). */
+    lineNumber: number;
+    /** Column number in the script (0-based). */
+    columnNumber?: number;
+}
+export interface CDPDebuggerCallFrame {
+    callFrameId: CDPCallFrameId;
+    functionName: string;
+    functionLocation?: DebuggerLocation;
+    location: DebuggerLocation;
+}
+export interface CDPRuntimeStackTrace {
+    description?: string;
+    callFrames: CDPDebuggerCallFrame;
+    parent?: CDPRuntimeStackTrace;
+}
+export interface CDPRuntimeRemoteObject<T = undefined> {
+    /** Object type. */
+    type: "object" | "function" | "undefined" | "string" | "number" | "boolean" | "symbol" | "bigint";
+    /** Object subtype hint. Specified for object type values only. */
+    subtype?: string;
+    /** Object class (constructor) name. Specified for object type values only. */
+    className?: string;
+    /** Remote object value in case of primitive values or JSON values (if it was requested). */
+    value: T;
+    /** Primitive value which can not be JSON-stringified does not have value, but gets this property. */
+    unserializableValue?: string;
+    /** String representation of the object. */
+    description?: string;
+}
+export interface CDPCSSStyleSheetHeader {
+    /** The stylesheet identifier. */
+    styleSheetId: CDPStyleSheetId;
+    /** Owner frame identifier. */
+    frameId: CDPFrameId;
+    /** Stylesheet resource URL. Empty if this is a constructed stylesheet created using new CSSStyleSheet() (but non-empty if this is a constructed stylesheet imported as a CSS module script). */
+    sourceURL: string;
+    /** URL of source map associated with the stylesheet (if any). */
+    sourceMapURL?: string;
+    /** Stylesheet origin. */
+    origin: CDPStyleSheetOrigin;
+    /** Stylesheet title. */
+    title: string;
+    /** The backend id for the owner node of the stylesheet. */
+    ownerNode?: CDPDOMBackendNodeId;
+    /** Denotes whether the stylesheet is disabled. */
+    disabled: boolean;
+    /** Whether the sourceURL field value comes from the sourceURL comment. */
+    hasSourceURL?: boolean;
+    /** Whether this stylesheet is created for STYLE tag by parser. This flag is not set for document.written STYLE tags. */
+    isInline: boolean;
+    /**
+     * Whether this stylesheet is mutable. Inline stylesheets become mutable after they have been modified via CSSOM API.
+     * <link> element's stylesheets become mutable only if DevTools modifies them.
+     * Constructed stylesheets (new CSSStyleSheet()) are mutable immediately after creation.
+     */
+    isMutable: boolean;
+    /** True if this stylesheet is created through new CSSStyleSheet() or imported as a CSS module script. */
+    isConstructed: boolean;
+    /** Line offset of the stylesheet within the resource (zero based). */
+    startLine: number;
+    /** Column offset of the stylesheet within the resource (zero based). */
+    startColumn: number;
+    /** Size of the content (in characters). */
+    length: number;
+    /** Line offset of the end of the stylesheet within the resource (zero based). */
+    endLine: number;
+    /** Column offset of the end of the stylesheet within the resource (zero based). */
+    endColumn: number;
+}
 export {};

package/build/src/browser/commands/index.js

@@ -13,5 +13,7 @@ exports.customCommandFileNames = [
     "moveCursorTo",
     "captureDomSnapshot",
     "waitForStaticToLoad",
+    "saveState",
+    "restoreState",
 ];
 //# sourceMappingURL=index.js.map

package/build/src/browser/commands/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/browser/commands/index.ts"],"names":[],"mappings":";;;AAAa,QAAA,sBAAsB,GAAG;IAClC,aAAa;IACb,cAAc;IACd,WAAW;IACX,cAAc;IACd,gBAAgB;IAChB,gBAAgB;IAChB,aAAa;IACb,cAAc;IACd,cAAc;IACd,oBAAoB;IACpB,qBAAqB;CACxB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/browser/commands/index.ts"],"names":[],"mappings":";;;AAAa,QAAA,sBAAsB,GAAG;IAClC,aAAa;IACb,cAAc;IACd,WAAW;IACX,cAAc;IACd,gBAAgB;IAChB,gBAAgB;IAChB,aAAa;IACb,cAAc;IACd,cAAc;IACd,oBAAoB;IACpB,qBAAqB;IACrB,WAAW;IACX,cAAc;CACjB,CAAC"}

package/build/src/browser/commands/restoreState/index.d.ts

@@ -0,0 +1,8 @@
+import type { Browser } from "../../types";
+import { SaveStateData, SaveStateOptions } from "../saveState";
+export type RestoreStateOptions = SaveStateOptions & {
+    data?: SaveStateData;
+    refresh?: boolean;
+};
+declare const _default: (browser: Browser) => void;
+export default _default;

package/build/src/browser/commands/restoreState/index.js

@@ -0,0 +1,124 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const fs_extra_1 = __importDefault(require("fs-extra"));
+const restoreStorage_1 = require("./restoreStorage");
+const logger = __importStar(require("../../../utils/logger"));
+const config_1 = require("../../../constants/config");
+const saveState_1 = require("../saveState");
+const existing_browser_1 = require("../../existing-browser");
+exports.default = (browser) => {
+    const { publicAPI: session } = browser;
+    session.addCommand("restoreState", async (_options) => {
+        const currentUrl = new URL(await session.getUrl());
+        if (!currentUrl.origin || currentUrl.origin === "null") {
+            logger.error("Before restoreState first open page using url command");
+            process.exit(1);
+        }
+        const options = { ...saveState_1.defaultOptions, refresh: true, ..._options };
+        let restoreState = options.data;
+        if (options.path) {
+            restoreState = await fs_extra_1.default.readJson(options.path);
+        }
+        if (!restoreState) {
+            logger.error("Can't restore state: please provide a path to file or data");
+            process.exit(1);
+        }
+        switch ((0, saveState_1.getCalculatedProtocol)(browser)) {
+            case config_1.WEBDRIVER_PROTOCOL: {
+                await session.switchToParentFrame();
+                if (restoreState.cookies && options.cookies) {
+                    await session.setCookies(restoreState.cookies.map(cookie => ({
+                        ...cookie,
+                        secure: cookie.secure || cookie.sameSite === "None", // fix for ff
+                        sameSite: cookie.sameSite && session.isBidi
+                            ? cookie.sameSite.toLowerCase()
+                            : cookie.sameSite,
+                    })));
+                }
+                if (restoreState.framesData) {
+                    await session.switchToParentFrame();
+                    const frames = await (0, saveState_1.getWebdriverFrames)(session);
+                    for (let i = 0; i <= frames.length; i++) {
+                        await session.switchToParentFrame();
+                        // after last element have to set data for parent frame
+                        if (i < frames.length) {
+                            await session.switchFrame(await session.$(`iframe[src="${frames[i]}"]`));
+                        }
+                        const origin = await session.execute(() => window.location.origin);
+                        const frameData = restoreState.framesData[origin];
+                        if (frameData) {
+                            if (frameData.localStorage && options.localStorage) {
+                                await session.execute(restoreStorage_1.restoreStorage, frameData.localStorage, "localStorage");
+                            }
+                            if (frameData.sessionStorage && options.sessionStorage) {
+                                await session.execute(restoreStorage_1.restoreStorage, frameData.sessionStorage, "sessionStorage");
+                            }
+                        }
+                    }
+                    await session.switchToParentFrame();
+                }
+                if (options.refresh) {
+                    await session.refresh();
+                }
+                break;
+            }
+            case config_1.DEVTOOLS_PROTOCOL: {
+                const page = await (0, existing_browser_1.getActivePuppeteerPage)(session);
+                if (!page) {
+                    return;
+                }
+                const frames = page.frames();
+                if (restoreState.cookies && options.cookies) {
+                    await page.setCookie(...restoreState.cookies);
+                }
+                for (const frame of frames) {
+                    const origin = new URL(frame.url()).origin;
+                    if (origin === "null" || !restoreState.framesData[origin]) {
+                        continue;
+                    }
+                    const frameData = restoreState.framesData[origin];
+                    if (!frameData) {
+                        continue;
+                    }
+                    if (frameData.localStorage && options.localStorage) {
+                        await frame.evaluate(restoreStorage_1.restoreStorage, frameData.localStorage, "localStorage");
+                    }
+                    if (frameData.sessionStorage && options.sessionStorage) {
+                        await frame.evaluate(restoreStorage_1.restoreStorage, frameData.sessionStorage, "sessionStorage");
+                    }
+                }
+                if (options.refresh) {
+                    await page.reload();
+                }
+                break;
+            }
+        }
+    });
+};
+//# sourceMappingURL=index.js.map

package/build/src/browser/commands/restoreState/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/browser/commands/restoreState/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,wDAA0B;AAE1B,qDAAkD;AAElD,8DAAgD;AAEhD,sDAAkF;AAClF,4CAMsB;AACtB,6DAAgE;AAQhE,kBAAe,CAAC,OAAgB,EAAQ,EAAE;IACtC,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;IAEvC,OAAO,CAAC,UAAU,CAAC,cAAc,EAAE,KAAK,EAAE,QAA6B,EAAE,EAAE;QACvE,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,MAAM,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAEnD,IAAI,CAAC,UAAU,CAAC,MAAM,IAAI,UAAU,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YACrD,MAAM,CAAC,KAAK,CAAC,uDAAuD,CAAC,CAAC;YACtE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC;QAED,MAAM,OAAO,GAAG,EAAE,GAAG,0BAAc,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,QAAQ,EAAE,CAAC;QAElE,IAAI,YAAY,GAA8B,OAAO,CAAC,IAAI,CAAC;QAE3D,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;YACf,YAAY,GAAG,MAAM,kBAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,CAAC,YAAY,EAAE,CAAC;YAChB,MAAM,CAAC,KAAK,CAAC,4DAA4D,CAAC,CAAC;YAC3E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC;QAED,QAAQ,IAAA,iCAAqB,EAAC,OAAO,CAAC,EAAE,CAAC;YACrC,KAAK,2BAAkB,CAAC,CAAC,CAAC;gBACtB,MAAM,OAAO,CAAC,mBAAmB,EAAE,CAAC;gBAEpC,IAAI,YAAY,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBAC1C,MAAM,OAAO,CAAC,UAAU,CACpB,YAAY,CAAC,OAAO,CAAC,GAAG,CACpB,MAAM,CAAC,EAAE,CACL,CAAC;wBACG,GAAG,MAAM;wBACT,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,QAAQ,KAAK,MAAM,EAAE,aAAa;wBAClE,QAAQ,EACJ,MAAM,CAAC,QAAQ,IAAI,OAAO,CAAC,MAAM;4BAC7B,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,EAAE;4BAC/B,CAAC,CAAC,MAAM,CAAC,QAAQ;qBACjB,CAAA,CACnB,CACJ,CAAC;gBACN,CAAC;gBAED,IAAI,YAAY,CAAC,UAAU,EAAE,CAAC;oBAC1B,MAAM,OAAO,CAAC,mBAAmB,EAAE,CAAC;oBAEpC,MAAM,MAAM,GAAG,MAAM,IAAA,8BAAkB,EAAC,OAAO,CAAC,CAAC;oBAEjD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;wBACtC,MAAM,OAAO,CAAC,mBAAmB,EAAE,CAAC;wBAEpC,uDAAuD;wBACvD,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;4BACpB,MAAM,OAAO,CAAC,WAAW,CAAC,MAAM,OAAO,CAAC,CAAC,CAAC,eAAe,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;wBAC7E,CAAC;wBAED,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAa,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;wBAE/E,MAAM,SAAS,GAAG,YAAY,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;wBAElD,IAAI,SAAS,EAAE,CAAC;4BACZ,IAAI,SAAS,CAAC,YAAY,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;gCACjD,MAAM,OAAO,CAAC,OAAO,CACjB,+BAAc,EACd,SAAS,CAAC,YAAY,EACtB,cAAc,CACjB,CAAC;4BACN,CAAC;4BAED,IAAI,SAAS,CAAC,cAAc,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;gCACrD,MAAM,OAAO,CAAC,OAAO,CACjB,+BAAc,EACd,SAAS,CAAC,cAAc,EACxB,gBAAgB,CACnB,CAAC;4BACN,CAAC;wBACL,CAAC;oBACL,CAAC;oBAED,MAAM,OAAO,CAAC,mBAAmB,EAAE,CAAC;gBACxC,CAAC;gBAED,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBAClB,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;gBAC5B,CAAC;gBAED,MAAM;YACV,CAAC;YACD,KAAK,0BAAiB,CAAC,CAAC,CAAC;gBACrB,MAAM,IAAI,GAAG,MAAM,IAAA,yCAAsB,EAAC,OAAO,CAAC,CAAC;gBAEnD,IAAI,CAAC,IAAI,EAAE,CAAC;oBACR,OAAO;gBACX,CAAC;gBAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;gBAE7B,IAAI,YAAY,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBAC1C,MAAM,IAAI,CAAC,SAAS,CAAC,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;gBAClD,CAAC;gBAED,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBACzB,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC;oBAE3C,IAAI,MAAM,KAAK,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;wBACxD,SAAS;oBACb,CAAC;oBAED,MAAM,SAAS,GAAG,YAAY,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;oBAElD,IAAI,CAAC,SAAS,EAAE,CAAC;wBACb,SAAS;oBACb,CAAC;oBAED,IAAI,SAAS,CAAC,YAAY,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;wBACjD,MAAM,KAAK,CAAC,QAAQ,CAChB,+BAAc,EACd,SAAS,CAAC,YAAsC,EAChD,cAAuB,CAC1B,CAAC;oBACN,CAAC;oBAED,IAAI,SAAS,CAAC,cAAc,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;wBACrD,MAAM,KAAK,CAAC,QAAQ,CAChB,+BAAc,EACd,SAAS,CAAC,cAAwC,EAClD,gBAAyB,CAC5B,CAAC;oBACN,CAAC;gBACL,CAAC;gBAED,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBAClB,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;gBACxB,CAAC;gBAED,MAAM;YACV,CAAC;QACL,CAAC;IACL,CAAC,CAAC,CAAC;AACP,CAAC,CAAC"}

package/build/src/browser/commands/restoreState/restoreStorage.d.ts

@@ -0,0 +1 @@
+export declare const restoreStorage: (data: Record<string, string>, type: "localStorage" | "sessionStorage") => void;

package/build/src/browser/commands/restoreState/restoreStorage.js

@@ -0,0 +1,12 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.restoreStorage = void 0;
+const restoreStorage = (data, type) => {
+    const storage = window[type];
+    storage.clear();
+    Object.keys(data).forEach(key => {
+        storage.setItem(key, data[key]);
+    });
+};
+exports.restoreStorage = restoreStorage;
+//# sourceMappingURL=restoreStorage.js.map

package/build/src/browser/commands/restoreState/restoreStorage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"restoreStorage.js","sourceRoot":"","sources":["../../../../../src/browser/commands/restoreState/restoreStorage.ts"],"names":[],"mappings":";;;AAAO,MAAM,cAAc,GAAG,CAAC,IAA4B,EAAE,IAAuC,EAAQ,EAAE;IAC1G,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;IAC7B,OAAO,CAAC,KAAK,EAAE,CAAC;IAEhB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;QAC5B,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;AACP,CAAC,CAAC;AAPW,QAAA,cAAc,kBAOzB"}

package/build/src/browser/commands/saveState/dumpStorage.d.ts

@@ -0,0 +1,5 @@
+export type StorageData = {
+    localStorage?: Record<string, string>;
+    sessionStorage?: Record<string, string>;
+};
+export declare const dumpStorage: () => StorageData;

package/build/src/browser/commands/saveState/dumpStorage.js

@@ -0,0 +1,29 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.dumpStorage = void 0;
+const dumpStorage = () => {
+    const getData = (storage) => {
+        const data = {};
+        for (let i = 0; i < storage.length; i++) {
+            const key = storage.key(i);
+            if (key) {
+                data[key] = storage.getItem(key);
+            }
+        }
+        return Object.keys(data).length === 0 ? undefined : data;
+    };
+    try {
+        return {
+            localStorage: getData(window?.localStorage),
+            sessionStorage: getData(window?.sessionStorage),
+        };
+    }
+    catch (error) {
+        return {
+            localStorage: undefined,
+            sessionStorage: undefined,
+        };
+    }
+};
+exports.dumpStorage = dumpStorage;
+//# sourceMappingURL=dumpStorage.js.map

package/build/src/browser/commands/saveState/dumpStorage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"dumpStorage.js","sourceRoot":"","sources":["../../../../../src/browser/commands/saveState/dumpStorage.ts"],"names":[],"mappings":";;;AAKO,MAAM,WAAW,GAAG,GAAgB,EAAE;IACzC,MAAM,OAAO,GAAG,CAAC,OAAgB,EAAsC,EAAE;QACrE,MAAM,IAAI,GAA2B,EAAE,CAAC;QAExC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAE3B,IAAI,GAAG,EAAE,CAAC;gBACN,IAAI,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAW,CAAC;YAC/C,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC;IAC7D,CAAC,CAAC;IAEF,IAAI,CAAC;QACD,OAAO;YACH,YAAY,EAAE,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC;YAC3C,cAAc,EAAE,OAAO,CAAC,MAAM,EAAE,cAAc,CAAC;SAClD,CAAC;IACN,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,OAAO;YACH,YAAY,EAAE,SAAS;YACvB,cAAc,EAAE,SAAS;SAC5B,CAAC;IACN,CAAC;AACL,CAAC,CAAC;AA1BW,QAAA,WAAW,eA0BtB"}

package/build/src/browser/commands/saveState/index.d.ts

@@ -0,0 +1,25 @@
+import type { Browser } from "../../types";
+import { StorageData } from "./dumpStorage";
+import { DEVTOOLS_PROTOCOL, WEBDRIVER_PROTOCOL } from "../../../constants/config";
+import { ExistingBrowser } from "../../existing-browser";
+import { Protocol } from "devtools-protocol";
+export type SaveStateOptions = {
+    path?: string;
+    cookies?: boolean;
+    localStorage?: boolean;
+    sessionStorage?: boolean;
+};
+export type FrameData = StorageData;
+export type SaveStateData = {
+    cookies?: Array<Protocol.Network.CookieParam>;
+    framesData: Record<string, FrameData>;
+};
+export declare const defaultOptions: {
+    cookies: boolean;
+    localStorage: boolean;
+    sessionStorage: boolean;
+};
+export declare const getCalculatedProtocol: (browser: Browser) => typeof DEVTOOLS_PROTOCOL | typeof WEBDRIVER_PROTOCOL;
+export declare const getWebdriverFrames: (session: WebdriverIO.Browser) => Promise<string[]>;
+declare const _default: (browser: ExistingBrowser) => void;
+export default _default;

package/build/src/browser/commands/saveState/index.js

@@ -0,0 +1,154 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getWebdriverFrames = exports.getCalculatedProtocol = exports.defaultOptions = void 0;
+const fs_extra_1 = __importDefault(require("fs-extra"));
+const lodash_1 = __importDefault(require("lodash"));
+const dumpStorage_1 = require("./dumpStorage");
+const config_1 = require("../../../constants/config");
+const browser_1 = require("../../../utils/browser");
+const existing_browser_1 = require("../../existing-browser");
+const logger = __importStar(require("../../../utils/logger"));
+exports.defaultOptions = {
+    cookies: true,
+    localStorage: true,
+    sessionStorage: true,
+};
+const getCalculatedProtocol = (browser) => {
+    const protocol = browser.config.automationProtocol;
+    if (protocol === config_1.DEVTOOLS_PROTOCOL) {
+        return config_1.DEVTOOLS_PROTOCOL;
+    }
+    if (protocol === config_1.WEBDRIVER_PROTOCOL &&
+        browser.config.isolation &&
+        (0, browser_1.isSupportIsolation)(browser.publicAPI.capabilities.browserName, browser.publicAPI.capabilities.browserVersion)) {
+        return config_1.DEVTOOLS_PROTOCOL;
+    }
+    return protocol;
+};
+exports.getCalculatedProtocol = getCalculatedProtocol;
+const getWebdriverFrames = async (session) => session.execute(() => Array.from(document.getElementsByTagName("iframe"))
+    .map(el => el.getAttribute("src"))
+    .filter(src => src !== null && src !== "about:blank"));
+exports.getWebdriverFrames = getWebdriverFrames;
+exports.default = (browser) => {
+    const { publicAPI: session } = browser;
+    session.addCommand("saveState", async (_options = {}) => {
+        const currentUrl = new URL(await session.getUrl());
+        if (!currentUrl.origin || currentUrl.origin === "null") {
+            logger.error("Before saveState first open page using url command");
+            process.exit(1);
+        }
+        const options = { ...exports.defaultOptions, ..._options };
+        const data = {
+            framesData: {},
+        };
+        switch ((0, exports.getCalculatedProtocol)(browser)) {
+            case config_1.WEBDRIVER_PROTOCOL: {
+                if (options.cookies) {
+                    const cookies = await session.getCookies();
+                    data.cookies = cookies.map(cookie => ({
+                        ...cookie,
+                        sameSite: cookie.sameSite ? lodash_1.default.startCase(cookie.sameSite) : cookie.sameSite,
+                    }));
+                }
+                await session.switchToParentFrame();
+                const frames = await (0, exports.getWebdriverFrames)(session);
+                const framesData = {};
+                for (let i = 0; i <= frames.length; i++) {
+                    await session.switchToParentFrame();
+                    // after last element have to get data from parent frame
+                    if (i < frames.length) {
+                        await session.switchFrame(await session.$(`iframe[src="${frames[i]}"]`));
+                    }
+                    const origin = await session.execute(() => window.location.origin);
+                    if (!origin || origin === "null" || framesData[origin]) {
+                        continue;
+                    }
+                    const frameData = {};
+                    if (options.localStorage || options.sessionStorage) {
+                        const { localStorage, sessionStorage } = await session.execute(dumpStorage_1.dumpStorage);
+                        if (localStorage && options.localStorage) {
+                            frameData.localStorage = localStorage;
+                        }
+                        if (sessionStorage && options.sessionStorage) {
+                            frameData.sessionStorage = sessionStorage;
+                        }
+                    }
+                    if (frameData.localStorage || frameData.sessionStorage) {
+                        framesData[origin] = frameData;
+                    }
+                }
+                await session.switchToParentFrame();
+                data.framesData = framesData;
+                break;
+            }
+            case config_1.DEVTOOLS_PROTOCOL: {
+                if (options.cookies) {
+                    const cookies = await browser.getAllRequestsCookies();
+                    data.cookies = cookies.map(cookie => ({
+                        ...cookie,
+                        sameSite: cookie.sameSite ? lodash_1.default.startCase(cookie.sameSite) : cookie.sameSite,
+                    }));
+                }
+                const page = await (0, existing_browser_1.getActivePuppeteerPage)(session);
+                if (!page) {
+                    break;
+                }
+                const frames = page.frames();
+                const framesData = {};
+                for (const frame of frames) {
+                    const origin = new URL(frame.url()).origin;
+                    if (origin === "null" || framesData[origin]) {
+                        continue;
+                    }
+                    const frameData = {};
+                    if (options.localStorage || options.sessionStorage) {
+                        const { localStorage, sessionStorage } = await frame.evaluate(dumpStorage_1.dumpStorage);
+                        if (localStorage && options.localStorage) {
+                            frameData.localStorage = localStorage;
+                        }
+                        if (sessionStorage && options.sessionStorage) {
+                            frameData.sessionStorage = sessionStorage;
+                        }
+                    }
+                    if (frameData.localStorage || frameData.sessionStorage) {
+                        framesData[origin] = frameData;
+                    }
+                }
+                data.framesData = framesData;
+                break;
+            }
+        }
+        if (options && options.path) {
+            await fs_extra_1.default.writeJson(options.path, data, { spaces: 2 });
+        }
+        return data;
+    });
+};
+//# sourceMappingURL=index.js.map

package/build/src/browser/commands/saveState/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/browser/commands/saveState/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,wDAA0B;AAE1B,oDAAuB;AAEvB,+CAAyD;AACzD,sDAAkF;AAClF,oDAA4D;AAC5D,6DAAiF;AAEjF,8DAAgD;AAiBnC,QAAA,cAAc,GAAG;IAC1B,OAAO,EAAE,IAAI;IACb,YAAY,EAAE,IAAI;IAClB,cAAc,EAAE,IAAI;CACvB,CAAC;AAEK,MAAM,qBAAqB,GAAG,CAAC,OAAgB,EAAwD,EAAE;IAC5G,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,kBAAkB,CAAC;IAEnD,IAAI,QAAQ,KAAK,0BAAiB,EAAE,CAAC;QACjC,OAAO,0BAAiB,CAAC;IAC7B,CAAC;IAED,IACI,QAAQ,KAAK,2BAAkB;QAC/B,OAAO,CAAC,MAAM,CAAC,SAAS;QACxB,IAAA,4BAAkB,EAAC,OAAO,CAAC,SAAS,CAAC,YAAY,CAAC,WAAY,EAAE,OAAO,CAAC,SAAS,CAAC,YAAY,CAAC,cAAe,CAAC,EACjH,CAAC;QACC,OAAO,0BAAiB,CAAC;IAC7B,CAAC;IAED,OAAO,QAAQ,CAAC;AACpB,CAAC,CAAC;AAhBW,QAAA,qBAAqB,yBAgBhC;AAEK,MAAM,kBAAkB,GAAG,KAAK,EAAE,OAA4B,EAAqB,EAAE,CACxF,OAAO,CAAC,OAAO,CAAe,GAAG,EAAE,CAC/B,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;KAC9C,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,YAAY,CAAC,KAAK,CAAW,CAAC;KAC3C,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,IAAI,IAAI,GAAG,KAAK,aAAa,CAAC,CAC5D,CAAC;AALO,QAAA,kBAAkB,sBAKzB;AAEN,kBAAe,CAAC,OAAwB,EAAQ,EAAE;IAC9C,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;IAEvC,OAAO,CAAC,UAAU,CAAC,WAAW,EAAE,KAAK,EAAE,WAA6B,EAAE,EAAsC,EAAE;QAC1G,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,MAAM,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAEnD,IAAI,CAAC,UAAU,CAAC,MAAM,IAAI,UAAU,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YACrD,MAAM,CAAC,KAAK,CAAC,oDAAoD,CAAC,CAAC;YACnE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC;QAED,MAAM,OAAO,GAAG,EAAE,GAAG,sBAAc,EAAE,GAAG,QAAQ,EAAE,CAAC;QAEnD,MAAM,IAAI,GAAkB;YACxB,UAAU,EAAE,EAAE;SACjB,CAAC;QAEF,QAAQ,IAAA,6BAAqB,EAAC,OAAO,CAAC,EAAE,CAAC;YACrC,KAAK,2BAAkB,CAAC,CAAC,CAAC;gBACtB,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBAClB,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;oBAE3C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,GAAG,CACtB,MAAM,CAAC,EAAE,CACL,CAAC;wBACG,GAAG,MAAM;wBACT,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,gBAAC,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ;qBAC5C,CAAA,CACzC,CAAC;gBACN,CAAC;gBAED,MAAM,OAAO,CAAC,mBAAmB,EAAE,CAAC;gBAEpC,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAkB,EAAC,OAAO,CAAC,CAAC;gBACjD,MAAM,UAAU,GAA8B,EAAE,CAAC;gBAEjD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBACtC,MAAM,OAAO,CAAC,mBAAmB,EAAE,CAAC;oBAEpC,wDAAwD;oBACxD,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;wBACpB,MAAM,OAAO,CAAC,WAAW,CAAC,MAAM,OAAO,CAAC,CAAC,CAAC,eAAe,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;oBAC7E,CAAC;oBAED,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAa,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;oBAE/E,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,MAAM,IAAI,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;wBACrD,SAAS;oBACb,CAAC;oBAED,MAAM,SAAS,GAAc,EAAE,CAAC;oBAEhC,IAAI,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;wBACjD,MAAM,EAAE,YAAY,EAAE,cAAc,EAAE,GAAG,MAAM,OAAO,CAAC,OAAO,CAAkB,yBAAW,CAAC,CAAC;wBAE7F,IAAI,YAAY,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;4BACvC,SAAS,CAAC,YAAY,GAAG,YAAY,CAAC;wBAC1C,CAAC;wBAED,IAAI,cAAc,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;4BAC3C,SAAS,CAAC,cAAc,GAAG,cAAc,CAAC;wBAC9C,CAAC;oBACL,CAAC;oBAED,IAAI,SAAS,CAAC,YAAY,IAAI,SAAS,CAAC,cAAc,EAAE,CAAC;wBACrD,UAAU,CAAC,MAAM,CAAC,GAAG,SAAS,CAAC;oBACnC,CAAC;gBACL,CAAC;gBAED,MAAM,OAAO,CAAC,mBAAmB,EAAE,CAAC;gBAEpC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;gBAC7B,MAAM;YACV,CAAC;YACD,KAAK,0BAAiB,CAAC,CAAC,CAAC;gBACrB,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBAClB,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,qBAAqB,EAAE,CAAC;oBAEtD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,GAAG,CACtB,MAAM,CAAC,EAAE,CACL,CAAC;wBACG,GAAG,MAAM;wBACT,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,gBAAC,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ;qBAC5C,CAAA,CACzC,CAAC;gBACN,CAAC;gBAED,MAAM,IAAI,GAAG,MAAM,IAAA,yCAAsB,EAAC,OAAO,CAAC,CAAC;gBAEnD,IAAI,CAAC,IAAI,EAAE,CAAC;oBACR,MAAM;gBACV,CAAC;gBAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;gBAE7B,MAAM,UAAU,GAA8B,EAAE,CAAC;gBAEjD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBACzB,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC;oBAE3C,IAAI,MAAM,KAAK,MAAM,IAAI,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC1C,SAAS;oBACb,CAAC;oBAED,MAAM,SAAS,GAAc,EAAE,CAAC;oBAEhC,IAAI,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;wBACjD,MAAM,EAAE,YAAY,EAAE,cAAc,EAAE,GAAgB,MAAM,KAAK,CAAC,QAAQ,CAAC,yBAAW,CAAC,CAAC;wBAExF,IAAI,YAAY,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;4BACvC,SAAS,CAAC,YAAY,GAAG,YAAY,CAAC;wBAC1C,CAAC;wBAED,IAAI,cAAc,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;4BAC3C,SAAS,CAAC,cAAc,GAAG,cAAc,CAAC;wBAC9C,CAAC;oBACL,CAAC;oBAED,IAAI,SAAS,CAAC,YAAY,IAAI,SAAS,CAAC,cAAc,EAAE,CAAC;wBACrD,UAAU,CAAC,MAAM,CAAC,GAAG,SAAS,CAAC;oBACnC,CAAC;gBACL,CAAC;gBAED,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;gBAC7B,MAAM;YACV,CAAC;QACL,CAAC;QAED,IAAI,OAAO,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;YAC1B,MAAM,kBAAE,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;QAC1D,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC,CAAC,CAAC;AACP,CAAC,CAAC"}

package/build/src/browser/existing-browser.d.ts

@@ -1,3 +1,4 @@
+import { Cookie } from "set-cookie-parser";
 import { Browser, BrowserOpts } from "./browser";
 import { Camera, PageMeta } from "./camera";
 import { type ClientBridge } from "./client-bridge";
@@ -5,6 +6,8 @@ import { Config } from "../config";
 import { Image, Rect } from "../image";
 import type { CalibrationResult, Calibrator } from "./calibrator";
 import type { SessionOptions } from "./types";
+import { Protocol } from "devtools-protocol";
+import { Page } from "puppeteer-core";
 import { CDP } from "./cdp";
 interface PrepareScreenshotOpts {
     disableAnimation?: boolean;
@@ -14,14 +17,19 @@ interface ScrollByParams {
     y: number;
     selector?: string;
 }
+export declare const getActivePuppeteerPage: (session: WebdriverIO.Browser) => Promise<Page | undefined>;
 export declare class ExistingBrowser extends Browser {
     protected _camera: Camera;
     protected _meta: Record<string, unknown>;
     protected _calibration?: CalibrationResult;
     protected _clientBridge?: ClientBridge;
     protected _cdp: CDP | null;
+    private _allCookies;
     constructor(config: Config, opts: BrowserOpts);
     init({ sessionId, sessionCaps, sessionOpts }: SessionOptions, calibrator: Calibrator): Promise<this>;
+    getCookieIndex(cookie: Cookie): string;
+    getAllRequestsCookies(): Promise<Array<Protocol.Network.CookieParam>>;
+    startCollectCookies(): Promise<void>;
     markAsBroken(): void;
     quit(): void;
     prepareScreenshot(selectors: string[] | Rect[], opts?: PrepareScreenshotOpts): Promise<unknown>;
@@ -60,5 +68,6 @@ export declare class ExistingBrowser extends Browser {
     protected _cleanupPageAnimations(): Promise<void>;
     _stubCommands(): void;
     get meta(): Record<string, unknown>;
+    get cdp(): CDP | null;
 }
 export {};