@vitest-agent/plugin 1.1.6 → 1.1.7

package/README.md

@@ -4,12 +4,12 @@
 [![License: MIT](https://img.shields.io/badge/License-MIT-4caf50.svg)](https://opensource.org/licenses/MIT)
 [![TypeScript 6.0](https://img.shields.io/badge/TypeScript-6.0-3178c6.svg)](https://www.typescriptlang.org/)
 
-Vitest plugin that turns your test suite into a live data source for LLM coding agents. Handles persistence, failure classification, coverage policy enforcement, project discovery and a configurable reporter chain — the CLI and MCP server arrive as required peers and are wired automatically.
+Vitest plugin that turns your test suite into a live data source for LLM coding agents. Handles persistence, failure classification, coverage policy enforcement, project discovery and a configurable reporter chain — the CLI and MCP server ship as dependencies and are wired automatically.
 
 ## Features
 
 - **`AgentPlugin`** — drop into `vitest.config.ts`; auto-detects human, agent and CI executors and adapts console output accordingly
-- **Project discovery** — `AgentPlugin.discover()` scans workspace packages, returns `{ projects, tags }` ready for `test.projects` and `test.tags`
+- **Project discovery** — `AgentPlugin.discover()` scans workspace packages, returns `{ projects, tags }` ready for `test.projects` and `test.tags`; classification tags apply at collection time, so every test declaration form inherits them, including wrapper testers like `@effect/vitest`'s `it.effect`
 - **Coverage presets** — `COVERAGE_LEVELS` and `COVERAGE_LEVELS_PER_FILE` return dual-output `{ thresholds, coverageTargets }` objects; `COVERAGE_AUTOUPDATE` tolerance functions plug into Vitest's native `autoUpdate`
 - **Failure classification** — persists per-test errors, computes failure signatures, classifies tests as stable, new-failure, persistent, flaky or recovered across runs
 - **Custom reporters** — pass any `VitestAgentReporterFactory` as the `reporter` option; the default wires `DefaultVitestAgentReporter` from `@vitest-agent/reporter`
@@ -23,7 +23,7 @@ npm install --save-dev @vitest-agent/plugin
 pnpm add -D @vitest-agent/plugin
 ```
 
-The required peers `@vitest-agent/cli` and `@vitest-agent/mcp` install automatically with npm 7+ and pnpm (`autoInstallPeers: true`).
+`@vitest-agent/cli` and `@vitest-agent/mcp` ship as regular dependencies of the plugin, so they install with it on every package manager.
 
 ## Quick start
 

package/index.d.ts

@@ -157,7 +157,7 @@ declare class DefaultDiscoverStrategy extends DiscoverStrategy {
  * @public
  */
 interface InjectTagsResult {
-  /** The transformed source code string with injected `tags` arguments. */
+  /** The transformed source code string with the prepended tag prelude. */
   code: string;
   /** Source map correlating transformed positions back to the original source. */
   map: SourceMap;
@@ -193,8 +193,9 @@ interface AgentPluginConstructorOptions extends AgentPluginOptions {
    */
   onRunEvent?: (event: RunEvent) => void;
   /**
-   * Controls the Vite transform hook that rewrites test() and it() call
-   * options to inject filename-derived tags. Pass a DiscoverStrategy
+   * Controls the Vite transform hook that prepends a tag prelude to each
+   * classified test file, applying filename-derived tags at collection
+   * time so every declaration form inherits them. Pass a DiscoverStrategy
    * instance to customize classification, or false to disable the
    * transform entirely.
    *

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@vitest-agent/plugin",
-	"version": "1.1.6",
+	"version": "1.1.7",
 	"private": false,
 	"description": "Vitest plugin for the vitest-agent ecosystem: owns persistence, classification, baselines, trends, and dispatches rendering to a configurable reporter.",
 	"keywords": [
@@ -46,17 +46,15 @@
 		"@effect/sql": "^0.51.1",
 		"@effect/sql-sqlite-node": "^0.52.0",
 		"@effect/workflow": "^0.18.2",
+		"@vitest-agent/cli": "1.0.6",
+		"@vitest-agent/mcp": "1.3.4",
 		"@vitest-agent/reporter": "1.0.6",
 		"@vitest-agent/sdk": "1.3.2",
-		"acorn": "^8.17.0",
-		"acorn-typescript": "^1.4.13",
 		"effect": "^3.21.4",
 		"magic-string": "^0.30.21",
 		"workspaces-effect": "^2.0.2"
 	},
 	"peerDependencies": {
-		"@vitest-agent/cli": "1.0.6",
-		"@vitest-agent/mcp": "1.3.4",
 		"@vitest/coverage-istanbul": "^4.1.0",
 		"@vitest/coverage-v8": "^4.1.0",
 		"vitest": "^4.1.0"

package/plugin.js

@@ -98,7 +98,7 @@ const aggregatedReporterByVitest = /* @__PURE__ */ new WeakSet();
 *
 * @public
 */
-const CURRENT_PLUGIN_VERSION = "1.1.6";
+const CURRENT_PLUGIN_VERSION = "1.1.7";
 const TEST_FILE_SUFFIX_RE = /\.(?:test|spec)\.(?:ts|tsx|js|jsx)$/;
 const TEST_FILE_DIR_RE = /\/(?:src|__test__)\//;
 const isTestFile = (id) => TEST_FILE_SUFFIX_RE.test(id) && TEST_FILE_DIR_RE.test(id);

package/utils/inject-tags.js

@@ -1,92 +1,36 @@
-import * as acorn from "acorn";
-import { tsPlugin } from "acorn-typescript";
 import MagicString from "magic-string";
 
 //#region src/utils/inject-tags.ts
-const Parser = acorn.Parser.extend(tsPlugin());
-const TEST_NAMES = /* @__PURE__ */ new Set(["test", "it"]);
-function rootIdentifier(n) {
-	if (n.type === "Identifier") return n;
-	if (n.type === "MemberExpression") return rootIdentifier(n.object);
-	if (n.type === "CallExpression") return rootIdentifier(n.callee);
-	return null;
-}
-function isTestCallee(callee) {
-	const root = rootIdentifier(callee);
-	return !!root && TEST_NAMES.has(root.name);
-}
-function hasTagsField(objectExpression) {
-	const props = objectExpression.properties ?? [];
-	for (const p of props) {
-		if (p.type !== "Property") continue;
-		const key = p.key;
-		if (key.type === "Identifier" && key.name === "tags") return true;
-		if (key.type === "Literal" && key.value === "tags") return true;
-	}
-	return false;
-}
 function tagsLiteral(tags) {
 	return `[${tags.map((t) => JSON.stringify(t)).join(", ")}]`;
 }
-function walk(node, visit) {
-	visit(node);
-	for (const key of Object.keys(node)) {
-		if (key === "type" || key === "start" || key === "end" || key === "loc") continue;
-		const v = node[key];
-		if (Array.isArray(v)) {
-			for (const item of v) if (item && typeof item === "object" && "type" in item) walk(item, visit);
-		} else if (v && typeof v === "object" && "type" in v) walk(v, visit);
-	}
+/**
+* Builds the guarded prelude that applies the classified tags to the file
+* task at collection time. `TestRunner.getCurrentSuite()` is a public
+* static on the main vitest entry (>= 4.1, the plugin's peer floor); the
+* `.suite ?? .file` fallback mirrors vitest's own parent-task resolution.
+* The runner unions parent tags into every suite and test it registers,
+* so tags set here are inherited by every declaration form in the file —
+* native it/test, wrapper testers like @effect/vitest's it.effect, aliases
+* from test.extend, and dynamically registered tests (issue #133). The
+* namespace import plus optional chaining on `TestRunner` means environments
+* whose "vitest" entry lacks the `TestRunner` export (e.g. vitest's browser-
+* mode entry) degrade to untagged tests instead of a module-instantiation
+* failure. If a future vitest changes the collector shape, the try/catch
+* degrades to "tests carry no tags" — never a crash.
+*/
+function preludeFor(tags) {
+	return `import * as __vitestAgentVitest from "vitest";
+try { const __vitestAgentCollector = __vitestAgentVitest.TestRunner?.getCurrentSuite?.(); const __vitestAgentTask = __vitestAgentCollector?.suite ?? __vitestAgentCollector?.file; if (__vitestAgentTask) { __vitestAgentTask.tags = [...new Set([...(__vitestAgentTask.tags ?? []), ...${tagsLiteral(tags)}])]; } } catch {}
+`;
 }
 function injectTags(source, tags) {
 	if (tags.length === 0) return null;
-	let ast;
-	try {
-		ast = Parser.parse(source, {
-			ecmaVersion: "latest",
-			sourceType: "module",
-			locations: true
-		});
-	} catch {
-		return null;
-	}
 	const ms = new MagicString(source);
-	let mutated = false;
-	walk(ast, (node) => {
-		if (node.type !== "CallExpression") return;
-		const callee = node.callee;
-		if (!isTestCallee(callee)) return;
-		const args = node.arguments ?? [];
-		if (args.length < 2) return;
-		const last = args[args.length - 1];
-		if (!(last.type === "FunctionExpression" || last.type === "ArrowFunctionExpression")) return;
-		const optsCandidate = args.length >= 3 ? args[args.length - 2] : null;
-		if (optsCandidate) {
-			if (optsCandidate.type === "ObjectExpression") {
-				if (hasTagsField(optsCandidate)) return;
-				const lastProp = (optsCandidate.properties ?? []).at(-1);
-				if (lastProp === void 0) {
-					const insertPoint = optsCandidate.end - 1;
-					ms.appendLeft(insertPoint, `tags: ${tagsLiteral(tags)} `);
-				} else ms.appendLeft(lastProp.end, `, tags: ${tagsLiteral(tags)}`);
-				mutated = true;
-				return;
-			}
-			const optsStart = optsCandidate.start;
-			const optsEnd = optsCandidate.end;
-			const origExpr = source.slice(optsStart, optsEnd);
-			ms.overwrite(optsStart, optsEnd, `{ ...(${origExpr}), tags: ${tagsLiteral(tags)} }`);
-			mutated = true;
-			return;
-		}
-		const nameArg = args[0];
-		ms.appendRight(nameArg.end, `, { tags: ${tagsLiteral(tags)} }`);
-		mutated = true;
-	});
-	if (!mutated) return null;
+	ms.prepend(preludeFor(tags));
 	return {
 		code: ms.toString(),
-		map: ms.generateMap({ hires: true })
+		map: ms.generateMap()
 	};
 }