@squide/firefly-rsbuild-storybook 0.0.1 → 1.0.0

package/CHANGELOG.md

@@ -1,5 +1,19 @@
 # @squide/firefly-rsbuild-storybook
 
+## 1.0.0
+
+### Major Changes
+
+- [#352](https://github.com/workleap/wl-squide/pull/352) [`7bd7af5`](https://github.com/workleap/wl-squide/commit/7bd7af5a757912309704b9e93bf8ecf1388e6787) Thanks [@patricklafrance](https://github.com/patricklafrance)! - Added the `withFeatureFlagOverrideDecorator` hook and first major release.
+
+### Patch Changes
+
+- Updated dependencies [[`7bd7af5`](https://github.com/workleap/wl-squide/commit/7bd7af5a757912309704b9e93bf8ecf1388e6787), [`7bd7af5`](https://github.com/workleap/wl-squide/commit/7bd7af5a757912309704b9e93bf8ecf1388e6787), [`7bd7af5`](https://github.com/workleap/wl-squide/commit/7bd7af5a757912309704b9e93bf8ecf1388e6787)]:
+  - @squide/env-vars@1.4.10
+  - @squide/msw@4.0.8
+  - @squide/launch-darkly@1.0.0
+  - @squide/firefly@16.1.0
+
 ## 0.0.1
 
 ### Patch Changes

package/dist/index.d.ts

@@ -1,3 +1,4 @@
 export { initializeFireflyForStorybook, type InitializeFireflyForStorybookOptions } from "./initializeFireflyForStorybook.ts";
 export { StorybookRuntime, StorybookRuntimeScope } from "./StorybookRuntime.ts";
+export { withFeatureFlagsOverrideDecorator } from "./withFeatureFlagsOverrideDecorator.tsx";
 export { FireflyDecorator, withFireflyDecorator, type FireflyDecoratorProps } from "./withFireflyDecorator.tsx";

package/dist/index.js

@@ -1,11 +1,14 @@
 import { initializeFireflyForStorybook } from "./initializeFireflyForStorybook.js";
 import { StorybookRuntime, StorybookRuntimeScope } from "./StorybookRuntime.js";
+import { withFeatureFlagsOverrideDecorator } from "./withFeatureFlagsOverrideDecorator.js";
 import { FireflyDecorator, withFireflyDecorator } from "./withFireflyDecorator.js";
 
 ;// CONCATENATED MODULE: external "./initializeFireflyForStorybook.js"
 
 ;// CONCATENATED MODULE: external "./StorybookRuntime.js"
 
+;// CONCATENATED MODULE: external "./withFeatureFlagsOverrideDecorator.js"
+
 ;// CONCATENATED MODULE: external "./withFireflyDecorator.js"
 
 ;// CONCATENATED MODULE: ./src/index.ts
@@ -13,6 +16,7 @@ import { FireflyDecorator, withFireflyDecorator } from "./withFireflyDecorator.j
 
 
 
-export { FireflyDecorator, StorybookRuntime, StorybookRuntimeScope, initializeFireflyForStorybook, withFireflyDecorator };
+
+export { FireflyDecorator, StorybookRuntime, StorybookRuntimeScope, initializeFireflyForStorybook, withFeatureFlagsOverrideDecorator, withFireflyDecorator };
 
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sources":["../src/index.ts"],"sourcesContent":["export { initializeFireflyForStorybook, type InitializeFireflyForStorybookOptions } from \"./initializeFireflyForStorybook.ts\";\nexport { StorybookRuntime, StorybookRuntimeScope } from \"./StorybookRuntime.ts\";\nexport { FireflyDecorator, withFireflyDecorator, type FireflyDecoratorProps } from \"./withFireflyDecorator.tsx\";\n\n"],"names":["initializeFireflyForStorybook","StorybookRuntime","StorybookRuntimeScope","FireflyDecorator","withFireflyDecorator"],"mappings":";;;;;;;;;;;AAA8H;AAC9C;AACgC"}
+{"version":3,"file":"index.js","sources":["../src/index.ts"],"sourcesContent":["export { initializeFireflyForStorybook, type InitializeFireflyForStorybookOptions } from \"./initializeFireflyForStorybook.ts\";\nexport { StorybookRuntime, StorybookRuntimeScope } from \"./StorybookRuntime.ts\";\nexport { withFeatureFlagsOverrideDecorator } from \"./withFeatureFlagsOverrideDecorator.tsx\";\nexport { FireflyDecorator, withFireflyDecorator, type FireflyDecoratorProps } from \"./withFireflyDecorator.tsx\";\n\n"],"names":["initializeFireflyForStorybook","StorybookRuntime","StorybookRuntimeScope","withFeatureFlagsOverrideDecorator","FireflyDecorator","withFireflyDecorator"],"mappings":";;;;;;;;;;;;;;AAA8H;AAC9C;AACY;AACoB"}

package/dist/initializeFireflyForStorybook.d.ts

@@ -1,8 +1,11 @@
 import { EnvironmentVariables } from "@squide/env-vars";
 import { FireflyRuntime, ModuleRegisterFunction } from "@squide/firefly";
+import { LDClient, LDFlagValue } from "launchdarkly-js-client-sdk";
 import { StorybookRuntime } from "./StorybookRuntime.ts";
 export interface InitializeFireflyForStorybookOptions {
     localModules?: ModuleRegisterFunction<FireflyRuntime>[];
     environmentVariables?: EnvironmentVariables;
+    featureFlags?: Map<string, LDFlagValue> | Record<string, LDFlagValue>;
+    launchDarklyClient?: LDClient;
 }
 export declare function initializeFireflyForStorybook(options?: InitializeFireflyForStorybookOptions): Promise<StorybookRuntime>;

package/dist/initializeFireflyForStorybook.js

@@ -1,5 +1,5 @@
 import { EnvironmentVariablesPlugin } from "@squide/env-vars";
-import { toLocalModuleDefinitions } from "@squide/firefly";
+import { InMemoryLaunchDarklyClient, LaunchDarklyPlugin, toLocalModuleDefinitions } from "@squide/firefly";
 import { MswPlugin } from "@squide/msw";
 import { StorybookRuntime } from "./StorybookRuntime.js";
 
@@ -17,19 +17,25 @@ import { StorybookRuntime } from "./StorybookRuntime.js";
 
 
 async function initializeFireflyForStorybook(options = {}) {
-    const { localModules = [], environmentVariables = {} } = options;
+    const { localModules, environmentVariables, featureFlags = {}, launchDarklyClient } = options;
     const runtime = new StorybookRuntime({
         mode: "development",
         plugins: [
             (x)=>new MswPlugin(x),
             (x)=>new EnvironmentVariablesPlugin(x, {
                     variables: environmentVariables
-                })
+                }),
+            (x)=>new LaunchDarklyPlugin(x, // 1- If no client is provided create it.
+                // 2- If feature flags are provided and it's an instance of Map, use the argument.
+                // 3- If feature flags are provided (or the default value) and it's an object literal, convert the object to a Map instance.
+                launchDarklyClient ?? new InMemoryLaunchDarklyClient(featureFlags instanceof Map ? featureFlags : new Map(Object.entries(featureFlags))))
         ]
     });
