33 lines
1.4 KiB
TypeScript
33 lines
1.4 KiB
TypeScript
import type { McpServer } from "@modelcontextprotocol/server";
|
|
import type { BinaryLayoutService } from "../application/binaryDiagnostics/BinaryLayoutService.js";
|
|
import type { EvidenceWriter } from "../application/investigation/InvestigationRecordPort.js";
|
|
import { toolContract } from "../contracts/toolContracts.js";
|
|
import type { Logger } from "../logger.js";
|
|
import { logToolExecution } from "./toolLogging.js";
|
|
import { toolRegistrationOptions } from "./toolRegistrationOptions.js";
|
|
import { toCallToolResult } from "./toolResult.js";
|
|
|
|
/** Bind offline layout inspection to its named contract and session Evidence owner. */
|
|
export const registerBinaryDiagnosticsTools = (
|
|
server: McpServer,
|
|
service: BinaryLayoutService,
|
|
logger: Logger,
|
|
recordEvidence?: EvidenceWriter["recordEvidence"],
|
|
): void => {
|
|
const contract = toolContract("inspect_binary_layout");
|
|
server.registerTool(
|
|
contract.name,
|
|
toolRegistrationOptions(contract),
|
|
async (input, context) => {
|
|
const result = await logToolExecution(logger, contract.name, () =>
|
|
service.inspect(input, { signal: context.mcpReq.signal }),
|
|
);
|
|
if (!result.ok) return toCallToolResult(result, contract);
|
|
const recorded = recordEvidence?.(result.value);
|
|
return recorded !== undefined && !recorded.ok
|
|
? toCallToolResult(recorded, contract)
|
|
: toCallToolResult(result, contract);
|
|
},
|
|
);
|
|
};
|