@storybook/codemod 11.0.0-alpha.7 → 11.0.0-alpha.8

package/dist/_node-chunks/chunk-5VYHV7J4.js

@@ -0,0 +1,105 @@
+import CJS_COMPAT_NODE_URL_77qkz8uuglf from 'node:url';
+import CJS_COMPAT_NODE_PATH_77qkz8uuglf from 'node:path';
+import CJS_COMPAT_NODE_MODULE_77qkz8uuglf from "node:module";
+
+var __filename = CJS_COMPAT_NODE_URL_77qkz8uuglf.fileURLToPath(import.meta.url);
+var __dirname = CJS_COMPAT_NODE_PATH_77qkz8uuglf.dirname(__filename);
+var require = CJS_COMPAT_NODE_MODULE_77qkz8uuglf.createRequire(import.meta.url);
+
+// ------------------------------------------------------------
+// end of CJS compatibility banner, injected by Storybook's esbuild configuration
+// ------------------------------------------------------------
+
+// src/transforms/upgrade-deprecated-types.ts
+import {
+  SourceEditor,
+  removeFromList,
+  specifierList,
+  storyShapeError,
+  walk
+} from "storybook/internal/csf-tools";
+import { logger } from "storybook/internal/node-logger";
+import prettier from "prettier";
+var deprecatedTypes = [
+  "ComponentStory",
+  "ComponentStoryFn",
+  "ComponentStoryObj",
+  "ComponentMeta",
+  "Story"
+];
+function migrateType(oldType) {
+  return oldType === "Story" || oldType === "ComponentStory" ? "StoryFn" : oldType.replace("Component", "");
+}
+async function transform(info) {
+  let editor = new SourceEditor(info.source, info.path);
+  upgradeDeprecatedTypes(editor);
+  let output = editor.toString();
+  try {
+    output = await prettier.format(output, {
+      ...await prettier.resolveConfig(info.path),
+      filepath: info.path
+    });
+  } catch {
+    logger.log(`Failed applying prettier to ${info.path}.`);
+  }
+  return output;
+}
+var parser = "tsx", importedName = (specifier) => specifier.type === "ImportSpecifier" && specifier.imported.type === "Identifier" ? specifier.imported.name : void 0;
+function upgradeDeprecatedTypes(editor) {
+  let importedNamespaces = /* @__PURE__ */ new Set(), typeReferencesToUpdate = /* @__PURE__ */ new Set(), existingImports = [];
+  for (let statement of editor.program.body) {
+    if (statement.type !== "ImportDeclaration")
+      continue;
+    let { specifiers } = statement;
+    if (existingImports.push(
+      ...specifiers.map((specifier) => ({
+        name: specifier.local.name,
+        isAlias: importedName(specifier) !== specifier.local.name,
+        node: specifier
+      }))
+    ), !statement.source.value.startsWith("@storybook"))
+      continue;
+    let removed = [];
+    for (let specifier of specifiers) {
+      specifier.type === "ImportNamespaceSpecifier" && importedNamespaces.add(specifier.local.name);
+      let imported = importedName(specifier);
+      if (specifier.type !== "ImportSpecifier" || !imported || !deprecatedTypes.includes(imported))
+        continue;
+      imported === specifier.local.name && typeReferencesToUpdate.add(specifier.local.name);
+      let newType = migrateType(imported);
+      if (!existingImports.some((it) => it.name === newType)) {
+        existingImports.push({ name: newType, isAlias: !1, node: specifier }), editor.edits.overwrite(specifier.imported.start, specifier.imported.end, newType);
+        continue;
+      }
+      let existingImport = existingImports.find((it) => it.name === newType && it.isAlias);
+      if (existingImport)
+        throw storyShapeError(
+          `This codemod does not support local imports that are called the same as a storybook import.
+Rename this local import and try again.`,
+          existingImport.node,
+          editor
+        );
+      removed.push(specifier);
+    }
+    removed.length > 0 && removeFromList(editor, specifierList(editor, statement), removed);
+  }
+  walk(editor.program, (node) => {
+    if (node.type !== "TSTypeReference")
+      return;
+    let { typeName } = node;
+    typeName.type === "Identifier" ? typeReferencesToUpdate.has(typeName.name) && editor.edits.overwrite(typeName.start, typeName.end, migrateType(typeName.name)) : (
+      // For example SB.StoryObj
+      typeName.type === "TSQualifiedName" && typeName.left.type === "Identifier" && importedNamespaces.has(typeName.left.name) && deprecatedTypes.includes(typeName.right.name) && editor.edits.overwrite(
+        typeName.right.start,
+        typeName.right.end,
+        migrateType(typeName.right.name)
+      )
+    );
+  }), editor.commit();
+}
+
+export {
+  transform,
+  parser,
+  upgradeDeprecatedTypes
+};

package/dist/index.js

