opencandle 0.11.1 → 0.12.0

package/gui/server/ws-hub.ts

@@ -1,403 +0,0 @@
-import type { IncomingMessage } from "node:http";
-import type { Duplex } from "node:stream";
-import {
-  type AgentSession,
-  type SessionEntry,
-  SessionManager,
-} from "@earendil-works/pi-coding-agent";
-import { probeProviderStatus } from "../../src/onboarding/provider-status.js";
-import { getProvider, type ProviderId } from "../../src/onboarding/providers.js";
-import {
-  clearProviderOnboardingEntry,
-  loadOnboardingState,
-  saveOnboardingState,
-} from "../../src/onboarding/state.js";
-import type { ChatEvent } from "../shared/chat-events.js";
-import type { BackgroundQuoteRefreshes } from "./background-quotes.js";
-import { sessionEntriesToChatEvents } from "./chat-event-adapter.js";
-import type { ToolInvokeController } from "./invoke-tool.js";
-import { getSavedMarketStateSymbols } from "./market-state-api.js";
-import type { ModelSetupController } from "./model-setup.js";
-import { projectDashboard } from "./projector.js";
-import type { SessionActionsController } from "./session-actions.js";
-import { buildCatalog, setToolEnabled } from "./tool-metadata.js";
-import { acceptWebSocket, type WsClient } from "./websocket.js";
-import { readWriterLock, writerLockScopeForSession } from "./writer-lock.js";
-
-interface AskUserBridge {
-  getPrompts(): unknown[];
-}
-
-export interface WsHub {
-  handleUpgrade(req: IncomingMessage, socket: Duplex): void;
-  getClientCount(): number;
-  closeClients(): void;
-  sendBoot(client: WsClient): void;
-  buildBootstrapPayload(): Promise<Record<string, unknown>>;
-  broadcast(message: unknown): void;
-  broadcastModelSetup(): void;
-  broadcastState(): void;
-  broadcastSessionSnapshot(sessionManager: SessionManager): void;
-  broadcastSessions(): void;
-  buildStateSnapshot(): {
-    sessionId: string;
-    state: unknown;
-    entries: SessionEntry[];
-    events: ChatEvent[];
-  };
-  currentChatEvents(entries?: SessionEntry[]): ChatEvent[];
-  subscribeToSessionEvents(): () => void;
-}
-
-export interface WsHubOptions {
-  role: string;
-  lock: unknown;
-  cwd: string;
-  sessionDir: string;
-  getSession: () => AgentSession;
-  getSessionManager: () => SessionManager;
-  backgroundQuoteRefreshes: BackgroundQuoteRefreshes;
-  askUserBridge: AskUserBridge;
-  modelSetupController: ModelSetupController;
-  toolInvokeController: ToolInvokeController;
-  sessionActionsController: SessionActionsController;
-  onClientCountChanged: () => void;
-  isTrustedRequest?: (req: IncomingMessage) => boolean;
-  acceptWebSocketFn?: typeof acceptWebSocket;
-}
-
-export function createWsHub({
-  role,
-  lock,
-  cwd,
-  sessionDir,
-  getSession,
-  getSessionManager,
-  backgroundQuoteRefreshes,
-  askUserBridge,
-  modelSetupController,
-  toolInvokeController,
-  sessionActionsController,
-  onClientCountChanged,
-  isTrustedRequest = () => true,
-  acceptWebSocketFn = acceptWebSocket,
-}: WsHubOptions): WsHub {
-  const clients = new Set<WsClient>();
-
-  function handleUpgrade(req: IncomingMessage, socket: Duplex): void {
-    if (req.url !== "/ws") {
-      socket.destroy();
-      return;
-    }
-    if (!isTrustedRequest(req)) {
-      socket.destroy();
-      return;
-    }
-
-    const client = acceptWebSocketFn(req, socket);
-    clients.add(client);
-    client.onClose(() => {
-      clients.delete(client);
-      onClientCountChanged();
-    });
-    client.onMessage((message) => void handleClientMessage(client, message));
-    sendBoot(client);
-    onClientCountChanged();
-  }
-
-  async function handleClientMessage(client: WsClient, message: unknown): Promise<void> {
-    const data = asRecord(message);
-    try {
-      switch (data.type) {
-        case "chat.prompt":
-          // Legacy implicit-active-session mutation; parity with the HTTP
-          // route's 410. Chat runs go through the session-addressed API.
-          throw new Error(
-            "Legacy active-session chat prompts are no longer supported. Use the session-addressed chat-run API.",
-          );
-        case "ask_user.answer":
-          await sessionActionsController.handleAskUserAnswer(String(data.id ?? ""), data.answer, {
-            actionId: String(data.actionId ?? ""),
-            sessionId: String(data.sessionId ?? ""),
-            source: "browser",
-          });
-          break;
-        case "ask_user.cancel":
-          await sessionActionsController.handleAskUserCancel(String(data.id ?? ""), {
-            actionId: String(data.actionId ?? ""),
-            sessionId: String(data.sessionId ?? ""),
-            source: "browser",
-          });
-          break;
-        case "tool.invoke":
-          await toolInvokeController.handleToolInvokeMessage(client, data);
-          break;
-        case "tool.enabled":
-          if (role !== "writer") throw new Error("Read-only follower mode");
-          setToolEnabled(String(data.toolName), Boolean(data.enabled));
-          broadcast({ type: "catalog", catalog: buildCatalog(), restartRequired: true });
-          break;
-        case "catalog.refresh":
-          client.send({ type: "catalog", catalog: buildCatalog() });
-          break;
-        case "model.setup.refresh":
-          getSession().modelRegistry.refresh();
-          broadcastModelSetup();
-          break;
-        case "model.setup.save_api_key":
-          await modelSetupController.handleSaveModelApiKey(
-            String(data.provider ?? ""),
-            String(data.apiKey ?? ""),
-          );
-          broadcastModelSetup();
-          break;
-        case "model.setup.select_model":
-          await modelSetupController.handleSelectModel(
-            String(data.provider ?? ""),
-            String(data.modelId ?? ""),
-          );
-          broadcastModelSetup();
-          break;
-        case "provider.save_api_key":
-          await modelSetupController.handleSaveProviderApiKey(
-            String(data.providerId ?? ""),
-            String(data.apiKey ?? ""),
-          );
-          broadcast({ type: "catalog", catalog: buildCatalog() });
-          break;
-        case "provider.status.check": {
-          const providerId = String(data.providerId ?? "") as ProviderId;
-          getProvider(providerId);
-          const mode = data.mode === "session" ? "session" : "install";
-          if (data.reenable === true) {
-            saveOnboardingState(clearProviderOnboardingEntry(loadOnboardingState(), providerId));
-            broadcast({ type: "catalog", catalog: buildCatalog() });
-          }
-          const status = await probeProviderStatus(providerId, { mode, force: true });
-          client.send({ type: "provider.status", providerId, status });
-          break;
-        }
-        case "session.new":
-          await sessionActionsController.handleNewSession();
-          sendBoot(client);
-          broadcastState();
-          broadcastSessions();
-          break;
-        case "session.open":
-          await sessionActionsController.handleOpenSession(String(data.path ?? ""));
-          sendBoot(client);
-          broadcastState();
-          broadcastSessions();
-          break;
-        case "session.rename":
-          await sessionActionsController.handleRenameSession(
-            String(data.path ?? ""),
-            String(data.name ?? ""),
-          );
-          broadcastState();
-          broadcastSessions();
-          break;
-        case "session.delete":
-          await sessionActionsController.handleDeleteSession(client, String(data.path ?? ""));
-          break;
-      }
-    } catch (error) {
-      const errorMessage = error instanceof Error ? error.message : String(error);
-      client.send({
-        type: "error",
-        message:
-          errorMessage === "Read-only follower mode"
-            ? "OpenCandle is reconnecting to this session."
-            : errorMessage,
-      });
-    }
-  }
-
-  function getClientCount(): number {
-    return clients.size;
-  }
-
-  function closeClients(): void {
-    for (const client of clients) client.close();
-    clients.clear();
-  }
-
-  function sendBoot(client: WsClient): void {
-    const snapshot = buildStateSnapshot();
-    const sessionManager = getSessionManager();
-    client.send({
-      type: "boot",
-      role,
-      lock: publicWriterLock(lock),
-      sessionId: sessionManager.getSessionId(),
-      coordination: coordinationStateForSession(sessionManager, role, lock),
-      catalog: buildCatalog(),
-      modelSetup: modelSetupController.buildCurrentModelSetupState(),
-      askUserPrompts: askUserBridge.getPrompts(),
-    });
-    client.send({
-      type: "state.snapshot",
-      ...snapshot,
-    });
-    void SessionManager.list(cwd, sessionDir).then((sessions) =>
-      client.send({ type: "sessions", sessions }),
-    );
-  }
-
-  async function buildBootstrapPayload(): Promise<Record<string, unknown>> {
-    const sessionManager = getSessionManager();
-    return {
-      role,
-      sessionId: sessionManager.getSessionId(),
-      coordination: coordinationStateForSession(sessionManager, role, lock),
-      catalog: buildCatalog(),
-      modelSetup: modelSetupController.buildCurrentModelSetupState(),
-      askUserPrompts: askUserBridge.getPrompts(),
-      sessions: await SessionManager.list(cwd, sessionDir),
-      snapshot: buildStateSnapshot(),
-    };
-  }
-
-  function broadcastModelSetup(): void {
-    broadcast({
-      type: "model.setup",
-      modelSetup: modelSetupController.buildCurrentModelSetupState(),
-    });
-  }
-
-  function broadcastState(): void {
-    broadcast({
-      type: "state.snapshot",
-      ...buildStateSnapshot(),
-    });
-  }
-
-  function broadcastSessionSnapshot(sessionManager: SessionManager): void {
-    broadcast({
-      type: "session.snapshot",
-      ...buildSnapshotForSession(sessionManager),
-    });
-  }
-
-  function buildStateSnapshot() {
-    const sessionManager = getSessionManager();
-    const sessionId = sessionManager.getSessionId();
-    const entries = sessionManager.getEntries();
-    return {
-      sessionId,
-      state: projectDashboard(
-        backgroundQuoteRefreshes.withEntries(entries),
-        sessionId,
-        getSavedMarketStateSymbols(),
-      ),
-      entries,
-      events: currentChatEvents(entries),
-      // Ownership can change mid-run (e.g. a TUI takes over the session), so
-      // snapshots re-derive coordination instead of relying on boot state.
-      coordination: coordinationStateForSession(sessionManager, role, lock),
-    };
-  }
-
-  function buildSnapshotForSession(sessionManager: SessionManager) {
-    const sessionId = sessionManager.getSessionId();
-    const entries = sessionManager.getEntries();
-    return {
-      sessionId,
-      state: projectDashboard(
-        backgroundQuoteRefreshes.withEntries(entries),
-        sessionId,
-        getSavedMarketStateSymbols(),
-      ),
-      entries,
-      events: sessionEntriesToChatEvents(entries, {
-        sessionId,
-        title: sessionManager.getSessionName(),
-      }),
-    };
-  }
-
-  function currentChatEvents(entries = getSessionManager().getEntries()): ChatEvent[] {
-    const sessionManager = getSessionManager();
-    const session = getSession();
-    return sessionEntriesToChatEvents(entries, {
-      sessionId: sessionManager.getSessionId(),
-      title: sessionManager.getSessionName(),
-      markUnresolvedToolCalls: !(session.isStreaming || session.pendingMessageCount > 0),
-    });
-  }
-
-  function broadcastSessions(): void {
-    void SessionManager.list(cwd, sessionDir).then((sessions) =>
-      broadcast({ type: "sessions", sessions }),
-    );
-  }
-
-  function broadcast(message: unknown): void {
-    for (const client of clients) client.send(message);
-  }
-
-  function subscribeToSessionEvents(): () => void {
-    return getSession().subscribe((event) => {
-      broadcast({ type: "session.event", event });
-      broadcastState();
-    });
-  }
-
-  return {
-    handleUpgrade,
-    getClientCount,
-    closeClients,
-    sendBoot,
-    buildBootstrapPayload,
-    broadcast,
-    broadcastModelSetup,
-    broadcastState,
-    broadcastSessionSnapshot,
-    broadcastSessions,
-    buildStateSnapshot,
-    currentChatEvents,
-    subscribeToSessionEvents,
-  };
-}
-
-function coordinationStateForSession(
-  sessionManager: SessionManager,
-  role: string,
-  fallbackLock: unknown,
-) {
-  let sessionLock: unknown = fallbackLock;
-  try {
-    sessionLock = readWriterLock(writerLockScopeForSession(sessionManager)) ?? fallbackLock;
-  } catch {
-    sessionLock = fallbackLock;
-  }
-  const sessionId = sessionManager.getSessionId();
-  const publicLock = asRecord(publicWriterLock(sessionLock));
-  const ownedByThisProcess = publicLock.pid === process.pid;
-  const status =
-    role === "writer" && (!publicLock.pid || ownedByThisProcess)
-      ? "ready"
-      : role === "connecting"
-        ? "connecting"
-        : role === "disconnected"
-          ? "reconnecting"
-          : "syncing";
-  return {
-    sessionId,
-    status,
-    ...(typeof publicLock.processKind === "string" ? { ownerKind: publicLock.processKind } : {}),
-  };
-}
-
-function publicWriterLock(lock: unknown): unknown {
-  const record = asRecord(lock);
-  if (Object.keys(record).length === 0) return lock;
-  const { coordinatorSecret: _coordinatorSecret, ...publicLock } = record;
-  return publicLock;
-}
-
-function asRecord(value: unknown): Record<string, unknown> {
-  return typeof value === "object" && value !== null && !Array.isArray(value)
-    ? (value as Record<string, unknown>)
-    : {};
-}

