@storybook/addon-vitest 10.1.0-alpha.11 → 10.1.0-alpha.13

package/dist/postinstall.js

@@ -1,33 +1,32 @@
-import CJS_COMPAT_NODE_URL_zfiyucf8tb from 'node:url';
-import CJS_COMPAT_NODE_PATH_zfiyucf8tb from 'node:path';
-import CJS_COMPAT_NODE_MODULE_zfiyucf8tb from "node:module";
+import CJS_COMPAT_NODE_URL_4nt6o51r4ah from 'node:url';
+import CJS_COMPAT_NODE_PATH_4nt6o51r4ah from 'node:path';
+import CJS_COMPAT_NODE_MODULE_4nt6o51r4ah from "node:module";
 
-var __filename = CJS_COMPAT_NODE_URL_zfiyucf8tb.fileURLToPath(import.meta.url);
-var __dirname = CJS_COMPAT_NODE_PATH_zfiyucf8tb.dirname(__filename);
-var require = CJS_COMPAT_NODE_MODULE_zfiyucf8tb.createRequire(import.meta.url);
+var __filename = CJS_COMPAT_NODE_URL_4nt6o51r4ah.fileURLToPath(import.meta.url);
+var __dirname = CJS_COMPAT_NODE_PATH_4nt6o51r4ah.dirname(__filename);
+var require = CJS_COMPAT_NODE_MODULE_4nt6o51r4ah.createRequire(import.meta.url);
 
 // ------------------------------------------------------------
 // end of CJS compatibility banner, injected by Storybook's esbuild configuration
 // ------------------------------------------------------------
 import {
-  execa,
   resolvePackageDir
-} from "./_node-chunks/chunk-BEGZGAUW.js";
+} from "./_node-chunks/chunk-HKZ24UC3.js";
 import {
-  any,
-  getAddonNames,
-  up
-} from "./_node-chunks/chunk-FD2MOKYY.js";
+  any
+} from "./_node-chunks/chunk-25N73QA4.js";
 import {
-  DOCUMENTATION_LINK,
-  SUPPORTED_FRAMEWORKS
-} from "./_node-chunks/chunk-TCQNKRGF.js";
+  DOCUMENTATION_LINK
+} from "./_node-chunks/chunk-FJKQJ53U.js";
+import {
+  require_dist
+} from "./_node-chunks/chunk-IZMDFVYT.js";
 import {
   dirname,
   join,
   relative,
   resolve
-} from "./_node-chunks/chunk-L7WYCUSW.js";
+} from "./_node-chunks/chunk-F3U6JEPX.js";
 import {
   require_compare,
   require_constants,
@@ -37,11 +36,11 @@ import {
   require_parse_options,
   require_re,
   require_semver
-} from "./_node-chunks/chunk-CHRIX3Z2.js";
+} from "./_node-chunks/chunk-CWEKNW53.js";
 import {
   __commonJS,
   __toESM
-} from "./_node-chunks/chunk-FCFZB5YS.js";
+} from "./_node-chunks/chunk-6K4PB6ZM.js";
 
 // ../../node_modules/semver/functions/parse.js
 var require_parse = __commonJS({
@@ -295,7 +294,7 @@ var require_cmp = __commonJS({
 var require_coerce = __commonJS({
   "../../node_modules/semver/functions/coerce.js"(exports, module) {
     "use strict";
-    var SemVer = require_semver(), parse = require_parse(), { safeRe: re, t } = require_re(), coerce2 = (version, options) => {
+    var SemVer = require_semver(), parse = require_parse(), { safeRe: re, t } = require_re(), coerce = (version, options) => {
       if (version instanceof SemVer)
         return version;
       if (typeof version == "number" && (version = String(version)), typeof version != "string")
@@ -315,7 +314,7 @@ var require_coerce = __commonJS({
       let major = match[2], minor = match[3] || "0", patch = match[4] || "0", prerelease = options.includePrerelease && match[5] ? `-${match[5]}` : "", build = options.includePrerelease && match[6] ? `+${match[6]}` : "";
       return parse(`${major}.${minor}.${patch}${prerelease}${build}`, options);
     };
-    module.exports = coerce2;
+    module.exports = coerce;
   }
 });
 
@@ -820,7 +819,7 @@ var require_subset = __commonJS({
 var require_semver2 = __commonJS({
   "../../node_modules/semver/index.js"(exports, module) {
     "use strict";
-    var internalRe = require_re(), constants = require_constants(), SemVer = require_semver(), identifiers = require_identifiers(), parse = require_parse(), valid = require_valid(), clean = require_clean(), inc = require_inc(), diff = require_diff(), major = require_major(), minor = require_minor(), patch = require_patch(), prerelease = require_prerelease(), compare = require_compare(), rcompare = require_rcompare(), compareLoose = require_compare_loose(), compareBuild = require_compare_build(), sort = require_sort(), rsort = require_rsort(), gt = require_gt(), lt = require_lt(), eq = require_eq(), neq = require_neq(), gte = require_gte(), lte = require_lte(), cmp = require_cmp(), coerce2 = require_coerce(), Comparator = require_comparator(), Range = require_range(), satisfies2 = require_satisfies(), toComparators = require_to_comparators(), maxSatisfying = require_max_satisfying(), minSatisfying = require_min_satisfying(), minVersion = require_min_version(), validRange = require_valid2(), outside = require_outside(), gtr = require_gtr(), ltr = require_ltr(), intersects = require_intersects(), simplifyRange = require_simplify(), subset = require_subset();
+    var internalRe = require_re(), constants = require_constants(), SemVer = require_semver(), identifiers = require_identifiers(), parse = require_parse(), valid = require_valid(), clean = require_clean(), inc = require_inc(), diff = require_diff(), major = require_major(), minor = require_minor(), patch = require_patch(), prerelease = require_prerelease(), compare = require_compare(), rcompare = require_rcompare(), compareLoose = require_compare_loose(), compareBuild = require_compare_build(), sort = require_sort(), rsort = require_rsort(), gt = require_gt(), lt = require_lt(), eq = require_eq(), neq = require_neq(), gte = require_gte(), lte = require_lte(), cmp = require_cmp(), coerce = require_coerce(), Comparator = require_comparator(), Range = require_range(), satisfies2 = require_satisfies(), toComparators = require_to_comparators(), maxSatisfying = require_max_satisfying(), minSatisfying = require_min_satisfying(), minVersion = require_min_version(), validRange = require_valid2(), outside = require_outside(), gtr = require_gtr(), ltr = require_ltr(), intersects = require_intersects(), simplifyRange = require_simplify(), subset = require_subset();
     module.exports = {
       parse,
       valid,
@@ -844,7 +843,7 @@ var require_semver2 = __commonJS({
       gte,
       lte,
       cmp,
-      coerce: coerce2,
+      coerce,
       Comparator,
       Range,
       satisfies: satisfies2,
@@ -875,42 +874,21 @@ 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);
-}
-
-// 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", step = colors.gray("\u203A"), info = colors.blue(fancy ? "\u2139" : "i"), success = colors.green(fancy ? "\u2714" : "\u221A"), warning = colors.orange(fancy ? "\u26A0" : "\u203C"), error = colors.red(fancy ? "\u2716" : "\xD7"), baseOptions = {
-  borderStyle: "round",
-  padding: 1
-}, print = (message, options) => {
-  logger.line(1), logger.logBox(message, { ...baseOptions, ...options });
-}, printInfo = (title, message, options) => print(message, { borderColor: "blue", title, ...options }), printWarning = (title, message, options) => print(message, { borderColor: "yellow", title, ...options }), printError = (title, message, options) => print(message, { borderColor: "red", title, ...options }), printSuccess = (title, message, options) => print(message, { borderColor: "green", title, ...options });
+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_ts_dedent = __toESM(require_dist(), 1);
 
 // src/updateVitestFile.ts
 import * as fs from "node:fs/promises";
@@ -1060,167 +1038,91 @@ var loadTemplate = async (name, replacements) => {
 };
 
 // src/postinstall.ts
-var ADDON_NAME = "@storybook/addon-vitest", EXTENSIONS = [".ts", ".tsx", ".js", ".jsx", ".cts", ".mts", ".cjs", ".mjs"], addonA11yName = "@storybook/addon-a11y", hasErrors = !1;
-function nameMatches(name, pattern) {
-  return !!(name === pattern || name.includes(`${pattern}${sep}`) || name.includes(`${pattern}${posix.sep}`));
-}
-var logErrors = (...args) => {
-  hasErrors = !0, printError(...args);
-}, findFile = (basename, extensions = EXTENSIONS) => any(
-  extensions.map((ext) => basename + ext),
-  { last: getProjectRoot() }
-);
+var ADDON_NAME = "@storybook/addon-vitest", EXTENSIONS = [".ts", ".tsx", ".js", ".jsx", ".cts", ".mts", ".cjs", ".mjs"], addonA11yName = "@storybook/addon-a11y";
 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...
-    `
-  );
-  let packageManager = JsPackageManagerFactory.getPackageManager({
+  let errors = [], { logger, prompt } = options, packageManager = JsPackageManagerFactory.getPackageManager({
     force: options.packageManager
-  }), vitestVersionSpecifier = await packageManager.getInstalledVersion("vitest"), coercedVitestVersion = vitestVersionSpecifier ? (0, import_semver.coerce)(vitestVersionSpecifier) : null, isVitest3_2To4 = vitestVersionSpecifier ? (0, import_semver.satisfies)(vitestVersionSpecifier, ">=3.2.0 <4.0.0") : !1, isVitest4OrNewer = vitestVersionSpecifier ? (0, import_semver.satisfies)(vitestVersionSpecifier, ">=4.0.0") : !0, info2 = await getStorybookInfo(options), allDeps = packageManager.getAllDependencies(), uniqueDependencies = [
-    "vitest",
-    "playwright",
-    isVitest4OrNewer ? "@vitest/browser-playwright" : "@vitest/browser"
-  ].filter((p) => !allDeps[p]), mainJsPath = getInterpretedFile(resolve(options.configDir, "main")), config = await readConfig(mainJsPath), hasCustomWebpackConfig = !!config.getFieldNode(["webpackFinal"]), isInteractive = process.stdout.isTTY && !isCI2();
-  nameMatches(info2.frameworkPackageName, "@storybook/nextjs") && !hasCustomWebpackConfig && (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
+  }), findFile = (basename, extensions = EXTENSIONS) => any(
+    extensions.map((ext) => basename + ext),
+    { last: getProjectRoot(), cwd: options.configDir }
+  ), vitestVersionSpecifier = await packageManager.getInstalledVersion("vitest");
+  logger.debug(`Vitest version specifier: ${vitestVersionSpecifier}`);
+  let isVitest3_2To4 = vitestVersionSpecifier ? (0, import_semver.satisfies)(vitestVersionSpecifier, ">=3.2.0 <4.0.0") : !1, isVitest4OrNewer = vitestVersionSpecifier ? (0, import_semver.satisfies)(vitestVersionSpecifier, ">=4.0.0") : !0, info = await getStorybookInfo(options.configDir), allDeps = packageManager.getAllDependencies(), addonVitestService = new AddonVitestService(), compatibilityResult = await addonVitestService.validateCompatibility({
+    packageManager,
+    framework: info.framework,
+    builder: info.builder
+  }), result = null;
+  if (!compatibilityResult.compatible && compatibilityResult.reasons) {
+    let reasons = compatibilityResult.reasons.map((r) => `\u2022 ${CLI_COLORS.error(r)}`);
+    reasons.unshift(import_ts_dedent.dedent`
+      Automated setup failed
+      The following packages have incompatibilities that prevent automated setup:
+    `), reasons.push(
+      import_ts_dedent.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: !0
-  })).migrateToNextjsVite && (await packageManager.addDependencies({ type: "devDependencies", skipInstall: !0 }, [
-    "@storybook/nextjs-vite"
-  ]), await packageManager.removeDependencies(["@storybook/nextjs"]), traverse(config._ast, {
-    StringLiteral(path) {
-      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: !1,
-    transformFn: (files, options2, dryRun) => transformImportFiles(files, options2, dryRun),
-    transformOptions: {
-      "@storybook/nextjs": "@storybook/nextjs-vite"
-    }
-  }));
-  let annotationsImport = SUPPORTED_FRAMEWORKS.find(
-    (f) => nameMatches(info2.frameworkPackageName, f)
-  ) ? info2.frameworkPackageName === "@storybook/nextjs" ? "@storybook/nextjs-vite" : info2.frameworkPackageName : null, isRendererSupported = !!annotationsImport, result = await (async () => {
-    let reasons = [];
-    hasCustomWebpackConfig && reasons.push("\u2022 The addon can not be used with a custom Webpack configuration."), !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."
-    ), isRendererSupported || reasons.push(dedent`
-        • The addon cannot yet be used with ${info2.frameworkPackageName}
-      `), 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.
-      `);
-    let mswVersionSpecifier = await packageManager.getInstalledVersion("msw"), coercedMswVersion = mswVersionSpecifier ? (0, import_semver.coerce)(mswVersionSpecifier) : null;
-    return 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.
-      `), nameMatches(info2.frameworkPackageName, "@storybook/nextjs") && (await packageManager.getInstalledVersion("next") || 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.
-        `)), reasons.length > 0 ? (reasons.unshift(
-      "@storybook/addon-vitest's automated setup failed due to the following package incompatibilities:"
-    ), reasons.push("--------------------------------"), 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.
-        `
-    ), isRendererSupported ? 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
-          `
-    ) : reasons.push(
-      dedent`
-            Please check the documentation for more information about its requirements and installation:
-            https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}
-          `
-    ), reasons.map((r) => r.trim()).join(`
+        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(`
 
-`)) : null;
-  })();
-  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.
+  if (result)
+    throw logger.error(result), new AddonVitestPostinstallPrerequisiteCheckError({
+      reasons: compatibilityResult.reasons
+    });
+  if (!options.skipDependencyManagement) {
+    let versionedDependencies = await addonVitestService.collectDependencies(packageManager);
+    info.framework === SupportedFramework.NEXTJS && packageManager.getAllDependencies()["@storybook/nextjs-vite"];
+    let v8Version = await packageManager.getInstalledVersion("@vitest/coverage-v8"), istanbulVersion = await packageManager.getInstalledVersion("@vitest/coverage-istanbul");
+    !v8Version && !istanbulVersion && logger.step(
+      import_ts_dedent.dedent`
+          You don't seem to have a coverage reporter installed. Vitest needs either V8 or Istanbul to generate coverage reports.
 
-        More info about the plugin at https://github.com/storybookjs/vite-plugin-storybook-nextjs
-      `
-    );
-    try {
-      let storybookVersion = await packageManager.getInstalledVersion("storybook");
-      uniqueDependencies.push(`@storybook/nextjs-vite@^${storybookVersion}`);
-    } catch {
-      console.error("Failed to install @storybook/nextjs-vite. Please install it manually");
-    }
+          Adding "@vitest/coverage-v8" to enable coverage reporting.
+          Read more about Vitest coverage providers at https://vitest.dev/guide/coverage.html#coverage-providers
+        `
+    ), versionedDependencies.length > 0 && (logger.step("Adding dependencies to your package.json"), logger.log("  " + versionedDependencies.join(", ")), await packageManager.addDependencies(
+      { type: "devDependencies", skipInstall: !0 },
+      versionedDependencies
+    )), options.skipInstall || await packageManager.installDependencies();
   }
-  let v8Version = await packageManager.getInstalledVersion("@vitest/coverage-v8"), istanbulVersion = await packageManager.getInstalledVersion("@vitest/coverage-istanbul");
-  !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"));
-  let versionedDependencies = uniqueDependencies.map((p) => p.includes("vitest") && vitestVersionSpecifier ? `${p}@${vitestVersionSpecifier}` : p);
-  if (versionedDependencies.length > 0 && (await packageManager.addDependencies(
-    { type: "devDependencies", skipInstall: !0 },
-    versionedDependencies
-  ), logger2.line(1), logger2.plain(`${step} Installing dependencies:`), logger2.plain("  " + versionedDependencies.join(", "))), await packageManager.installDependencies(), logger2.line(1), 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"]
-      });
-    } catch {
-      console.error("Failed to install Playwright. Please install it manually");
+  if (!options.skipDependencyManagement)
+    if (options.skipInstall)
+      logger.warn(import_ts_dedent.dedent`
+        Playwright browser binaries installation skipped. Please run the following command manually later:
+        ${CLI_COLORS.cta("npx playwright install chromium --with-deps")}
+      `);
+    else {
+      let { errors: playwrightErrors } = await addonVitestService.installPlaywright(
+        packageManager,
+        {
+          yes: options.yes
+        }
+      );
+      errors.push(...playwrightErrors);
     }
-  }
   let fileExtension = allDeps.typescript || findFile("tsconfig", [...EXTENSIONS, ".json"]) ? "ts" : "js", 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
-      `
-    ), logger2.line(1);
-    return;
-  }
-  logger2.line(1), logger2.plain(`${step} Creating a Vitest setup file for Storybook:`), logger2.plain(`  ${vitestSetupFile}`);
-  let previewExists = EXTENSIONS.map((ext) => resolve(options.configDir, `preview${ext}`)).some(
-    existsSync
-  ), imports = [`import { setProjectAnnotations } from '${annotationsImport}';`], projectAnnotations = [];
-  previewExists && (imports.push("import * as projectAnnotations from './preview';"), projectAnnotations.push("projectAnnotations")), await writeFile(
-    vitestSetupFile,
-    dedent`
+    let errorMessage = import_ts_dedent.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}
+`);
+    let previewExists = EXTENSIONS.map((ext) => resolve(options.configDir, `preview${ext}`)).some(
+      existsSync
+    ), imports = [`import { setProjectAnnotations } from '${info.frameworkPackage}';`], projectAnnotations = [];
+    previewExists && (imports.push("import * as projectAnnotations from './preview';"), projectAnnotations.push("projectAnnotations")), await writeFile(
+      vitestSetupFile,
+      import_ts_dedent.dedent`
       ${imports.join(`
 `)}
 
@@ -1228,7 +1130,8 @@ async function postInstall(options) {
       // More info at: https://storybook.js.org/docs/api/portable-stories/portable-stories-vitest#setprojectannotations
       setProjectAnnotations([${projectAnnotations.join(", ")}]);
     `
-  );
+    );
+  }
   let vitestWorkspaceFile = findFile("vitest.workspace", [".ts", ".js", ".json"]), viteConfigFile = findFile("vite.config"), vitestConfigFile = findFile("vitest.config"), vitestShimFile = findFile("vitest.shims.d"), rootConfig = vitestConfigFile || viteConfigFile;
   fileExtension === "ts" && !vitestShimFile && await writeFile(
     "vitest.shims.d.ts",
@@ -1243,13 +1146,12 @@ async function postInstall(options) {
     }).then((t) => t.replace(`
   'ROOT_CONFIG',`, "").replace(/\s+extends: '',/, "")), workspaceFile = await fs2.readFile(vitestWorkspaceFile, "utf8"), source = babelParse(workspaceTemplate), target = babelParse(workspaceFile);
     if (updateWorkspaceFile(source, target)) {
-      logger2.line(1), logger2.plain(`${step} Updating your Vitest workspace file:`), logger2.plain(`  ${vitestWorkspaceFile}`);
+      logger.step("Updating your Vitest workspace file..."), logger.log(`${vitestWorkspaceFile}`);
       let formattedContent = await formatFileContent(vitestWorkspaceFile, generate(target).code);
       await writeFile(vitestWorkspaceFile, formattedContent);
-    } else {
-      logErrors(
-        "\u{1F6A8} Oh no!",
-        dedent`
+    } else
+      logger.error(
+        import_ts_dedent.dedent`
           Could not update existing Vitest workspace file:
           ${vitestWorkspaceFile}
 
@@ -1259,9 +1161,7 @@ async function postInstall(options) {
           Please refer to the documentation to complete the setup manually:
           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, configFile = await fs2.readFile(rootConfig, "utf8"), configFileHasTypeReference = configFile.match(
       /\/\/\/\s*<reference\s+types=["']vitest\/config["']\s*\/>/
@@ -1274,7 +1174,7 @@ async function postInstall(options) {
       target = babelParse(configFile), 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}`);
       let formattedContent = await formatFileContent(rootConfig, generate(target).code);
       await writeFile(
         rootConfig,
@@ -1282,108 +1182,70 @@ async function postInstall(options) {
 ` + formattedContent : formattedContent
       );
     } else
-      logErrors(
-        "\u{1F6A8} Oh no!",
-        dedent`
+      logger.error(import_ts_dedent.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 {
-    let newConfigFile = resolve(`vitest.config.${fileExtension}`), configTemplate = await loadTemplate(getTemplateName(), {
+    let parentDir = dirname(options.configDir), newConfigFile = resolve(parentDir, `vitest.config.${fileExtension}`), configTemplate = await loadTemplate(getTemplateName(), {
       CONFIG_DIR: options.configDir,
       SETUP_FILE: relative(dirname(newConfigFile), vitestSetupFile)
     });
-    logger2.line(1), logger2.plain(`${step} Creating a Vitest config file:`), logger2.plain(`  ${newConfigFile}`);
+    logger.step("Creating a Vitest config file:"), logger.log(`${newConfigFile}`);
     let formattedContent = await formatFileContent(newConfigFile, configTemplate);
     await writeFile(newConfigFile, formattedContent);
   }
-  if (info2.addons.find((addon) => addon.includes(addonA11yName)))
+  if (info.addons.find((addon) => addon.includes(addonA11yName)))
     try {
-      logger2.plain(`${step} Setting up ${addonA11yName} for @storybook/addon-vitest:`), await execa(
+      let command = [
         "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}"`] : []
-        ],
+        "automigrate",
+        "addon-a11y-addon-test",
+        "--loglevel",
+        "silent",
+        "--yes",
+        "--skip-doctor"
+      ];
+      options.packageManager && command.push("--package-manager", options.packageManager), options.skipInstall && command.push("--skip-install"), options.configDir !== ".storybook" && command.push("--config-dir", options.configDir), await prompt.executeTask(
+        // TODO: Remove stdio: 'ignore' once we have a way to log the output of the command properly
+        () => packageManager.runPackageCommand({ args: command, stdio: "ignore" }),
         {
-          stdio: "inherit"
+          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(import_ts_dedent.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))
       );
     }
-  hasErrors ? printWarning(
-    "\u26A0\uFE0F Done, but with errors!",
-    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:
-        https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#manual-setup
-      `
-  ) : printSuccess(
-    "\u{1F389} All done!",
-    dedent`
+  let runCommand = rootConfig ? "npx vitest --project=storybook" : "npx vitest";
+  if (errors.length === 0)
+    logger.step(CLI_COLORS.success("@storybook/addon-vitest setup completed successfully")), logger.log(import_ts_dedent.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 "${rootConfig ? "npx vitest --project=storybook" : "npx vitest"}"
-        • 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
+    throw logger.warn(
+      import_ts_dedent.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:
+        https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#manual-setup
       `
-  ), logger2.line(1);
-}
-async function getPackageNameFromPath(input) {
-  let path = input.startsWith("file://") ? fileURLToPath(input) : input;
-  if (!isAbsolute(path))
-    return path;
-  let packageJsonPath = up2({ cwd: path });
-  if (!packageJsonPath)
-    throw new Error(`Could not find package.json in path: ${path}`);
-  let { default: packageJson } = await import(pathToFileURL(packageJsonPath).href, {
-    with: { type: "json" }
-  });
-  return packageJson.name;
-}
-async function getStorybookInfo({ configDir, packageManager: pkgMgr }) {
-  let packageManager = JsPackageManagerFactory.getPackageManager({ force: pkgMgr, configDir }), { packageJson } = packageManager.primaryPackageJson, config = await loadMainConfig({ configDir }), { presets } = await experimental_loadStorybook({
-    configDir,
-    packageJson
-  }), framework = await presets.apply("framework", {}), core = await presets.apply("core", {}), { builder, renderer } = core;
-  if (!builder)
-    throw new Error("Could not detect your Storybook builder.");
-  let frameworkPackageName = await getPackageNameFromPath(
-    typeof framework == "string" ? framework : framework.name
-  ), builderPackageName = await getPackageNameFromPath(
-    typeof builder == "string" ? builder : builder.name
-  ), rendererPackageName;
-  return renderer && (rendererPackageName = await getPackageNameFromPath(renderer)), {
-    frameworkPackageName,
-    builderPackageName,
-    rendererPackageName,
-    addons: getAddonNames(config)
-  };
+    ), new AddonVitestPostinstallError({ errors });
 }
 export {
   postInstall as default