@delego/runner 0.1.0

package/README.md

@@ -0,0 +1,64 @@
+# @delego/sdk
+
+Delego SDK for registering and executing actions.
+
+## Installation
+
+```bash
+npm install @delego/sdk
+```
+
+## Usage
+
+```typescript
+import { DelegoClient } from '@delego/sdk';
+
+const client = new DelegoClient({
+  apiKey: process.env.DELEGO_API_KEY!,
+  baseUrl: 'http://localhost:4000', // optional, defaults to localhost:4000
+  runnerName: 'my-runner', // optional, defaults to hostname
+});
+
+// Register an action
+client.registerAction({
+  name: 'reset_password',
+  description: 'Reset a user password',
+  parameters: [
+    {
+      name: 'email',
+      type: 'string',
+      description: 'User email',
+      required: true,
+    },
+  ],
+  handler: async ({ email }) => {
+    // Your handler logic here
+    return { success: true };
+  },
+});
+
+// Connect to Delego (registers actions and opens WebSocket)
+await client.connect();
+```
+
+## API Reference
+
+### `DelegoClient`
+
+Main client class for interacting with Delego.
+
+#### Constructor
+
+```typescript
+new DelegoClient(config: DelegoConfig)
+```
+
+#### Methods
+
+- `registerAction(action: Action): void` - Register an action with a handler
+- `connect(): Promise<void>` - Connect to Delego and register actions
+- `disconnect(): Promise<void>` - Disconnect from Delego
+
+## Types
+
+See `src/types.ts` for full type definitions.

package/dist/client.d.ts

@@ -0,0 +1,28 @@
+import { Action, DelegoConfig } from './types';
+export declare class DelegoClient {
+    private runnerId;
+    private runnerIdPromise;
+    private config;
+    private actions;
+    private connected;
+    private encoreClient;
+    private actionIds;
+    private wsConnection;
+    constructor(config: DelegoConfig);
+    /**
+     * Get the runner ID, ensuring it's initialized if needed
+     */
+    getRunnerId(): Promise<string>;
+    registerAction(action: Action): void;
+    getRegisteredActions(): Action[];
+    getHandler(actionName: string): Action['handler'] | undefined;
+    connect(): Promise<void>;
+    /**
+     * Validates if user has required roles to execute an action.
+     * Returns true if action is unrestricted (no requiredRoles) or user has at least one required role.
+     */
+    private validateRoles;
+    private handleInvocation;
+    private connectWebSocket;
+    disconnect(): Promise<void>;
+}

package/dist/client.js

