File size: 2,044 Bytes
fcd8223 | 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 | import { isPromiseLike } from "@openclaw/normalization-core/promise-like";
import { attachPluginApiFacades } from "./api-facades.js";
import type { OpenClawPluginApi, OpenClawPluginDefinition } from "./types.js";
const LATE_CALLABLE_PLUGIN_API_METHODS: ReadonlySet<string> = new Set<keyof OpenClawPluginApi>([
"clearRunContext",
"emitAgentEvent",
"enqueueNextTurnInjection",
"getRunContext",
"sendSessionAttachment",
"scheduleSessionTurn",
"setRunContext",
"unscheduleSessionTurnsByTag",
]);
function createGuardedPluginRegistrationApi(api: OpenClawPluginApi): {
api: OpenClawPluginApi;
close: () => void;
} {
let closed = false;
const guardedApi = attachPluginApiFacades(
new Proxy(api, {
get(target, prop, receiver) {
const value = Reflect.get(target, prop, receiver);
if (typeof value !== "function") {
return value;
}
if (typeof prop === "string" && LATE_CALLABLE_PLUGIN_API_METHODS.has(prop)) {
return (...args: unknown[]) => Reflect.apply(value, target, args);
}
return (...args: unknown[]) => {
if (closed) {
return undefined;
}
return Reflect.apply(value, target, args);
};
},
}),
);
return {
api: guardedApi,
close: () => {
closed = true;
},
};
}
export function runPluginRegistration(
register: NonNullable<OpenClawPluginDefinition["register"]>,
api: Parameters<NonNullable<OpenClawPluginDefinition["register"]>>[0],
asyncResult: "reject" | "ignore" = "reject",
trackPending?: (pending: Promise<unknown>) => void,
): void {
const guarded = createGuardedPluginRegistrationApi(api);
try {
const result = register(guarded.api);
if (isPromiseLike(result)) {
const pending = Promise.resolve(result);
trackPending?.(pending);
void pending.catch(() => {});
if (asyncResult === "reject") {
throw new Error("plugin register must be synchronous");
}
}
} finally {
guarded.close();
}
}
|