@wrongstack/webui 0.6.0 → 0.6.3

package/dist/server/entry.js

@@ -1,6 +1,7 @@
 #!/usr/bin/env node
 // src/server/index.ts
 import * as fs2 from "fs/promises";
+import * as http from "http";
 import * as path from "path";
 import {
   Agent,
@@ -13,6 +14,7 @@ import {
   DefaultSkillLoader,
   DefaultSystemPromptBuilder,
   DefaultTokenCounter,
+  estimateRequestTokens,
   EventBus,
   HybridCompactor,
   ProviderRegistry,
@@ -85,7 +87,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
@@ -153,22 +161,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,
@@ -194,16 +226,7 @@ async function startWebUI(opts = {}) {
     autoCompactor = new AutoCompactionMiddleware(
       compactor,
       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;
-      },
+      (ctx) => estimateRequestTokens(ctx.messages, ctx.systemPrompt, ctx.tools ?? []).total,
       {
         warn: initialContextPolicy.thresholds.warn,
         soft: initialContextPolicy.thresholds.soft,
@@ -473,8 +496,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);
         }
       }
@@ -554,10 +577,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;
       }
@@ -1419,6 +1442,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, wsHost2, () => {
+    console.log(`[WebUI] HTTP server running on http://${wsHost2}:${httpPort}`);
+  });
   const shutdown = async () => {
     console.log("[WebUI] Shutting down...");
     try {
@@ -1432,6 +1507,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);