@agentuity/pi 1.0.22

package/dist/client.d.ts

@@ -0,0 +1,11 @@
+import type { InitMessage, HubRequest, HubResponse } from './protocol.ts';
+export declare class HubClient {
+    private ws;
+    private pending;
+    connect(url: string): Promise<InitMessage>;
+    nextId(): string;
+    send(request: HubRequest): Promise<HubResponse>;
+    close(): void;
+    get connected(): boolean;
+}
+//# sourceMappingURL=client.d.ts.map

package/dist/client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAQ1E,qBAAa,SAAS;IACrB,OAAO,CAAC,EAAE,CAA0B;IACpC,OAAO,CAAC,OAAO,CAOX;IACE,OAAO,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IA6FhD,MAAM,IAAI,MAAM;IAIV,IAAI,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,WAAW,CAAC;IAwBrD,KAAK,IAAI,IAAI;IAOb,IAAI,SAAS,IAAI,OAAO,CAEvB;CACD"}

package/dist/client.js

@@ -0,0 +1,118 @@
+/** How long to wait for a response before rejecting the pending promise (ms). */
+const SEND_TIMEOUT_MS = 30_000;
+/** How long to wait for the init message after connecting (ms). */
+const CONNECT_TIMEOUT_MS = 30_000;
+export class HubClient {
+    ws = null;
+    pending = new Map();
+    async connect(url) {
+        // Guard against overlapping connections — old socket handlers would corrupt shared state
+        if (this.ws && this.ws.readyState !== WebSocket.CLOSED) {
+            throw new Error('Already connected or connecting — call close() first');
+        }
+        return new Promise((resolve, reject) => {
+            // Ensure ws:// or wss:// protocol — upgrade http(s) URLs automatically
+            let wsUrl = url;
+            if (wsUrl.startsWith('http://')) {
+                wsUrl = 'ws://' + wsUrl.slice(7);
+            }
+            else if (wsUrl.startsWith('https://')) {
+                wsUrl = 'wss://' + wsUrl.slice(8);
+            }
+            else if (!wsUrl.startsWith('ws://') &&
+                !wsUrl.startsWith('wss://')) {
+                wsUrl = 'ws://' + wsUrl;
+            }
+            this.ws = new WebSocket(wsUrl);
+            // Guard against duplicate init messages re-invoking the resolve
+            let initResolved = false;
+            const connectTimer = setTimeout(() => {
+                if (!initResolved) {
+                    reject(new Error(`Hub did not send init message within ${CONNECT_TIMEOUT_MS}ms`));
+                    this.ws?.close();
+                }
+            }, CONNECT_TIMEOUT_MS);
+            this.ws.onmessage = (event) => {
+                let data;
+                try {
+                    const raw = typeof event.data === 'string'
+                        ? event.data
+                        : new TextDecoder().decode(event.data);
+                    data = JSON.parse(raw);
+                }
+                catch {
+                    // Malformed or non-JSON frame — ignore
+                    return;
+                }
+                // First message should be init
+                if (data.type === 'init' && !initResolved) {
+                    initResolved = true;
+                    clearTimeout(connectTimer);
+                    resolve(data);
+                    return;
+                }
+                // Otherwise it's a response to a pending request
+                const response = data;
+                const entry = this.pending.get(response.id);
+                if (entry) {
+                    clearTimeout(entry.timer);
+                    this.pending.delete(response.id);
+                    entry.resolve(response);
+                }
+            };
+            this.ws.onerror = (err) => {
+                // ErrorEvent has a message property; plain Event does not
+                const message = 'message' in err &&
+                    typeof err.message === 'string'
+                    ? err.message
+                    : `connection to ${wsUrl} failed`;
+                clearTimeout(connectTimer);
+                reject(new Error(`WebSocket error: ${message}`));
+            };
+            this.ws.onclose = () => {
+                clearTimeout(connectTimer);
+                // Reject connect() promise if init was never received
+                if (!initResolved) {
+                    reject(new Error('WebSocket closed before init message received'));
+                }
+                // Reject all pending requests
+                for (const [id, entry] of this.pending) {
+                    clearTimeout(entry.timer);
+                    entry.reject(new Error('WebSocket closed'));
+                    this.pending.delete(id);
+                }
+                this.ws = null;
+            };
+        });
+    }
+    nextId() {
+        return crypto.randomUUID();
+    }
+    async send(request) {
+        if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
+            // Return a default ACK if not connected
+            return { id: request.id, actions: [{ action: 'ACK' }] };
+        }
+        return new Promise((resolve, reject) => {
+            const timer = setTimeout(() => {
+                const entry = this.pending.get(request.id);
+                if (entry) {
+                    this.pending.delete(request.id);
+                    entry.reject(new Error(`Hub response timeout after ${SEND_TIMEOUT_MS}ms for request ${request.id}`));
+                }
+            }, SEND_TIMEOUT_MS);
+            this.pending.set(request.id, { resolve, reject, timer });
+            this.ws.send(JSON.stringify(request));
+        });
+    }
+    close() {
+        if (this.ws) {
+            this.ws.close();
+            this.ws = null;
+        }
+    }
+    get connected() {
+        return this.ws !== null && this.ws.readyState === WebSocket.OPEN;
+    }
+}
+//# 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,iFAAiF;AACjF,MAAM,eAAe,GAAG,MAAM,CAAC;AAE/B,mEAAmE;AACnE,MAAM,kBAAkB,GAAG,MAAM,CAAC;AAElC,MAAM,OAAO,SAAS;IACb,EAAE,GAAqB,IAAI,CAAC;IAC5B,OAAO,GAAG,IAAI,GAAG,EAOtB,CAAC;IACJ,KAAK,CAAC,OAAO,CAAC,GAAW;QACxB,yFAAyF;QACzF,IAAI,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,MAAM,EAAE,CAAC;YACxD,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;QACzE,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACtC,uEAAuE;YACvE,IAAI,KAAK,GAAG,GAAG,CAAC;YAChB,IAAI,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;gBACjC,KAAK,GAAG,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAClC,CAAC;iBAAM,IAAI,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;gBACzC,KAAK,GAAG,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACnC,CAAC;iBAAM,IACN,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC;gBAC1B,CAAC,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,EAC1B,CAAC;gBACF,KAAK,GAAG,OAAO,GAAG,KAAK,CAAC;YACzB,CAAC;YAED,IAAI,CAAC,EAAE,GAAG,IAAI,SAAS,CAAC,KAAK,CAAC,CAAC;YAE/B,gEAAgE;YAChE,IAAI,YAAY,GAAG,KAAK,CAAC;YAEzB,MAAM,YAAY,GAAG,UAAU,CAAC,GAAG,EAAE;gBACpC,IAAI,CAAC,YAAY,EAAE,CAAC;oBACnB,MAAM,CAAC,IAAI,KAAK,CAAC,wCAAwC,kBAAkB,IAAI,CAAC,CAAC,CAAC;oBAClF,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,CAAC;gBAClB,CAAC;YACF,CAAC,EAAE,kBAAkB,CAAC,CAAC;YAEvB,IAAI,CAAC,EAAE,CAAC,SAAS,GAAG,CAAC,KAAmB,EAAE,EAAE;gBAC3C,IAAI,IAA6B,CAAC;gBAClC,IAAI,CAAC;oBACJ,MAAM,GAAG,GACR,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;wBAC7B,CAAC,CAAC,KAAK,CAAC,IAAI;wBACZ,CAAC,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CACxB,KAAK,CAAC,IAAmB,CACzB,CAAC;oBACL,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAA4B,CAAC;gBACnD,CAAC;gBAAC,MAAM,CAAC;oBACR,uCAAuC;oBACvC,OAAO;gBACR,CAAC;gBAED,+BAA+B;gBAC/B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;oBAC3C,YAAY,GAAG,IAAI,CAAC;oBACpB,YAAY,CAAC,YAAY,CAAC,CAAC;oBAC3B,OAAO,CAAC,IAA8B,CAAC,CAAC;oBACxC,OAAO;gBACR,CAAC;gBAED,iDAAiD;gBACjD,MAAM,QAAQ,GAAG,IAA8B,CAAC;gBAChD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;gBAC5C,IAAI,KAAK,EAAE,CAAC;oBACX,YAAY,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBAC1B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;oBACjC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBACzB,CAAC;YACF,CAAC,CAAC;YAEF,IAAI,CAAC,EAAE,CAAC,OAAO,GAAG,CAAC,GAAU,EAAE,EAAE;gBAChC,0DAA0D;gBAC1D,MAAM,OAAO,GACZ,SAAS,IAAI,GAAG;oBAChB,OAAQ,GAAkB,CAAC,OAAO,KAAK,QAAQ;oBAC9C,CAAC,CAAE,GAAkB,CAAC,OAAO;oBAC7B,CAAC,CAAC,iBAAiB,KAAK,SAAS,CAAC;gBACpC,YAAY,CAAC,YAAY,CAAC,CAAC;gBAC3B,MAAM,CAAC,IAAI,KAAK,CAAC,oBAAoB,OAAO,EAAE,CAAC,CAAC,CAAC;YAClD,CAAC,CAAC;YAEF,IAAI,CAAC,EAAE,CAAC,OAAO,GAAG,GAAG,EAAE;gBACtB,YAAY,CAAC,YAAY,CAAC,CAAC;gBAC3B,sDAAsD;gBACtD,IAAI,CAAC,YAAY,EAAE,CAAC;oBACnB,MAAM,CAAC,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC,CAAC;gBACpE,CAAC;gBACD,8BAA8B;gBAC9B,KAAK,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;oBACxC,YAAY,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBAC1B,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC,CAAC;oBAC5C,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACzB,CAAC;gBACD,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;YAChB,CAAC,CAAC;QACH,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,MAAM;QACL,OAAO,MAAM,CAAC,UAAU,EAAE,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAmB;QAC7B,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YACvD,wCAAwC;YACxC,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QACzD,CAAC;QAED,OAAO,IAAI,OAAO,CAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACnD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC3C,IAAI,KAAK,EAAE,CAAC;oBACX,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBAChC,KAAK,CAAC,MAAM,CACX,IAAI,KAAK,CACR,8BAA8B,eAAe,kBAAkB,OAAO,CAAC,EAAE,EAAE,CAC3E,CACD,CAAC;gBACH,CAAC;YACF,CAAC,EAAE,eAAe,CAAC,CAAC;YAEpB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YACzD,IAAI,CAAC,EAAG,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK;QACJ,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACb,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QAChB,CAAC;IACF,CAAC;IAED,IAAI,SAAS;QACZ,OAAO,IAAI,CAAC,EAAE,KAAK,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,CAAC;IAClE,CAAC;CACD"}

