106 lines
3.0 KiB
TypeScript
106 lines
3.0 KiB
TypeScript
|
const MCP_TOOL_NAME_PREFIX = 'mcp__';
|
||
|
|
|
||
|
|
const KNOWN_ARTIFACT_TOOL_NAMES = [
|
||
|
|
'terminal_read_context',
|
||
|
|
'vault_notes_create',
|
||
|
|
'vault_notes_update',
|
||
|
|
'vault_notes_get',
|
||
|
|
'vault_notes_list',
|
||
|
|
'vault_hosts_create',
|
||
|
|
'vault_hosts_import',
|
||
|
|
'vault_hosts_list',
|
||
|
|
'host_get',
|
||
|
|
'snippets_list',
|
||
|
|
'snippets_get',
|
||
|
|
'snippets_create',
|
||
|
|
'snippets_update',
|
||
|
|
'snippets_delete',
|
||
|
|
'snippets_run',
|
||
|
|
'scripts_list',
|
||
|
|
'scripts_get',
|
||
|
|
'scripts_create',
|
||
|
|
'scripts_update',
|
||
|
|
'scripts_delete',
|
||
|
|
'scripts_run',
|
||
|
|
'scripts_reference',
|
||
|
|
'scripts_runs_list',
|
||
|
|
'scripts_run_stop',
|
||
|
|
'scripts_run_pause',
|
||
|
|
'scripts_run_resume',
|
||
|
|
'scripts_targets_set',
|
||
|
|
] as const;
|
||
|
|
|
||
|
|
const CLI_ARTIFACT_TOOL_NAMES = new Map<string, string>([
|
||
|
|
['vault host get', 'host_get'],
|
||
|
|
]);
|
||
|
|
|
||
|
|
function readCommandString(args: Record<string, unknown> | undefined): string | null {
|
||
|
|
if (!args) return null;
|
||
|
|
const raw = args.command;
|
||
|
|
if (typeof raw === 'string') return raw || null;
|
||
|
|
if (!Array.isArray(raw) || raw.length === 0) return null;
|
||
|
|
|
||
|
|
const isShellWrap =
|
||
|
|
raw.length >= 3 &&
|
||
|
|
/(?:^|\/)(sh|bash|zsh|fish|ash|dash)$/.test(String(raw[0] ?? '')) &&
|
||
|
|
/^-l?c$/.test(String(raw[1] ?? ''));
|
||
|
|
|
||
|
|
return isShellWrap
|
||
|
|
? String(raw[raw.length - 1] ?? '') || null
|
||
|
|
: raw.map((part) => String(part)).join(' ');
|
||
|
|
}
|
||
|
|
|
||
|
|
function unwrapShellCommand(command: string): string {
|
||
|
|
const strWrap = command.match(
|
||
|
|
/^(?:\S*\/)?(?:sh|bash|zsh|fish|ash|dash)\s+-l?c\s+(['"])([\s\S]*)\1\s*$/,
|
||
|
|
);
|
||
|
|
return strWrap ? strWrap[2] : command;
|
||
|
|
}
|
||
|
|
|
||
|
|
function stripWrappingQuote(value: string): string {
|
||
|
|
const trimmed = value.trim();
|
||
|
|
if (trimmed.length < 2) return trimmed;
|
||
|
|
const first = trimmed[0];
|
||
|
|
const last = trimmed[trimmed.length - 1];
|
||
|
|
return (first === last && (first === '"' || first === "'"))
|
||
|
|
? trimmed.slice(1, -1)
|
||
|
|
: trimmed;
|
||
|
|
}
|
||
|
|
|
||
|
|
export function normalizeArtifactToolName(toolName: string | undefined): string | undefined {
|
||
|
|
const trimmed = toolName?.trim();
|
||
|
|
if (!trimmed) return undefined;
|
||
|
|
|
||
|
|
if (trimmed.startsWith(MCP_TOOL_NAME_PREFIX)) {
|
||
|
|
const segments = trimmed.split('__').filter(Boolean);
|
||
|
|
return segments[segments.length - 1] || trimmed;
|
||
|
|
}
|
||
|
|
|
||
|
|
const prefixedArtifactToolName = KNOWN_ARTIFACT_TOOL_NAMES.find((candidate) => (
|
||
|
|
trimmed.endsWith(`_${candidate}`) || trimmed.endsWith(`-${candidate}`)
|
||
|
|
));
|
||
|
|
if (prefixedArtifactToolName) return prefixedArtifactToolName;
|
||
|
|
|
||
|
|
return trimmed;
|
||
|
|
}
|
||
|
|
|
||
|
|
export function inferArtifactToolNameFromCliArgs(
|
||
|
|
args: Record<string, unknown> | undefined,
|
||
|
|
): string | undefined {
|
||
|
|
const command = readCommandString(args);
|
||
|
|
if (!command) return undefined;
|
||
|
|
|
||
|
|
const unwrapped = unwrapShellCommand(command);
|
||
|
|
const cliMatch = unwrapped.match(/(?:^|\s|["'])(?:\S*\/)?netcatty-tool-cli(?:\.(?:cjs|cmd))?(?=["'\s]|$)([\s\S]*)$/);
|
||
|
|
if (!cliMatch) return undefined;
|
||
|
|
|
||
|
|
const afterCli = stripWrappingQuote(cliMatch[1] ?? '').replace(/^["']?\s*/, '');
|
||
|
|
const commandKey = afterCli
|
||
|
|
.split(/\s+/)
|
||
|
|
.filter((part) => part && !part.startsWith('-'))
|
||
|
|
.slice(0, 3)
|
||
|
|
.join(' ');
|
||
|
|
|
||
|
|
return CLI_ARTIFACT_TOOL_NAMES.get(commandKey);
|
||
|
|
}
|