-    await runtime.moduleManager.registerModules([
-        ...toLocalModuleDefinitions(localModules)
-    ]);
+    if (localModules && localModules.length > 0) {
+        await runtime.moduleManager.registerModules([
+            ...toLocalModuleDefinitions(localModules)
+        ]);
+    }
     return runtime;
 }
 

package/dist/initializeFireflyForStorybook.js.map

@@ -1 +1 @@
-{"version":3,"file":"initializeFireflyForStorybook.js","sources":["../src/initializeFireflyForStorybook.ts"],"sourcesContent":["import { EnvironmentVariables, EnvironmentVariablesPlugin } from \"@squide/env-vars\";\nimport { FireflyRuntime, ModuleRegisterFunction, toLocalModuleDefinitions } from \"@squide/firefly\";\nimport { MswPlugin } from \"@squide/msw\";\nimport { StorybookRuntime } from \"./StorybookRuntime.ts\";\n\nexport interface InitializeFireflyForStorybookOptions {\n    localModules?: ModuleRegisterFunction<FireflyRuntime>[];\n    environmentVariables?: EnvironmentVariables;\n}\n\nexport async function initializeFireflyForStorybook(options: InitializeFireflyForStorybookOptions = {}) {\n    const {\n        localModules = [],\n        environmentVariables = {}\n    } = options;\n\n    const runtime = new StorybookRuntime({\n        mode: \"development\",\n        plugins: [\n            x => new MswPlugin(x),\n            x => new EnvironmentVariablesPlugin(x, {\n                variables: environmentVariables\n            })\n        ]\n    });\n\n    await runtime.moduleManager.registerModules([\n        ...toLocalModuleDefinitions(localModules)\n    ]);\n\n    return runtime;\n}\n"],"names":["EnvironmentVariablesPlugin","toLocalModuleDefinitions","MswPlugin","StorybookRuntime","initializeFireflyForStorybook","options","localModules","environmentVariables","runtime","x"],"mappings":";;;;;;;;;;;;;;AAAoF;AACe;AAC3D;AACiB;AAOlD,eAAeI,8BAA8BC,UAAgD,CAAC,CAAC;IAClG,MAAM,EACFC,eAAe,EAAE,EACjBC,uBAAuB,CAAC,CAAC,EAC5B,GAAGF;IAEJ,MAAMG,UAAU,IAAIL,gBAAgBA,CAAC;QACjC,MAAM;QACN,SAAS;YACLM,CAAAA,IAAK,IAAIP,SAASA,CAACO;YACnBA,CAAAA,IAAK,IAAIT,0BAA0BA,CAACS,GAAG;oBACnC,WAAWF;gBACf;SACH;IACL;IAEA,MAAMC,QAAQ,aAAa,CAAC,eAAe,CAAC;WACrCP,wBAAwBA,CAACK;KAC/B;IAED,OAAOE;AACX"}
+{"version":3,"file":"initializeFireflyForStorybook.js","sources":["../src/initializeFireflyForStorybook.ts"],"sourcesContent":["import { EnvironmentVariables, EnvironmentVariablesPlugin } from \"@squide/env-vars\";\nimport { FireflyRuntime, InMemoryLaunchDarklyClient, LaunchDarklyPlugin, ModuleRegisterFunction, toLocalModuleDefinitions } from \"@squide/firefly\";\nimport { MswPlugin } from \"@squide/msw\";\nimport { LDClient, LDFlagValue } from \"launchdarkly-js-client-sdk\";\nimport { StorybookRuntime } from \"./StorybookRuntime.ts\";\n\nexport interface InitializeFireflyForStorybookOptions {\n    localModules?: ModuleRegisterFunction<FireflyRuntime>[];\n    environmentVariables?: EnvironmentVariables;\n    featureFlags?: Map<string, LDFlagValue> | Record<string, LDFlagValue>;\n    launchDarklyClient?: LDClient;\n}\n\nexport async function initializeFireflyForStorybook(options: InitializeFireflyForStorybookOptions = {}) {\n    const {\n        localModules,\n        environmentVariables,\n        featureFlags = {},\n        launchDarklyClient\n    } = options;\n\n    const runtime = new StorybookRuntime({\n        mode: \"development\",\n        plugins: [\n            x => new MswPlugin(x),\n            x => new EnvironmentVariablesPlugin(x, {\n                variables: environmentVariables\n            }),\n            x => new LaunchDarklyPlugin(\n                x,\n                // 1- If no client is provided create it.\n                // 2- If feature flags are provided and it's an instance of Map, use the argument.\n                // 3- If feature flags are provided (or the default value) and it's an object literal, convert the object to a Map instance.\n                launchDarklyClient ?? new InMemoryLaunchDarklyClient(featureFlags instanceof Map\n                    ? featureFlags\n                    : new Map<string, LDFlagValue>(Object.entries(featureFlags))\n                )\n            )\n        ]\n    });\n\n    if (localModules && localModules.length > 0) {\n        await runtime.moduleManager.registerModules([\n            ...toLocalModuleDefinitions(localModules)\n        ]);\n    }\n\n    return runtime;\n}\n"],"names":["EnvironmentVariablesPlugin","InMemoryLaunchDarklyClient","LaunchDarklyPlugin","toLocalModuleDefinitions","MswPlugin","StorybookRuntime","initializeFireflyForStorybook","options","localModules","environmentVariables","featureFlags","launchDarklyClient","runtime","x","Map","Object"],"mappings":";;;;;;;;;;;;;;AAAoF;AAC+D;AAC3G;AAEiB;AASlD,eAAeM,8BAA8BC,UAAgD,CAAC,CAAC;IAClG,MAAM,EACFC,YAAY,EACZC,oBAAoB,EACpBC,eAAe,CAAC,CAAC,EACjBC,kBAAkB,EACrB,GAAGJ;IAEJ,MAAMK,UAAU,IAAIP,gBAAgBA,CAAC;QACjC,MAAM;QACN,SAAS;YACLQ,CAAAA,IAAK,IAAIT,SAASA,CAACS;YACnBA,CAAAA,IAAK,IAAIb,0BAA0BA,CAACa,GAAG;oBACnC,WAAWJ;gBACf;YACAI,CAAAA,IAAK,IAAIX,kBAAkBA,CACvBW,GACA,yCAAyC;gBACzC,kFAAkF;gBAClF,4HAA4H;gBAC5HF,sBAAsB,IAAIV,0BAA0BA,CAACS,wBAAwBI,MACvEJ,eACA,IAAII,IAAyBC,OAAO,OAAO,CAACL;SAGzD;IACL;IAEA,IAAIF,gBAAgBA,aAAa,MAAM,GAAG,GAAG;QACzC,MAAMI,QAAQ,aAAa,CAAC,eAAe,CAAC;eACrCT,wBAAwBA,CAACK;SAC/B;IACL;IAEA,OAAOI;AACX"}

