Maintenance: Add flag to toggle default automigrations

code/lib/cli-storybook/src/automigrate/fixes/addon-experimental-test.ts

@@ -1,5 +1,4 @@
 import { readFileSync, writeFileSync } from 'fs';
-import picocolors from 'picocolors';
 
 import { findFilesUp } from '../../util';
 import type { Fix } from '../types';

code/lib/cli-storybook/src/automigrate/fixes/addon-mdx-gfm-remove.ts

@@ -1,7 +1,5 @@
 import { getAddonNames, removeAddon } from 'storybook/internal/common';
 
-import picocolors from 'picocolors';
-
 import type { Fix } from '../types';
 
 type AddonMdxGfmOptions = true;

code/lib/cli-storybook/src/automigrate/fixes/upgrade-storybook-related-dependencies.ts

@@ -67,7 +67,7 @@ function isValidVersionType(packageName: string, specifier: string) {
 export const upgradeStorybookRelatedDependencies = {
   id: 'upgrade-storybook-related-dependencies',
   promptType: 'auto',
-  promptDefaultValue: false,
+  defaultSelected: false,
 
   async check({ packageManager, storybookVersion }) {
     logger.debug('Checking for incompatible storybook packages...');

code/lib/cli-storybook/src/automigrate/index.ts

@@ -327,7 +327,7 @@ export async function runFixes({
           const shouldRun = await prompt.confirm(
             {
               message: `Do you want to run the '${picocolors.cyan(f.id)}' migration on your project?`,
-              initialValue: f.promptDefaultValue ?? true,
+              initialValue: f.defaultSelected ?? true,
             },
             {
               onCancel: () => {

code/lib/cli-storybook/src/automigrate/multi-project.ts

@@ -235,13 +235,14 @@ export async function promptForAutomigrations(
       value: am.fix.id,
       label,
       hint: hint.join('\n'),
+      defaultSelected: am.fix.defaultSelected ?? true,
     };
   });
 
   const selectedIds = await prompt.multiselect({
     message: 'Select automigrations to run',
     options: choices,
-    initialValues: choices.map((c) => c.value),
+    initialValues: choices.filter((c) => c.defaultSelected).map((c) => c.value),
   });
 
   return automigrations.filter((am) => selectedIds.includes(am.fix.id));

code/lib/cli-storybook/src/automigrate/types.ts

@@ -40,7 +40,8 @@ type BaseFix<ResultType = any> = {
   check: (options: CheckOptions) => Promise<ResultType | null>;
   /** Keep the prompt message short and concise. */
   prompt: () => string;
-  promptDefaultValue?: boolean;
+  /** Whether the automigration is selected by default when the user is prompted. */
+  defaultSelected?: boolean;
   link?: string;
 };
 

code/lib/cli-storybook/src/bin/index.ts

@@ -228,6 +228,7 @@ command('automigrate [fixId]')
   )
   .option('--skip-doctor', 'Skip doctor check')
   .action(async (fixId, options) => {
+    prompt.setPromptLibrary('clack');
     await doAutomigrate({ fixId, ...options }).catch(handleCommandFailure);
   });