testplane 9.0.0-rc.3 → 9.0.0

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

@@ -0,0 +1,41 @@
+"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 logger = __importStar(require("../../utils/logger"));
+exports.default = (browser) => {
+    const { publicAPI: session } = browser;
+    session.addCommand("getState", async (options = browser.config.stateOpts || {}) => {
+        if (options.path) {
+            return fs_extra_1.default.readJson(options.path);
+        }
+        logger.error("Please provide the path to the state file for getState");
+        return null;
+    });
+};
+//# sourceMappingURL=getState.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"getState.js","sourceRoot":"","sources":["../../../../src/browser/commands/getState.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,wDAA0B;AAK1B,2DAA6C;AAI7C,kBAAe,CAAC,OAAgB,EAAQ,EAAE;IACtC,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;IAEvC,OAAO,CAAC,UAAU,CACd,UAAU,EACV,KAAK,EAAE,UAA2B,OAAO,CAAC,MAAM,CAAC,SAAS,IAAI,EAAE,EAAiC,EAAE;QAC/F,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;YACf,OAAO,kBAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACrC,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,wDAAwD,CAAC,CAAC;QAEvE,OAAO,IAAI,CAAC;IAChB,CAAC,CACJ,CAAC;AACN,CAAC,CAAC"}

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

@@ -12,5 +12,10 @@ exports.customCommandFileNames = [
     "switchToRepl",
     "moveCursorTo",
     "captureDomSnapshot",
+    "waitForStaticToLoad",
+    "saveState",
+    "restoreState",
+    "getState",
+    "unstable_getCdp",
 ];
 //# 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;CACvB,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;IACd,UAAU;IACV,iBAAiB;CACpB,CAAC"}

package/build/src/browser/commands/moveCursorTo.d.ts

@@ -4,6 +4,6 @@ type MoveToOptions = {
     xOffset?: number;
     yOffset?: number;
 };
-export type MoveCursorToCommand = (this: WebdriverIO.Element | ChainablePromiseElement<WebdriverIO.Element>, options: MoveToOptions) => Promise<void>;
+export type MoveCursorToCommand = (this: WebdriverIO.Element | ChainablePromiseElement<WebdriverIO.Element>, options?: MoveToOptions) => Promise<void>;
 declare const _default: (browser: Browser) => void;
 export default _default;

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