package/dist/withFeatureFlagsOverrideDecorator.d.ts

@@ -0,0 +1,3 @@
+import { LDFlagValue } from "launchdarkly-js-client-sdk";
+import { Decorator } from "storybook-react-rsbuild";
+export declare function withFeatureFlagsOverrideDecorator<const T extends string>(featureFlags: Map<T, LDFlagValue>, overrides: Partial<Record<T, LDFlagValue>>): Decorator;

package/dist/withFeatureFlagsOverrideDecorator.js

@@ -0,0 +1,47 @@
+import { jsx } from "react/jsx-runtime";
+import { useLayoutEffect } from "react";
+
+;// CONCATENATED MODULE: external "react/jsx-runtime"
+
+;// CONCATENATED MODULE: external "react"
+
+;// CONCATENATED MODULE: ./src/withFeatureFlagsOverrideDecorator.tsx
+
+
+function OverrideFeatureFlags(props) {
+    const { featureFlags, overrides, children } = props;
+    // Must use a layout effect to override the feature flags before the story renders.
+    useLayoutEffect(()=>{
+        const originalValues = {};
+        Object.entries(overrides).forEach(([key, value])=>{
+            originalValues[key] = featureFlags.get(key);
+            featureFlags.set(key, value);
+        });
+        return ()=>{
+            // Reset the feature flags to the original values.
+            Object.entries(originalValues).forEach(([key, value])=>{
+                featureFlags.set(key, value);
+            });
+        };
+    }, [
+        featureFlags,
+        overrides
+    ]);
+    return children;
+}
+function withFeatureFlagsOverrideDecorator(featureFlags, overrides) {
+    if (!(featureFlags instanceof Map)) {
+        throw new TypeError("[squide] The \"featureFlags\" argument must be a Map instance.");
+    }
+    return (story)=>{
+        return /*#__PURE__*/ jsx(OverrideFeatureFlags, {
+            featureFlags: featureFlags,
+            overrides: overrides,
+            children: story()
+        });
+    };
+}
+
+export { withFeatureFlagsOverrideDecorator };
+
+//# sourceMappingURL=withFeatureFlagsOverrideDecorator.js.map

