refactor: centralize plugin tool policy helpers
This commit is contained in:
@@ -85,6 +85,16 @@ export function normalizeToolList(list?: string[]) {
|
||||
return list.map(normalizeToolName).filter(Boolean);
|
||||
}
|
||||
|
||||
export type ToolPolicyLike = {
|
||||
allow?: string[];
|
||||
deny?: string[];
|
||||
};
|
||||
|
||||
export type PluginToolGroups = {
|
||||
all: string[];
|
||||
byPlugin: Map<string, string[]>;
|
||||
};
|
||||
|
||||
export function expandToolGroups(list?: string[]) {
|
||||
const normalized = normalizeToolList(list);
|
||||
const expanded: string[] = [];
|
||||
@@ -99,6 +109,77 @@ export function expandToolGroups(list?: string[]) {
|
||||
return Array.from(new Set(expanded));
|
||||
}
|
||||
|
||||
export function collectExplicitAllowlist(
|
||||
policies: Array<ToolPolicyLike | undefined>,
|
||||
): string[] {
|
||||
const entries: string[] = [];
|
||||
for (const policy of policies) {
|
||||
if (!policy?.allow) continue;
|
||||
for (const value of policy.allow) {
|
||||
if (typeof value !== "string") continue;
|
||||
const trimmed = value.trim();
|
||||
if (trimmed) entries.push(trimmed);
|
||||
}
|
||||
}
|
||||
return entries;
|
||||
}
|
||||
|
||||
export function buildPluginToolGroups(params: {
|
||||
tools: Array<{ name: string }>;
|
||||
toolMeta: (tool: { name: string }) => { pluginId: string } | undefined;
|
||||
}): PluginToolGroups {
|
||||
const all: string[] = [];
|
||||
const byPlugin = new Map<string, string[]>();
|
||||
for (const tool of params.tools) {
|
||||
const meta = params.toolMeta(tool);
|
||||
if (!meta) continue;
|
||||
const name = normalizeToolName(tool.name);
|
||||
all.push(name);
|
||||
const pluginId = meta.pluginId.toLowerCase();
|
||||
const list = byPlugin.get(pluginId) ?? [];
|
||||
list.push(name);
|
||||
byPlugin.set(pluginId, list);
|
||||
}
|
||||
return { all, byPlugin };
|
||||
}
|
||||
|
||||
export function expandPluginGroups(
|
||||
list: string[] | undefined,
|
||||
groups: PluginToolGroups,
|
||||
): string[] | undefined {
|
||||
if (!list || list.length === 0) return list;
|
||||
const expanded: string[] = [];
|
||||
for (const entry of list) {
|
||||
const normalized = normalizeToolName(entry);
|
||||
if (normalized === "group:plugins") {
|
||||
if (groups.all.length > 0) {
|
||||
expanded.push(...groups.all);
|
||||
} else {
|
||||
expanded.push(normalized);
|
||||
}
|
||||
continue;
|
||||
}
|
||||
const tools = groups.byPlugin.get(normalized);
|
||||
if (tools && tools.length > 0) {
|
||||
expanded.push(...tools);
|
||||
continue;
|
||||
}
|
||||
expanded.push(normalized);
|
||||
}
|
||||
return Array.from(new Set(expanded));
|
||||
}
|
||||
|
||||
export function expandPolicyWithPluginGroups(
|
||||
policy: ToolPolicyLike | undefined,
|
||||
groups: PluginToolGroups,
|
||||
): ToolPolicyLike | undefined {
|
||||
if (!policy) return undefined;
|
||||
return {
|
||||
allow: expandPluginGroups(policy.allow, groups),
|
||||
deny: expandPluginGroups(policy.deny, groups),
|
||||
};
|
||||
}
|
||||
|
||||
export function resolveToolProfilePolicy(profile?: string): ToolProfilePolicy | undefined {
|
||||
if (!profile) return undefined;
|
||||
const resolved = TOOL_PROFILES[profile as ToolProfileId];
|
||||
|
||||
Reference in New Issue
Block a user