@eggjs/controller-decorator
Version:
tegg controller decorator
108 lines • 12.2 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.MCPInfoUtil = void 0;
const tegg_types_1 = require("@eggjs/tegg-types");
const core_decorator_1 = require("@eggjs/core-decorator");
class MCPInfoUtil {
static setMCPName(name, clazz) {
core_decorator_1.MetadataUtil.defineMetaData(tegg_types_1.CONTROLLER_MCP_NAME, name, clazz);
}
static getMCPName(clazz) {
return core_decorator_1.MetadataUtil.getMetaData(tegg_types_1.CONTROLLER_MCP_NAME, clazz);
}
static setMCPVersion(version, clazz) {
core_decorator_1.MetadataUtil.defineMetaData(tegg_types_1.CONTROLLER_MCP_VERSION, version, clazz);
}
static getMCPVersion(clazz) {
return core_decorator_1.MetadataUtil.getMetaData(tegg_types_1.CONTROLLER_MCP_VERSION, clazz);
}
static setMCPControllerParams(params, clazz) {
core_decorator_1.MetadataUtil.defineMetaData(tegg_types_1.CONTROLLER_MCP_CONTROLLER_PARAMS_MAP, params, clazz);
}
static getMCPControllerParams(clazz) {
return core_decorator_1.MetadataUtil.getMetaData(tegg_types_1.CONTROLLER_MCP_CONTROLLER_PARAMS_MAP, clazz);
}
static setMCPResource(clazz, methodName) {
const methodMap = core_decorator_1.MetadataUtil.initOwnMapMetaData(tegg_types_1.CONTROLLER_MCP_RESOURCE_MAP, clazz, new Map());
methodMap.set(methodName, true);
}
static getMCPResource(clazz) {
const methodMap = core_decorator_1.MetadataUtil.getMetaData(tegg_types_1.CONTROLLER_MCP_RESOURCE_MAP, clazz);
if (!methodMap) {
return [];
}
return Array.from(methodMap.keys());
}
static setMCPResourceParams(params, clazz, resourceName) {
const methodMap = core_decorator_1.MetadataUtil.initOwnMapMetaData(tegg_types_1.CONTROLLER_MCP_RESOURCE_PARAMS_MAP, clazz, new Map());
methodMap.set(resourceName, params);
}
static getMCPResourceParams(clazz, resourceName) {
const methodMap = core_decorator_1.MetadataUtil.getMetaData(tegg_types_1.CONTROLLER_MCP_RESOURCE_PARAMS_MAP, clazz);
return methodMap === null || methodMap === void 0 ? void 0 : methodMap.get(resourceName);
}
static setMCPTool(clazz, methodName) {
const methodMap = core_decorator_1.MetadataUtil.initOwnMapMetaData(tegg_types_1.CONTROLLER_MCP_TOOL_MAP, clazz, new Map());
methodMap.set(methodName, true);
}
static getMCPTool(clazz) {
const methodMap = core_decorator_1.MetadataUtil.getMetaData(tegg_types_1.CONTROLLER_MCP_TOOL_MAP, clazz);
if (!methodMap) {
return [];
}
return Array.from(methodMap.keys());
}
static getMCPToolParams(clazz, resourceName) {
const methodMap = core_decorator_1.MetadataUtil.getMetaData(tegg_types_1.CONTROLLER_MCP_TOOL_PARAMS_MAP, clazz);
return methodMap === null || methodMap === void 0 ? void 0 : methodMap.get(resourceName);
}
static setMCPToolParams(params, clazz, resourceName) {
const methodMap = core_decorator_1.MetadataUtil.initOwnMapMetaData(tegg_types_1.CONTROLLER_MCP_TOOL_PARAMS_MAP, clazz, new Map());
methodMap.set(resourceName, params);
}
static setMCPPrompt(clazz, methodName) {
const methodMap = core_decorator_1.MetadataUtil.initOwnMapMetaData(tegg_types_1.CONTROLLER_MCP_PROMPT_MAP, clazz, new Map());
methodMap.set(methodName, true);
}
static getMCPPrompt(clazz) {
const methodMap = core_decorator_1.MetadataUtil.getMetaData(tegg_types_1.CONTROLLER_MCP_PROMPT_MAP, clazz);
if (!methodMap) {
return [];
}
return Array.from(methodMap.keys());
}
static setMCPPromptParams(params, clazz, resourceName) {
const methodMap = core_decorator_1.MetadataUtil.initOwnMapMetaData(tegg_types_1.CONTROLLER_MCP_PROMPT_PARAMS_MAP, clazz, new Map());
methodMap.set(resourceName, params);
}
static getMCPPromptParams(clazz, resourceName) {
const methodMap = core_decorator_1.MetadataUtil.getMetaData(tegg_types_1.CONTROLLER_MCP_PROMPT_PARAMS_MAP, clazz);
return methodMap === null || methodMap === void 0 ? void 0 : methodMap.get(resourceName);
}
static setMCPToolArgsInArgs(detail, clazz, methodName) {
const methodContextIndexMap = core_decorator_1.MetadataUtil.initOwnMapMetaData(tegg_types_1.CONTROLLER_MCP_TOOL_ARGS_INDEX, clazz, new Map());
methodContextIndexMap.set(methodName, detail);
}
static getMCPToolArgsIndex(clazz, methodName) {
const methodContextIndexMap = core_decorator_1.MetadataUtil.getMetaData(tegg_types_1.CONTROLLER_MCP_TOOL_ARGS_INDEX, clazz);
return methodContextIndexMap === null || methodContextIndexMap === void 0 ? void 0 : methodContextIndexMap.get(methodName);
}
static setMCPExtra(index, clazz, methodName) {
const methodContextIndexMap = core_decorator_1.MetadataUtil.initOwnMapMetaData(tegg_types_1.CONTROLLER_MCP_EXTRA_INDEX, clazz, new Map());
methodContextIndexMap.set(methodName, index);
}
static getMCPExtra(clazz, methodName) {
const methodContextIndexMap = core_decorator_1.MetadataUtil.getMetaData(tegg_types_1.CONTROLLER_MCP_EXTRA_INDEX, clazz);
return methodContextIndexMap === null || methodContextIndexMap === void 0 ? void 0 : methodContextIndexMap.get(methodName);
}
static setMCPPromptArgsInArgs(detail, clazz, methodName) {
const methodContextIndexMap = core_decorator_1.MetadataUtil.initOwnMapMetaData(tegg_types_1.CONTROLLER_MCP_PROMPT_ARGS_INDEX, clazz, new Map());
methodContextIndexMap.set(methodName, detail);
}
static getMCPPromptArgsIndex(clazz, methodName) {
const methodContextIndexMap = core_decorator_1.MetadataUtil.getMetaData(tegg_types_1.CONTROLLER_MCP_PROMPT_ARGS_INDEX, clazz);
return methodContextIndexMap === null || methodContextIndexMap === void 0 ? void 0 : methodContextIndexMap.get(methodName);
}
}
exports.MCPInfoUtil = MCPInfoUtil;
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiTUNQSW5mb1V0aWwuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi9zcmMvdXRpbC9NQ1BJbmZvVXRpbC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7QUFBQSxrREFrQjJCO0FBQzNCLDBEQUFxRDtBQXNCckQsTUFBYSxXQUFXO0lBRXRCLE1BQU0sQ0FBQyxVQUFVLENBQUMsSUFBWSxFQUFFLEtBQXdCO1FBQ3RELDZCQUFZLENBQUMsY0FBYyxDQUFDLGdDQUFtQixFQUFFLElBQUksRUFBRSxLQUFLLENBQUMsQ0FBQztJQUNoRSxDQUFDO0lBRUQsTUFBTSxDQUFDLFVBQVUsQ0FBQyxLQUF3QjtRQUN4QyxPQUFPLDZCQUFZLENBQUMsV0FBVyxDQUFDLGdDQUFtQixFQUFFLEtBQUssQ0FBQyxDQUFDO0lBQzlELENBQUM7SUFFRCxNQUFNLENBQUMsYUFBYSxDQUFDLE9BQWUsRUFBRSxLQUF3QjtRQUM1RCw2QkFBWSxDQUFDLGNBQWMsQ0FBQyxtQ0FBc0IsRUFBRSxPQUFPLEVBQUUsS0FBSyxDQUFDLENBQUM7SUFDdEUsQ0FBQztJQUVELE1BQU0sQ0FBQyxhQUFhLENBQUMsS0FBd0I7UUFDM0MsT0FBTyw2QkFBWSxDQUFDLFdBQVcsQ0FBQyxtQ0FBc0IsRUFBRSxLQUFLLENBQUMsQ0FBQztJQUNqRSxDQUFDO0lBRUQsTUFBTSxDQUFDLHNCQUFzQixDQUFDLE1BQXVDLEVBQUUsS0FBd0I7UUFDN0YsNkJBQVksQ0FBQyxjQUFjLENBQUMsaURBQW9DLEVBQUUsTUFBTSxFQUFFLEtBQUssQ0FBQyxDQUFDO0lBQ25GLENBQUM7SUFFRCxNQUFNLENBQUMsc0JBQXNCLENBQUMsS0FBd0I7UUFDcEQsT0FBTyw2QkFBWSxDQUFDLFdBQVcsQ0FBQyxpREFBb0MsRUFBRSxLQUFLLENBQUMsQ0FBQztJQUMvRSxDQUFDO0lBRUQsTUFBTSxDQUFDLGNBQWMsQ0FBQyxLQUF3QixFQUFFLFVBQWtCO1FBQ2hFLE1BQU0sU0FBUyxHQUFpQiw2QkFBWSxDQUFDLGtCQUFrQixDQUFDLHdDQUEyQixFQUFFLEtBQUssRUFBRSxJQUFJLEdBQUcsRUFBRSxDQUFDLENBQUM7UUFDL0csU0FBUyxDQUFDLEdBQUcsQ0FBQyxVQUFVLEVBQUUsSUFBSSxDQUFDLENBQUM7SUFDbEMsQ0FBQztJQUVELE1BQU0sQ0FBQyxjQUFjLENBQUMsS0FBd0I7UUFDNUMsTUFBTSxTQUFTLEdBQTZCLDZCQUFZLENBQUMsV0FBVyxDQUFDLHdDQUEyQixFQUFFLEtBQUssQ0FBQyxDQUFDO1FBQ3pHLElBQUksQ0FBQyxTQUFTLEVBQUUsQ0FBQztZQUNmLE9BQU8sRUFBRSxDQUFDO1FBQ1osQ0FBQztRQUNELE9BQU8sS0FBSyxDQUFDLElBQUksQ0FBQyxTQUFTLENBQUMsSUFBSSxFQUFFLENBQUMsQ0FBQztJQUN0QyxDQUFDO0lBRUQsTUFBTSxDQUFDLG9CQUFvQixDQUFDLE1BQWdELEVBQUUsS0FBd0IsRUFBRSxZQUFvQjtRQUMxSCxNQUFNLFNBQVMsR0FBbUIsNkJBQVksQ0FBQyxrQkFBa0IsQ0FBQywrQ0FBa0MsRUFBRSxLQUFLLEVBQUUsSUFBSSxHQUFHLEVBQUUsQ0FBQyxDQUFDO1FBQ3hILFNBQVMsQ0FBQyxHQUFHLENBQUMsWUFBWSxFQUFFLE1BQU0sQ0FBQyxDQUFDO0lBQ3RDLENBQUM7SUFFRCxNQUFNLENBQUMsb0JBQW9CLENBQUMsS0FBd0IsRUFBRSxZQUFvQjtRQUN4RSxNQUFNLFNBQVMsR0FBK0IsNkJBQVksQ0FBQyxXQUFXLENBQUMsK0NBQWtDLEVBQUUsS0FBSyxDQUFDLENBQUM7UUFDbEgsT0FBTyxTQUFTLGFBQVQsU0FBUyx1QkFBVCxTQUFTLENBQUUsR0FBRyxDQUFDLFlBQVksQ0FBQyxDQUFDO0lBQ3RDLENBQUM7SUFFRCxNQUFNLENBQUMsVUFBVSxDQUFDLEtBQXdCLEVBQUUsVUFBa0I7UUFDNUQsTUFBTSxTQUFTLEdBQWlCLDZCQUFZLENBQUMsa0JBQWtCLENBQUMsb0NBQXVCLEVBQUUsS0FBSyxFQUFFLElBQUksR0FBRyxFQUFFLENBQUMsQ0FBQztRQUMzRyxTQUFTLENBQUMsR0FBRyxDQUFDLFVBQVUsRUFBRSxJQUFJLENBQUMsQ0FBQztJQUNsQyxDQUFDO0lBRUQsTUFBTSxDQUFDLFVBQVUsQ0FBQyxLQUF3QjtRQUN4QyxNQUFNLFNBQVMsR0FBNkIsNkJBQVksQ0FBQyxXQUFXLENBQUMsb0NBQXVCLEVBQUUsS0FBSyxDQUFDLENBQUM7UUFDckcsSUFBSSxDQUFDLFNBQVMsRUFBRSxDQUFDO1lBQ2YsT0FBTyxFQUFFLENBQUM7UUFDWixDQUFDO1FBQ0QsT0FBTyxLQUFLLENBQUMsSUFBSSxDQUFDLFNBQVMsQ0FBQyxJQUFJLEVBQUUsQ0FBQyxDQUFDO0lBQ3RDLENBQUM7SUFFRCxNQUFNLENBQUMsZ0JBQWdCLENBQUMsS0FBd0IsRUFBRSxZQUFvQjtRQUNwRSxNQUFNLFNBQVMsR0FBMkIsNkJBQVksQ0FBQyxXQUFXLENBQUMsMkNBQThCLEVBQUUsS0FBSyxDQUFDLENBQUM7UUFDMUcsT0FBTyxTQUFTLGFBQVQsU0FBUyx1QkFBVCxTQUFTLENBQUUsR0FBRyxDQUFDLFlBQVksQ0FBQyxDQUFDO0lBQ3RDLENBQUM7SUFFRCxNQUFNLENBQUMsZ0JBQWdCLENBQUMsTUFBNEMsRUFBRSxLQUF3QixFQUFFLFlBQW9CO1FBQ2xILE1BQU0sU0FBUyxHQUFlLDZCQUFZLENBQUMsa0JBQWtCLENBQUMsMkNBQThCLEVBQUUsS0FBSyxFQUFFLElBQUksR0FBRyxFQUFFLENBQUMsQ0FBQztRQUNoSCxTQUFTLENBQUMsR0FBRyxDQUFDLFlBQVksRUFBRSxNQUFNLENBQUMsQ0FBQztJQUN0QyxDQUFDO0lBRUQsTUFBTSxDQUFDLFlBQVksQ0FBQyxLQUF3QixFQUFFLFVBQWtCO1FBQzlELE1BQU0sU0FBUyxHQUFpQiw2QkFBWSxDQUFDLGtCQUFrQixDQUFDLHNDQUF5QixFQUFFLEtBQUssRUFBRSxJQUFJLEdBQUcsRUFBRSxDQUFDLENBQUM7UUFDN0csU0FBUyxDQUFDLEdBQUcsQ0FBQyxVQUFVLEVBQUUsSUFBSSxDQUFDLENBQUM7SUFDbEMsQ0FBQztJQUVELE1BQU0sQ0FBQyxZQUFZLENBQUMsS0FBd0I7UUFDMUMsTUFBTSxTQUFTLEdBQTZCLDZCQUFZLENBQUMsV0FBVyxDQUFDLHNDQUF5QixFQUFFLEtBQUssQ0FBQyxDQUFDO1FBQ3ZHLElBQUksQ0FBQyxTQUFTLEVBQUUsQ0FBQztZQUNmLE9BQU8sRUFBRSxDQUFDO1FBQ1osQ0FBQztRQUNELE9BQU8sS0FBSyxDQUFDLElBQUksQ0FBQyxTQUFTLENBQUMsSUFBSSxFQUFFLENBQUMsQ0FBQztJQUN0QyxDQUFDO0lBRUQsTUFBTSxDQUFDLGtCQUFrQixDQUFDLE1BQThDLEVBQUUsS0FBd0IsRUFBRSxZQUFvQjtRQUN0SCxNQUFNLFNBQVMsR0FBaUIsNkJBQVksQ0FBQyxrQkFBa0IsQ0FBQyw2Q0FBZ0MsRUFBRSxLQUFLLEVBQUUsSUFBSSxHQUFHLEVBQUUsQ0FBQyxDQUFDO1FBQ3BILFNBQVMsQ0FBQyxHQUFHLENBQUMsWUFBWSxFQUFFLE1BQU0sQ0FBQyxDQUFDO0lBQ3RDLENBQUM7SUFFRCxNQUFNLENBQUMsa0JBQWtCLENBQUMsS0FBd0IsRUFBRSxZQUFvQjtRQUN0RSxNQUFNLFNBQVMsR0FBNkIsNkJBQVksQ0FBQyxXQUFXLENBQUMsNkNBQWdDLEVBQUUsS0FBSyxDQUFDLENBQUM7UUFDOUcsT0FBTyxTQUFTLGFBQVQsU0FBUyx1QkFBVCxTQUFTLENBQUUsR0FBRyxDQUFDLFlBQVksQ0FBQyxDQUFDO0lBQ3RDLENBQUM7SUFFRCxNQUFNLENBQUMsb0JBQW9CLENBQUMsTUFBNEIsRUFBRSxLQUF3QixFQUFFLFVBQWtCO1FBQ3BHLE1BQU0scUJBQXFCLEdBQXlCLDZCQUFZLENBQUMsa0JBQWtCLENBQUMsMkNBQThCLEVBQUUsS0FBSyxFQUFFLElBQUksR0FBRyxFQUFFLENBQUMsQ0FBQztRQUN0SSxxQkFBcUIsQ0FBQyxHQUFHLENBQUMsVUFBVSxFQUFFLE1BQU0sQ0FBQyxDQUFDO0lBQ2hELENBQUM7SUFFRCxNQUFNLENBQUMsbUJBQW1CLENBQUMsS0FBd0IsRUFBRSxVQUFrQjtRQUNyRSxNQUFNLHFCQUFxQixHQUFxQyw2QkFBWSxDQUFDLFdBQVcsQ0FBQywyQ0FBOEIsRUFBRSxLQUFLLENBQUMsQ0FBQztRQUNoSSxPQUFPLHFCQUFxQixhQUFyQixxQkFBcUIsdUJBQXJCLHFCQUFxQixDQUFFLEdBQUcsQ0FBQyxVQUFVLENBQUMsQ0FBQztJQUNoRCxDQUFDO0lBRUQsTUFBTSxDQUFDLFdBQVcsQ0FBQyxLQUFhLEVBQUUsS0FBd0IsRUFBRSxVQUFrQjtRQUM1RSxNQUFNLHFCQUFxQixHQUFnQiw2QkFBWSxDQUFDLGtCQUFrQixDQUFDLHVDQUEwQixFQUFFLEtBQUssRUFBRSxJQUFJLEdBQUcsRUFBRSxDQUFDLENBQUM7UUFDekgscUJBQXFCLENBQUMsR0FBRyxDQUFDLFVBQVUsRUFBRSxLQUFLLENBQUMsQ0FBQztJQUMvQyxDQUFDO0lBRUQsTUFBTSxDQUFDLFdBQVcsQ0FBQyxLQUF3QixFQUFFLFVBQWtCO1FBQzdELE1BQU0scUJBQXFCLEdBQTRCLDZCQUFZLENBQUMsV0FBVyxDQUFDLHVDQUEwQixFQUFFLEtBQUssQ0FBQyxDQUFDO1FBQ25ILE9BQU8scUJBQXFCLGFBQXJCLHFCQUFxQix1QkFBckIscUJBQXFCLENBQUUsR0FBRyxDQUFDLFVBQVUsQ0FBQyxDQUFDO0lBQ2hELENBQUM7SUFFRCxNQUFNLENBQUMsc0JBQXNCLENBQUMsTUFBOEIsRUFBRSxLQUF3QixFQUFFLFVBQWtCO1FBQ3hHLE1BQU0scUJBQXFCLEdBQTJCLDZCQUFZLENBQUMsa0JBQWtCLENBQUMsNkNBQWdDLEVBQUUsS0FBSyxFQUFFLElBQUksR0FBRyxFQUFFLENBQUMsQ0FBQztRQUMxSSxxQkFBcUIsQ0FBQyxHQUFHLENBQUMsVUFBVSxFQUFFLE1BQU0sQ0FBQyxDQUFDO0lBQ2hELENBQUM7SUFFRCxNQUFNLENBQUMscUJBQXFCLENBQUMsS0FBd0IsRUFBRSxVQUFrQjtRQUN2RSxNQUFNLHFCQUFxQixHQUF1Qyw2QkFBWSxDQUFDLFdBQVcsQ0FBQyw2Q0FBZ0MsRUFBRSxLQUFLLENBQUMsQ0FBQztRQUNwSSxPQUFPLHFCQUFxQixhQUFyQixxQkFBcUIsdUJBQXJCLHFCQUFxQixDQUFFLEdBQUcsQ0FBQyxVQUFVLENBQUMsQ0FBQztJQUNoRCxDQUFDO0NBQ0Y7QUE1SEQsa0NBNEhDIn0=