File size: 2,818 Bytes
78d0c37 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 | import { normalizeOptionalString } from "@openclaw/normalization-core/string-coerce";
import type { SessionEntry } from "../../config/sessions.js";
import {
consumeSessionWorkAdmissionHandoff,
type SessionWorkAdmissionLease,
} from "../../sessions/session-lifecycle-admission.js";
export type ExpectedExistingSessionConstraint = {
handoffId?: string;
sessionId: string;
};
export class ExpectedExistingSessionChangedError extends Error {
constructor(message: string) {
super(message);
this.name = "ExpectedExistingSessionChangedError";
}
}
export function resolveExpectedExistingSessionConstraint(params: {
canUseInternalRuntimeHandoff: boolean;
expectedExistingSessionId?: unknown;
internalRuntimeHandoffId?: unknown;
}): { ok: true; constraint?: ExpectedExistingSessionConstraint } | { ok: false; error: string } {
const sessionId = normalizeOptionalString(params.expectedExistingSessionId);
if (!sessionId) {
return { ok: true };
}
if (!params.canUseInternalRuntimeHandoff) {
return {
ok: false,
error: "expectedExistingSessionId is reserved for backend callers.",
};
}
const handoffId = normalizeOptionalString(params.internalRuntimeHandoffId);
return {
ok: true,
constraint: { sessionId, ...(handoffId ? { handoffId } : {}) },
};
}
export function validateExpectedExistingSessionTarget(params: {
constraint?: ExpectedExistingSessionConstraint;
requestedSessionId?: string;
requestedSessionKey?: string;
}): string | undefined {
if (!params.constraint) {
return undefined;
}
if (!params.requestedSessionKey) {
return "expectedExistingSessionId requires an explicit session key.";
}
if (params.requestedSessionId && params.requestedSessionId !== params.constraint.sessionId) {
return "conflicting session identity constraints.";
}
return undefined;
}
export function assertExpectedExistingSession(params: {
constraint?: ExpectedExistingSessionConstraint;
entry?: SessionEntry;
message: string;
}): void {
if (params.constraint && params.entry?.sessionId !== params.constraint.sessionId) {
throw new ExpectedExistingSessionChangedError(params.message);
}
}
export function consumeExpectedSessionWorkAdmission(params: {
constraint?: ExpectedExistingSessionConstraint;
identities: Iterable<string | undefined>;
onInterrupt: (reason?: Error) => void;
scope: string;
}): SessionWorkAdmissionLease | undefined {
const handoffId = params.constraint?.handoffId;
if (!handoffId) {
return undefined;
}
const lease = consumeSessionWorkAdmissionHandoff({
handoffId,
scope: params.scope,
identities: params.identities,
onInterrupt: params.onInterrupt,
});
if (!lease) {
throw new Error("session work admission handoff is unavailable");
}
return lease;
}
|