WIP: Codemods and refactorings

code/.storybook/main.ts

@@ -1,12 +1,12 @@
 import { join } from 'node:path';
 
-import type { StorybookConfig } from '../frameworks/react-vite';
+import { defineMain } from '../frameworks/react-vite';
 
 const componentsPath = join(__dirname, '../core/src/components');
 const managerApiPath = join(__dirname, '../core/src/manager-api');
 const imageContextPath = join(__dirname, '..//frameworks/nextjs/src/image-context.ts');
 
-const config: StorybookConfig = {
+const config = defineMain({
   stories: [
     './*.stories.@(js|jsx|ts|tsx)',
     {
@@ -169,6 +169,6 @@ const config: StorybookConfig = {
     } satisfies typeof viteConfig);
   },
   // logLevel: 'debug',
-};
+});
 
 export default config;

code/.storybook/preview.tsx

@@ -18,7 +18,7 @@ import {
 import { DocsContext } from '@storybook/blocks';
 import { global } from '@storybook/global';
 import type { Decorator, Loader, ReactRenderer } from '@storybook/react';
-import { defineConfig } from '@storybook/react/preview';
+import { definePreview } from '@storybook/react/preview';
 
 // TODO add empty preview
 // import * as storysource from '@storybook/addon-storysource';
@@ -375,7 +375,7 @@ const parameters = {
   },
 };
 