package/dist/handlers.d.ts

@@ -0,0 +1,19 @@
+import type { HubAction } from './protocol.ts';
+export interface ActionResult {
+    block?: {
+        block: true;
+        reason: string;
+    };
+    returnValue?: unknown;
+}
+/** Minimal UI surface used by action handlers — avoids a hard dep on pi-coding-agent. */
+interface ActionContext {
+    ui?: {
+        notify(message: string, level?: 'info' | 'warning' | 'error'): void;
+        confirm(title: string, message: string): Promise<boolean>;
+        setStatus(key: string, text?: string): void;
+    };
+}
+export declare function processActions(actions: HubAction[], ctx: ActionContext): Promise<ActionResult>;
+export {};
+//# sourceMappingURL=handlers.d.ts.map

package/dist/handlers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"handlers.d.ts","sourceRoot":"","sources":["../src/handlers.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAE/C,MAAM,WAAW,YAAY;IAC5B,KAAK,CAAC,EAAE;QAAE,KAAK,EAAE,IAAI,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC;IACxC,WAAW,CAAC,EAAE,OAAO,CAAC;CAEtB;AAED,yFAAyF;AACzF,UAAU,aAAa;IACtB,EAAE,CAAC,EAAE;QACJ,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,GAAG,OAAO,GAAG,IAAI,CAAC;QACpE,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;QAC1D,SAAS,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;KAC5C,CAAC;CACF;AAED,wBAAsB,cAAc,CACnC,OAAO,EAAE,SAAS,EAAE,EACpB,GAAG,EAAE,aAAa,GAChB,OAAO,CAAC,YAAY,CAAC,CAkEvB"}

package/dist/handlers.js