package/dist/withFeatureFlagsOverrideDecorator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"withFeatureFlagsOverrideDecorator.js","sources":["../src/withFeatureFlagsOverrideDecorator.tsx"],"sourcesContent":["import { LDFlagValue } from \"launchdarkly-js-client-sdk\";\nimport { PropsWithChildren, useLayoutEffect } from \"react\";\nimport { Decorator } from \"storybook-react-rsbuild\";\n\ninterface OverrideFeatureFlagsProps extends PropsWithChildren {\n    featureFlags: Map<string, LDFlagValue>;\n    overrides: Record<string, LDFlagValue>;\n}\n\nfunction OverrideFeatureFlags(props: OverrideFeatureFlagsProps) {\n    const {\n        featureFlags,\n        overrides,\n        children\n    } = props;\n\n    // Must use a layout effect to override the feature flags before the story renders.\n    useLayoutEffect(() => {\n        const originalValues: Record<string, LDFlagValue> = {};\n\n        Object.entries(overrides).forEach(([key, value]) => {\n            originalValues[key] = featureFlags.get(key);\n            featureFlags.set(key, value);\n        });\n\n        return () => {\n            // Reset the feature flags to the original values.\n            Object.entries(originalValues).forEach(([key, value]) => {\n                featureFlags.set(key, value);\n            });\n        };\n    }, [featureFlags, overrides]);\n\n    return children;\n}\n\nexport function withFeatureFlagsOverrideDecorator<const T extends string>(featureFlags: Map<T, LDFlagValue>, overrides: Partial<Record<T, LDFlagValue>>): Decorator {\n    if (!(featureFlags instanceof Map)) {\n        throw new TypeError(\"[squide] The \\\"featureFlags\\\" argument must be a Map instance.\");\n    }\n\n    return story => {\n        return (\n            <OverrideFeatureFlags featureFlags={featureFlags} overrides={overrides}>\n                {story()}\n            </OverrideFeatureFlags>\n        );\n    };\n}\n"],"names":["useLayoutEffect","OverrideFeatureFlags","props","featureFlags","overrides","children","originalValues","Object","key","value","withFeatureFlagsOverrideDecorator","Map","TypeError","story"],"mappings":";;;;;;;;;AAC2D;AAQ3D,SAASC,qBAAqBC,KAAgC;IAC1D,MAAM,EACFC,YAAY,EACZC,SAAS,EACTC,QAAQ,EACX,GAAGH;IAEJ,mFAAmF;IACnFF,eAAeA,CAAC;QACZ,MAAMM,iBAA8C,CAAC;QAErDC,OAAO,OAAO,CAACH,WAAW,OAAO,CAAC,CAAC,CAACI,KAAKC,MAAM;YAC3CH,cAAc,CAACE,IAAI,GAAGL,aAAa,GAAG,CAACK;YACvCL,aAAa,GAAG,CAACK,KAAKC;QAC1B;QAEA,OAAO;YACH,kDAAkD;YAClDF,OAAO,OAAO,CAACD,gBAAgB,OAAO,CAAC,CAAC,CAACE,KAAKC,MAAM;gBAChDN,aAAa,GAAG,CAACK,KAAKC;YAC1B;QACJ;IACJ,GAAG;QAACN;QAAcC;KAAU;IAE5B,OAAOC;AACX;AAEO,SAASK,kCAA0DP,YAAiC,EAAEC,SAA0C;IACnJ,IAAI,CAAED,CAAAA,wBAAwBQ,GAAE,GAAI;QAChC,MAAM,IAAIC,UAAU;IACxB;IAEA,OAAOC,CAAAA;QACH,qBACI,IAACZ;YAAqB,cAAcE;YAAc,WAAWC;sBACxDS;;IAGb;AACJ"}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@squide/firefly-rsbuild-storybook",
   "author": "Workleap",
