@wrongstack/webui 0.5.7 → 0.6.1

package/dist/server/index.js

@@ -1,5 +1,6 @@
 // src/server/index.ts
 import * as fs2 from "fs/promises";
+import * as http from "http";
 import * as path from "path";
 import {
   Agent,
@@ -24,6 +25,7 @@ import {
   repairToolUseAdjacency,
   resolveContextWindowPolicy
 } from "@wrongstack/core";
+import { ToolExecutor } from "@wrongstack/core/execution";
 import { decryptConfigSecrets, encryptConfigSecrets } from "@wrongstack/core/security";
 import { buildProviderFactoriesFromRegistry, makeProviderFromConfig } from "@wrongstack/providers";
 import { builtinToolsPack, forgetTool, rememberTool } from "@wrongstack/tools";
@@ -83,7 +85,13 @@ async function startWebUI(opts = {}) {
   const { config: baseConfig, vault, globalConfigPath, projectRoot, wpaths, logger } = boot;
   let config = baseConfig;
   let configWriteLock = Promise.resolve();
-  console.log("[WebUI] Config loaded:", config.provider, "/", config.model);
+  console.log("[WebUI] Config loaded:", config.provider ?? "(none)", "/", config.model ?? "(none)");
+  if (!config.provider && config.providers && Object.keys(config.providers).length > 0) {
+    const firstKey = Object.keys(config.providers)[0];
+    config = patchConfig(config, { provider: firstKey });
+    console.log("[WebUI] No active provider \u2014 auto-selected:", firstKey);
+  }
+  const needsProvider = !config.provider || !config.model;
   const modelsRegistry = new DefaultModelsRegistry({
     cacheFile: wpaths.modelsCache,
     ttlSeconds: 24 * 3600
@@ -151,22 +159,46 @@ async function startWebUI(opts = {}) {
     provider: config.provider,
     model: config.model
   });
-  const providerConfig = config.providers?.[config.provider] ?? {
-    type: config.provider,
-    apiKey: config.apiKey,
-    baseUrl: config.baseUrl
-  };
   let provider;
-  try {
-    const cfgWithType = { ...providerConfig, type: config.provider };
-    if (config.features.modelsRegistry && providerRegistry.has(config.provider)) {
-      provider = providerRegistry.create(cfgWithType);
+  if (!needsProvider) {
+    const providerConfig = config.providers?.[config.provider] ?? {
+      type: config.provider,
+      apiKey: config.apiKey,
+      baseUrl: config.baseUrl
+    };
+    try {
+      const cfgWithType = { ...providerConfig, type: config.provider };
+      if (config.features.modelsRegistry && providerRegistry.has(config.provider)) {
+        provider = providerRegistry.create(cfgWithType);
+      } else {
+        provider = makeProviderFromConfig(config.provider, cfgWithType);
+      }
+    } catch (err) {
+      console.error("[WebUI] Failed to create provider:", err);
+      throw err;
+    }
+  } else {
+    const savedProviders = config.providers ?? {};
+    const firstKey = Object.keys(savedProviders)[0];
+    if (firstKey) {
+      const firstProvider = savedProviders[firstKey];
+      try {
+        provider = makeProviderFromConfig(firstKey, {
+          ...firstProvider,
+          type: firstKey,
+          family: firstProvider.family,
+          apiKey: firstProvider.apiKey
+        });
+        console.log("[WebUI] Using saved provider:", firstKey);
+      } catch (err) {
+        console.error("[WebUI] Could not create provider stub:", err);
+        throw err;
+      }
     } else {
-      provider = makeProviderFromConfig(config.provider, cfgWithType);
+      throw new Error(
+        "No provider configured. Run `wrongstack init` first, or configure via the WebUI."
+      );
     }
-  } catch (err) {
-    console.error("[WebUI] Failed to create provider:", err);
-    throw err;
   }
   const context = new Context({
     systemPrompt,
@@ -228,6 +260,18 @@ async function startWebUI(opts = {}) {
     }
     autoCompactor.setMaxContext(newMaxContext);
   }
+  const secretScrubber = container.resolve(TOKENS.SecretScrubber);
+  const renderer = container.has(TOKENS.Renderer) ? container.resolve(TOKENS.Renderer) : void 0;
+  const toolExecutor = new ToolExecutor(toolRegistry, {
+    permissionPolicy: container.resolve(TOKENS.PermissionPolicy),
+    secretScrubber,
+    renderer,
+    events,
+    confirmAwaiter: void 0,
+    iterationTimeoutMs: config.tools?.iterationTimeoutMs ?? 12e4,
+    perIterationOutputCapBytes: config.tools?.perIterationOutputCapBytes ?? 5e4,
+    tracer: void 0
+  });
   const agent = new Agent({
     container,
     tools: toolRegistry,
@@ -239,7 +283,8 @@ async function startWebUI(opts = {}) {
     iterationTimeoutMs: config.tools?.iterationTimeoutMs ?? 12e4,
     executionStrategy: config.tools?.defaultExecutionStrategy ?? "sequential",
     perIterationOutputCapBytes: config.tools?.perIterationOutputCapBytes ?? 5e4,
-    confirmAwaiter: void 0
+    confirmAwaiter: void 0,
+    toolExecutor
   });
   console.log("[WebUI] Agent initialized");
   async function sessionStartPayload() {
@@ -458,8 +503,8 @@ async function startWebUI(opts = {}) {
       rateLimits.delete(ws);
       console.log("[WebUI] Client disconnected, total:", clients.size);
       if (pendingConfirms.size > 0) {
-        for (const [id, resolve] of pendingConfirms) {
-          resolve("no");
+        for (const [id, resolve2] of pendingConfirms) {
+          resolve2("no");
           pendingConfirms.delete(id);
         }
       }
@@ -539,10 +584,10 @@ async function startWebUI(opts = {}) {
       }
       case "tool.confirm_result": {
         const { id, decision } = msg.payload;
-        const resolve = pendingConfirms.get(id);
-        if (resolve) {
+        const resolve2 = pendingConfirms.get(id);
+        if (resolve2) {
           pendingConfirms.delete(id);
-          resolve(decision);
+          resolve2(decision);
         }
         break;
       }
@@ -1404,6 +1449,58 @@ async function startWebUI(opts = {}) {
       sendResult(ws, false, err instanceof Error ? err.message : String(err));
     }
   }
+  const httpPort = Number.parseInt(process.env["PORT"] ?? "3456", 10);
+  const DIST_DIR = path.resolve(import.meta.dirname, "../../dist");
+  const mimeTypes = {
+    ".html": "text/html",
+    ".js": "application/javascript",
+    ".css": "text/css",
+    ".json": "application/json",
+    ".svg": "image/svg+xml",
+    ".png": "image/png",
+    ".ico": "image/x-icon"
+  };
+  const httpServer = http.createServer(async (req, res) => {
+    try {
+      const url = new URL(req.url ?? "/", `http://127.0.0.1:${httpPort}`);
+      let filePath;
+      if (url.pathname === "/" || url.pathname === "") {
+        filePath = path.join(DIST_DIR, "index.html");
+      } else if (url.pathname.startsWith("/assets/")) {
+        filePath = path.join(DIST_DIR, url.pathname);
+      } else if (url.pathname.startsWith("/")) {
+        filePath = path.join(DIST_DIR, url.pathname);
+      } else {
+        filePath = path.join(DIST_DIR, "index.html");
+      }
+      const ext = path.extname(filePath);
+      const contentType = mimeTypes[ext] ?? "application/octet-stream";
+      res.setHeader("Content-Type", contentType);
+      if (ext === ".html") {
+        res.setHeader("Cache-Control", "no-cache");
+      }
+      const fileContent = await fs2.readFile(filePath);
+      res.writeHead(200);
+      res.end(fileContent);
+    } catch (err) {
+      if (err.code === "ENOENT") {
+        try {
+          const fileContent = await fs2.readFile(path.join(DIST_DIR, "index.html"));
+          res.writeHead(200, { "Content-Type": "text/html" });
+          res.end(fileContent);
+        } catch {
+          res.writeHead(404);
+          res.end("Not found");
+        }
+      } else {
+        res.writeHead(500);
+        res.end("Server error");
+      }
+    }
+  });
+  httpServer.listen(httpPort, wsHost, () => {
+    console.log(`[WebUI] HTTP server running on http://${wsHost}:${httpPort}`);
+  });
   const shutdown = async () => {
     console.log("[WebUI] Shutting down...");
     try {
@@ -1417,6 +1514,7 @@ async function startWebUI(opts = {}) {
       console.warn("[WebUI] Error closing session:", e);
     }
     for (const [ws] of clients) ws.close();
+    httpServer.close();
     wssPrimary.close();
     wssSecondary?.close();
     process.exit(0);