@@ -0,0 +1,58 @@
+export async function processActions(actions, ctx) {
+    let result = {};
+    for (const action of actions) {
+        switch (action.action) {
+            case 'ACK':
+                // Terminal: proceed normally
+                result = {};
+                break;
+            case 'BLOCK':
+                // Terminal: block
+                result = { block: { block: true, reason: action.reason } };
+                break;
+            case 'RETURN':
+                // Terminal: return a specific result
+                result = { returnValue: action.result };
+                break;
+            case 'NOTIFY':
+                // Side effect: show notification, continue
+                if (ctx?.ui) {
+                    ctx.ui.notify(action.message, action.level ?? 'info');
+                }
+                break;
+            case 'STATUS':
+                // Side effect: set status, continue
+                if (ctx?.ui) {
+                    ctx.ui.setStatus(action.key, action.text);
+                }
+                break;
+            case 'CONFIRM': {
+                // Gate: if user denies, stop and block
+                if (ctx?.ui) {
+                    const confirmed = await ctx.ui.confirm(action.title, action.message);
+                    if (!confirmed) {
+                        return {
+                            block: {
+                                block: true,
+                                reason: action.deny_reason ?? 'Denied by user',
+                            },
+                        };
+                    }
+                }
+                else {
+                    // No UI available — block by default for safety
+                    result = {
+                        block: {
+                            block: true,
+                            reason: action.deny_reason ??
+                                'Confirmation required but no UI available',
+                        },
+                    };
+                }
+                break;
+            }
+        }
+    }
+    return result;
+}
+//# sourceMappingURL=handlers.js.map

package/dist/handlers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handlers.js","sourceRoot":"","sources":["../src/handlers.ts"],"names":[],"mappings":"AAiBA,MAAM,CAAC,KAAK,UAAU,cAAc,CACnC,OAAoB,EACpB,GAAkB;IAElB,IAAI,MAAM,GAAiB,EAAE,CAAC;IAE9B,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC9B,QAAQ,MAAM,CAAC,MAAM,EAAE,CAAC;YACvB,KAAK,KAAK;gBACT,6BAA6B;gBAC7B,MAAM,GAAG,EAAE,CAAC;gBACZ,MAAM;YAEP,KAAK,OAAO;gBACX,kBAAkB;gBAClB,MAAM,GAAG,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC3D,MAAM;YAEP,KAAK,QAAQ;gBACZ,qCAAqC;gBACrC,MAAM,GAAG,EAAE,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;gBACxC,MAAM;YAEP,KAAK,QAAQ;gBACZ,2CAA2C;gBAC3C,IAAI,GAAG,EAAE,EAAE,EAAE,CAAC;oBACb,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,CAAC;gBACvD,CAAC;gBACD,MAAM;YAEP,KAAK,QAAQ;gBACZ,oCAAoC;gBACpC,IAAI,GAAG,EAAE,EAAE,EAAE,CAAC;oBACb,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC3C,CAAC;gBACD,MAAM;YAEP,KAAK,SAAS,CAAC,CAAC,CAAC;gBAChB,uCAAuC;gBACvC,IAAI,GAAG,EAAE,EAAE,EAAE,CAAC;oBACb,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,OAAO,CACrC,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,OAAO,CACd,CAAC;oBACF,IAAI,CAAC,SAAS,EAAE,CAAC;wBAChB,OAAO;4BACN,KAAK,EAAE;gCACN,KAAK,EAAE,IAAI;gCACX,MAAM,EAAE,MAAM,CAAC,WAAW,IAAI,gBAAgB;6BAC9C;yBACD,CAAC;oBACH,CAAC;gBACF,CAAC;qBAAM,CAAC;oBACP,gDAAgD;oBAChD,MAAM,GAAG;wBACR,KAAK,EAAE;4BACN,KAAK,EAAE,IAAI;4BACX,MAAM,EACL,MAAM,CAAC,WAAW;gCAClB,2CAA2C;yBAC5C;qBACD,CAAC;gBACH,CAAC;gBACD,MAAM;YACP,CAAC;QACF,CAAC;IACF,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,4 @@
+import type { ExtensionAPI } from '@mariozechner/pi-coding-agent';
+export declare function agentuityCoderHub(pi: ExtensionAPI): void;
+export default agentuityCoderHub;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAEX,YAAY,EAGZ,MAAM,+BAA+B,CAAC;AA6CvC,wBAAgB,iBAAiB,CAAC,EAAE,EAAE,YAAY,QAsMjD;AAED,eAAe,iBAAiB,CAAC"}

package/dist/index.js