-  "version": "0.0.1",
+  "version": "1.0.0",
   "description": "Squide firefly helpers for Storybook and Rsbuild.",
   "license": "Apache-2.0",
   "repository": {
@@ -28,14 +28,15 @@
   ],
   "peerDependencies": {
     "@opentelemetry/api": "^1.9.0",
-    "@rsbuild/core": "^1.6.9",
-    "@tanstack/react-query": "^5.90.11",
-    "msw": "^2.12.3",
+    "@rsbuild/core": "^1.6.14",
+    "@tanstack/react-query": "^5.90.12",
+    "launchdarkly-js-client-sdk": "^3.9.0",
+    "msw": "^2.12.4",
     "react": "^18.0.0 || ^19.0.0",
     "react-dom": "^18.0.0 || ^19.0.0",
-    "react-router": "^7.9.6",
-    "storybook": "^10.1.0",
-    "storybook-react-rsbuild": "^3.0.0"
+    "react-router": "^7.10.1",
+    "storybook": "^10.1.6",
+    "storybook-react-rsbuild": "^3.1.0"
   },
   "peerDependenciesMeta": {
     "@opentelemetry/api": {
@@ -43,30 +44,35 @@
     }
   },
   "dependencies": {
-    "@workleap-telemetry/core": "^1.0.4",
+    "@workleap-telemetry/core": "^1.0.5",
     "@workleap/logging": "^1.3.2",
-    "@squide/env-vars": "1.4.9",
-    "@squide/firefly": "16.0.3",
-    "@squide/msw": "4.0.7"
+    "@squide/env-vars": "1.4.10",
+    "@squide/firefly": "16.1.0",
+    "@squide/launch-darkly": "1.0.0",
+    "@squide/msw": "4.0.8"
   },
   "devDependencies": {
     "@eslint/js": "9.39.1",
-    "@rsbuild/core": "1.6.9",
-    "@rslib/core": "0.18.2",
+    "@rsbuild/core": "1.6.14",
+    "@rslib/core": "0.18.4",
     "@types/react": "19.2.7",
     "@types/react-dom": "19.2.3",
-    "@typescript-eslint/parser": "8.48.0",
-    "@workleap/eslint-configs": "1.1.5",
+    "@typescript-eslint/parser": "8.49.0",
+    "@vitejs/plugin-react": "5.1.2",
+    "@workleap/eslint-configs": "1.1.6",
     "@workleap/rslib-configs": "1.1.3",
     "@workleap/typescript-configs": "3.0.7",
     "eslint": "9.39.1",
+    "happy-dom": "20.0.11",
     "typescript": "5.9.3",
-    "typescript-eslint": "8.48.0"
+    "typescript-eslint": "8.49.0",
+    "vitest": "4.0.15"
   },
   "sideEffects": false,
   "scripts": {
     "build": "rslib build --config ./rslib.build.ts",
     "eslint": "eslint . --max-warnings=0 --cache --cache-location node_modules/.cache/eslint",
-    "typecheck": "tsc"
+    "typecheck": "tsc",
+    "test": "vitest --config vitest.config.ts --no-watch"
   }
 }