@@ -0,0 +1,242 @@
+import os from 'os';
+import Client from './generated-client';
+import { WebSocketConnection } from './connection';
+import { getOrCreateRunnerId } from './runner-id';
+export class DelegoClient {
+    runnerId = null;
+    runnerIdPromise = null;
+    config;
+    actions = new Map();
+    connected = false;
+    encoreClient = null;
+    actionIds = [];
+    wsConnection = null;
+    constructor(config) {
+        // If runnerId is explicitly provided, use it immediately
+        if (config.runnerId) {
+            this.runnerId = config.runnerId;
+        }
+        else {
+            // Otherwise, initialize async (will be resolved when first accessed)
+            this.runnerIdPromise = getOrCreateRunnerId().then((id) => {
+                this.runnerId = id;
+                return id;
+            });
+        }
+        this.config = {
+            apiKey: config.apiKey,
+            baseUrl: config.baseUrl || 'https://production-delego-backend-trui.encr.app',
+            runnerName: config.runnerName || os.hostname(),
+            debug: config.debug ?? false,
+            emailRoles: config.emailRoles,
+        };
+    }
+    /**
+     * Get the runner ID, ensuring it's initialized if needed
+     */
+    async getRunnerId() {
+        if (this.runnerId !== null) {
+            return this.runnerId;
+        }
+        if (this.runnerIdPromise) {
+            return await this.runnerIdPromise;
+        }
+        // Fallback (shouldn't happen)
+        this.runnerIdPromise = getOrCreateRunnerId().then((id) => {
+            this.runnerId = id;
+            return id;
+        });
+        return await this.runnerIdPromise;
+    }
+    registerAction(action) {
+        // Validate required fields
+        if (!action.name) {
+            throw new Error('Action name is required');
+        }
+        if (!action.description) {
+            throw new Error('Action description is required');
+        }
+        if (!action.handler || typeof action.handler !== 'function') {
+            throw new Error('Action handler must be a function');
+        }
+        // Check for duplicates
+        if (this.actions.has(action.name)) {
+            throw new Error(`Action "${action.name}" already registered`);
+        }
+        // Store action
+        this.actions.set(action.name, action);
+    }
+    getRegisteredActions() {
+        return Array.from(this.actions.values());
+    }
+    getHandler(actionName) {
+        return this.actions.get(actionName)?.handler;
+    }
+    async connect() {
+        if (this.connected) {
+            throw new Error('Already connected');
+        }
+        if (this.actions.size === 0) {
+            throw new Error('No actions registered. Call registerAction() first.');
+        }
+        // Ensure runner ID is initialized before connecting
+        const runnerId = await this.getRunnerId();
+        // Create Encore client instance
+        this.encoreClient = new Client(this.config.baseUrl);
+        try {
+            // 1. Establish WebSocket connection first (this creates the runner record)
+            await this.connectWebSocket();
+            // 2. Register actions via REST using generated client (runner now exists)
+            const actions = Array.from(this.actions.values()).map((a) => ({
+                name: a.name,
+                description: a.description,
+                parameters: a.parameters || [],
+                requiredRoles: a.requiredRoles,
+            }));
+            // Set up timeout for registration request
+            const timeoutPromise = new Promise((_, reject) => {
+                setTimeout(() => reject(new Error('Registration request timed out after 30 seconds')), 30000);
+            });
+            try {
+                const result = await Promise.race([
+                    this.encoreClient.core.registerActions({
+                        runnerId,
+                        actions,
+                        apiKey: this.config.apiKey,
+                        emailRoles: this.config.emailRoles,
+                    }),
+                    timeoutPromise,
+                ]);
+                // Store action IDs if needed
+                this.actionIds = result.actionIds;
+            }
+            catch (error) {
+                // Check if it's a timeout error
+                if (error.message === 'Registration request timed out after 30 seconds') {
+                    throw error;
+                }
+                // Handle API errors
+                if (error.status) {
+                    if (error.status === 401) {
+                        throw new Error('Invalid API key');
+                    }
+                    else if (error.status === 404) {
+                        throw new Error(`Runner ${this.runnerId} not found`);
+                    }
+                    else if (error.status === 403) {
+                        throw new Error('Permission denied: Runner does not belong to tenant');
+                    }
+                    else if (error.status === 400) {
+                        throw new Error(`Invalid request: ${error.message || 'Bad request'}`);
+                    }
+                    else {
+                        throw new Error(`Registration failed: ${error.message || `HTTP ${error.status}`}`);
+                    }
+                }
+                throw new Error(`Failed to register actions: ${error.message || String(error)}`);
+            }
+            this.connected = true;
+            console.log(`Connected to Delego as runner ${runnerId}`);
+        }
+        catch (error) {
+            // Clean up on failure
+            this.connected = false;
+            if (this.wsConnection) {
+                this.wsConnection.disconnect();
+                this.wsConnection = null;
+            }
+            throw error;
+        }
+    }
+    /**
+     * Validates if user has required roles to execute an action.
+     * Returns true if action is unrestricted (no requiredRoles) or user has at least one required role.
+     */
+    validateRoles(action, userContext) {
+        console.log('validateRoles', action, userContext);
+        // No required roles = unrestricted (allow all users)
+        if (!action.requiredRoles || action.requiredRoles.length === 0) {
+            return true;
+        }
+        // Get user's roles from email mapping
+        const userEmail = userContext?.email;
+        if (!userEmail) {
+            // User has no email = no roles
+            return false;
+        }
+        const userRoles = this.config.emailRoles?.[userEmail] ?? [];
+        // User needs at least one of the required roles (OR logic)
+        return action.requiredRoles.some((role) => userRoles.includes(role));
+    }
+    async handleInvocation(invocationId, actionName, parameters, userContext) {
+        const action = this.actions.get(actionName);
+        if (!action) {
+            // Unknown action - send error response immediately
+            await this.wsConnection?.sendResult(invocationId, false, undefined, `Unknown action: ${actionName}`);
+            return;
+        }
+        // Validate roles before executing handler
+        if (!this.validateRoles(action, userContext)) {
+            // User is unauthorized - send special error format
+            const errorMessage = JSON.stringify({
+                success: false,
+                error: 'UNAUTHORIZED',
+                message: 'You do not have permission to execute this action',
+                requiredRoles: action.requiredRoles,
+            });
+            await this.wsConnection?.sendResult(invocationId, false, undefined, errorMessage);
+            return;
+        }
+        try {
+            // Construct InvocationContext - user context defaults to empty if not provided
+            const context = {
+                invocationId,
+                parameters,
+                user: userContext || {
+                    slackUserId: '',
+                    groupIds: [],
+                },
+            };
+            console.log(`Executing action: ${actionName}`, context);
+            const result = await action.handler(context);
+            await this.wsConnection?.sendResult(invocationId, true, result);
+            console.log(`Action completed: ${actionName}`, result);
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            console.error(`Action failed: ${actionName}`, errorMessage);
+            await this.wsConnection?.sendResult(invocationId, false, undefined, errorMessage);
+        }
+    }
+    async connectWebSocket() {
+        // Ensure runner ID is initialized
+        const runnerId = await this.getRunnerId();
+        const connectionOptions = {
+            baseUrl: this.config.baseUrl,
+            apiKey: this.config.apiKey,
+            runnerId,
+            runnerName: this.config.runnerName,
+            debug: this.config.debug,
+            onInvoke: (invocationId, actionName, params, userContext) => {
+                // Handle async without blocking - fire and forget pattern
+                this.handleInvocation(invocationId, actionName, params, userContext).catch((err) => {
+                    console.error('Unexpected error handling invocation:', err);
+                });
+            },
+            onDisconnect: () => {
+                this.connected = false;
+                console.log('Disconnected from Delego');
+            },
+        };
+        this.wsConnection = new WebSocketConnection(connectionOptions);
+        await this.wsConnection.connect();
+    }
+    async disconnect() {
+        if (this.wsConnection) {
+            this.wsConnection.disconnect();
+            this.wsConnection = null;
+        }
+        this.connected = false;
+    }
+}
+//# sourceMappingURL=client.js.map