@@ -0,0 +1,198 @@
+import { HubClient } from "./client.js";
+import { processActions } from "./handlers.js";
+const HUB_URL_ENV = 'AGENTUITY_CODER_HUB_URL';
+// All Pi events we subscribe to (order matters — session_start is handled separately)
+const PROXY_EVENTS = [
+    'session_shutdown',
+    'session_before_switch',
+    'session_switch',
+    'session_before_fork',
+    'session_fork',
+    'session_before_compact',
+    'session_compact',
+    'before_agent_start',
+    'agent_start',
+    'agent_end',
+    'turn_start',
+    'turn_end',
+    'tool_call',
+    'tool_result',
+    'tool_execution_start',
+    'tool_execution_update',
+    'tool_execution_end',
+    'message_start',
+    'message_update',
+    'message_end',
+    'input',
+    'model_select',
+    'context',
+];
+function log(msg) {
+    console.error(`[agentuity-pi] ${msg}`);
+}
+export function agentuityCoderHub(pi) {
+    const hubUrl = process.env[HUB_URL_ENV];
+    // No-op if not configured
+    if (!hubUrl) {
+        return;
+    }
+    log(`Hub URL: ${hubUrl}`);
+    const client = new HubClient();
+    // Connect to the Hub and register tools/commands
+    async function connectAndRegister() {
+        if (client.connected)
+            return;
+        log('Connecting to Hub...');
+        let initMsg;
+        try {
+            initMsg = await client.connect(hubUrl);
+        }
+        catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            log(`Failed to connect: ${msg}`);
+            return;
+        }
+        log(`Connected. Init: ${initMsg.tools?.length ?? 0} tools, ${initMsg.commands?.length ?? 0} commands`);
+        // Register tools from Hub
+        if (initMsg.tools) {
+            for (const toolDef of initMsg.tools) {
+                log(`Registering tool: ${toolDef.name}`);
+                pi.registerTool({
+                    name: toolDef.name,
+                    label: toolDef.label,
+                    description: toolDef.description,
+                    parameters: toolDef.parameters,
+                    async execute(toolCallId, params, _signal, _onUpdate, ctx) {
+                        log(`Tool execute: ${toolDef.name}`);
+                        const id = client.nextId();
+                        let response;
+                        try {
+                            response = await client.send({
+                                id,
+                                type: 'tool',
+                                name: toolDef.name,
+                                toolCallId,
+                                params: params,
+                            });
+                        }
+                        catch {
+                            return {
+                                content: [
+                                    {
+                                        type: 'text',
+                                        text: 'Error: Hub connection lost',
+                                    },
+                                ],
+                                details: {},
+                            };
+                        }
+                        const result = await processActions(response.actions, ctx);
+                        if (result.returnValue !== undefined) {
+                            return result.returnValue;
+                        }
+                        return {
+                            content: [
+                                { type: 'text', text: 'Done' },
+                            ],
+                            details: {},
+                        };
+                    },
+                });
+            }
+        }
+        // Register commands from Hub
+        if (initMsg.commands) {
+            for (const cmdDef of initMsg.commands) {
+                log(`Registering command: /${cmdDef.name}`);
+                pi.registerCommand(cmdDef.name, {
+                    description: cmdDef.description,
+                    handler: async (args, ctx) => {
+                        log(`Command execute: /${cmdDef.name}`);
+                        const id = client.nextId();
+                        let response;
+                        try {
+                            response = await client.send({
+                                id,
+                                type: 'command',
+                                name: cmdDef.name,
+                                args,
+                            });
+                        }
+                        catch {
+                            ctx.ui.notify('Hub connection lost', 'error');
+                            return;
+                        }
+                        await processActions(response.actions, ctx);
+                    },
+                });
+            }
+        }
+        log('Registration complete');
+    }
+    // Helper to send event and process response actions
+    async function sendEvent(eventName, data, ctx) {
+        if (!client.connected)
+            return undefined;
+        const id = client.nextId();
+        let response;
+        try {
+            response = await client.send({
+                id,
+                type: 'event',
+                event: eventName,
+                data,
+            });
+        }
+        catch {
+            return undefined;
+        }
+        const result = await processActions(response.actions, ctx);
+        if (result.block)
+            return result.block;
+        if (result.returnValue !== undefined)
+            return result.returnValue;
+        return undefined;
+    }
+    // Serialize event data — strip non-serializable fields
+    function serializeEvent(event) {
+        const data = {};
+        if (event && typeof event === 'object') {
+            for (const [key, value] of Object.entries(event)) {
+                if (typeof value !== 'function' && key !== 'signal') {
+                    try {
+                        JSON.stringify(value);
+                        data[key] = value;
+                    }
+                    catch {
+                        // Skip non-serializable values
+                    }
+                }
+            }
+        }
+        return data;
+    }
+    // Cast pi.on to generic handler since we iterate over a union of event names
+    // TypeScript overloads require exact string literal matching which doesn't work in loops
+    const onEvent = pi.on.bind(pi);
+    // ── session_start: connect + register BEFORE Pi builds the tool list ──
+    onEvent('session_start', async (event, ctx) => {
+        await connectAndRegister();
+        if (client.connected) {
+            return sendEvent('session_start', serializeEvent(event), ctx);
+        }
+    });
+    for (const eventName of PROXY_EVENTS) {
+        onEvent(eventName, async (event, ctx) => {
+            if (!client.connected)
+                return undefined;
+            return sendEvent(eventName, serializeEvent(event), ctx);
+        });
+    }
+    // Clean up on shutdown
+    pi.on('session_shutdown', async () => {
+        log('Shutting down');
+        client.close();
+    });
+}
+export default agentuityCoderHub;
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAG/C,MAAM,WAAW,GAAG,yBAAyB,CAAC;AAE9C,sFAAsF;AACtF,MAAM,YAAY,GAAG;IACpB,kBAAkB;IAClB,uBAAuB;IACvB,gBAAgB;IAChB,qBAAqB;IACrB,cAAc;IACd,wBAAwB;IACxB,iBAAiB;IACjB,oBAAoB;IACpB,aAAa;IACb,WAAW;IACX,YAAY;IACZ,UAAU;IACV,WAAW;IACX,aAAa;IACb,sBAAsB;IACtB,uBAAuB;IACvB,oBAAoB;IACpB,eAAe;IACf,gBAAgB;IAChB,aAAa;IACb,OAAO;IACP,cAAc;IACd,SAAS;CACA,CAAC;AAQX,SAAS,GAAG,CAAC,GAAW;IACvB,OAAO,CAAC,KAAK,CAAC,kBAAkB,GAAG,EAAE,CAAC,CAAC;AACxC,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,EAAgB;IACjD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAExC,0BAA0B;IAC1B,IAAI,CAAC,MAAM,EAAE,CAAC;QACb,OAAO;IACR,CAAC;IAED,GAAG,CAAC,YAAY,MAAM,EAAE,CAAC,CAAC;IAE1B,MAAM,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;IAE/B,iDAAiD;IACjD,KAAK,UAAU,kBAAkB;QAChC,IAAI,MAAM,CAAC,SAAS;YAAE,OAAO;QAE7B,GAAG,CAAC,sBAAsB,CAAC,CAAC;QAC5B,IAAI,OAAoB,CAAC;QAEzB,IAAI,CAAC;YACJ,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,MAAO,CAAC,CAAC;QACzC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACd,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7D,GAAG,CAAC,sBAAsB,GAAG,EAAE,CAAC,CAAC;YACjC,OAAO;QACR,CAAC;QAED,GAAG,CACF,oBAAoB,OAAO,CAAC,KAAK,EAAE,MAAM,IAAI,CAAC,WAAW,OAAO,CAAC,QAAQ,EAAE,MAAM,IAAI,CAAC,WAAW,CACjG,CAAC;QAEF,0BAA0B;QAC1B,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YACnB,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBACrC,GAAG,CAAC,qBAAqB,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;gBACzC,EAAE,CAAC,YAAY,CAAC;oBACf,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,KAAK,EAAE,OAAO,CAAC,KAAK;oBACpB,WAAW,EAAE,OAAO,CAAC,WAAW;oBAChC,UAAU,EAAE,OAAO,CAAC,UAAgC;oBACpD,KAAK,CAAC,OAAO,CACZ,UAAkB,EAClB,MAAe,EACf,OAAgC,EAChC,SAAkB,EAClB,GAAqB;wBAErB,GAAG,CAAC,iBAAiB,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;wBACrC,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;wBAC3B,IAAI,QAAqB,CAAC;wBAE1B,IAAI,CAAC;4BACJ,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC;gCAC5B,EAAE;gCACF,IAAI,EAAE,MAAM;gCACZ,IAAI,EAAE,OAAO,CAAC,IAAI;gCAClB,UAAU;gCACV,MAAM,EAAE,MAAiC;6BACzC,CAAC,CAAC;wBACJ,CAAC;wBAAC,MAAM,CAAC;4BACR,OAAO;gCACN,OAAO,EAAE;oCACR;wCACC,IAAI,EAAE,MAAe;wCACrB,IAAI,EAAE,4BAA4B;qCAClC;iCACD;gCACD,OAAO,EAAE,EAAE;6BACX,CAAC;wBACH,CAAC;wBAED,MAAM,MAAM,GAAG,MAAM,cAAc,CAClC,QAAQ,CAAC,OAAO,EAChB,GAAG,CACH,CAAC;wBAEF,IAAI,MAAM,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;4BACtC,OAAO,MAAM,CAAC,WAAuC,CAAC;wBACvD,CAAC;wBAED,OAAO;4BACN,OAAO,EAAE;gCACR,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,MAAM,EAAE;6BACvC;4BACD,OAAO,EAAE,EAAE;yBACX,CAAC;oBACH,CAAC;iBACD,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QAED,6BAA6B;QAC7B,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACtB,KAAK,MAAM,MAAM,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;gBACvC,GAAG,CAAC,yBAAyB,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC5C,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,EAAE;oBAC/B,WAAW,EAAE,MAAM,CAAC,WAAW;oBAC/B,OAAO,EAAE,KAAK,EACb,IAAY,EACZ,GAA4B,EAC3B,EAAE;wBACH,GAAG,CAAC,qBAAqB,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;wBACxC,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;wBAC3B,IAAI,QAAqB,CAAC;wBAE1B,IAAI,CAAC;4BACJ,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC;gCAC5B,EAAE;gCACF,IAAI,EAAE,SAAS;gCACf,IAAI,EAAE,MAAM,CAAC,IAAI;gCACjB,IAAI;6BACJ,CAAC,CAAC;wBACJ,CAAC;wBAAC,MAAM,CAAC;4BACR,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC;4BAC9C,OAAO;wBACR,CAAC;wBAED,MAAM,cAAc,CAAC,QAAQ,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;oBAC7C,CAAC;iBACD,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QAED,GAAG,CAAC,uBAAuB,CAAC,CAAC;IAC9B,CAAC;IAED,oDAAoD;IACpD,KAAK,UAAU,SAAS,CACvB,SAAiB,EACjB,IAA6B,EAC7B,GAAqB;QAErB,IAAI,CAAC,MAAM,CAAC,SAAS;YAAE,OAAO,SAAS,CAAC;QAExC,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;QAC3B,IAAI,QAAqB,CAAC;QAE1B,IAAI,CAAC;YACJ,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC;gBAC5B,EAAE;gBACF,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,SAAS;gBAChB,IAAI;aACJ,CAAC,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACR,OAAO,SAAS,CAAC;QAClB,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,QAAQ,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QAE3D,IAAI,MAAM,CAAC,KAAK;YAAE,OAAO,MAAM,CAAC,KAAK,CAAC;QACtC,IAAI,MAAM,CAAC,WAAW,KAAK,SAAS;YAAE,OAAO,MAAM,CAAC,WAAW,CAAC;QAChE,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,uDAAuD;IACvD,SAAS,cAAc,CAAC,KAAc;QACrC,MAAM,IAAI,GAA4B,EAAE,CAAC;QACzC,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACxC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBAClD,IAAI,OAAO,KAAK,KAAK,UAAU,IAAI,GAAG,KAAK,QAAQ,EAAE,CAAC;oBACrD,IAAI,CAAC;wBACJ,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;wBACtB,IAAI,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;oBACnB,CAAC;oBAAC,MAAM,CAAC;wBACR,+BAA+B;oBAChC,CAAC;gBACF,CAAC;YACF,CAAC;QACF,CAAC;QACD,OAAO,IAAI,CAAC;IACb,CAAC;IAED,6EAA6E;IAC7E,yFAAyF;IACzF,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAwB,CAAC;IAEtD,yEAAyE;IACzE,OAAO,CAAC,eAAe,EAAE,KAAK,EAAE,KAAc,EAAE,GAAqB,EAAE,EAAE;QACxE,MAAM,kBAAkB,EAAE,CAAC;QAE3B,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YACtB,OAAO,SAAS,CAAC,eAAe,EAAE,cAAc,CAAC,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC;QAC/D,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,KAAK,MAAM,SAAS,IAAI,YAAY,EAAE,CAAC;QACtC,OAAO,CAAC,SAAS,EAAE,KAAK,EAAE,KAAc,EAAE,GAAqB,EAAE,EAAE;YAClE,IAAI,CAAC,MAAM,CAAC,SAAS;gBAAE,OAAO,SAAS,CAAC;YACxC,OAAO,SAAS,CAAC,SAAS,EAAE,cAAc,CAAC,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,uBAAuB;IACvB,EAAE,CAAC,EAAE,CAAC,kBAAkB,EAAE,KAAK,IAAI,EAAE;QACpC,GAAG,CAAC,eAAe,CAAC,CAAC;QACrB,MAAM,CAAC,KAAK,EAAE,CAAC;IAChB,CAAC,CAAC,CAAC;AACJ,CAAC;AAED,eAAe,iBAAiB,CAAC"}

