Add support for passing in sourceLoader options from the docs addon u…

…sing the vite builder

code/lib/builder-vite/src/plugins/source-loader-plugin.ts

@@ -1,33 +1,44 @@
 import type { Plugin } from 'vite';
 import sourceLoaderTransform from '@storybook/source-loader';
 import MagicString from 'magic-string';
+import type { StorybookConfig } from '@storybook/types';
 import type { ExtendedOptions } from '../types';
 
 const storyPattern = /\.stories\.[jt]sx?$/;
 const storySourcePattern = /var __STORY__ = "(.*)"/;
 const storySourceReplacement = '--STORY_SOURCE_REPLACEMENT--';
 
-const mockClassLoader = (id: string) => ({
+const mockClassLoader = (id: string, options: any) => ({
   // eslint-disable-next-line no-console
   emitWarning: (message: string) => console.warn(message),
   resourcePath: id,
-  getOptions: () => ({}),
+  getOptions: () => ({ injectStoryParameters: true, inspectLocalDependencies: true, ...options }),
 });
 
 // HACK: Until we can support only node 15+ and use string.prototype.replaceAll
 const replaceAll = (str: string, search: string, replacement: string) => {
   return str.split(search).join(replacement);
 };
 
-export function sourceLoaderPlugin(config: ExtendedOptions): Plugin | Plugin[] {
+export async function sourceLoaderPlugin(config: ExtendedOptions): Promise<Plugin | Plugin[]> {
+  const { presets } = config;
+
+  const addons = await presets.apply('addons', [] as StorybookConfig['addons']);
+
+  const docsOptions =
+    // @ts-expect-error - not sure what type to use here
+    addons.find((a) => [a, a.name].includes('@storybook/addon-docs'))?.options ?? {};
   if (config.configType === 'DEVELOPMENT') {
     return {
       name: 'storybook:source-loader-plugin',
       enforce: 'pre',
       async transform(src: string, id: string) {
         if (id.match(storyPattern)) {
           // @ts-expect-error - source loader doesn't have types
-          const code: string = await sourceLoaderTransform.call(mockClassLoader(id), src);
+          const code: string = await sourceLoaderTransform.call(
+            mockClassLoader(id, docsOptions.sourceLoaderOptions),
+            src
+          );
           const s = new MagicString(src);
           // Entirely replace with new code
           s.overwrite(0, src.length, code);

code/lib/builder-vite/src/vite-config.ts

@@ -80,7 +80,7 @@ export async function pluginConfig(options: ExtendedOptions) {
 
   const plugins = [
     codeGeneratorPlugin(options),
-    sourceLoaderPlugin(options),
+    await sourceLoaderPlugin(options),
     mdxPlugin(),
     injectExportOrderPlugin,
     stripStoryHMRBoundary(),