@storybook/addon-vitest 0.0.0-pr-32810-sha-b3901628 → 0.0.0-pr-32717-sha-6b32df73

package/dist/postinstall.js

@@ -1,33 +1,29 @@
-import CJS_COMPAT_NODE_URL_w2xsksad47 from 'node:url';
-import CJS_COMPAT_NODE_PATH_w2xsksad47 from 'node:path';
-import CJS_COMPAT_NODE_MODULE_w2xsksad47 from "node:module";
+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";
 
-var __filename = CJS_COMPAT_NODE_URL_w2xsksad47.fileURLToPath(import.meta.url);
-var __dirname = CJS_COMPAT_NODE_PATH_w2xsksad47.dirname(__filename);
-var require = CJS_COMPAT_NODE_MODULE_w2xsksad47.createRequire(import.meta.url);
+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);
 
 // ------------------------------------------------------------
 // end of CJS compatibility banner, injected by Storybook's esbuild configuration
 // ------------------------------------------------------------
 import {
-  execa,
   resolvePackageDir
-} from "./_node-chunks/chunk-HLYWN43O.js";
+} from "./_node-chunks/chunk-C3FRTXFF.js";
 import {
-  any,
-  getAddonNames,
-  up
-} from "./_node-chunks/chunk-H6JZJZLI.js";
+  any
+} from "./_node-chunks/chunk-NC7DRZM4.js";
 import {
-  DOCUMENTATION_LINK,
-  SUPPORTED_FRAMEWORKS
-} from "./_node-chunks/chunk-6MS5DPHO.js";
+  DOCUMENTATION_LINK
+} from "./_node-chunks/chunk-KIUHNMC3.js";
 import {
   dirname,
   join,
   relative,
   resolve
-} from "./_node-chunks/chunk-LZCJ7HPK.js";
+} from "./_node-chunks/chunk-7U3E5OTZ.js";
 import {
   require_compare,
   require_constants,
@@ -37,12 +33,12 @@ import {
   require_parse_options,
   require_re,
   require_semver
-} from "./_node-chunks/chunk-SQGILNIE.js";
+} from "./_node-chunks/chunk-OQHY5QS2.js";
 import {
   __commonJS,
   __name,
   __toESM
-} from "./_node-chunks/chunk-UHNP62OV.js";
+} from "./_node-chunks/chunk-RU4LPJPR.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;
   }
 });
 
@@ -972,7 +968,7 @@ var require_max_satisfying = __commonJS({
     "use strict";
     var SemVer = require_semver();
     var Range = require_range();
-    var maxSatisfying = /* @__PURE__ */ __name((versions, range, options) => {
+    var maxSatisfying = /* @__PURE__ */ __name((versions2, range, options) => {
       let max = null;
       let maxSV = null;
       let rangeObj = null;
@@ -981,7 +977,7 @@ var require_max_satisfying = __commonJS({
       } catch (er) {
         return null;
       }
-      versions.forEach((v) => {
+      versions2.forEach((v) => {
         if (rangeObj.test(v)) {
           if (!max || maxSV.compare(v) === -1) {
             max = v;
@@ -1001,7 +997,7 @@ var require_min_satisfying = __commonJS({
     "use strict";
     var SemVer = require_semver();
     var Range = require_range();
-    var minSatisfying = /* @__PURE__ */ __name((versions, range, options) => {
+    var minSatisfying = /* @__PURE__ */ __name((versions2, range, options) => {
       let min = null;
       let minSV = null;
       let rangeObj = null;
@@ -1010,7 +1006,7 @@ var require_min_satisfying = __commonJS({
       } catch (er) {
         return null;
       }
-      versions.forEach((v) => {
+      versions2.forEach((v) => {
         if (rangeObj.test(v)) {
           if (!min || minSV.compare(v) === 1) {
             min = v;
@@ -1208,11 +1204,11 @@ var require_simplify = __commonJS({
     "use strict";
     var satisfies2 = require_satisfies();
     var compare = require_compare();
-    module.exports = (versions, range, options) => {
+    module.exports = (versions2, range, options) => {
       const set = [];
       let first = null;
       let prev = null;
-      const v = versions.sort((a, b) => compare(a, b, options));
+      const v = versions2.sort((a, b) => compare(a, b, options));
       for (const version of v) {
         const included = satisfies2(version, range, options);
         if (included) {
@@ -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,24 @@ 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,
+  versions
 } 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) => {
@@ -1771,274 +1735,133 @@ 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 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 info2 = await getStorybookInfo(options);
+  const info = await getStorybookInfo(options.configDir);
   const allDeps = packageManager.getAllDependencies();
-  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
+  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");
-      uniqueDependencies.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
-      `
-    );
-    uniqueDependencies.push(`@vitest/coverage-v8`);
-  }
-  const versionedDependencies = uniqueDependencies.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"]);
   const viteConfigFile = findFile("vite.config");
   const vitestConfigFile = findFile("vitest.config");
@@ -2070,14 +1893,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}
@@ -2089,8 +1910,7 @@ 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;
@@ -2109,9 +1929,8 @@ 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(
@@ -2119,139 +1938,92 @@ async function postInstall(options) {
         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 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)
     });
-    logger2.line(1);
-    logger2.plain(`${step} Creating a Vitest config file:`);
-    logger2.plain(`  ${newConfigFile}`);
+    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:`);
-      await execa(
+      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(
         "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"
-        }
+        command,
+        versions.storybook
       );
+      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) {
-      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)}
-
+      console.log(e);
+      logger.line();
+      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`
+  logger.line();
+  if (errors.length === 0) {
+    logger.step(CLI_COLORS.success("All done!"));
+    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
 };