mirror of
https://github.com/openclaw/openclaw.git
synced 2026-04-13 10:11:20 +00:00
refactor: share plugin cli registration helpers
This commit is contained in:
83
src/plugins/register-plugin-cli-command-groups.ts
Normal file
83
src/plugins/register-plugin-cli-command-groups.ts
Normal file
@@ -0,0 +1,83 @@
|
||||
import type { Command } from "commander";
|
||||
import {
|
||||
findCommandGroupEntry,
|
||||
getCommandGroupNames,
|
||||
registerLazyCommandGroup,
|
||||
removeCommandGroupNames,
|
||||
type CommandGroupEntry,
|
||||
} from "../cli/program/register-command-groups.js";
|
||||
import type { OpenClawPluginCliCommandDescriptor, PluginLogger } from "./types.js";
|
||||
|
||||
export type PluginCliCommandGroupEntry = CommandGroupEntry & {
|
||||
pluginId: string;
|
||||
};
|
||||
|
||||
export type PluginCliCommandGroupMode = "eager" | "lazy";
|
||||
|
||||
function canRegisterPluginCliLazily(entry: PluginCliCommandGroupEntry): boolean {
|
||||
if (entry.placeholders.length === 0) {
|
||||
return false;
|
||||
}
|
||||
const descriptorNames = new Set(
|
||||
(entry.placeholders as readonly OpenClawPluginCliCommandDescriptor[]).map(
|
||||
(descriptor) => descriptor.name,
|
||||
),
|
||||
);
|
||||
return getCommandGroupNames(entry).every((command) => descriptorNames.has(command));
|
||||
}
|
||||
|
||||
export async function registerPluginCliCommandGroups(
|
||||
program: Command,
|
||||
entries: readonly PluginCliCommandGroupEntry[],
|
||||
params: {
|
||||
mode: PluginCliCommandGroupMode;
|
||||
primary: string | null;
|
||||
existingCommands: Set<string>;
|
||||
logger: PluginLogger;
|
||||
},
|
||||
) {
|
||||
for (const entry of entries) {
|
||||
const registerEntry = async () => {
|
||||
await entry.register(program);
|
||||
for (const command of getCommandGroupNames(entry)) {
|
||||
params.existingCommands.add(command);
|
||||
}
|
||||
};
|
||||
|
||||
if (params.primary && findCommandGroupEntry([entry], params.primary)) {
|
||||
removeCommandGroupNames(program, entry);
|
||||
await registerEntry();
|
||||
continue;
|
||||
}
|
||||
|
||||
const overlaps = getCommandGroupNames(entry).filter((command) =>
|
||||
params.existingCommands.has(command),
|
||||
);
|
||||
if (overlaps.length > 0) {
|
||||
params.logger.debug?.(
|
||||
`plugin CLI register skipped (${entry.pluginId}): command already registered (${overlaps.join(
|
||||
", ",
|
||||
)})`,
|
||||
);
|
||||
continue;
|
||||
}
|
||||
|
||||
try {
|
||||
if (params.mode === "lazy" && canRegisterPluginCliLazily(entry)) {
|
||||
for (const placeholder of entry.placeholders) {
|
||||
registerLazyCommandGroup(program, entry, placeholder);
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
if (params.mode === "lazy" && entry.placeholders.length > 0) {
|
||||
params.logger.debug?.(
|
||||
`plugin CLI lazy register fallback to eager (${entry.pluginId}): descriptors do not cover all command roots`,
|
||||
);
|
||||
}
|
||||
await registerEntry();
|
||||
} catch (error) {
|
||||
params.logger.warn(`plugin CLI register failed (${entry.pluginId}): ${String(error)}`);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user