package/dist/protocol.d.ts

@@ -0,0 +1,68 @@
+export interface HubToolDefinition {
+    name: string;
+    label: string;
+    description: string;
+    parameters: Record<string, unknown>;
+}
+export interface HubCommandDefinition {
+    name: string;
+    description: string;
+}
+export interface InitMessage {
+    type: 'init';
+    tools?: HubToolDefinition[];
+    commands?: HubCommandDefinition[];
+}
+export interface EventRequest {
+    id: string;
+    type: 'event';
+    event: string;
+    data: Record<string, unknown>;
+}
+export interface ToolRequest {
+    id: string;
+    type: 'tool';
+    name: string;
+    toolCallId: string;
+    params: Record<string, unknown>;
+}
+export interface CommandRequest {
+    id: string;
+    type: 'command';
+    name: string;
+    args: string;
+}
+export type HubRequest = EventRequest | ToolRequest | CommandRequest;
+export interface AckAction {
+    action: 'ACK';
+}
+export interface BlockAction {
+    action: 'BLOCK';
+    reason: string;
+}
+export interface ConfirmAction {
+    action: 'CONFIRM';
+    title: string;
+    message: string;
+    deny_reason?: string;
+}
+export interface NotifyAction {
+    action: 'NOTIFY';
+    message: string;
+    level?: 'info' | 'warning' | 'error';
+}
+export interface ReturnAction {
+    action: 'RETURN';
+    result: unknown;
+}
+export interface StatusAction {
+    action: 'STATUS';
+    key: string;
+    text?: string;
+}
+export type HubAction = AckAction | BlockAction | ConfirmAction | NotifyAction | ReturnAction | StatusAction;
+export interface HubResponse {
+    id: string;
+    actions: HubAction[];
+}
+//# sourceMappingURL=protocol.d.ts.map