@@ -37,8 +37,26 @@ const is = {
 const makeOpenAndWaitCommand = (config, session) => async function openAndWait(uri, { selector = [], predicate, waitNetworkIdle = config.openAndWaitOpts?.waitNetworkIdle, waitNetworkIdleTimeout = config.openAndWaitOpts?.waitNetworkIdleTimeout, failOnNetworkError = config.openAndWaitOpts?.failOnNetworkError, shouldThrowError = shouldThrowErrorDefault, ignoreNetworkErrorsPatterns = config.openAndWaitOpts?.ignoreNetworkErrorsPatterns, timeout = config.openAndWaitOpts?.timeout || config?.pageLoadTimeout || 0, } = {}) {
     const PageLoader = await Promise.resolve().then(() => __importStar(require("../../utils/page-loader"))).then(m => m.default);
     const isChrome = config.desiredCapabilities?.browserName === "chrome";
-    const isCDP = config.automationProtocol === "devtools";
-    waitNetworkIdle &&= isChrome || isCDP;
+    waitNetworkIdle &&= isChrome;
+    const originalPageLoadTimeout = config.pageLoadTimeout;
+    const shouldUpdateTimeout = timeout && timeout !== originalPageLoadTimeout;
+    const setPageLoadTimeout = async (value) => {
+        if (!value) {
+            return;
+        }
+        try {
+            await session.setTimeout({ pageLoad: value });
+        }
+        catch {
+            /* */
+        }
+    };
+    const restorePageLoadTimeout = () => {
+        if (shouldUpdateTimeout && originalPageLoadTimeout) {
+            // No await because session might be stuck on url() command
+            setPageLoadTimeout(originalPageLoadTimeout).catch(() => { });
+        }
+    };
     if (!uri || uri === emptyPageUrl) {
         return new Promise(resolve => {
             session.url(uri).then(() => resolve());
@@ -55,7 +73,10 @@ const makeOpenAndWaitCommand = (config, session) => async function openAndWait(u
     let selectorsResolved = !selectors.length;
     let predicateResolved = !predicate;
     let networkResolved = !waitNetworkIdle;
-    return new Promise((resolve, reject) => {
+    if (shouldUpdateTimeout) {
+        await setPageLoadTimeout(timeout);
+    }
+    const loadPromise = new Promise((resolve, reject) => {
         const handleError = (err) => {
             reject(new Error(`url: ${err.message}`));
         };
@@ -65,7 +86,27 @@ const makeOpenAndWaitCommand = (config, session) => async function openAndWait(u
             }
         };
         const goToPage = async () => {
-            await session.url(uri);
+            const promises = [];
+            promises.push(session.url(uri, { timeout, _noFallbackOnCancel: true }).finally(() => {
+                restorePageLoadTimeout();
+            }));
+            let hardTimeoutId;
+            /* BiDi doesn't respect pageLoadTimeout, so we need to set a hard timeout and cancel previous navigation
+             * by navigating to about:blank */
+            if (session.isBidi && timeout > 0) {
+                promises.push(new Promise((_, reject) => {
+                    hardTimeoutId = setTimeout(async () => {
+                        await session.url(emptyPageUrl);
+                        reject(new Error(`openAndWait timed out after ${timeout}ms while trying to navigate to ${uri}`));
+                    }, timeout);
+                }));
+            }
+            try {
+                await Promise.race(promises);
+            }
+            finally {
+                clearTimeout(hardTimeoutId);
+            }
         };
         pageLoader.on("pageLoadError", handleError);
         pageLoader.on("selectorsError", handleError);
@@ -92,7 +133,8 @@ const makeOpenAndWaitCommand = (config, session) => async function openAndWait(u
             checkLoaded();
         });
         pageLoader.load(goToPage).then(checkLoaded);
-    }).finally(() => pageLoader.unsubscribe());
+    });
+    return loadPromise.finally(() => pageLoader.unsubscribe());
 };
 exports.default = (browser) => {
     const { publicAPI: session, config } = browser;

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

@@ -1 +1 @@
-{"version":3,"file":"openAndWait.js","sourceRoot":"","sources":["../../../../src/browser/commands/openAndWait.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,oDAAuB;AAgBvB,MAAM,YAAY,GAAG,aAAa,CAAC;AAEnC,MAAM,EAAE,GAAgD;IACpD,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,OAAO,CAAC;IACxD,UAAU,EAAE,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,UAAU,CAAC;IAChE,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC,gBAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IACxG,OAAO,EAAE,KAAK,CAAC,EAAE,CAAC,gBAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,cAAc,CAAC;CAChF,CAAC;AAEF,MAAM,sBAAsB,GAAG,CAAC,MAAqB,EAAE,OAA4B,EAAE,EAAE,CACnF,KAAK,UAAU,WAAW,CAEtB,GAAW,EACX,EACI,QAAQ,GAAG,EAAE,EACb,SAAS,EACT,eAAe,GAAG,MAAM,CAAC,eAAe,EAAE,eAAe,EACzD,sBAAsB,GAAG,MAAM,CAAC,eAAe,EAAE,sBAAsB,EACvE,kBAAkB,GAAG,MAAM,CAAC,eAAe,EAAE,kBAAkB,EAC/D,gBAAgB,GAAG,uBAAuB,EAC1C,2BAA2B,GAAG,MAAM,CAAC,eAAe,EAAE,2BAA2B,EACjF,OAAO,GAAG,MAAM,CAAC,eAAe,EAAE,OAAO,IAAI,MAAM,EAAE,eAAe,IAAI,CAAC,MAC/D,EAAE;IAEhB,MAAM,UAAU,GAAG,MAAM,kDAAO,yBAAyB,IAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IAChF,MAAM,QAAQ,GAAG,MAAM,CAAC,mBAAmB,EAAE,WAAW,KAAK,QAAQ,CAAC;IACtE,MAAM,KAAK,GAAG,MAAM,CAAC,kBAAkB,KAAK,UAAU,CAAC;IAEvD,eAAe,KAAK,QAAQ,IAAI,KAAK,CAAC;IAEtC,IAAI,CAAC,GAAG,IAAI,GAAG,KAAK,YAAY,EAAE,CAAC;QAC/B,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE;YACzB,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACP,CAAC;IAED,MAAM,SAAS,GAAG,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;IAEvE,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,OAAO,EAAE;QACvC,SAAS;QACT,SAAS;QACT,OAAO;QACP,eAAe;QACf,sBAAsB;KACzB,CAAC,CAAC;IAEH,IAAI,iBAAiB,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC;IAC1C,IAAI,iBAAiB,GAAG,CAAC,SAAS,CAAC;IACnC,IAAI,eAAe,GAAG,CAAC,eAAe,CAAC;IAEvC,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACzC,MAAM,WAAW,GAAG,CAAC,GAAU,EAAQ,EAAE;YACrC,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC7C,CAAC,CAAC;QAEF,MAAM,WAAW,GAAG,GAAS,EAAE;YAC3B,IAAI,iBAAiB,IAAI,iBAAiB,IAAI,eAAe,EAAE,CAAC;gBAC5D,OAAO,EAAE,CAAC;YACd,CAAC;QACL,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,KAAK,IAAmB,EAAE;YACvC,MAAM,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC3B,CAAC,CAAC;QAEF,UAAU,CAAC,EAAE,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC;QAC5C,UAAU,CAAC,EAAE,CAAC,gBAAgB,EAAE,WAAW,CAAC,CAAC;QAC7C,UAAU,CAAC,EAAE,CAAC,gBAAgB,EAAE,WAAW,CAAC,CAAC;QAC7C,UAAU,CAAC,EAAE,CAAC,cAAc,EAAE,KAAK,CAAC,EAAE;YAClC,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBACtB,OAAO;YACX,CAAC;YAED,MAAM,YAAY,GAAG,eAAe,CAAC,2BAA2B,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;YAE7E,IAAI,CAAC,YAAY,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC3C,MAAM,CAAC,IAAI,KAAK,CAAC,kCAAkC,KAAK,CAAC,GAAG,KAAK,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;YAC1F,CAAC;QACL,CAAC,CAAC,CAAC;QACH,UAAU,CAAC,EAAE,CAAC,gBAAgB,EAAE,GAAG,EAAE;YACjC,iBAAiB,GAAG,IAAI,CAAC;YACzB,WAAW,EAAE,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,UAAU,CAAC,EAAE,CAAC,mBAAmB,EAAE,GAAG,EAAE;YACpC,iBAAiB,GAAG,IAAI,CAAC;YACzB,WAAW,EAAE,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,UAAU,CAAC,EAAE,CAAC,iBAAiB,EAAE,GAAG,EAAE;YAClC,eAAe,GAAG,IAAI,CAAC;YACvB,WAAW,EAAE,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC;AAC/C,CAAC,CAAC;AAIN,kBAAe,CAAC,OAAgB,EAAQ,EAAE;IACtC,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;IAE/C,OAAO,CAAC,UAAU,CAAC,aAAa,EAAE,sBAAsB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;AAC/E,CAAC,CAAC;AAEF,SAAS,eAAe,CAAC,WAAmC,EAAE,EAAE,GAAW;IACvE,OAAO,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,gBAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;AACvG,CAAC;AAED,SAAS,uBAAuB,CAAC,KAAc;IAC3C,IAAI,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACpB,OAAO,KAAK,CAAC;IACjB,CAAC;IAED,OAAO,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACrE,CAAC"}
+{"version":3,"file":"openAndWait.js","sourceRoot":"","sources":["../../../../src/browser/commands/openAndWait.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,oDAAuB;AAgBvB,MAAM,YAAY,GAAG,aAAa,CAAC;AAEnC,MAAM,EAAE,GAAgD;IACpD,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,OAAO,CAAC;IACxD,UAAU,EAAE,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,UAAU,CAAC;IAChE,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC,gBAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IACxG,OAAO,EAAE,KAAK,CAAC,EAAE,CAAC,gBAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,cAAc,CAAC;CAChF,CAAC;AAEF,MAAM,sBAAsB,GAAG,CAAC,MAAqB,EAAE,OAA4B,EAAE,EAAE,CACnF,KAAK,UAAU,WAAW,CAEtB,GAAW,EACX,EACI,QAAQ,GAAG,EAAE,EACb,SAAS,EACT,eAAe,GAAG,MAAM,CAAC,eAAe,EAAE,eAAe,EACzD,sBAAsB,GAAG,MAAM,CAAC,eAAe,EAAE,sBAAsB,EACvE,kBAAkB,GAAG,MAAM,CAAC,eAAe,EAAE,kBAAkB,EAC/D,gBAAgB,GAAG,uBAAuB,EAC1C,2BAA2B,GAAG,MAAM,CAAC,eAAe,EAAE,2BAA2B,EACjF,OAAO,GAAG,MAAM,CAAC,eAAe,EAAE,OAAO,IAAI,MAAM,EAAE,eAAe,IAAI,CAAC,MAC/D,EAAE;IAEhB,MAAM,UAAU,GAAG,MAAM,kDAAO,yBAAyB,IAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IAChF,MAAM,QAAQ,GAAG,MAAM,CAAC,mBAAmB,EAAE,WAAW,KAAK,QAAQ,CAAC;IAEtE,eAAe,KAAK,QAAQ,CAAC;IAE7B,MAAM,uBAAuB,GAAG,MAAM,CAAC,eAAe,CAAC;IACvD,MAAM,mBAAmB,GAAG,OAAO,IAAI,OAAO,KAAK,uBAAuB,CAAC;IAE3E,MAAM,kBAAkB,GAAG,KAAK,EAAE,KAAoB,EAAiB,EAAE;QACrE,IAAI,CAAC,KAAK,EAAE,CAAC;YACT,OAAO;QACX,CAAC;QACD,IAAI,CAAC;YACD,MAAM,OAAO,CAAC,UAAU,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAClD,CAAC;QAAC,MAAM,CAAC;YACL,KAAK;QACT,CAAC;IACL,CAAC,CAAC;IAEF,MAAM,sBAAsB,GAAG,GAAS,EAAE;QACtC,IAAI,mBAAmB,IAAI,uBAAuB,EAAE,CAAC;YACjD,2DAA2D;YAC3D,kBAAkB,CAAC,uBAAuB,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAChE,CAAC;IACL,CAAC,CAAC;IAEF,IAAI,CAAC,GAAG,IAAI,GAAG,KAAK,YAAY,EAAE,CAAC;QAC/B,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE;YACzB,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACP,CAAC;IAED,MAAM,SAAS,GAAG,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;IAEvE,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,OAAO,EAAE;QACvC,SAAS;QACT,SAAS;QACT,OAAO;QACP,eAAe;QACf,sBAAsB;KACzB,CAAC,CAAC;IAEH,IAAI,iBAAiB,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC;IAC1C,IAAI,iBAAiB,GAAG,CAAC,SAAS,CAAC;IACnC,IAAI,eAAe,GAAG,CAAC,eAAe,CAAC;IAEvC,IAAI,mBAAmB,EAAE,CAAC;QACtB,MAAM,kBAAkB,CAAC,OAAO,CAAC,CAAC;IACtC,CAAC;IAED,MAAM,WAAW,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACtD,MAAM,WAAW,GAAG,CAAC,GAAU,EAAQ,EAAE;YACrC,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC7C,CAAC,CAAC;QAEF,MAAM,WAAW,GAAG,GAAS,EAAE;YAC3B,IAAI,iBAAiB,IAAI,iBAAiB,IAAI,eAAe,EAAE,CAAC;gBAC5D,OAAO,EAAE,CAAC;YACd,CAAC;QACL,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,KAAK,IAAmB,EAAE;YACvC,MAAM,QAAQ,GAAuB,EAAE,CAAC;YAExC,QAAQ,CAAC,IAAI,CACT,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,mBAAmB,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;gBAClE,sBAAsB,EAAE,CAAC;YAC7B,CAAC,CAAC,CACL,CAAC;YAEF,IAAI,aAAyC,CAAC;YAE9C;8CACkC;YAClC,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;gBAChC,QAAQ,CAAC,IAAI,CACT,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;oBAC7B,aAAa,GAAG,UAAU,CAAC,KAAK,IAAI,EAAE;wBAClC,MAAM,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;wBAEhC,MAAM,CACF,IAAI,KAAK,CACL,+BAA+B,OAAO,kCAAkC,GAAG,EAAE,CAChF,CACJ,CAAC;oBACN,CAAC,EAAE,OAAO,CAAC,CAAC;gBAChB,CAAC,CAAC,CACL,CAAC;YACN,CAAC;YAED,IAAI,CAAC;gBACD,MAAM,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACjC,CAAC;oBAAS,CAAC;gBACP,YAAY,CAAC,aAAa,CAAC,CAAC;YAChC,CAAC;QACL,CAAC,CAAC;QAEF,UAAU,CAAC,EAAE,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC;QAC5C,UAAU,CAAC,EAAE,CAAC,gBAAgB,EAAE,WAAW,CAAC,CAAC;QAC7C,UAAU,CAAC,EAAE,CAAC,gBAAgB,EAAE,WAAW,CAAC,CAAC;QAC7C,UAAU,CAAC,EAAE,CAAC,cAAc,EAAE,KAAK,CAAC,EAAE;YAClC,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBACtB,OAAO;YACX,CAAC;YAED,MAAM,YAAY,GAAG,eAAe,CAAC,2BAA2B,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;YAE7E,IAAI,CAAC,YAAY,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC3C,MAAM,CAAC,IAAI,KAAK,CAAC,kCAAkC,KAAK,CAAC,GAAG,KAAK,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;YAC1F,CAAC;QACL,CAAC,CAAC,CAAC;QACH,UAAU,CAAC,EAAE,CAAC,gBAAgB,EAAE,GAAG,EAAE;YACjC,iBAAiB,GAAG,IAAI,CAAC;YACzB,WAAW,EAAE,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,UAAU,CAAC,EAAE,CAAC,mBAAmB,EAAE,GAAG,EAAE;YACpC,iBAAiB,GAAG,IAAI,CAAC;YACzB,WAAW,EAAE,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,UAAU,CAAC,EAAE,CAAC,iBAAiB,EAAE,GAAG,EAAE;YAClC,eAAe,GAAG,IAAI,CAAC;YACvB,WAAW,EAAE,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,OAAO,WAAW,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC;AAC/D,CAAC,CAAC;AAIN,kBAAe,CAAC,OAAgB,EAAQ,EAAE;IACtC,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;IAE/C,OAAO,CAAC,UAAU,CAAC,aAAa,EAAE,sBAAsB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;AAC/E,CAAC,CAAC;AAEF,SAAS,eAAe,CAAC,WAAmC,EAAE,EAAE,GAAW;IACvE,OAAO,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,gBAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;AACvG,CAAC;AAED,SAAS,uBAAuB,CAAC,KAAc;IAC3C,IAAI,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACpB,OAAO,KAAK,CAAC;IACjB,CAAC;IAED,OAAO,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACrE,CAAC"}

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

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

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

@@ -0,0 +1,112 @@
+"use strict";
+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 saveState_1 = require("../saveState");
+const existing_browser_1 = require("../../existing-browser");
+const normalizeCookiePrefixConstraints = (cookie) => {
+    if (cookie.name.startsWith("__Host-")) {
+        const cookieWithoutDomain = { ...cookie };
+        delete cookieWithoutDomain.domain;
+        return { ...cookieWithoutDomain, path: "/", secure: true };
+    }
+    if (cookie.name.startsWith("__Secure-")) {
+        return { ...cookie, secure: true };
+    }
+    return cookie;
+};
+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") {
+            throw new Error("Before restoreState first open page using url command");
+        }
+        const options = { ...browser.config.stateOpts, refresh: true, ..._options };
+        let restoreState = options.data;
+        if (options.path) {
+            restoreState = await fs_extra_1.default.readJson(options.path);
+        }
+        if (!restoreState) {
+            throw new Error("Can't restore state: please provide a path to file or data");
+        }
+        if (restoreState?.cookies && options.cookieFilter) {
+            restoreState.cookies = restoreState?.cookies.filter(options.cookieFilter);
+        }
+        const cookies = restoreState.cookies?.map(normalizeCookiePrefixConstraints);
+        switch ((0, saveState_1.isBidiWithIsolation)(browser)) {
+            case false: {
+                await session.switchToParentFrame();
+                if (cookies && options.cookies) {
+                    await session.setCookies(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 true: {
+                const page = await (0, existing_browser_1.getActivePuppeteerPage)(session);
+                if (!page) {
+                    return;
+                }
+                const frames = page.frames();
+                if (cookies && options.cookies) {
+                    await page.setCookie(...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;AAGlD,4CAAsF;AACtF,6DAAgE;AAUhE,MAAM,gCAAgC,GAAG,CAAC,MAAc,EAAU,EAAE;IAChE,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QACpC,MAAM,mBAAmB,GAAG,EAAE,GAAG,MAAM,EAAE,CAAC;QAC1C,OAAO,mBAAmB,CAAC,MAAM,CAAC;QAElC,OAAO,EAAE,GAAG,mBAAmB,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IAC/D,CAAC;IAED,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;QACtC,OAAO,EAAE,GAAG,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACvC,CAAC;IAED,OAAO,MAAM,CAAC;AAClB,CAAC,CAAC;AAEF,kBAAe,CAAC,OAAgB,EAAQ,EAAE;IACtC,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;IAEvC,OAAO,CAAC,UAAU,CAAC,cAAc,EAAE,KAAK,EAAE,WAAgC,EAAE,EAAE,EAAE;QAC5E,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,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;QAC7E,CAAC;QAED,MAAM,OAAO,GAAG,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,QAAQ,EAAE,CAAC;QAE5E,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,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;QAClF,CAAC;QAED,IAAI,YAAY,EAAE,OAAO,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;YAChD,YAAY,CAAC,OAAO,GAAG,YAAY,EAAE,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAC9E,CAAC;QAED,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,EAAE,GAAG,CAAC,gCAAgC,CAAC,CAAC;QAE5E,QAAQ,IAAA,+BAAmB,EAAC,OAAO,CAAC,EAAE,CAAC;YACnC,KAAK,KAAK,CAAC,CAAC,CAAC;gBACT,MAAM,OAAO,CAAC,mBAAmB,EAAE,CAAC;gBAEpC,IAAI,OAAO,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBAC7B,MAAM,OAAO,CAAC,UAAU,CACpB,OAAO,CAAC,GAAG,CACP,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;qBACR,CAAA,CAC5B,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,IAAI,CAAC,CAAC,CAAC;gBACR,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,OAAO,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBAC7B,MAAM,IAAI,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,CAAC;gBACrC,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,13 @@
+import { StorageData } from "./dumpStorage";
+import { ExistingBrowser } from "../../existing-browser";
+import { Cookie } from "../../../types";
+import type { Browser } from "../../types";
+export type FrameData = StorageData;
+export type SaveStateData = {
+    cookies?: Array<Cookie>;
+    framesData: Record<string, FrameData>;
+};
+export declare const isBidiWithIsolation: (browser: Browser) => boolean;
+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,156 @@
+"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.isBidiWithIsolation = void 0;
+const fs_extra_1 = __importDefault(require("fs-extra"));
+const lodash_1 = __importDefault(require("lodash"));
+const dumpStorage_1 = require("./dumpStorage");
+const existing_browser_1 = require("../../existing-browser");
+const logger = __importStar(require("../../../utils/logger"));
+const events_1 = require("../../../events");
+const globalFilesToRemove_1 = require("../../../globalFilesToRemove");
+// in case when we use bidi and isolation
+// we have to use puppeteer, because we use it for creating incognito window
+const isBidiWithIsolation = (browser) => browser.publicAPI.isBidi && browser.config.isolation;
+exports.isBidiWithIsolation = isBidiWithIsolation;
+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") {
+            throw new Error("Before saveState first open page using url command");
+        }
+        const options = { ...browser.config.stateOpts, ..._options };
+        const data = {
+            framesData: {},
+        };
+        switch ((0, exports.isBidiWithIsolation)(browser)) {
+            case false: {
+                if (options.cookies) {
+                    const cookies = await session.getAllCookies();
+                    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 true: {
+                if (options.cookies) {
+                    const cookies = await session.getAllCookies();
+                    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.cookieFilter && data.cookies) {
+            data.cookies = data.cookies.filter(options.cookieFilter);
+        }
+        const dataIsEmpty = data.cookies?.length === 0 && lodash_1.default.isEmpty(data.framesData);
+        if (options && options.path && !dataIsEmpty) {
+            await fs_extra_1.default.outputJson(options.path, data, { spaces: 2 });
+            if (options.keepFile) {
+                logger.warn("\x1b[31mOption keepFile in stateOpts now is true. Please be aware that the file containing authorization data will not be automatically deleted after the tests are completed!\x1b[0m");
+            }
+            else {
+                if (process.send) {
+                    process.send({
+                        event: events_1.MasterEvents.ADD_FILE_TO_REMOVE,
+                        data: options.path,
+                    });
+                }
+                (0, globalFilesToRemove_1.addGlobalFileToRemove)(options.path);
+                browser.emitter.emit(events_1.MasterEvents.ADD_FILE_TO_REMOVE, options.path);
+            }
+        }
+        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;AACvB,+CAAyD;AACzD,6DAAiF;AACjF,8DAAgD;AAGhD,4CAA+C;AAE/C,sEAAqE;AASrE,yCAAyC;AACzC,4EAA4E;AACrE,MAAM,mBAAmB,GAAG,CAAC,OAAgB,EAAW,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC;AAA1G,QAAA,mBAAmB,uBAAuF;AAEhH,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,WAAsB,EAAE,EAAsC,EAAE;QACnG,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,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;QAC1E,CAAC;QAED,MAAM,OAAO,GAAG,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,GAAG,QAAQ,EAAE,CAAC;QAE7D,MAAM,IAAI,GAAkB;YACxB,UAAU,EAAE,EAAE;SACjB,CAAC;QAEF,QAAQ,IAAA,2BAAmB,EAAC,OAAO,CAAC,EAAE,CAAC;YACnC,KAAK,KAAK,CAAC,CAAC,CAAC;gBACT,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBAClB,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,aAAa,EAAE,CAAC;oBAE9C,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;qBAClE,CAAA,CACnB,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,IAAI,CAAC,CAAC,CAAC;gBACR,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBAClB,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,aAAa,EAAE,CAAC;oBAE9C,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;qBAClE,CAAA,CACnB,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,YAAY,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,KAAK,CAAC,IAAI,gBAAC,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAE7E,IAAI,OAAO,IAAI,OAAO,CAAC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YAC1C,MAAM,kBAAE,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;YAEvD,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;gBACnB,MAAM,CAAC,IAAI,CACP,uLAAuL,CAC1L,CAAC;YACN,CAAC;iBAAM,CAAC;gBACJ,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;oBACf,OAAO,CAAC,IAAI,CAAC;wBACT,KAAK,EAAE,qBAAY,CAAC,kBAAkB;wBACtC,IAAI,EAAE,OAAO,CAAC,IAAI;qBACrB,CAAC,CAAC;gBACP,CAAC;gBAED,IAAA,2CAAqB,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBAEpC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,qBAAY,CAAC,kBAAkB,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YACxE,CAAC;QACL,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC,CAAC,CAAC;AACP,CAAC,CAAC"}