@trim21/personal-pi-extensions 0.0.357 → 0.0.359

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@trim21/personal-pi-extensions",
-  "version": "0.0.357",
+  "version": "0.0.359",
   "type": "module",
   "description": "Custom pi coding-agent extensions: bwrap sandbox, workspace guard, opencode edit, and more",
   "keywords": [

package/src/lib/lsp/client.ts

@@ -99,6 +99,7 @@ export interface LspClient {
     version: number;
     mode?: "document" | "full";
     after?: number;
+    signal?: AbortSignal;
   }): Promise<void>;
   shutdown(): Promise<void>;
 }
@@ -168,6 +169,11 @@ async function withTimeout<T>(promise: Promise<T>, ms: number): Promise<T> {
   }
 }
 
+const sleep = (ms: number) => new Promise<void>((resolve) => setTimeout(resolve, ms));
+
+/** pull 诊断失败后的重试间隔。 */
+const PULL_RETRY_INTERVAL_MS = 100;
+
 function stopProcess(process: LspServerHandle["process"]): Promise<void> {
   if (process.exitCode !== null) return Promise.resolve();
   try {
@@ -201,6 +207,14 @@ export async function create(input: CreateInput): Promise<LspClient> {
     new StreamMessageWriter(input.server.process.stdin),
   );
   input.server.process.stderr?.resume();
+  /** 连接或服务器进程已关闭；pull 重试循环以此终止，避免无界等待。 */
+  let connectionClosed = false;
+  input.server.process.once("exit", () => {
+    connectionClosed = true;
+  });
+  connection.onDispose(() => {
+    connectionClosed = true;
+  });
 
   // ── 连接状态 ────────────────────────────────────────────────────────────────
 
@@ -589,8 +603,20 @@ export async function create(input: CreateInput): Promise<LspClient> {
     path: string;
     version: number;
     after?: number;
+    signal?: AbortSignal;
   }): Promise<void> {
     const startedAt = request.after ?? Date.now();
+    // 支持 pull 的服务器：push 已被忽略，pull 是唯一通道。无窗口截断——
+    // 一直重试直到拿到当前文档结果（matched），确保诊断不因服务器重算慢而丢失。
+    if (supportsPullDiagnostics()) {
+      while (!connectionClosed && !request.signal?.aborted) {
+        const result = await requestDocumentDiagnostics(request.path);
+        if (result.matched) return;
+        await sleep(PULL_RETRY_INTERVAL_MS);
+      }
+      return;
+    }
+
     const pushWait = waitForFreshPush({
       path: request.path,
       version: request.version,
@@ -601,9 +627,6 @@ export async function create(input: CreateInput): Promise<LspClient> {
     while (Date.now() - startedAt < diagnosticsDocumentWaitTimeoutMs) {
       const result = await requestDocumentDiagnostics(request.path);
       if (result.matched) return;
-      // 支持 pull 的服务器：pull 未匹配（失败/超时）即返回，不依赖 push 兜底
-      // （push 已被忽略）；纯 push 服务器继续走下面的 push 等待。
-      if (supportsPullDiagnostics()) return;
       const remaining = diagnosticsDocumentWaitTimeoutMs - (Date.now() - startedAt);
       if (remaining <= 0) return;
       const next = await Promise.race([
@@ -620,8 +643,18 @@ export async function create(input: CreateInput): Promise<LspClient> {
     path: string;
     version: number;
     after?: number;
+    signal?: AbortSignal;
   }): Promise<void> {
     const startedAt = request.after ?? Date.now();
+    if (supportsPullDiagnostics()) {
+      while (!connectionClosed && !request.signal?.aborted) {
+        const result = await requestFullDiagnostics(request.path);
+        if (result.handled || result.matched) return;
+        await sleep(PULL_RETRY_INTERVAL_MS);
+      }
+      return;
+    }
+
     const pushWait = waitForFreshPush({
       path: request.path,
       version: request.version,
@@ -632,7 +665,6 @@ export async function create(input: CreateInput): Promise<LspClient> {
     while (Date.now() - startedAt < diagnosticsFullWaitTimeoutMs) {
       const result = await requestFullDiagnostics(request.path);
       if (result.handled || result.matched) return;
-      if (supportsPullDiagnostics()) return;
       const remaining = diagnosticsFullWaitTimeoutMs - (Date.now() - startedAt);
       if (remaining <= 0) return;
       const next = await Promise.race([
@@ -724,6 +756,7 @@ export async function create(input: CreateInput): Promise<LspClient> {
           path: normalizedPath,
           version: request.version,
           after: request.after,
+          signal: request.signal,
         });
         return;
       }
@@ -731,6 +764,7 @@ export async function create(input: CreateInput): Promise<LspClient> {
         path: normalizedPath,
         version: request.version,
         after: request.after,
+        signal: request.signal,
       });
     },
     async shutdown() {

package/src/lib/lsp/lsp.ts

@@ -348,7 +348,13 @@ export function createLspService(
         const version = await client.notify.open({ path: file });
         if (!diagnostics) return;
         await abortable(
-          client.waitForDiagnostics({ path: file, version, mode: diagnostics, after }),
+          client.waitForDiagnostics({
+            path: file,
+            version,
+            mode: diagnostics,
+            after,
+            signal: options?.signal,
+          }),
           options?.signal,
         );
       }),

package/src/web/fetch.ts

@@ -124,7 +124,8 @@ export async function fetchPage(url: string, signal?: AbortSignal): Promise<Fetc
     throw new Error(`HTTP ${response.status} ${response.statusText}`);
   }
   const contentType = response.headers.get("content-type") ?? "";
-  if (!contentType.includes("text/html") && !contentType.includes("text/plain")) {
+  const category = classifyContentType(contentType);
+  if (category === null) {
     throw new Error(`不支持的内容类型: ${contentType || "unknown"}`);
   }
 
@@ -133,22 +134,36 @@ export async function fetchPage(url: string, signal?: AbortSignal): Promise<Fetc
     throw new Error(`页面过大 (${declaredLength} bytes)，上限 ${MAX_BYTES}`);
   }
 
-  let html = "";
+  let body = "";
   if (response.body) {
     const reader = response.body.getReader();
     const decoder = new TextDecoder();
     for (;;) {
       const chunk = (await reader.read()) as { done: boolean; value: Uint8Array };
       if (chunk.done) break;
-      html += decoder.decode(chunk.value, { stream: true });
-      if (Buffer.byteLength(html, "utf8") > MAX_BYTES) {
+      body += decoder.decode(chunk.value, { stream: true });
+      if (Buffer.byteLength(body, "utf8") > MAX_BYTES) {
         throw new Error(`页面过大，上限 ${MAX_BYTES} bytes`);
       }
     }
-    html += decoder.decode();
+    body += decoder.decode();
   }
 
-  return extractMarkdown(html, response.url);
+  if (category === "html") {
+    return extractMarkdown(body, response.url);
+  }
+  // JSON / XML / text/*：原样返回
+  return { url: response.url, title: response.url, markdown: body.trim() };
+}
+
+/** 按 mime 主体分类响应；html 走 readability，其余文本类原样返回 */
+function classifyContentType(contentType: string): "html" | "text" | null {
+  const mime = contentType.split(";", 1)[0]?.trim().toLowerCase() ?? "";
+  if (mime === "text/html" || mime === "application/xhtml+xml") return "html";
+  if (mime.startsWith("text/")) return "text";
+  if (mime === "application/json" || mime.endsWith("+json")) return "text";
+  if (mime === "application/xml" || mime.endsWith("+xml")) return "text";
+  return null;
 }
 
 /** 提取用到的 document 最小接口（linkedom 类型是 any，显式标注避免 unsafe） */

package/src/web/index.ts

@@ -140,9 +140,10 @@ export default function webTools(pi: ExtensionAPI) {
     name: "web_fetch",
     label: "Web Fetch",
     description:
-      "Fetch a URL and extract the main content as markdown. SSRF-protected: refuses " +
-      "private/internal addresses. Only http/https HTML pages are supported.",
-    promptSnippet: "Fetch a web page and extract its content",
+      "Fetch a URL and return its content as markdown (HTML pages) or raw text " +
+      "(JSON/XML/plain-text API responses). SSRF-protected: refuses private/internal " +
+      "addresses.",
+    promptSnippet: "Fetch a web page or API response",
     parameters: Type.Object({
       url: Type.String({ description: "The URL to fetch" }),
     }),