mslxdff 0.1.43 → 0.1.45

package/src/routes/peers.js

@@ -1,6 +1,7 @@
 import { performance } from "node:perf_hooks";
 import { isAutoModel } from "../auto.js";
 import { errMsg } from "./helpers.js";
+import { runHook } from "../plugins.js";
 
 const PEER_TIMEOUT_MS = 30_000;
 const PEER_STATUS_TIMEOUT_MS = 2_000;
@@ -86,11 +87,19 @@ export async function racePeerCandidates(candidates, ctx) {
       const total = prepared.length;
       for (const { peer, target } of prepared) {
         ctx.evt("peer-request", { peer: peer.url, model: target, hops: ctx.hops + 1 });
+        // 插件 hook：peer:beforeForward — 转发给组员前观察
+        if (ctx.plugins?.length) {
+          runHook(ctx.plugins, "peer:beforeForward", { reqId: ctx.reqId, peer: peer.url, model: target, hops: ctx.hops + 1 }).catch(() => {});
+        }
         const t0 = performance.now();
         forwardToPeer(peer, ctx.body, target, ctx.hops).then((res) => {
           const latencyMs = Math.round(performance.now() - t0);
           const failed = res instanceof Error || res.status >= 400;
           ctx.evt("peer-forward", { peer: peer.url, model: target, hops: ctx.hops + 1, latencyMs, ok: !failed });
+          // 插件 hook：peer:result — 组员响应后观察
+          if (ctx.plugins?.length) {
+            runHook(ctx.plugins, "peer:result", { reqId: ctx.reqId, peer: peer.url, model: target, ok: !failed, status: failed ? (res instanceof Error ? 502 : res.status) : res.status, latencyMs }).catch(() => {});
+          }
           if (failed) {
             const status = res instanceof Error ? 502 : res.status;
             ctx.logError(ctx.model, status, res instanceof Error ? errMsg(res) : `peer ${status}`);

package/src/server.js

@@ -1,7 +1,7 @@
 import { createServer as httpCreateServer } from "node:http";
 import { DEFAULT_PORT, getPort } from "./state.js";
 
-export function startServer({ router, signals = true, host }, port = resolvePort()) {
+export function startServer({ router, signals = true, host, onBeforeClose }, port = resolvePort()) {
   const listenHost = host ?? resolveHost();
   const server = httpCreateServer((req, res) => {
     router(req, res).catch((err) => {
@@ -18,13 +18,16 @@ export function startServer({ router, signals = true, host }, port = resolvePort
       else server.listen(port, resolve);
     });
 
-  const close = () =>
-    new Promise((resolve) => {
+  const close = async () => {
+    // 插件 hook：server:stop — 关闭前触发（fire-and-forget，不阻塞关停）
+    try { await onBeforeClose?.(); } catch {}
+    await new Promise((resolve) => {
       server.close(resolve);
       // SSE debug streams keep connections open — force-close so shutdown
       // never waits on them.
       server.closeAllConnections?.();
     });
+  };
 
   if (signals) {
     process.on("SIGINT", close);

package/src/state.js

@@ -58,6 +58,17 @@ export function saveModelLatencies(latencies, { file = defaultStateFile() } = {}
   return latencies;
 }
 
+export function loadPreferredModel({ file = defaultStateFile() } = {}) {
+  const v = readState(file).preferredModel;
+  return typeof v === "string" && v.trim() ? v.trim() : null;
+}
+
+export function savePreferredModel(id, { file = defaultStateFile() } = {}) {
+  const state = readState(file);
+  writeState(file, { ...state, preferredModel: String(id || "").trim() });
+  return String(id || "").trim();
+}
+
 export function loadPeers({ file = defaultStateFile() } = {}) {
   const peers = readState(file).peers;
   return Array.isArray(peers) ? peers : [];

package/src/upstream.js

@@ -38,9 +38,19 @@ export function createUpstreamClient({
     504: { attempts: 1, delayMs: 500 },
   },
   fetchImpl,
+  hooks,
 } = {}) {
   // 使用 undici 的 fetch 与 Agent 配对，避免 global fetch 与 npm undici Agent 不兼容
   if (!fetchImpl) fetchImpl = UndiciFetch || fetch;
+  // hooks: async (name, ctx) => ({ value, changed }) | null — 由 bin 用 runHook 绑定；错误已在上层隔离
+  const applyHook = async (name, ctx) => {
+    if (!hooks) return null;
+    try {
+      return await hooks(name, ctx);
+    } catch {
+      return null;
+    }
+  };
   const baseHeaders = {
     "Content-Type": "application/json",
     "Authorization": `Bearer ${authToken}`,
@@ -129,7 +139,17 @@ export function createUpstreamClient({
       connectTimeoutMs
     );
     try {
-      const headers = buildHeaders(body);
+      let reqUrl = url;
+      let headers = buildHeaders(body);
+      // 插件 hook：upstream:headers — 返回 { headers } 可替换请求头
+      const hh = await applyHook("upstream:headers", { url, body, headers });
+      if (hh?.changed && hh.value?.headers) headers = hh.value.headers;
+      // 插件 hook：upstream:before-request — 返回 { url, headers } 可改目标地址/头（上游不限于 opencode）
+      const br = await applyHook("upstream:before-request", { url, method: "POST", body, headers });
+      if (br?.changed && br.value) {
+        if (typeof br.value.url === "string" && br.value.url) reqUrl = br.value.url;
+        if (br.value.headers && typeof br.value.headers === "object") headers = br.value.headers;
+      }
       const opts = {
         method: "POST",
         headers,
@@ -137,7 +157,7 @@ export function createUpstreamClient({
         signal: controller.signal,
       };
       if (dispatcher) opts.dispatcher = dispatcher;
-      const res = await fetchImpl(url, opts);
+      const res = await fetchImpl(reqUrl, opts);
       return res;
     } catch (err) {
       return err;