@xalia/agent 0.5.8 → 0.6.1

package/src/chat/data/dbMcpServerConfigs.ts

@@ -0,0 +1,59 @@
+import { McpServerConfiguration } from "@xalia/xmcp/sdk";
+import { DbClientBase } from "./database";
+
+export class DbMcpServerConfigs extends DbClientBase {
+  async clearConfigsForUser(user_uuid: string): Promise<void> {
+    const { error } = await this.client
+      .from("user_server_configs")
+      .delete()
+      .eq("user_uuid", user_uuid);
+    if (error) {
+      throw error;
+    }
+  }
+
+  async deleteConfigForUser(
+    user_uuid: string,
+    server_name: string
+  ): Promise<void> {
+    const { error } = await this.client
+      .from("user_server_configs")
+      .delete()
+      .eq("user_uuid", user_uuid)
+      .eq("server_name", server_name);
+    if (error) {
+      throw error;
+    }
+  }
+
+  async getConfigForUser(
+    user_uuid: string,
+    server_name: string
+  ): Promise<McpServerConfiguration | undefined> {
+    const { error, data } = await this.client
+      .from("user_server_configs")
+      .select("config")
+      .eq("user_uuid", user_uuid)
+      .eq("server_name", server_name)
+      .maybeSingle();
+    if (error) {
+      throw error;
+    }
+
+    return data ? (data.config as McpServerConfiguration) : undefined;
+  }
+
+  async setConfigForUser(
+    user_uuid: string,
+    server_name: string,
+    config: McpServerConfiguration
+  ): Promise<void> {
+    const payload = { user_uuid, server_name, config };
+    const { error } = await this.client
+      .from("user_server_configs")
+      .upsert([payload], { onConflict: "user_uuid,server_name" });
+    if (error) {
+      throw error;
+    }
+  }
+}

package/src/chat/protocol/connectionMessages.ts

@@ -0,0 +1,49 @@
+//
+// Connection-level Client Messages
+//
+
+export type ClientConnectionData<T> = {
+  t: "data";
+  d: T;
+};
+
+export type ClientConnectionPong = {
+  t: "pong";
+};
+
+export type ClientConnectionMessage<T> =
+  | ClientConnectionData<T> // TODO: generic
+  | ClientConnectionPong;
+
+//
+// Connection-Level Server Messages (Transport)
+//
+
+/**
+ * Response to connection init
+ */
+export type ServerConnectionReady = {
+  t: "ready";
+  c_id: string; // TODO: remove
+  user_uuid: string;
+};
+
+export type ServerConnectionError = {
+  t: "error";
+  e: string;
+};
+
+export type ServerConnectionPing = {
+  t: "ping";
+};
+
+export type ServerConnectionData<T> = {
+  t: "data";
+  d: T;
+};
+
+export type ServerConnectionMessage<T> =
+  | ServerConnectionReady
+  | ServerConnectionError
+  | ServerConnectionPing
+  | ServerConnectionData<T>;

package/src/chat/protocol/constants.ts

@@ -0,0 +1,55 @@
+/**
+ * Protocol constants for the multi-session chat system
+ */
+
+/**
+ * Current version of the agent package
+ */
+export const AGENT_VERSION = "0.5.9";
+
+/**
+ * Protocol version for WebSocket communication
+ */
+export const PROTOCOL_VERSION = AGENT_VERSION;
+
+/**
+ * Client version sent in connection handshake
+ */
+export const CLIENT_VERSION = AGENT_VERSION;
+
+/**
+ * Server version sent in connection handshake responses
+ */
+export const SERVER_VERSION = AGENT_VERSION;
+
+/**
+ * Default message queue limits
+ */
+export const DEFAULT_MESSAGE_QUEUE_SIZE = 100;
+export const DEFAULT_USER_MESSAGE_QUEUE_SIZE = 10;
+
+/**
+ * Connection timeouts (in milliseconds)
+ */
+export const CONNECTION_READY_TIMEOUT = 2000; // 2 seconds
+export const MESSAGE_PROCESSING_TIMEOUT = 10000; // 10 seconds
+
+/**
+ * WebSocket close codes used by the chat system
+ * see: https://datatracker.ietf.org/doc/html/rfc6455#section-7.4.1
+ */
+export const CHAT_CLOSE_CODES = {
+  NORMAL_CLOSURE: 1000,
+  GOING_AWAY: 1001,
+  PROTOCOL_ERROR: 1002,
+  INVALID_DATA: 1007,
+  // 4000-4999: application-defined error codes
+  AUTH_FAILED: 4003,
+  SESSION_ERROR: 4402,
+  RATE_LIMITED: 4403,
+} as const;
+
+/**
+ * Maximum message size in bytes
+ */
+export const MAX_MESSAGE_SIZE = 64 * 1024; // 64KB

package/src/chat/protocol/errors.ts

@@ -0,0 +1,16 @@
+/**
+ * An error resulting the the connection being closed
+ */
+export class ChatFatalError extends Error {}
+
+/**
+ * An error to be reported, but not resulting in termination of the
+ * connection.
+ */
+export class ChatErrorMessage extends Error {}
+
+export class UserAlreadyConnected extends ChatFatalError {
+  constructor() {
+    super("User already connected to the conversation");
+  }
+}