@@ -1,10 +1,10 @@
-import CJS_COMPAT_NODE_URL_ppo5qkb13mm from 'node:url';
-import CJS_COMPAT_NODE_PATH_ppo5qkb13mm from 'node:path';
-import CJS_COMPAT_NODE_MODULE_ppo5qkb13mm from "node:module";
+import CJS_COMPAT_NODE_URL_77qkz8uuglf from 'node:url';
+import CJS_COMPAT_NODE_PATH_77qkz8uuglf from 'node:path';
+import CJS_COMPAT_NODE_MODULE_77qkz8uuglf from "node:module";
 
-var __filename = CJS_COMPAT_NODE_URL_ppo5qkb13mm.fileURLToPath(import.meta.url);
-var __dirname = CJS_COMPAT_NODE_PATH_ppo5qkb13mm.dirname(__filename);
-var require = CJS_COMPAT_NODE_MODULE_ppo5qkb13mm.createRequire(import.meta.url);
+var __filename = CJS_COMPAT_NODE_URL_77qkz8uuglf.fileURLToPath(import.meta.url);
+var __dirname = CJS_COMPAT_NODE_PATH_77qkz8uuglf.dirname(__filename);
+var require = CJS_COMPAT_NODE_MODULE_77qkz8uuglf.createRequire(import.meta.url);
 
 // ------------------------------------------------------------
 // end of CJS compatibility banner, injected by Storybook's esbuild configuration

package/dist/transforms/csf-2-to-3.js

@@ -1,115 +1,107 @@
-import CJS_COMPAT_NODE_URL_ppo5qkb13mm from 'node:url';
-import CJS_COMPAT_NODE_PATH_ppo5qkb13mm from 'node:path';
-import CJS_COMPAT_NODE_MODULE_ppo5qkb13mm from "node:module";
+import CJS_COMPAT_NODE_URL_77qkz8uuglf from 'node:url';
+import CJS_COMPAT_NODE_PATH_77qkz8uuglf from 'node:path';
+import CJS_COMPAT_NODE_MODULE_77qkz8uuglf from "node:module";
 
-var __filename = CJS_COMPAT_NODE_URL_ppo5qkb13mm.fileURLToPath(import.meta.url);
-var __dirname = CJS_COMPAT_NODE_PATH_ppo5qkb13mm.dirname(__filename);
-var require = CJS_COMPAT_NODE_MODULE_ppo5qkb13mm.createRequire(import.meta.url);
+var __filename = CJS_COMPAT_NODE_URL_77qkz8uuglf.fileURLToPath(import.meta.url);
+var __dirname = CJS_COMPAT_NODE_PATH_77qkz8uuglf.dirname(__filename);
+var require = CJS_COMPAT_NODE_MODULE_77qkz8uuglf.createRequire(import.meta.url);
 
 // ------------------------------------------------------------
 // end of CJS compatibility banner, injected by Storybook's esbuild configuration
 // ------------------------------------------------------------
 import {
   upgradeDeprecatedTypes
-} from "../_node-chunks/chunk-I4FB73AT.js";
+} from "../_node-chunks/chunk-5VYHV7J4.js";
 
 // src/transforms/csf-2-to-3.ts
-import { core as babel, types as t } from "storybook/internal/babel";
-import { loadCsf, printCsf } from "storybook/internal/csf-tools";
+import {
+  declaratorList,
+  loadCsf,
+  placeableTextOf,
+  printCsf,
+  removeFromList,
+  removeStatement,
+  walk
+} from "storybook/internal/csf-tools";
 import { logger } from "storybook/internal/node-logger";
 import prettier from "prettier";
 import invariant from "tiny-invariant";