package/dist/protocol.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"protocol.d.ts","sourceRoot":"","sources":["../src/protocol.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,iBAAiB;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,oBAAoB;IACpC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,WAAW;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,iBAAiB,EAAE,CAAC;IAC5B,QAAQ,CAAC,EAAE,oBAAoB,EAAE,CAAC;CAClC;AAID,MAAM,WAAW,YAAY;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,OAAO,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC9B;AAED,MAAM,WAAW,WAAW;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAChC;AAED,MAAM,WAAW,cAAc;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,SAAS,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACb;AAED,MAAM,MAAM,UAAU,GAAG,YAAY,GAAG,WAAW,GAAG,cAAc,CAAC;AAIrE,MAAM,WAAW,SAAS;IACzB,MAAM,EAAE,KAAK,CAAC;CACd;AAED,MAAM,WAAW,WAAW;IAC3B,MAAM,EAAE,OAAO,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,aAAa;IAC7B,MAAM,EAAE,SAAS,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,YAAY;IAC5B,MAAM,EAAE,QAAQ,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,GAAG,OAAO,CAAC;CACrC;AAED,MAAM,WAAW,YAAY;IAC5B,MAAM,EAAE,QAAQ,CAAC;IACjB,MAAM,EAAE,OAAO,CAAC;CAChB;AAED,MAAM,WAAW,YAAY;IAC5B,MAAM,EAAE,QAAQ,CAAC;IACjB,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,MAAM,CAAC;CACd;AAED,MAAM,MAAM,SAAS,GAClB,SAAS,GACT,WAAW,GACX,aAAa,GACb,YAAY,GACZ,YAAY,GACZ,YAAY,CAAC;AAIhB,MAAM,WAAW,WAAW;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,SAAS,EAAE,CAAC;CACrB"}

package/dist/protocol.js

@@ -0,0 +1,3 @@
+// ---- Init Message (Server → Client on connect) ----
+export {};
+//# sourceMappingURL=protocol.js.map

package/dist/protocol.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"protocol.js","sourceRoot":"","sources":["../src/protocol.ts"],"names":[],"mappings":"AAAA,sDAAsD"}

package/package.json

@@ -0,0 +1,42 @@
+{
+	"name": "@agentuity/pi",
+	"version": "1.0.22",
+	"description": "Agentuity Coder Hub extension for Pi coding agent",
+	"license": "Apache-2.0",
+	"author": "Agentuity employees and contributors",
+	"type": "module",
+	"main": "./dist/index.js",
+	"types": "./dist/index.d.ts",
+	"files": [
+		"README.md",
+		"src",
+		"dist"
+	],
+	"exports": {
+		".": {
+			"import": "./dist/index.js",
+			"types": "./dist/index.d.ts"
+		}
+	},
+	"scripts": {
+		"clean": "rm -rf dist tsconfig.tsbuildinfo",
+		"build": "bunx tsc --build",
+		"typecheck": "bunx tsc --noEmit",
+		"prepublishOnly": "bun run clean && bun run build"
+	},
+	"peerDependencies": {
+		"@mariozechner/pi-coding-agent": "*",
+		"@sinclair/typebox": "*"
+	},
+	"devDependencies": {
+		"@mariozechner/pi-coding-agent": "^0.54.2",
+		"@sinclair/typebox": "^0.34.48",
+		"@types/bun": "^1.3.9",
+		"bun-types": "^1.3.9",
+		"typescript": "^5.9.0"
+	},
+	"publishConfig": {
+		"access": "public"
+	},
+	"sideEffects": false
+}

package/src/client.ts

@@ -0,0 +1,150 @@
+import type { InitMessage, HubRequest, HubResponse } from './protocol.ts';
+
+/** How long to wait for a response before rejecting the pending promise (ms). */
+const SEND_TIMEOUT_MS = 30_000;
+
+/** How long to wait for the init message after connecting (ms). */
+const CONNECT_TIMEOUT_MS = 30_000;
+
+export class HubClient {
+	private ws: WebSocket | null = null;
+	private pending = new Map<
+		string,
+		{
+			resolve: (resp: HubResponse) => void;
+			reject: (err: Error) => void;
+			timer: ReturnType<typeof setTimeout>;
+		}
+	>();
+	async connect(url: string): Promise<InitMessage> {
+		// Guard against overlapping connections — old socket handlers would corrupt shared state
+		if (this.ws && this.ws.readyState !== WebSocket.CLOSED) {
+			throw new Error('Already connected or connecting — call close() first');
+		}
+
+		return new Promise((resolve, reject) => {
+			// Ensure ws:// or wss:// protocol — upgrade http(s) URLs automatically
+			let wsUrl = url;
+			if (wsUrl.startsWith('http://')) {
+				wsUrl = 'ws://' + wsUrl.slice(7);
+			} else if (wsUrl.startsWith('https://')) {
+				wsUrl = 'wss://' + wsUrl.slice(8);
+			} else if (
+				!wsUrl.startsWith('ws://') &&
+				!wsUrl.startsWith('wss://')
+			) {
+				wsUrl = 'ws://' + wsUrl;
+			}
+
+			this.ws = new WebSocket(wsUrl);
+
+			// Guard against duplicate init messages re-invoking the resolve
+			let initResolved = false;
+
+			const connectTimer = setTimeout(() => {
+				if (!initResolved) {
+					reject(new Error(`Hub did not send init message within ${CONNECT_TIMEOUT_MS}ms`));
+					this.ws?.close();
+				}
+			}, CONNECT_TIMEOUT_MS);
+
+			this.ws.onmessage = (event: MessageEvent) => {
+				let data: Record<string, unknown>;
+				try {
+					const raw =
+						typeof event.data === 'string'
+							? event.data
+							: new TextDecoder().decode(
+									event.data as ArrayBuffer,
+								);
+					data = JSON.parse(raw) as Record<string, unknown>;
+				} catch {
+					// Malformed or non-JSON frame — ignore
+					return;
+				}
+
+				// First message should be init
+				if (data.type === 'init' && !initResolved) {
+					initResolved = true;
+					clearTimeout(connectTimer);
+					resolve(data as unknown as InitMessage);
+					return;
+				}
+
+				// Otherwise it's a response to a pending request
+				const response = data as unknown as HubResponse;
+				const entry = this.pending.get(response.id);
+				if (entry) {
+					clearTimeout(entry.timer);
+					this.pending.delete(response.id);
+					entry.resolve(response);
+				}
+			};
+
+			this.ws.onerror = (err: Event) => {
+				// ErrorEvent has a message property; plain Event does not
+				const message =
+					'message' in err &&
+					typeof (err as ErrorEvent).message === 'string'
+						? (err as ErrorEvent).message
+						: `connection to ${wsUrl} failed`;
+				clearTimeout(connectTimer);
+				reject(new Error(`WebSocket error: ${message}`));
+			};
+
+			this.ws.onclose = () => {
+				clearTimeout(connectTimer);
+				// Reject connect() promise if init was never received
+				if (!initResolved) {
+					reject(new Error('WebSocket closed before init message received'));
+				}
+				// Reject all pending requests
+				for (const [id, entry] of this.pending) {
+					clearTimeout(entry.timer);
+					entry.reject(new Error('WebSocket closed'));
+					this.pending.delete(id);
+				}
+				this.ws = null;
+			};
+		});
+	}
+
+	nextId(): string {
+		return crypto.randomUUID();
+	}
+
+	async send(request: HubRequest): Promise<HubResponse> {
+		if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
+			// Return a default ACK if not connected
+			return { id: request.id, actions: [{ action: 'ACK' }] };
+		}
+
+		return new Promise<HubResponse>((resolve, reject) => {
+			const timer = setTimeout(() => {
+				const entry = this.pending.get(request.id);
+				if (entry) {
+					this.pending.delete(request.id);
+					entry.reject(
+						new Error(
+							`Hub response timeout after ${SEND_TIMEOUT_MS}ms for request ${request.id}`,
+						),
+					);
+				}
+			}, SEND_TIMEOUT_MS);
+
+			this.pending.set(request.id, { resolve, reject, timer });
+			this.ws!.send(JSON.stringify(request));
+		});
+	}
+
+	close(): void {
+		if (this.ws) {
+			this.ws.close();
+			this.ws = null;
+		}
+	}
+
+	get connected(): boolean {
+		return this.ws !== null && this.ws.readyState === WebSocket.OPEN;
+	}
+}

