@wrongstack/webui 0.2.0 → 0.3.2

package/dist/server/index.js

@@ -1,24 +1,13 @@
 // src/server/index.ts
-import * as fs from "fs/promises";
-import * as os from "os";
+import * as fs2 from "fs/promises";
 import * as path from "path";
 import {
   Agent,
   AutoCompactionMiddleware,
-  Container,
   Context,
-  DefaultConfigLoader,
-  DefaultConfigStore,
-  DefaultErrorHandler,
-  DefaultLogger,
   DefaultMemoryStore,
   DefaultModeStore,
   DefaultModelsRegistry,
-  DefaultPathResolver,
-  DefaultPermissionPolicy,
-  DefaultRetryPolicy,
-  DefaultSecretScrubber,
-  DefaultSecretVault,
   DefaultSessionStore,
   DefaultSkillLoader,
   DefaultSystemPromptBuilder,
@@ -30,13 +19,29 @@ import {
   ToolRegistry,
   atomicWrite,
   createDefaultPipelines,
-  migratePlaintextSecrets,
-  resolveWstackPaths
+  DEFAULT_CONTEXT_WINDOW_MODE_ID,
+  listContextWindowModes,
+  repairToolUseAdjacency,
+  resolveContextWindowPolicy
 } from "@wrongstack/core";
 import { buildProviderFactoriesFromRegistry, makeProviderFromConfig } from "@wrongstack/providers";
 import { forgetTool, rememberTool } from "@wrongstack/tools";
-import { builtinTools } from "@wrongstack/tools/builtin";
+import { builtinToolsPack } from "@wrongstack/tools/pack";
 import { WebSocket, WebSocketServer } from "ws";
+import { randomBytes } from "crypto";
+import { createDefaultContainer } from "@wrongstack/runtime";
+
+// src/server/boot.ts
+import * as fs from "fs/promises";
+import * as os from "os";
+import {
+  DefaultConfigLoader,
+  DefaultLogger,
+  DefaultPathResolver,
+  DefaultSecretVault,
+  migratePlaintextSecrets,
+  resolveWstackPaths
+} from "@wrongstack/core";
 async function bootConfig() {
   const cwd = process.cwd();
   const pathResolver = new DefaultPathResolver(cwd);
@@ -63,18 +68,13 @@ async function bootConfig() {
     level: config.log?.level ?? "info",
     file: wpaths.logFile
   });
-  return {
-    config,
-    vault,
-    globalConfigPath: wpaths.globalConfig,
-    projectRoot,
-    wpaths,
-    logger
-  };
+  return { config, vault, globalConfigPath: wpaths.globalConfig, projectRoot, wpaths, logger };
 }
 function patchConfig(config, updates) {
   return Object.freeze({ ...config, ...updates });
 }
+
+// src/server/index.ts
 async function startWebUI(opts = {}) {
   const wsPort = opts.wsPort ?? 3457;
   const wsHost = opts.wsHost ?? "127.0.0.1";
@@ -87,22 +87,8 @@ async function startWebUI(opts = {}) {
     cacheFile: wpaths.modelsCache,
     ttlSeconds: 24 * 3600
   });