-export const config = defineConfig({
+export const config = definePreview({
   addons: [addonThemes, essentials, a11y, addonsPreview, templatePreview],
   parameters,
   decorators,

code/core/src/csf-tools/ConfigFile.test.ts

@@ -245,13 +245,25 @@ describe('ConfigFile', () => {
     });
 
     describe('factory config', () => {
+      it('parses correctly', () => {
+        const source = dedent`
+          import { definePreview } from '@storybook/react-vite/preview';
+
+          const config = definePreview({
+            framework: 'foo',
+          });
+          export default config;
+        `;
+        const config = loadConfig(source).parse();
+        expect(config.getNameFromPath(['framework'])).toEqual('foo');
+      });
       it('found scalar', () => {
         expect(
           getField(
             ['core', 'builder'],
             dedent`
-            import { defineConfig } from '@storybook/react-vite/preview';
-            export const foo = defineConfig({ core: { builder: 'webpack5' } });
+            import { definePreview } from '@storybook/react-vite/preview';
+            export const foo = definePreview({ core: { builder: 'webpack5' } });
             `
           )
         ).toEqual('webpack5');
@@ -261,9 +273,9 @@ describe('ConfigFile', () => {
           getField(
             ['tags'],
             dedent`
-              import { defineConfig } from '@storybook/react-vite/preview';
+              import { definePreview } from '@storybook/react-vite/preview';
               const parameters = {};
-              export const config = defineConfig({
+              export const config = definePreview({
                 parameters,
                 tags: ['test', 'vitest', '!a11ytest'],
               });
@@ -516,15 +528,15 @@ describe('ConfigFile', () => {
             ['core', 'builder'],
             'webpack5',
             dedent`
-              import { defineConfig } from '@storybook/react-vite/preview';
-              export const foo = defineConfig({
+              import { definePreview } from '@storybook/react-vite/preview';
+              export const foo = definePreview({
                 addons: [],
               });
             `
           )
         ).toMatchInlineSnapshot(`
-          import { defineConfig } from '@storybook/react-vite/preview';
-          export const foo = defineConfig({
+          import { definePreview } from '@storybook/react-vite/preview';
+          export const foo = definePreview({
             addons: [],
 
             core: {
@@ -539,15 +551,15 @@ describe('ConfigFile', () => {
             ['core', 'builder'],
             'webpack5',
             dedent`
-              import { defineConfig } from '@storybook/react-vite/preview';
-              export const foo = defineConfig({
+              import { definePreview } from '@storybook/react-vite/preview';
+              export const foo = definePreview({
                 core: { foo: 'bar' },
               });
             `
           )
         ).toMatchInlineSnapshot(`
-          import { defineConfig } from '@storybook/react-vite/preview';
-          export const foo = defineConfig({
+          import { definePreview } from '@storybook/react-vite/preview';
+          export const foo = definePreview({
             core: {
               foo: 'bar',
               builder: 'webpack5'
@@ -561,15 +573,15 @@ describe('ConfigFile', () => {
             ['core', 'builder'],
             'webpack5',
             dedent`
-              import { defineConfig } from '@storybook/react-vite/preview';
-              export const foo = defineConfig({
+              import { definePreview } from '@storybook/react-vite/preview';
+              export const foo = definePreview({
                 core: { builder: 'webpack4' },
               });
             `
           )
         ).toMatchInlineSnapshot(`
-          import { defineConfig } from '@storybook/react-vite/preview';
-          export const foo = defineConfig({
+          import { definePreview } from '@storybook/react-vite/preview';
+          export const foo = definePreview({
             core: { builder: 'webpack5' },
           });
         `);
@@ -1017,6 +1029,19 @@ describe('ConfigFile', () => {
         expect(config.getNameFromPath(['otherField'])).toEqual('foo');
       });
 
+      it(`supports pnp wrapped names`, () => {
+        const source = dedent`
+          import type { StorybookConfig } from '@storybook/react-webpack5';
+
+          const config: StorybookConfig = {
+            framework: getAbsolutePath('foo'),
+          }
+          export default config;
+        `;
+        const config = loadConfig(source).parse();
+        expect(config.getNameFromPath(['framework'])).toEqual('foo');
+      });
+
       it(`returns undefined when accessing a field that does not exist`, () => {
         const source = dedent`
           import type { StorybookConfig } from '@storybook/react-webpack5';

code/core/src/csf-tools/ConfigFile.ts

@@ -25,18 +25,8 @@ const getCsfParsingErrorMessage = ({
   foundType: string | undefined;
   node: any | undefined;
 }) => {
-  let nodeInfo = '';
-  if (node) {
-    try {
-      nodeInfo = JSON.stringify(node);
-    } catch (e) {
-      //
-    }
-  }
-
   return dedent`
       CSF Parsing error: Expected '${expectedType}' but found '${foundType}' instead in '${node?.type}'.
-      ${nodeInfo}
     `;
 };
 
@@ -215,6 +205,11 @@ export class ConfigFile {
 
           decl = unwrap(decl);
 
+          // csf factory
+          if (t.isCallExpression(decl) && t.isObjectExpression(decl.arguments[0])) {
+            decl = decl.arguments[0];
+          }
+
           if (t.isObjectExpression(decl)) {
             self._parseExportsObject(decl);
           } else {
@@ -323,7 +318,7 @@ export class ConfigFile {
         enter: ({ node }) => {
           if (
             t.isIdentifier(node.callee) &&
-            node.callee.name === 'defineConfig' &&
+            node.callee.name === 'definePreview' &&
             node.arguments.length === 1 &&
             t.isObjectExpression(node.arguments[0])
           ) {
@@ -499,6 +494,8 @@ export class ConfigFile {
           value = prop.value.value;
         }
       });
+    } else if (t.isCallExpression(node)) {
+      value = this._getPnpWrappedValue(node);
     }
 
     if (!value) {

code/core/src/csf-tools/CsfFile.ts

@@ -4,6 +4,7 @@ import { readFile, writeFile } from 'node:fs/promises';
 import {
   BabelFileClass,
   type GeneratorOptions,
+  type NodePath,
   type RecastOptions,
   babelParse,
   generate,
@@ -255,10 +256,14 @@ export class CsfFile {
 
   _storyExports: Record<string, t.VariableDeclarator | t.FunctionDeclaration> = {};
 
+  _storyPaths: Record<string, NodePath<t.ExportNamedDeclaration>> = {};
+
   _metaStatement: t.Statement | undefined;
 
   _metaNode: t.Expression | undefined;
 
+  _metaPath: NodePath<t.ExportDefaultDeclaration> | undefined;
+
   _metaVariableName: string | undefined;
 
   _metaIsFactory: boolean | undefined;
@@ -466,10 +471,13 @@ export class CsfFile {
               self._options.fileName
             );
           }
+
+          self._metaPath = path;
         },
       },
       ExportNamedDeclaration: {
-        enter({ node, parent }) {
+        enter(path) {
+          const { node, parent } = path;
           let declarations;
           if (t.isVariableDeclaration(node.declaration)) {
             declarations = node.declaration.declarations.filter((d) => t.isVariableDeclarator(d));
@@ -487,6 +495,7 @@ export class CsfFile {
                   return;
                 }
                 self._storyExports[exportName] = decl;
+                self._storyPaths[exportName] = path;
                 self._storyStatements[exportName] = node;
                 let name = storyNameFromExport(exportName);
                 if (self._storyAnnotations[exportName]) {
@@ -611,6 +620,7 @@ export class CsfFile {
                 } else {
                   self._storyAnnotations[exportName] = {};
                   self._storyStatements[exportName] = decl;
+                  self._storyPaths[exportName] = path;
                   self._stories[exportName] = {
                     id: 'FIXME',
                     name: exportName,

code/core/src/csf-tools/enrichCsf.test.ts

@@ -149,6 +149,51 @@ describe('enrichCsf', () => {
         };
       `);
     });
+    it('csf factories', () => {
+      expect(
+        enrich(
+          dedent`
+          // compiled code
+          import {config} from "/.storybook/preview.ts";
+          const meta = config.meta({
+              args: {
+                label: "Hello world!"
+              }
+          });
+          export const Story = meta.story({});
+        `,
+          dedent`
+          // original code
+          import {config} from "#.storybook/preview.ts";
+          const meta = config.meta({
+              args: {
+                label: "Hello world!"
+              }
+          });
+          export const Story = meta.story({});
+        `
+        )
+      ).toMatchInlineSnapshot(`
+        // compiled code
+        import { config } from "/.storybook/preview.ts";
+        const meta = config.meta({
+          args: {
+            label: "Hello world!"
+          }
+        });
+        export const Story = meta.story({});
+        Story.annotations.parameters = {
+          ...Story.annotations.parameters,
+          docs: {
+            ...Story.annotations.parameters?.docs,
+            source: {
+              originalSource: "meta.story({})",
+              ...Story.annotations.parameters?.docs?.source
+            }
+          }
+        };
+      `);
+    });
     it('multiple stories', () => {
       expect(
         enrich(

code/core/src/csf-tools/enrichCsf.ts

@@ -15,11 +15,20 @@ export const enrichCsfStory = (
   options?: EnrichCsfOptions
 ) => {
   const storyExport = csfSource.getStoryExport(key);
+  const isCsfFactory =
+    t.isCallExpression(storyExport) &&
+    t.isMemberExpression(storyExport.callee) &&
+    t.isIdentifier(storyExport.callee.object) &&
+    storyExport.callee.object.name === 'meta';
   const source = !options?.disableSource && extractSource(storyExport);
   const description =
     !options?.disableDescription && extractDescription(csfSource._storyStatements[key]);
   const parameters = [];
-  const originalParameters = t.memberExpression(t.identifier(key), t.identifier('parameters'));
+  // in csf 1/2/3 use Story.parameters; CSF factories use Story.annotations.parameters
+  const baseStoryObject = isCsfFactory
+    ? t.memberExpression(t.identifier(key), t.identifier('annotations'))
+    : t.identifier(key);
+  const originalParameters = t.memberExpression(baseStoryObject, t.identifier('parameters'));
   parameters.push(t.spreadElement(originalParameters));
   const optionalDocs = t.optionalMemberExpression(
     originalParameters,

code/frameworks/angular/package.json

@@ -20,6 +20,16 @@
     "url": "https://opencollective.com/storybook"
   },
   "license": "MIT",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "node": "./dist/index.js",
+      "import": "./dist/index.mjs",
+      "require": "./dist/index.js"
+    },
+    "./preset": "./preset.js",
+    "./package.json": "./package.json"
+  },
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",

code/frameworks/angular/src/defineMainConfig.ts

@@ -0,0 +1,5 @@
+import { StorybookConfig } from './types';
+
+export function defineMain(config: StorybookConfig) {
+  return config;
+}

code/frameworks/angular/src/index.ts

@@ -1,5 +1,6 @@
 export * from './client/index';
 export * from './types';
+export * from './defineMainConfig';
 
 /*
  * ATTENTION:

code/frameworks/ember/package.json

@@ -16,6 +16,16 @@
     "url": "https://opencollective.com/storybook"
   },
   "license": "MIT",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "node": "./dist/index.js",
+      "import": "./dist/index.mjs",
+      "require": "./dist/index.js"
+    },
+    "./preset": "./preset.js",
+    "./package.json": "./package.json"
+  },
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",

code/frameworks/ember/src/defineMainConfig.ts

@@ -0,0 +1,5 @@
+import type { StorybookConfig } from './types';
+
+export function defineMain(config: StorybookConfig) {
+  return config;
+}

code/frameworks/ember/src/index.ts

@@ -1,6 +1,7 @@
 /// <reference types="webpack-env" />
 
 export * from './types';
+export * from './defineMainConfig';
 
 // optimization: stop HMR propagation in webpack