package com.zy.ai.mcp.config;
|
|
import com.alibaba.fastjson.JSONObject;
|
import com.zy.ai.mcp.dto.McpToolHandler;
|
import com.zy.ai.mcp.dto.ToolDefinition;
|
import com.zy.ai.mcp.dto.ToolRegistry;
|
import com.zy.ai.mcp.service.WcsDataFacade;
|
|
import java.util.*;
|
|
public class McpToolsBootstrap {
|
|
public static void registerAll(ToolRegistry registry, final WcsDataFacade facade) {
|
|
registry.register(tool(
|
"device.get_crn_status",
|
"Query realtime status of a crn device by deviceNo.",
|
schemaObj(
|
propInt("crnNos", true)
|
),
|
schemaObj(
|
propObj("devices", true)
|
),
|
new McpToolHandler() {
|
public Object handle(JSONObject args) {
|
return facade.getCrnDeviceStatus(args);
|
}
|
}
|
));
|
|
registry.register(tool(
|
"device.get_station_status",
|
"Query realtime status of a station device",
|
schemaObj(
|
|
),
|
schemaObj(
|
propObj("stations", true)
|
),
|
new McpToolHandler() {
|
public Object handle(JSONObject args) {
|
return facade.getStationDeviceStatus(args);
|
}
|
}
|
));
|
|
registry.register(tool(
|
"device.get_rgv_status",
|
"Query realtime status of a rgv device by deviceNo.",
|
schemaObj(
|
propInt("rgvNos", true)
|
),
|
schemaObj(
|
propObj("devices", true)
|
),
|
new McpToolHandler() {
|
public Object handle(JSONObject args) {
|
return facade.getRgvDeviceStatus(args);
|
}
|
}
|
));
|
|
registry.register(tool(
|
"task.list",
|
"List tasks by filters (status/CrnDevice/RgvDevice//time window).",
|
schemaObj(
|
propInt("crnNo", false),
|
propInt("rgvNo", false),
|
propArr("taskNos", false, "integer"),
|
propInt("limit", false)
|
),
|
schemaObj(propArr("tasks", true, "object")),
|
new McpToolHandler() {
|
public Object handle(JSONObject args) {
|
return facade.getTasks(args);
|
}
|
}
|
));
|
|
registry.register(tool(
|
"log.query",
|
"Query logs by keyword/level/time window/device/task. Return clipped log lines.",
|
schemaObj(
|
propInt("limit", false)
|
),
|
schemaObj(propArr("logs", true, "object")),
|
new McpToolHandler() {
|
public Object handle(JSONObject args) {
|
return facade.getLogs(args);
|
}
|
}
|
));
|
|
registry.register(tool(
|
"config.get_device_config",
|
"Get device config by deviceCode.",
|
schemaObj(
|
propArr("crnNos", false, "integer"),
|
propArr("rgvNos", false, "integer"),
|
propArr("devpNos", false, "integer")
|
),
|
schemaObj(propObj("deviceConfigs", true)),
|
new McpToolHandler() {
|
public Object handle(JSONObject args) {
|
return facade.getDeviceConfig(args);
|
}
|
}
|
));
|
|
registry.register(tool(
|
"config.get_system_config",
|
"Get key system configs for diagnosis.",
|
schemaObj(
|
|
),
|
schemaObj(
|
propObj("systemConfigs", true)
|
),
|
new McpToolHandler() {
|
public Object handle(JSONObject args) {
|
return facade.getSystemConfig(args);
|
}
|
}
|
));
|
|
// // ★ 诊断聚合快照：一次拿全
|
// registry.register(tool(
|
// "build_diagnosis_snapshot",
|
// "Aggregate diagnosis snapshot: tasks + device realtime + configs + clipped logs for diagnosis.",
|
// schemaObj(
|
// propStr("warehouseCode", true),
|
// propArr("deviceCodes", false, "string"), // 不传则按任务涉及设备推导
|
// propStr("taskNo", false),
|
// propStr("fromTime", true),
|
// propStr("toTime", true),
|
// propInt("taskLimit", false),
|
// propInt("logLimit", false),
|
// propArr("logKeywords", false, "string"),
|
// propBool("includeSystemConfig", false),
|
// propBool("includeDeviceConfig", false)
|
// ),
|
// schemaObj(
|
// propObj("snapshot", true),
|
// propArr("hints", false, "string")
|
// ),
|
// new McpToolHandler() {
|
// public Object handle(JSONObject args) {
|
// return facade.buildDiagnosisSnapshot(args);
|
// }
|
// }
|
// ));
|
}
|
|
// ---------- schema helpers ----------
|
private static ToolDefinition tool(String name, String desc,
|
Map<String, Object> in, Map<String, Object> out,
|
McpToolHandler handler) {
|
ToolDefinition d = new ToolDefinition();
|
d.setName(name);
|
d.setDescription(desc);
|
d.setInputSchema(in);
|
d.setOutputSchema(out);
|
d.setHandler(handler);
|
return d;
|
}
|
|
private static Map<String, Object> schemaObj(Object... props) {
|
Map<String, Object> m = new LinkedHashMap<String, Object>();
|
m.put("type", "object");
|
|
Map<String, Object> properties = new LinkedHashMap<String, Object>();
|
List<String> required = new ArrayList<String>();
|
|
for (Object p : props) {
|
@SuppressWarnings("unchecked")
|
Map<String, Object> pm = (Map<String, Object>) p;
|
String name = String.valueOf(pm.get("name"));
|
boolean req = Boolean.TRUE.equals(pm.get("required"));
|
pm.remove("name");
|
pm.remove("required");
|
properties.put(name, pm);
|
if (req) required.add(name);
|
}
|
|
m.put("properties", properties);
|
if (!required.isEmpty()) m.put("required", required);
|
return m;
|
}
|
|
private static Map<String, Object> propStr(String name, boolean required) {
|
Map<String, Object> m = new LinkedHashMap<String, Object>();
|
m.put("name", name);
|
m.put("required", required);
|
m.put("type", "string");
|
return m;
|
}
|
|
private static Map<String, Object> propInt(String name, boolean required) {
|
Map<String, Object> m = new LinkedHashMap<String, Object>();
|
m.put("name", name);
|
m.put("required", required);
|
m.put("type", "integer");
|
return m;
|
}
|
|
private static Map<String, Object> propBool(String name, boolean required) {
|
Map<String, Object> m = new LinkedHashMap<String, Object>();
|
m.put("name", name);
|
m.put("required", required);
|
m.put("type", "boolean");
|
return m;
|
}
|
|
private static Map<String, Object> propObj(String name, boolean required) {
|
Map<String, Object> m = new LinkedHashMap<String, Object>();
|
m.put("name", name);
|
m.put("required", required);
|
m.put("type", "object");
|
return m;
|
}
|
|
private static Map<String, Object> propArr(String name, boolean required, String itemType) {
|
Map<String, Object> m = new LinkedHashMap<String, Object>();
|
m.put("name", name);
|
m.put("required", required);
|
m.put("type", "array");
|
Map<String, Object> items = new LinkedHashMap<String, Object>();
|
items.put("type", itemType);
|
m.put("items", items);
|
return m;
|
}
|
}
|
