openclaw/dist/migrations-B9Z9aYrA.mjs

Multi-channel AI gateway with extensible messaging integrations

import { t as createDeferredCore } from "./deferred-D0La5CRk.mjs";
import "./src-CdRlGRnW.mjs";
import { t as stableStringify } from "./stable-stringify-Djn3uxNo.mjs";
import { t as formatErrorMessage } from "./errors-O33D01XW.mjs";
import { n as sha256Hex } from "./node-crypto-BG4u3UnX.mjs";
import { n as normalizeAgentId, t as isValidAgentId } from "./agent-id-D1-Q6vR4.mjs";
import { d as resolveAgentWorkspaceDir } from "./agent-scope-config-CDYpzA0r.mjs";
import "./session-key-C09ow4h0.mjs";
import { i as listAgentIds } from "./agent-roster-BXD1uMJD.mjs";
import "./agent-scope-CG17uXFn.mjs";
import { df as validateMigrationsMemoryPlanParams, uf as validateMigrationsMemoryApplyParams } from "./src-l2dWvEsN.mjs";
import { t as ErrorCodes } from "./gateway-error-details-DIefe7Zy.mjs";
import { f as errorShape } from "./error-codes-Dh56LniQ.mjs";
import { y as summarizeMigrationItems } from "./migration-BeM6IrM8.mjs";
import { n as defineValidatedGatewayHandler } from "./validation-Bu27N0nv.mjs";
import { n as planProviderMemoryImport, r as withMemoryMigrationProviders, t as applyProviderMemoryImport } from "./memory-import-DmMrSaYe.mjs";
//#region src/gateway/server-methods/migrations.ts
const MEMORY_APPLY_DEDUPE_PREFIX = "migrations.memory.apply:";
const activeApplies = /* @__PURE__ */ new Set();
const inFlightMemoryApplies = /* @__PURE__ */ new WeakMap();
function memoryApplyInflightMap(dedupe) {
	let active = inFlightMemoryApplies.get(dedupe);
	if (!active) {
		active = /* @__PURE__ */ new Map();
		inFlightMemoryApplies.set(dedupe, active);
	}
	return active;
}
function isCachedMemoryApply(value) {
	if (!value || typeof value !== "object") return false;
	const candidate = value;
	return typeof candidate.requestFingerprint === "string" && candidate.outcome !== void 0;
}
function respondMemoryApply(outcome, respond, cached = false) {
	const meta = cached ? { cached: true } : void 0;
	if (outcome.ok) respond(true, JSON.parse(outcome.resultJson), void 0, meta);
	else respond(false, void 0, outcome.error, meta);
}
function toWireItem(item) {
	return {
		id: item.id,
		status: item.status,
		...item.source ? { source: item.source } : {},
		...item.target ? { target: item.target } : {},
		...item.message !== void 0 ? { message: item.message } : {},
		...item.reason !== void 0 ? { reason: item.reason } : {},
		...item.details !== void 0 ? { details: item.details } : {}
	};
}
function fingerprintMemoryPlan(params) {
	return sha256Hex(stableStringify({
		version: 3,
		agentId: params.agentId,
		workspace: params.workspace,
		providerId: params.providerId,
		overwrite: params.overwrite === true,
		plan: params.plan
	}));
}
function targetAgentOrRespond(rawAgentId, config, respond) {
	if (!isValidAgentId(rawAgentId)) {
		respond(false, void 0, errorShape(ErrorCodes.INVALID_REQUEST, "invalid agent id"));
		return;
	}
	const agentId = normalizeAgentId(rawAgentId);
	if (!listAgentIds(config).includes(agentId)) {
		respond(false, void 0, errorShape(ErrorCodes.INVALID_REQUEST, "unknown agent id"));
		return;
	}
	return agentId;
}
async function planMemoryProvider(params) {
	const base = {
		providerId: params.provider.id,
		label: params.provider.label,
		...params.provider.description ? { description: params.provider.description } : {}
	};
	try {
		const { detection, plan } = await planProviderMemoryImport({
			provider: params.provider,
			config: params.config,
			agentId: params.agentId,
			overwrite: params.overwrite
		});
		if (detection && !detection.found) return {
			...base,
			found: false,
			...detection.source ? { source: detection.source } : {},
			...detection.confidence ? { confidence: detection.confidence } : {},
			...detection.message ? { message: detection.message } : {},
			summary: summarizeMigrationItems([]),
			items: []
		};
		const found = plan.items.length > 0;
		const workspace = resolveAgentWorkspaceDir(params.config, params.agentId);
		return {
			...base,
			found,
			planFingerprint: fingerprintMemoryPlan({
				agentId: params.agentId,
				workspace,
				providerId: params.provider.id,
				overwrite: params.overwrite,
				plan
			}),
			source: plan.source,
			...plan.target ? { target: plan.target } : {},
			...detection?.confidence ? { confidence: detection.confidence } : {},
			...detection?.message ? { message: detection.message } : {},
			summary: plan.summary,
			items: plan.items.map(toWireItem),
			...plan.warnings?.length ? { warnings: plan.warnings } : {}
		};
	} catch (error) {
		return {
			...base,
			found: false,
			error: formatErrorMessage(error),
			summary: summarizeMigrationItems([]),
			items: []
		};
	}
}
const migrationsHandlers = {
	"migrations.memory.plan": defineValidatedGatewayHandler("migrations.memory.plan", validateMigrationsMemoryPlanParams, async ({ params, respond, context }) => {
		const config = context.getRuntimeConfig();
		const agentId = targetAgentOrRespond(params.agentId, config, respond);
		if (!agentId) return;
		const resultJson = await withMemoryMigrationProviders(config, async (providers) => {
			const planning = providers.map(async (provider) => await planMemoryProvider({
				provider,
				config,
				agentId,
				overwrite: params.overwrite
			}));
			let planned;
			try {
				planned = await Promise.all(planning);
			} catch (error) {
				await Promise.allSettled(planning);
				throw error;
			}
			const result = {
				agentId,
				workspace: resolveAgentWorkspaceDir(config, agentId),
				providers: planned
			};
			return JSON.stringify(result);
		});
		respond(true, JSON.parse(resultJson), void 0);
	}),
	"migrations.memory.apply": defineValidatedGatewayHandler("migrations.memory.apply", validateMigrationsMemoryApplyParams, async ({ params, respond, context }) => {
		const config = context.getRuntimeConfig();
		const agentId = targetAgentOrRespond(params.agentId, config, respond);
		if (!agentId) return;
		const requestFingerprint = stableStringify({
			agentId,
			providerId: params.providerId,
			planFingerprint: params.planFingerprint,
			itemIds: params.itemIds,
			overwrite: params.overwrite === true
		});
		const dedupeKey = `${MEMORY_APPLY_DEDUPE_PREFIX}${params.idempotencyKey}`;
		const cached = context.dedupe.get(dedupeKey);
		if (cached && isCachedMemoryApply(cached.payload)) {
			if (cached.payload.requestFingerprint !== requestFingerprint) {
				respond(false, void 0, errorShape(ErrorCodes.INVALID_REQUEST, "memory import idempotency key was reused"));
				return;
			}
			respondMemoryApply(cached.payload.outcome, respond, true);
			return;
		}
		const inFlightMap = memoryApplyInflightMap(context.dedupe);
		const inFlight = inFlightMap.get(dedupeKey);
		if (inFlight) {
			if (inFlight.requestFingerprint !== requestFingerprint) {
				respond(false, void 0, errorShape(ErrorCodes.INVALID_REQUEST, "memory import idempotency key was reused"));
				return;
			}
			respondMemoryApply(await inFlight.completion, respond, true);
			return;
		}
		const completion = createDeferredCore();
		inFlightMap.set(dedupeKey, {
			requestFingerprint,
			completion: completion.promise
		});
		let applyCompleted = false;
		let producedOutcome;
		let outcome;
		const runApply = async (providers) => {
			const provider = providers.find((candidate) => candidate.id === params.providerId);
			if (!provider) return {
				ok: false,
				error: errorShape(ErrorCodes.INVALID_REQUEST, "unknown memory migration provider")
			};
			const applyKey = `${agentId}:${provider.id}`;
			if (activeApplies.has(applyKey)) return {
				ok: false,
				error: errorShape(ErrorCodes.UNAVAILABLE, "memory import already running", {
					retryable: true,
					retryAfterMs: 1e3
				})
			};
			activeApplies.add(applyKey);
			try {
				const { plan } = await planProviderMemoryImport({
					provider,
					config,
					agentId,
					overwrite: params.overwrite
				});
				if (fingerprintMemoryPlan({
					agentId,
					workspace: resolveAgentWorkspaceDir(config, agentId),
					providerId: provider.id,
					overwrite: params.overwrite,
					plan
				}) !== params.planFingerprint) return {
					ok: false,
					error: errorShape(ErrorCodes.INVALID_REQUEST, "memory migration plan changed; refresh the plan before importing")
				};
				const selectable = new Map(plan.items.filter((item) => item.status === "planned" || item.status === "conflict").map((item) => [item.id, item]));
				const unavailable = params.itemIds.filter((id) => !selectable.has(id));
				if (unavailable.length > 0) return {
					ok: false,
					error: errorShape(ErrorCodes.INVALID_REQUEST, `memory migration items changed; refresh the plan (${unavailable.join(", ")})`)
				};
				const selectedConflicts = params.itemIds.filter((id) => selectable.get(id)?.status === "conflict");
				if (!params.overwrite && selectedConflicts.length > 0) return {
					ok: false,
					error: errorShape(ErrorCodes.INVALID_REQUEST, "selected memory was already imported; enable replacement and refresh the plan")
				};
				const applied = await applyProviderMemoryImport({
					provider,
					config,
					agentId,
					itemIds: params.itemIds,
					overwrite: params.overwrite,
					preflightPlan: plan,
					onApplyCompleted: () => {
						applyCompleted = true;
					}
				});
				const result = {
					providerId: applied.providerId,
					source: applied.source,
					...applied.target ? { target: applied.target } : {},
					summary: applied.summary,
					items: applied.items.map(toWireItem),
					...applied.warnings?.length ? { warnings: applied.warnings } : {},
					...applied.backupPath ? { backupPath: applied.backupPath } : {},
					...applied.reportDir ? { reportDir: applied.reportDir } : {}
				};
				return {
					ok: true,
					resultJson: JSON.stringify(result)
				};
			} finally {
				activeApplies.delete(applyKey);
			}
		};
		try {
			outcome = await withMemoryMigrationProviders(config, async (providers) => {
				try {
					producedOutcome = await runApply(providers);
				} catch (error) {
					producedOutcome = {
						ok: false,
						error: errorShape(ErrorCodes.UNAVAILABLE, applyCompleted ? `Memory import apply completed, but its result could not be returned: ${formatErrorMessage(error)}. Inspect the migration report before starting another import.` : formatErrorMessage(error))
					};
				}
				if (applyCompleted) context.dedupe.set(dedupeKey, {
					ts: Date.now(),
					ok: producedOutcome.ok,
					payload: {
						requestFingerprint,
						outcome: producedOutcome
					}
				});
				return producedOutcome;
			});
		} catch (error) {
			if (producedOutcome) {
				context.logGateway.warn(`Memory migration plugin cleanup failed: ${formatErrorMessage(error)}`);
				outcome = producedOutcome;
			} else outcome = {
				ok: false,
				error: errorShape(ErrorCodes.UNAVAILABLE, formatErrorMessage(error))
			};
		} finally {
			inFlightMap.delete(dedupeKey);
		}
		completion.resolve(outcome);
		respondMemoryApply(outcome, respond);
	})
};
//#endregion
export { migrationsHandlers };