fix(vitest-plugin): stop importing the deprecated vitest subpaths

Vitest 4.1 moved the benchmark runner and the suite helpers to the main
`vitest` entry point and deprecated the `vitest/runners` and
`vitest/suite` subpaths, which now print a warning on every import.

Add a compat module that reads `BenchmarkRunner` and the `TestRunner`
statics from the main entry when they are available, and falls back to
the subpaths otherwise, keeping the vitest 3.2 and 4.0 support the peer
dependency range advertises. Resolution happens through a top-level
await because the runner is needed as a base class, so the two runner
bundles are built for es2022; `jsPlugins` now takes an optional target
and keeps es2020 for every other entry point.

Co-Authored-By: Claude <noreply@anthropic.com>

Author: adriencaccia

packages/vitest-plugin/rollup.config.mjs

@@ -26,13 +26,15 @@ export default defineConfig([
   {
     input: "src/analysis.ts",
     output: { file: "dist/analysis.mjs", format: "es" },
-    plugins: jsPlugins(pkg.version),
+    // top-level await
+    plugins: jsPlugins(pkg.version, "es2022"),
     external: ["@codspeed/core", /^vitest/],
   },
   {
     input: "src/walltime/index.ts",
     output: { file: "dist/walltime.mjs", format: "es" },
-    plugins: jsPlugins(pkg.version),
+    // top-level await
+    plugins: jsPlugins(pkg.version, "es2022"),
     external: ["@codspeed/core", /^vitest/],
   },
 ]);

packages/vitest-plugin/src/__tests__/instrumented.test.ts

@@ -1,7 +1,7 @@
 import { fromPartial } from "@total-typescript/shoehorn";
 import { describe, expect, it, vi, type RunnerTestSuite } from "vitest";
-import { getBenchFn } from "vitest/suite";
 import { AnalysisRunner as CodSpeedRunner } from "../analysis";
+import { getBenchFn } from "../compat";
 
 const coreMocks = vi.hoisted(() => {
   return {
@@ -28,8 +28,8 @@ vi.mock("@codspeed/core", async (importOriginal) => {
 
 console.log = vi.fn();
 
-vi.mock("vitest/suite", async (importOriginal) => {
-  const actual = await importOriginal<typeof import("vitest/suite")>();
+vi.mock("../compat", async (importOriginal) => {
+  const actual = await importOriginal<typeof import("../compat")>();
   return {
     ...actual,
     getBenchFn: vi.fn(),

packages/vitest-plugin/src/analysis.ts

@@ -8,13 +8,12 @@ import {
   wrapWithRootFrame,
 } from "@codspeed/core";
 import { Benchmark, type RunnerTestSuite } from "vitest";
-import { NodeBenchmarkRunner } from "vitest/runners";
-import { getBenchFn } from "vitest/suite";
 import {
   callSuiteHook,
   isVitestTaskBenchmark,
   patchRootSuiteWithFullFilePath,
 } from "./common";
+import { getBenchFn, NodeBenchmarkRunner } from "./compat";
 
 const currentFileName =
   typeof __filename === "string"

packages/vitest-plugin/src/common.ts

@@ -1,7 +1,7 @@
 import { getGitDir } from "@codspeed/core";
 import path from "path";
 import { Benchmark, type RunnerTask, type RunnerTestSuite } from "vitest";
-import { getHooks } from "vitest/suite";
+import { getHooks } from "./compat";
 type SuiteHooks = ReturnType<typeof getHooks>;
 
 function getSuiteHooks(suite: RunnerTestSuite, name: keyof SuiteHooks) {

packages/vitest-plugin/src/compat.ts

@@ -0,0 +1,35 @@
+type VitestExports = typeof import("vitest");
+
+/**
+ * Vitest 4.1 moved the benchmark runner and the suite helpers to the main
+ * `vitest` entry point and deprecated the `vitest/runners` and `vitest/suite`
+ * subpaths, which warn on import.
+ */
+async function resolveVitestApi() {
+  const { BenchmarkRunner, TestRunner }: Partial<VitestExports> =
+    await import("vitest");
+
+  if (BenchmarkRunner && TestRunner) {
+    return {
+      NodeBenchmarkRunner: BenchmarkRunner,
+      getHooks: TestRunner.getSuiteHooks,
+      getBenchFn: TestRunner.getBenchFn,
+      getBenchOptions: TestRunner.getBenchOptions,
+    };
+  }
+
+  const [runners, suite] = await Promise.all([
+    import("vitest/runners"),
+    import("vitest/suite"),
+  ]);
+
+  return {
+    NodeBenchmarkRunner: runners.NodeBenchmarkRunner,
+    getHooks: suite.getHooks,
+    getBenchFn: suite.getBenchFn,
+    getBenchOptions: suite.getBenchOptions,
+  };
+}
+
+export const { NodeBenchmarkRunner, getHooks, getBenchFn, getBenchOptions } =
+  await resolveVitestApi();

packages/vitest-plugin/src/walltime/index.ts

@@ -12,8 +12,8 @@ import {
   RunnerTaskResultPack,
   type RunnerTestSuite,
 } from "vitest";
-import { NodeBenchmarkRunner } from "vitest/runners";
 import { patchRootSuiteWithFullFilePath } from "../common";
+import { NodeBenchmarkRunner } from "../compat";
 import { extractBenchmarkResults } from "./utils";
 
 type Tinybench = typeof tinybench;

packages/vitest-plugin/src/walltime/utils.ts

@@ -10,8 +10,8 @@ import {
   type RunnerTestSuite,
   type Benchmark as VitestBenchmark,
 } from "vitest";
-import { getBenchOptions } from "vitest/suite";
 import { isVitestTaskBenchmark } from "../common";
+import { getBenchOptions } from "../compat";
 
 export async function extractBenchmarkResults(
   suite: RunnerTestSuite,

rollup.options.mjs

@@ -13,9 +13,10 @@ import esbuild from "rollup-plugin-esbuild";
  */
 export const declarationsPlugin = (options) => [dts(options)];
 
-export const jsPlugins = (version) => [
+export const jsPlugins = (version, target = "es2020") => [
   json(),
   esbuild({
+    target,
     define: {
       __VERSION__: '"' + version + '"',
     },