@storybook/addon-vitest 0.0.0-pr-32921-sha-12ed7665 → 0.0.0-pr-32928-sha-4befe9a9

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,29 +1,33 @@
1
- import CJS_COMPAT_NODE_URL_juf8p1r8sqm from 'node:url';
2
- import CJS_COMPAT_NODE_PATH_juf8p1r8sqm from 'node:path';
3
- import CJS_COMPAT_NODE_MODULE_juf8p1r8sqm from "node:module";
1
+ import CJS_COMPAT_NODE_URL_6s72upzd1a9 from 'node:url';
2
+ import CJS_COMPAT_NODE_PATH_6s72upzd1a9 from 'node:path';
3
+ import CJS_COMPAT_NODE_MODULE_6s72upzd1a9 from "node:module";
4
4
 
5
- var __filename = CJS_COMPAT_NODE_URL_juf8p1r8sqm.fileURLToPath(import.meta.url);
6
- var __dirname = CJS_COMPAT_NODE_PATH_juf8p1r8sqm.dirname(__filename);
7
- var require = CJS_COMPAT_NODE_MODULE_juf8p1r8sqm.createRequire(import.meta.url);
5
+ var __filename = CJS_COMPAT_NODE_URL_6s72upzd1a9.fileURLToPath(import.meta.url);
6
+ var __dirname = CJS_COMPAT_NODE_PATH_6s72upzd1a9.dirname(__filename);
7
+ var require = CJS_COMPAT_NODE_MODULE_6s72upzd1a9.createRequire(import.meta.url);
8
8
 
9
9
  // ------------------------------------------------------------
10
10
  // end of CJS compatibility banner, injected by Storybook's esbuild configuration
11
11
  // ------------------------------------------------------------
12
12
  import {
13
+ execa,
13
14
  resolvePackageDir
14
- } from "./_node-chunks/chunk-DKHTLFGI.js";
15
+ } from "./_node-chunks/chunk-MDHHO4LN.js";
15
16
  import {
16
- any
17
- } from "./_node-chunks/chunk-UM4KUWJO.js";
17
+ any,
18
+ getAddonNames,
19
+ up
20
+ } from "./_node-chunks/chunk-XKIZF73O.js";
18
21
  import {
19
- DOCUMENTATION_LINK
20
- } from "./_node-chunks/chunk-4NV37VMK.js";
22
+ DOCUMENTATION_LINK,
23
+ SUPPORTED_FRAMEWORKS
24
+ } from "./_node-chunks/chunk-QDQ5ORRW.js";
21
25
  import {
22
26
  dirname,
23
27
  join,
24
28
  relative,
25
29
  resolve
26
- } from "./_node-chunks/chunk-YDCELX4B.js";
30
+ } from "./_node-chunks/chunk-MD2GMYRF.js";
27
31
  import {
28
32
  require_compare,
29
33
  require_constants,
@@ -33,12 +37,12 @@ import {
33
37
  require_parse_options,
34
38
  require_re,
35
39
  require_semver
36
- } from "./_node-chunks/chunk-EKF5O6IB.js";
40
+ } from "./_node-chunks/chunk-FV5ODK67.js";
37
41
  import {
38
42
  __commonJS,
39
43
  __name,
40
44
  __toESM
41
- } from "./_node-chunks/chunk-BXS7J2KF.js";
45
+ } from "./_node-chunks/chunk-IFVSP7X6.js";
42
46
 
43
47
  // ../../node_modules/semver/functions/parse.js