package/src/index.ts

@@ -1,4 +1,5 @@
 export { initializeFireflyForStorybook, type InitializeFireflyForStorybookOptions } from "./initializeFireflyForStorybook.ts";
 export { StorybookRuntime, StorybookRuntimeScope } from "./StorybookRuntime.ts";
+export { withFeatureFlagsOverrideDecorator } from "./withFeatureFlagsOverrideDecorator.tsx";
 export { FireflyDecorator, withFireflyDecorator, type FireflyDecoratorProps } from "./withFireflyDecorator.tsx";
 

package/src/initializeFireflyForStorybook.ts

@@ -1,17 +1,22 @@
 import { EnvironmentVariables, EnvironmentVariablesPlugin } from "@squide/env-vars";
-import { FireflyRuntime, ModuleRegisterFunction, toLocalModuleDefinitions } from "@squide/firefly";
+import { FireflyRuntime, InMemoryLaunchDarklyClient, LaunchDarklyPlugin, ModuleRegisterFunction, toLocalModuleDefinitions } from "@squide/firefly";
 import { MswPlugin } from "@squide/msw";
+import { LDClient, LDFlagValue } from "launchdarkly-js-client-sdk";
 import { StorybookRuntime } from "./StorybookRuntime.ts";
 
 export interface InitializeFireflyForStorybookOptions {
     localModules?: ModuleRegisterFunction<FireflyRuntime>[];
     environmentVariables?: EnvironmentVariables;
+    featureFlags?: Map<string, LDFlagValue> | Record<string, LDFlagValue>;
+    launchDarklyClient?: LDClient;
 }
 
 export async function initializeFireflyForStorybook(options: InitializeFireflyForStorybookOptions = {}) {
     const {
-        localModules = [],
-        environmentVariables = {}
+        localModules,
+        environmentVariables,
+        featureFlags = {},
+        launchDarklyClient
     } = options;
 
     const runtime = new StorybookRuntime({
@@ -20,13 +25,25 @@ export async function initializeFireflyForStorybook(options: InitializeFireflyFo
             x => new MswPlugin(x),
             x => new EnvironmentVariablesPlugin(x, {
                 variables: environmentVariables
-            })
+            }),
+            x => new LaunchDarklyPlugin(
+                x,
+                // 1- If no client is provided create it.
+                // 2- If feature flags are provided and it's an instance of Map, use the argument.
+                // 3- If feature flags are provided (or the default value) and it's an object literal, convert the object to a Map instance.
+                launchDarklyClient ?? new InMemoryLaunchDarklyClient(featureFlags instanceof Map
+                    ? featureFlags
+                    : new Map<string, LDFlagValue>(Object.entries(featureFlags))
+                )
+            )
         ]
     });
 
