@copilotkit/aimock 1.38.0 → 1.40.0

package/dist/server.cjs

@@ -145,23 +145,31 @@ function handleNotFound(res, message) {
 }
 const CONTROL_PREFIX = "/__aimock";
 /**
-* Perform a full fixtures reset: clear the fixtures array, journal, video/fal
-* generation state, and the interaction/event-id counters, then zero the
-* `aimock_fixtures_loaded` gauge. Shared by `/reset/fixtures` and the
-* deprecated `/reset` alias.
+* Perform a full reset: clear the fixtures array, the journal (entries *and*
+* per-test fixture match-counts, i.e. sequence position), the video and fal.ai
+* job/queue state, and the Gemini interaction/event-id counters, then re-zero
+* the `aimock_fixtures_loaded` gauge.
+*
+* `targets` is `null` when no server is running (an in-process `reset()` before
+* `start()`). The process-global generation state is reset either way, since it
+* is not owned by any one server instance.
+*
+* Shared by `POST /__aimock/reset` (canonical), its deprecated
+* `POST /__aimock/reset/fixtures` alias, and `LLMock.reset()`.
 */
-function performFixturesReset(fixtures, journal, videoStates, openRouterVideoJobs, veoVideoJobs, grokVideoJobs, defaults) {
+function performFullReset(fixtures, targets) {
 	fixtures.length = 0;
-	journal.clear();
-	videoStates.clear();
-	openRouterVideoJobs.clear();
-	veoVideoJobs.clear();
-	grokVideoJobs.clear();
 	require_fal_audio.falJobs.clear();
 	require_fal.falQueueStates.clear();
 	require_gemini_interactions.resetInteractionCounter();
 	require_gemini_interactions.resetEventIdCounter();
-	if (defaults.registry) defaults.registry.setGauge("aimock_fixtures_loaded", {}, fixtures.length);
+	if (!targets) return;
+	targets.journal.clear();
+	targets.videoStates.clear();
+	targets.openRouterVideoJobs.clear();
+	targets.veoVideoJobs.clear();
+	targets.grokVideoJobs.clear();
+	if (targets.defaults.registry) targets.defaults.registry.setGauge("aimock_fixtures_loaded", {}, fixtures.length);
 }
 /**
 * Handle requests under `/__aimock/`. Returns `true` if the request was
@@ -181,6 +189,11 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		res.end(JSON.stringify(journal.getAll()));
 		return true;
 	}
+	if (subPath === "/fixtures" && req.method === "GET") {
+		res.writeHead(200, { "Content-Type": "application/json" });
+		res.end(JSON.stringify({ count: fixtures.length }));
+		return true;
+	}
 	if (subPath === "/fixtures" && req.method === "POST") {
 		let raw;
 		try {
@@ -230,8 +243,16 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		res.end(JSON.stringify({ cleared: true }));
 		return true;
 	}
-	if (subPath === "/reset/fixtures" && req.method === "POST") {
-		performFixturesReset(fixtures, journal, videoStates, openRouterVideoJobs, veoVideoJobs, grokVideoJobs, defaults);
+	const resetTargets = () => ({
+		journal,
+		videoStates,
+		openRouterVideoJobs,
+		veoVideoJobs,
+		grokVideoJobs,
+		defaults
+	});
+	if (subPath === "/reset" && req.method === "POST") {
+		performFullReset(fixtures, resetTargets());
 		res.writeHead(200, { "Content-Type": "application/json" });
 		res.end(JSON.stringify({ reset: true }));
 		return true;
@@ -242,10 +263,10 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		res.end(JSON.stringify({ reset: true }));
 		return true;
 	}
-	if (subPath === "/reset" && req.method === "POST") {
-		performFixturesReset(fixtures, journal, videoStates, openRouterVideoJobs, veoVideoJobs, grokVideoJobs, defaults);
-		const deprecation = "POST /__aimock/reset is deprecated; use POST /__aimock/reset/fixtures (full reset) or POST /__aimock/reset/journal (journal only)";
-		defaults.logger.warn("POST /__aimock/reset is deprecated; use /__aimock/reset/fixtures or /__aimock/reset/journal");
+	if (subPath === "/reset/fixtures" && req.method === "POST") {
+		performFullReset(fixtures, resetTargets());
+		const deprecation = "POST /__aimock/reset/fixtures is deprecated; use POST /__aimock/reset (full reset) or POST /__aimock/reset/journal (journal only)";
+		defaults.logger.warn("POST /__aimock/reset/fixtures is deprecated; use /__aimock/reset or /__aimock/reset/journal");
 		res.writeHead(200, {
 			"Content-Type": "application/json",
 			Deprecation: "true"
@@ -1839,7 +1860,48 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 			if (!socket.destroyed) socket.destroy();
 		});
 	});
+	/**
+	* Body bytes the platform already parsed onto `req` for an upgrade request.
+	*
+	* Node >= 26 delivers an upgrade request's body here and leaves `head`
+	* empty; Node <= 24 leaves the body in `head` and ends `req` with nothing.
+	* The stream has already ended in both cases, so this resolves on the next
+	* tick either way — it never waits on the network.
+	*/
+	function drainParsedRequestBody(req) {
+		return new Promise((resolve) => {
+			const chunks = [];
+			let settled = false;
+			const done = () => {
+				if (settled) return;
+				settled = true;
+				resolve(Buffer.concat(chunks));
+			};
+			req.on("data", (c) => chunks.push(c));
+			req.on("end", done);
+			req.on("error", done);
+			req.on("close", done);
+			req.resume();
+		});
+	}
 	async function handleUpgradeRequest(req, socket, head) {
+		if ((req.headers.upgrade ?? "").toLowerCase() !== "websocket") {
+			const parsedBody = await drainParsedRequestBody(req);
+			const requestLine = `${req.method} ${req.url} HTTP/${req.httpVersion}\r\n`;
+			const headerLines = [];
+			for (let i = 0; i < req.rawHeaders.length; i += 2) {
+				if (/^(?:upgrade|connection)$/i.test(req.rawHeaders[i])) continue;
+				headerLines.push(`${req.rawHeaders[i]}: ${req.rawHeaders[i + 1]}`);
+			}
+			const rebuilt = Buffer.from(requestLine + headerLines.join("\r\n") + "\r\n\r\n");
+			socket.unshift(Buffer.concat([
+				rebuilt,
+				parsedBody,
+				head
+			]));
+			server.emit("connection", socket);
+			return;
+		}
 		const parsedUrl = new URL(req.url ?? "/", `http://${req.headers.host ?? "localhost"}`);
 		let pathname = parsedUrl.pathname;
 		if (!require_api_key_auth.validateRequestApiKey(req, resolvedAuth.policy).ok) {
@@ -1940,4 +2002,5 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 //#endregion
 exports.createServer = createServer;
 exports.createServerWithResolvedAuth = createServerWithResolvedAuth;
+exports.performFullReset = performFullReset;
 //# sourceMappingURL=server.cjs.map