-  const container = new Container();
-  const configStore = new DefaultConfigStore(config);
-  container.bind(TOKENS.ConfigStore, () => configStore);
-  container.bind(TOKENS.Logger, () => logger);
-  container.bind(TOKENS.SecretScrubber, () => new DefaultSecretScrubber());
-  container.bind(TOKENS.RetryPolicy, () => new DefaultRetryPolicy());
-  container.bind(TOKENS.ErrorHandler, () => new DefaultErrorHandler());
-  container.bind(TOKENS.ModelsRegistry, () => modelsRegistry);
-  container.bind(
-    TOKENS.PermissionPolicy,
-    () => new DefaultPermissionPolicy({
-      trustFile: wpaths.projectTrust,
-      yolo: false,
-      promptDelegate: void 0
-    })
-  );
+  const container = createDefaultContainer({ config, wpaths, logger, modelsRegistry });
+  const configStore = container.resolve(TOKENS.ConfigStore);
   const providerRegistry = new ProviderRegistry();
   try {
     const factories = await buildProviderFactoriesFromRegistry({
@@ -115,7 +101,7 @@ async function startWebUI(opts = {}) {
     console.warn("[WebUI] Failed to load provider registry:", err);
   }
   const toolRegistry = new ToolRegistry();
-  for (const t of builtinTools) toolRegistry.register(t);
+  toolRegistry.registerAllOrThrow([...builtinToolsPack.tools ?? []], builtinToolsPack.name);
   const memoryStore = new DefaultMemoryStore({ paths: wpaths });
   if (config.features.memory) {
     toolRegistry.register(rememberTool(memoryStore));
@@ -191,24 +177,42 @@ async function startWebUI(opts = {}) {
     projectRoot,
     model: config.model
   });
+  const initialContextPolicy = resolveContextWindowPolicy(config.context);
+  context.meta["contextWindowMode"] = initialContextPolicy.id;
+  context.meta["contextWindowPolicy"] = initialContextPolicy;
   const pipelines = createDefaultPipelines();
   const compactor = new HybridCompactor({
     preserveK: config.context?.preserveK ?? 20,
     eliseThreshold: config.context?.eliseThreshold ?? 0.7
   });
   if (config.context?.autoCompact !== false) {
+    const effectiveMaxContext = config.context?.effectiveMaxContext ?? provider.capabilities.maxContext;
     const autoCompactor = new AutoCompactionMiddleware(
       compactor,
-      2e5,
+      effectiveMaxContext,
       (ctx) => {
         let total = 0;
         for (const m of ctx.messages) {
           if (typeof m.content === "string") total += Math.ceil(m.content.length / 4);
+          else if (Array.isArray(m.content)) {
+            for (const b of m.content) total += Math.ceil(JSON.stringify(b).length / 4);
+          }
         }
         return total;
       },
-      { warn: 0.7, soft: 0.85, hard: 0.95 },
-      { events }
+      {
+        warn: initialContextPolicy.thresholds.warn,
+        soft: initialContextPolicy.thresholds.soft,
+        hard: initialContextPolicy.thresholds.hard
+      },
+      {
+        events,
+        aggressiveOn: initialContextPolicy.aggressiveOn,
+        policyProvider: (ctx) => {
+          const policy = ctx.meta["contextWindowPolicy"];
+          return policy && typeof policy === "object" ? policy : initialContextPolicy;
+        }
+      }
     );
     pipelines.contextWindow.use({ name: "AutoCompaction", handler: autoCompactor.handler() });
   }
@@ -250,15 +254,27 @@ async function startWebUI(opts = {}) {
       cacheReadCost,
       projectName: path.basename(projectRoot) || projectRoot,
       cwd: projectRoot,
-      mode: modeId
+      mode: modeId,
+      contextMode: String(context.meta["contextWindowMode"] ?? DEFAULT_CONTEXT_WINDOW_MODE_ID),
+      wsToken
     };
   }
+  const wsToken = randomBytes(16).toString("hex");
+  console.log(`[WebUI] WS auth token: ${wsToken}`);
+  const isLoopback = (hostname) => hostname === "localhost" || hostname === "127.0.0.1" || hostname === "::1" || hostname === "[::1]";
   const verifyClient = (info) => {
     const origin = info.origin;
-    if (!origin) return true;
+    const url = info.req.url ?? "";
+    const tokenMatch = url.match(/[?&]token=([^&]+)/);
+    const providedToken = tokenMatch ? tokenMatch[1] : void 0;
+    const tokenOk = providedToken === wsToken;
+    if (!origin) {
+      return tokenOk || wsHost === "127.0.0.1" || wsHost === "::1" || wsHost === "localhost";
+    }
     try {
       const { hostname } = new URL(origin);
-      return hostname === "localhost" || hostname === "127.0.0.1" || hostname === "::1";
+      if (isLoopback(hostname)) return true;
+      return tokenOk;
     } catch {
       return false;
     }
@@ -335,6 +351,16 @@ async function startWebUI(opts = {}) {
         }
       });
     });
