updated generate-stories-script

Author: bashmish

packages/storybook-builder/src/generate-stories-script.ts

@@ -1,38 +1,30 @@
 // based on https://github.com/storybookjs/storybook/blob/v8.5.0/code/builders/builder-vite/src/codegen-importfn-script.ts
 
 import { normalizePath } from '@rollup/pluginutils';
-import { logger } from '@storybook/node-logger';
 import type { Options } from '@storybook/types';
-import { extname, relative } from 'node:path';
-import { listStories } from './list-stories.js';
+import { relative } from 'node:path';
+import { listStories } from './list-stories';
 
-/**
- * This file is largely based on https://github.com/storybookjs/storybook/blob/d1195cbd0c61687f1720fefdb772e2f490a46584/lib/core-common/src/utils/to-importFn.ts
- */
-
-export async function generateStoriesScript(options: Options) {
+export async function generateStoriesScript(options: Options): Promise<string> {
   // First we need to get an array of stories and their absolute paths.
   const stories = await listStories(options);
 
   // We can then call toImportFn to create a function that can be used to load each story dynamically.
-  return (await toImportFn(stories)).trim();
+  return await toImportFn(stories);
 }
 
 /**
  * This function takes an array of stories and creates a mapping between the stories' relative paths
- * to the working directory and their dynamic imports. The import is done in an asynchronous function
- * to delay loading. It then creates a function, `importFn(path)`, which resolves a path to an import
- * function and this is called by Storybook to fetch a story dynamically when needed.
+ * to the working directory and their dynamic imports. The import is done in an asynchronous
+ * function to delay loading and to allow rollup to split the code into smaller chunks. It then
+ * creates a function, `importFn(path)`, which resolves a path to an import function and this is
+ * called by Storybook to fetch a story dynamically when needed.
+ *
  * @param stories An array of absolute story paths.
  */
 async function toImportFn(stories: string[]) {
   const objectEntries = stories.map(file => {
-    const ext = extname(file);
     const relativePath = normalizePath(relative(process.cwd(), file));
-    if (!['.js', '.jsx', '.ts', '.tsx', '.mdx', '.svelte', '.vue'].includes(ext)) {
-      logger.warn(`Cannot process ${ext} file with storyStoreV7: ${relativePath}`);
-    }
-
     const importPath = toImportPath(relativePath);
     let actualPath = file;
     if (actualPath.endsWith('.mdx')) {