@storybook/addon-vitest 0.0.0-pr-32799-sha-e8c4dd98 → 0.0.0-pr-32921-sha-16b7aab0

package/dist/postinstall.js

@@ -1,33 +1,29 @@
-import CJS_COMPAT_NODE_URL_pdgzsigohck from 'node:url';
-import CJS_COMPAT_NODE_PATH_pdgzsigohck from 'node:path';
-import CJS_COMPAT_NODE_MODULE_pdgzsigohck from "node:module";
+import CJS_COMPAT_NODE_URL_6px7swmj8pe from 'node:url';
+import CJS_COMPAT_NODE_PATH_6px7swmj8pe from 'node:path';
+import CJS_COMPAT_NODE_MODULE_6px7swmj8pe from "node:module";
 
-var __filename = CJS_COMPAT_NODE_URL_pdgzsigohck.fileURLToPath(import.meta.url);
-var __dirname = CJS_COMPAT_NODE_PATH_pdgzsigohck.dirname(__filename);
-var require = CJS_COMPAT_NODE_MODULE_pdgzsigohck.createRequire(import.meta.url);
+var __filename = CJS_COMPAT_NODE_URL_6px7swmj8pe.fileURLToPath(import.meta.url);
+var __dirname = CJS_COMPAT_NODE_PATH_6px7swmj8pe.dirname(__filename);
+var require = CJS_COMPAT_NODE_MODULE_6px7swmj8pe.createRequire(import.meta.url);
 
 // ------------------------------------------------------------
 // end of CJS compatibility banner, injected by Storybook's esbuild configuration
 // ------------------------------------------------------------
 import {
-  execa,
   resolvePackageDir
-} from "./_node-chunks/chunk-APU7XC4Q.js";
+} from "./_node-chunks/chunk-IVUROXNL.js";
 import {
-  any,
-  getAddonNames,
-  up
-} from "./_node-chunks/chunk-GW4MQP76.js";
+  any
+} from "./_node-chunks/chunk-ZJEXRKEJ.js";
 import {
-  DOCUMENTATION_LINK,
-  SUPPORTED_FRAMEWORKS
-} from "./_node-chunks/chunk-CEYAHLH7.js";
+  DOCUMENTATION_LINK
+} from "./_node-chunks/chunk-ITGNSETM.js";
 import {
   dirname,
   join,
   relative,
   resolve
-} from "./_node-chunks/chunk-PJS3NMXR.js";
+} from "./_node-chunks/chunk-YT4V4DMB.js";
 import {
   require_compare,
   require_constants,
@@ -37,12 +33,12 @@ import {
   require_parse_options,
   require_re,
   require_semver
-} from "./_node-chunks/chunk-4KCNAHUA.js";
+} from "./_node-chunks/chunk-WPTT6UNN.js";
 import {
   __commonJS,
   __name,
   __toESM
-} from "./_node-chunks/chunk-YW2JS56O.js";
+} from "./_node-chunks/chunk-XJAFV5RM.js";
 
 // ../../node_modules/semver/functions/parse.js
 var require_parse = __commonJS({
@@ -364,7 +360,7 @@ var require_coerce = __commonJS({
     var SemVer = require_semver();
     var parse = require_parse();
     var { safeRe: re, t } = require_re();
-    var coerce2 = /* @__PURE__ */ __name((version, options) => {
+    var coerce = /* @__PURE__ */ __name((version, options) => {
       if (version instanceof SemVer) {
         return version;
       }
@@ -399,7 +395,7 @@ var require_coerce = __commonJS({
       const build = options.includePrerelease && match[6] ? `+${match[6]}` : "";
       return parse(`${major}.${minor}.${patch}${prerelease}${build}`, options);
     }, "coerce");
-    module.exports = coerce2;
+    module.exports = coerce;
   }
 });
 
@@ -1444,7 +1440,7 @@ var require_semver2 = __commonJS({
     var gte = require_gte();
     var lte = require_lte();
     var cmp = require_cmp();
-    var coerce2 = require_coerce();
+    var coerce = require_coerce();
     var Comparator = require_comparator();
     var Range = require_range();
     var satisfies2 = require_satisfies();
@@ -1482,7 +1478,7 @@ var require_semver2 = __commonJS({
       gte,
       lte,
       cmp,
-      coerce: coerce2,
+      coerce,
       Comparator,
       Range,
       satisfies: satisfies2,
@@ -1513,56 +1509,23 @@ var require_semver2 = __commonJS({
 import { existsSync } from "node:fs";
 import * as fs2 from "node:fs/promises";
 import { writeFile } from "node:fs/promises";
-import { isAbsolute, posix, sep } from "node:path";
-import { fileURLToPath, pathToFileURL } from "node:url";
-import { babelParse, generate, traverse } from "storybook/internal/babel";
+import { babelParse, generate } from "storybook/internal/babel";
+import { AddonVitestService } from "storybook/internal/cli";
 import {
   JsPackageManagerFactory,
   formatFileContent,
-  getInterpretedFile,
   getProjectRoot,
-  isCI as isCI2,
-  loadMainConfig,
-  scanAndTransformFiles,
-  transformImportFiles
+  getStorybookInfo
 } from "storybook/internal/common";
-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
+import { CLI_COLORS } from "storybook/internal/node-logger";
+import {
+  AddonVitestPostinstallError,
+  AddonVitestPostinstallPrerequisiteCheckError
+} from "storybook/internal/server-errors";
+import { SupportedFramework } from "storybook/internal/types";
 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) => {
@@ -1616,10 +1579,7 @@ 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) {
-    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;
+    canHandleConfig = true;
   }
   if (!canHandleConfig) {
     return false;
@@ -1652,20 +1612,24 @@ 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 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(
+          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(
               (p) => p.type === "ObjectProperty" && p.key.type === "Identifier" && p.key.name === "test"
             )
           );
-          const defineConfigNode = defineConfigNodeWithTest || defineConfigNodes[0];
-          if (!defineConfigNode) {
+          const targetConfigObject = configObjectWithTest || configObjectNodes[0];
+          if (!targetConfigObject) {
             return false;
           }
-          const defineConfigProps = defineConfigNode.arguments[0];
-          const existingTestProp = defineConfigProps.properties.find(
+          const existingTestProp = targetConfigObject.properties.find(
             (p) => p.type === "ObjectProperty" && p.key.type === "Identifier" && p.key.name === "test"
           );
           if (existingTestProp && existingTestProp.value.type === "ObjectExpression") {
@@ -1677,6 +1641,16 @@ 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: [
@@ -1690,25 +1664,28 @@ var updateConfigFile = /* @__PURE__ */ __name((source, target) => {
                     {
                       type: "ObjectProperty",
                       key: { type: "Identifier", name: "test" },
-                      value: existingTestProp.value,
+                      value: testConfigForProject,
                       computed: false,
                       shorthand: false
                     }
                   ]
                 };
                 workspaceOrProjectsProp.value.elements.unshift(existingTestProject);
-                defineConfigProps.properties = defineConfigProps.properties.filter(
+                targetConfigObject.properties = targetConfigObject.properties.filter(
                   (p) => p !== existingTestProp
                 );
-                mergeProperties(properties, defineConfigProps.properties);
+                if (coverageProp && templateTestProp.value.type === "ObjectExpression") {
+                  templateTestProp.value.properties.unshift(coverageProp);
+                }
+                mergeProperties(properties, targetConfigObject.properties);
               } else {
-                mergeProperties(properties, defineConfigProps.properties);
+                mergeProperties(properties, targetConfigObject.properties);
               }
             } else {
-              mergeProperties(properties, defineConfigProps.properties);
+              mergeProperties(properties, targetConfigObject.properties);
             }
           } else {
-            mergeProperties(properties, defineConfigProps.properties);
+            mergeProperties(properties, targetConfigObject.properties);
           }
           updated = true;
         }
@@ -1757,290 +1734,156 @@ 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) {
-  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 errors = [];
+  const { logger, prompt } = options;
   const packageManager = JsPackageManagerFactory.getPackageManager({
     force: options.packageManager
   });
-  const info2 = await getStorybookInfo(options);
-  const allDeps = packageManager.getAllDependencies();
-  const dependencies = ["vitest", "@vitest/browser", "playwright"].filter((p) => !allDeps[p]);
+  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_2OrNewer = vitestVersionSpecifier ? (0, import_semver.satisfies)(vitestVersionSpecifier, ">=3.2.0") : true;
-  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
+  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 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.
 
-            Do you want to migrate?
-          `,
-      initial: true
+        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
     });
-    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"
-        }
-      });
-    }
   }
-  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.
-        `);
+  if (!options.skipDependencyManagement) {
+    const versionedDependencies = await addonVitestService.collectDependencies(packageManager);
+    if (info.framework === SupportedFramework.NEXTJS) {
+      const allDeps2 = packageManager.getAllDependencies();
+      if (!allDeps2["@storybook/nextjs-vite"]) {
       }
     }
-    if (reasons.length > 0) {
-      reasons.unshift(
-        `@storybook/addon-vitest's automated setup failed due to the following package incompatibilities:`
-      );
-      reasons.push("--------------------------------");
-      reasons.push(
+    const v8Version = await packageManager.getInstalledVersion("@vitest/coverage-v8");
+    const istanbulVersion = await packageManager.getInstalledVersion("@vitest/coverage-istanbul");
+    if (!v8Version && !istanbulVersion) {
+      logger.step(
         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.
+          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
         `
       );
-      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");
     }
-    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");
-      dependencies.push(`@storybook/nextjs-vite@^${storybookVersion}`);
-    } catch (e) {
-      console.error("Failed to install @storybook/nextjs-vite. Please install it manually");
+    if (versionedDependencies.length > 0) {
+      logger.step("Adding dependencies to your package.json");
+      logger.log("  " + versionedDependencies.join(", "));
+      await packageManager.addDependencies(
+        { type: "devDependencies", skipInstall: true },
+        versionedDependencies
+      );
     }
-  }
-  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
-      `
-    );
-    dependencies.push(`@vitest/coverage-v8`);
-  }
-  const versionedDependencies = dependencies.map((p) => {
-    if (p.includes("vitest")) {
-      return vitestVersionSpecifier ? `${p}@${vitestVersionSpecifier}` : p;
+    if (!options.skipInstall) {
+      await packageManager.installDependencies();
     }
-    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(", "));
   }
-  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"]
+  if (!options.skipDependencyManagement) {
+    if (!options.skipInstall) {
+      const playwrightErrors = await addonVitestService.installPlaywright(packageManager, {
+        yes: options.yes
       });
-    } catch (e) {
-      console.error("Failed to install Playwright. Please install it manually");
+      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")}
+      `);
     }
   }
   const fileExtension = allDeps.typescript || findFile("tsconfig", [...EXTENSIONS, ".json"]) ? "ts" : "js";
   const vitestSetupFile = resolve(options.configDir, `vitest.setup.${fileExtension}`);
   if (existsSync(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
-      `
+    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
     );
-    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`
+    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,
+      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"]) || findFile("vitest.projects", [".ts", ".js", ".json"]);
+    );
+  }
+  const vitestWorkspaceFile = findFile("vitest.workspace", [".ts", ".js", ".json"]);
   const viteConfigFile = findFile("vite.config");
   const vitestConfigFile = findFile("vitest.config");
   const vitestShimFile = findFile("vitest.shims.d");
   const rootConfig = vitestConfigFile || viteConfigFile;
-  const browserConfig = `{
-        enabled: true,
-        headless: true,
-        provider: 'playwright',
-        instances: [{ browser: 'chromium' }]
-      }`;
   if (fileExtension === "ts" && !vitestShimFile) {
     await writeFile(
       "vitest.shims.d.ts",
-      '/// <reference types="@vitest/browser/providers/playwright" />'
+      isVitest4OrNewer ? '/// <reference types="@vitest/browser-playwright" />' : '/// <reference types="@vitest/browser/providers/playwright" />'
     );
   }
+  const getTemplateName = /* @__PURE__ */ __name(() => {
+    if (isVitest4OrNewer) {
+      return "vitest.config.4.template.ts";
+    } else if (isVitest3_2To4) {
+      return "vitest.config.3.2.template.ts";
+    }
+    return "vitest.config.template.ts";
+  }, "getTemplateName");
   if (vitestWorkspaceFile) {
     const workspaceTemplate = await loadTemplate("vitest.workspace.template.ts", {
       EXTENDS_WORKSPACE: viteConfigFile ? relative(dirname(vitestWorkspaceFile), viteConfigFile) : "",
       CONFIG_DIR: options.configDir,
-      BROWSER_CONFIG: browserConfig,
       SETUP_FILE: relative(dirname(vitestWorkspaceFile), vitestSetupFile)
     }).then((t) => t.replace(`
   'ROOT_CONFIG',`, "").replace(/\s+extends: '',/, ""));
@@ -2049,14 +1892,12 @@ async function postInstall(options) {
     const target = babelParse(workspaceFile);
     const updated = updateWorkspaceFile(source, target);
     if (updated) {
-      logger2.line(1);
-      logger2.plain(`${step} Updating your Vitest workspace file:`);
-      logger2.plain(`  ${vitestWorkspaceFile}`);
+      logger.step(`Updating your Vitest workspace file...`);
+      logger.log(`${vitestWorkspaceFile}`);
       const formattedContent = await formatFileContent(vitestWorkspaceFile, generate(target).code);
       await writeFile(vitestWorkspaceFile, formattedContent);
     } else {
-      logErrors(
-        "\u{1F6A8} Oh no!",
+      logger.error(
         dedent`
           Could not update existing Vitest workspace file:
           ${vitestWorkspaceFile}
@@ -2068,21 +1909,18 @@ async function postInstall(options) {
           https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#manual-setup
         `
       );
-      logger2.line(1);
-      return;
+      errors.push("Unable to update existing Vitest workspace file");
     }
   } else if (rootConfig) {
     let target, updated;
     const configFile = await fs2.readFile(rootConfig, "utf8");
-    const hasProjectsConfig = configFile.includes("projects:");
     const configFileHasTypeReference = configFile.match(
       /\/\/\/\s*<reference\s+types=["']vitest\/config["']\s*\/>/
     );
-    const templateName = hasProjectsConfig || isVitest3_2OrNewer ? "vitest.config.3.2.template.ts" : "vitest.config.template.ts";
+    const templateName = getTemplateName();
     if (templateName) {
       const configTemplate = await loadTemplate(templateName, {
         CONFIG_DIR: options.configDir,
-        BROWSER_CONFIG: browserConfig,
         SETUP_FILE: relative(dirname(rootConfig), vitestSetupFile)
       });
       const source = babelParse(configTemplate);
@@ -2090,45 +1928,38 @@ async function postInstall(options) {
       updated = updateConfigFile(source, target);
     }
     if (target && updated) {
-      logger2.line(1);
-      logger2.plain(`${step} Updating your ${vitestConfigFile ? "Vitest" : "Vite"} config file:`);
-      logger2.plain(`  ${rootConfig}`);
+      logger.step(`Updating your ${vitestConfigFile ? "Vitest" : "Vite"} config file:`);
+      logger.log(`  ${rootConfig}`);
       const formattedContent = await formatFileContent(rootConfig, generate(target).code);
+      const shouldAddReference = !configFileHasTypeReference && !vitestConfigFile;
       await writeFile(
         rootConfig,
-        configFileHasTypeReference ? formattedContent : '/// <reference types="vitest/config" />\n' + formattedContent
+        shouldAddReference ? '/// <reference types="vitest/config" />\n' + formattedContent : formattedContent
       );
     } else {
-      logErrors(
-        "\u{1F6A8} Oh no!",
-        dedent`
+      logger.error(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 newConfigFile = resolve(`vitest.config.${fileExtension}`);
-    const configTemplate = await loadTemplate(
-      isVitest3_2OrNewer ? "vitest.config.3.2.template.ts" : "vitest.config.template.ts",
-      {
-        CONFIG_DIR: options.configDir,
-        BROWSER_CONFIG: browserConfig,
-        SETUP_FILE: relative(dirname(newConfigFile), vitestSetupFile)
-      }
-    );
-    logger2.line(1);
-    logger2.plain(`${step} Creating a Vitest config file:`);
-    logger2.plain(`  ${newConfigFile}`);
+    const parentDir = dirname(options.configDir);
+    const newConfigFile = resolve(parentDir, `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}`);
     const formattedContent = await formatFileContent(newConfigFile, configTemplate);
     await writeFile(newConfigFile, formattedContent);
   }
-  const a11yAddon = info2.addons.find((addon) => addon.includes(addonA11yName));
+  const a11yAddon = info.addons.find((addon) => addon.includes(addonA11yName));
   if (a11yAddon) {
     try {
-      logger2.plain(`${step} Setting up ${addonA11yName} for @storybook/addon-vitest:`);
       const command = ["automigrate", "addon-a11y-addon-test"];
       command.push("--loglevel", "silent");
       command.push("--yes", "--skip-doctor");
@@ -2141,99 +1972,53 @@ async function postInstall(options) {
       if (options.configDir !== ".storybook") {
         command.push("--config-dir", `"${options.configDir}"`);
       }
-      await execa("storybook", command, {
-        stdio: "inherit"
-      });
+      const remoteCommand = packageManager.getRemoteRunCommand("storybook", command);
+      const [cmd, ...args] = remoteCommand.split(" ");
+      await prompt.executeTask(
+        // TODO: Remove stdio: 'ignore' once we have a way to log the output of the command properly
+        () => packageManager.executeCommand({ command: cmd, args, stdio: "ignore" }),
+        {
+          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) {
-      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)}
-
+      logger.error(dedent`
+        Could not automatically set up ${addonA11yName} for @storybook/addon-vitest.
         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`;
-  if (!hasErrors) {
-    printSuccess(
-      "\u{1F389} All done!",
-      dedent`
+  if (errors.length === 0) {
+    logger.step(CLI_COLORS.success("@storybook/addon-vitest setup completed successfully"));
+    logger.log(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 "${runCommand}"
-        • When using the Vitest extension in your editor, all of your stories will be shown as tests!
-  
+
+        • You can run tests with "${CLI_COLORS.cta(runCommand)}"
+        • Vitest IDE extension shows all stories as tests in your editor!
+
         Check the documentation for more information about its features and options at:
         https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}
-      `
-    );
+      `);
   } else {
-    printWarning(
-      "\u26A0\uFE0F Done, but with errors!",
+    logger.warn(
       dedent`
-        @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:
+        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:
         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
 };