From aa3a03eca82b128dea6142cbd89b3e28f177b0de Mon Sep 17 00:00:00 2001
From: Mads Rasmussen <madsr@hey.com>
Date: Tue, 18 Oct 2022 12:24:51 +0200
Subject: [PATCH] fix storybook build

---
 src/Umbraco.Web.UI.Client/.storybook/main.js    |  9 +++++++++
 src/Umbraco.Web.UI.Client/.storybook/preview.js | 10 +++++-----
 2 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/src/Umbraco.Web.UI.Client/.storybook/main.js b/src/Umbraco.Web.UI.Client/.storybook/main.js
index 73a7c5ea34..23de765208 100644
--- a/src/Umbraco.Web.UI.Client/.storybook/main.js
+++ b/src/Umbraco.Web.UI.Client/.storybook/main.js
@@ -1,3 +1,6 @@
+const tsconfigPaths = require('vite-tsconfig-paths').default;
+const { mergeConfig } = require('vite');
+
 module.exports = {
 	stories: ['../src/**/*.stories.mdx', '../src/**/*.stories.@(js|jsx|ts|tsx)'],
 	addons: ['@storybook/addon-links', '@storybook/addon-essentials', '@storybook/addon-a11y'],
@@ -10,4 +13,10 @@ module.exports = {
 		builder: '@storybook/builder-vite',
 	},
 	staticDirs: ['../public-assets'],
+	async viteFinal(config, { configType }) {
+		return mergeConfig(config, {
+			// customize the Vite config here
+			plugins: [tsconfigPaths()],
+		});
+	},
 };
diff --git a/src/Umbraco.Web.UI.Client/.storybook/preview.js b/src/Umbraco.Web.UI.Client/.storybook/preview.js
index 63c61cb5d8..5c2f95c9f5 100644
--- a/src/Umbraco.Web.UI.Client/.storybook/preview.js
+++ b/src/Umbraco.Web.UI.Client/.storybook/preview.js
@@ -1,5 +1,5 @@
-import '../src/core/context/context-provider.element';
-import '../src/css/custom-properties.css';
+import '../src/core/context-api/provide/context-provider.element';
+import '../src/core/css/custom-properties.css';
 import '../src/backoffice/components/backoffice-modal-container.element';
 import '@umbraco-ui/uui';
 import '@umbraco-ui/uui-modal';
@@ -12,15 +12,15 @@ import { initialize, mswDecorator } from 'msw-storybook-addon';
 import { setCustomElements } from '@storybook/web-components';
 
 import customElementManifests from '../custom-elements.json';
-import { UmbExtensionRegistry } from '@umbraco-cms/extensions-api';
+import { UmbExtensionRegistry } from '../src/core/extensions-api';
 import { UmbDataTypeStore } from '../src/core/stores/data-type/data-type.store';
 import { UmbDocumentTypeStore } from '../src/core/stores/document-type.store';
 import { UmbNodeStore } from '../src/core/stores/node.store';
 import { UmbPropertyEditorStore } from '../src/core/stores/property-editor/property-editor.store';
 import { UmbPropertyEditorConfigStore } from '../src/core/stores/property-editor-config/property-editor-config.store';
 import { UmbIconStore } from '../src/core/stores/icon/icon.store';
-import { onUnhandledRequest } from '../src/mocks/browser';
-import { handlers } from '../src/mocks/browser-handlers';
+import { onUnhandledRequest } from '../src/core/mocks/browser';
+import { handlers } from '../src/core/mocks/browser-handlers';
 import { internalManifests } from '../src/temp-internal-manifests';
 import { LitElement } from 'lit';
 import { UmbModalService } from '../src/core/services/modal';