handle multiple browser config

lost-pixel · Jan 14, 2024 · 1c62ff2 · 1c62ff2
1 parent 4bce039
commit 1c62ff2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/config.ts b/src/config.ts
@@ -4,7 +4,7 @@ import type { BrowserContextOptions, Page } from 'playwright-core';
 import z from 'zod';
 import { loadProjectConfigFile, loadTSProjectConfigFile } from './configHelper';
 import { log } from './log';
-import { ShotModeSchema } from './types';
+import { BrowserSchema, ShotModeSchema } from './types';
 
 const MaskSchema = z.object({
   /**
@@ -228,7 +228,9 @@ const BaseConfigSchema = z.object({
    * Browser to use: chromium, firefox, or webkit
    * @default 'chromium'
    */
-  browser: z.enum(['chromium', 'firefox', 'webkit']).default('chromium'),
+  browser: z
+    .union([BrowserSchema, z.array(BrowserSchema).default(['chromium'])])
+    .default('chromium'),
 
   /**
    * Enable Storybook mode

diff --git a/src/types.ts b/src/types.ts
@@ -1,6 +1,8 @@
 import z from 'zod';
 import type { BrowserContextOptions } from 'playwright-core';
 
+export const BrowserSchema = z.enum(['chromium', 'firefox', 'webkit']);
+
 export const ShotModeSchema = z.enum([
   'storybook',
   'ladle',