chainlesschain 0.38.1 → 0.40.1

package/src/lib/ws-server.js

@@ -0,0 +1,474 @@
+/**
+ * ChainlessChain WebSocket Server
+ *
+ * Exposes CLI commands over WebSocket for remote access by IDE plugins,
+ * web frontends, automation scripts, etc. Commands are executed by spawning
+ * child processes — all 60+ CLI commands are available immediately.
+ */
+
+import { EventEmitter } from "node:events";
+import { spawn } from "node:child_process";
+import { fileURLToPath } from "node:url";
+import { dirname, join } from "node:path";
+import { WebSocketServer } from "ws";
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+/** Absolute path to the CLI entry point */
+const BIN_PATH = join(__dirname, "..", "..", "bin", "chainlesschain.js");
+
+/** Commands that must not be executed via WebSocket */
+const BLOCKED_COMMANDS = new Set(["serve", "chat", "agent", "setup"]);
+
+/** Heartbeat interval (ms) */
+const HEARTBEAT_INTERVAL = 30_000;
+
+/**
+ * Tokenize a command string into an array of arguments.
+ * Handles double-quoted and single-quoted strings. Does NOT invoke a shell.
+ */
+export function tokenizeCommand(input) {
+  const args = [];
+  let current = "";
+  let inDouble = false;
+  let inSingle = false;
+  let escape = false;
+
+  for (const ch of input) {
+    if (escape) {
+      current += ch;
+      escape = false;
+      continue;
+    }
+    if (ch === "\\" && inDouble) {
+      escape = true;
+      continue;
+    }
+    if (ch === '"' && !inSingle) {
+      inDouble = !inDouble;
+      continue;
+    }
+    if (ch === "'" && !inDouble) {
+      inSingle = !inSingle;
+      continue;
+    }
+    if ((ch === " " || ch === "\t") && !inDouble && !inSingle) {
+      if (current.length > 0) {
+        args.push(current);
+        current = "";
+      }
+      continue;
+    }
+    current += ch;
+  }
+  if (current.length > 0) {
+    args.push(current);
+  }
+  return args;
+}
+
+export class ChainlessChainWSServer extends EventEmitter {
+  /**
+   * @param {object} options
+   * @param {number}  [options.port=18800]
+   * @param {string}  [options.host="127.0.0.1"]
+   * @param {string}  [options.token]           - If set, clients must authenticate first
+   * @param {number}  [options.maxConnections=10]
+   * @param {number}  [options.timeout=30000]   - Command execution timeout (ms)
+   */
+  constructor(options = {}) {
+    super();
+    this.port = options.port || 18800;
+    this.host = options.host || "127.0.0.1";
+    this.token = options.token || null;
+    this.maxConnections = options.maxConnections || 10;
+    this.timeout = options.timeout || 30000;
+
+    /** @type {WebSocketServer|null} */
+    this.wss = null;
+
+    /** Connected clients: clientId → { ws, authenticated, connectedAt } */
+    this.clients = new Map();
+
+    /** Running child processes: requestId → ChildProcess */
+    this.processes = new Map();
+
+    this._heartbeatTimer = null;
+    this._clientCounter = 0;
+  }
+
+  /** Start the WebSocket server */
+  start() {
+    return new Promise((resolve, reject) => {
+      this.wss = new WebSocketServer({
+        port: this.port,
+        host: this.host,
+      });
+
+      this.wss.on("listening", () => {
+        this._startHeartbeat();
+        this.emit("listening", { port: this.port, host: this.host });
+        resolve();
+      });
+
+      this.wss.on("error", (err) => {
+        this.emit("error", err);
+        reject(err);
+      });
+
+      this.wss.on("connection", (ws, req) => this._handleConnection(ws, req));
+    });
+  }
+
+  /** Stop the server and clean up */
+  async stop() {
+    if (this._heartbeatTimer) {
+      clearInterval(this._heartbeatTimer);
+      this._heartbeatTimer = null;
+    }
+
+    // Kill all running child processes
+    for (const [id, child] of this.processes) {
+      try {
+        child.kill("SIGTERM");
+      } catch (_err) {
+        // Process may have already exited
+      }
+      this.processes.delete(id);
+    }
+
+    // Close all client connections
+    for (const [, client] of this.clients) {
+      try {
+        client.ws.close(1001, "Server shutting down");
+      } catch (_err) {
+        // Connection may already be closed
+      }
+    }
+    this.clients.clear();
+
+    // Close the server
+    if (this.wss) {
+      await new Promise((resolve) => {
+        this.wss.close(() => resolve());
+      });
+      this.wss = null;
+    }
+
+    this.emit("stopped");
+  }
+
+  /** @private */
+  _handleConnection(ws, req) {
+    if (this.clients.size >= this.maxConnections) {
+      ws.close(1013, "Max connections reached");
+      return;
+    }
+
+    const clientId = `client-${++this._clientCounter}`;
+    const clientIp =
+      req.socket.remoteAddress || req.headers["x-forwarded-for"] || "unknown";
+
+    this.clients.set(clientId, {
+      ws,
+      authenticated: !this.token, // If no token required, auto-authenticated
+      connectedAt: Date.now(),
+      ip: clientIp,
+      alive: true,
+    });
+
+    this.emit("connection", { clientId, ip: clientIp });
+
+    ws.on("message", (data) => {
+      try {
+        const message = JSON.parse(data.toString("utf8"));
+        this._handleMessage(clientId, ws, message);
+      } catch (_err) {
+        this._send(ws, {
+          type: "error",
+          code: "INVALID_JSON",
+          message: "Failed to parse message as JSON",
+        });
+      }
+    });
+
+    ws.on("close", () => {
+      this.clients.delete(clientId);
+      this.emit("disconnection", { clientId });
+    });
+
+    ws.on("pong", () => {
+      const client = this.clients.get(clientId);
+      if (client) client.alive = true;
+    });
+  }
+
+  /** @private */
+  _handleMessage(clientId, ws, message) {
+    const { id, type } = message;
+
+    if (!id) {
+      this._send(ws, {
+        type: "error",
+        code: "MISSING_ID",
+        message: 'Message must include an "id" field',
+      });
+      return;
+    }
+
+    // Check authentication
+    const client = this.clients.get(clientId);
+    if (this.token && !client.authenticated && type !== "auth") {
+      this._send(ws, {
+        id,
+        type: "error",
+        code: "AUTH_REQUIRED",
+        message: "Authentication required. Send an auth message first.",
+      });
+      return;
+    }
+
+    switch (type) {
+      case "auth":
+        this._handleAuth(clientId, ws, message);
+        break;
+      case "ping":
+        this._send(ws, { id, type: "pong", serverTime: Date.now() });
+        break;
+      case "execute":
+        this._executeCommand(id, ws, message.command, false);
+        break;
+      case "stream":
+        this._executeCommand(id, ws, message.command, true);
+        break;
+      case "cancel":
+        this._cancelRequest(id, ws);
+        break;
+      default:
+        this._send(ws, {
+          id,
+          type: "error",
+          code: "UNKNOWN_TYPE",
+          message: `Unknown message type: ${type}`,
+        });
+    }
+  }
+
+  /** @private */
+  _handleAuth(clientId, ws, message) {
+    const { id, token } = message;
+    const success = token === this.token;
+    const client = this.clients.get(clientId);
+
+    if (success && client) {
+      client.authenticated = true;
+    }
+
+    this._send(ws, {
+      id,
+      type: "auth-result",
+      success,
+      ...(success ? {} : { message: "Invalid token" }),
+    });
+
+    if (!success) {
+      // Disconnect after failed auth
+      setTimeout(() => ws.close(4001, "Authentication failed"), 100);
+    }
+  }
+
+  /** @private */
+  _executeCommand(id, ws, command, stream) {
+    if (!command || typeof command !== "string") {
+      this._send(ws, {
+        id,
+        type: "error",
+        code: "INVALID_COMMAND",
+        message: "Command must be a non-empty string",
+      });
+      return;
+    }
+
+    const args = tokenizeCommand(command.trim());
+    if (args.length === 0) {
+      this._send(ws, {
+        id,
+        type: "error",
+        code: "INVALID_COMMAND",
+        message: "Empty command",
+      });
+      return;
+    }
+
+    // Block dangerous/interactive commands
+    const baseCmd = args[0];
+    if (BLOCKED_COMMANDS.has(baseCmd)) {
+      this._send(ws, {
+        id,
+        type: "error",
+        code: "COMMAND_BLOCKED",
+        message: `Command "${baseCmd}" cannot be executed via WebSocket (interactive or recursive)`,
+      });
+      return;
+    }
+
+    const child = spawn(process.execPath, [BIN_PATH, ...args], {
+      env: {
+        ...process.env,
+        FORCE_COLOR: "0",
+        NO_SPINNER: "1",
+      },
+      stdio: ["pipe", "pipe", "pipe"],
+      windowsHide: true,
+    });
+
+    this.processes.set(id, child);
+    this.emit("command:start", { id, command, stream });
+
+    // Timeout handling
+    const timer = setTimeout(() => {
+      if (this.processes.has(id)) {
+        try {
+          child.kill("SIGTERM");
+        } catch (_err) {
+          // Process may have already exited
+        }
+        this.processes.delete(id);
+        this._send(ws, {
+          id,
+          type: "error",
+          code: "COMMAND_TIMEOUT",
+          message: `Command timed out after ${this.timeout}ms`,
+        });
+      }
+    }, this.timeout);
+
+    if (stream) {
+      // Stream mode: send chunks as they arrive
+      child.stdout.on("data", (data) => {
+        this._send(ws, {
+          id,
+          type: "stream-data",
+          channel: "stdout",
+          data: data.toString("utf8"),
+        });
+      });
+
+      child.stderr.on("data", (data) => {
+        this._send(ws, {
+          id,
+          type: "stream-data",
+          channel: "stderr",
+          data: data.toString("utf8"),
+        });
+      });
+
+      child.on("close", (exitCode) => {
+        clearTimeout(timer);
+        this.processes.delete(id);
+        this._send(ws, {
+          id,
+          type: "stream-end",
+          exitCode: exitCode ?? 1,
+        });
+        this.emit("command:end", { id, exitCode });
+      });
+    } else {
+      // Buffered mode: collect all output then send result
+      const stdoutChunks = [];
+      const stderrChunks = [];
+
+      child.stdout.on("data", (data) => stdoutChunks.push(data));
+      child.stderr.on("data", (data) => stderrChunks.push(data));
+
+      child.on("close", (exitCode) => {
+        clearTimeout(timer);
+        this.processes.delete(id);
+
+        const stdout = Buffer.concat(stdoutChunks).toString("utf8");
+        const stderr = Buffer.concat(stderrChunks).toString("utf8");
+
+        this._send(ws, {
+          id,
+          type: "result",
+          success: exitCode === 0,
+          exitCode: exitCode ?? 1,
+          stdout,
+          stderr,
+        });
+        this.emit("command:end", { id, exitCode });
+      });
+    }
+
+    child.on("error", (err) => {
+      clearTimeout(timer);
+      this.processes.delete(id);
+      this._send(ws, {
+        id,
+        type: "error",
+        code: "SPAWN_ERROR",
+        message: err.message,
+      });
+    });
+  }
+
+  /** @private */
+  _cancelRequest(id, ws) {
+    const child = this.processes.get(id);
+    if (child) {
+      try {
+        child.kill("SIGTERM");
+      } catch (_err) {
+        // Process may have already exited
+      }
+      this.processes.delete(id);
+      this._send(ws, {
+        id,
+        type: "result",
+        success: false,
+        exitCode: -1,
+        stdout: "",
+        stderr: "Cancelled by client",
+      });
+    } else {
+      this._send(ws, {
+        id,
+        type: "error",
+        code: "NOT_FOUND",
+        message: `No running command with id "${id}"`,
+      });
+    }
+  }
+
+  /** @private — ping/pong heartbeat to detect dead connections */
+  _startHeartbeat() {
+    this._heartbeatTimer = setInterval(() => {
+      for (const [clientId, client] of this.clients) {
+        if (!client.alive) {
+          client.ws.terminate();
+          this.clients.delete(clientId);
+          this.emit("disconnection", { clientId, reason: "heartbeat timeout" });
+          continue;
+        }
+        client.alive = false;
+        try {
+          client.ws.ping();
+        } catch (_err) {
+          // Connection may be closing
+        }
+      }
+    }, HEARTBEAT_INTERVAL);
+  }
+
+  /** @private — safe JSON send */
+  _send(ws, data) {
+    if (ws.readyState === ws.OPEN) {
+      try {
+        ws.send(JSON.stringify(data));
+      } catch (_err) {
+        // Connection may have just closed
+      }
+    }
+  }
+}