-var { isIdentifier, isTSTypeAnnotation, isTSTypeReference } = t, renameAnnotation = (annotation) => annotation === "storyName" ? "name" : annotation, annotationProperties = (annotations) => {
+var renameAnnotation = (annotation) => annotation === "storyName" ? "name" : annotation, annotationProperties = (annotations) => {
   let fields = /* @__PURE__ */ new Map();
   for (let [annotation, value] of Object.entries(annotations)) {
-    if (annotation === "story" && t.isObjectExpression(value)) {
+    if (annotation === "story" && value.type === "ObjectExpression") {
       for (let prop of value.properties)
-        t.isObjectProperty(prop) && t.isIdentifier(prop.key) && fields.set(prop.key.name, prop.value);
+        prop.type === "Property" && prop.key.type === "Identifier" && fields.set(prop.key.name, prop.value);
       continue;
     }
     fields.set(annotation, value);
   }
   return [...fields].map(
-    ([annotation, value]) => t.objectProperty(t.identifier(renameAnnotation(annotation)), value)
+    ([annotation, value]) => `${renameAnnotation(annotation)}: ${placeableTextOf(value)}`
   );
-}, getTemplateBindVariable = (init) => t.isCallExpression(init) && t.isMemberExpression(init.callee) && t.isIdentifier(init.callee.object) && t.isIdentifier(init.callee.property) && init.callee.property.name === "bind" && (init.arguments.length === 0 || init.arguments.length === 1 && t.isObjectExpression(init.arguments[0]) && init.arguments[0].properties.length === 0) ? init.callee.object.name : null, isStoryAnnotation = (stmt, objectExports) => t.isExpressionStatement(stmt) && t.isAssignmentExpression(stmt.expression) && t.isMemberExpression(stmt.expression.left) && t.isIdentifier(stmt.expression.left.object) && objectExports[stmt.expression.left.object.name], getNewExport = (stmt, objectExports) => {
-  if (t.isExportNamedDeclaration(stmt) && t.isVariableDeclaration(stmt.declaration) && stmt.declaration.declarations.length === 1) {
-    let decl = stmt.declaration.declarations[0];
-    if (t.isVariableDeclarator(decl) && t.isIdentifier(decl.id))
-      return objectExports[decl.id.name];
-  }
-  return null;
-}, isReactGlobalRenderFn = (csf, storyFn) => {
-  if (csf._meta?.component && t.isArrowFunctionExpression(storyFn) && storyFn.params.length === 1 && t.isJSXElement(storyFn.body)) {
+}, getTemplateBindVariable = (init) => init.type === "CallExpression" && init.callee.type === "MemberExpression" && init.callee.object.type === "Identifier" && init.callee.property.type === "Identifier" && init.callee.property.name === "bind" && (init.arguments.length === 0 || init.arguments.length === 1 && init.arguments[0].type === "ObjectExpression" && init.arguments[0].properties.length === 0) ? init.callee.object.name : null, isStoryAnnotation = (stmt, objectExports) => stmt.type === "ExpressionStatement" && stmt.expression.type === "AssignmentExpression" && stmt.expression.left.type === "MemberExpression" && stmt.expression.left.object.type === "Identifier" && objectExports.has(stmt.expression.left.object.name), isSingleConstExport = (stmt) => stmt?.type === "ExportNamedDeclaration" && stmt.declaration?.type === "VariableDeclaration" && stmt.declaration.declarations.length === 1, isReactGlobalRenderFn = (csf, storyFn) => {
+  if (csf._meta?.component && storyFn?.type === "ArrowFunctionExpression" && storyFn.params.length === 1 && storyFn.body.type === "JSXElement") {
     let { openingElement } = storyFn.body;
-    if (openingElement.selfClosing && t.isJSXIdentifier(openingElement.name) && openingElement.attributes.length === 1) {
+    if (openingElement.selfClosing && openingElement.name.type === "JSXIdentifier" && openingElement.attributes.length === 1) {
       let attr = openingElement.attributes[0], param = storyFn.params[0];
-      if (t.isJSXSpreadAttribute(attr) && t.isIdentifier(attr.argument) && t.isIdentifier(param) && param.name === attr.argument.name && csf._meta.component === openingElement.name.name)
+      if (attr.type === "JSXSpreadAttribute" && attr.argument.type === "Identifier" && param.type === "Identifier" && param.name === attr.argument.name && csf._meta.component === openingElement.name.name)
         return !0;
     }
   }
   return !1;
-}, isSimpleCSFStory = (init, annotations) => annotations.length === 0 && t.isArrowFunctionExpression(init) && init.params.length === 0;
-function removeUnusedTemplates(csf) {
-  Object.entries(csf._templates).forEach(([template, templateExpression]) => {
+}, isSimpleCSFStory = (init, annotations) => annotations.length === 0 && init.type === "ArrowFunctionExpression" && init.params.length === 0;
+function removeUnusedTemplates(editor, templates) {
+  templates.forEach((template) => {
     let references = [];
-    if (babel.traverse(csf._ast, {
-      Identifier: (path) => {
-        path.node.name === template && references.push(path);
-      }
-    }), references.length === 1) {
-      let reference = references[0];
-      reference.parentPath?.isVariableDeclarator() && reference.parentPath.node.init === templateExpression && reference.parentPath.remove();
-    }
-  });
+    walk(editor.program, (node, parent) => {
+      (node.type === "Identifier" || node.type === "JSXIdentifier") && node.name === template && references.push({ node, parent });
+    });
+    let [reference] = references;
+    if (references.length !== 1 || reference.parent?.type !== "VariableDeclarator" || reference.parent.id !== reference.node || !reference.parent.init)
+      return;
+    let declarator = reference.parent, declaration = editor.parentOf(declarator);
+    declaration.declarations.length === 1 ? removeStatement(editor, declaration) : removeFromList(editor, declaratorList(editor, declaration), [declarator]);
+  }), editor.commit();
 }
-async function transform(info, api, options) {
-  let makeTitle = (userTitle) => userTitle || "FIXME", csf = loadCsf(info.source, { makeTitle });
+async function transform(info) {
+  if (info.path.endsWith(".svelte"))
+    return info.source;
+  let makeTitle = (userTitle) => userTitle || "FIXME", csf = loadCsf(info.source, { makeTitle, fileName: info.path });
   try {
     csf.parse();
   } catch (err) {
     return logger.log(`Error ${err}, skipping`), info.source;
   }
-  let file = new babel.File(
-    { filename: info.path },
-    { code: info.source, ast: csf._ast }
-  ), importHelper = new StorybookImportHelper(file, info), objectExports = {};
+  let editor = csf.editor, importHelper = new StorybookImportHelper(editor), templates = Object.keys(csf._templates), objectExports = /* @__PURE__ */ new Set();
   Object.entries(csf._storyExports).forEach(([key, decl]) => {
     let annotations = annotationProperties(csf._storyAnnotations[key]);
-    if (t.isVariableDeclarator(decl)) {
-      let { init, id } = decl;
-      invariant(init, "Inital value should be declared");
-      let template = getTemplateBindVariable(init);
-      if (!t.isArrowFunctionExpression(init) && !template)
-        return;
-      if (isSimpleCSFStory(init, annotations)) {
-        objectExports[key] = t.exportNamedDeclaration(
-          t.variableDeclaration("const", [
-            t.variableDeclarator(importHelper.updateTypeTo(id, "StoryFn"), init)
-          ])
-        );
-        return;
-      }
-      let storyFn = template ? t.identifier(template) : init, renderAnnotation = isReactGlobalRenderFn(
-        csf,
-        template ? csf._templates[template] : storyFn
-      ) ? [] : [t.objectProperty(t.identifier("render"), storyFn)];
-      objectExports[key] = t.exportNamedDeclaration(
-        t.variableDeclaration("const", [
-          t.variableDeclarator(
-            importHelper.updateTypeTo(id, "StoryObj"),
-            t.objectExpression([...renderAnnotation, ...annotations])
-          )
-        ])
-      );
+    if (decl.type !== "VariableDeclarator")
+      return;
+    let { init, id } = decl;
+    invariant(init, "Inital value should be declared");
+    let template = getTemplateBindVariable(init);
+    if (init.type !== "ArrowFunctionExpression" && !template)
+      return;
+    let replaceExport = isSingleConstExport(csf._storyStatements[key]);
+    if (isSimpleCSFStory(init, annotations)) {
+      importHelper.updateTypeTo(id, "StoryFn", replaceExport);
+      return;
     }
-  }), csf._ast.program.body = csf._ast.program.body.reduce((acc, stmt) => {
-    let statement = stmt;
-    if (isStoryAnnotation(statement, objectExports))
-      return acc;
-    let newExport = getNewExport(statement, objectExports);
-    return newExport ? (acc.push(newExport), acc) : (acc.push(statement), acc);
-  }, []), upgradeDeprecatedTypes(file), importHelper.removeDeprecatedStoryImport(), removeUnusedTemplates(csf);
+    if (importHelper.updateTypeTo(id, "StoryObj", replaceExport), !replaceExport)
+      return;
+    objectExports.add(key);
+    let properties = [...isReactGlobalRenderFn(csf, template ? csf._templates[template] : init) ? [] : [`render: ${template ?? editor.source(init)}`], ...annotations];
+    editor.edits.overwrite(
+      init.start,
+      init.end,
+      properties.length === 0 ? "{}" : `{
+${properties.map((it) => `  ${it},
+`).join("")}}`
+    );
+  });
+  for (let statement of editor.program.body)
+    isStoryAnnotation(statement, objectExports) && removeStatement(editor, statement);
+  editor.commit(), upgradeDeprecatedTypes(editor), removeUnusedTemplates(editor, templates);
   let output = printCsf(csf).code;
   try {
     output = await prettier.format(output, {
@@ -121,87 +113,56 @@ async function transform(info, api, options) {
   }
   return output;
 }
-var StorybookImportHelper = class {
-  constructor(file, info) {
-    this.getAllSbImportDeclarations = (file) => {
-      let found = [];
-      return file.path.traverse({
-        ImportDeclaration: (path) => {
-          let source = path.node.source.value;
-          if (source.startsWith("@storybook/csf") || !source.startsWith("@storybook"))
-            return;
-          path.get("specifiers").some((specifier) => {
-            if (specifier.isImportNamespaceSpecifier())
-              throw new Error(
-                `This codemod does not support namespace imports for a ${path.node.source.value} package.
-Replace the namespace import with named imports and try again.`
-              );
-            if (!specifier.isImportSpecifier())
-              return !1;
-            let imported = specifier.get("imported");
-            return Array.isArray(imported) ? imported.some((importedSpecifier) => importedSpecifier.isIdentifier() ? [
-              "Story",
-              "StoryFn",
-              "StoryObj",
-              "Meta",
-              "ComponentStory",
-              "ComponentStoryFn",
-              "ComponentStoryObj",
-              "ComponentMeta"
-            ].includes(importedSpecifier.node.name) : !1) : imported.isIdentifier() ? [
-              "Story",
-              "StoryFn",
-              "StoryObj",
-              "Meta",
-              "ComponentStory",
-              "ComponentStoryFn",
-              "ComponentStoryObj",
-              "ComponentMeta"
-            ].includes(imported.node.name) : !1;
-          }) && found.push(path);
-        }
-      }), found;
-    };
+var storyTypes = [
+  "Story",
+  "StoryFn",
+  "StoryObj",
+  "Meta",
+  "ComponentStory",
+  "ComponentStoryFn",
+  "ComponentStoryObj",
+  "ComponentMeta"
+], StorybookImportHelper = class {
+  constructor(editor) {
+    this.editor = editor;
+    this.addedImports = [];
     this.getOrAddImport = (type) => {
-      let sbImport = this.sbImportDeclarations.find((path) => path.node.importKind === "type") ?? this.sbImportDeclarations[0];
+      let sbImport = this.sbImportDeclarations.find((it) => it.importKind === "type") ?? this.sbImportDeclarations[0];
       if (sbImport == null)
         return;
-      let specifiers = sbImport.get("specifiers"), importSpecifier = specifiers.find((specifier) => {
-        if (!specifier.isImportSpecifier())
-          return !1;
-        let imported = specifier.get("imported");
-        return imported.isIdentifier() ? imported.node.name === type : !1;
-      });
-      return importSpecifier ? importSpecifier.node.local.name : (specifiers[0].insertBefore(t.importSpecifier(t.identifier(type), t.identifier(type))), type);
-    };
-    this.removeDeprecatedStoryImport = () => {
-      this.sbImportDeclarations.flatMap((it) => it.get("specifiers")).filter((specifier) => {
-        if (!specifier.isImportSpecifier())
-          return !1;
-        let imported = specifier.get("imported");
-        return imported.isIdentifier() ? imported.node.name === "Story" : !1;
-      }).forEach((path) => path.remove());
+      let importSpecifier = sbImport.specifiers.find(
+        (specifier) => specifier.type === "ImportSpecifier" && specifier.imported.type === "Identifier" && specifier.imported.name === type
+      );
+      return importSpecifier ? importSpecifier.local.name : (this.addedImports.includes(type) || (this.addedImports.push(type), this.editor.edits.appendLeft(sbImport.specifiers[0].start, `${type}, `)), type);
     };
-    this.getAllLocalImports = () => this.sbImportDeclarations.flatMap((it) => it.get("specifiers")).map((it) => it.node.local.name);
-    this.updateTypeTo = (id, type) => {
-      if (isIdentifier(id) && isTSTypeAnnotation(id.typeAnnotation) && isTSTypeReference(id.typeAnnotation.typeAnnotation) && isIdentifier(id.typeAnnotation.typeAnnotation.typeName)) {
-        let { name } = id.typeAnnotation.typeAnnotation.typeName;
-        if (this.getAllLocalImports().includes(name)) {
-          let localTypeImport = this.getOrAddImport(type);
-          return {
-            ...id,
-            typeAnnotation: t.tsTypeAnnotation(
-              t.tsTypeReference(
-                t.identifier(localTypeImport ?? ""),
-                id.typeAnnotation.typeAnnotation.typeParameters
-              )
-            )
-          };
-        }
+    this.getAllLocalImports = () => [
+      ...this.sbImportDeclarations.flatMap((it) => it.specifiers).map((it) => it.local.name),
+      ...this.addedImports
+    ];
+    this.updateTypeTo = (id, type, apply) => {
+      if (id.type !== "Identifier")
+        return;
+      let annotation = id.typeAnnotation?.typeAnnotation;
+      if (annotation?.type === "TSTypeReference" && annotation.typeName.type === "Identifier" && this.getAllLocalImports().includes(annotation.typeName.name) && apply) {
+        let localTypeImport = this.getOrAddImport(type), { typeName } = annotation;
+        this.editor.edits.overwrite(typeName.start, typeName.end, localTypeImport ?? "");
       }
-      return id;
     };
-    this.sbImportDeclarations = this.getAllSbImportDeclarations(file);
+    this.sbImportDeclarations = editor.program.body.filter(
+      (statement) => {
+        if (statement.type !== "ImportDeclaration")
+          return !1;
+        let source = statement.source.value;
+        return source.startsWith("@storybook/csf") || !source.startsWith("@storybook") ? !1 : statement.specifiers.some((specifier) => {
+          if (specifier.type === "ImportNamespaceSpecifier")
+            throw new Error(
+              `This codemod does not support namespace imports for a ${source} package.
+Replace the namespace import with named imports and try again.`
+            );
+          return specifier.type === "ImportSpecifier" && specifier.imported.type === "Identifier" && storyTypes.includes(specifier.imported.name);
+        });
+      }
+    );
   }
 }, parser = "tsx";
 export {

package/dist/transforms/find-implicit-spies.js

@@ -1,73 +1,44 @@
-import CJS_COMPAT_NODE_URL_ppo5qkb13mm from 'node:url';
-import CJS_COMPAT_NODE_PATH_ppo5qkb13mm from 'node:path';
-import CJS_COMPAT_NODE_MODULE_ppo5qkb13mm from "node:module";
+import CJS_COMPAT_NODE_URL_77qkz8uuglf from 'node:url';
+import CJS_COMPAT_NODE_PATH_77qkz8uuglf from 'node:path';
+import CJS_COMPAT_NODE_MODULE_77qkz8uuglf from "node:module";
 
-var __filename = CJS_COMPAT_NODE_URL_ppo5qkb13mm.fileURLToPath(import.meta.url);
-var __dirname = CJS_COMPAT_NODE_PATH_ppo5qkb13mm.dirname(__filename);
-var require = CJS_COMPAT_NODE_MODULE_ppo5qkb13mm.createRequire(import.meta.url);
+var __filename = CJS_COMPAT_NODE_URL_77qkz8uuglf.fileURLToPath(import.meta.url);
+var __dirname = CJS_COMPAT_NODE_PATH_77qkz8uuglf.dirname(__filename);
+var require = CJS_COMPAT_NODE_MODULE_77qkz8uuglf.createRequire(import.meta.url);
 
 // ------------------------------------------------------------
 // end of CJS compatibility banner, injected by Storybook's esbuild configuration
 // ------------------------------------------------------------
 
 // src/transforms/find-implicit-spies.ts
-import { core as babel, types as t } from "storybook/internal/babel";
-import { loadCsf } from "storybook/internal/csf-tools";
-function findImplicitSpies(path, file, keys) {
-  path.traverse({
-    Identifier: (identifier) => {
-      !keys.includes(identifier.node.name) && /^on[A-Z].*/.test(identifier.node.name) && console.warn(identifier.buildCodeFrameError(`${file} Possible implicit spy found`).message);
-    }
+import {
+  loadCsf,
+  storyShapeError,
+  walk
+} from "storybook/internal/csf-tools";
+function findImplicitSpies(editor, node, keys) {
+  walk(node, (identifier) => {
+    identifier.type === "Identifier" && !keys.includes(identifier.name) && /^on[A-Z].*/.test(identifier.name) && console.warn(storyShapeError("Possible implicit spy found", identifier, editor).message);
   });
 }
-function getAnnotationKeys(file, storyName, annotationName) {
-  let argKeys = [];
-  return file.path.traverse({
-    // CSF2 play function Story.args =
-    AssignmentExpression: (path) => {
-      let left = path.get("left");
-      if (!left.isMemberExpression())
-        return;
-      let object = left.get("object");
-      if (!(object.isIdentifier() && object.node.name === storyName))
-        return;
-      let property = left.get("property"), right = path.get("right");
-      property.isIdentifier() && property.node.name === annotationName && right.isObjectExpression() && argKeys.push(
-        ...right.node.properties.flatMap(
-          (value) => t.isObjectProperty(value) && t.isIdentifier(value.key) ? [value.key.name] : []
-        )
-      );
-    },
-    // CSF3 const Story = {args: () => {} };
-    VariableDeclarator: (path) => {
-      let id = path.get("id"), init = path.get("init");
-      if (!(id.isIdentifier() && id.node.name === storyName) || !init.isObjectExpression())
-        return;
-      let args = init.get("properties").flatMap((it) => it.isObjectProperty() ? [it] : []).find((it) => {
-        let argKey = it.get("key");
-        return argKey.isIdentifier() && argKey.node.name === annotationName;
-      });
-      if (!args)
-        return;
-      let argsValue = args.get("value");
-      !argsValue || !argsValue.isObjectExpression() || argKeys.push(
-        ...argsValue.node.properties.flatMap(
-          (value) => t.isObjectProperty(value) && t.isIdentifier(value.key) ? [value.key.name] : []
-        )
-      );
+var getObjectExpressionKeys = (node) => node?.type === "ObjectExpression" ? node.properties.flatMap(
+  (value) => value.type === "Property" && !value.method && value.key.type === "Identifier" ? [value.key.name] : []
+) : [], findProperty = (object, name) => object.properties.find(
+  (it) => it.type === "Property" && !it.method && it.key.type === "Identifier" && it.key.name === name
+);
+function findAnnotations(program, storyName, annotationName) {
+  let values = [];
+  return walk(program, (node) => {
+    if (node.type === "AssignmentExpression" && node.left.type === "MemberExpression" && node.left.object.type === "Identifier" && node.left.object.name === storyName && node.left.property.type === "Identifier" && node.left.property.name === annotationName && values.push(node.right), node.type === "VariableDeclarator" && node.id.type === "Identifier" && node.id.name === storyName && node.init?.type === "ObjectExpression") {
+      let property = findProperty(node.init, annotationName);
+      property && values.push(property.value);
     }
-  }), argKeys;
+  }), values;
 }
-var getObjectExpressionKeys = (node) => t.isObjectExpression(node) ? node.properties.flatMap(
-  (value) => t.isObjectProperty(value) && t.isIdentifier(value.key) ? [value.key.name] : []
-) : [];
 async function transform(info) {
-  let csf = loadCsf(info.source, { makeTitle: (title) => title }), fileNode = csf._ast, file = new babel.File(
-    { filename: info.path },
-    { code: info.source, ast: fileNode }
-  );
+  let csf = loadCsf(info.source, { makeTitle: (title) => title, fileName: info.path });
   csf.parse();
-  let metaKeys = [
+  let editor = csf.editor, metaKeys = [
     ...getObjectExpressionKeys(csf._metaAnnotations.args),
     ...getObjectExpressionKeys(csf._metaAnnotations.argTypes)
   ];
@@ -76,31 +47,13 @@ async function transform(info) {
       return;
     let allKeys = [
       ...metaKeys,
-      ...getAnnotationKeys(file, name, "args"),
-      ...getAnnotationKeys(file, name, "argTypes")
+      ...findAnnotations(editor.program, name, "args").flatMap(getObjectExpressionKeys),
+      ...findAnnotations(editor.program, name, "argTypes").flatMap(getObjectExpressionKeys)
     ];
-    file.path.traverse({
-      // CSF2 play function Story.play =
-      AssignmentExpression: (path) => {
-        let left = path.get("left");
-        if (!left.isMemberExpression())
-          return;
-        let object = left.get("object");
-        if (!(object.isIdentifier() && object.node.name === name))
-          return;
-        let property = left.get("property");
-        property.isIdentifier() && property.node.name === "play" && findImplicitSpies(path, info.path, allKeys);
-      },
-      // CSF3 play function: const Story = {play: () => {} };
-      VariableDeclarator: (path) => {
-        let id = path.get("id"), init = path.get("init");
-        if (!(id.isIdentifier() && id.node.name === name) || !init.isObjectExpression())
-          return;
-        let play = init.get("properties").flatMap((it) => it.isObjectProperty() ? [it] : []).find((it) => {
-          let argKey = it.get("key");
-          return argKey.isIdentifier() && argKey.node.name === "play";
-        });
-        play && findImplicitSpies(play, info.path, allKeys);
+    walk(editor.program, (node) => {
+      if (node.type === "AssignmentExpression" && node.left.type === "MemberExpression" && node.left.object.type === "Identifier" && node.left.object.name === name && node.left.property.type === "Identifier" && node.left.property.name === "play" && findImplicitSpies(editor, node, allKeys), node.type === "VariableDeclarator" && node.id.type === "Identifier" && node.id.name === name && node.init?.type === "ObjectExpression") {
+        let play = findProperty(node.init, "play");
+        play && findImplicitSpies(editor, play, allKeys);
       }
     });
   });

package/dist/transforms/upgrade-deprecated-types.js

@@ -1,10 +1,10 @@
-import CJS_COMPAT_NODE_URL_ppo5qkb13mm from 'node:url';
-import CJS_COMPAT_NODE_PATH_ppo5qkb13mm from 'node:path';
-import CJS_COMPAT_NODE_MODULE_ppo5qkb13mm from "node:module";
+import CJS_COMPAT_NODE_URL_77qkz8uuglf from 'node:url';
+import CJS_COMPAT_NODE_PATH_77qkz8uuglf from 'node:path';
+import CJS_COMPAT_NODE_MODULE_77qkz8uuglf from "node:module";
 
-var __filename = CJS_COMPAT_NODE_URL_ppo5qkb13mm.fileURLToPath(import.meta.url);
-var __dirname = CJS_COMPAT_NODE_PATH_ppo5qkb13mm.dirname(__filename);
-var require = CJS_COMPAT_NODE_MODULE_ppo5qkb13mm.createRequire(import.meta.url);
+var __filename = CJS_COMPAT_NODE_URL_77qkz8uuglf.fileURLToPath(import.meta.url);
+var __dirname = CJS_COMPAT_NODE_PATH_77qkz8uuglf.dirname(__filename);
+var require = CJS_COMPAT_NODE_MODULE_77qkz8uuglf.createRequire(import.meta.url);
 
 // ------------------------------------------------------------
 // end of CJS compatibility banner, injected by Storybook's esbuild configuration
@@ -13,7 +13,7 @@ import {
   parser,
   transform,
   upgradeDeprecatedTypes
-} from "../_node-chunks/chunk-I4FB73AT.js";
+} from "../_node-chunks/chunk-5VYHV7J4.js";
 export {
   transform as default,
   parser,

package/dist/transforms/upgrade-hierarchy-separators.js

@@ -1,10 +1,10 @@
-import CJS_COMPAT_NODE_URL_ppo5qkb13mm from 'node:url';
-import CJS_COMPAT_NODE_PATH_ppo5qkb13mm from 'node:path';
-import CJS_COMPAT_NODE_MODULE_ppo5qkb13mm from "node:module";
+import CJS_COMPAT_NODE_URL_77qkz8uuglf from 'node:url';
+import CJS_COMPAT_NODE_PATH_77qkz8uuglf from 'node:path';
+import CJS_COMPAT_NODE_MODULE_77qkz8uuglf from "node:module";
 
-var __filename = CJS_COMPAT_NODE_URL_ppo5qkb13mm.fileURLToPath(import.meta.url);
-var __dirname = CJS_COMPAT_NODE_PATH_ppo5qkb13mm.dirname(__filename);
-var require = CJS_COMPAT_NODE_MODULE_ppo5qkb13mm.createRequire(import.meta.url);
+var __filename = CJS_COMPAT_NODE_URL_77qkz8uuglf.fileURLToPath(import.meta.url);
+var __dirname = CJS_COMPAT_NODE_PATH_77qkz8uuglf.dirname(__filename);
+var require = CJS_COMPAT_NODE_MODULE_77qkz8uuglf.createRequire(import.meta.url);
 
 // ------------------------------------------------------------
 // end of CJS compatibility banner, injected by Storybook's esbuild configuration

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@storybook/codemod",
-  "version": "11.0.0-alpha.7",
+  "version": "11.0.0-alpha.8",
   "description": "A collection of codemod scripts written with JSCodeshift",
   "keywords": [
     "storybook"
@@ -46,7 +46,7 @@
     "es-toolkit": "^1.43.0",
     "jscodeshift": "^17.4.0",
     "prettier": "^3.7.1",
-    "storybook": "11.0.0-alpha.7",
+    "storybook": "11.0.0-alpha.8",
     "tiny-invariant": "^1.3.1",
     "tinyglobby": "^0.2.13"
   },

package/dist/_node-chunks/chunk-I4FB73AT.js

@@ -1,97 +0,0 @@
-import CJS_COMPAT_NODE_URL_ppo5qkb13mm from 'node:url';
-import CJS_COMPAT_NODE_PATH_ppo5qkb13mm from 'node:path';
-import CJS_COMPAT_NODE_MODULE_ppo5qkb13mm from "node:module";
-
-var __filename = CJS_COMPAT_NODE_URL_ppo5qkb13mm.fileURLToPath(import.meta.url);
-var __dirname = CJS_COMPAT_NODE_PATH_ppo5qkb13mm.dirname(__filename);
-var require = CJS_COMPAT_NODE_MODULE_ppo5qkb13mm.createRequire(import.meta.url);
-
-// ------------------------------------------------------------
-// end of CJS compatibility banner, injected by Storybook's esbuild configuration
-// ------------------------------------------------------------
-
-// src/transforms/upgrade-deprecated-types.ts
-import { core as babel, types as t } from "storybook/internal/babel";
-import { loadCsf, printCsf } from "storybook/internal/csf-tools";
-import { logger } from "storybook/internal/node-logger";
-import prettier from "prettier";
-var deprecatedTypes = [
-  "ComponentStory",
-  "ComponentStoryFn",
-  "ComponentStoryObj",
-  "ComponentMeta",
-  "Story"
-];
-function migrateType(oldType) {
-  return oldType === "Story" || oldType === "ComponentStory" ? "StoryFn" : oldType.replace("Component", "");
-}
-async function transform(info, api, options) {
-  let csf = loadCsf(info.source, { makeTitle: (title) => title }), fileNode = csf._ast, file = new babel.File(
-    { filename: info.path },
-    { code: info.source, ast: fileNode }
-  );
-  upgradeDeprecatedTypes(file);
-  let output = printCsf(csf).code;
-  try {
-    output = await prettier.format(output, {
-      ...await prettier.resolveConfig(info.path),
-      filepath: info.path
-    });
-  } catch {
-    logger.log(`Failed applying prettier to ${info.path}.`);
-  }
-  return output;
-}
-var parser = "tsx";
-function upgradeDeprecatedTypes(file) {
-  let importedNamespaces = /* @__PURE__ */ new Set(), typeReferencesToUpdate = /* @__PURE__ */ new Set(), existingImports = [];
-  file.path.traverse({
-    ImportDeclaration: (path) => {
-      existingImports.push(
-        ...path.get("specifiers").map((specifier) => ({
-          name: specifier.node.local.name,
-          isAlias: !(specifier.isImportSpecifier() && t.isIdentifier(specifier.node.imported) && specifier.node.local.name === specifier.node.imported.name),
-          path: specifier
-        }))
-      ), path.node.source.value.startsWith("@storybook") && path.get("specifiers").forEach((specifier) => {
-        if (specifier.isImportNamespaceSpecifier() && importedNamespaces.add(specifier.node.local.name), !specifier.isImportSpecifier())
-          return;
-        let imported = specifier.get("imported");
-        if (imported.isIdentifier() && deprecatedTypes.includes(imported.node.name)) {
-          imported.node.name === specifier.node.local.name && typeReferencesToUpdate.add(specifier.node.local.name);
-          let newType = migrateType(imported.node.name);
-          if (!existingImports.some((it) => it.name === newType))
-            imported.replaceWith(t.identifier(newType)), existingImports.push({ name: newType, isAlias: !1, path: specifier });
-          else {
-            let existingImport = existingImports.find((it) => it.name === newType && it.isAlias);
-            if (existingImport)
-              throw existingImport.path.buildCodeFrameError(
-                `This codemod does not support local imports that are called the same as a storybook import.
-Rename this local import and try again.`
-              );
-            specifier.remove();
-          }
-        }
-      });
-    }
-  }), file.path.traverse({
-    TSTypeReference: (path) => {
-      let typeName = path.get("typeName");
-      if (typeName.isIdentifier())
-        typeReferencesToUpdate.has(typeName.node.name) && typeName.replaceWith(t.identifier(migrateType(typeName.node.name)));
-      else if (typeName.isTSQualifiedName()) {
-        let namespace = typeName.get("left");
-        if (namespace.isIdentifier() && importedNamespaces.has(namespace.node.name)) {
-          let right = typeName.get("right");
-          deprecatedTypes.includes(right.node.name) && right.replaceWith(t.identifier(migrateType(right.node.name)));
-        }
-      }
-    }
-  });
-}
-
-export {
-  transform,
-  parser,
-  upgradeDeprecatedTypes
-};