package/src/handlers.ts

@@ -0,0 +1,87 @@
+import type { HubAction } from './protocol.ts';
+
+export interface ActionResult {
+	block?: { block: true; reason: string };
+	returnValue?: unknown;
+	// undefined means ACK (proceed normally)
+}
+
+/** Minimal UI surface used by action handlers — avoids a hard dep on pi-coding-agent. */
+interface ActionContext {
+	ui?: {
+		notify(message: string, level?: 'info' | 'warning' | 'error'): void;
+		confirm(title: string, message: string): Promise<boolean>;
+		setStatus(key: string, text?: string): void;
+	};
+}
+
+export async function processActions(
+	actions: HubAction[],
+	ctx: ActionContext,
+): Promise<ActionResult> {
+	let result: ActionResult = {};
+
+	for (const action of actions) {
+		switch (action.action) {
+			case 'ACK':
+				// Terminal: proceed normally
+				result = {};
+				break;
+
+			case 'BLOCK':
+				// Terminal: block
+				result = { block: { block: true, reason: action.reason } };
+				break;
+
+			case 'RETURN':
+				// Terminal: return a specific result
+				result = { returnValue: action.result };
+				break;
+
+			case 'NOTIFY':
+				// Side effect: show notification, continue
+				if (ctx?.ui) {
+					ctx.ui.notify(action.message, action.level ?? 'info');
+				}
+				break;
+
+			case 'STATUS':
+				// Side effect: set status, continue
+				if (ctx?.ui) {
+					ctx.ui.setStatus(action.key, action.text);
+				}
+				break;
+
+			case 'CONFIRM': {
+				// Gate: if user denies, stop and block
+				if (ctx?.ui) {
+					const confirmed = await ctx.ui.confirm(
+						action.title,
+						action.message,
+					);
+					if (!confirmed) {
+						return {
+							block: {
+								block: true,
+								reason: action.deny_reason ?? 'Denied by user',
+							},
+						};
+					}
+				} else {
+					// No UI available — block by default for safety
+					result = {
+						block: {
+							block: true,
+							reason:
+								action.deny_reason ??
+								'Confirmation required but no UI available',
+						},
+					};
+				}
+				break;
+			}
+		}
+	}
+
+	return result;
+}

package/src/index.ts