package/src/repl/agent-repl.js

@@ -35,6 +35,10 @@ import {
 import { storeMemory, consolidateMemory } from "../lib/hierarchical-memory.js";
 import { CLIContextEngineering } from "../lib/cli-context-engineering.js";
 import { createChatFn } from "../lib/cowork-adapter.js";
+import {
+  detectTaskType,
+  selectModelForTask,
+} from "../lib/task-model-selector.js";
 import { executeHooks, HookEvents } from "../lib/hook-manager.js";
 import { CLIPermanentMemory } from "../lib/permanent-memory.js";
 import { CLIAutonomousAgent, GoalStatus } from "../lib/autonomous-agent.js";
@@ -565,6 +569,7 @@ async function chatWithTools(rawMessages, options) {
     dashscope: "https://dashscope.aliyuncs.com/compatible-mode/v1",
     mistral: "https://api.mistral.ai/v1",
     gemini: "https://generativelanguage.googleapis.com/v1beta/openai",
+    volcengine: "https://ark.cn-beijing.volces.com/api/v3",
   };
 
   const providerApiKeyEnvs = {
@@ -573,6 +578,7 @@ async function chatWithTools(rawMessages, options) {
     dashscope: "DASHSCOPE_API_KEY",
     mistral: "MISTRAL_API_KEY",
     gemini: "GEMINI_API_KEY",
+    volcengine: "VOLCENGINE_API_KEY",
   };
 
   const url =
@@ -582,7 +588,7 @@ async function chatWithTools(rawMessages, options) {
 
   if (!url) {
     throw new Error(
-      `Unsupported provider: ${provider}. Supported: ollama, anthropic, openai, deepseek, dashscope, mistral, gemini`,
+      `Unsupported provider: ${provider}. Supported: ollama, anthropic, openai, deepseek, dashscope, mistral, gemini, volcengine`,
     );
   }
 
@@ -596,6 +602,7 @@ async function chatWithTools(rawMessages, options) {
     dashscope: "qwen-turbo",
     mistral: "mistral-large-latest",
     gemini: "gemini-2.0-flash",
+    volcengine: "doubao-seed-1-6-251015",
   };
 
   const response = await fetch(`${url}/chat/completions`, {
@@ -769,7 +776,7 @@ export async function startAgentRepl(options = {}) {
   let model = options.model || "qwen2:7b";
   let provider = options.provider || "ollama";
   const baseUrl = options.baseUrl || "http://localhost:11434";
-  const apiKey = options.apiKey || process.env.OPENAI_API_KEY;
+  const apiKey = options.apiKey || null;
 
   // Bootstrap runtime (best-effort, DB not required)
   let db = null;
@@ -974,6 +981,7 @@ export async function startAgentRepl(options = {}) {
           "dashscope",
           "mistral",
           "gemini",
+          "volcengine",
         ];
         if (supported.includes(arg)) {
           provider = arg;
@@ -987,7 +995,7 @@ export async function startAgentRepl(options = {}) {
         logger.info(`Current provider: ${chalk.cyan(provider)}`);
         logger.info(
           chalk.gray(
-            "Available: ollama, anthropic, openai, deepseek, dashscope, mistral, gemini",
+            "Available: ollama, anthropic, openai, deepseek, dashscope, mistral, gemini, volcengine",
           ),
         );
       }
@@ -1556,11 +1564,24 @@ export async function startAgentRepl(options = {}) {
     // Add user message
     messages.push({ role: "user", content: trimmed });
 
+    // Auto-select best model based on task type
+    let activeModel = model;
+    const taskDetection = detectTaskType(trimmed);
+    if (taskDetection.confidence > 0.3) {
+      const recommended = selectModelForTask(provider, taskDetection.taskType);
+      if (recommended && recommended !== activeModel) {
+        activeModel = recommended;
+        logger.info(
+          chalk.gray(`[auto] ${taskDetection.name} → ${activeModel}`),
+        );
+      }
+    }
+
     try {
       process.stdout.write("\n");
       const response = await agentLoop(messages, {
         provider,
-        model,
+        model: activeModel,
         baseUrl,
         apiKey,
         contextEngine,

package/src/repl/chat-repl.js

@@ -11,6 +11,7 @@
 import readline from "readline";
 import chalk from "chalk";
 import { logger } from "../lib/logger.js";
+import { BUILT_IN_PROVIDERS } from "../lib/llm-providers.js";
 
 const SLASH_COMMANDS = {
   "/exit": "Exit the chat",
@@ -128,7 +129,7 @@ export async function startChatRepl(options = {}) {
   let model = options.model || "qwen2:7b";
   let provider = options.provider || "ollama";
   const baseUrl = options.baseUrl || "http://localhost:11434";
-  const apiKey = options.apiKey || process.env.OPENAI_API_KEY;
+  const apiKey = options.apiKey || null;
 
   const messages = [];
 
@@ -236,14 +237,21 @@ export async function startChatRepl(options = {}) {
 
       if (provider === "ollama") {
         response = await streamOllama(messages, model, baseUrl, onToken);
-      } else if (provider === "openai") {
+      } else {
+        // OpenAI-compatible providers (openai, volcengine, deepseek, dashscope, mistral, gemini, anthropic-proxy)
+        const providerDef = BUILT_IN_PROVIDERS[provider];
         const url =
           baseUrl !== "http://localhost:11434"
             ? baseUrl
-            : "https://api.openai.com/v1";
-        response = await streamOpenAI(messages, model, url, apiKey, onToken);
-      } else {
-        throw new Error(`Unsupported provider: ${provider}`);
+            : providerDef?.baseUrl || "https://api.openai.com/v1";
+        const key =
+          apiKey ||
+          (providerDef?.apiKeyEnv ? process.env[providerDef.apiKeyEnv] : null);
+        if (!key)
+          throw new Error(
+            `API key required for ${provider} (set ${providerDef?.apiKeyEnv || "API key"})`,
+          );
+        response = await streamOpenAI(messages, model, url, key, onToken);
       }
 
       process.stdout.write("\n\n");