dsh-code-ui 1.4.0 → 1.5.0

package/lib/index.js

@@ -664,15 +664,22 @@ function rpcHandler(ctx) {
       }
 
       if (endpoint === "add-reference") {
+        const path = args.path || "";
+        const kind = args.kind || "cite";
+        const lineStart = args.lineStart || null;
+        const lineEnd = args.lineEnd || null;
+        // 允许重复引用：chip 在输入框句中可视化，重复了用户看得见、可 ✕ 删除，无需 host 去重拦截。
+        // （原逻辑曾对同 path+行区间+kind 的未使用引用去重，行内 chip 落地后按用户要求取消）
         const id = "r" + (idSeq++);
         const ref = {
           id,
-          path: args.path || "",
+          path,
           language: args.language || "",
-          lineStart: args.lineStart || null,
-          lineEnd: args.lineEnd || null,
+          lineStart,
+          lineEnd,
           text: args.text || "",
-          kind: args.kind || "cite",
+          kind,
+          used: false, // 一次性消耗标记：随消息发送后由 use-references 置 true（chip 清空，记录保留）
           createdAt: now(),
         };
         references.push(ref);
@@ -683,6 +690,17 @@ function rpcHandler(ctx) {
         return ok({ references });
       }
 
+      if (endpoint === "use-references") {
+        // 引用一次性消耗：把本条消息附带的引用标记为 used（不删除——AI 在消息发出后才经
+        // get_referenced_snippets 读取，删除就读不到了；used 记录保留供用户在【引用/备注】查找）。
+        const ids = Array.isArray(args.ids) ? args.ids : [];
+        let n = 0;
+        for (const r of references) {
+          if (!r.used && ids.indexOf(r.id) >= 0) { r.used = true; r.usedAt = now(); n++; }
+        }
+        return ok({ ok: true, used: n });
+      }
+
       if (endpoint === "delete-reference") {
         const i = references.findIndex((r) => r.id === args.id);
         if (i >= 0) references.splice(i, 1);
@@ -911,7 +929,8 @@ function registerSnippetsTool(tools) {
             const whole = !r.lineStart;
             const label = r.kind === "translate" ? "翻译请求" : whole ? "文件引用" : "代码引用";
             const loc = (r.path || "") + (whole ? "（整个文件）" : ":" + r.lineStart + "-" + r.lineEnd);
-            const lang = r.language ? " [" + r.language + "]" : "";
+            const num = r.seq ? "，用户 tag 编号 " + r.seq : "";
+            const lang = (r.language ? " [" + r.language + "]" : "") + (r.used ? "（已使用" + (r.usedAt ? " " + r.usedAt : "") + num + "）" : "");
             const body = whole
               ? "整个文件的引用：请用你的文件读取工具打开上述绝对路径自行读取。"
               : (r.text || "");