package/dist/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,mBAAmB,EAAqB,MAAM,cAAc,CAAC;AACtE,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAElD,MAAM,OAAO,YAAY;IACf,QAAQ,GAAkB,IAAI,CAAC;IAC/B,eAAe,GAA2B,IAAI,CAAC;IAC/C,MAAM,CAOZ;IACM,OAAO,GAAwB,IAAI,GAAG,EAAE,CAAC;IACzC,SAAS,GAAY,KAAK,CAAC;IAC3B,YAAY,GAAkB,IAAI,CAAC;IACnC,SAAS,GAAa,EAAE,CAAC;IACzB,YAAY,GAA+B,IAAI,CAAC;IAExD,YAAY,MAAoB;QAC9B,yDAAyD;QACzD,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACpB,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAClC,CAAC;aAAM,CAAC;YACN,qEAAqE;YACrE,IAAI,CAAC,eAAe,GAAG,mBAAmB,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE;gBACvD,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;gBACnB,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,MAAM,GAAG;YACZ,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,iDAAiD;YAC5E,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC,QAAQ,EAAE;YAC9C,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,KAAK;YAC5B,UAAU,EAAE,MAAM,CAAC,UAAU;SAC9B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW;QACf,IAAI,IAAI,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;YAC3B,OAAO,IAAI,CAAC,QAAQ,CAAC;QACvB,CAAC;QACD,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,OAAO,MAAM,IAAI,CAAC,eAAe,CAAC;QACpC,CAAC;QACD,8BAA8B;QAC9B,IAAI,CAAC,eAAe,GAAG,mBAAmB,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE;YACvD,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;YACnB,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,IAAI,CAAC,eAAe,CAAC;IACpC,CAAC;IAED,cAAc,CAAC,MAAc;QAC3B,2BAA2B;QAC3B,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,OAAO,MAAM,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;YAC5D,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,uBAAuB;QACvB,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,WAAW,MAAM,CAAC,IAAI,sBAAsB,CAAC,CAAC;QAChE,CAAC;QAED,eAAe;QACf,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACxC,CAAC;IAED,oBAAoB;QAClB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,UAAU,CAAC,UAAkB;QAC3B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,OAAO,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACzE,CAAC;QAED,oDAAoD;QACpD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QAE1C,gCAAgC;QAChC,IAAI,CAAC,YAAY,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAEpD,IAAI,CAAC;YACH,2EAA2E;YAC3E,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAE9B,0EAA0E;YAC1E,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC5D,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;gBAC1B,UAAU,EAAE,CAAC,CAAC,UAAU,IAAI,EAAE;gBAC9B,aAAa,EAAE,CAAC,CAAC,aAAa;aAC/B,CAAC,CAAC,CAAC;YAEJ,0CAA0C;YAC1C,MAAM,cAAc,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;gBACtD,UAAU,CACR,GAAG,EAAE,CACH,MAAM,CACJ,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAC7D,EACH,KAAK,CACN,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;oBAChC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,eAAe,CAAC;wBACrC,QAAQ;wBACR,OAAO;wBACP,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;wBAC1B,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU;qBACnC,CAAC;oBACF,cAAc;iBACf,CAAC,CAAC;gBAEH,6BAA6B;gBAC7B,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;YACpC,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,gCAAgC;gBAChC,IACE,KAAK,CAAC,OAAO,KAAK,iDAAiD,EACnE,CAAC;oBACD,MAAM,KAAK,CAAC;gBACd,CAAC;gBAED,oBAAoB;gBACpB,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;oBACjB,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;wBACzB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;oBACrC,CAAC;yBAAM,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;wBAChC,MAAM,IAAI,KAAK,CAAC,UAAU,IAAI,CAAC,QAAQ,YAAY,CAAC,CAAC;oBACvD,CAAC;yBAAM,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;wBAChC,MAAM,IAAI,KAAK,CACb,qDAAqD,CACtD,CAAC;oBACJ,CAAC;yBAAM,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;wBAChC,MAAM,IAAI,KAAK,CACb,oBAAoB,KAAK,CAAC,OAAO,IAAI,aAAa,EAAE,CACrD,CAAC;oBACJ,CAAC;yBAAM,CAAC;wBACN,MAAM,IAAI,KAAK,CACb,wBAAwB,KAAK,CAAC,OAAO,IAAI,QAAQ,KAAK,CAAC,MAAM,EAAE,EAAE,CAClE,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,MAAM,IAAI,KAAK,CACb,+BAA+B,KAAK,CAAC,OAAO,IAAI,MAAM,CAAC,KAAK,CAAC,EAAE,CAChE,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,OAAO,CAAC,GAAG,CAAC,iCAAiC,QAAQ,EAAE,CAAC,CAAC;QAC3D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,sBAAsB;YACtB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;YACvB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACtB,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;gBAC/B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;YAC3B,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,aAAa,CACnB,MAAc,EACd,WAIC;QAED,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;QAElD,qDAAqD;QACrD,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/D,OAAO,IAAI,CAAC;QACd,CAAC;QAED,sCAAsC;QACtC,MAAM,SAAS,GAAG,WAAW,EAAE,KAAK,CAAC;QACrC,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,+BAA+B;YAC/B,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;QAE5D,2DAA2D;QAC3D,OAAO,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;IACvE,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,YAAoB,EACpB,UAAkB,EAClB,UAA+B,EAC/B,WAIC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAE5C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,mDAAmD;YACnD,MAAM,IAAI,CAAC,YAAY,EAAE,UAAU,CACjC,YAAY,EACZ,KAAK,EACL,SAAS,EACT,mBAAmB,UAAU,EAAE,CAChC,CAAC;YACF,OAAO;QACT,CAAC;QAED,0CAA0C;QAC1C,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,WAAW,CAAC,EAAE,CAAC;YAC7C,mDAAmD;YACnD,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC;gBAClC,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,cAAc;gBACrB,OAAO,EAAE,mDAAmD;gBAC5D,aAAa,EAAE,MAAM,CAAC,aAAa;aACpC,CAAC,CAAC;YACH,MAAM,IAAI,CAAC,YAAY,EAAE,UAAU,CACjC,YAAY,EACZ,KAAK,EACL,SAAS,EACT,YAAY,CACb,CAAC;YACF,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,+EAA+E;YAC/E,MAAM,OAAO,GAAsB;gBACjC,YAAY;gBACZ,UAAU;gBACV,IAAI,EAAE,WAAW,IAAI;oBACnB,WAAW,EAAE,EAAE;oBACf,QAAQ,EAAE,EAAE;iBACb;aACF,CAAC;YAEF,OAAO,CAAC,GAAG,CAAC,qBAAqB,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;YACxD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC,YAAY,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;YAChE,OAAO,CAAC,GAAG,CAAC,qBAAqB,UAAU,EAAE,EAAE,MAAM,CAAC,CAAC;QACzD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACzD,OAAO,CAAC,KAAK,CAAC,kBAAkB,UAAU,EAAE,EAAE,YAAY,CAAC,CAAC;YAC5D,MAAM,IAAI,CAAC,YAAY,EAAE,UAAU,CACjC,YAAY,EACZ,KAAK,EACL,SAAS,EACT,YAAY,CACb,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,gBAAgB;QAC5B,kCAAkC;QAClC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QAE1C,MAAM,iBAAiB,GAAsB;YAC3C,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;YAC5B,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;YAC1B,QAAQ;YACR,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU;YAClC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;YACxB,QAAQ,EAAE,CACR,YAAoB,EACpB,UAAkB,EAClB,MAA2B,EAC3B,WAIC,EACD,EAAE;gBACF,0DAA0D;gBAC1D,IAAI,CAAC,gBAAgB,CACnB,YAAY,EACZ,UAAU,EACV,MAAM,EACN,WAAW,CACZ,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;oBACd,OAAO,CAAC,KAAK,CAAC,uCAAuC,EAAE,GAAG,CAAC,CAAC;gBAC9D,CAAC,CAAC,CAAC;YACL,CAAC;YACD,YAAY,EAAE,GAAG,EAAE;gBACjB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;gBACvB,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;YAC1C,CAAC;SACF,CAAC;QAEF,IAAI,CAAC,YAAY,GAAG,IAAI,mBAAmB,CAAC,iBAAiB,CAAC,CAAC;QAC/D,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;YAC/B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAC3B,CAAC;QACD,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;IACzB,CAAC;CACF"}

