vitest 4.0.6 → 4.0.8

package/dist/chunks/console.Cf-YriPC.js

@@ -0,0 +1,146 @@
+import { Console } from 'node:console';
+import { relative } from 'node:path';
+import { Writable } from 'node:stream';
+import { getSafeTimers } from '@vitest/utils/timers';
+import c from 'tinyrainbow';
+import { R as RealDate } from './date.Bq6ZW5rf.js';
+import { g as getWorkerState } from './utils.DvEY5TfP.js';
+
+const UNKNOWN_TEST_ID = "__vitest__unknown_test__";
+function getTaskIdByStack(root) {
+	const stack = (/* @__PURE__ */ new Error("STACK_TRACE_ERROR")).stack?.split("\n");
+	if (!stack) return UNKNOWN_TEST_ID;
+	const index = stack.findIndex((line) => line.includes("at Console.value"));
+	const line = index === -1 ? null : stack[index + 2];
+	if (!line) return UNKNOWN_TEST_ID;
+	const filepath = line.match(/at\s(.*)\s?/)?.[1];
+	if (filepath) return relative(root, filepath);
+	return UNKNOWN_TEST_ID;
+}
+function createCustomConsole(defaultState) {
+	const stdoutBuffer = /* @__PURE__ */ new Map();
+	const stderrBuffer = /* @__PURE__ */ new Map();
+	const timers = /* @__PURE__ */ new Map();
+	const { queueMicrotask } = getSafeTimers();
+	function queueCancelableMicrotask(callback) {
+		let canceled = false;
+		queueMicrotask(() => {
+			if (!canceled) callback();
+		});
+		return () => {
+			canceled = true;
+		};
+	}
+	const state = () => defaultState || getWorkerState();
+	// group sync console.log calls with micro task
+	function schedule(taskId) {
+		const timer = timers.get(taskId);
+		const { stdoutTime, stderrTime } = timer;
+		timer.cancel?.();
+		timer.cancel = queueCancelableMicrotask(() => {
+			if (stderrTime < stdoutTime) {
+				sendStderr(taskId);
+				sendStdout(taskId);
+			} else {
+				sendStdout(taskId);
+				sendStderr(taskId);
+			}
+		});
+	}
+	function sendStdout(taskId) {
+		sendBuffer("stdout", taskId);
+	}
+	function sendStderr(taskId) {
+		sendBuffer("stderr", taskId);
+	}
+	function sendBuffer(type, taskId) {
+		const buffers = type === "stdout" ? stdoutBuffer : stderrBuffer;
+		const buffer = buffers.get(taskId);
+		if (!buffer) return;
+		if (state().config.printConsoleTrace) buffer.forEach(([buffer, origin]) => {
+			sendLog(type, taskId, String(buffer), buffer.length, origin);
+		});
+		else sendLog(type, taskId, buffer.map((i) => String(i[0])).join(""), buffer.length);
+		const timer = timers.get(taskId);
+		buffers.delete(taskId);
+		if (type === "stderr") timer.stderrTime = 0;
+		else timer.stdoutTime = 0;
+	}
+	function sendLog(type, taskId, content, size, origin) {
+		const timer = timers.get(taskId);
+		const time = type === "stderr" ? timer.stderrTime : timer.stdoutTime;
+		state().rpc.onUserConsoleLog({
+			type,
+			content: content || "<empty line>",
+			taskId,
+			time: time || RealDate.now(),
+			size,
+			origin
+		});
+	}
+	return new Console({
+		stdout: new Writable({ write(data, encoding, callback) {
+			const s = state();
+			const id = s?.current?.id || s?.current?.suite?.id || s.current?.file.id || getTaskIdByStack(s.config.root);
+			let timer = timers.get(id);
+			if (timer) timer.stdoutTime = timer.stdoutTime || RealDate.now();
+			else {
+				timer = {
+					stdoutTime: RealDate.now(),
+					stderrTime: RealDate.now()
+				};
+				timers.set(id, timer);
+			}
+			let buffer = stdoutBuffer.get(id);
+			if (!buffer) {
+				buffer = [];
+				stdoutBuffer.set(id, buffer);
+			}
+			if (state().config.printConsoleTrace) {
+				const limit = Error.stackTraceLimit;
+				Error.stackTraceLimit = limit + 6;
+				const trace = (/* @__PURE__ */ new Error("STACK_TRACE")).stack?.split("\n").slice(7).join("\n");
+				Error.stackTraceLimit = limit;
+				buffer.push([data, trace]);
+			} else buffer.push([data, void 0]);
+			schedule(id);
+			callback();
+		} }),
+		stderr: new Writable({ write(data, encoding, callback) {
+			const s = state();
+			const id = s?.current?.id || s?.current?.suite?.id || s.current?.file.id || getTaskIdByStack(s.config.root);
+			let timer = timers.get(id);
+			if (timer) timer.stderrTime = timer.stderrTime || RealDate.now();
+			else {
+				timer = {
+					stderrTime: RealDate.now(),
+					stdoutTime: RealDate.now()
+				};
+				timers.set(id, timer);
+			}
+			let buffer = stderrBuffer.get(id);
+			if (!buffer) {
+				buffer = [];
+				stderrBuffer.set(id, buffer);
+			}
+			if (state().config.printConsoleTrace) {
+				const limit = Error.stackTraceLimit;
+				Error.stackTraceLimit = limit + 6;
+				const stack = (/* @__PURE__ */ new Error("STACK_TRACE")).stack?.split("\n");
+				Error.stackTraceLimit = limit;
+				if (stack?.some((line) => line.includes("at Console.trace"))) buffer.push([data, void 0]);
+				else {
+					const trace = stack?.slice(7).join("\n");
+					Error.stackTraceLimit = limit;
+					buffer.push([data, trace]);
+				}
+			} else buffer.push([data, void 0]);
+			schedule(id);
+			callback();
+		} }),
+		colorMode: c.isColorSupported,
+		groupIndentation: 2
+	});
+}
+
+export { UNKNOWN_TEST_ID, createCustomConsole };