Merge branch 'kasper/install-blocks' into kasper/mdx-update-glob-in-c…

…odemod # Conflicts: # code/lib/cli/src/migrate.ts
storybookjs · Mar 30, 2023 · 919d964 · 919d964
2 parents 299293a + 6bcd662
commit 919d964
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/code/lib/cli/src/migrate.ts b/code/lib/cli/src/migrate.ts
@@ -1,16 +1,31 @@
 import { listCodemods, runCodemod } from '@storybook/codemod';
 import { runFixes } from './automigrate';
 import { bareMdxStoriesGlob } from './automigrate/fixes/bare-mdx-stories-glob';
+import { JsPackageManagerFactory } from './js-package-manager';
+import { getStorybookVersionSpecifier } from './helpers';
 
-export async function migrate(migration: any, { glob, dryRun, list, rename, logger, parser }: any) {
+const logger = console;
+
+export async function migrate(migration: any, { glob, dryRun, list, rename, parser }: any) {
   if (list) {
     listCodemods().forEach((key: any) => logger.log(key));
   } else if (migration) {
-    if (migration === 'mdx-to-csf') {
+    if (migration === 'mdx-to-csf' && !dryRun) {
       await runFixes({ fixes: [bareMdxStoriesGlob] });
+      await addStorybookBlocksPackage();
     }
     await runCodemod(migration, { glob, dryRun, logger, rename, parser });
   } else {
     throw new Error('Migrate: please specify a migration name or --list');
   }
 }
+
+export async function addStorybookBlocksPackage() {
+  const packageManager = JsPackageManagerFactory.getPackageManager();
+  const packageJson = packageManager.retrievePackageJson();
+  const versionToInstall = getStorybookVersionSpecifier(packageManager.retrievePackageJson());
+  logger.info(`✅ Adding "@storybook/blocks" package`);
+  await packageManager.addDependencies({ installAsDevDependencies: true, packageJson }, [
+    `@storybook/blocks@${versionToInstall}`,
+  ]);
+}