From 4d6afdbe183223da5c3bbf44a6e558c5bd6c2c0e Mon Sep 17 00:00:00 2001
From: Mads Rasmussen <madsr@hey.com>
Date: Mon, 8 Aug 2022 15:09:13 +0200
Subject: [PATCH] add global providers in storybook

---
 .../.storybook/preview.js                     | 46 +++++++++++++++++--
 1 file changed, 41 insertions(+), 5 deletions(-)

diff --git a/src/Umbraco.Web.UI.Client/.storybook/preview.js b/src/Umbraco.Web.UI.Client/.storybook/preview.js
index 84d7f64230..29941bcc69 100644
--- a/src/Umbraco.Web.UI.Client/.storybook/preview.js
+++ b/src/Umbraco.Web.UI.Client/.storybook/preview.js
@@ -1,16 +1,52 @@
 import '@umbraco-ui/uui';
 import '@umbraco-ui/uui-css/dist/uui-css.css';
+import { html } from 'lit-html';
 
 import { initialize, mswDecorator } from 'msw-storybook-addon';
 
 import { onUnhandledRequest } from '../src/mocks/browser';
 import { handlers } from '../src/mocks/handlers';
 
+import { UmbExtensionRegistry } from '../src/core/extension';
+import { internalManifests } from '../src/temp-internal-manifests';
+
+import '../src/core/context/context-provider.element';
+import { UmbNodeStore } from '../src/core/stores/node.store';
+import { UmbDataTypeStore } from '../src/core/stores/data-type.store';
+import { UmbDocumentTypeStore } from '../src/core/stores/document-type.store';
+
+const extensionRegistry = new UmbExtensionRegistry();
+internalManifests.forEach((manifest) => extensionRegistry.register(manifest));
+
+const extensionRegistryProvider = (story) => html`
+	<umb-context-provider key="umbExtensionRegistry" .value=${extensionRegistry}>${story()}</umb-context-provider>
+`;
+
+const nodeStoreProvider = (story) => html`
+	<umb-context-provider key="umbNodeStore" .value=${new UmbNodeStore()}>${story()}</umb-context-provider>
+`;
+
+const dataTypeStoreProvider = (story) => html`
+	<umb-context-provider key="umbDataTypeStore" .value=${new UmbDataTypeStore()}>${story()}</umb-context-provider>
+`;
+
+const documentTypeStoreProvider = (story) => html`
+	<umb-context-provider key="umbDocumentTypeStore" .value=${new UmbDocumentTypeStore()}
+		>${story()}</umb-context-provider
+	>
+`;
+
 // Initialize MSW
-initialize({onUnhandledRequest});
+initialize({ onUnhandledRequest });
 
 // Provide the MSW addon decorator globally
-export const decorators = [mswDecorator];
+export const decorators = [
+	mswDecorator,
+	extensionRegistryProvider,
+	nodeStoreProvider,
+	dataTypeStoreProvider,
+	documentTypeStoreProvider,
+];
 
 export const parameters = {
 	actions: { argTypesRegex: '^on.*' },
@@ -23,7 +59,7 @@ export const parameters = {
 	},
 	msw: {
 		handlers: {
-			global: handlers
-		}
-	}
+			global: handlers,
+		},
+	},
 };