@aipanel/dsh-plugin 1.2.4 → 1.2.5

package/dist/index.js

@@ -15,6 +15,25 @@ var JS_EXTENSIONS = /* @__PURE__ */ new Set([
   ".vue"
 ]);
 var MUTATING_TOOLS = /* @__PURE__ */ new Set(["write", "edit", "apply_patch"]);
+var DEFAULT_CONTEXT_API_PATH = "/__aipanel_context__";
+function collectNodeIds(text) {
+  const ids = [];
+  const re = new RegExp("@\u8282\u70B9\\[(n[0-9a-z]+)\\]", "g");
+  let m;
+  while ((m = re.exec(text)) !== null) ids.push(m[1]);
+  return ids;
+}
+function buildNodeContext(e) {
+  const lines = [`\u8282\u70B9 ID\uFF1A${e.id ?? ""}`];
+  if (e.filePath) lines.push(`\u6E90\u7801\u6587\u4EF6\u8DEF\u5F84\uFF1A${e.filePath}${e.line ? `:${e.line}` : ""}`);
+  if (e.line) lines.push(`\u4EE3\u7801\u6240\u5728\u884C\u53F7\uFF1A${e.line}`);
+  if (e.column) lines.push(`\u4EE3\u7801\u6240\u5728\u5217\u53F7\uFF1A${e.column}`);
+  if (e.description) lines.push(`DOM \u5143\u7D20\u9009\u62E9\u5668\uFF1A${e.description}`);
+  if (e.innerText) lines.push(`DOM \u5143\u7D20\u5185\u90E8\u6587\u672C\uFF1A${e.innerText.slice(0, 200)}`);
+  if (e.previewPageUrl) lines.push(`\u7528\u6237\u9009\u4E2D\u8282\u70B9\u65F6\u7684\u9875\u9762 URL\uFF1A${e.previewPageUrl}`);
+  if (e.previewPageTitle) lines.push(`\u9875\u9762\u6807\u9898\uFF1A${e.previewPageTitle}`);
+  return lines.join("\n");
+}
 var STDOUT_MAX_BYTES = 2e5;
 var STDOUT_SPILL_MAX_BYTES = 2e6;
 var STDERR_MAX_BYTES = 1e5;
@@ -22,6 +41,8 @@ var GRACE_MS = 3e4;
 function apply(ctx, config = {}) {
   const cwd = config.cwd ?? process.cwd();
   const autoDiagnose = config.autoDiagnose ?? true;
+  const vitePort = config.vitePort ?? 0;
+  const contextApiPath = config.contextApiPath ?? DEFAULT_CONTEXT_API_PATH;
   const tools = ctx.tools;
   const subprocess = ctx.subprocess;
   async function collectOutput(argv, signal) {
@@ -107,6 +128,60 @@ ${diag}`
       return { kind: "accept", additionalContexts: [message] };
     }
   );
+  if (vitePort > 0) {
+    const contextBase = `http://127.0.0.1:${vitePort}${contextApiPath}`;
+    ctx.on(
+      "agent/pre-step",
+      async ({ signal }, next) => {
+        const decision = await next();
+        if (decision.kind === "reject") return decision;
+        const ids = /* @__PURE__ */ new Set();
+        for (const message of decision.messages) {
+          if (message.source.kind !== "user") continue;
+          for (const block of message.content) {
+            if (block.type !== "text") continue;
+            for (const id of collectNodeIds(block.text)) ids.add(id);
+          }
+        }
+        if (ids.size === 0) return decision;
+        let elements = [];
+        try {
+          const res = await fetch(contextBase, { signal });
+          if (res.ok) {
+            const pc = await res.json();
+            elements = pc.selectedElements ?? [];
+          }
+        } catch {
+        }
+        const byId = /* @__PURE__ */ new Map();
+        for (const el of elements) {
+          if (el.id) byId.set(el.id, el);
+        }
+        const injected = [...ids].map((id) => byId.get(id)).filter((el) => el !== void 0);
+        if (injected.length === 0) return decision;
+        const contextText = injected.map(buildNodeContext).join("\n\n---\n\n");
+        const contextMessage = {
+          role: "user",
+          id: randomUUID(),
+          content: [
+            {
+              type: "text",
+              text: `\u4EE5\u4E0B\u662F\u7528\u6237\u5F15\u7528\u8282\u70B9\u7684\u5B8C\u6574\u4E0A\u4E0B\u6587\uFF08\u8282\u70B9 ID \u4E0E\u6D88\u606F\u4E2D\u7684 @\u8282\u70B9[id] \u6807\u8BB0\u5BF9\u5E94\uFF09\uFF1A
+
+${contextText}`
+            }
+          ],
+          source: { kind: "plugin", plugin: name }
+        };
+        try {
+          await fetch(contextBase, { method: "DELETE", signal });
+        } catch {
+        }
+        return { ...decision, messages: [...decision.messages, contextMessage] };
+      },
+      { prepend: true }
+    );
+  }
 }
 export {
   apply,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aipanel/dsh-plugin",
-  "version": "1.2.4",
+  "version": "1.2.5",
   "type": "module",
   "description": "AIPanel for DeepSeek Harness (dsh)：注入审查工具 run_diagnostics、编辑后自动诊断。",
   "main": "./dist/index.js",
@@ -13,6 +13,7 @@
   },
   "devDependencies": {
     "@deepseek-ai/cordis": "^4.0.1",
+    "@deepseek-ai/dsh-agent": "^0.1.1-rc.2",
     "@deepseek-ai/dsh-llm": "^0.1.1-rc.2",
     "@deepseek-ai/dsh-subprocess": "^0.1.1-rc.2",
     "@deepseek-ai/dsh-tools": "^0.1.1-rc.2",