@ainias42/wdio-utils 0.0.1 → 0.0.3

package/dist/ChromeCleanupService.d.ts

@@ -1,9 +1,25 @@
-export declare class ChromeCleanupService {
+import type { Awaitable } from '@ainias42/js-helper';
+import type { Services } from '@wdio/types';
+export declare class ChromeCleanupService implements Services.ServiceInstance {
     private cacheDir;
-    constructor(_1: any, _browser: any, config: {
+    private newSessionPerTest;
+    private afterBeforeTest;
+    private myCapabilities;
+    constructor(pluginConfig: {
+        newSessionPerTest?: boolean;
+        afterBeforeTest?: () => Awaitable<void>;
+    } | undefined, _browser: any, config: {
         cacheDir: string;
     });
+    private getCacheDir;
     private sortByVersion;
     private cleanCache;
+    private updateCapabilities;
+    onPrepare(): Promise<void>;
     onComplete(): Promise<void>;
+    beforeTest(): Promise<void>;
+    beforeSession(config: any, capabilities: WebdriverIO.Capabilities | Record<string, {
+        capabilities: WebdriverIO.Capabilities;
+    }>): Promise<void>;
+    afterSession(config: any, capabilities: WebdriverIO.Capabilities | Record<string, undefined | WebdriverIO.Capabilities>): Promise<void>;
 }

package/dist/createChromeCapability.d.ts

@@ -1,4 +1,4 @@
-export declare function createChromeCapability(cacheDir: string, headless: boolean, extraArgs?: string[]): {
+export declare function createChromeCapability(cacheDir: string, headless: boolean, extraArgs?: string[], id?: string): {
     browserName: string;
     browserVersion: string;
     'goog:chromeOptions': {

package/dist/index.js

@@ -1,3 +1,5 @@
+import { ObjectHelper, Random } from "@ainias42/js-helper";
+import { driver } from "@wdio/globals";
 import { default as default_0 } from "fs/promises";
 import { default as default_1 } from "path";
 /******/ // The require scope
@@ -24,6 +26,10 @@ import { default as default_1 } from "path";
 /************************************************************************/
 var __webpack_exports__ = {};
 
+;// external "@ainias42/js-helper"
+
+;// external "@wdio/globals"
+
 ;// external "fs/promises"
 
 ;// external "path"
@@ -31,9 +37,25 @@ var __webpack_exports__ = {};
 ;// ./src/ChromeCleanupService.ts
 
 
+
+
+const browser = driver;
+function isMultiBrowser(browserInstance) {
+  return Array.isArray(browserInstance.instances);
+}
 class ChromeCleanupService {
-  constructor(_1, _browser, config) {
+  constructor(pluginConfig, _browser, config) {
+    var _a;
+    this.newSessionPerTest = true;
+    // TODO handle single capability
+    this.myCapabilities = {};
     this.cacheDir = config.cacheDir;
+    this.newSessionPerTest = (_a = pluginConfig === null || pluginConfig === void 0 ? void 0 : pluginConfig.newSessionPerTest) !== null && _a !== void 0 ? _a : this.newSessionPerTest;
+    this.afterBeforeTest = pluginConfig === null || pluginConfig === void 0 ? void 0 : pluginConfig.afterBeforeTest;
+  }
+  getCacheDir() {
+    var _a;
+    return (_a = this.cacheDir) !== null && _a !== void 0 ? _a : 'test/tmp';
   }
   // eslint-disable-next-line class-methods-use-this
   sortByVersion(items) {
@@ -78,21 +100,90 @@ class ChromeCleanupService {
       console.error(`[WDIO Cleanup] Error cleaning cache at ${cachePath}:`, error);
     }
   }
+  updateCapabilities(capabilities) {
+    var _a, _b;
+    const cacheDir = this.getCacheDir();
+    for (const cap of capabilities) {
+      const options = (_a = cap['goog:chromeOptions']) !== null && _a !== void 0 ? _a : {};
+      const args = (_b = options.args) !== null && _b !== void 0 ? _b : [];
+      if (args.every(arg => !arg.startsWith('--user-data-dir='))) {
+        args.push(`--user-data-dir=${default_1.join(cacheDir, 'profiles', Random.getStringRandom(12))}`);
+      }
+      options.args = args;
+      cap['goog:chromeOptions'] = options;
+    }
+  }
+  async onPrepare() {
+    const cacheDir = this.getCacheDir();
+    await default_0.rm(`${cacheDir}/profiles`, {
+      recursive: true,
+      force: true
+    }).catch();
+  }
   async onComplete() {
-    var _a;
-    const cacheDir = (_a = this.cacheDir) !== null && _a !== void 0 ? _a : 'test/tmp';
+    const cacheDir = this.getCacheDir();
     await this.cleanCache(`${cacheDir}/chrome`);
     await this.cleanCache(`${cacheDir}/chromedriver`);
     await default_0.rm(`${cacheDir}/profiles`, {
       recursive: true,
       force: true
-    });
+    }).catch();
+  }
+  async beforeTest() {
+    var _a, _b, _c;
+    if (!this.newSessionPerTest) {
+      return;
+    }
+    this.updateCapabilities(ObjectHelper.values(this.myCapabilities));
+    for (const cap of ObjectHelper.values(this.myCapabilities)) {
+      const options = (_a = cap['goog:chromeOptions']) !== null && _a !== void 0 ? _a : {};
+      const args = ((_b = options.args) !== null && _b !== void 0 ? _b : []).filter(value => !value.startsWith('--user-data-dir='));
+      args.push(`--profile-dir=profile_${Random.getStringRandom(12)}`);
+      options.args = args;
+      cap['goog:chromeOptions'] = options;
+    }
+    if (isMultiBrowser(browser)) {
+      const promises = browser.instances.map(instanceName =>
+      // @ts-expect-error there are typing errors here
+      browser[instanceName].reloadSession(this.myCapabilities[instanceName]));
+      await Promise.all(promises);
+    }
+    (_c = this.afterBeforeTest) === null || _c === void 0 ? void 0 : _c.call(this);
+  }
+  async beforeSession(config, capabilities) {
+    const resolvedCapabilities = [];
+    if ('browserName' in capabilities) {
+      resolvedCapabilities.push(capabilities);
+    } else {
+      resolvedCapabilities.push(...Object.values(capabilities).filter(obj => !!obj).map(cap => cap.capabilities));
+      this.myCapabilities = Object.entries(capabilities).filter(([name, obj]) => !!obj).reduce((acc, [name, cap]) => {
+        acc[name] = cap.capabilities;
+        return acc;
+      }, {});
+    }
+    this.updateCapabilities(resolvedCapabilities);
+  }
+  async afterSession(config, capabilities) {
+    const resolvedCapabilities = [];
+    if ('browserName' in capabilities) {
+      resolvedCapabilities.push(capabilities);
+    } else {
+      resolvedCapabilities.push(...Object.values(capabilities).filter(obj => !!obj));
+    }
+    const profileDirs = resolvedCapabilities.filter(cap => {
+      var _a;
+      return 'chrome' in cap && typeof cap.chrome === 'object' && cap.chrome !== null && 'userDataDir' in cap.chrome && !!((_a = cap === null || cap === void 0 ? void 0 : cap.chrome) === null || _a === void 0 ? void 0 : _a.userDataDir);
+    }).map(cap => cap.chrome.userDataDir);
+    const deletionPromises = profileDirs.map(profileDir => default_0.rm(`${profileDir}`, {
+      recursive: true,
+      force: true
+    }).catch());
+    await Promise.all(deletionPromises);
   }
 }
 ;// ./src/createChromeCapability.ts
-
-function createChromeCapability(cacheDir, headless, extraArgs = []) {
-  const args = ['--disable-infobars', '--enable-automation', '--disable-web-security', '--disable-popup-blocking', '--disable-notifications', '--disable-features=MacAppCodeSignClone', '--disable-blink-features=AutomationControlled', `--user-data-dir=${default_1.join(cacheDir, 'profiles')}`, ...extraArgs];
+function createChromeCapability(cacheDir, headless, extraArgs = [], id = 'default') {
+  const args = ['--disable-infobars', '--enable-automation', '--disable-web-security', '--disable-popup-blocking', '--disable-notifications', '--disable-features=MacAppCodeSignClone', '--disable-blink-features=AutomationControlled', ...extraArgs];
   if (headless) {
     args.push('--headless=new');
   }

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","mappings":";;SAAA;SACA;;;;;UCDA;UACA;UACA;UACA;UACA,yCAAyC,wCAAwC;UACjF;UACA;UACA,E;;;;;UCPA,wF;;;;;;;;;;;AGA6B;AACL;AAElB,MAAOE,oBAAoB;EAG7BC,YAAYC,EAAO,EAAEC,QAAa,EAAEC,MAA4B;IAC5D,IAAI,CAACC,QAAQ,GAAGD,MAAM,CAACC,QAAQ;EACnC;EAEA;EACQC,aAAaA,CAACC,KAAe;IACjC,OAAOA,KAAK,CAACC,IAAI,CAAC,CAACC,CAAC,EAAEC,CAAC,KAAI;MACvB,MAAMC,SAAS,GAAGF,CAAC,CAACG,KAAK,CAAC,mBAAmB,CAAC;MAC9C,MAAMC,SAAS,GAAGH,CAAC,CAACE,KAAK,CAAC,mBAAmB,CAAC;MAE9C,MAAME,IAAI,GAAGH,SAAS,GAAGA,SAAS,CAAC,CAAC,CAAC,GAAG,SAAS;MACjD,MAAMI,IAAI,GAAGF,SAAS,GAAGA,SAAS,CAAC,CAAC,CAAC,GAAG,SAAS;MAEjD,OAAOE,IAAI,CAACC,aAAa,CAACF,IAAI,EAAEG,SAAS,EAAE;QAAEC,OAAO,EAAE;MAAI,CAAE,CAAC;IACjE,CAAC,CAAC;EACN;EAEQ,MAAMC,UAAUA,CAACC,SAAiB;IACtC,IAAI;MACA,MAAMC,MAAM,GAAG,MAAMvB,cACZ,CAACsB,SAAS,CAAC,CACfG,IAAI,CAAC,MAAM,IAAI,CAAC,CAChBC,KAAK,CAAC,MAAM,KAAK,CAAC;MACvB,IAAI,CAACH,MAAM,EAAE;QACT;MACJ;MAEA,MAAMI,OAAO,GAAG,MAAM3B,iBAAU,CAACsB,SAAS,CAAC;MAC3C,MAAMb,KAAK,GAAa,EAAE;MAE1B,MAAMoB,YAAY,GAAGF,OAAO,CAACG,GAAG,CAAC,MAAOC,KAAK,IAAI;QAC7C,MAAMC,QAAQ,GAAG/B,cAAS,CAACqB,SAAS,EAAES,KAAK,CAAC;QAC5C,MAAMP,IAAI,GAAG,MAAMxB,cAAO,CAACgC,QAAQ,CAAC;QACpC,IAAIR,IAAI,CAACU,WAAW,EAAE,EAAE;UACpBzB,KAAK,CAAC0B,IAAI,CAACH,QAAQ,CAAC;QACxB;MACJ,CAAC,CAAC;MACF,MAAMI,OAAO,CAACC,GAAG,CAACR,YAAY,CAAC;MAE/B,IAAIpB,KAAK,CAAC6B,MAAM,IAAI,CAAC,EAAE;QACnB;MACJ;MAEA,MAAMC,WAAW,GAAG,IAAI,CAAC/B,aAAa,CAACC,KAAK,CAAC;MAC7C,MAAM,CAAC+B,MAAM,EAAE,GAAGC,QAAQ,CAAC,GAAGF,WAAW;MAEzC,MAAMG,cAAc,GAAGD,QAAQ,CAACX,GAAG,CAAEa,OAAO,IAAK3C,YAAK,CAAC2C,OAAO,EAAE;QAAEE,SAAS,EAAE,IAAI;QAAEC,KAAK,EAAE;MAAI,CAAE,CAAC,CAAC;MAClG,MAAMV,OAAO,CAACC,GAAG,CAACK,cAAc,CAAC;MAEjCK,OAAO,CAACC,GAAG,CAAC,sDAAsDR,MAAM,EAAE,CAAC;IAC/E,CAAC,CAAC,OAAOS,KAAK,EAAE;MACZF,OAAO,CAACE,KAAK,CAAC,0CAA0C3B,SAAS,GAAG,EAAE2B,KAAK,CAAC;IAChF;EACJ;EAEA,MAAMC,UAAUA,CAAA;;IACZ,MAAM3C,QAAQ,GAAG,CAAA4C,EAAA,OAAI,CAAC5C,QAAQ,cAAA4C,EAAA,cAAAA,EAAA,GAAI,UAAU;IAC5C,MAAM,IAAI,CAAC9B,UAAU,CAAC,GAAGd,QAAQ,SAAS,CAAC;IAC3C,MAAM,IAAI,CAACc,UAAU,CAAC,GAAGd,QAAQ,eAAe,CAAC;IACjD,MAAMP,YAAK,CAAC,GAAGO,QAAQ,WAAW,EAAE;MAAEsC,SAAS,EAAE,IAAI;MAAEC,KAAK,EAAE;IAAI,CAAE,CAAC;EACzE;;;AClEoB;AAElB,SAAUM,sBAAsBA,CAAC7C,QAAgB,EAAE8C,QAAiB,EAAEC,SAAA,GAAsB,EAAE;EAChG,MAAMC,IAAI,GAAG,CACT,oBAAoB,EACpB,qBAAqB,EACrB,wBAAwB,EACxB,0BAA0B,EAC1B,yBAAyB,EACzB,wCAAwC,EACxC,+CAA+C,EAC/C,mBAAmBtD,cAAS,CAACM,QAAQ,EAAE,UAAU,CAAC,EAAE,EACpD,GAAG+C,SAAS,CACf;EAED,IAAID,QAAQ,EAAE;IACVE,IAAI,CAACpB,IAAI,CAAC,gBAAgB,CAAC;EAC/B;EAEA,OAAO;IACHqB,WAAW,EAAE,QAAQ;IACrBC,cAAc,EAAE,QAAQ;IACxB,oBAAoB,EAAE;MAClBC,eAAe,EAAE,CAAC,mBAAmB,CAAC;MACtCH,IAAI;MACJI,KAAK,EAAE;QACH,uBAAuB,EAAE,IAAI;QAC7BC,0BAA0B,EAAE,KAAK;QACjC,kCAAkC,EAAE;;;GAG/C;AACL,C;;AChCuC","sources":["webpack://@ainias42/wdio-utils/webpack/bootstrap","webpack://@ainias42/wdio-utils/webpack/runtime/define property getters","webpack://@ainias42/wdio-utils/webpack/runtime/hasOwnProperty shorthand","webpack://@ainias42/wdio-utils/external module \"fs/promises\"","webpack://@ainias42/wdio-utils/external module \"path\"","webpack://@ainias42/wdio-utils/./src/ChromeCleanupService.ts","webpack://@ainias42/wdio-utils/./src/createChromeCapability.ts","webpack://@ainias42/wdio-utils/./src/index.ts"],"sourcesContent":["// The require scope\nvar __webpack_require__ = {};\n\n","// define getter functions for harmony exports\n__webpack_require__.d = (exports, definition) => {\n\tfor(var key in definition) {\n\t\tif(__webpack_require__.o(definition, key) && !__webpack_require__.o(exports, key)) {\n\t\t\tObject.defineProperty(exports, key, { enumerable: true, get: definition[key] });\n\t\t}\n\t}\n};","__webpack_require__.o = (obj, prop) => (Object.prototype.hasOwnProperty.call(obj, prop))","","","import fs from 'fs/promises';\nimport path from 'path';\n\nexport class ChromeCleanupService {\n    private cacheDir: string;\n\n    constructor(_1: any, _browser: any, config: { cacheDir: string }) {\n        this.cacheDir = config.cacheDir;\n    }\n\n    // eslint-disable-next-line class-methods-use-this\n    private sortByVersion(items: string[]) {\n        return items.sort((a, b) => {\n            const aVerMatch = a.match(/((?:\\d+\\.){3}\\d+)/);\n            const bVerMatch = b.match(/((?:\\d+\\.){3}\\d+)/);\n\n            const aVer = aVerMatch ? aVerMatch[0] : '0.0.0.0';\n            const bVer = bVerMatch ? bVerMatch[0] : '0.0.0.0';\n\n            return bVer.localeCompare(aVer, undefined, { numeric: true });\n        });\n    }\n\n    private async cleanCache(cachePath: string) {\n        try {\n            const exists = await fs\n                .stat(cachePath)\n                .then(() => true)\n                .catch(() => false);\n            if (!exists) {\n                return;\n            }\n\n            const entries = await fs.readdir(cachePath);\n            const items: string[] = [];\n\n            const itemPromises = entries.map(async (entry) => {\n                const fullPath = path.join(cachePath, entry);\n                const stat = await fs.stat(fullPath);\n                if (stat.isDirectory()) {\n                    items.push(fullPath);\n                }\n            });\n            await Promise.all(itemPromises);\n\n            if (items.length <= 1) {\n                return;\n            }\n\n            const sortedItems = this.sortByVersion(items);\n            const [latest, ...oldItems] = sortedItems;\n\n            const deletePromises = oldItems.map((oldItem) => fs.rm(oldItem, { recursive: true, force: true }));\n            await Promise.all(deletePromises);\n\n            console.log(`[WDIO Cleanup] Deleted old drivers/binaries, kept: ${latest}`);\n        } catch (error) {\n            console.error(`[WDIO Cleanup] Error cleaning cache at ${cachePath}:`, error);\n        }\n    }\n\n    async onComplete() {\n        const cacheDir = this.cacheDir ?? 'test/tmp';\n        await this.cleanCache(`${cacheDir}/chrome`);\n        await this.cleanCache(`${cacheDir}/chromedriver`);\n        await fs.rm(`${cacheDir}/profiles`, { recursive: true, force: true });\n    }\n}\n","import path from 'path';\n\nexport function createChromeCapability(cacheDir: string, headless: boolean, extraArgs: string[] = []) {\n    const args = [\n        '--disable-infobars',\n        '--enable-automation',\n        '--disable-web-security',\n        '--disable-popup-blocking',\n        '--disable-notifications',\n        '--disable-features=MacAppCodeSignClone',\n        '--disable-blink-features=AutomationControlled',\n        `--user-data-dir=${path.join(cacheDir, 'profiles')}`,\n        ...extraArgs,\n    ];\n\n    if (headless) {\n        args.push('--headless=new');\n    }\n\n    return {\n        browserName: 'chrome',\n        browserVersion: 'stable',\n        'goog:chromeOptions': {\n            excludeSwitches: ['enable-automation'],\n            args,\n            prefs: {\n                'intl.accept_languages': 'en',\n                credentials_enable_service: false,\n                'profile.password_manager_enabled': false,\n            },\n        },\n    };\n}\n","export * from './ChromeCleanupService';\nexport * from './createChromeCapability';\n"],"names":["fs","path","ChromeCleanupService","constructor","_1","_browser","config","cacheDir","sortByVersion","items","sort","a","b","aVerMatch","match","bVerMatch","aVer","bVer","localeCompare","undefined","numeric","cleanCache","cachePath","exists","stat","then","catch","entries","readdir","itemPromises","map","entry","fullPath","join","isDirectory","push","Promise","all","length","sortedItems","latest","oldItems","deletePromises","oldItem","rm","recursive","force","console","log","error","onComplete","_a","createChromeCapability","headless","extraArgs","args","browserName","browserVersion","excludeSwitches","prefs","credentials_enable_service"],"sourceRoot":""}
+{"version":3,"file":"index.js","mappings":";;;;SAAA;SACA;;;;;UCDA;UACA;UACA;UACA;UACA,yCAAyC,wCAAwC;UACjF;UACA;UACA,E;;;;;UCPA,wF;;;;;;;;;;;;;;;AKA2D;AACpB;AACV;AACL;AAYxB,MAAMK,OAAO,GAAGH,MAAiB;AAEjC,SAASI,cAAcA,CAACC,eAAwB;EAC5C,OAAOC,KAAK,CAACC,OAAO,CAACF,eAAe,CAACG,SAAS,CAAC;AACnD;AAEM,MAAOC,oBAAoB;EAO7BC,YACIC,YAAkG,EAClGC,QAAa,EACbC,MAA4B;;IARxB,KAAAC,iBAAiB,GAAY,IAAI;IAEzC;IACQ,KAAAC,cAAc,GAA8C,EAAE;IAOlE,IAAI,CAACC,QAAQ,GAAGH,MAAM,CAACG,QAAQ;IAC/B,IAAI,CAACF,iBAAiB,GAAG,CAAAG,EAAA,GAAAN,YAAY,aAAZA,YAAY,uBAAZA,YAAY,CAAEG,iBAAiB,cAAAG,EAAA,cAAAA,EAAA,GAAI,IAAI,CAACH,iBAAiB;IAClF,IAAI,CAACI,eAAe,GAAGP,YAAY,aAAZA,YAAY,uBAAZA,YAAY,CAAEO,eAAe;EACxD;EAEQC,WAAWA,CAAA;;IACf,OAAO,CAAAF,EAAA,OAAI,CAACD,QAAQ,cAAAC,EAAA,cAAAA,EAAA,GAAI,UAAU;EACtC;EAEA;EACQG,aAAaA,CAACC,KAAe;IACjC,OAAOA,KAAK,CAACC,IAAI,CAAC,CAACC,CAAC,EAAEC,CAAC,KAAI;MACvB,MAAMC,SAAS,GAAGF,CAAC,CAACG,KAAK,CAAC,mBAAmB,CAAC;MAC9C,MAAMC,SAAS,GAAGH,CAAC,CAACE,KAAK,CAAC,mBAAmB,CAAC;MAE9C,MAAME,IAAI,GAAGH,SAAS,GAAGA,SAAS,CAAC,CAAC,CAAC,GAAG,SAAS;MACjD,MAAMI,IAAI,GAAGF,SAAS,GAAGA,SAAS,CAAC,CAAC,CAAC,GAAG,SAAS;MAEjD,OAAOE,IAAI,CAACC,aAAa,CAACF,IAAI,EAAEG,SAAS,EAAE;QAAEC,OAAO,EAAE;MAAI,CAAE,CAAC;IACjE,CAAC,CAAC;EACN;EAEQ,MAAMC,UAAUA,CAACC,SAAiB;IACtC,IAAI;MACA,MAAMC,MAAM,GAAG,MAAMlC,cACZ,CAACiC,SAAS,CAAC,CACfG,IAAI,CAAC,MAAM,IAAI,CAAC,CAChBC,KAAK,CAAC,MAAM,KAAK,CAAC;MACvB,IAAI,CAACH,MAAM,EAAE;QACT;MACJ;MAEA,MAAMI,OAAO,GAAG,MAAMtC,iBAAU,CAACiC,SAAS,CAAC;MAC3C,MAAMb,KAAK,GAAa,EAAE;MAE1B,MAAMoB,YAAY,GAAGF,OAAO,CAACG,GAAG,CAAC,MAAOC,KAAK,IAAI;QAC7C,MAAMC,QAAQ,GAAG1C,cAAS,CAACgC,SAAS,EAAES,KAAK,CAAC;QAC5C,MAAMP,IAAI,GAAG,MAAMnC,cAAO,CAAC2C,QAAQ,CAAC;QACpC,IAAIR,IAAI,CAACU,WAAW,EAAE,EAAE;UACpBzB,KAAK,CAAC0B,IAAI,CAACH,QAAQ,CAAC;QACxB;MACJ,CAAC,CAAC;MACF,MAAMI,OAAO,CAACC,GAAG,CAACR,YAAY,CAAC;MAE/B,IAAIpB,KAAK,CAAC6B,MAAM,IAAI,CAAC,EAAE;QACnB;MACJ;MAEA,MAAMC,WAAW,GAAG,IAAI,CAAC/B,aAAa,CAACC,KAAK,CAAC;MAC7C,MAAM,CAAC+B,MAAM,EAAE,GAAGC,QAAQ,CAAC,GAAGF,WAAW;MAEzC,MAAMG,cAAc,GAAGD,QAAQ,CAACX,GAAG,CAAEa,OAAO,IAAKtD,YAAK,CAACsD,OAAO,EAAE;QAAEE,SAAS,EAAE,IAAI;QAAEC,KAAK,EAAE;MAAI,CAAE,CAAC,CAAC;MAClG,MAAMV,OAAO,CAACC,GAAG,CAACK,cAAc,CAAC;MAEjCK,OAAO,CAACC,GAAG,CAAC,sDAAsDR,MAAM,EAAE,CAAC;IAC/E,CAAC,CAAC,OAAOS,KAAK,EAAE;MACZF,OAAO,CAACE,KAAK,CAAC,0CAA0C3B,SAAS,GAAG,EAAE2B,KAAK,CAAC;IAChF;EACJ;EAEQC,kBAAkBA,CAACC,YAAwC;;IAC/D,MAAM/C,QAAQ,GAAG,IAAI,CAACG,WAAW,EAAE;IAEnC,KAAK,MAAM6C,GAAG,IAAID,YAAY,EAAE;MAC5B,MAAME,OAAO,GAAG,CAAAhD,EAAA,GAAA+C,GAAG,CAAC,oBAAoB,CAAC,cAAA/C,EAAA,cAAAA,EAAA,GAAI,EAAE;MAC/C,MAAMiD,IAAI,GAAG,CAAAC,EAAA,GAAAF,OAAO,CAACC,IAAI,cAAAC,EAAA,cAAAA,EAAA,GAAI,EAAE;MAC/B,IAAID,IAAI,CAACE,KAAK,CAAEC,GAAG,IAAK,CAACA,GAAG,CAACC,UAAU,CAAC,kBAAkB,CAAC,CAAC,EAAE;QAC1DJ,IAAI,CAACnB,IAAI,CAAC,mBAAmB7C,cAAS,CAACc,QAAQ,EAAE,UAAU,EAAEjB,MAAM,CAACwE,eAAe,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;MAC/F;MACAN,OAAO,CAACC,IAAI,GAAGA,IAAI;MACnBF,GAAG,CAAC,oBAAoB,CAAC,GAAGC,OAAO;IACvC;EACJ;EAEA,MAAMO,SAASA,CAAA;IACX,MAAMxD,QAAQ,GAAG,IAAI,CAACG,WAAW,EAAE;IACnC,MAAMlB,YAAK,CAAC,GAAGe,QAAQ,WAAW,EAAE;MAAEyC,SAAS,EAAE,IAAI;MAAEC,KAAK,EAAE;IAAI,CAAE,CAAC,CAACpB,KAAK,EAAE;EACjF;EAEA,MAAMmC,UAAUA,CAAA;IACZ,MAAMzD,QAAQ,GAAG,IAAI,CAACG,WAAW,EAAE;IACnC,MAAM,IAAI,CAACc,UAAU,CAAC,GAAGjB,QAAQ,SAAS,CAAC;IAC3C,MAAM,IAAI,CAACiB,UAAU,CAAC,GAAGjB,QAAQ,eAAe,CAAC;IACjD,MAAMf,YAAK,CAAC,GAAGe,QAAQ,WAAW,EAAE;MAAEyC,SAAS,EAAE,IAAI;MAAEC,KAAK,EAAE;IAAI,CAAE,CAAC,CAACpB,KAAK,EAAE;EACjF;EAEA,MAAMoC,UAAUA,CAAA;;IACZ,IAAI,CAAC,IAAI,CAAC5D,iBAAiB,EAAE;MACzB;IACJ;IACA,IAAI,CAACgD,kBAAkB,CAAChE,YAAY,CAAC6E,MAAM,CAAC,IAAI,CAAC5D,cAAc,CAAC,CAAC;IAEjE,KAAK,MAAMiD,GAAG,IAAIlE,YAAY,CAAC6E,MAAM,CAAC,IAAI,CAAC5D,cAAc,CAAC,EAAE;MACxD,MAAMkD,OAAO,GAAG,CAAAhD,EAAA,GAAA+C,GAAG,CAAC,oBAAoB,CAAC,cAAA/C,EAAA,cAAAA,EAAA,GAAI,EAAE;MAC/C,MAAMiD,IAAI,GAAG,CAAC,CAAAC,EAAA,GAAAF,OAAO,CAACC,IAAI,cAAAC,EAAA,cAAAA,EAAA,GAAI,EAAE,EAAES,MAAM,CAAEC,KAAK,IAAK,CAACA,KAAK,CAACP,UAAU,CAAC,kBAAkB,CAAC,CAAC;MAC1FJ,IAAI,CAACnB,IAAI,CAAC,yBAAyBhD,MAAM,CAACwE,eAAe,CAAC,EAAE,CAAC,EAAE,CAAC;MAChEN,OAAO,CAACC,IAAI,GAAGA,IAAI;MACnBF,GAAG,CAAC,oBAAoB,CAAC,GAAGC,OAAO;IACvC;IAEA,IAAI7D,cAAc,CAACD,OAAO,CAAC,EAAE;MACzB,MAAM2E,QAAQ,GAAG3E,OAAO,CAACK,SAAS,CAACkC,GAAG,CAAEqC,YAAY;MAChD;MACA5E,OAAO,CAAC4E,YAAY,CAAC,CAACC,aAAa,CAAC,IAAI,CAACjE,cAAc,CAACgE,YAAY,CAAC,CAAC,CACzE;MACD,MAAM/B,OAAO,CAACC,GAAG,CAAC6B,QAAQ,CAAC;IAC/B;IAEA,CAAAG,EAAA,OAAI,CAAC/D,eAAe,cAAA+D,EAAA,uBAAAA,EAAA,CAAAC,IAAA,MAAI;EAC5B;EAEA,MAAMC,aAAaA,CACftE,MAAW,EACXkD,YAAmG;IAEnG,MAAMqB,oBAAoB,GAA+B,EAAE;IAC3D,IAAI,aAAa,IAAIrB,YAAY,EAAE;MAC/BqB,oBAAoB,CAACrC,IAAI,CAACgB,YAAY,CAAC;IAC3C,CAAC,MAAM;MACHqB,oBAAoB,CAACrC,IAAI,CACrB,GAAGsC,MAAM,CAACV,MAAM,CAACZ,YAAY,CAAC,CACzBa,MAAM,CAAEU,GAAG,IAAwD,CAAC,CAACA,GAAG,CAAC,CACzE5C,GAAG,CAAEsB,GAAG,IAAKA,GAAG,CAACD,YAAY,CAAC,CACtC;MAED,IAAI,CAAChD,cAAc,GAAGsE,MAAM,CAAC9C,OAAO,CAACwB,YAAY,CAAC,CAC7Ca,MAAM,CAAC,CAAC,CAACW,IAAI,EAAED,GAAG,CAAC,KAAK,CAAC,CAACA,GAAG,CAAC,CAC9BE,MAAM,CACH,CAACC,GAAG,EAAE,CAACF,IAAI,EAAEvB,GAAG,CAAC,KAAI;QACjByB,GAAG,CAACF,IAAI,CAAC,GAAGvB,GAAG,CAACD,YAAY;QAC5B,OAAO0B,GAAG;MACd,CAAC,EACD,EAA+C,CAClD;IACT;IAEA,IAAI,CAAC3B,kBAAkB,CAACsB,oBAAoB,CAAC;EACjD;EAEA,MAAMM,YAAYA,CACd7E,MAAW,EACXkD,YAA6F;IAE7F,MAAMqB,oBAAoB,GAA+B,EAAE;IAC3D,IAAI,aAAa,IAAIrB,YAAY,EAAE;MAC/BqB,oBAAoB,CAACrC,IAAI,CAACgB,YAAY,CAAC;IAC3C,CAAC,MAAM;MACHqB,oBAAoB,CAACrC,IAAI,CACrB,GAAGsC,MAAM,CAACV,MAAM,CAACZ,YAAY,CAAC,CAACa,MAAM,CAAEU,GAAG,IAAsC,CAAC,CAACA,GAAG,CAAC,CACzF;IACL;IAEA,MAAMK,WAAW,GAAGP,oBAAoB,CACnCR,MAAM,CACFZ,GAAG,IAA2E;;MAC3E,eAAQ,IAAIA,GAAG,IACf,OAAOA,GAAG,CAAC4B,MAAM,KAAK,QAAQ,IAC9B5B,GAAG,CAAC4B,MAAM,KAAK,IAAI,IACnB,aAAa,IAAI5B,GAAG,CAAC4B,MAAM,IAC3B,CAAC,EAAC,CAAA3E,EAAA,GAAA+C,GAAG,aAAHA,GAAG,uBAAHA,GAAG,CAAE4B,MAAM,cAAA3E,EAAA,uBAAAA,EAAA,CAAE4E,WAAW;KAAA,CACjC,CACAnD,GAAG,CAAEsB,GAAG,IAAKA,GAAG,CAAC4B,MAAM,CAACC,WAAW,CAAC;IAEzC,MAAMC,gBAAgB,GAAGH,WAAW,CAACjD,GAAG,CAAEqD,UAAU,IAChD9F,YAAK,CAAC,GAAG8F,UAAU,EAAE,EAAE;MAAEtC,SAAS,EAAE,IAAI;MAAEC,KAAK,EAAE;IAAI,CAAE,CAAC,CAACpB,KAAK,EAAE,CACnE;IACD,MAAMU,OAAO,CAACC,GAAG,CAAC6C,gBAAgB,CAAC;EACvC;;;ACvME,SAAUE,sBAAsBA,CAAChF,QAAgB,EAAEiF,QAAiB,EAAEC,SAAA,GAAsB,EAAE,EAAEC,EAAE,GAAG,SAAS;EAChH,MAAMjC,IAAI,GAAG,CACT,oBAAoB,EACpB,qBAAqB,EACrB,wBAAwB,EACxB,0BAA0B,EAC1B,yBAAyB,EACzB,wCAAwC,EACxC,+CAA+C,EAE/C,GAAGgC,SAAS,CACf;EAED,IAAID,QAAQ,EAAE;IACV/B,IAAI,CAACnB,IAAI,CAAC,gBAAgB,CAAC;EAC/B;EAEA,OAAO;IACHqD,WAAW,EAAE,QAAQ;IACrBC,cAAc,EAAE,QAAQ;IACxB,oBAAoB,EAAE;MAClBC,eAAe,EAAE,CAAC,mBAAmB,CAAC;MACtCpC,IAAI;MACJqC,KAAK,EAAE;QACH,uBAAuB,EAAE,IAAI;QAC7BC,0BAA0B,EAAE,KAAK;QACjC,kCAAkC,EAAE;;;GAG/C;AACL,C;;AC/BuC","sources":["webpack://@ainias42/wdio-utils/webpack/bootstrap","webpack://@ainias42/wdio-utils/webpack/runtime/define property getters","webpack://@ainias42/wdio-utils/webpack/runtime/hasOwnProperty shorthand","webpack://@ainias42/wdio-utils/external module \"@ainias42/js-helper\"","webpack://@ainias42/wdio-utils/external module \"@wdio/globals\"","webpack://@ainias42/wdio-utils/external module \"fs/promises\"","webpack://@ainias42/wdio-utils/external module \"path\"","webpack://@ainias42/wdio-utils/./src/ChromeCleanupService.ts","webpack://@ainias42/wdio-utils/./src/createChromeCapability.ts","webpack://@ainias42/wdio-utils/./src/index.ts"],"sourcesContent":["// The require scope\nvar __webpack_require__ = {};\n\n","// define getter functions for harmony exports\n__webpack_require__.d = (exports, definition) => {\n\tfor(var key in definition) {\n\t\tif(__webpack_require__.o(definition, key) && !__webpack_require__.o(exports, key)) {\n\t\t\tObject.defineProperty(exports, key, { enumerable: true, get: definition[key] });\n\t\t}\n\t}\n};","__webpack_require__.o = (obj, prop) => (Object.prototype.hasOwnProperty.call(obj, prop))","","","","","import { ObjectHelper, Random } from '@ainias42/js-helper';\nimport { driver } from '@wdio/globals';\nimport fs from 'fs/promises';\nimport path from 'path';\nimport type { Awaitable, URecord } from '@ainias42/js-helper';\nimport type { Services } from '@wdio/types';\n\ntype SingleBrowser = {\n    capabilities: WebdriverIO.Capabilities;\n    instances?: undefined;\n};\n\ntype MultiBrowser<Names extends string = string> = Record<Names, SingleBrowser> & { instances: Names[] };\ntype Browser = MultiBrowser | SingleBrowser;\n\nconst browser = driver as Browser;\n\nfunction isMultiBrowser(browserInstance: Browser): browserInstance is MultiBrowser {\n    return Array.isArray(browserInstance.instances);\n}\n\nexport class ChromeCleanupService implements Services.ServiceInstance {\n    private cacheDir: string;\n    private newSessionPerTest: boolean = true;\n    private afterBeforeTest: (() => Awaitable<void>) | undefined;\n    // TODO handle single capability\n    private myCapabilities: URecord<string, WebdriverIO.Capabilities> = {};\n\n    constructor(\n        pluginConfig: { newSessionPerTest?: boolean; afterBeforeTest?: () => Awaitable<void> } | undefined,\n        _browser: any,\n        config: { cacheDir: string },\n    ) {\n        this.cacheDir = config.cacheDir;\n        this.newSessionPerTest = pluginConfig?.newSessionPerTest ?? this.newSessionPerTest;\n        this.afterBeforeTest = pluginConfig?.afterBeforeTest;\n    }\n\n    private getCacheDir() {\n        return this.cacheDir ?? 'test/tmp';\n    }\n\n    // eslint-disable-next-line class-methods-use-this\n    private sortByVersion(items: string[]) {\n        return items.sort((a, b) => {\n            const aVerMatch = a.match(/((?:\\d+\\.){3}\\d+)/);\n            const bVerMatch = b.match(/((?:\\d+\\.){3}\\d+)/);\n\n            const aVer = aVerMatch ? aVerMatch[0] : '0.0.0.0';\n            const bVer = bVerMatch ? bVerMatch[0] : '0.0.0.0';\n\n            return bVer.localeCompare(aVer, undefined, { numeric: true });\n        });\n    }\n\n    private async cleanCache(cachePath: string) {\n        try {\n            const exists = await fs\n                .stat(cachePath)\n                .then(() => true)\n                .catch(() => false);\n            if (!exists) {\n                return;\n            }\n\n            const entries = await fs.readdir(cachePath);\n            const items: string[] = [];\n\n            const itemPromises = entries.map(async (entry) => {\n                const fullPath = path.join(cachePath, entry);\n                const stat = await fs.stat(fullPath);\n                if (stat.isDirectory()) {\n                    items.push(fullPath);\n                }\n            });\n            await Promise.all(itemPromises);\n\n            if (items.length <= 1) {\n                return;\n            }\n\n            const sortedItems = this.sortByVersion(items);\n            const [latest, ...oldItems] = sortedItems;\n\n            const deletePromises = oldItems.map((oldItem) => fs.rm(oldItem, { recursive: true, force: true }));\n            await Promise.all(deletePromises);\n\n            console.log(`[WDIO Cleanup] Deleted old drivers/binaries, kept: ${latest}`);\n        } catch (error) {\n            console.error(`[WDIO Cleanup] Error cleaning cache at ${cachePath}:`, error);\n        }\n    }\n\n    private updateCapabilities(capabilities: WebdriverIO.Capabilities[]) {\n        const cacheDir = this.getCacheDir();\n\n        for (const cap of capabilities) {\n            const options = cap['goog:chromeOptions'] ?? {};\n            const args = options.args ?? [];\n            if (args.every((arg) => !arg.startsWith('--user-data-dir='))) {\n                args.push(`--user-data-dir=${path.join(cacheDir, 'profiles', Random.getStringRandom(12))}`);\n            }\n            options.args = args;\n            cap['goog:chromeOptions'] = options;\n        }\n    }\n\n    async onPrepare() {\n        const cacheDir = this.getCacheDir();\n        await fs.rm(`${cacheDir}/profiles`, { recursive: true, force: true }).catch();\n    }\n\n    async onComplete() {\n        const cacheDir = this.getCacheDir();\n        await this.cleanCache(`${cacheDir}/chrome`);\n        await this.cleanCache(`${cacheDir}/chromedriver`);\n        await fs.rm(`${cacheDir}/profiles`, { recursive: true, force: true }).catch();\n    }\n\n    async beforeTest() {\n        if (!this.newSessionPerTest) {\n            return;\n        }\n        this.updateCapabilities(ObjectHelper.values(this.myCapabilities));\n\n        for (const cap of ObjectHelper.values(this.myCapabilities)) {\n            const options = cap['goog:chromeOptions'] ?? {};\n            const args = (options.args ?? []).filter((value) => !value.startsWith('--user-data-dir='));\n            args.push(`--profile-dir=profile_${Random.getStringRandom(12)}`);\n            options.args = args;\n            cap['goog:chromeOptions'] = options;\n        }\n\n        if (isMultiBrowser(browser)) {\n            const promises = browser.instances.map((instanceName) =>\n                // @ts-expect-error there are typing errors here\n                browser[instanceName].reloadSession(this.myCapabilities[instanceName]),\n            );\n            await Promise.all(promises);\n        }\n\n        this.afterBeforeTest?.();\n    }\n\n    async beforeSession(\n        config: any,\n        capabilities: WebdriverIO.Capabilities | Record<string, { capabilities: WebdriverIO.Capabilities }>,\n    ) {\n        const resolvedCapabilities: WebdriverIO.Capabilities[] = [];\n        if ('browserName' in capabilities) {\n            resolvedCapabilities.push(capabilities);\n        } else {\n            resolvedCapabilities.push(\n                ...Object.values(capabilities)\n                    .filter((obj): obj is { capabilities: WebdriverIO.Capabilities } => !!obj)\n                    .map((cap) => cap.capabilities),\n            );\n\n            this.myCapabilities = Object.entries(capabilities)\n                .filter(([name, obj]) => !!obj)\n                .reduce(\n                    (acc, [name, cap]) => {\n                        acc[name] = cap.capabilities;\n                        return acc;\n                    },\n                    {} as URecord<string, WebdriverIO.Capabilities>,\n                );\n        }\n\n        this.updateCapabilities(resolvedCapabilities);\n    }\n\n    async afterSession(\n        config: any,\n        capabilities: WebdriverIO.Capabilities | Record<string, undefined | WebdriverIO.Capabilities>,\n    ) {\n        const resolvedCapabilities: WebdriverIO.Capabilities[] = [];\n        if ('browserName' in capabilities) {\n            resolvedCapabilities.push(capabilities);\n        } else {\n            resolvedCapabilities.push(\n                ...Object.values(capabilities).filter((obj): obj is WebdriverIO.Capabilities => !!obj),\n            );\n        }\n\n        const profileDirs = resolvedCapabilities\n            .filter(\n                (cap): cap is WebdriverIO.Capabilities & { chrome: { userDataDir: string } } =>\n                    'chrome' in cap &&\n                    typeof cap.chrome === 'object' &&\n                    cap.chrome !== null &&\n                    'userDataDir' in cap.chrome &&\n                    !!cap?.chrome?.userDataDir,\n            )\n            .map((cap) => cap.chrome.userDataDir);\n\n        const deletionPromises = profileDirs.map((profileDir) =>\n            fs.rm(`${profileDir}`, { recursive: true, force: true }).catch(),\n        );\n        await Promise.all(deletionPromises);\n    }\n}\n","\nexport function createChromeCapability(cacheDir: string, headless: boolean, extraArgs: string[] = [], id = 'default') {\n    const args = [\n        '--disable-infobars',\n        '--enable-automation',\n        '--disable-web-security',\n        '--disable-popup-blocking',\n        '--disable-notifications',\n        '--disable-features=MacAppCodeSignClone',\n        '--disable-blink-features=AutomationControlled',\n\n        ...extraArgs,\n    ];\n\n    if (headless) {\n        args.push('--headless=new');\n    }\n\n    return {\n        browserName: 'chrome',\n        browserVersion: 'stable',\n        'goog:chromeOptions': {\n            excludeSwitches: ['enable-automation'],\n            args,\n            prefs: {\n                'intl.accept_languages': 'en',\n                credentials_enable_service: false,\n                'profile.password_manager_enabled': false,\n            },\n        },\n    };\n}\n","export * from './ChromeCleanupService';\nexport * from './createChromeCapability';\n"],"names":["ObjectHelper","Random","driver","fs","path","browser","isMultiBrowser","browserInstance","Array","isArray","instances","ChromeCleanupService","constructor","pluginConfig","_browser","config","newSessionPerTest","myCapabilities","cacheDir","_a","afterBeforeTest","getCacheDir","sortByVersion","items","sort","a","b","aVerMatch","match","bVerMatch","aVer","bVer","localeCompare","undefined","numeric","cleanCache","cachePath","exists","stat","then","catch","entries","readdir","itemPromises","map","entry","fullPath","join","isDirectory","push","Promise","all","length","sortedItems","latest","oldItems","deletePromises","oldItem","rm","recursive","force","console","log","error","updateCapabilities","capabilities","cap","options","args","_b","every","arg","startsWith","getStringRandom","onPrepare","onComplete","beforeTest","values","filter","value","promises","instanceName","reloadSession","_c","call","beforeSession","resolvedCapabilities","Object","obj","name","reduce","acc","afterSession","profileDirs","chrome","userDataDir","deletionPromises","profileDir","createChromeCapability","headless","extraArgs","id","browserName","browserVersion","excludeSwitches","prefs","credentials_enable_service"],"sourceRoot":""}

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@ainias42/wdio-utils",
-    "version": "0.0.1",
+    "version": "0.0.3",
     "description": "Utils for WDIO",
     "publishConfig": {
         "access": "public"
@@ -26,9 +26,12 @@
     },
     "peerDependencies": {},
     "dependencies": {
-        "@ainias42/config": "*"
+        "@ainias42/config": "*",
+        "@wdio/globals": "^9.23.0",
+        "@ainias42/js-helper": ">=0.8.19"
     },
     "devDependencies": {
-        "@ainias42/webpack-config": "*"
+        "@ainias42/webpack-config": "*",
+        "@wdio/types": "^9.24.0"
     }
 }

package/src/ChromeCleanupService.ts

@@ -1,11 +1,43 @@
+import { ObjectHelper, Random } from '@ainias42/js-helper';
+import { driver } from '@wdio/globals';
 import fs from 'fs/promises';
 import path from 'path';
+import type { Awaitable, URecord } from '@ainias42/js-helper';
+import type { Services } from '@wdio/types';
 
-export class ChromeCleanupService {
+type SingleBrowser = {
+    capabilities: WebdriverIO.Capabilities;
+    instances?: undefined;
+};
+
+type MultiBrowser<Names extends string = string> = Record<Names, SingleBrowser> & { instances: Names[] };
+type Browser = MultiBrowser | SingleBrowser;
+
+const browser = driver as Browser;
+
+function isMultiBrowser(browserInstance: Browser): browserInstance is MultiBrowser {
+    return Array.isArray(browserInstance.instances);
+}
+
+export class ChromeCleanupService implements Services.ServiceInstance {
     private cacheDir: string;
+    private newSessionPerTest: boolean = true;
+    private afterBeforeTest: (() => Awaitable<void>) | undefined;
+    // TODO handle single capability
+    private myCapabilities: URecord<string, WebdriverIO.Capabilities> = {};
 
-    constructor(_1: any, _browser: any, config: { cacheDir: string }) {
+    constructor(
+        pluginConfig: { newSessionPerTest?: boolean; afterBeforeTest?: () => Awaitable<void> } | undefined,
+        _browser: any,
+        config: { cacheDir: string },
+    ) {
         this.cacheDir = config.cacheDir;
+        this.newSessionPerTest = pluginConfig?.newSessionPerTest ?? this.newSessionPerTest;
+        this.afterBeforeTest = pluginConfig?.afterBeforeTest;
+    }
+
+    private getCacheDir() {
+        return this.cacheDir ?? 'test/tmp';
     }
 
     // eslint-disable-next-line class-methods-use-this
@@ -59,10 +91,112 @@ export class ChromeCleanupService {
         }
     }
 
+    private updateCapabilities(capabilities: WebdriverIO.Capabilities[]) {
+        const cacheDir = this.getCacheDir();
+
+        for (const cap of capabilities) {
+            const options = cap['goog:chromeOptions'] ?? {};
+            const args = options.args ?? [];
+            if (args.every((arg) => !arg.startsWith('--user-data-dir='))) {
+                args.push(`--user-data-dir=${path.join(cacheDir, 'profiles', Random.getStringRandom(12))}`);
+            }
+            options.args = args;
+            cap['goog:chromeOptions'] = options;
+        }
+    }
+
+    async onPrepare() {
+        const cacheDir = this.getCacheDir();
+        await fs.rm(`${cacheDir}/profiles`, { recursive: true, force: true }).catch();
+    }
+
     async onComplete() {
-        const cacheDir = this.cacheDir ?? 'test/tmp';
+        const cacheDir = this.getCacheDir();
         await this.cleanCache(`${cacheDir}/chrome`);
         await this.cleanCache(`${cacheDir}/chromedriver`);
-        await fs.rm(`${cacheDir}/profiles`, { recursive: true, force: true });
+        await fs.rm(`${cacheDir}/profiles`, { recursive: true, force: true }).catch();
+    }
+
+    async beforeTest() {
+        if (!this.newSessionPerTest) {
+            return;
+        }
+        this.updateCapabilities(ObjectHelper.values(this.myCapabilities));
+
+        for (const cap of ObjectHelper.values(this.myCapabilities)) {
+            const options = cap['goog:chromeOptions'] ?? {};
+            const args = (options.args ?? []).filter((value) => !value.startsWith('--user-data-dir='));
+            args.push(`--profile-dir=profile_${Random.getStringRandom(12)}`);
+            options.args = args;
+            cap['goog:chromeOptions'] = options;
+        }
+
+        if (isMultiBrowser(browser)) {
+            const promises = browser.instances.map((instanceName) =>
+                // @ts-expect-error there are typing errors here
+                browser[instanceName].reloadSession(this.myCapabilities[instanceName]),
+            );
+            await Promise.all(promises);
+        }
+
+        this.afterBeforeTest?.();
+    }
+
+    async beforeSession(
+        config: any,
+        capabilities: WebdriverIO.Capabilities | Record<string, { capabilities: WebdriverIO.Capabilities }>,
+    ) {
+        const resolvedCapabilities: WebdriverIO.Capabilities[] = [];
+        if ('browserName' in capabilities) {
+            resolvedCapabilities.push(capabilities);
+        } else {
+            resolvedCapabilities.push(
+                ...Object.values(capabilities)
+                    .filter((obj): obj is { capabilities: WebdriverIO.Capabilities } => !!obj)
+                    .map((cap) => cap.capabilities),
+            );
+
+            this.myCapabilities = Object.entries(capabilities)
+                .filter(([name, obj]) => !!obj)
+                .reduce(
+                    (acc, [name, cap]) => {
+                        acc[name] = cap.capabilities;
+                        return acc;
+                    },
+                    {} as URecord<string, WebdriverIO.Capabilities>,
+                );
+        }
+
+        this.updateCapabilities(resolvedCapabilities);
+    }
+
+    async afterSession(
+        config: any,
+        capabilities: WebdriverIO.Capabilities | Record<string, undefined | WebdriverIO.Capabilities>,
+    ) {
+        const resolvedCapabilities: WebdriverIO.Capabilities[] = [];
+        if ('browserName' in capabilities) {
+            resolvedCapabilities.push(capabilities);
+        } else {
+            resolvedCapabilities.push(
+                ...Object.values(capabilities).filter((obj): obj is WebdriverIO.Capabilities => !!obj),
+            );
+        }
+
+        const profileDirs = resolvedCapabilities
+            .filter(
+                (cap): cap is WebdriverIO.Capabilities & { chrome: { userDataDir: string } } =>
+                    'chrome' in cap &&
+                    typeof cap.chrome === 'object' &&
+                    cap.chrome !== null &&
+                    'userDataDir' in cap.chrome &&
+                    !!cap?.chrome?.userDataDir,
+            )
+            .map((cap) => cap.chrome.userDataDir);
+
+        const deletionPromises = profileDirs.map((profileDir) =>
+            fs.rm(`${profileDir}`, { recursive: true, force: true }).catch(),
+        );
+        await Promise.all(deletionPromises);
     }
 }

package/src/createChromeCapability.ts

@@ -1,6 +1,5 @@
-import path from 'path';
 
-export function createChromeCapability(cacheDir: string, headless: boolean, extraArgs: string[] = []) {
+export function createChromeCapability(cacheDir: string, headless: boolean, extraArgs: string[] = [], id = 'default') {
     const args = [
         '--disable-infobars',
         '--enable-automation',
@@ -9,7 +8,7 @@ export function createChromeCapability(cacheDir: string, headless: boolean, extr
         '--disable-notifications',
         '--disable-features=MacAppCodeSignClone',
         '--disable-blink-features=AutomationControlled',
-        `--user-data-dir=${path.join(cacheDir, 'profiles')}`,
+
         ...extraArgs,
     ];
 

package/tsconfig.json

@@ -1,18 +1,16 @@
 {
-  "extends": "@ainias42/config/tsconfig.json",
-  "compilerOptions": {
-    "jsx": "react",
-    "outDir": "./dist",
-    "declaration": true
-  },
-  "exclude": [
-    "../../node_modules",
-    "dist/**/*",
-    "dist/src/**/*"
-  ],
-  "include": [
-    "src/**/*.tsx",
-    "src/**/*.ts",
-    "src/**/*.d.ts",
-  ]
+    "extends": "@ainias42/config/tsconfig.json",
+    "compilerOptions": {
+        "jsx": "react",
+        "outDir": "./dist",
+        "declaration": true,
+        "typeRoots": [
+            "node_modules/@types",
+            "../../node_modules/@types",
+            "../../node_modules/@wdio/globals",
+            "../../node_modules/webdriverio/build/@types"
+        ]
+    },
+    "exclude": ["../../node_modules", "dist/**/*", "dist/src/**/*"],
+    "include": ["src/**/*.tsx", "src/**/*.ts", "src/**/*.d.ts"]
 }