package/gui/shared/chat-events.ts

@@ -1,165 +0,0 @@
-export type ChatRole = "user" | "assistant" | "system";
-
-export interface MessageTextContent {
-  type: "text";
-  text: string;
-}
-
-export interface MessageToolContent {
-  type: "tool";
-  toolCallId: string;
-}
-
-export interface MessageImageContent {
-  type: "image";
-  url?: string;
-  data?: string;
-  mimeType?: string;
-  alt?: string;
-}
-
-export type MessageContent = MessageTextContent | MessageToolContent | MessageImageContent;
-
-export interface MessageAttachmentChip {
-  kind: string;
-  label: string;
-}
-
-export interface ToolOutput {
-  content: Array<{ type: string; text?: string; data?: string; mimeType?: string }>;
-  details?: unknown;
-  isError?: boolean;
-  source?: string;
-}
-
-export interface ToolError {
-  message: string;
-  code?: string;
-  details?: unknown;
-}
-
-export interface RunError {
-  message: string;
-  code?: string;
-  details?: unknown;
-}
-
-export interface Usage {
-  input?: number;
-  output?: number;
-  totalTokens?: number;
-  cost?: unknown;
-}
-
-interface ChatEventBase {
-  sessionId: string;
-  seq: number;
-}
-
-interface RunScopedChatEventBase extends ChatEventBase {
-  runId: string;
-}
-
-export type ChatEvent =
-  | ({ type: "run.started" } & RunScopedChatEventBase)
-  | ({ type: "thinking.delta"; text: string } & RunScopedChatEventBase)
-  | ({ type: "thinking.completed"; text?: string } & RunScopedChatEventBase)
-  | ({ type: "message.created"; messageId: string; role: ChatRole; runId?: string } & ChatEventBase)
-  | ({ type: "message.delta"; messageId: string; text: string; runId?: string } & ChatEventBase)
-  | ({
-      type: "message.completed";
-      messageId: string;
-      content: MessageContent[];
-      attachments?: MessageAttachmentChip[];
-      runId?: string;
-    } & ChatEventBase)
-  | ({
-      type: "custom.message";
-      messageId: string;
-      customType: string;
-      content: MessageContent[];
-    } & ChatEventBase)
-  | ({
-      type: "tool.started";
-      toolCallId: string;
-      messageId: string;
-      name: string;
-      input: unknown;
-      runId?: string;
-    } & ChatEventBase)
-  | ({ type: "tool.delta"; toolCallId: string; chunk: unknown; runId?: string } & ChatEventBase)
-  | ({
-      type: "tool.completed";
-      toolCallId: string;
-      output: ToolOutput;
-      runId?: string;
-    } & ChatEventBase)
-  | ({ type: "tool.failed"; toolCallId: string; error: ToolError; runId?: string } & ChatEventBase)
-  | ({ type: "run.completed"; usage?: Usage } & RunScopedChatEventBase)
-  | ({ type: "run.failed"; error: RunError } & RunScopedChatEventBase)
-  | ({ type: "session.updated"; title?: string; updatedAt: string } & ChatEventBase);
-
-export interface RenderMessage {
-  id: string;
-  sessionId?: string;
-  role: ChatRole;
-  status: "streaming" | "completed";
-  content: MessageContent[];
-  text: string;
-  attachments?: MessageAttachmentChip[];
-  customType?: string;
-}
-
-export interface RenderToolCall {
-  id: string;
-  sessionId?: string;
-  messageId: string;
-  name: string;
-  input: unknown;
-  status: "queued" | "running" | "completed" | "failed";
-  chunks: unknown[];
-  output?: ToolOutput;
-  error?: ToolError;
-}
-
-export interface RenderThinking {
-  runId: string;
-  sessionId?: string;
-  status: "streaming" | "completed";
-  text: string;
-}
-
-export interface ChatRunState {
-  id: string;
-  sessionId?: string;
-  status: "running" | "completed" | "failed";
-  usage?: Usage;
-  error?: RunError;
-}
-
-export interface ChatRenderState {
-  lastSeq: number;
-  seenSeq: Set<string>;
-  lastSeqBySession: Map<string, number>;
-  messages: RenderMessage[];
-  messageById: Map<string, RenderMessage>;
-  tools: Map<string, RenderToolCall>;
-  runs: Map<string, ChatRunState>;
-  thinking: Map<string, RenderThinking>;
-  session?: { id: string; title?: string; updatedAt?: string };
-  gaps: Array<{ expected: number; received: number }>;
-}
-
-export function createChatRenderState(): ChatRenderState {
-  return {
-    lastSeq: 0,
-    seenSeq: new Set(),
-    lastSeqBySession: new Map(),
-    messages: [],
-    messageById: new Map(),
-    tools: new Map(),
-    runs: new Map(),
-    thinking: new Map(),
-    gaps: [],
-  };
-}