-    await runtime.moduleManager.registerModules([
-        ...toLocalModuleDefinitions(localModules)
-    ]);
+    if (localModules && localModules.length > 0) {
+        await runtime.moduleManager.registerModules([
+            ...toLocalModuleDefinitions(localModules)
+        ]);
+    }
 
     return runtime;
 }

package/src/withFeatureFlagsOverrideDecorator.tsx

@@ -0,0 +1,49 @@
+import { LDFlagValue } from "launchdarkly-js-client-sdk";
+import { PropsWithChildren, useLayoutEffect } from "react";
+import { Decorator } from "storybook-react-rsbuild";
+
+interface OverrideFeatureFlagsProps extends PropsWithChildren {
+    featureFlags: Map<string, LDFlagValue>;
+    overrides: Record<string, LDFlagValue>;
+}
+
+function OverrideFeatureFlags(props: OverrideFeatureFlagsProps) {
+    const {
+        featureFlags,
+        overrides,
+        children
+    } = props;
+
+    // Must use a layout effect to override the feature flags before the story renders.
+    useLayoutEffect(() => {
+        const originalValues: Record<string, LDFlagValue> = {};
+
+        Object.entries(overrides).forEach(([key, value]) => {
+            originalValues[key] = featureFlags.get(key);
+            featureFlags.set(key, value);
+        });
+
+        return () => {
+            // Reset the feature flags to the original values.
+            Object.entries(originalValues).forEach(([key, value]) => {
+                featureFlags.set(key, value);
+            });
+        };
+    }, [featureFlags, overrides]);
+
+    return children;
+}
+
+export function withFeatureFlagsOverrideDecorator<const T extends string>(featureFlags: Map<T, LDFlagValue>, overrides: Partial<Record<T, LDFlagValue>>): Decorator {
+    if (!(featureFlags instanceof Map)) {
+        throw new TypeError("[squide] The \"featureFlags\" argument must be a Map instance.");
+    }
+
+    return story => {
+        return (
+            <OverrideFeatureFlags featureFlags={featureFlags} overrides={overrides}>
+                {story()}
+            </OverrideFeatureFlags>
+        );
+    };
+}