Addon Test: Prompt switch to experimental-nextjs-vite

code/addons/test/src/postinstall.ts

@@ -7,8 +7,10 @@ import {
   extractProperFrameworkName,
   loadAllPresets,
   loadMainConfig,
+  serverResolve,
   validateFrameworkName,
 } from 'storybook/internal/common';
+import { readConfig, writeConfig } from 'storybook/internal/csf-tools';
 import { colors, logger } from 'storybook/internal/node-logger';
 
 // eslint-disable-next-line depend/ban-dependencies
@@ -53,6 +55,34 @@ export default async function postInstall(options: PostinstallOptions) {
   // if Vitest is installed, we use the same version to keep consistency across Vitest packages
   const vitestVersionToInstall = vitestVersionSpecifier ?? 'latest';
 
+  if (info.frameworkPackageName === '@storybook/nextjs') {
+    const out = await prompts({
+      type: 'confirm',
+      name: 'migrateToExperimentalNextjsVite',
+      message: `To use this addon, you should migrate to use @storybook/experimental-nextjs-vite, do you want to migrate?`,
+      initial: true,
+    });
+
+    if (out.migrateToExperimentalNextjsVite) {
+      await packageManager.addDependencies({ installAsDevDependencies: true }, [
+        '@storybook/experimental-nextjs-vite',
+      ]);
+
+      await packageManager.removeDependencies({}, ['@storybook/next']);
+
+      // update the storybook main.ts config file
+      const mainJsPath = serverResolve(resolve(options.configDir, 'main')) as string;
+
+      const config = await readConfig(mainJsPath);
+
+      config.setFieldValue(['framework'], '@storybook/experimental-nextjs-vite');
+
+      await writeConfig(config, mainJsPath);
+
+      info.frameworkPackageName = '@storybook/experimental-nextjs-vite';
+    }
+  }
+
   const annotationsImport = [
     '@storybook/nextjs',
     '@storybook/experimental-nextjs-vite',