44
48
  var require_parse = __commonJS({
@@ -360,7 +364,7 @@ var require_coerce = __commonJS({
360
364
  var SemVer = require_semver();
361
365
  var parse = require_parse();
362
366
  var { safeRe: re, t } = require_re();
363
- var coerce = /* @__PURE__ */ __name((version, options) => {
367
+ var coerce2 = /* @__PURE__ */ __name((version, options) => {
364
368
  if (version instanceof SemVer) {
365
369
  return version;
366
370
  }
@@ -395,7 +399,7 @@ var require_coerce = __commonJS({
395
399
  const build = options.includePrerelease && match[6] ? `+${match[6]}` : "";
396
400
  return parse(`${major}.${minor}.${patch}${prerelease}${build}`, options);
397
401
  }, "coerce");
398
- module.exports = coerce;
402
+ module.exports = coerce2;
399
403
  }
400
404
  });
401
405
 
@@ -1440,7 +1444,7 @@ var require_semver2 = __commonJS({
1440
1444
  var gte = require_gte();
1441
1445
  var lte = require_lte();
1442
1446
  var cmp = require_cmp();
1443
- var coerce = require_coerce();
1447
+ var coerce2 = require_coerce();
1444
1448
  var Comparator = require_comparator();
1445
1449
  var Range = require_range();
1446
1450
  var satisfies2 = require_satisfies();
@@ -1478,7 +1482,7 @@ var require_semver2 = __commonJS({
1478
1482
  gte,
1479
1483
  lte,
1480
1484
  cmp,
1481
- coerce,
1485
+ coerce: coerce2,
1482
1486
  Comparator,
1483
1487
  Range,
1484
1488
  satisfies: satisfies2,
@@ -1509,23 +1513,56 @@ var require_semver2 = __commonJS({
1509
1513
  import { existsSync } from "node:fs";
1510
1514
  import * as fs2 from "node:fs/promises";
1511
1515
  import { writeFile } from "node:fs/promises";
1512
- import { babelParse, generate } from "storybook/internal/babel";
1513
- import { AddonVitestService } from "storybook/internal/cli";
1516
+ import { isAbsolute, posix, sep } from "node:path";
1517
+ import { fileURLToPath, pathToFileURL } from "node:url";
1518
+ import { babelParse, generate, traverse } from "storybook/internal/babel";
1514
1519
  import {
1515
1520
  JsPackageManagerFactory,
1516
1521
  formatFileContent,
1522
+ getInterpretedFile,
1517
1523
  getProjectRoot,
1518
- getStorybookInfo
1524
+ isCI as isCI2,
1525
+ loadMainConfig,
1526
+ scanAndTransformFiles,
1527
+ transformImportFiles
1519
1528
  } from "storybook/internal/common";
1520
- import { CLI_COLORS } from "storybook/internal/node-logger";
1521
- import {
1522
- AddonVitestPostinstallError,
1523
- AddonVitestPostinstallPrerequisiteCheckError
1524
- } from "storybook/internal/server-errors";
1525
- import { SupportedFramework } from "storybook/internal/types";
1529
+ import { experimental_loadStorybook } from "storybook/internal/core-server";
1530
+ import { readConfig, writeConfig } from "storybook/internal/csf-tools";
1531
+ import { logger as logger2 } from "storybook/internal/node-logger";
1532
+
1533
+ // ../../node_modules/empathic/package.mjs
1534
+ function up2(options) {
1535
+ return up("package.json", options);
1536
+ }
1537
+ __name(up2, "up");
1538
+
1539
+ // src/postinstall.ts
1526
1540
  var import_semver = __toESM(require_semver2(), 1);
1541
+ import prompts from "prompts";
1527
1542
  import { dedent } from "ts-dedent";
1528
1543
 
1544
+ // src/postinstall-logger.ts
1545
+ import { isCI } from "storybook/internal/common";
1546
+ import { colors, logger } from "storybook/internal/node-logger";
1547
+ var fancy = process.platform !== "win32" || isCI() || process.env.TERM === "xterm-256color";
1548
+ var step = colors.gray("\u203A");
1549
+ var info = colors.blue(fancy ? "\u2139" : "i");
1550
+ var success = colors.green(fancy ? "\u2714" : "\u221A");
1551
+ var warning = colors.orange(fancy ? "\u26A0" : "\u203C");
1552
+ var error = colors.red(fancy ? "\u2716" : "\xD7");
1553
+ var baseOptions = {
1554
+ borderStyle: "round",
1555
+ padding: 1
1556
+ };
1557
+ var print = /* @__PURE__ */ __name((message, options) => {
1558
+ logger.line(1);
1559
+ logger.logBox(message, { ...baseOptions, ...options });
1560
+ }, "print");
1561
+ var printInfo = /* @__PURE__ */ __name((title, message, options) => print(message, { borderColor: "blue", title, ...options }), "printInfo");
1562
+ var printWarning = /* @__PURE__ */ __name((title, message, options) => print(message, { borderColor: "yellow", title, ...options }), "printWarning");
1563
+ var printError = /* @__PURE__ */ __name((title, message, options) => print(message, { borderColor: "red", title, ...options }), "printError");
1564
+ var printSuccess = /* @__PURE__ */ __name((title, message, options) => print(message, { borderColor: "green", title, ...options }), "printSuccess");
1565
+
1529
1566
  // src/updateVitestFile.ts
1530
1567
  import * as fs from "node:fs/promises";
1531
1568
  var loadTemplate = /* @__PURE__ */ __name(async (name, replacements) => {
@@ -1734,133 +1771,274 @@ var updateWorkspaceFile = /* @__PURE__ */ __name((source, target) => {
1734
1771
  var ADDON_NAME = "@storybook/addon-vitest";
1735
1772
  var EXTENSIONS = [".ts", ".tsx", ".js", ".jsx", ".cts", ".mts", ".cjs", ".mjs"];
1736
1773
  var addonA11yName = "@storybook/addon-a11y";
1774
+ var hasErrors = false;
1775
+ function nameMatches(name, pattern) {
1776
+ if (name === pattern) {
1777
+ return true;
1778
+ }
1779
+ if (name.includes(`${pattern}${sep}`)) {
1780
+ return true;
1781
+ }
1782
+ if (name.includes(`${pattern}${posix.sep}`)) {
1783
+ return true;
1784
+ }
1785
+ return false;
1786
+ }
1787
+ __name(nameMatches, "nameMatches");
1788
+ var logErrors = /* @__PURE__ */ __name((...args) => {
1789
+ hasErrors = true;
1790
+ printError(...args);
1791
+ }, "logErrors");
1792
+ var findFile = /* @__PURE__ */ __name((basename, extensions = EXTENSIONS) => any(
1793
+ extensions.map((ext) => basename + ext),
1794
+ { last: getProjectRoot() }
1795
+ ), "findFile");
1737
1796
  async function postInstall(options) {
1738
- const errors = [];
1739
- const { logger, prompt } = options;
1797
+ printSuccess(
1798
+ "\u{1F44B} Howdy!",
1799
+ dedent`
1800
+ I'm the installation helper for ${ADDON_NAME}
1801
+
1802
+ Hold on for a moment while I look at your project and get it set up...
1803
+ `
1804
+ );
1740
1805
  const packageManager = JsPackageManagerFactory.getPackageManager({
1741
1806
  force: options.packageManager
1742
1807
  });
1743
- const findFile = /* @__PURE__ */ __name((basename, extensions = EXTENSIONS) => any(
1744
- extensions.map((ext) => basename + ext),
1745
- { last: getProjectRoot(), cwd: options.configDir }
1746
- ), "findFile");
1747
1808
  const vitestVersionSpecifier = await packageManager.getInstalledVersion("vitest");
1809
+ const coercedVitestVersion = vitestVersionSpecifier ? (0, import_semver.coerce)(vitestVersionSpecifier) : null;
1748
1810
  const isVitest3_2To4 = vitestVersionSpecifier ? (0, import_semver.satisfies)(vitestVersionSpecifier, ">=3.2.0 <4.0.0") : false;
1749
1811
  const isVitest4OrNewer = vitestVersionSpecifier ? (0, import_semver.satisfies)(vitestVersionSpecifier, ">=4.0.0") : true;
1750
- const info = await getStorybookInfo(options.configDir);
1812
+ const info2 = await getStorybookInfo(options);
1751
1813
  const allDeps = packageManager.getAllDependencies();
1752
- const addonVitestService = new AddonVitestService();
1753
- const compatibilityResult = await addonVitestService.validateCompatibility({
1754
- packageManager,
1755
- framework: info.framework,
1756
- builder: info.builder
1757
- });
1758
- let result = null;
1759
- if (!compatibilityResult.compatible && compatibilityResult.reasons) {
1760
- const reasons = compatibilityResult.reasons.map((r) => `\u2022 ${CLI_COLORS.error(r)}`);
1761
- reasons.unshift(dedent`
1762
- Automated setup failed
1763
- We have found incompatibilities due to the following package incompatibilities:
1764
- `);
1765
- reasons.push(
1766
- dedent`
1767
- 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
1768
- in your main Storybook config file and remove the dependency from your package.json file.
1814
+ const dependencies = [
1815
+ "vitest",
1816
+ "playwright",
1817
+ isVitest4OrNewer ? "@vitest/browser-playwright" : "@vitest/browser"
1818
+ ];
1819
+ const uniqueDependencies = dependencies.filter((p) => !allDeps[p]);
1820
+ const mainJsPath = getInterpretedFile(resolve(options.configDir, "main"));
1821
+ const config = await readConfig(mainJsPath);
1822
+ const hasCustomWebpackConfig = !!config.getFieldNode(["webpackFinal"]);
1823
+ const isInteractive = process.stdout.isTTY && !isCI2();
1824
+ if (nameMatches(info2.frameworkPackageName, "@storybook/nextjs") && !hasCustomWebpackConfig) {
1825
+ const out = options.yes || !isInteractive ? { migrateToNextjsVite: !!options.yes } : await prompts({
1826
+ type: "confirm",
1827
+ name: "migrateToNextjsVite",
1828
+ message: dedent`
1829
+ The addon requires the use of @storybook/nextjs-vite to work with Next.js.
1830
+ https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#install-and-set-up
1769
1831
 
1770
- Please check the documentation for more information about its requirements and installation:
1771
- https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}
1772
- `
1773
- );
1774
- result = reasons.map((r) => r.trim()).join("\n\n");
1775
- }
1776
- if (result) {
1777
- logger.error(result);
1778
- throw new AddonVitestPostinstallPrerequisiteCheckError({
1779
- reasons: compatibilityResult.reasons
1832
+ Do you want to migrate?
1833
+ `,
1834
+ initial: true
1780
1835
  });
1836
+ if (out.migrateToNextjsVite) {
1837
+ await packageManager.addDependencies({ type: "devDependencies", skipInstall: true }, [
1838
+ "@storybook/nextjs-vite"
1839
+ ]);
1840
+ await packageManager.removeDependencies(["@storybook/nextjs"]);
1841
+ traverse(config._ast, {
1842
+ StringLiteral(path) {
1843
+ if (path.node.value === "@storybook/nextjs") {
1844
+ path.node.value = "@storybook/nextjs-vite";
1845
+ }
1846
+ }
1847
+ });
1848
+ await writeConfig(config, mainJsPath);
1849
+ info2.frameworkPackageName = "@storybook/nextjs-vite";
1850
+ info2.builderPackageName = "@storybook/builder-vite";
1851
+ await scanAndTransformFiles({
1852
+ promptMessage: "Enter a glob to scan for all @storybook/nextjs imports to substitute with @storybook/nextjs-vite:",
1853
+ force: options.yes,
1854
+ dryRun: false,
1855
+ transformFn: /* @__PURE__ */ __name((files, options2, dryRun) => transformImportFiles(files, options2, dryRun), "transformFn"),
1856
+ transformOptions: {
1857
+ "@storybook/nextjs": "@storybook/nextjs-vite"
1858
+ }
1859
+ });
1860
+ }
1781
1861
  }
1782
- if (!options.skipDependencyManagement) {
1783
- const versionedDependencies = await addonVitestService.collectDependencies(packageManager);
1784
- if (info.framework === SupportedFramework.NEXTJS) {
1785
- const allDeps2 = packageManager.getAllDependencies();
1786
- if (!allDeps2["@storybook/nextjs-vite"]) {
1862
+ const annotationsImport = SUPPORTED_FRAMEWORKS.find(
1863
+ (f) => nameMatches(info2.frameworkPackageName, f)
1864
+ ) ? info2.frameworkPackageName === "@storybook/nextjs" ? "@storybook/nextjs-vite" : info2.frameworkPackageName : null;
1865
+ const isRendererSupported = !!annotationsImport;
1866
+ const prerequisiteCheck = /* @__PURE__ */ __name(async () => {
1867
+ const reasons = [];
1868
+ if (hasCustomWebpackConfig) {
1869
+ reasons.push("\u2022 The addon can not be used with a custom Webpack configuration.");
1870
+ }
1871
+ if (!nameMatches(info2.frameworkPackageName, "@storybook/nextjs") && !nameMatches(info2.builderPackageName, "@storybook/builder-vite")) {
1872
+ reasons.push(
1873
+ "\u2022 The addon can only be used with a Vite-based Storybook framework or Next.js."
1874
+ );
1875
+ }
1876
+ if (!isRendererSupported) {
1877
+ reasons.push(dedent`
1878
+ • The addon cannot yet be used with ${info2.frameworkPackageName}
1879
+ `);
1880
+ }
1881
+ if (coercedVitestVersion && !(0, import_semver.satisfies)(coercedVitestVersion, ">=3.0.0")) {
1882
+ reasons.push(dedent`
1883
+ • The addon requires Vitest 3.0.0 or higher. You are currently using ${vitestVersionSpecifier}.
1884
+ Please update all of your Vitest dependencies and try again.
1885
+ `);
1886
+ }
1887
+ const mswVersionSpecifier = await packageManager.getInstalledVersion("msw");
1888
+ const coercedMswVersion = mswVersionSpecifier ? (0, import_semver.coerce)(mswVersionSpecifier) : null;
1889
+ if (coercedMswVersion && !(0, import_semver.satisfies)(coercedMswVersion, ">=2.0.0")) {
1890
+ reasons.push(dedent`
1891
+ • 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.
1892
+ Please update the 'msw' package and try again.
1893
+ `);
1894
+ }
1895
+ if (nameMatches(info2.frameworkPackageName, "@storybook/nextjs")) {
1896
+ const nextVersion = await packageManager.getInstalledVersion("next");
1897
+ if (!nextVersion) {
1898
+ reasons.push(dedent`
1899
+ • You are using @storybook/nextjs without having "next" installed.
1900
+ Please install "next" or use a different Storybook framework integration and try again.
1901
+ `);
1787
1902
  }
1788
1903
  }
1789
- const v8Version = await packageManager.getInstalledVersion("@vitest/coverage-v8");
1790
- const istanbulVersion = await packageManager.getInstalledVersion("@vitest/coverage-istanbul");
1791
- if (!v8Version && !istanbulVersion) {
1792
- logger.step(
1904
+ if (reasons.length > 0) {
1905
+ reasons.unshift(
1906
+ `@storybook/addon-vitest's automated setup failed due to the following package incompatibilities:`
1907
+ );
1908
+ reasons.push("--------------------------------");
1909
+ reasons.push(
1793
1910
  dedent`
1794
- You don't seem to have a coverage reporter installed. Vitest needs either V8 or Istanbul to generate coverage reports.
1795
-
1796
- Adding "@vitest/coverage-v8" to enable coverage reporting.
1797
- Read more about Vitest coverage providers at https://vitest.dev/guide/coverage.html#coverage-providers
1911
+ 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
1912
+ in your main Storybook config file and remove the dependency from your package.json file.
1798
1913
  `
1799
1914
  );
1915
+ if (!isRendererSupported) {
1916
+ reasons.push(
1917
+ dedent`
1918
+ Please check the documentation for more information about its requirements and installation:
1919
+ https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}
1920
+ `
1921
+ );
1922
+ } else {
1923
+ reasons.push(
1924
+ dedent`
1925
+ Fear not, however, you can follow the manual installation process instead at:
1926
+ https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#manual-setup
1927
+ `
1928
+ );
1929
+ }
1930
+ return reasons.map((r) => r.trim()).join("\n\n");
1800
1931
  }
1801
- if (versionedDependencies.length > 0) {
1802
- logger.step("Adding dependencies to your package.json");
1803
- logger.log(" " + versionedDependencies.join(", "));
1804
- await packageManager.addDependencies(
1805
- { type: "devDependencies", skipInstall: true },
1806
- versionedDependencies
1807
- );
1932
+ return null;
1933
+ }, "prerequisiteCheck");
1934
+ const result = await prerequisiteCheck();
1935
+ if (result) {
1936
+ logErrors("\u26D4\uFE0F Sorry!", result);
1937
+ logger2.line(1);
1938
+ return;
1939
+ }
1940
+ if (info2.frameworkPackageName === "@storybook/nextjs") {
1941
+ printInfo(
1942
+ "\u{1F37F} Just so you know...",
1943
+ dedent`
1944
+ It looks like you're using Next.js.
1945
+
1946
+ Adding "@storybook/nextjs-vite/vite-plugin" so you can use it with Vitest.
1947
+
1948
+ More info about the plugin at https://github.com/storybookjs/vite-plugin-storybook-nextjs
1949
+ `
1950
+ );
1951
+ try {
1952
+ const storybookVersion = await packageManager.getInstalledVersion("storybook");
1953
+ uniqueDependencies.push(`@storybook/nextjs-vite@^${storybookVersion}`);
1954
+ } catch (e) {
1955
+ console.error("Failed to install @storybook/nextjs-vite. Please install it manually");
1808
1956
  }
1809
- if (!options.skipInstall) {
1810
- await packageManager.installDependencies();
1957
+ }
1958
+ const v8Version = await packageManager.getInstalledVersion("@vitest/coverage-v8");
1959
+ const istanbulVersion = await packageManager.getInstalledVersion("@vitest/coverage-istanbul");
1960
+ if (!v8Version && !istanbulVersion) {
1961
+ printInfo(
1962
+ "\u{1F648} Let me cover this for you",
1963
+ dedent`
1964
+ You don't seem to have a coverage reporter installed. Vitest needs either V8 or Istanbul to generate coverage reports.
1965
+
1966
+ Adding "@vitest/coverage-v8" to enable coverage reporting.
1967
+ Read more about Vitest coverage providers at https://vitest.dev/guide/coverage.html#coverage-providers
1968
+ `
1969
+ );
1970
+ uniqueDependencies.push(`@vitest/coverage-v8`);
1971
+ }
1972
+ const versionedDependencies = uniqueDependencies.map((p) => {
1973
+ if (p.includes("vitest")) {
1974
+ return vitestVersionSpecifier ? `${p}@${vitestVersionSpecifier}` : p;
1811
1975
  }
1976
+ return p;
1977
+ });
1978
+ if (versionedDependencies.length > 0) {
1979
+ await packageManager.addDependencies(
1980
+ { type: "devDependencies", skipInstall: true },
1981
+ versionedDependencies
1982
+ );
1983
+ logger2.line(1);
1984
+ logger2.plain(`${step} Installing dependencies:`);
1985
+ logger2.plain(" " + versionedDependencies.join(", "));
1812
1986
  }
1813
- if (!options.skipDependencyManagement) {
1814
- if (!options.skipInstall) {
1815
- const playwrightErrors = await addonVitestService.installPlaywright(packageManager, {
1816
- yes: options.yes
1987
+ await packageManager.installDependencies();
1988
+ logger2.line(1);
1989
+ if (options.skipInstall) {
1990
+ logger2.plain("Skipping Playwright installation, please run this command manually:");
1991
+ logger2.plain(" npx playwright install chromium --with-deps");
1992
+ } else {
1993
+ logger2.plain(`${step} Configuring Playwright with Chromium (this might take some time):`);
1994
+ logger2.plain(" npx playwright install chromium --with-deps");
1995
+ try {
1996
+ await packageManager.executeCommand({
1997
+ command: "npx",
1998
+ args: ["playwright", "install", "chromium", "--with-deps"]
1817
1999
  });
1818
- errors.push(...playwrightErrors);
1819
- } else {
1820
- logger.warn(dedent`
1821
- Playwright browser binaries installation skipped. Please run the following command manually later:
1822
- ${CLI_COLORS.cta("npx playwright install chromium --with-deps")}
1823
- `);
2000
+ } catch (e) {
2001
+ console.error("Failed to install Playwright. Please install it manually");
1824
2002
  }
1825
2003
  }
1826
2004
  const fileExtension = allDeps.typescript || findFile("tsconfig", [...EXTENSIONS, ".json"]) ? "ts" : "js";
1827
2005
  const vitestSetupFile = resolve(options.configDir, `vitest.setup.${fileExtension}`);
1828
2006
  if (existsSync(vitestSetupFile)) {
1829
- const errorMessage = dedent`
1830
- Found an existing Vitest setup file:
1831
- ${vitestSetupFile}
1832
- Please refer to the documentation to complete the setup manually:
1833
- https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#manual-setup
1834
- `;
1835
- logger.line();
1836
- logger.error(`${errorMessage}
1837
- `);
1838
- errors.push("Found existing Vitest setup file");
1839
- } else {
1840
- logger.step(`Creating a Vitest setup file for Storybook:`);
1841
- logger.log(`${vitestSetupFile}
1842
- `);
1843
- const previewExists = EXTENSIONS.map((ext) => resolve(options.configDir, `preview${ext}`)).some(
1844
- existsSync
1845
- );
1846
- const annotationsImport = info.frameworkPackage;
1847
- const imports = [`import { setProjectAnnotations } from '${annotationsImport}';`];
1848
- const projectAnnotations = [];
1849
- if (previewExists) {
1850
- imports.push(`import * as projectAnnotations from './preview';`);
1851
- projectAnnotations.push("projectAnnotations");
1852
- }
1853
- await writeFile(
1854
- vitestSetupFile,
2007
+ logErrors(
2008
+ "\u{1F6A8} Oh no!",
1855
2009
  dedent`
2010
+ Found an existing Vitest setup file:
2011
+ ${vitestSetupFile}
2012
+
2013
+ Please refer to the documentation to complete the setup manually:
2014
+ https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#manual-setup
2015
+ `
2016
+ );
2017
+ logger2.line(1);
2018
+ return;
2019
+ }
2020
+ logger2.line(1);
2021
+ logger2.plain(`${step} Creating a Vitest setup file for Storybook:`);
2022
+ logger2.plain(` ${vitestSetupFile}`);
2023
+ const previewExists = EXTENSIONS.map((ext) => resolve(options.configDir, `preview${ext}`)).some(
2024
+ existsSync
2025
+ );
2026
+ const imports = [`import { setProjectAnnotations } from '${annotationsImport}';`];
2027
+ const projectAnnotations = [];
2028
+ if (previewExists) {
2029
+ imports.push(`import * as projectAnnotations from './preview';`);
2030
+ projectAnnotations.push("projectAnnotations");
2031
+ }
2032
+ await writeFile(
2033
+ vitestSetupFile,
2034
+ dedent`
1856
2035
  ${imports.join("\n")}
1857
2036
 
1858
2037
  // This is an important step to apply the right configuration when testing your stories.
1859
2038
  // More info at: https://storybook.js.org/docs/api/portable-stories/portable-stories-vitest#setprojectannotations
1860
2039
  setProjectAnnotations([${projectAnnotations.join(", ")}]);
1861
2040
  `
1862
- );
1863
- }
2041
+ );
1864
2042
  const vitestWorkspaceFile = findFile("vitest.workspace", [".ts", ".js", ".json"]);
1865
2043
  const viteConfigFile = findFile("vite.config");
1866
2044
  const vitestConfigFile = findFile("vitest.config");
@@ -1892,12 +2070,14 @@ async function postInstall(options) {
1892
2070
  const target = babelParse(workspaceFile);
1893
2071
  const updated = updateWorkspaceFile(source, target);
1894
2072
  if (updated) {
1895
- logger.step(`Updating your Vitest workspace file...`);
1896
- logger.log(`${vitestWorkspaceFile}`);
2073
+ logger2.line(1);
2074
+ logger2.plain(`${step} Updating your Vitest workspace file:`);
2075
+ logger2.plain(` ${vitestWorkspaceFile}`);
1897
2076
  const formattedContent = await formatFileContent(vitestWorkspaceFile, generate(target).code);
1898
2077
  await writeFile(vitestWorkspaceFile, formattedContent);
1899
2078
  } else {
1900
- logger.error(
2079
+ logErrors(
2080
+ "\u{1F6A8} Oh no!",
1901
2081
  dedent`
1902
2082
  Could not update existing Vitest workspace file:
1903
2083
  ${vitestWorkspaceFile}
@@ -1909,7 +2089,8 @@ async function postInstall(options) {
1909
2089
  https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#manual-setup
1910
2090
  `
1911
2091
  );
1912
- errors.push("Unable to update existing Vitest workspace file");
2092
+ logger2.line(1);
2093
+ return;
1913
2094
  }
1914
2095
  } else if (rootConfig) {
1915
2096
  let target, updated;
@@ -1928,8 +2109,9 @@ async function postInstall(options) {
1928
2109
  updated = updateConfigFile(source, target);
1929
2110
  }
1930
2111
  if (target && updated) {
1931
- logger.step(`Updating your ${vitestConfigFile ? "Vitest" : "Vite"} config file:`);
1932
- logger.log(` ${rootConfig}`);
2112
+ logger2.line(1);
2113
+ logger2.plain(`${step} Updating your ${vitestConfigFile ? "Vitest" : "Vite"} config file:`);
2114
+ logger2.plain(` ${rootConfig}`);
1933
2115
  const formattedContent = await formatFileContent(rootConfig, generate(target).code);
1934
2116
  const shouldAddReference = !configFileHasTypeReference && !vitestConfigFile;
1935
2117
  await writeFile(
@@ -1937,90 +2119,139 @@ async function postInstall(options) {
1937
2119
  shouldAddReference ? '/// <reference types="vitest/config" />\n' + formattedContent : formattedContent
1938
2120
  );
1939
2121
  } else {
1940
- logger.error(dedent`
2122
+ logErrors(
2123
+ "\u{1F6A8} Oh no!",
2124
+ dedent`
1941
2125
  We were unable to update your existing ${vitestConfigFile ? "Vitest" : "Vite"} config file.
1942
2126
 
1943
2127
  Please refer to the documentation to complete the setup manually:
1944
2128
  https://storybook.js.org/docs/writing-tests/integrations/vitest-addon#manual-setup
1945
- `);
1946
- errors.push("Unable to update existing Vitest config file");
2129
+ `
2130
+ );
1947
2131
  }
1948
2132
  } else {
1949
- const parentDir = dirname(options.configDir);
1950
- const newConfigFile = resolve(parentDir, `vitest.config.${fileExtension}`);
2133
+ const newConfigFile = resolve(`vitest.config.${fileExtension}`);
1951
2134
  const configTemplate = await loadTemplate(getTemplateName(), {
1952
2135
  CONFIG_DIR: options.configDir,
1953
2136
  SETUP_FILE: relative(dirname(newConfigFile), vitestSetupFile)
1954
2137
  });
1955
- logger.step(`Creating a Vitest config file:`);
1956
- logger.log(`${newConfigFile}`);
2138
+ logger2.line(1);
2139
+ logger2.plain(`${step} Creating a Vitest config file:`);
2140
+ logger2.plain(` ${newConfigFile}`);
1957
2141
  const formattedContent = await formatFileContent(newConfigFile, configTemplate);
1958
2142
  await writeFile(newConfigFile, formattedContent);
1959
2143
  }
1960
- const a11yAddon = info.addons.find((addon) => addon.includes(addonA11yName));
2144
+ const a11yAddon = info2.addons.find((addon) => addon.includes(addonA11yName));
1961
2145
  if (a11yAddon) {
1962
2146
  try {
1963
- const command = ["automigrate", "addon-a11y-addon-test"];
1964
- command.push("--loglevel", "silent");
1965
- command.push("--yes", "--skip-doctor");
1966
- if (options.packageManager) {
1967
- command.push("--package-manager", options.packageManager);
1968
- }
1969
- if (options.skipInstall) {
1970
- command.push("--skip-install");
1971
- }
1972
- if (options.configDir !== ".storybook") {
1973
- command.push("--config-dir", `"${options.configDir}"`);
1974
- }
1975
- const remoteCommand = packageManager.getRemoteRunCommand("storybook", command);
1976
- const [cmd, ...args] = remoteCommand.split(" ");
1977
- await prompt.executeTask(
1978
- // TODO: Remove stdio: 'ignore' once we have a way to log the output of the command properly
1979
- () => packageManager.executeCommand({ command: cmd, args, stdio: "ignore" }),
2147
+ logger2.plain(`${step} Setting up ${addonA11yName} for @storybook/addon-vitest:`);
2148
+ await execa(
2149
+ "storybook",
2150
+ [
2151
+ "automigrate",
2152
+ "addon-a11y-addon-test",
2153
+ "--loglevel",
2154
+ "silent",
2155
+ "--yes",
2156
+ "--skip-doctor",
2157
+ ...options.packageManager ? ["--package-manager", options.packageManager] : [],
2158
+ ...options.skipInstall ? ["--skip-install"] : [],
2159
+ ...options.configDir !== ".storybook" ? ["--config-dir", `"${options.configDir}"`] : []
2160
+ ],
1980
2161
  {
1981
- intro: "Setting up a11y addon for @storybook/addon-vitest",
1982
- error: "Failed to setup a11y addon for @storybook/addon-vitest",
1983
- success: "a11y addon setup successfully"
2162
+ stdio: "inherit"
1984
2163
  }
1985
2164
  );
1986
2165
  } catch (e) {
1987
- logger.line();
1988
- logger.error(dedent`
1989
- Could not automatically set up ${addonA11yName} for @storybook/addon-vitest.
2166
+ logErrors(
2167
+ "\u{1F6A8} Oh no!",
2168
+ dedent`
2169
+ We have detected that you have ${addonA11yName} installed but could not automatically set it up for @storybook/addon-vitest:
2170
+
2171
+ ${e instanceof Error ? e.message : String(e)}
2172
+
1990
2173
  Please refer to the documentation to complete the setup manually:
1991
2174
  https://storybook.js.org/docs/writing-tests/accessibility-testing#test-addon-integration
1992
- `);
1993
- errors.push(
1994
- "The @storybook/addon-a11y couldn't be set up for the Vitest addon" + (e instanceof Error ? e.stack : String(e))
2175
+ `
1995
2176
  );
1996
2177
  }
1997
2178
  }
1998
2179
  const runCommand = rootConfig ? `npx vitest --project=storybook` : `npx vitest`;
1999
- logger.line();
2000
- if (errors.length === 0) {
2001
- logger.step(CLI_COLORS.success("@storybook/addon-vitest setup completed successfully"));
2002
- logger.log(dedent`
2180
+ if (!hasErrors) {
2181
+ printSuccess(
2182
+ "\u{1F389} All done!",
2183
+ dedent`
2003
2184
  @storybook/addon-vitest is now configured and you're ready to run your tests!
2185
+
2004
2186
  Here are a couple of tips to get you started:
2005
-
2006
- You can run tests with "${CLI_COLORS.cta(runCommand)}"
2007
- • Vitest IDE extension shows all stories as tests in your editor!
2008
-
2187
+ • You can run tests with "${runCommand}"
2188
+ When using the Vitest extension in your editor, all of your stories will be shown as tests!
2189
+
2009
2190
  Check the documentation for more information about its features and options at:
2010
2191
  https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}
2011
- `);
2192
+ `
2193
+ );
2012
2194
  } else {
2013
- logger.warn(
2195
+ printWarning(
2196
+ "\u26A0\uFE0F Done, but with errors!",
2014
2197
  dedent`
2015
- Done, but with errors!
2016
- @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:
2198
+ @storybook/addon-vitest was installed successfully, but there were some errors during the setup process.
2199
+
2200
+ Please refer to the documentation to complete the setup manually and check the errors above:
2017
2201
  https://storybook.js.org/docs/next/${DOCUMENTATION_LINK}#manual-setup
2018
2202
  `
2019
2203
  );
2020
- throw new AddonVitestPostinstallError({ errors });
2021
2204
  }
2205
+ logger2.line(1);
2022
2206
  }
2023
2207
  __name(postInstall, "postInstall");
2208
+ async function getPackageNameFromPath(input) {
2209
+ const path = input.startsWith("file://") ? fileURLToPath(input) : input;
2210
+ if (!isAbsolute(path)) {
2211
+ return path;
2212
+ }
2213
+ const packageJsonPath = up2({ cwd: path });
2214
+ if (!packageJsonPath) {
2215
+ throw new Error(`Could not find package.json in path: ${path}`);
2216
+ }
2217
+ const { default: packageJson } = await import(pathToFileURL(packageJsonPath).href, {
2218
+ with: { type: "json" }
2219
+ });
2220
+ return packageJson.name;
2221
+ }
2222
+ __name(getPackageNameFromPath, "getPackageNameFromPath");
2223
+ async function getStorybookInfo({ configDir, packageManager: pkgMgr }) {
2224
+ const packageManager = JsPackageManagerFactory.getPackageManager({ force: pkgMgr, configDir });
2225
+ const { packageJson } = packageManager.primaryPackageJson;
2226
+ const config = await loadMainConfig({ configDir });
2227
+ const { presets } = await experimental_loadStorybook({
2228
+ configDir,
2229
+ packageJson
2230
+ });
2231
+ const framework = await presets.apply("framework", {});
2232
+ const core = await presets.apply("core", {});
2233
+ const { builder, renderer } = core;
2234
+ if (!builder) {
2235
+ throw new Error("Could not detect your Storybook builder.");
2236
+ }
2237
+ const frameworkPackageName = await getPackageNameFromPath(
2238
+ typeof framework === "string" ? framework : framework.name
2239
+ );
2240
+ const builderPackageName = await getPackageNameFromPath(
2241
+ typeof builder === "string" ? builder : builder.name
2242
+ );
2243
+ let rendererPackageName;
2244
+ if (renderer) {
2245
+ rendererPackageName = await getPackageNameFromPath(renderer);
2246
+ }
2247
+ return {
2248
+ frameworkPackageName,
2249
+ builderPackageName,
2250
+ rendererPackageName,
2251
+ addons: getAddonNames(config)
2252
+ };
2253
+ }
2254
+ __name(getStorybookInfo, "getStorybookInfo");
2024
2255
  export {
2025
2256
  postInstall as default
2026
2257
  };