@duckmind/dm-windows-x64 0.60.6 → 0.60.9

package/extensions/dm-9router-ext/src/index.js

@@ -1,5 +1,410 @@
-import{randomUUID as T}from"node:crypto";import{chmodSync as w,existsSync as S,mkdirSync as M,readFileSync as $,renameSync as x,rmSync as E,statSync as B,writeFileSync as I}from"node:fs";import{homedir as D}from"node:os";import{dirname as L,join as O}from"node:path";import{Type as j}from"typebox";var F="http://localhost:20128",W=process.env.NINE_ROUTER_BASE_URL,G=process.env.NINE_ROUTER_API_KEY,f=O(D(),".dm","agent","9router-config.json"),A="9router-config",Y="9router-last-route",z=["x-9router-model","x-routed-model","x-actual-model","x-upstream-model","x-provider-model"];function C(e){return e.replace(/\/$/,"")}function U(e){return{baseUrl:C(W||e.baseUrl),apiKey:G||e.apiKey}}function H(){try{if(!S(f))return null;let e=JSON.parse($(f,"utf8"));if(typeof e.baseUrl!=="string"||!e.baseUrl.trim())return null;return{baseUrl:C(e.baseUrl.trim()),apiKey:typeof e.apiKey==="string"&&e.apiKey.trim()?e.apiKey.trim():void 0}}catch{return console.error("[dm-9router-ext] Failed to load persisted config"),null}}function J(e){let n={baseUrl:e.baseUrl};if(e.apiKey)n.apiKey=e.apiKey;return Buffer.from(`${JSON.stringify(n,null,2)}
-`)}function V(){if(!S(f))return{existed:!1};let e=B(f);return{existed:!0,contents:$(f),mode:e.mode&511}}function K(e,n=384){let s=`${f}.tmp-${process.pid}-${T()}`;try{return M(L(f),{recursive:!0,mode:448}),I(s,e,{flag:"wx",mode:n}),w(s,n),s}catch(c){throw E(s,{force:!0}),c}}function q(e){let n=V();return{tempPath:K(J(e)),snapshot:n}}function Q(e){x(e.tempPath,f),w(f,384)}function X(e){if(!e.existed){E(f,{force:!0});return}let n=K(e.contents??Buffer.alloc(0),e.mode??384);x(n,f),w(f,e.mode??384)}function Z(e){let n=e.sessionManager,s=typeof n.getBranch==="function"?n.getBranch():n.getEntries();for(let c=s.length-1;c>=0;c-=1){let t=s[c];if(t.type!=="custom"||t.customType!==A)continue;let o=t.data;if(typeof o?.baseUrl!=="string"||!o.baseUrl.trim())continue;return{baseUrl:C(o.baseUrl.trim()),apiKey:void 0}}return null}async function h(e,n){let s={Accept:"application/json"};if(e.apiKey)s.Authorization=`Bearer ${e.apiKey}`;let c=await fetch(`${e.baseUrl}/v1/models`,{method:"GET",headers:s,signal:n});if(!c.ok)throw Error(`9router returned ${c.status}: ${c.statusText}`);let t;try{t=await c.json()}catch{throw Error("9router returned invalid model metadata")}if(typeof t!=="object"||t===null||!Array.isArray(t.data)||t.data.some((o)=>typeof o!=="object"||o===null||typeof o.id!=="string"||!o.id.trim()))throw Error("9router returned invalid model metadata");return t.data}async function R(e,n){try{return await h(e,n),{ok:!0}}catch(s){return{ok:!1,error:s instanceof Error?s.message:String(s)}}}function ee(e){return{id:e.id,name:e.owned_by==="combo"?`\uD83D\uDD00 ${e.id}`:e.id,reasoning:!1,input:["text"],cost:{input:0,output:0,cacheRead:0,cacheWrite:0},contextWindow:128000,maxTokens:4096,compat:{supportsDeveloperRole:!1,supportsReasoningEffort:!1}}}function b(e,n,s){e.registerProvider("9router",{name:"9router",baseUrl:`${n.baseUrl}/v1`,apiKey:n.apiKey||"9router-no-api-key",api:"openai-completions",models:s.map(ee)})}function P(e){e.unregisterProvider("9router")}async function te(e){let n=H(),s=n||{baseUrl:F,apiKey:void 0},c=n!==null,t=U(s),o=[],m,v,y=!1,_=()=>({persistedConfig:s,hasPersistedConfig:c,config:t,discoveredModels:o,isConnected:y}),k=(u)=>{try{if(u.isConnected)b(e,u.config,u.discoveredModels);else P(e)}catch{console.error("[dm-9router-ext] Failed to restore the previous provider snapshot")}},N=(u,r,i,l)=>{let d=_(),a,p=!1,g=!1;try{if(a=q(u),p=!0,b(e,r,i),g=!0,Q(a),l)e.appendEntry(A,{baseUrl:u.baseUrl});s=u,c=!0,t=r,o=i,y=!0}catch{if(a)E(a.tempPath,{force:!0});if(g&&a)try{X(a.snapshot)}catch{console.error("[dm-9router-ext] Failed to restore the previous config file")}if(p)k(d);throw Error("9router configuration update failed; previous configuration remains active")}};try{o=await h(t,AbortSignal.timeout(1500)),y=!0,b(e,t,o)}catch{P(e)}e.on("session_start",async(u,r)=>{let i=Z(r);if(!c&&i){let l=U(i);try{let d=await h(l,r.signal);N(i,l,d,!1)}catch{}}if(y&&o.length>0)r.ui.notify(`9router connected — ${o.length} models available`,"info")}),e.on("after_provider_response",(u)=>{if(u.status>=400||v!=="9router")return;for(let r of z){let i=u.headers[r];if(!i||typeof i!=="string")continue;m=i,e.appendEntry(Y,{model:i,timestamp:Date.now()});break}}),e.on("model_select",async(u)=>{if(v=u.model.provider,v!=="9router")m=void 0}),e.registerCommand("9router-status",{description:"Show 9router connection status and configuration",handler:async(u,r)=>{let i=await R(t,r.signal),l=["\uD83D\uDD17 9router Status","",`Base URL:    ${t.baseUrl}`,`API Key:     ${t.apiKey?"set":"not set"}`,`Connection:  ${i.ok?"\uD83D\uDFE2 connected":`\uD83D\uDD34 ${i.error||"disconnected"}`}`,`Models:      ${o.length} available`];if(m)l.push(`Last routed: ${m}`);r.ui.notify(l.join(`
-`),i.ok?"info":"warning")}}),e.registerCommand("9router-models",{description:"Browse 9router available models and combos",handler:async(u,r)=>{if(o.length===0){r.ui.notify("No 9router models discovered. Check connection with /9router-status","warning");return}let i=o.map((a)=>({value:a.id,label:a.owned_by==="combo"?`\uD83D\uDD00 ${a.id}`:a.id})),l=await r.ui.select("Select a 9router model to use:",i.map((a)=>a.label)),d=i.find((a)=>a.label===l)?.value;if(!d)return;e.sendUserMessage(`/model 9router/${d}`,{deliverAs:"followUp"})}}),e.registerCommand("9router-config",{description:"Configure 9router base URL and API key",handler:async(u,r)=>{let i=await R(t,r.signal);r.ui.notify(["Current config:",`  Base URL:  ${t.baseUrl}`,`  API Key:   ${t.apiKey?"set":"not set"}`,`  Status:    ${i.ok?"\uD83D\uDFE2 connected":"\uD83D\uDD34 disconnected"}`,"","Enter new values. Blank keeps the current value."].join(`
-`),"info");let l=await r.ui.input("Base URL",t.baseUrl);if(l===void 0)return;let d=await r.ui.input("API key (blank keeps current, '-' clears):","");if(d===void 0)return;let a={baseUrl:C(l.trim()||s.baseUrl),apiKey:d.trim()===""?s.apiKey:d.trim()==="-"?void 0:d.trim()},p=U(a);try{let g=await h(p,r.signal);N(a,p,g,!0),r.ui.notify(`9router updated — ${o.length} models at ${t.baseUrl}`,"info")}catch{r.ui.notify("9router update failed; the previous configuration remains active","error")}}}),e.registerCommand("9router-reload",{description:"Reload models from 9router",handler:async(u,r)=>{let i=_(),l=!1;try{let d=await h(t,r.signal);l=!0,b(e,t,d),o=d,y=!0,r.ui.notify(`9router reloaded — ${o.length} models`,"info")}catch{if(l)k(i);r.ui.notify("9router reload failed; the previous provider and model list remain active","error")}}}),e.registerTool({name:"ninerouter_status",label:"9router Status",description:"Check 9router connection status and list available models",parameters:j.Object({}),async execute(u,r,i,l,d){let a=await R(t,d.signal),p=o.filter((g)=>g.owned_by==="combo");return{content:[{type:"text",text:[`9router: ${a.ok?"connected":`disconnected (${a.error})`}`,`Base URL: ${t.baseUrl}`,`Total models: ${o.length}`,`  Regular: ${o.length-p.length}`,`  Combos:  ${p.length}`,m?`Last routed model: ${m}`:""].filter(Boolean).join(`
-`)}],details:{connected:a.ok,baseUrl:t.baseUrl,modelCount:o.length,comboCount:p.length,lastRoutedModel:m,models:o.map((g)=>g.id)}}}})}export{h as fetchModels,te as default};
+import { randomUUID } from "node:crypto";
+import {
+  chmodSync,
+  existsSync,
+  mkdirSync,
+  readFileSync,
+  renameSync,
+  rmSync,
+  statSync,
+  writeFileSync
+} from "node:fs";
+import { homedir } from "node:os";
+import { dirname, join } from "node:path";
+import { Type } from "typebox";
+const DEFAULT_BASE_URL = "http://localhost:20128";
+const ENV_BASE_URL = process.env.NINE_ROUTER_BASE_URL;
+const ENV_API_KEY = process.env.NINE_ROUTER_API_KEY;
+const CONFIG_PATH = join(homedir(), ".dm", "agent", "9router-config.json");
+const CUSTOM_TYPE_CONFIG = "9router-config";
+const CUSTOM_TYPE_LAST_ROUTE = "9router-last-route";
+const ROUTING_HEADERS = [
+  "x-9router-model",
+  "x-routed-model",
+  "x-actual-model",
+  "x-upstream-model",
+  "x-provider-model"
+];
+function normalizeBaseUrl(url) {
+  return url.replace(/\/$/, "");
+}
+function applyEnvOverrides(config) {
+  return {
+    baseUrl: normalizeBaseUrl(ENV_BASE_URL || config.baseUrl),
+    apiKey: ENV_API_KEY || config.apiKey
+  };
+}
+function loadConfigFromDisk() {
+  try {
+    if (!existsSync(CONFIG_PATH))
+      return null;
+    const data = JSON.parse(readFileSync(CONFIG_PATH, "utf8"));
+    if (typeof data.baseUrl !== "string" || !data.baseUrl.trim())
+      return null;
+    return {
+      baseUrl: normalizeBaseUrl(data.baseUrl.trim()),
+      apiKey: typeof data.apiKey === "string" && data.apiKey.trim() ? data.apiKey.trim() : undefined
+    };
+  } catch {
+    console.error("[dm-9router-ext] Failed to load persisted config");
+    return null;
+  }
+}
+function serializeConfig(config) {
+  const persisted = {
+    baseUrl: config.baseUrl
+  };
+  if (config.apiKey)
+    persisted.apiKey = config.apiKey;
+  return Buffer.from(`${JSON.stringify(persisted, null, 2)}
+`);
+}
+function captureDiskConfigSnapshot() {
+  if (!existsSync(CONFIG_PATH))
+    return { existed: false };
+  const stat = statSync(CONFIG_PATH);
+  return {
+    existed: true,
+    contents: readFileSync(CONFIG_PATH),
+    mode: stat.mode & 511
+  };
+}
+function writeConfigTemp(contents, mode = 384) {
+  const tempPath = `${CONFIG_PATH}.tmp-${process.pid}-${randomUUID()}`;
+  try {
+    mkdirSync(dirname(CONFIG_PATH), { recursive: true, mode: 448 });
+    writeFileSync(tempPath, contents, { flag: "wx", mode });
+    chmodSync(tempPath, mode);
+    return tempPath;
+  } catch (error) {
+    rmSync(tempPath, { force: true });
+    throw error;
+  }
+}
+function prepareConfigWrite(config) {
+  const snapshot = captureDiskConfigSnapshot();
+  return {
+    tempPath: writeConfigTemp(serializeConfig(config)),
+    snapshot
+  };
+}
+function commitPreparedConfig(prepared) {
+  renameSync(prepared.tempPath, CONFIG_PATH);
+  chmodSync(CONFIG_PATH, 384);
+}
+function restoreDiskSnapshot(snapshot) {
+  if (!snapshot.existed) {
+    rmSync(CONFIG_PATH, { force: true });
+    return;
+  }
+  const tempPath = writeConfigTemp(snapshot.contents ?? Buffer.alloc(0), snapshot.mode ?? 384);
+  renameSync(tempPath, CONFIG_PATH);
+  chmodSync(CONFIG_PATH, snapshot.mode ?? 384);
+}
+function loadConfigFromSession(ctx) {
+  const sessionManager = ctx.sessionManager;
+  const entries = typeof sessionManager.getBranch === "function" ? sessionManager.getBranch() : sessionManager.getEntries();
+  for (let index = entries.length - 1;index >= 0; index -= 1) {
+    const entry = entries[index];
+    if (entry.type !== "custom" || entry.customType !== CUSTOM_TYPE_CONFIG)
+      continue;
+    const data = entry.data;
+    if (typeof data?.baseUrl !== "string" || !data.baseUrl.trim())
+      continue;
+    return {
+      baseUrl: normalizeBaseUrl(data.baseUrl.trim()),
+      apiKey: undefined
+    };
+  }
+  return null;
+}
+export async function fetchModels(config, signal) {
+  const headers = { Accept: "application/json" };
+  if (config.apiKey)
+    headers.Authorization = `Bearer ${config.apiKey}`;
+  const response = await fetch(`${config.baseUrl}/v1/models`, {
+    method: "GET",
+    headers,
+    signal
+  });
+  if (!response.ok) {
+    throw new Error(`9router returned ${response.status}: ${response.statusText}`);
+  }
+  let payload;
+  try {
+    payload = await response.json();
+  } catch {
+    throw new Error("9router returned invalid model metadata");
+  }
+  if (typeof payload !== "object" || payload === null || !Array.isArray(payload.data) || payload.data.some((model) => typeof model !== "object" || model === null || typeof model.id !== "string" || !model.id.trim())) {
+    throw new Error("9router returned invalid model metadata");
+  }
+  return payload.data;
+}
+async function testConnection(config, signal) {
+  try {
+    await fetchModels(config, signal);
+    return { ok: true };
+  } catch (error) {
+    return {
+      ok: false,
+      error: error instanceof Error ? error.message : String(error)
+    };
+  }
+}
+function mapNineRouterModel(model) {
+  return {
+    id: model.id,
+    name: model.owned_by === "combo" ? `\uD83D\uDD00 ${model.id}` : model.id,
+    reasoning: false,
+    input: ["text"],
+    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+    contextWindow: 128000,
+    maxTokens: 4096,
+    compat: {
+      supportsDeveloperRole: false,
+      supportsReasoningEffort: false
+    }
+  };
+}
+function registerNineRouterProvider(dm, config, models) {
+  dm.registerProvider("9router", {
+    name: "9router",
+    baseUrl: `${config.baseUrl}/v1`,
+    apiKey: config.apiKey || "9router-no-api-key",
+    api: "openai-completions",
+    models: models.map(mapNineRouterModel)
+  });
+}
+function unregisterNineRouterProvider(dm) {
+  dm.unregisterProvider("9router");
+}
+export default async function dm9router(dm) {
+  const diskConfig = loadConfigFromDisk();
+  let persistedConfig = diskConfig || {
+    baseUrl: DEFAULT_BASE_URL,
+    apiKey: undefined
+  };
+  let hasPersistedConfig = diskConfig !== null;
+  let config = applyEnvOverrides(persistedConfig);
+  let discoveredModels = [];
+  let lastRoutedModel;
+  let activeProvider;
+  let isConnected = false;
+  const snapshotRuntime = () => ({
+    persistedConfig,
+    hasPersistedConfig,
+    config,
+    discoveredModels,
+    isConnected
+  });
+  const restoreProviderSnapshot = (snapshot) => {
+    try {
+      if (snapshot.isConnected) {
+        registerNineRouterProvider(dm, snapshot.config, snapshot.discoveredModels);
+      } else {
+        unregisterNineRouterProvider(dm);
+      }
+    } catch {
+      console.error("[dm-9router-ext] Failed to restore the previous provider snapshot");
+    }
+  };
+  const commitConfigTransaction = (nextPersistedConfig, nextConfig, nextModels, appendSessionEntry) => {
+    const previous = snapshotRuntime();
+    let prepared;
+    let providerTouched = false;
+    let diskTouched = false;
+    try {
+      prepared = prepareConfigWrite(nextPersistedConfig);
+      providerTouched = true;
+      registerNineRouterProvider(dm, nextConfig, nextModels);
+      diskTouched = true;
+      commitPreparedConfig(prepared);
+      if (appendSessionEntry) {
+        dm.appendEntry(CUSTOM_TYPE_CONFIG, {
+          baseUrl: nextPersistedConfig.baseUrl
+        });
+      }
+      persistedConfig = nextPersistedConfig;
+      hasPersistedConfig = true;
+      config = nextConfig;
+      discoveredModels = nextModels;
+      isConnected = true;
+    } catch {
+      if (prepared)
+        rmSync(prepared.tempPath, { force: true });
+      if (diskTouched && prepared) {
+        try {
+          restoreDiskSnapshot(prepared.snapshot);
+        } catch {
+          console.error("[dm-9router-ext] Failed to restore the previous config file");
+        }
+      }
+      if (providerTouched)
+        restoreProviderSnapshot(previous);
+      throw new Error("9router configuration update failed; previous configuration remains active");
+    }
+  };
+  try {
+    discoveredModels = await fetchModels(config, AbortSignal.timeout(1500));
+    isConnected = true;
+    registerNineRouterProvider(dm, config, discoveredModels);
+  } catch {
+    unregisterNineRouterProvider(dm);
+  }
+  dm.on("session_start", async (_event, ctx) => {
+    const restored = loadConfigFromSession(ctx);
+    if (!hasPersistedConfig && restored) {
+      const restoredConfig = applyEnvOverrides(restored);
+      try {
+        const nextModels = await fetchModels(restoredConfig, ctx.signal);
+        commitConfigTransaction(restored, restoredConfig, nextModels, false);
+      } catch {}
+    }
+    if (isConnected && discoveredModels.length > 0) {
+      ctx.ui.notify(`9router connected — ${discoveredModels.length} models available`, "info");
+    }
+  });
+  dm.on("after_provider_response", (event) => {
+    if (event.status >= 400 || activeProvider !== "9router")
+      return;
+    for (const header of ROUTING_HEADERS) {
+      const value = event.headers[header];
+      if (!value || typeof value !== "string")
+        continue;
+      lastRoutedModel = value;
+      dm.appendEntry(CUSTOM_TYPE_LAST_ROUTE, {
+        model: value,
+        timestamp: Date.now()
+      });
+      break;
+    }
+  });
+  dm.on("model_select", async (event) => {
+    activeProvider = event.model.provider;
+    if (activeProvider !== "9router")
+      lastRoutedModel = undefined;
+  });
+  dm.registerCommand("9router-status", {
+    description: "Show 9router connection status and configuration",
+    handler: async (_args, ctx) => {
+      const test = await testConnection(config, ctx.signal);
+      const lines = [
+        "\uD83D\uDD17 9router Status",
+        "",
+        `Base URL:    ${config.baseUrl}`,
+        `API Key:     ${config.apiKey ? "set" : "not set"}`,
+        `Connection:  ${test.ok ? "\uD83D\uDFE2 connected" : `\uD83D\uDD34 ${test.error || "disconnected"}`}`,
+        `Models:      ${discoveredModels.length} available`
+      ];
+      if (lastRoutedModel)
+        lines.push(`Last routed: ${lastRoutedModel}`);
+      ctx.ui.notify(lines.join(`
+`), test.ok ? "info" : "warning");
+    }
+  });
+  dm.registerCommand("9router-models", {
+    description: "Browse 9router available models and combos",
+    handler: async (_args, ctx) => {
+      if (discoveredModels.length === 0) {
+        ctx.ui.notify("No 9router models discovered. Check connection with /9router-status", "warning");
+        return;
+      }
+      const items = discoveredModels.map((model) => ({
+        value: model.id,
+        label: model.owned_by === "combo" ? `\uD83D\uDD00 ${model.id}` : model.id
+      }));
+      const selected = await ctx.ui.select("Select a 9router model to use:", items.map((item) => item.label));
+      const modelId = items.find((item) => item.label === selected)?.value;
+      if (!modelId)
+        return;
+      dm.sendUserMessage(`/model 9router/${modelId}`, { deliverAs: "followUp" });
+    }
+  });
+  dm.registerCommand("9router-config", {
+    description: "Configure 9router base URL and API key",
+    handler: async (_args, ctx) => {
+      const test = await testConnection(config, ctx.signal);
+      ctx.ui.notify([
+        "Current config:",
+        `  Base URL:  ${config.baseUrl}`,
+        `  API Key:   ${config.apiKey ? "set" : "not set"}`,
+        `  Status:    ${test.ok ? "\uD83D\uDFE2 connected" : "\uD83D\uDD34 disconnected"}`,
+        "",
+        "Enter new values. Blank keeps the current value."
+      ].join(`
+`), "info");
+      const newBaseUrl = await ctx.ui.input("Base URL", config.baseUrl);
+      if (newBaseUrl === undefined)
+        return;
+      const newApiKey = await ctx.ui.input("API key (blank keeps current, '-' clears):", "");
+      if (newApiKey === undefined)
+        return;
+      const nextPersistedConfig = {
+        baseUrl: normalizeBaseUrl(newBaseUrl.trim() || persistedConfig.baseUrl),
+        apiKey: newApiKey.trim() === "" ? persistedConfig.apiKey : newApiKey.trim() === "-" ? undefined : newApiKey.trim()
+      };
+      const nextConfig = applyEnvOverrides(nextPersistedConfig);
+      try {
+        const nextModels = await fetchModels(nextConfig, ctx.signal);
+        commitConfigTransaction(nextPersistedConfig, nextConfig, nextModels, true);
+        ctx.ui.notify(`9router updated — ${discoveredModels.length} models at ${config.baseUrl}`, "info");
+      } catch {
+        ctx.ui.notify("9router update failed; the previous configuration remains active", "error");
+      }
+    }
+  });
+  dm.registerCommand("9router-reload", {
+    description: "Reload models from 9router",
+    handler: async (_args, ctx) => {
+      const previous = snapshotRuntime();
+      let providerTouched = false;
+      try {
+        const nextModels = await fetchModels(config, ctx.signal);
+        providerTouched = true;
+        registerNineRouterProvider(dm, config, nextModels);
+        discoveredModels = nextModels;
+        isConnected = true;
+        ctx.ui.notify(`9router reloaded — ${discoveredModels.length} models`, "info");
+      } catch {
+        if (providerTouched)
+          restoreProviderSnapshot(previous);
+        ctx.ui.notify("9router reload failed; the previous provider and model list remain active", "error");
+      }
+    }
+  });
+  dm.registerTool({
+    name: "ninerouter_status",
+    label: "9router Status",
+    description: "Check 9router connection status and list available models",
+    parameters: Type.Object({}),
+    async execute(_toolCallId, _params, _signal, _onUpdate, ctx) {
+      const test = await testConnection(config, ctx.signal);
+      const combos = discoveredModels.filter((model) => model.owned_by === "combo");
+      return {
+        content: [
+          {
+            type: "text",
+            text: [
+              `9router: ${test.ok ? "connected" : `disconnected (${test.error})`}`,
+              `Base URL: ${config.baseUrl}`,
+              `Total models: ${discoveredModels.length}`,
+              `  Regular: ${discoveredModels.length - combos.length}`,
+              `  Combos:  ${combos.length}`,
+              lastRoutedModel ? `Last routed model: ${lastRoutedModel}` : ""
+            ].filter(Boolean).join(`
+`)
+          }
+        ],
+        details: {
+          connected: test.ok,
+          baseUrl: config.baseUrl,
+          modelCount: discoveredModels.length,
+          comboCount: combos.length,
+          lastRoutedModel,
+          models: discoveredModels.map((model) => model.id)
+        }
+      };
+    }
+  });
+}