@storybook/addon-vitest 0.0.0-pr-32717-sha-6b32df73 → 0.0.0-pr-32882-sha-d65d0e7f

package/dist/postinstall.js

@@ -1,29 +1,33 @@
-import CJS_COMPAT_NODE_URL_hwd9utotfy8 from 'node:url';
-import CJS_COMPAT_NODE_PATH_hwd9utotfy8 from 'node:path';
-import CJS_COMPAT_NODE_MODULE_hwd9utotfy8 from "node:module";
+import CJS_COMPAT_NODE_URL_2v687h26vhe from 'node:url';
+import CJS_COMPAT_NODE_PATH_2v687h26vhe from 'node:path';
+import CJS_COMPAT_NODE_MODULE_2v687h26vhe from "node:module";
 
-var __filename = CJS_COMPAT_NODE_URL_hwd9utotfy8.fileURLToPath(import.meta.url);
-var __dirname = CJS_COMPAT_NODE_PATH_hwd9utotfy8.dirname(__filename);
-var require = CJS_COMPAT_NODE_MODULE_hwd9utotfy8.createRequire(import.meta.url);
+var __filename = CJS_COMPAT_NODE_URL_2v687h26vhe.fileURLToPath(import.meta.url);
+var __dirname = CJS_COMPAT_NODE_PATH_2v687h26vhe.dirname(__filename);
+var require = CJS_COMPAT_NODE_MODULE_2v687h26vhe.createRequire(import.meta.url);
 
 // ------------------------------------------------------------
 // end of CJS compatibility banner, injected by Storybook's esbuild configuration
 // ------------------------------------------------------------
 import {
+  execa,
   resolvePackageDir
-} from "./_node-chunks/chunk-C3FRTXFF.js";
+} from "./_node-chunks/chunk-POD5FBDR.js";
 import {
-  any
-} from "./_node-chunks/chunk-NC7DRZM4.js";
+  any,
+  getAddonNames,
+  up
+} from "./_node-chunks/chunk-PWT7VWXJ.js";
 import {
-  DOCUMENTATION_LINK
-} from "./_node-chunks/chunk-KIUHNMC3.js";
+  DOCUMENTATION_LINK,
+  SUPPORTED_FRAMEWORKS
+} from "./_node-chunks/chunk-VETR5WKO.js";
 import {
   dirname,
   join,
   relative,
   resolve
-} from "./_node-chunks/chunk-7U3E5OTZ.js";
+} from "./_node-chunks/chunk-RFFTF3P3.js";
 import {
   require_compare,
   require_constants,
@@ -33,12 +37,12 @@ import {
   require_parse_options,
   require_re,
   require_semver
-} from "./_node-chunks/chunk-OQHY5QS2.js";
+} from "./_node-chunks/chunk-A7VETTJF.js";
 import {
   __commonJS,
   __name,
   __toESM
-} from "./_node-chunks/chunk-RU4LPJPR.js";
+} from "./_node-chunks/chunk-ZGPYNFM4.js";
 
 // ../../node_modules/semver/functions/parse.js
 var require_parse = __commonJS({
@@ -360,7 +364,7 @@ var require_coerce = __commonJS({
     var SemVer = require_semver();
     var parse = require_parse();
     var { safeRe: re, t } = require_re();
-    var coerce = /* @__PURE__ */ __name((version, options) => {
+    var coerce2 = /* @__PURE__ */ __name((version, options) => {
       if (version instanceof SemVer) {
         return version;
       }
@@ -395,7 +399,7 @@ var require_coerce = __commonJS({
       const build = options.includePrerelease && match[6] ? `+${match[6]}` : "";
       return parse(`${major}.${minor}.${patch}${prerelease}${build}`, options);
     }, "coerce");
-    module.exports = coerce;
+    module.exports = coerce2;
   }
 });
 
@@ -968,7 +972,7 @@ var require_max_satisfying = __commonJS({
     "use strict";
     var SemVer = require_semver();
     var Range = require_range();
-    var maxSatisfying = /* @__PURE__ */ __name((versions2, range, options) => {
+    var maxSatisfying = /* @__PURE__ */ __name((versions, range, options) => {
       let max = null;
       let maxSV = null;
       let rangeObj = null;
@@ -977,7 +981,7 @@ var require_max_satisfying = __commonJS({
       } catch (er) {
         return null;
       }
-      versions2.forEach((v) => {
+      versions.forEach((v) => {
         if (rangeObj.test(v)) {
           if (!max || maxSV.compare(v) === -1) {
             max = v;
@@ -997,7 +1001,7 @@ var require_min_satisfying = __commonJS({
     "use strict";
     var SemVer = require_semver();
     var Range = require_range();
-    var minSatisfying = /* @__PURE__ */ __name((versions2, range, options) => {
+    var minSatisfying = /* @__PURE__ */ __name((versions, range, options) => {
       let min = null;
       let minSV = null;
       let rangeObj = null;
@@ -1006,7 +1010,7 @@ var require_min_satisfying = __commonJS({
       } catch (er) {
         return null;
       }
-      versions2.forEach((v) => {
+      versions.forEach((v) => {
         if (rangeObj.test(v)) {
           if (!min || minSV.compare(v) === 1) {
             min = v;
@@ -1204,11 +1208,11 @@ var require_simplify = __commonJS({
     "use strict";
     var satisfies2 = require_satisfies();
     var compare = require_compare();
-    module.exports = (versions2, range, options) => {
+    module.exports = (versions, range, options) => {
       const set = [];
       let first = null;
       let prev = null;
-      const v = versions2.sort((a, b) => compare(a, b, options));
+      const v = versions.sort((a, b) => compare(a, b, options));
       for (const version of v) {
         const included = satisfies2(version, range, options);
         if (included) {
@@ -1440,7 +1444,7 @@ var require_semver2 = __commonJS({
     var gte = require_gte();
     var lte = require_lte();
     var cmp = require_cmp();
-    var coerce = require_coerce();
+    var coerce2 = require_coerce();
     var Comparator = require_comparator();
     var Range = require_range();
     var satisfies2 = require_satisfies();
@@ -1478,7 +1482,7 @@ var require_semver2 = __commonJS({
       gte,
       lte,
       cmp,
-      coerce,
+      coerce: coerce2,
       Comparator,
       Range,
       satisfies: satisfies2,
@@ -1509,24 +1513,56 @@ var require_semver2 = __commonJS({
 import { existsSync } from "node:fs";
 import * as fs2 from "node:fs/promises";
 import { writeFile } from "node:fs/promises";
-import { babelParse, generate } from "storybook/internal/babel";
-import { AddonVitestService } from "storybook/internal/cli";
+import { isAbsolute, posix, sep } from "node:path";
+import { fileURLToPath, pathToFileURL } from "node:url";
+import { babelParse, generate, traverse } from "storybook/internal/babel";
 import {
   JsPackageManagerFactory,
   formatFileContent,
+  getInterpretedFile,
   getProjectRoot,
-  getStorybookInfo,
-  versions
+  isCI as isCI2,
+  loadMainConfig,
+  scanAndTransformFiles,
+  transformImportFiles
 } from "storybook/internal/common";
-import { CLI_COLORS } from "storybook/internal/node-logger";
-import {
-  AddonVitestPostinstallError,
-  AddonVitestPostinstallPrerequisiteCheckError
-} from "storybook/internal/server-errors";
-import { SupportedFramework } from "storybook/internal/types";
+import { experimental_loadStorybook } from "storybook/internal/core-server";
+import { readConfig, writeConfig } from "storybook/internal/csf-tools";
+import { logger as logger2 } from "storybook/internal/node-logger";
+
+// ../../node_modules/empathic/package.mjs
+function up2(options) {
+  return up("package.json", options);
+}
+__name(up2, "up");
+
+// src/postinstall.ts
 var import_semver = __toESM(require_semver2(), 1);
+import prompts from "prompts";
 import { dedent } from "ts-dedent";
 
+// src/postinstall-logger.ts
+import { isCI } from "storybook/internal/common";
+import { colors, logger } from "storybook/internal/node-logger";
+var fancy = process.platform !== "win32" || isCI() || process.env.TERM === "xterm-256color";
+var step = colors.gray("\u203A");
+var info = colors.blue(fancy ? "\u2139" : "i");
+var success = colors.green(fancy ? "\u2714" : "\u221A");
+var warning = colors.orange(fancy ? "\u26A0" : "\u203C");
+var error = colors.red(fancy ? "\u2716" : "\xD7");
+var baseOptions = {
+  borderStyle: "round",
+  padding: 1
+};
+var print = /* @__PURE__ */ __name((message, options) => {
+  logger.line(1);
+  logger.logBox(message, { ...baseOptions, ...options });
+}, "print");
+var printInfo = /* @__PURE__ */ __name((title, message, options) => print(message, { borderColor: "blue", title, ...options }), "printInfo");
+var printWarning = /* @__PURE__ */ __name((title, message, options) => print(message, { borderColor: "yellow", title, ...options }), "printWarning");
+var printError = /* @__PURE__ */ __name((title, message, options) => print(message, { borderColor: "red", title, ...options }), "printError");
+var printSuccess = /* @__PURE__ */ __name((title, message, options) => print(message, { borderColor: "green", title, ...options }), "printSuccess");
+
 // src/updateVitestFile.ts
 import * as fs from "node:fs/promises";
 var loadTemplate = /* @__PURE__ */ __name(async (name, replacements) => {
@@ -1580,7 +1616,10 @@ var updateConfigFile = /* @__PURE__ */ __name((source, target) => {
   } else if (targetExportDefault.declaration.type === "CallExpression" && targetExportDefault.declaration.callee.type === "Identifier" && targetExportDefault.declaration.callee.name === "defineConfig" && targetExportDefault.declaration.arguments[0]?.type === "ObjectExpression") {
     canHandleConfig = true;
   } else if (targetExportDefault.declaration.type === "CallExpression" && targetExportDefault.declaration.callee.type === "Identifier" && targetExportDefault.declaration.callee.name === "mergeConfig" && targetExportDefault.declaration.arguments.length >= 2) {
-    canHandleConfig = true;
+    const defineConfigNodes = targetExportDefault.declaration.arguments.filter(
+      (arg) => arg?.type === "CallExpression" && arg.callee.type === "Identifier" && arg.callee.name === "defineConfig" && arg.arguments[0]?.type === "ObjectExpression"
+    );
+    canHandleConfig = defineConfigNodes.length > 0;
   }
   if (!canHandleConfig) {
     return false;
@@ -1613,24 +1652,20 @@ var updateConfigFile = /* @__PURE__ */ __name((source, target) => {
           mergeProperties(properties, exportDefault.declaration.arguments[0].properties);
           updated = true;
         } else if (exportDefault.declaration.type === "CallExpression" && exportDefault.declaration.callee.type === "Identifier" && exportDefault.declaration.callee.name === "mergeConfig" && exportDefault.declaration.arguments.length >= 2) {
-          const configObjectNodes = [];
-          for (const arg of exportDefault.declaration.arguments) {
-            if (arg?.type === "CallExpression" && arg.callee.type === "Identifier" && arg.callee.name === "defineConfig" && arg.arguments[0]?.type === "ObjectExpression") {
-              configObjectNodes.push(arg.arguments[0]);
-            } else if (arg?.type === "ObjectExpression") {
-              configObjectNodes.push(arg);
-            }
-          }
-          const configObjectWithTest = configObjectNodes.find(
-            (obj) => obj.properties.some(
+          const defineConfigNodes = exportDefault.declaration.arguments.filter(
+            (arg) => arg?.type === "CallExpression" && arg.callee.type === "Identifier" && arg.callee.name === "defineConfig" && arg.arguments[0]?.type === "ObjectExpression"
+          );
+          const defineConfigNodeWithTest = defineConfigNodes.find(
+            (node) => node.arguments[0].type === "ObjectExpression" && node.arguments[0].properties.some(
               (p) => p.type === "ObjectProperty" && p.key.type === "Identifier" && p.key.name === "test"
             )
           );
-          const targetConfigObject = configObjectWithTest || configObjectNodes[0];
-          if (!targetConfigObject) {
+          const defineConfigNode = defineConfigNodeWithTest || defineConfigNodes[0];
+          if (!defineConfigNode) {
             return false;
           }
-          const existingTestProp = targetConfigObject.properties.find(
+          const defineConfigProps = defineConfigNode.arguments[0];
+          const existingTestProp = defineConfigProps.properties.find(
             (p) => p.type === "ObjectProperty" && p.key.type === "Identifier" && p.key.name === "test"
           );
           if (existingTestProp && existingTestProp.value.type === "ObjectExpression") {
@@ -1642,16 +1677,6 @@ var updateConfigFile = /* @__PURE__ */ __name((source, target) => {
                 (p) => p.type === "ObjectProperty" && p.key.type === "Identifier" && (p.key.name === "workspace" || p.key.name === "projects")
               );
               if (workspaceOrProjectsProp && workspaceOrProjectsProp.value.type === "ArrayExpression") {
-                const coverageProp = existingTestProp.value.properties.find(
-                  (p) => p.type === "ObjectProperty" && p.key.type === "Identifier" && p.key.name === "coverage"
-                );
-                const testPropsWithoutCoverage = existingTestProp.value.properties.filter(
-                  (p) => p !== coverageProp
-                );
-                const testConfigForProject = {
-                  type: "ObjectExpression",
-                  properties: testPropsWithoutCoverage
-                };
                 const existingTestProject = {
                   type: "ObjectExpression",
                   properties: [
@@ -1665,28 +1690,25 @@ var updateConfigFile = /* @__PURE__ */ __name((source, target) => {
                     {
                       type: "ObjectProperty",
                       key: { type: "Identifier", name: "test" },
-                      value: testConfigForProject,
+                      value: existingTestProp.value,
                       computed: false,
                       shorthand: false
                     }
                   ]
                 };
                 workspaceOrProjectsProp.value.elements.unshift(existingTestProject);
-                targetConfigObject.properties = targetConfigObject.properties.filter(
+                defineConfigProps.properties = defineConfigProps.properties.filter(
                   (p) => p !== existingTestProp
                 );
-                if (coverageProp && templateTestProp.value.type === "ObjectExpression") {
-                  templateTestProp.value.properties.unshift(coverageProp);
-                }
-                mergeProperties(properties, targetConfigObject.properties);
+                mergeProperties(properties, defineConfigProps.properties);
               } else {
-                mergeProperties(properties, targetConfigObject.properties);
+                mergeProperties(properties, defineConfigProps.properties);
               }
             } else {
-              mergeProperties(properties, targetConfigObject.properties);
+              mergeProperties(properties, defineConfigProps.properties);
             }
           } else {
-            mergeProperties(properties, targetConfigObject.properties);
+            mergeProperties(properties, defineConfigProps.properties);
           }
           updated = true;
         }
@@ -1735,133 +1757,274 @@ var updateWorkspaceFile = /* @__PURE__ */ __name((source, target) => {
 var ADDON_NAME = "@storybook/addon-vitest";
 var EXTENSIONS = [".ts", ".tsx", ".js", ".jsx", ".cts", ".mts", ".cjs", ".mjs"];
 var addonA11yName = "@storybook/addon-a11y";
+var hasErrors = false;
+function nameMatches(name, pattern) {
+  if (name === pattern) {
+    return true;
+  }
+  if (name.includes(`${pattern}${sep}`)) {
+    return true;
+  }
+  if (name.includes(`${pattern}${posix.sep}`)) {
+    return true;
+  }
+  return false;
+}
+__name(nameMatches, "nameMatches");
+var logErrors = /* @__PURE__ */ __name((...args) => {
+  hasErrors = true;
+  printError(...args);
+}, "logErrors");
+var findFile = /* @__PURE__ */ __name((basename, extensions = EXTENSIONS) => any(
+  extensions.map((ext) => basename + ext),
+  { last: getProjectRoot() }
+), "findFile");
 async function postInstall(options) {
-  const errors = [];
-  const { logger, prompt } = options;
+  printSuccess(
+    "\u{1F44B} Howdy!",
+    dedent`
+      I'm the installation helper for ${ADDON_NAME}
+
+      Hold on for a moment while I look at your project and get it set up...
+    `
+  );
   const packageManager = JsPackageManagerFactory.getPackageManager({
     force: options.packageManager
   });
-  const findFile = /* @__PURE__ */ __name((basename, extensions = EXTENSIONS) => any(
-    extensions.map((ext) => basename + ext),
-    { last: getProjectRoot(), cwd: options.configDir }
-  ), "findFile");
   const vitestVersionSpecifier = await packageManager.getInstalledVersion("vitest");
+  const coercedVitestVersion = vitestVersionSpecifier ? (0, import_semver.coerce)(vitestVersionSpecifier) : null;
   const isVitest3_2To4 = vitestVersionSpecifier ? (0, import_semver.satisfies)(vitestVersionSpecifier, ">=3.2.0 <4.0.0") : false;
   const isVitest4OrNewer = vitestVersionSpecifier ? (0, import_semver.satisfies)(vitestVersionSpecifier, ">=4.0.0") : true;
-  const info = await getStorybookInfo(options.configDir);
+  const info2 = await getStorybookInfo(options);
   const allDeps = packageManager.getAllDependencies();
-  const addonVitestService = new AddonVitestService();
-  const compatibilityResult = await addonVitestService.validateCompatibility({
-    packageManager,
-    framework: info.framework,
-    builder: info.builder
-  });
-  let result = null;
-  if (!compatibilityResult.compatible && compatibilityResult.reasons) {
-    const reasons = compatibilityResult.reasons.map((r) => `\u2022 ${CLI_COLORS.error(r)}`);
-    reasons.unshift(dedent`
-      Automated setup failed
-      We have found incompatibilities due to the following package incompatibilities:
-    `);
-    reasons.push(
-      dedent`
-        You can fix these issues and rerun the command to reinstall. If you wish to roll back the installation, remove ${ADDON_NAME} from the "addons" array
-        in your main Storybook config file and remove the dependency from your package.json file.
+  const dependencies = [
+    "vitest",
+    "playwright",
+    isVitest4OrNewer ? "@vitest/browser-playwright" : "@vitest/browser"
+  ];
+  const uniqueDependencies = dependencies.filter((p) => !allDeps[p]);
+  const mainJsPath = getInterpretedFile(resolve(options.configDir, "main"));
+  const config = await readConfig(mainJsPath);
+  const hasCustomWebpackConfig = !!config.getFieldNode(["webpackFinal"]);
+  const isInteractive = process.stdout.isTTY && !isCI2();
+  if (nameMatches(info2.frameworkPackageName, "@storybook/nextjs") && !hasCustomWebpackConfig) {
+    const out = options.yes || !isInteractive ? { migrateToNextjsVite: !!options.yes } : await prompts({
+      type: "confirm",
+      name: "migrateToNextjsVite",
+      message: dedent`
+            The addon requires the use of @storybook/nextjs-vite to work with Next.js.
+            https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#install-and-set-up
 
-        Please check the documentation for more information about its requirements and installation:
-        https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}
-      `
-    );
-    result = reasons.map((r) => r.trim()).join("\n\n");
-  }
-  if (result) {
-    logger.error(result);
-    throw new AddonVitestPostinstallPrerequisiteCheckError({
-      reasons: compatibilityResult.reasons
+            Do you want to migrate?
+          `,
+      initial: true
     });
+    if (out.migrateToNextjsVite) {
+      await packageManager.addDependencies({ type: "devDependencies", skipInstall: true }, [
+        "@storybook/nextjs-vite"
+      ]);
+      await packageManager.removeDependencies(["@storybook/nextjs"]);
+      traverse(config._ast, {
+        StringLiteral(path) {
+          if (path.node.value === "@storybook/nextjs") {
+            path.node.value = "@storybook/nextjs-vite";
+          }
+        }
+      });
+      await writeConfig(config, mainJsPath);
+      info2.frameworkPackageName = "@storybook/nextjs-vite";
+      info2.builderPackageName = "@storybook/builder-vite";
+      await scanAndTransformFiles({
+        promptMessage: "Enter a glob to scan for all @storybook/nextjs imports to substitute with @storybook/nextjs-vite:",
+        force: options.yes,
+        dryRun: false,
+        transformFn: /* @__PURE__ */ __name((files, options2, dryRun) => transformImportFiles(files, options2, dryRun), "transformFn"),
+        transformOptions: {
+          "@storybook/nextjs": "@storybook/nextjs-vite"
+        }
+      });
+    }
   }
-  if (!options.skipDependencyManagement) {
-    const versionedDependencies = await addonVitestService.collectDependencies(packageManager);
-    if (info.framework === SupportedFramework.NEXTJS) {
-      const allDeps2 = packageManager.getAllDependencies();
-      if (!allDeps2["@storybook/nextjs-vite"]) {
+  const annotationsImport = SUPPORTED_FRAMEWORKS.find(
+    (f) => nameMatches(info2.frameworkPackageName, f)
+  ) ? info2.frameworkPackageName === "@storybook/nextjs" ? "@storybook/nextjs-vite" : info2.frameworkPackageName : null;
+  const isRendererSupported = !!annotationsImport;
+  const prerequisiteCheck = /* @__PURE__ */ __name(async () => {
+    const reasons = [];
+    if (hasCustomWebpackConfig) {
+      reasons.push("\u2022 The addon can not be used with a custom Webpack configuration.");
+    }
+    if (!nameMatches(info2.frameworkPackageName, "@storybook/nextjs") && !nameMatches(info2.builderPackageName, "@storybook/builder-vite")) {
+      reasons.push(
+        "\u2022 The addon can only be used with a Vite-based Storybook framework or Next.js."
+      );
+    }
+    if (!isRendererSupported) {
+      reasons.push(dedent`
+        • The addon cannot yet be used with ${info2.frameworkPackageName}
+      `);
+    }
+    if (coercedVitestVersion && !(0, import_semver.satisfies)(coercedVitestVersion, ">=3.0.0")) {
+      reasons.push(dedent`
+        • The addon requires Vitest 3.0.0 or higher. You are currently using ${vitestVersionSpecifier}.
+          Please update all of your Vitest dependencies and try again.
+      `);
+    }
+    const mswVersionSpecifier = await packageManager.getInstalledVersion("msw");
+    const coercedMswVersion = mswVersionSpecifier ? (0, import_semver.coerce)(mswVersionSpecifier) : null;
+    if (coercedMswVersion && !(0, import_semver.satisfies)(coercedMswVersion, ">=2.0.0")) {
+      reasons.push(dedent`
+        • The addon uses Vitest behind the scenes, which supports only version 2 and above of MSW. However, we have detected version ${coercedMswVersion.version} in this project.
+        Please update the 'msw' package and try again.
+      `);
+    }
+    if (nameMatches(info2.frameworkPackageName, "@storybook/nextjs")) {
+      const nextVersion = await packageManager.getInstalledVersion("next");
+      if (!nextVersion) {
+        reasons.push(dedent`
+          • You are using @storybook/nextjs without having "next" installed.
+            Please install "next" or use a different Storybook framework integration and try again.
+        `);
       }
     }
-    const v8Version = await packageManager.getInstalledVersion("@vitest/coverage-v8");
-    const istanbulVersion = await packageManager.getInstalledVersion("@vitest/coverage-istanbul");
-    if (!v8Version && !istanbulVersion) {
-      logger.step(
+    if (reasons.length > 0) {
+      reasons.unshift(
+        `@storybook/addon-vitest's automated setup failed due to the following package incompatibilities:`
+      );
+      reasons.push("--------------------------------");
+      reasons.push(
         dedent`
-          You don't seem to have a coverage reporter installed. Vitest needs either V8 or Istanbul to generate coverage reports.
-
-          Adding "@vitest/coverage-v8" to enable coverage reporting.
-          Read more about Vitest coverage providers at https://vitest.dev/guide/coverage.html#coverage-providers
+          You can fix these issues and rerun the command to reinstall. If you wish to roll back the installation, remove ${ADDON_NAME} from the "addons" array
+          in your main Storybook config file and remove the dependency from your package.json file.
         `
       );
+      if (!isRendererSupported) {
+        reasons.push(
+          dedent`
+            Please check the documentation for more information about its requirements and installation:
+            https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}
+          `
+        );
+      } else {
+        reasons.push(
+          dedent`
+            Fear not, however, you can follow the manual installation process instead at:
+            https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#manual-setup
+          `
+        );
+      }
+      return reasons.map((r) => r.trim()).join("\n\n");
     }
-    if (versionedDependencies.length > 0) {
-      logger.step("Adding dependencies to your package.json");
-      logger.log("  " + versionedDependencies.join(", "));
-      await packageManager.addDependencies(
-        { type: "devDependencies", skipInstall: true },
-        versionedDependencies
-      );
+    return null;
+  }, "prerequisiteCheck");
+  const result = await prerequisiteCheck();
+  if (result) {
+    logErrors("\u26D4\uFE0F Sorry!", result);
+    logger2.line(1);
+    return;
+  }
+  if (info2.frameworkPackageName === "@storybook/nextjs") {
+    printInfo(
+      "\u{1F37F} Just so you know...",
+      dedent`
+        It looks like you're using Next.js.
+
+        Adding "@storybook/nextjs-vite/vite-plugin" so you can use it with Vitest.
+
+        More info about the plugin at https://github.com/storybookjs/vite-plugin-storybook-nextjs
+      `
+    );
+    try {
+      const storybookVersion = await packageManager.getInstalledVersion("storybook");
+      uniqueDependencies.push(`@storybook/nextjs-vite@^${storybookVersion}`);
+    } catch (e) {
+      console.error("Failed to install @storybook/nextjs-vite. Please install it manually");
     }
-    if (!options.skipInstall) {
-      await packageManager.installDependencies();
+  }
+  const v8Version = await packageManager.getInstalledVersion("@vitest/coverage-v8");
+  const istanbulVersion = await packageManager.getInstalledVersion("@vitest/coverage-istanbul");
+  if (!v8Version && !istanbulVersion) {
+    printInfo(
+      "\u{1F648} Let me cover this for you",
+      dedent`
+        You don't seem to have a coverage reporter installed. Vitest needs either V8 or Istanbul to generate coverage reports.
+
+        Adding "@vitest/coverage-v8" to enable coverage reporting.
+        Read more about Vitest coverage providers at https://vitest.dev/guide/coverage.html#coverage-providers
+      `
+    );
+    uniqueDependencies.push(`@vitest/coverage-v8`);
+  }
+  const versionedDependencies = uniqueDependencies.map((p) => {
+    if (p.includes("vitest")) {
+      return vitestVersionSpecifier ? `${p}@${vitestVersionSpecifier}` : p;
     }
+    return p;
+  });
+  if (versionedDependencies.length > 0) {
+    await packageManager.addDependencies(
+      { type: "devDependencies", skipInstall: true },
+      versionedDependencies
+    );
+    logger2.line(1);
+    logger2.plain(`${step} Installing dependencies:`);
+    logger2.plain("  " + versionedDependencies.join(", "));
   }
-  if (!options.skipDependencyManagement) {
-    if (!options.skipInstall) {
-      const playwrightErrors = await addonVitestService.installPlaywright(packageManager, {
-        yes: options.yes
+  await packageManager.installDependencies();
+  logger2.line(1);
+  if (options.skipInstall) {
+    logger2.plain("Skipping Playwright installation, please run this command manually:");
+    logger2.plain("  npx playwright install chromium --with-deps");
+  } else {
+    logger2.plain(`${step} Configuring Playwright with Chromium (this might take some time):`);
+    logger2.plain("  npx playwright install chromium --with-deps");
+    try {
+      await packageManager.executeCommand({
+        command: "npx",
+        args: ["playwright", "install", "chromium", "--with-deps"]
       });
-      errors.push(...playwrightErrors);
-    } else {
-      logger.warn(dedent`
-        Playwright browser binaries installation skipped. Please run the following command manually later:
-        ${CLI_COLORS.cta("npx playwright install chromium --with-deps")}
-      `);
+    } catch (e) {
+      console.error("Failed to install Playwright. Please install it manually");
     }
   }
   const fileExtension = allDeps.typescript || findFile("tsconfig", [...EXTENSIONS, ".json"]) ? "ts" : "js";
   const vitestSetupFile = resolve(options.configDir, `vitest.setup.${fileExtension}`);
   if (existsSync(vitestSetupFile)) {
-    const errorMessage = dedent`
-    Found an existing Vitest setup file:
-    ${vitestSetupFile}
-    Please refer to the documentation to complete the setup manually:
-    https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#manual-setup
-  `;
-    logger.line();
-    logger.error(`${errorMessage}
-`);
-    errors.push("Found existing Vitest setup file");
-  } else {
-    logger.step(`Creating a Vitest setup file for Storybook:`);
-    logger.log(`${vitestSetupFile}
-`);
-    const previewExists = EXTENSIONS.map((ext) => resolve(options.configDir, `preview${ext}`)).some(
-      existsSync
-    );
-    const annotationsImport = info.frameworkPackage;
-    const imports = [`import { setProjectAnnotations } from '${annotationsImport}';`];
-    const projectAnnotations = [];
-    if (previewExists) {
-      imports.push(`import * as projectAnnotations from './preview';`);
-      projectAnnotations.push("projectAnnotations");
-    }
-    await writeFile(
-      vitestSetupFile,
+    logErrors(
+      "\u{1F6A8} Oh no!",
       dedent`
+        Found an existing Vitest setup file:
+        ${vitestSetupFile}
+
+        Please refer to the documentation to complete the setup manually:
+        https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#manual-setup
+      `
+    );
+    logger2.line(1);
+    return;
+  }
+  logger2.line(1);
+  logger2.plain(`${step} Creating a Vitest setup file for Storybook:`);
+  logger2.plain(`  ${vitestSetupFile}`);
+  const previewExists = EXTENSIONS.map((ext) => resolve(options.configDir, `preview${ext}`)).some(
+    existsSync
+  );
+  const imports = [`import { setProjectAnnotations } from '${annotationsImport}';`];
+  const projectAnnotations = [];
+  if (previewExists) {
+    imports.push(`import * as projectAnnotations from './preview';`);
+    projectAnnotations.push("projectAnnotations");
+  }
+  await writeFile(
+    vitestSetupFile,
+    dedent`
       ${imports.join("\n")}
 
       // This is an important step to apply the right configuration when testing your stories.
       // More info at: https://storybook.js.org/docs/api/portable-stories/portable-stories-vitest#setprojectannotations
       setProjectAnnotations([${projectAnnotations.join(", ")}]);
     `
-    );
-  }
+  );
   const vitestWorkspaceFile = findFile("vitest.workspace", [".ts", ".js", ".json"]);
   const viteConfigFile = findFile("vite.config");
   const vitestConfigFile = findFile("vitest.config");
@@ -1893,12 +2056,14 @@ async function postInstall(options) {
     const target = babelParse(workspaceFile);
     const updated = updateWorkspaceFile(source, target);
     if (updated) {
-      logger.step(`Updating your Vitest workspace file...`);
-      logger.log(`${vitestWorkspaceFile}`);
+      logger2.line(1);
+      logger2.plain(`${step} Updating your Vitest workspace file:`);
+      logger2.plain(`  ${vitestWorkspaceFile}`);
       const formattedContent = await formatFileContent(vitestWorkspaceFile, generate(target).code);
       await writeFile(vitestWorkspaceFile, formattedContent);
     } else {
-      logger.error(
+      logErrors(
+        "\u{1F6A8} Oh no!",
         dedent`
           Could not update existing Vitest workspace file:
           ${vitestWorkspaceFile}
@@ -1910,7 +2075,8 @@ async function postInstall(options) {
           https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#manual-setup
         `
       );
-      errors.push("Unable to update existing Vitest workspace file");
+      logger2.line(1);
+      return;
     }
   } else if (rootConfig) {
     let target, updated;
@@ -1929,101 +2095,148 @@ async function postInstall(options) {
       updated = updateConfigFile(source, target);
     }
     if (target && updated) {
-      logger.step(`Updating your ${vitestConfigFile ? "Vitest" : "Vite"} config file:`);
-      logger.log(`  ${rootConfig}`);
+      logger2.line(1);
+      logger2.plain(`${step} Updating your ${vitestConfigFile ? "Vitest" : "Vite"} config file:`);
+      logger2.plain(`  ${rootConfig}`);
       const formattedContent = await formatFileContent(rootConfig, generate(target).code);
-      const shouldAddReference = !configFileHasTypeReference && !vitestConfigFile;
       await writeFile(
         rootConfig,
-        shouldAddReference ? '/// <reference types="vitest/config" />\n' + formattedContent : formattedContent
+        configFileHasTypeReference ? formattedContent : '/// <reference types="vitest/config" />\n' + formattedContent
       );
     } else {
-      logger.error(dedent`
+      logErrors(
+        "\u{1F6A8} Oh no!",
+        dedent`
         We were unable to update your existing ${vitestConfigFile ? "Vitest" : "Vite"} config file.
 
         Please refer to the documentation to complete the setup manually:
         https://storybook.js.org/docs/writing-tests/integrations/vitest-addon#manual-setup
-      `);
-      errors.push("Unable to update existing Vitest config file");
+      `
+      );
     }
   } else {
-    const parentDir = dirname(options.configDir);
-    const newConfigFile = resolve(parentDir, `vitest.config.${fileExtension}`);
+    const newConfigFile = resolve(`vitest.config.${fileExtension}`);
     const configTemplate = await loadTemplate(getTemplateName(), {
       CONFIG_DIR: options.configDir,
       SETUP_FILE: relative(dirname(newConfigFile), vitestSetupFile)
     });
-    logger.step(`Creating a Vitest config file:`);
-    logger.log(`${newConfigFile}`);
+    logger2.line(1);
+    logger2.plain(`${step} Creating a Vitest config file:`);
+    logger2.plain(`  ${newConfigFile}`);
     const formattedContent = await formatFileContent(newConfigFile, configTemplate);
     await writeFile(newConfigFile, formattedContent);
   }
-  const a11yAddon = info.addons.find((addon) => addon.includes(addonA11yName));
+  const a11yAddon = info2.addons.find((addon) => addon.includes(addonA11yName));
   if (a11yAddon) {
     try {
-      const command = ["automigrate", "addon-a11y-addon-test"];
-      command.push("--loglevel", "silent");
-      command.push("--yes", "--skip-doctor");
-      if (options.packageManager) {
-        command.push("--package-manager", options.packageManager);
-      }
-      if (options.skipInstall) {
-        command.push("--skip-install");
-      }
-      if (options.configDir !== ".storybook") {
-        command.push("--config-dir", `"${options.configDir}"`);
-      }
-      const remoteCommand = packageManager.getRemoteRunCommand(
+      logger2.plain(`${step} Setting up ${addonA11yName} for @storybook/addon-vitest:`);
+      await execa(
         "storybook",
-        command,
-        versions.storybook
+        [
+          "automigrate",
+          "addon-a11y-addon-test",
+          "--loglevel",
+          "silent",
+          "--yes",
+          "--skip-doctor",
+          ...options.packageManager ? ["--package-manager", options.packageManager] : [],
+          ...options.skipInstall ? ["--skip-install"] : [],
+          ...options.configDir !== ".storybook" ? ["--config-dir", `"${options.configDir}"`] : []
+        ],
+        {
+          stdio: "inherit"
+        }
       );
-      const [cmd, ...args] = remoteCommand.split(" ");
-      await prompt.executeTask(() => packageManager.executeCommand({ command: cmd, args }), {
-        id: "a11y-addon-setup",
-        intro: "Setting up a11y addon for @storybook/addon-vitest",
-        error: "Failed to setup a11y addon for @storybook/addon-vitest",
-        success: "a11y addon setup successfully"
-      });
     } catch (e) {
-      console.log(e);
-      logger.line();
-      logger.error(dedent`
-        Could not automatically set up ${addonA11yName} for @storybook/addon-vitest.
+      logErrors(
+        "\u{1F6A8} Oh no!",
+        dedent`
+        We have detected that you have ${addonA11yName} installed but could not automatically set it up for @storybook/addon-vitest:
+
+        ${e instanceof Error ? e.message : String(e)}
+
         Please refer to the documentation to complete the setup manually:
         https://storybook.js.org/docs/writing-tests/accessibility-testing#test-addon-integration
-      `);
-      errors.push(
-        "The @storybook/addon-a11y couldn't be set up for the Vitest addon" + (e instanceof Error ? e.stack : String(e))
+      `
       );
     }
   }
   const runCommand = rootConfig ? `npx vitest --project=storybook` : `npx vitest`;
-  logger.line();
-  if (errors.length === 0) {
-    logger.step(CLI_COLORS.success("All done!"));
-    logger.log(dedent`
+  if (!hasErrors) {
+    printSuccess(
+      "\u{1F389} All done!",
+      dedent`
         @storybook/addon-vitest is now configured and you're ready to run your tests!
+  
         Here are a couple of tips to get you started:
-
-        • You can run tests with "${CLI_COLORS.cta(runCommand)}"
-        • Vitest IDE extension shows all stories as tests in your editor!
-
+        • You can run tests with "${runCommand}"
+        • When using the Vitest extension in your editor, all of your stories will be shown as tests!
+  
         Check the documentation for more information about its features and options at:
         https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}
-      `);
+      `
+    );
   } else {
-    logger.warn(
+    printWarning(
+      "\u26A0\uFE0F Done, but with errors!",
       dedent`
-        Done, but with errors!
-        @storybook/addon-vitest was installed successfully, but there were some errors during the setup process. Please refer to the documentation to complete the setup manually and check the errors above:
+        @storybook/addon-vitest was installed successfully, but there were some errors during the setup process.
+
+        Please refer to the documentation to complete the setup manually and check the errors above:
         https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#manual-setup
       `
     );
-    throw new AddonVitestPostinstallError({ errors });
   }
+  logger2.line(1);
 }
 __name(postInstall, "postInstall");
+async function getPackageNameFromPath(input) {
+  const path = input.startsWith("file://") ? fileURLToPath(input) : input;
+  if (!isAbsolute(path)) {
+    return path;
+  }
+  const packageJsonPath = up2({ cwd: path });
+  if (!packageJsonPath) {
+    throw new Error(`Could not find package.json in path: ${path}`);
+  }
+  const { default: packageJson } = await import(pathToFileURL(packageJsonPath).href, {
+    with: { type: "json" }
+  });
+  return packageJson.name;
+}
+__name(getPackageNameFromPath, "getPackageNameFromPath");
+async function getStorybookInfo({ configDir, packageManager: pkgMgr }) {
+  const packageManager = JsPackageManagerFactory.getPackageManager({ force: pkgMgr, configDir });
+  const { packageJson } = packageManager.primaryPackageJson;
+  const config = await loadMainConfig({ configDir });
+  const { presets } = await experimental_loadStorybook({
+    configDir,
+    packageJson
+  });
+  const framework = await presets.apply("framework", {});
+  const core = await presets.apply("core", {});
+  const { builder, renderer } = core;
+  if (!builder) {
+    throw new Error("Could not detect your Storybook builder.");
+  }
+  const frameworkPackageName = await getPackageNameFromPath(
+    typeof framework === "string" ? framework : framework.name
+  );
+  const builderPackageName = await getPackageNameFromPath(
+    typeof builder === "string" ? builder : builder.name
+  );
+  let rendererPackageName;
+  if (renderer) {
+    rendererPackageName = await getPackageNameFromPath(renderer);
+  }
+  return {
+    frameworkPackageName,
+    builderPackageName,
+    rendererPackageName,
+    addons: getAddonNames(config)
+  };
+}
+__name(getStorybookInfo, "getStorybookInfo");
 export {
   postInstall as default
 };