package/dist/connection.js

@@ -0,0 +1,336 @@
+// WebSocket connection management for Delego SDK
+import { WebSocket as WS } from 'ws';
+import { StreamInOut } from './generated-client';
+if (typeof globalThis.WebSocket === 'undefined') {
+    globalThis.WebSocket = WS;
+}
+// ============================================================================
+// URL HELPERS
+// ============================================================================
+function buildWebSocketUrl(baseUrl) {
+    return baseUrl
+        .replace(/^http:\/\//, 'ws://')
+        .replace(/^https:\/\//, 'wss://');
+}
+function buildConnectionUrl(baseUrl, apiKey, runnerId, runnerName) {
+    const wsUrl = buildWebSocketUrl(baseUrl);
+    const queryParams = new URLSearchParams({ apiKey, runnerId });
+    if (runnerName) {
+        queryParams.set('runnerName', runnerName);
+    }
+    return `${wsUrl}/core/runners/connect?${queryParams.toString()}`;
+}
+// ============================================================================
+// MESSAGE PARSER
+// ============================================================================
+class MessageParser {
+    parse(data) {
+        const rawData = this.extractRawData(data);
+        return this.parseJson(rawData);
+    }
+    extractRawData(data) {
+        if (data && typeof data === 'object' && 'data' in data) {
+            return data.data;
+        }
+        return data;
+    }
+    parseJson(rawData) {
+        if (typeof rawData === 'string') {
+            return JSON.parse(rawData);
+        }
+        if (rawData instanceof Buffer || ArrayBuffer.isView(rawData)) {
+            return JSON.parse(rawData.toString());
+        }
+        throw new Error('Unable to parse message data');
+    }
+}
+// ============================================================================
+// CONNECTION LOGGER
+// ============================================================================
+class ConnectionLogger {
+    debug;
+    constructor(debug = false) {
+        this.debug = debug;
+    }
+    log(message, ...args) {
+        if (this.debug) {
+            console.log(`[Delego] ${message}`, ...args);
+        }
+    }
+    error(message, ...args) {
+        console.error(`[Delego] ${message}`, ...args);
+    }
+}
+class HeartbeatManager {
+    callbacks;
+    logger;
+    interval = null;
+    intervalMs = 10000;
+    constructor(callbacks, logger) {
+        this.callbacks = callbacks;
+        this.logger = logger;
+    }
+    start() {
+        this.stop();
+        this.logger.log('Starting heartbeat (every 10s)');
+        this.interval = setInterval(() => this.tick(), this.intervalMs);
+    }
+    stop() {
+        if (this.interval) {
+            clearInterval(this.interval);
+            this.interval = null;
+        }
+    }
+    async tick() {
+        this.logger.log('Sending heartbeat...');
+        try {
+            await this.callbacks.sendHeartbeat();
+            this.logger.log('Heartbeat sent');
+        }
+        catch (error) {
+            this.logger.error('Failed to send heartbeat:', error);
+            this.stop();
+            this.callbacks.onHeartbeatFailed(error);
+        }
+    }
+}
+class ReconnectionManager {
+    logger;
+    attempts = 0;
+    stopped = false;
+    config;
+    constructor(config = {}, logger) {
+        this.logger = logger;
+        this.config = {
+            initialDelayMs: config.initialDelayMs ?? 1000,
+            maxDelayMs: config.maxDelayMs ?? 30000,
+        };
+    }
+    reset() {
+        this.attempts = 0;
+        this.stopped = false;
+    }
+    canRetry() {
+        return !this.stopped;
+    }
+    stop() {
+        this.stopped = true;
+    }
+    async waitAndIncrement() {
+        this.attempts++;
+        const delay = Math.min(this.config.initialDelayMs * Math.pow(2, this.attempts - 1), this.config.maxDelayMs);
+        this.logger.log(`Waiting ${delay}ms before reconnecting (attempt ${this.attempts})...`);
+        await new Promise((resolve) => setTimeout(resolve, delay));
+    }
+    getAttemptCount() {
+        return this.attempts;
+    }
+}
+// ============================================================================
+// MAIN CLASS
+// ============================================================================
+export class WebSocketConnection {
+    stream = null;
+    state = 'disconnected';
+    options;
+    logger;
+    messageParser = new MessageParser();
+    heartbeat;
+    reconnection;
+    constructor(options) {
+        this.options = options;
+        this.logger = new ConnectionLogger(options.debug ?? false);
+        this.heartbeat = new HeartbeatManager({
+            sendHeartbeat: () => this.sendHeartbeat(),
+            onHeartbeatFailed: () => this.handleHeartbeatFailure(),
+        }, this.logger);
+        this.reconnection = new ReconnectionManager({}, this.logger);
+    }
+    getState() {
+        return this.state;
+    }
+    async connect() {
+        this.guardNotAlreadyConnecting();
+        this.state = 'connecting';
+        this.reconnection.reset();
+        await this.attemptConnection();
+    }
+    async attemptConnection() {
+        try {
+            this.stream = this.createStream();
+            await this.waitForReady();
+            this.startMessageLoop();
+        }
+        catch (error) {
+            this.state = 'disconnected';
+            if (!this.reconnection.canRetry()) {
+                throw error;
+            }
+            this.logger.error('Connection failed:', error);
+            await this.reconnection.waitAndIncrement();
+            this.state = 'connecting';
+            await this.attemptConnection();
+        }
+    }
+    async send(msg) {
+        if (!this.stream) {
+            throw new Error('Not connected');
+        }
+        if (this.state !== 'connected') {
+            throw new Error('Connection not established');
+        }
+        await this.stream.send(msg);
+    }
+    async sendResult(invocationId, success, result, error) {
+        await this.send({
+            type: 'invocation_result',
+            invocationId,
+            success,
+            result,
+            error,
+        });
+    }
+    disconnect() {
+        this.state = 'disconnected';
+        this.heartbeat.stop();
+        this.reconnection.stop();
+        if (this.stream) {
+            this.stream.close();
+            this.stream = null;
+        }
+    }
+    // ===========================================================================
+    // Private: Connection Setup
+    // ===========================================================================
+    guardNotAlreadyConnecting() {
+        if (this.state === 'connected' || this.state === 'connecting') {
+            throw new Error('Already connected or connecting');
+        }
+    }
+    createStream() {
+        const url = buildConnectionUrl(this.options.baseUrl, this.options.apiKey, this.options.runnerId, this.options.runnerName);
+        return new StreamInOut(url);
+    }
+    waitForReady() {
+        return new Promise((resolve, reject) => {
+            const timeout = setTimeout(() => {
+                reject(new Error('WebSocket connection timeout'));
+            }, 10000);
+            const cleanup = () => clearTimeout(timeout);
+            let isOpen = false;
+            this.stream.socket.on('open', () => {
+                isOpen = true;
+                this.logger.log('WebSocket opened, waiting for ready message...');
+            });
+            this.stream.socket.on('message', (data) => {
+                if (!isOpen)
+                    return;
+                try {
+                    const msg = this.messageParser.parse(data);
+                    if (msg.type === 'ready') {
+                        cleanup();
+                        this.onReady();
+                        resolve();
+                    }
+                }
+                catch {
+                    // Ignore parse errors for non-ready messages
+                }
+            });
+            this.stream.socket.on('error', (error) => {
+                cleanup();
+                this.state = 'disconnected';
+                reject(error);
+            });
+            this.stream.socket.on('close', () => {
+                if (this.state !== 'connected') {
+                    cleanup();
+                    this.state = 'disconnected';
+                    reject(new Error('WebSocket closed before ready'));
+                }
+            });
+        });
+    }
+    onReady() {
+        this.state = 'connected';
+        this.reconnection.reset();
+        this.logger.log('Connected - received ready message');
+        this.heartbeat.start();
+    }
+    // ===========================================================================
+    // Private: Message Loop
+    // ===========================================================================
+    async startMessageLoop() {
+        if (!this.stream)
+            return;
+        try {
+            for await (const msg of this.stream) {
+                this.handleMessage(msg);
+            }
+            this.logger.log('Stream ended');
+        }
+        catch (error) {
+            this.logger.error('Stream error:', error);
+        }
+        await this.handleDisconnection();
+    }
+    handleMessage(msg) {
+        switch (msg.type) {
+            case 'heartbeat_ack':
+                this.logger.log('Heartbeat acknowledged');
+                break;
+            case 'ready':
+                // Already handled during connection setup, ignore
+                break;
+            case 'invoke':
+                if (msg.invocationId && msg.actionName) {
+                    this.logger.log(`Received invocation: ${msg.actionName} (${msg.invocationId})`);
+                    this.options.onInvoke(msg.invocationId, msg.actionName, msg.parameters || {}, msg.userContext);
+                }
+                break;
+        }
+    }
+    // ===========================================================================
+    // Private: Heartbeat
+    // ===========================================================================
+    async sendHeartbeat() {
+        if (this.stream && this.state === 'connected') {
+            await this.send({ type: 'heartbeat' });
+        }
+    }
+    handleHeartbeatFailure() {
+        this.state = 'disconnected';
+        this.options.onDisconnect?.();
+    }
+    // ===========================================================================
+    // Private: Reconnection
+    // ===========================================================================
+    async handleDisconnection() {
+        this.logger.log('Disconnected');
+        this.state = 'disconnected';
+        this.heartbeat.stop();
+        this.options.onDisconnect?.();
+        if (this.reconnection.canRetry()) {
+            this.logger.log('Will attempt to reconnect...');
+            await this.attemptReconnect();
+        }
+    }
+    async attemptReconnect() {
+        await this.reconnection.waitAndIncrement();
+        try {
+            this.state = 'connecting';
+            this.stream = this.createStream();
+            await this.waitForReady();
+            this.logger.log('Reconnected successfully');
+            this.startMessageLoop();
+        }
+        catch (error) {
+            this.state = 'disconnected';
+            if (this.reconnection.canRetry()) {
+                this.logger.error('Reconnection failed:', error);
+                await this.attemptReconnect();
+            }
+        }
+    }
+}
+//# sourceMappingURL=connection.js.map