Feature/auto waiting

src/decorate_browser.ts

@@ -8,7 +8,7 @@
  */
 
 import { decoratePage } from './decorate_page.js'
-import type { Decorator } from './types/main.js'
+import type { Decorator, PluginConfig } from './types/main.js'
 import { decorateContext } from './decorate_context.js'
 import type { Browser as PlayWrightBrowser, BrowserContextOptions } from 'playwright'
 
@@ -18,19 +18,20 @@ import type { Browser as PlayWrightBrowser, BrowserContextOptions } from 'playwr
  */
 export function decorateBrowser(
   browser: PlayWrightBrowser,
-  decorators: Decorator[]
+  decorators: Decorator[],
+  config: PluginConfig = {}
 ): PlayWrightBrowser {
   const originalNewContext: typeof browser.newContext = browser.newContext.bind(browser)
   const originalNewPage: typeof browser.newPage = browser.newPage.bind(browser)
 
   browser.newContext = async function (options?: BrowserContextOptions) {
     const context = await originalNewContext(options)
-    return decorateContext(context, decorators)
+    return decorateContext(context, decorators, config)
   }
 
   browser.newPage = async function (...args: Parameters<typeof browser.newPage>) {
     const page = await originalNewPage(...args)
-    return decoratePage(page, page.context(), decorators)
+    return decoratePage(page, page.context(), decorators, config)
   }
 
   return browser

src/decorate_context.ts

@@ -8,19 +8,23 @@
  */
 
 import { decoratePage } from './decorate_page.js'
-import type { Decorator } from './types/main.js'
+import type { Decorator, PluginConfig } from './types/main.js'
 import type { BrowserContext } from 'playwright'
 
 /**
  * Decorates the playwright browser context
  */
-export function decorateContext(context: BrowserContext, decorators: Decorator[]): BrowserContext {
+export function decorateContext(
+  context: BrowserContext,
+  decorators: Decorator[],
+  config: PluginConfig
+): BrowserContext {
   decorators.forEach((decorator) => {
     if (decorator.context) {
       decorator.context(context)
     }
   })
 
-  context.on('page', (page) => decoratePage(page, context, decorators))
+  context.on('page', (page) => decoratePage(page, context, decorators, config))
   return context
 }

src/decorate_page.ts

@@ -7,16 +7,21 @@
  * file that was distributed with this source code.
  */
 
-import { Decorator } from './types/main.js'
+import { Decorator, PluginConfig } from './types/main.js'
 import type { BrowserContext, Page } from 'playwright'
 
 /**
  * Decorates the playwright page object
  */
-export function decoratePage(page: Page, context: BrowserContext, decorators: Decorator[]): Page {
+export function decoratePage(
+  page: Page,
+  context: BrowserContext,
+  decorators: Decorator[],
+  config: PluginConfig
+): Page {
   decorators.forEach((decorator) => {
     if (decorator.page) {
-      decorator.page(page, context)
+      decorator.page(page, context, config)
     }
   })