@@ -0,0 +1,251 @@
+import type {
+	AgentToolResult,
+	ExtensionAPI,
+	ExtensionContext,
+	ExtensionCommandContext,
+} from '@mariozechner/pi-coding-agent';
+import type { TSchema } from '@sinclair/typebox';
+import { HubClient } from './client.ts';
+import { processActions } from './handlers.ts';
+import type { HubResponse, InitMessage } from './protocol.ts';
+
+const HUB_URL_ENV = 'AGENTUITY_CODER_HUB_URL';
+
+// All Pi events we subscribe to (order matters — session_start is handled separately)
+const PROXY_EVENTS = [
+	'session_shutdown',
+	'session_before_switch',
+	'session_switch',
+	'session_before_fork',
+	'session_fork',
+	'session_before_compact',
+	'session_compact',
+	'before_agent_start',
+	'agent_start',
+	'agent_end',
+	'turn_start',
+	'turn_end',
+	'tool_call',
+	'tool_result',
+	'tool_execution_start',
+	'tool_execution_update',
+	'tool_execution_end',
+	'message_start',
+	'message_update',
+	'message_end',
+	'input',
+	'model_select',
+	'context',
+] as const;
+
+// Generic event handler type for the iteration loop
+type GenericEventHandler = (
+	event: string,
+	handler: (event: unknown, ctx: ExtensionContext) => Promise<unknown>,
+) => void;
+
+function log(msg: string): void {
+	console.error(`[agentuity-pi] ${msg}`);
+}
+
+export function agentuityCoderHub(pi: ExtensionAPI) {
+	const hubUrl = process.env[HUB_URL_ENV];
+
+	// No-op if not configured
+	if (!hubUrl) {
+		return;
+	}
+
+	log(`Hub URL: ${hubUrl}`);
+
+	const client = new HubClient();
+
+	// Connect to the Hub and register tools/commands
+	async function connectAndRegister(): Promise<void> {
+		if (client.connected) return;
+
+		log('Connecting to Hub...');
+		let initMsg: InitMessage;
+
+		try {
+			initMsg = await client.connect(hubUrl!);
+		} catch (err) {
+			const msg = err instanceof Error ? err.message : String(err);
+			log(`Failed to connect: ${msg}`);
+			return;
+		}
+
+		log(
+			`Connected. Init: ${initMsg.tools?.length ?? 0} tools, ${initMsg.commands?.length ?? 0} commands`,
+		);
+
+		// Register tools from Hub
+		if (initMsg.tools) {
+			for (const toolDef of initMsg.tools) {
+				log(`Registering tool: ${toolDef.name}`);
+				pi.registerTool({
+					name: toolDef.name,
+					label: toolDef.label,
+					description: toolDef.description,
+					parameters: toolDef.parameters as unknown as TSchema,
+					async execute(
+						toolCallId: string,
+						params: unknown,
+						_signal: AbortSignal | undefined,
+						_onUpdate: unknown,
+						ctx: ExtensionContext,
+					) {
+						log(`Tool execute: ${toolDef.name}`);
+						const id = client.nextId();
+						let response: HubResponse;
+
+						try {
+							response = await client.send({
+								id,
+								type: 'tool',
+								name: toolDef.name,
+								toolCallId,
+								params: params as Record<string, unknown>,
+							});
+						} catch {
+							return {
+								content: [
+									{
+										type: 'text' as const,
+										text: 'Error: Hub connection lost',
+									},
+								],
+								details: {},
+							};
+						}
+
+						const result = await processActions(
+							response.actions,
+							ctx,
+						);
+
+						if (result.returnValue !== undefined) {
+							return result.returnValue as AgentToolResult<unknown>;
+						}
+
+						return {
+							content: [
+								{ type: 'text' as const, text: 'Done' },
+							],
+							details: {},
+						};
+					},
+				});
+			}
+		}
+
+		// Register commands from Hub
+		if (initMsg.commands) {
+			for (const cmdDef of initMsg.commands) {
+				log(`Registering command: /${cmdDef.name}`);
+				pi.registerCommand(cmdDef.name, {
+					description: cmdDef.description,
+					handler: async (
+						args: string,
+						ctx: ExtensionCommandContext,
+					) => {
+						log(`Command execute: /${cmdDef.name}`);
+						const id = client.nextId();
+						let response: HubResponse;
+
+						try {
+							response = await client.send({
+								id,
+								type: 'command',
+								name: cmdDef.name,
+								args,
+							});
+						} catch {
+							ctx.ui.notify('Hub connection lost', 'error');
+							return;
+						}
+
+						await processActions(response.actions, ctx);
+					},
+				});
+			}
+		}
+
+		log('Registration complete');
+	}
+
+	// Helper to send event and process response actions
+	async function sendEvent(
+		eventName: string,
+		data: Record<string, unknown>,
+		ctx: ExtensionContext,
+	): Promise<unknown> {
+		if (!client.connected) return undefined;
+
+		const id = client.nextId();
+		let response: HubResponse;
+
+		try {
+			response = await client.send({
+				id,
+				type: 'event',
+				event: eventName,
+				data,
+			});
+		} catch {
+			return undefined;
+		}
+
+		const result = await processActions(response.actions, ctx);
+
+		if (result.block) return result.block;
+		if (result.returnValue !== undefined) return result.returnValue;
+		return undefined;
+	}
+
+	// Serialize event data — strip non-serializable fields
+	function serializeEvent(event: unknown): Record<string, unknown> {
+		const data: Record<string, unknown> = {};
+		if (event && typeof event === 'object') {
+			for (const [key, value] of Object.entries(event)) {
+				if (typeof value !== 'function' && key !== 'signal') {
+					try {
+						JSON.stringify(value);
+						data[key] = value;
+					} catch {
+						// Skip non-serializable values
+					}
+				}
+			}
+		}
+		return data;
+	}
+
+	// Cast pi.on to generic handler since we iterate over a union of event names
+	// TypeScript overloads require exact string literal matching which doesn't work in loops
+	const onEvent = pi.on.bind(pi) as GenericEventHandler;
+
+	// ── session_start: connect + register BEFORE Pi builds the tool list ──
+	onEvent('session_start', async (event: unknown, ctx: ExtensionContext) => {
+		await connectAndRegister();
+
+		if (client.connected) {
+			return sendEvent('session_start', serializeEvent(event), ctx);
+		}
+	});
+
+	for (const eventName of PROXY_EVENTS) {
+		onEvent(eventName, async (event: unknown, ctx: ExtensionContext) => {
+			if (!client.connected) return undefined;
+			return sendEvent(eventName, serializeEvent(event), ctx);
+		});
+	}
+
+	// Clean up on shutdown
+	pi.on('session_shutdown', async () => {
+		log('Shutting down');
+		client.close();
+	});
+}
+
+export default agentuityCoderHub;

package/src/protocol.ts

@@ -0,0 +1,95 @@
+// ---- Init Message (Server → Client on connect) ----
+
+export interface HubToolDefinition {
+	name: string;
+	label: string;
+	description: string;
+	parameters: Record<string, unknown>; // JSON Schema object
+}
+
+export interface HubCommandDefinition {
+	name: string;
+	description: string;
+}
+
+export interface InitMessage {
+	type: 'init';
+	tools?: HubToolDefinition[];
+	commands?: HubCommandDefinition[];
+}
+
+// ---- Request Messages (Client → Server) ----
+
+export interface EventRequest {
+	id: string;
+	type: 'event';
+	event: string;
+	data: Record<string, unknown>;
+}
+
+export interface ToolRequest {
+	id: string;
+	type: 'tool';
+	name: string;
+	toolCallId: string;
+	params: Record<string, unknown>;
+}
+
+export interface CommandRequest {
+	id: string;
+	type: 'command';
+	name: string;
+	args: string;
+}
+
+export type HubRequest = EventRequest | ToolRequest | CommandRequest;
+
+// ---- Actions ----
+
+export interface AckAction {
+	action: 'ACK';
+}
+
+export interface BlockAction {
+	action: 'BLOCK';
+	reason: string;
+}
+
+export interface ConfirmAction {
+	action: 'CONFIRM';
+	title: string;
+	message: string;
+	deny_reason?: string;
+}
+
+export interface NotifyAction {
+	action: 'NOTIFY';
+	message: string;
+	level?: 'info' | 'warning' | 'error';
+}
+
+export interface ReturnAction {
+	action: 'RETURN';
+	result: unknown;
+}
+
+export interface StatusAction {
+	action: 'STATUS';
+	key: string;
+	text?: string; // undefined = clear status
+}
+
+export type HubAction =
+	| AckAction
+	| BlockAction
+	| ConfirmAction
+	| NotifyAction
+	| ReturnAction
+	| StatusAction;
+
+// ---- Response Message (Server → Client) ----
+
+export interface HubResponse {
+	id: string;
+	actions: HubAction[];
+}