package com.xly.tool;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.xly.agent.AgentIdentity;
import com.xly.service.ErpClient;
import com.xly.service.FormResolverService;
import com.xly.service.OpService;
import dev.langchain4j.agent.tool.P;
import dev.langchain4j.agent.tool.Tool;
import org.springframework.jdbc.core.JdbcTemplate;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
/**
* ProposeWrite 工具(写操作,人在环)。
*
*
**只提议并暂存,绝不立即执行**:把「改某条记录的某字段」解析成具体的 表/记录id/列/新值,
* 写一条 draft 到 ai_op_queue,返回一个提议。真正执行发生在用户点【确认】后的确定性端点里
* (见 OpController),不经过 LLM。
*/
public class ProposeWriteTool {
private final ErpClient erp;
private final JdbcTemplate jdbc;
private final OpService ops;
private final ObjectMapper mapper;
private final AgentIdentity identity;
private final FormResolverService resolver;
public ProposeWriteTool(ErpClient erp, JdbcTemplate jdbc, OpService ops, ObjectMapper mapper,
AgentIdentity identity, FormResolverService resolver) {
this.erp = erp;
this.jdbc = jdbc;
this.ops = ops;
this.mapper = mapper;
this.identity = identity;
this.resolver = resolver;
}
@Tool("提议修改某条现有记录的某个字段(写操作)。本工具**只提议并暂存、绝不立即执行**——"
+ "必须等用户在对话内点【确认】后才真正修改。用于「把某个客户/物料的某字段改成X」这类需求。"
+ "直接给出实体类型(如 客户)即可,本工具会自行定位主表,无需先 findForms。")
public String proposeUpdate(
@P("实体/单据类型关键词,如 客户 / 物料 / 供应商") String entityKeyword,
@P("要修改的那条记录的名称关键词(如某个客户名)") String recordKeyword,
@P("要修改的字段中文名(如 简称 / 备注 / 联系电话)") String fieldChinese,
@P("修改后的新值") String newValue) {
if (isBlank(entityKeyword) || isBlank(recordKeyword) || isBlank(fieldChinese)) {
return err("缺少信息:需要实体类型、记录名称关键词、字段中文名、新值。");
}
// 1) 定位可修改的主表(该实体名下、table 类型、最常用的一张)
Map form = resolveForm(entityKeyword.trim());
if (form == null) {
return err("找不到「" + entityKeyword + "」对应的可修改主表。");
}
String formId = str(form.get("sFormId"));
String moduleId = str(form.get("sModuleId"));
String table = str(form.get("sDataSource"));
if (!identity.canAccessModule(moduleId)) {
return err("你没有修改「" + entityKeyword + "」的权限。");
}
// 2) 字段中文名 -> 技术列名
String field = queryOne(
"SELECT sField FROM viw_kg_field_dict WHERE sTable=? AND sChinese=? ORDER BY iFormUses DESC LIMIT 1",
table, fieldChinese.trim());
if (field == null) {
field = queryOne(
"SELECT sField FROM viw_kg_field_dict WHERE sTable=? AND sChinese LIKE ? ORDER BY iFormUses DESC LIMIT 1",
table, "%" + fieldChinese.trim() + "%");
}
if (field == null) {
return err("在该表单里找不到叫「" + fieldChinese + "」的字段,请换个字段名或先查看该表单有哪些字段。");
}
// 3) 定位唯一记录 + 旧值(用名称字段过滤)
String nameField = queryOne(
"SELECT sField FROM viw_kg_field_dict WHERE sTable=? AND sField LIKE '%Name' AND sField NOT LIKE '%NameId' " +
"ORDER BY iFormUses DESC LIMIT 1", table);
JsonNode root;
try {
root = erp.readForm(identity.token(), formId.trim(), moduleId.trim(), 1, 5, nameField, recordKeyword.trim());
} catch (Exception e) {
return err("定位记录时读取失败:" + e.getMessage());
}
if (root.path("code").asInt(0) < 0) {
return err("定位记录失败:" + root.path("msg").asText("未知错误"));
}
JsonNode rows = root.path("dataset").path("rows");
JsonNode data = (rows.isArray() && rows.size() > 0) ? rows.get(0).path("dataSet") : null;
int n = (data != null && data.isArray()) ? data.size() : 0;
if (n == 0) {
return err("没有找到名称含「" + recordKeyword + "」的记录,无法修改。");
}
if (n > 1) {
StringBuilder names = new StringBuilder();
for (int i = 0; i < data.size() && i < 5; i++) {
if (i > 0) names.append("、");
names.append(data.get(i).path(nameField == null ? "" : nameField).asText(""));
}
return err("匹配到多条记录(" + names + "),请提供更精确的名称,只改其中一条。");
}
JsonNode rec = data.get(0);
String billId = rec.path("sId").asText(null);
if (isBlank(billId)) {
return err("定位到的记录缺少主键 sId,无法安全修改。");
}
String oldValue = rec.path(field).asText("");
String recordName = nameField == null ? recordKeyword : rec.path(nameField).asText(recordKeyword);
// 4) 暂存 draft(不执行)
String description = "将【" + recordName + "】的【" + fieldChinese + "】"
+ (oldValue.isBlank() ? "" : ("由「" + oldValue + "」")) + "改为「" + newValue + "」";
String opId = ops.createDraft(identity.userId(), "update", formId.trim(), moduleId.trim(), table, billId,
field, fieldChinese, oldValue, newValue, description);
Map out = new LinkedHashMap<>();
out.put("opId", opId);
out.put("summary", description);
out.put("message", "已为你生成一条待确认的修改,请在下方点【确认】执行、或【取消】。");
return toJson(out);
}
@Tool("提议**删除**某条现有记录(写操作)。本工具只提议并暂存、绝不立即执行——必须等用户点【确认】后才真正删除。"
+ "删除不可恢复,请慎用。直接给出实体类型与记录名即可。")
public String proposeDelete(
@P("实体类型,如 客户 / 物料 / 供应商") String entityKeyword,
@P("要删除的记录的名称关键词") String recordKeyword) {
if (isBlank(entityKeyword) || isBlank(recordKeyword)) {
return err("缺少实体类型或记录名称。");
}
Map form = resolveForm(entityKeyword.trim());
if (form == null) {
return err("找不到「" + entityKeyword + "」对应的可操作主表。");
}
String formId = str(form.get("sFormId"));
String moduleId = str(form.get("sModuleId"));
String table = str(form.get("sDataSource"));
if (!identity.canAccessModule(moduleId)) {
return err("你没有操作「" + entityKeyword + "」的权限。");
}
String nameField = queryOne(
"SELECT sField FROM viw_kg_field_dict WHERE sTable=? AND sField LIKE '%Name' AND sField NOT LIKE '%NameId' " +
"ORDER BY iFormUses DESC LIMIT 1", table);
JsonNode root;
try {
root = erp.readForm(identity.token(), formId, moduleId, 1, 5, nameField, recordKeyword.trim());
} catch (Exception e) {
return err("定位记录失败:" + e.getMessage());
}
if (root.path("code").asInt(0) < 0) {
return err("定位记录失败:" + root.path("msg").asText("未知错误"));
}
JsonNode rows = root.path("dataset").path("rows");
JsonNode data = (rows.isArray() && rows.size() > 0) ? rows.get(0).path("dataSet") : null;
int n = (data != null && data.isArray()) ? data.size() : 0;
if (n == 0) {
return err("没有找到名称含「" + recordKeyword + "」的记录。");
}
if (n > 1) {
StringBuilder names = new StringBuilder();
for (int i = 0; i < data.size() && i < 5; i++) {
if (i > 0) names.append("、");
names.append(nameField == null ? "" : data.get(i).path(nameField).asText(""));
}
return err("匹配到多条记录(" + names + "),请提供更精确的名称,只删其中一条。");
}
JsonNode rec = data.get(0);
String billId = rec.path("sId").asText(null);
if (isBlank(billId)) {
return err("定位到的记录缺少主键 sId,无法安全删除。");
}
String recordName = nameField == null ? recordKeyword : rec.path(nameField).asText(recordKeyword);
String description = "删除【" + recordName + "】(" + entityKeyword + ")";
String opId = ops.createDraft(identity.userId(), "delete", formId, moduleId, table, billId,
null, null, recordName, null, description);
Map out = new LinkedHashMap<>();
out.put("opId", opId);
out.put("summary", description);
out.put("message", "已为你生成一条待确认的删除,请在下方点【确认】执行、或【取消】。删除不可恢复,请谨慎。");
return toJson(out);
}
@Tool("提议**新增**一条记录(写操作)。只提议并暂存、绝不立即执行——用户点确认后才真正新增。"
+ "给出实体类型 + 已知字段(JSON:字段中文名->值);主键与必填字段会自动补齐。")
public String proposeCreate(
@P("实体类型,如 客户 / 物料") String entityKeyword,
@P("已知字段的 JSON,键=字段中文名、值=字段值,例如 {\"客户名称\":\"常州测试公司\",\"客户简称\":\"常测\"}") String fieldsJson) {
if (isBlank(entityKeyword)) {
return err("缺少实体类型。");
}
Map form = resolveForm(entityKeyword.trim());
if (form == null) {
return err("找不到「" + entityKeyword + "」对应的可新增主表。");
}
String formId = str(form.get("sFormId"));
String moduleId = str(form.get("sModuleId"));
String table = str(form.get("sDataSource"));
if (!identity.canAccessModule(moduleId)) {
return err("你没有新增「" + entityKeyword + "」的权限。");
}
// 报价是跨表主-从单据 → 走多表创建(主表 + 印刷/部件从表 + 多数量)
if ("quoquotationmaster".equalsIgnoreCase(table)) {
return proposeQuote(fieldsJson, formId, moduleId, table);
}
Map types = resolver.columnTypes(table);
// 权威 label -> {col, fk} 映射(与 collectForm 同源:策展/字段字典),保证前端标签能被正确回映射
Map> labelMap = new LinkedHashMap<>();
for (Map bfm : resolver.businessFields(table, 40)) {
labelMap.put(normLabel(String.valueOf(bfm.get("label"))), bfm);
}
Map col = new LinkedHashMap<>();
List descParts = new ArrayList<>();
try {
if (!isBlank(fieldsJson)) {
JsonNode fj = mapper.readTree(fieldsJson.trim());
Iterator> it = fj.fields();
while (it.hasNext()) {
Map.Entry e = it.next();
String zh = e.getKey();
String v = e.getValue().asText("");
if (isBlank(v)) {
continue;
}
Map fm = labelMap.get(normLabel(zh));
if (fm == null) {
fm = resolveColumn(table, zh); // 回退字段字典
}
if (fm == null) {
continue; // 该实体没有这个字段,忽略
}
String colName = str(fm.get("col"));
// 别让 LLM/用户直填系统列(制单人/单据日期/租户/单号…),交给 ERP 与下方系统列处理
if (resolver.isSystemColumn(colName)) {
continue;
}
String fk = str(fm.get("fk"));
if (fk != null && !fk.isBlank()) {
String id = resolver.resolveFk(fk, v); // 外键:名称 -> id
if (id == null) {
String ent = zh.replace("名称", "");
return err("「" + v + "」不是系统里已有的" + ent + "。请让用户从下拉里选真实存在的"
+ ent + ";**不要**因此新建" + ent + "。");
}
col.put(colName, id);
} else {
Object cv = resolver.coerce(types.get(colName), v); // 按列类型强转
if (cv != null) {
col.put(colName, cv);
}
}
descParts.add(zh + "=" + v);
}
}
} catch (Exception ex) {
return err("字段解析失败:" + ex.getMessage());
}
if (descParts.isEmpty()) {
return err("请至少提供一个有效字段(如客户名称)。");
}
// 自动补齐 NOT-NULL 无默认列(系统列跳过;其余按类型给默认 0/'')
for (String rc : requiredCols(table)) {
if (col.containsKey(rc) || resolver.isSystemColumn(rc)) {
continue;
}
col.put(rc, resolver.typeDefault(rc, types.get(rc)));
}
// 系统列:主键 + 表单id + 单号(租户/制单人/日期由 ERP 注入,不填)
col.put("sId", erp.newUuid(identity.token()));
col.put("sFormId", formId);
String billNo = resolver.nextBillNo(table, identity.brandsId());
if (billNo != null) {
col.put("sBillNo", billNo);
}
String payload;
try {
payload = mapper.writeValueAsString(col);
} catch (Exception e) {
return err("内部错误:" + e.getMessage());
}
String description = "新增【" + entityKeyword + "】:" + String.join(",", descParts);
String opId = ops.createDraftPayload(identity.userId(), "create", formId, moduleId, table, payload, description);
Map out = new LinkedHashMap<>();
out.put("opId", opId);
out.put("summary", description);
out.put("message", "已为你生成一条待确认的新增,请在下方点【确认】执行、或【取消】。");
return toJson(out);
}
@Tool("提议**审核/过账**某条单据(写操作)。只提议并暂存、绝不立即执行——用户点确认后才真正审核。"
+ "用于「审核 / 过账某张单据」这类需求;审核有业务后果,请谨慎。给出单据类型与单号/名称即可。")
public String proposeExamine(
@P("单据类型,如 销售订单 / 采购订单 / 报价") String entityKeyword,
@P("要审核的单据编号或名称关键词") String recordKeyword) {
if (isBlank(entityKeyword) || isBlank(recordKeyword)) {
return err("缺少单据类型或单号/名称。");
}
Map form = resolveForm(entityKeyword.trim());
if (form == null) {
return err("找不到「" + entityKeyword + "」对应的可审核单据主表。");
}
String formId = str(form.get("sFormId"));
String moduleId = str(form.get("sModuleId"));
String table = str(form.get("sDataSource"));
if (!identity.canAccessModule(moduleId)) {
return err("你没有审核「" + entityKeyword + "」的权限。");
}
String nameField = queryOne(
"SELECT sField FROM viw_kg_field_dict WHERE sTable=? AND sField LIKE '%Name' AND sField NOT LIKE '%NameId' " +
"ORDER BY iFormUses DESC LIMIT 1", table);
JsonNode root;
try {
root = erp.readForm(identity.token(), formId, moduleId, 1, 5, nameField, recordKeyword.trim());
} catch (Exception e) {
return err("定位单据失败:" + e.getMessage());
}
if (root.path("code").asInt(0) < 0) {
return err("定位单据失败:" + root.path("msg").asText("未知错误"));
}
JsonNode rows = root.path("dataset").path("rows");
JsonNode data = (rows.isArray() && rows.size() > 0) ? rows.get(0).path("dataSet") : null;
int n = (data != null && data.isArray()) ? data.size() : 0;
if (n == 0) {
return err("没有找到含「" + recordKeyword + "」的单据。");
}
if (n > 1) {
StringBuilder names = new StringBuilder();
for (int i = 0; i < data.size() && i < 5; i++) {
if (i > 0) names.append("、");
names.append(nameField == null ? "" : data.get(i).path(nameField).asText(""));
}
return err("匹配到多条单据(" + names + "),请提供更精确的单号,只审核其中一条。");
}
JsonNode rec = data.get(0);
String billId = rec.path("sId").asText(null);
if (isBlank(billId)) {
return err("定位到的单据缺少主键 sId,无法审核。");
}
String recordName = nameField == null ? recordKeyword : rec.path(nameField).asText(recordKeyword);
String description = "审核【" + recordName + "】(" + entityKeyword + ")";
// examine:sNewValue 存 iFlag(1=审核);执行走 ERP doExamine(存储过程驱动)
String opId = ops.createDraft(identity.userId(), "examine", formId, moduleId, table, billId,
null, null, null, "1", description);
Map out = new LinkedHashMap<>();
out.put("opId", opId);
out.put("summary", description);
out.put("message", "已为你生成一条待确认的审核,请在下方点【确认】执行、或【取消】。审核有业务后果,请谨慎。");
return toJson(out);
}
/**
* 报价多表创建:主表(quoquotationmaster) + 印刷/部件从表(quoquotationslave) + 多数量(quoquotationmanyqtys)。
* 字段按策展的 target 表分流;印刷/颜色/单双面无独立列 → 合进从表 sMaterialsMemo;多数量按逗号拆多行。
* 价格由 ERP【核价】计算,本工具只落主-从明细。生成 __tables__ 结构化 payload,确认端点做多表写入。
*/
private String proposeQuote(String fieldsJson, String formId, String moduleId, String table) {
Map masterTypes = resolver.columnTypes(table);
Map slaveTypes = resolver.columnTypes("quoquotationslave");
Map> labelMap = new LinkedHashMap<>();
for (Map bfm : resolver.businessFields(table, 40)) {
labelMap.put(normLabel(String.valueOf(bfm.get("label"))), bfm);
}
Map masterCol = new LinkedHashMap<>();
Map slaveCol = new LinkedHashMap<>();
List notes = new ArrayList<>();
List manyQtys = new ArrayList<>();
List descParts = new ArrayList<>();
String custId = null;
try {
if (!isBlank(fieldsJson)) {
JsonNode fj = mapper.readTree(fieldsJson.trim());
Iterator> it = fj.fields();
while (it.hasNext()) {
Map.Entry e = it.next();
String zh = e.getKey();
String v = e.getValue().asText("");
if (isBlank(v)) {
continue;
}
Map fm = labelMap.get(normLabel(zh));
if (fm == null) {
continue;
}
String colName = str(fm.get("col"));
String tgt = fm.get("table") == null ? "master" : str(fm.get("table"));
String fk = str(fm.get("fk"));
if ("note".equals(tgt)) {
notes.add(zh + "=" + v);
descParts.add(zh + "=" + v);
continue;
}
if ("manyqtys".equals(tgt)) {
for (String q : v.split("[,,、\\s]+")) {
String n = q.replaceAll("[^0-9.]", "");
if (!n.isEmpty()) {
manyQtys.add(n);
}
}
descParts.add(zh + "=" + v);
continue;
}
Object value;
if (fk != null && !fk.isBlank()) {
String id = resolver.resolveFk(fk, v);
if (id == null) {
String ent = zh.replace("名称", "");
return err("「" + v + "」不是系统里已有的" + ent + "。请用 collectForm 让用户从下拉里选真实存在的"
+ ent + ";**不要**因此新建" + ent + ",也不要把产品/规格当成客户。");
}
value = id;
if ("sCustomerId".equals(colName)) {
custId = id;
}
} else {
String dt = "slave".equals(tgt) ? slaveTypes.get(colName) : masterTypes.get(colName);
value = resolver.coerce(dt, v);
}
if (value != null) {
if ("slave".equals(tgt)) {
slaveCol.put(colName, value);
} else {
masterCol.put(colName, value);
}
}
descParts.add(zh + "=" + v);
}
}
} catch (Exception ex) {
return err("字段解析失败:" + ex.getMessage());
}
if (descParts.isEmpty()) {
return err("请至少填写一个字段(如客户名称/产品名称/数量)。");
}
// 主表:补必填(非系统) + 主键 + 表单id + 单号
for (String rc : requiredCols(table)) {
if (masterCol.containsKey(rc) || resolver.isSystemColumn(rc)) {
continue;
}
masterCol.put(rc, resolver.typeDefault(rc, masterTypes.get(rc)));
}
String masterId = erp.newUuid(identity.token());
masterCol.put("sId", masterId);
masterCol.put("sFormId", formId);
String billNo = resolver.nextBillNo(table, identity.brandsId());
if (billNo != null) {
masterCol.put("sBillNo", billNo);
}
List