+    events.on("context.repaired", (e) => {
+      broadcast({
+        type: "context.repaired",
+        payload: {
+          removedToolUses: e.removedToolUses,
+          removedToolResults: e.removedToolResults,
+          removedMessages: e.removedMessages
+        }
+      });
+    });
     events.on("error", (e) => {
       broadcast({
         type: "error",
@@ -535,6 +561,8 @@ async function startWebUI(opts = {}) {
           type: "context.debug",
           payload: {
             total,
+            mode: context.meta["contextWindowMode"] ?? DEFAULT_CONTEXT_WINDOW_MODE_ID,
+            policy: context.meta["contextWindowPolicy"],
             systemPrompt: sysTokens,
             tools: { total: toolTokens, count: tools.length, breakdown: toolBreakdown },
             messages: {
@@ -556,7 +584,8 @@ async function startWebUI(opts = {}) {
               before: report.before,
               after: report.after,
               saved: Math.max(0, report.before - report.after),
-              reductions: report.reductions
+              reductions: report.reductions,
+              repaired: report.repaired
             }
           });
           sendResult(
@@ -569,6 +598,63 @@ async function startWebUI(opts = {}) {
         }
         break;
       }
+      case "context.repair": {
+        const beforeMessages = context.messages.length;
+        const repaired = repairToolUseAdjacency(context.messages);
+        if (repaired.report.changed) {
+          context.state.replaceMessages(repaired.messages);
+        }
+        const payload = {
+          removedToolUses: repaired.report.removedToolUses,
+          removedToolResults: repaired.report.removedToolResults,
+          removedMessages: repaired.report.removedMessages,
+          beforeMessages,
+          afterMessages: context.messages.length
+        };
+        broadcast({ type: "context.repaired", payload });
+        const removed = payload.removedToolUses.length + payload.removedToolResults.length + payload.removedMessages;
+        sendResult(
+          ws,
+          true,
+          removed > 0 ? `Context repaired: removed ${removed} orphan protocol item(s)` : "Context repair found no orphan protocol blocks"
+        );
+        break;
+      }
+      case "context.modes.list": {
+        const active = String(context.meta["contextWindowMode"] ?? DEFAULT_CONTEXT_WINDOW_MODE_ID);
+        send(ws, {
+          type: "context.modes.list",
+          payload: {
+            activeId: active,
+            modes: listContextWindowModes().map((m) => ({
+              id: m.id,
+              name: m.name,
+              description: m.description,
+              isActive: m.id === active,
+              thresholds: m.thresholds,
+              preserveK: m.preserveK,
+              eliseThreshold: m.eliseThreshold
+            }))
+          }
+        });
+        break;
+      }
+      case "context.mode.switch": {
+        const { id } = msg.payload;
+        const policy = resolveContextWindowPolicy({}, id);
+        if (policy.id !== id) {
+          sendResult(ws, false, `Unknown context mode "${id}"`);
+          break;
+        }
+        context.meta["contextWindowMode"] = policy.id;
+        context.meta["contextWindowPolicy"] = policy;
+        sendResult(ws, true, `Context mode switched to ${policy.id}`);
+        broadcast({
+          type: "context.mode.changed",
+          payload: { id: policy.id, name: policy.name, policy }
+        });
+        break;
+      }
       case "providers.list": {
         const providers = await modelsRegistry.listProviders();
         const savedIds = new Set(Object.keys(config.providers ?? {}));
@@ -623,7 +709,7 @@ async function startWebUI(opts = {}) {
           const newProv = providerRegistry.has(newProvider) ? providerRegistry.create({ ...providerCfg, type: newProvider }) : makeProviderFromConfig(newProvider, providerCfg);
           context.provider = newProv;
           try {
-            const raw = await fs.readFile(globalConfigPath, "utf8");
+            const raw = await fs2.readFile(globalConfigPath, "utf8");
             const parsed = JSON.parse(raw);
             parsed.provider = newProvider;
             parsed.model = newModel;
@@ -925,7 +1011,7 @@ async function startWebUI(opts = {}) {
           if (depth > 8 || results.length >= 600) return;
           let entries = [];
           try {
-            entries = await fs.readdir(dir, { withFileTypes: true });
+            entries = await fs2.readdir(dir, { withFileTypes: true });
           } catch {
             return;
           }
@@ -1062,7 +1148,7 @@ async function startWebUI(opts = {}) {
   }
   async function loadSavedProviders() {
     try {
-      const raw = await fs.readFile(globalConfigPath, "utf8");
+      const raw = await fs2.readFile(globalConfigPath, "utf8");
       const parsed = JSON.parse(raw);
       return parsed.providers ?? {};
     } catch {
@@ -1072,7 +1158,7 @@ async function startWebUI(opts = {}) {
   async function saveProviders(providers) {
     let parsed;
     try {
-      const raw = await fs.readFile(globalConfigPath, "utf8");
+      const raw = await fs2.readFile(globalConfigPath, "utf8");
       parsed = JSON.parse(raw);
     } catch {
       parsed = {};