import { randomUUID } from "./utils.mjs";
import { EventType } from "@ag-ui/client";

//#region src/finalize-events.ts
const defaultStopMessage = "Run stopped by user";
const defaultAbruptEndMessage = "Run ended without emitting a terminal event";
function finalizeRunEvents(events, options = {}) {
	const { stopRequested = false, interruptionMessage } = options;
	const resolvedStopMessage = interruptionMessage ?? defaultStopMessage;
	const resolvedAbruptMessage = interruptionMessage && interruptionMessage !== defaultStopMessage ? interruptionMessage : defaultAbruptEndMessage;
	const appended = [];
	const openMessageIds = /* @__PURE__ */ new Set();
	const openToolCalls = /* @__PURE__ */ new Map();
	for (const event of events) switch (event.type) {
		case EventType.TEXT_MESSAGE_START: {
			const messageId = event.messageId;
			if (typeof messageId === "string") openMessageIds.add(messageId);
			break;
		}
		case EventType.TEXT_MESSAGE_END: {
			const messageId = event.messageId;
			if (typeof messageId === "string") openMessageIds.delete(messageId);
			break;
		}
		case EventType.TOOL_CALL_START: {
			const toolCallId = event.toolCallId;
			if (typeof toolCallId === "string") openToolCalls.set(toolCallId, {
				hasEnd: false,
				hasResult: false
			});
			break;
		}
		case EventType.TOOL_CALL_END: {
			const toolCallId = event.toolCallId;
			const info = toolCallId ? openToolCalls.get(toolCallId) : void 0;
			if (info) info.hasEnd = true;
			break;
		}
		case EventType.TOOL_CALL_RESULT: {
			const toolCallId = event.toolCallId;
			const info = toolCallId ? openToolCalls.get(toolCallId) : void 0;
			if (info) info.hasResult = true;
			break;
		}
		default: break;
	}
	const hasRunFinished = events.some((event) => event.type === EventType.RUN_FINISHED);
	const hasRunError = events.some((event) => event.type === EventType.RUN_ERROR);
	const terminalEventMissing = !(hasRunFinished || hasRunError);
	for (const messageId of openMessageIds) {
		const endEvent = {
			type: EventType.TEXT_MESSAGE_END,
			messageId
		};
		events.push(endEvent);
		appended.push(endEvent);
	}
	for (const [toolCallId, info] of openToolCalls) {
		if (!info.hasEnd) {
			const endEvent = {
				type: EventType.TOOL_CALL_END,
				toolCallId
			};
			events.push(endEvent);
			appended.push(endEvent);
		}
		if (terminalEventMissing && !info.hasResult) {
			const resultEvent = {
				type: EventType.TOOL_CALL_RESULT,
				toolCallId,
				messageId: `${toolCallId ?? randomUUID()}-result`,
				role: "tool",
				content: JSON.stringify(stopRequested ? {
					status: "stopped",
					reason: "stop_requested",
					message: resolvedStopMessage
				} : {
					status: "error",
					reason: "missing_terminal_event",
					message: resolvedAbruptMessage
				})
			};
			events.push(resultEvent);
			appended.push(resultEvent);
		}
	}
	if (terminalEventMissing) if (stopRequested) {
		const finishedEvent = { type: EventType.RUN_FINISHED };
		events.push(finishedEvent);
		appended.push(finishedEvent);
	} else {
		const errorEvent = {
			type: EventType.RUN_ERROR,
			message: resolvedAbruptMessage,
			code: "INCOMPLETE_STREAM"
		};
		events.push(errorEvent);
		appended.push(errorEvent);
	}
	return appended;
}

//#endregion
export { finalizeRunEvents };
//# sourceMappingURL=finalize-events.mjs.map