Method handleWsConnection

src/runtime/adapters/web.ts:477–534

(
    ws: WebSocket,
    channelId: string,
    agent: IPackAgent,
  )

  // -------------------------------------------------------------------------

  private handleWsConnection(
    ws: WebSocket,
    channelId: string,
    agent: IPackAgent,
  ): void {
    this.addSocket(channelId, ws);

    ws.on("message", async (data) => {
      try {
        const payload = JSON.parse(data.toString());
        if (!payload.text) return;

        const text: string = payload.text;

        // Check for bot commands
        const command = parseCommand(text);
        if (command) {
          const result = await agent.handleCommand(command, channelId);

          // sendWsEvent(ws, { type: "agent_start" });
          // sendWsEvent(ws, { type: "message_start", role: "assistant" });

          if (result.message) {
            sendWsEvent(ws, { type: "text_delta", delta: result.message });
          }

          // sendWsEvent(ws, { type: "message_end", role: "assistant" });
          // sendWsEvent(ws, { type: "agent_end" });
          ws.send(JSON.stringify({ done: true }));
          return;
        }

        // Regular message → stream events via WebSocket
        const onEvent = (event: AgentEvent) => {
          sendWsEvent(ws, event);
          this.ipcBroadcaster?.broadcastAgentEvent(channelId, event);
        };

        const result = await agent.handleMessage("web", channelId, text, onEvent);

        if (result.errorMessage) {
          ws.send(JSON.stringify({ error: result.errorMessage }));
          return;
        }

        ws.send(JSON.stringify({ done: true }));
      } catch (err) {
        ws.send(JSON.stringify({ error: String(err) }));
      }
    });

    ws.on("close", () => {
      this.removeSocket(channelId, ws);
      if (channelId !== DEFAULT_WEB_CHANNEL_ID) {
        agent.dispose(channelId);
      }
    });
  }