@zgsm/costrict-cli 0.0.1

package/dist/src/zed-integration/acp.test.js

@@ -0,0 +1,214 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi, describe, it, expect, beforeEach } from 'vitest';
+import { AgentSideConnection, RequestError, } from './acp.js';
+import {} from './schema.js';
+import {} from './connection.js';
+import { ReadableStream, WritableStream } from 'node:stream/web';
+const mockConnectionConstructor = vi.hoisted(() => vi.fn(() => ({
+    sendRequest: vi.fn(),
+    sendNotification: vi.fn(),
+})));
+vi.mock('./connection.js', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        Connection: mockConnectionConstructor,
+    };
+});
+describe('acp', () => {
+    describe('RequestError', () => {
+        it('should create a parse error', () => {
+            const error = RequestError.parseError('details');
+            expect(error.code).toBe(-32700);
+            expect(error.message).toBe('Parse error');
+            expect(error.data?.details).toBe('details');
+        });
+        it('should create a method not found error', () => {
+            const error = RequestError.methodNotFound('details');
+            expect(error.code).toBe(-32601);
+            expect(error.message).toBe('Method not found');
+            expect(error.data?.details).toBe('details');
+        });
+        it('should convert to a result', () => {
+            const error = RequestError.internalError('details');
+            const result = error.toResult();
+            expect(result.error.code).toBe(-32603);
+            expect(result.error.message).toBe('Internal error');
+            expect(result.error.data).toEqual({ details: 'details' });
+        });
+    });
+    describe('AgentSideConnection', () => {
+        let mockAgent;
+        let toAgent;
+        let fromAgent;
+        let agentSideConnection;
+        let connectionInstance;
+        beforeEach(() => {
+            vi.clearAllMocks();
+            const initializeResponse = {
+                agentCapabilities: { loadSession: true },
+                authMethods: [],
+                protocolVersion: 1,
+            };
+            const newSessionResponse = { sessionId: 'session-1' };
+            const loadSessionResponse = { sessionId: 'session-1' };
+            mockAgent = {
+                initialize: vi.fn().mockResolvedValue(initializeResponse),
+                newSession: vi.fn().mockResolvedValue(newSessionResponse),
+                loadSession: vi.fn().mockResolvedValue(loadSessionResponse),
+                authenticate: vi.fn(),
+                prompt: vi.fn(),
+                cancel: vi.fn(),
+            };
+            toAgent = new WritableStream();
+            fromAgent = new ReadableStream();
+            agentSideConnection = new AgentSideConnection((_client) => mockAgent, toAgent, fromAgent);
+            // Get the mocked Connection instance
+            connectionInstance = mockConnectionConstructor.mock.results[0].value;
+        });
+        it('should initialize Connection with the correct handler and streams', () => {
+            expect(mockConnectionConstructor).toHaveBeenCalledTimes(1);
+            expect(mockConnectionConstructor).toHaveBeenCalledWith(expect.any(Function), toAgent, fromAgent);
+        });
+        it('should call agent.initialize when Connection handler receives initialize method', async () => {
+            const initializeParams = {
+                clientCapabilities: { fs: { readTextFile: true, writeTextFile: true } },
+                protocolVersion: 1,
+            };
+            const initializeResponse = {
+                agentCapabilities: { loadSession: true },
+                authMethods: [],
+                protocolVersion: 1,
+            };
+            const handler = mockConnectionConstructor.mock.calls[0][0];
+            const result = await handler('initialize', initializeParams);
+            expect(mockAgent.initialize).toHaveBeenCalledWith(initializeParams);
+            expect(result).toEqual(initializeResponse);
+        });
+        it('should call agent.newSession when Connection handler receives session_new method', async () => {
+            const newSessionParams = { cwd: '/tmp', mcpServers: [] };
+            const newSessionResponse = { sessionId: 'session-1' };
+            const handler = mockConnectionConstructor.mock.calls[0][0];
+            const result = await handler('session/new', newSessionParams);
+            expect(mockAgent.newSession).toHaveBeenCalledWith(newSessionParams);
+            expect(result).toEqual(newSessionResponse);
+        });
+        it('should call agent.loadSession when Connection handler receives session_load method', async () => {
+            const loadSessionParams = {
+                cwd: '/tmp',
+                mcpServers: [],
+                sessionId: 'session-1',
+            };
+            const loadSessionResponse = { sessionId: 'session-1' };
+            const handler = mockConnectionConstructor.mock.calls[0][0];
+            const result = await handler('session/load', loadSessionParams);
+            expect(mockAgent.loadSession).toHaveBeenCalledWith(loadSessionParams);
+            expect(result).toEqual(loadSessionResponse);
+        });
+        it('should throw methodNotFound if agent.loadSession is not implemented', async () => {
+            mockAgent.loadSession = undefined; // Simulate not implemented
+            const loadSessionParams = {
+                cwd: '/tmp',
+                mcpServers: [],
+                sessionId: 'session-1',
+            };
+            const handler = mockConnectionConstructor.mock.calls[0][0];
+            await expect(handler('session/load', loadSessionParams)).rejects.toThrow(RequestError.methodNotFound().message);
+        });
+        it('should call agent.authenticate when Connection handler receives authenticate method', async () => {
+            const authenticateParams = {
+                methodId: 'test-auth-method',
+            };
+            const handler = mockConnectionConstructor.mock.calls[0][0];
+            const result = await handler('authenticate', authenticateParams);
+            expect(mockAgent.authenticate).toHaveBeenCalledWith(authenticateParams);
+            expect(result).toBeUndefined();
+        });
+        it('should call agent.prompt when Connection handler receives session_prompt method', async () => {
+            const promptParams = {
+                prompt: [{ type: 'text', text: 'hi' }],
+                sessionId: 'session-1',
+            };
+            const promptResponse = {
+                response: [{ type: 'text', text: 'hello' }],
+                traceId: 'trace-1',
+            };
+            mockAgent.prompt.mockResolvedValue(promptResponse);
+            const handler = mockConnectionConstructor.mock.calls[0][0];
+            const result = await handler('session/prompt', promptParams);
+            expect(mockAgent.prompt).toHaveBeenCalledWith(promptParams);
+            expect(result).toEqual(promptResponse);
+        });
+        it('should call agent.cancel when Connection handler receives session_cancel method', async () => {
+            const cancelParams = { sessionId: 'session-1' };
+            const handler = mockConnectionConstructor.mock.calls[0][0];
+            const result = await handler('session/cancel', cancelParams);
+            expect(mockAgent.cancel).toHaveBeenCalledWith(cancelParams);
+            expect(result).toBeUndefined();
+        });
+        it('should throw methodNotFound for unknown methods', async () => {
+            const handler = mockConnectionConstructor.mock.calls[0][0];
+            await expect(handler('unknown_method', {})).rejects.toThrow(RequestError.methodNotFound().message);
+        });
+        it('should send sessionUpdate notification via connection', async () => {
+            const params = {
+                sessionId: '123',
+                update: {
+                    sessionUpdate: 'user_message_chunk',
+                    content: { type: 'text', text: 'hello' },
+                },
+            };
+            await agentSideConnection.sessionUpdate(params);
+        });
+        it('should send requestPermission request via connection', async () => {
+            const params = {
+                sessionId: '123',
+                toolCall: {
+                    toolCallId: 'tool-1',
+                    title: 'Test Tool',
+                    kind: 'other',
+                    status: 'pending',
+                },
+                options: [
+                    {
+                        optionId: 'option-1',
+                        name: 'Allow',
+                        kind: 'allow_once',
+                    },
+                ],
+            };
+            const response = {
+                outcome: { outcome: 'selected', optionId: 'option-1' },
+            };
+            connectionInstance.sendRequest.mockResolvedValue(response);
+            const result = await agentSideConnection.requestPermission(params);
+            expect(connectionInstance.sendRequest).toHaveBeenCalledWith('session/request_permission', params);
+            expect(result).toEqual(response);
+        });
+        it('should send readTextFile request via connection', async () => {
+            const params = { path: '/a/b.txt', sessionId: 'session-1' };
+            const response = { content: 'file content' };
+            connectionInstance.sendRequest.mockResolvedValue(response);
+            const result = await agentSideConnection.readTextFile(params);
+            expect(connectionInstance.sendRequest).toHaveBeenCalledWith('fs/read_text_file', params);
+            expect(result).toEqual(response);
+        });
+        it('should send writeTextFile request via connection', async () => {
+            const params = {
+                path: '/a/b.txt',
+                content: 'new content',
+                sessionId: 'session-1',
+            };
+            const response = { success: true };
+            connectionInstance.sendRequest.mockResolvedValue(response);
+            const result = await agentSideConnection.writeTextFile(params);
+            expect(connectionInstance.sendRequest).toHaveBeenCalledWith('fs/write_text_file', params);
+            expect(result).toEqual(response);
+        });
+    });
+});
+//# sourceMappingURL=acp.test.js.map

package/dist/src/zed-integration/acp.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"acp.test.js","sourceRoot":"","sources":["../../../src/zed-integration/acp.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAa,MAAM,QAAQ,CAAC;AACzE,OAAO,EACL,mBAAmB,EACnB,YAAY,GAGb,MAAM,UAAU,CAAC;AAClB,OAAO,EAAsB,MAAM,aAAa,CAAC;AACjD,OAAO,EAAsB,MAAM,iBAAiB,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAEjE,MAAM,yBAAyB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAChD,EAAE,CAAC,EAAE,CAMH,GAAG,EAAE,CAAC,CAAC;IACP,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;IACpB,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CACJ,CAAC;AAEF,EAAE,CAAC,IAAI,CAAC,iBAAiB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAClD,MAAM,MAAM,GAAG,MAAM,cAAc,EAAE,CAAC;IACtC,OAAO;QACL,GAAI,MAAiB;QACrB,UAAU,EAAE,yBAAyB;KACtC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE;IACnB,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,KAAK,GAAG,YAAY,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YACjD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC;YAChC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC1C,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,KAAK,GAAG,YAAY,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;YACrD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC;YAChC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YAC/C,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,KAAK,GAAG,YAAY,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YACpD,MAAM,MAAM,GAAG,KAAK,CAAC,QAAQ,EAA8B,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,IAAI,SAAgB,CAAC;QAErB,IAAI,OAAmC,CAAC;QACxC,IAAI,SAAqC,CAAC;QAC1C,IAAI,mBAAwC,CAAC;QAC7C,IAAI,kBAAkE,CAAC;QAEvE,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,aAAa,EAAE,CAAC;YAEnB,MAAM,kBAAkB,GAAG;gBACzB,iBAAiB,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE;gBACxC,WAAW,EAAE,EAAE;gBACf,eAAe,EAAE,CAAC;aACnB,CAAC;YACF,MAAM,kBAAkB,GAAG,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC;YACtD,MAAM,mBAAmB,GAAG,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC;YAEvD,SAAS,GAAG;gBACV,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,kBAAkB,CAAC;gBACzD,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,kBAAkB,CAAC;gBACzD,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,mBAAmB,CAAC;gBAC3D,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;gBACrB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;gBACf,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;aAChB,CAAC;YAEF,OAAO,GAAG,IAAI,cAAc,EAAc,CAAC;YAC3C,SAAS,GAAG,IAAI,cAAc,EAAc,CAAC;YAE7C,mBAAmB,GAAG,IAAI,mBAAmB,CAC3C,CAAC,OAAe,EAAE,EAAE,CAAC,SAAS,EAC9B,OAAO,EACP,SAAS,CACV,CAAC;YAEF,qCAAqC;YACrC,kBAAkB,GAAG,yBAAyB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;YAC3E,MAAM,CAAC,yBAAyB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,CAAC,yBAAyB,CAAC,CAAC,oBAAoB,CACpD,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EACpB,OAAO,EACP,SAAS,CACV,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;YAC/F,MAAM,gBAAgB,GAAG;gBACvB,kBAAkB,EAAE,EAAE,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE;gBACvE,eAAe,EAAE,CAAC;aACnB,CAAC;YACF,MAAM,kBAAkB,GAAG;gBACzB,iBAAiB,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE;gBACxC,WAAW,EAAE,EAAE;gBACf,eAAe,EAAE,CAAC;aACnB,CAAC;YACF,MAAM,OAAO,GAAG,yBAAyB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,YAAY,EAAE,gBAAgB,CAAC,CAAC;YAE7D,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;YACpE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;YAChG,MAAM,gBAAgB,GAAG,EAAE,GAAG,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;YACzD,MAAM,kBAAkB,GAAG,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC;YACtD,MAAM,OAAO,GAAG,yBAAyB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,aAAa,EAAE,gBAAgB,CAAC,CAAC;YAE9D,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;YACpE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oFAAoF,EAAE,KAAK,IAAI,EAAE;YAClG,MAAM,iBAAiB,GAAG;gBACxB,GAAG,EAAE,MAAM;gBACX,UAAU,EAAE,EAAE;gBACd,SAAS,EAAE,WAAW;aACvB,CAAC;YACF,MAAM,mBAAmB,GAAG,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC;YACvD,MAAM,OAAO,GAAG,yBAAyB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,cAAc,EAAE,iBAAiB,CAAC,CAAC;YAEhE,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;YACtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,SAAS,CAAC,WAAW,GAAG,SAAS,CAAC,CAAC,2BAA2B;YAC9D,MAAM,iBAAiB,GAAG;gBACxB,GAAG,EAAE,MAAM;gBACX,UAAU,EAAE,EAAE;gBACd,SAAS,EAAE,WAAW;aACvB,CAAC;YACF,MAAM,OAAO,GAAG,yBAAyB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,iBAAiB,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACtE,YAAY,CAAC,cAAc,EAAE,CAAC,OAAO,CACtC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qFAAqF,EAAE,KAAK,IAAI,EAAE;YACnG,MAAM,kBAAkB,GAAG;gBACzB,QAAQ,EAAE,kBAAkB;aAC7B,CAAC;YACF,MAAM,OAAO,GAAG,yBAAyB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,cAAc,EAAE,kBAAkB,CAAC,CAAC;YAEjE,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,kBAAkB,CAAC,CAAC;YACxE,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;YAC/F,MAAM,YAAY,GAAG;gBACnB,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBACtC,SAAS,EAAE,WAAW;aACvB,CAAC;YACF,MAAM,cAAc,GAAG;gBACrB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC3C,OAAO,EAAE,SAAS;aACnB,CAAC;YACD,SAAS,CAAC,MAAe,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAC7D,MAAM,OAAO,GAAG,yBAAyB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;YAE7D,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;YAC/F,MAAM,YAAY,GAAG,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC;YAChD,MAAM,OAAO,GAAG,yBAAyB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;YAE7D,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,OAAO,GAAG,yBAAyB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,MAAM,CAAC,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzD,YAAY,CAAC,cAAc,EAAE,CAAC,OAAO,CACtC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,MAAM,GAAG;gBACb,SAAS,EAAE,KAAK;gBAChB,MAAM,EAAE;oBACN,aAAa,EAAE,oBAA6B;oBAC5C,OAAO,EAAE,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,OAAO,EAAE;iBAClD;aACF,CAAC;YACF,MAAM,mBAAmB,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,MAAM,GAAG;gBACb,SAAS,EAAE,KAAK;gBAChB,QAAQ,EAAE;oBACR,UAAU,EAAE,QAAQ;oBACpB,KAAK,EAAE,WAAW;oBAClB,IAAI,EAAE,OAAgB;oBACtB,MAAM,EAAE,SAAkB;iBAC3B;gBACD,OAAO,EAAE;oBACP;wBACE,QAAQ,EAAE,UAAU;wBACpB,IAAI,EAAE,OAAO;wBACb,IAAI,EAAE,YAAqB;qBAC5B;iBACF;aACF,CAAC;YACF,MAAM,QAAQ,GAAG;gBACf,OAAO,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,UAAU,EAAE;aACvD,CAAC;YACF,kBAAkB,CAAC,WAAW,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YACnE,MAAM,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACzD,4BAA4B,EAC5B,MAAM,CACP,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC;YAC5D,MAAM,QAAQ,GAAG,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC;YAC7C,kBAAkB,CAAC,WAAW,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAC9D,MAAM,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACzD,mBAAmB,EACnB,MAAM,CACP,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,MAAM,GAAG;gBACb,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,aAAa;gBACtB,SAAS,EAAE,WAAW;aACvB,CAAC;YACF,MAAM,QAAQ,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACnC,kBAAkB,CAAC,WAAW,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;YAC/D,MAAM,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACzD,oBAAoB,EACpB,MAAM,CACP,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/zed-integration/connection.d.ts

@@ -0,0 +1,28 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type Result } from './schema.js';
+import type { WritableStream, ReadableStream } from 'node:stream/web';
+export declare class RequestError extends Error {
+    code: number;
+    data?: {
+        details?: string;
+    };
+    constructor(code: number, message: string, details?: string);
+    static parseError(details?: string): RequestError;
+    static invalidRequest(details?: string): RequestError;
+    static methodNotFound(details?: string): RequestError;
+    static invalidParams(details?: string): RequestError;
+    static internalError(details?: string): RequestError;
+    static authRequired(details?: string): RequestError;
+    toResult<T>(): Result<T>;
+}
+export type MethodHandler = (method: string, params: unknown) => Promise<unknown>;
+export declare class Connection {
+    #private;
+    constructor(handler: MethodHandler, peerInput: WritableStream<Uint8Array>, peerOutput: ReadableStream<Uint8Array>);
+    sendRequest<Req, Resp>(method: string, params?: Req): Promise<Resp>;
+    sendNotification<N>(method: string, params?: N): Promise<void>;
+}

package/dist/src/zed-integration/connection.js

@@ -0,0 +1,165 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { z } from 'zod';
+import { coreEvents } from '@google/gemini-cli-core';
+import {} from './schema.js';
+export class RequestError extends Error {
+    code;
+    data;
+    constructor(code, message, details) {
+        super(message);
+        this.code = code;
+        this.name = 'RequestError';
+        if (details) {
+            this.data = { details };
+        }
+    }
+    static parseError(details) {
+        return new RequestError(-32700, 'Parse error', details);
+    }
+    static invalidRequest(details) {
+        return new RequestError(-32600, 'Invalid request', details);
+    }
+    static methodNotFound(details) {
+        return new RequestError(-32601, 'Method not found', details);
+    }
+    static invalidParams(details) {
+        return new RequestError(-32602, 'Invalid params', details);
+    }
+    static internalError(details) {
+        return new RequestError(-32603, 'Internal error', details);
+    }
+    static authRequired(details) {
+        return new RequestError(-32000, 'Authentication required', details);
+    }
+    toResult() {
+        return {
+            error: {
+                code: this.code,
+                message: this.message,
+                data: this.data,
+            },
+        };
+    }
+}
+export class Connection {
+    #pendingResponses = new Map();
+    #nextRequestId = 0;
+    #handler;
+    #peerInput;
+    #writeQueue = Promise.resolve();
+    #textEncoder;
+    constructor(handler, peerInput, peerOutput) {
+        this.#handler = handler;
+        this.#peerInput = peerInput;
+        this.#textEncoder = new TextEncoder();
+        // eslint-disable-next-line @typescript-eslint/no-floating-promises
+        this.#receive(peerOutput);
+    }
+    async #receive(output) {
+        let content = '';
+        const decoder = new TextDecoder();
+        for await (const chunk of output) {
+            content += decoder.decode(chunk, { stream: true });
+            const lines = content.split('\n');
+            content = lines.pop() || '';
+            for (const line of lines) {
+                const trimmedLine = line.trim();
+                if (trimmedLine) {
+                    const message = JSON.parse(trimmedLine);
+                    // eslint-disable-next-line @typescript-eslint/no-floating-promises
+                    this.#processMessage(message);
+                }
+            }
+        }
+    }
+    async #processMessage(message) {
+        if ('method' in message && 'id' in message) {
+            // It's a request
+            const response = await this.#tryCallHandler(message.method, message.params);
+            await this.#sendMessage({
+                jsonrpc: '2.0',
+                id: message.id,
+                ...response,
+            });
+        }
+        else if ('method' in message) {
+            // It's a notification
+            await this.#tryCallHandler(message.method, message.params);
+        }
+        else if ('id' in message) {
+            // It's a response
+            this.#handleResponse(message);
+        }
+    }
+    async #tryCallHandler(method, params) {
+        try {
+            const result = await this.#handler(method, params);
+            return { result: result ?? null };
+        }
+        catch (error) {
+            if (error instanceof RequestError) {
+                return error.toResult();
+            }
+            if (error instanceof z.ZodError) {
+                return RequestError.invalidParams(JSON.stringify(error.format(), undefined, 2)).toResult();
+            }
+            let details;
+            if (error instanceof Error) {
+                details = error.message;
+            }
+            else if (typeof error === 'object' &&
+                error != null &&
+                'message' in error &&
+                typeof error.message === 'string') {
+                details = error.message;
+            }
+            return RequestError.internalError(details).toResult();
+        }
+    }
+    #handleResponse(response) {
+        const pendingResponse = this.#pendingResponses.get(response.id);
+        if (pendingResponse) {
+            if ('result' in response) {
+                pendingResponse.resolve(response.result);
+            }
+            else if ('error' in response) {
+                pendingResponse.reject(response.error);
+            }
+            this.#pendingResponses.delete(response.id);
+        }
+    }
+    async sendRequest(method, params) {
+        const id = this.#nextRequestId++;
+        const responsePromise = new Promise((resolve, reject) => {
+            this.#pendingResponses.set(id, { resolve, reject });
+        });
+        await this.#sendMessage({ jsonrpc: '2.0', id, method, params });
+        return responsePromise;
+    }
+    async sendNotification(method, params) {
+        await this.#sendMessage({ jsonrpc: '2.0', method, params });
+    }
+    async #sendMessage(json) {
+        const content = JSON.stringify(json) + '\n';
+        this.#writeQueue = this.#writeQueue
+            .then(async () => {
+            const writer = this.#peerInput.getWriter();
+            try {
+                await writer.write(this.#textEncoder.encode(content));
+            }
+            finally {
+                writer.releaseLock();
+            }
+        })
+            .catch((error) => {
+            // Continue processing writes on error
+            coreEvents.emitFeedback('error', 'ACP write error.', error);
+        });
+        return this.#writeQueue;
+    }
+}
+//# sourceMappingURL=connection.js.map

package/dist/src/zed-integration/connection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"connection.js","sourceRoot":"","sources":["../../../src/zed-integration/connection.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AACrD,OAAO,EAAmC,MAAM,aAAa,CAAC;AAG9D,MAAM,OAAO,YAAa,SAAQ,KAAK;IAI5B;IAHT,IAAI,CAAwB;IAE5B,YACS,IAAY,EACnB,OAAe,EACf,OAAgB;QAEhB,KAAK,CAAC,OAAO,CAAC,CAAC;QAJR,SAAI,GAAJ,IAAI,CAAQ;QAKnB,IAAI,CAAC,IAAI,GAAG,cAAc,CAAC;QAC3B,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,IAAI,GAAG,EAAE,OAAO,EAAE,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,MAAM,CAAC,UAAU,CAAC,OAAgB;QAChC,OAAO,IAAI,YAAY,CAAC,CAAC,KAAK,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;IAC1D,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,OAAgB;QACpC,OAAO,IAAI,YAAY,CAAC,CAAC,KAAK,EAAE,iBAAiB,EAAE,OAAO,CAAC,CAAC;IAC9D,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,OAAgB;QACpC,OAAO,IAAI,YAAY,CAAC,CAAC,KAAK,EAAE,kBAAkB,EAAE,OAAO,CAAC,CAAC;IAC/D,CAAC;IAED,MAAM,CAAC,aAAa,CAAC,OAAgB;QACnC,OAAO,IAAI,YAAY,CAAC,CAAC,KAAK,EAAE,gBAAgB,EAAE,OAAO,CAAC,CAAC;IAC7D,CAAC;IAED,MAAM,CAAC,aAAa,CAAC,OAAgB;QACnC,OAAO,IAAI,YAAY,CAAC,CAAC,KAAK,EAAE,gBAAgB,EAAE,OAAO,CAAC,CAAC;IAC7D,CAAC;IAED,MAAM,CAAC,YAAY,CAAC,OAAgB;QAClC,OAAO,IAAI,YAAY,CAAC,CAAC,KAAK,EAAE,yBAAyB,EAAE,OAAO,CAAC,CAAC;IACtE,CAAC;IAED,QAAQ;QACN,OAAO;YACL,KAAK,EAAE;gBACL,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,IAAI,EAAE,IAAI,CAAC,IAAI;aAChB;SACF,CAAC;IACJ,CAAC;CACF;AAgCD,MAAM,OAAO,UAAU;IACrB,iBAAiB,GAA0C,IAAI,GAAG,EAAE,CAAC;IACrE,cAAc,GAAW,CAAC,CAAC;IAC3B,QAAQ,CAAgB;IACxB,UAAU,CAA6B;IACvC,WAAW,GAAkB,OAAO,CAAC,OAAO,EAAE,CAAC;IAC/C,YAAY,CAAc;IAE1B,YACE,OAAsB,EACtB,SAAqC,EACrC,UAAsC;QAEtC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QACxB,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC5B,IAAI,CAAC,YAAY,GAAG,IAAI,WAAW,EAAE,CAAC;QACtC,mEAAmE;QACnE,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAkC;QAC/C,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,OAAO,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YACnD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAClC,OAAO,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;YAE5B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;gBAEhC,IAAI,WAAW,EAAE,CAAC;oBAChB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;oBACxC,mEAAmE;oBACnE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBAChC,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,OAAmB;QACvC,IAAI,QAAQ,IAAI,OAAO,IAAI,IAAI,IAAI,OAAO,EAAE,CAAC;YAC3C,iBAAiB;YACjB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,eAAe,CACzC,OAAO,CAAC,MAAM,EACd,OAAO,CAAC,MAAM,CACf,CAAC;YAEF,MAAM,IAAI,CAAC,YAAY,CAAC;gBACtB,OAAO,EAAE,KAAK;gBACd,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,GAAG,QAAQ;aACZ,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,QAAQ,IAAI,OAAO,EAAE,CAAC;YAC/B,sBAAsB;YACtB,MAAM,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QAC7D,CAAC;aAAM,IAAI,IAAI,IAAI,OAAO,EAAE,CAAC;YAC3B,kBAAkB;YAClB,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,MAAc,EACd,MAAgB;QAEhB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACnD,OAAO,EAAE,MAAM,EAAE,MAAM,IAAI,IAAI,EAAE,CAAC;QACpC,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,IAAI,KAAK,YAAY,YAAY,EAAE,CAAC;gBAClC,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC1B,CAAC;YAED,IAAI,KAAK,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC;gBAChC,OAAO,YAAY,CAAC,aAAa,CAC/B,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC,CAC7C,CAAC,QAAQ,EAAE,CAAC;YACf,CAAC;YAED,IAAI,OAAO,CAAC;YAEZ,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;YAC1B,CAAC;iBAAM,IACL,OAAO,KAAK,KAAK,QAAQ;gBACzB,KAAK,IAAI,IAAI;gBACb,SAAS,IAAI,KAAK;gBAClB,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,EACjC,CAAC;gBACD,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;YAC1B,CAAC;YAED,OAAO,YAAY,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;QACxD,CAAC;IACH,CAAC;IAED,eAAe,CAAC,QAAqB;QACnC,MAAM,eAAe,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAChE,IAAI,eAAe,EAAE,CAAC;YACpB,IAAI,QAAQ,IAAI,QAAQ,EAAE,CAAC;gBACzB,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAC3C,CAAC;iBAAM,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YACzC,CAAC;YACD,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAY,MAAc,EAAE,MAAY;QACvD,MAAM,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QACjC,MAAM,eAAe,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACtD,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QACH,MAAM,IAAI,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QAChE,OAAO,eAAgC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAI,MAAc,EAAE,MAAU;QAClD,MAAM,IAAI,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,IAAgB;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;QAC5C,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW;aAChC,IAAI,CAAC,KAAK,IAAI,EAAE;YACf,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC;YAC3C,IAAI,CAAC;gBACH,MAAM,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;YACxD,CAAC;oBAAS,CAAC;gBACT,MAAM,CAAC,WAAW,EAAE,CAAC;YACvB,CAAC;QACH,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACf,sCAAsC;YACtC,UAAU,CAAC,YAAY,CAAC,OAAO,EAAE,kBAAkB,EAAE,KAAK,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QACL,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;CACF"}

package/dist/src/zed-integration/connection.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/zed-integration/connection.test.js

@@ -0,0 +1,175 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { Connection, RequestError } from './connection.js';
+import { ReadableStream, WritableStream } from 'node:stream/web';
+describe('Connection', () => {
+    let toPeer;
+    let fromPeer;
+    let peerController;
+    let receivedChunks = [];
+    let connection;
+    let handler;
+    beforeEach(() => {
+        receivedChunks = [];
+        toPeer = new WritableStream({
+            write(chunk) {
+                const str = new TextDecoder().decode(chunk);
+                receivedChunks.push(str);
+            },
+        });
+        fromPeer = new ReadableStream({
+            start(controller) {
+                peerController = controller;
+            },
+        });
+        handler = vi.fn();
+        connection = new Connection(handler, toPeer, fromPeer);
+    });
+    afterEach(() => {
+        vi.clearAllMocks();
+    });
+    it('should send a request and receive a response', async () => {
+        const responsePromise = connection.sendRequest('testMethod', {
+            key: 'value',
+        });
+        // Verify request was sent
+        await vi.waitFor(() => {
+            expect(receivedChunks.length).toBeGreaterThan(0);
+        });
+        const request = JSON.parse(receivedChunks[0]);
+        expect(request).toMatchObject({
+            jsonrpc: '2.0',
+            method: 'testMethod',
+            params: { key: 'value' },
+        });
+        expect(request.id).toBeDefined();
+        // Simulate response
+        const response = {
+            jsonrpc: '2.0',
+            id: request.id,
+            result: { success: true },
+        };
+        peerController.enqueue(new TextEncoder().encode(JSON.stringify(response) + '\n'));
+        const result = await responsePromise;
+        expect(result).toEqual({ success: true });
+    });
+    it('should send a notification', async () => {
+        await connection.sendNotification('notifyMethod', { key: 'value' });
+        await vi.waitFor(() => {
+            expect(receivedChunks.length).toBeGreaterThan(0);
+        });
+        const notification = JSON.parse(receivedChunks[0]);
+        expect(notification).toMatchObject({
+            jsonrpc: '2.0',
+            method: 'notifyMethod',
+            params: { key: 'value' },
+        });
+        expect(notification.id).toBeUndefined();
+    });
+    it('should handle incoming requests', async () => {
+        handler.mockResolvedValue({ result: 'ok' });
+        const request = {
+            jsonrpc: '2.0',
+            id: 1,
+            method: 'incomingMethod',
+            params: { foo: 'bar' },
+        };
+        peerController.enqueue(new TextEncoder().encode(JSON.stringify(request) + '\n'));
+        // Wait for handler to be called and response to be written
+        await vi.waitFor(() => {
+            expect(handler).toHaveBeenCalledWith('incomingMethod', { foo: 'bar' });
+            expect(receivedChunks.length).toBeGreaterThan(0);
+        });
+        const response = JSON.parse(receivedChunks[receivedChunks.length - 1]);
+        expect(response).toMatchObject({
+            jsonrpc: '2.0',
+            id: 1,
+            result: { result: 'ok' },
+        });
+    });
+    it('should handle incoming notifications', async () => {
+        const notification = {
+            jsonrpc: '2.0',
+            method: 'incomingNotify',
+            params: { foo: 'bar' },
+        };
+        peerController.enqueue(new TextEncoder().encode(JSON.stringify(notification) + '\n'));
+        // Wait for handler to be called
+        await vi.waitFor(() => {
+            expect(handler).toHaveBeenCalledWith('incomingNotify', { foo: 'bar' });
+        });
+        // Notifications don't send responses
+        expect(receivedChunks.length).toBe(0);
+    });
+    it('should handle request errors from handler', async () => {
+        handler.mockRejectedValue(new Error('Handler failed'));
+        const request = {
+            jsonrpc: '2.0',
+            id: 2,
+            method: 'failMethod',
+        };
+        peerController.enqueue(new TextEncoder().encode(JSON.stringify(request) + '\n'));
+        await vi.waitFor(() => {
+            expect(receivedChunks.length).toBeGreaterThan(0);
+        });
+        const response = JSON.parse(receivedChunks[receivedChunks.length - 1]);
+        expect(response).toMatchObject({
+            jsonrpc: '2.0',
+            id: 2,
+            error: {
+                code: -32603,
+                message: 'Internal error',
+                data: { details: 'Handler failed' },
+            },
+        });
+    });
+    it('should handle RequestError from handler', async () => {
+        handler.mockRejectedValue(RequestError.methodNotFound('Unknown method'));
+        const request = {
+            jsonrpc: '2.0',
+            id: 3,
+            method: 'unknown',
+        };
+        peerController.enqueue(new TextEncoder().encode(JSON.stringify(request) + '\n'));
+        await vi.waitFor(() => {
+            expect(receivedChunks.length).toBeGreaterThan(0);
+        });
+        const response = JSON.parse(receivedChunks[receivedChunks.length - 1]);
+        expect(response).toMatchObject({
+            jsonrpc: '2.0',
+            id: 3,
+            error: {
+                code: -32601,
+                message: 'Method not found',
+                data: { details: 'Unknown method' },
+            },
+        });
+    });
+    it('should handle response errors', async () => {
+        const responsePromise = connection.sendRequest('testMethod');
+        // Verify request was sent
+        await vi.waitFor(() => {
+            expect(receivedChunks.length).toBeGreaterThan(0);
+        });
+        const request = JSON.parse(receivedChunks[0]);
+        // Simulate error response
+        const response = {
+            jsonrpc: '2.0',
+            id: request.id,
+            error: {
+                code: -32000,
+                message: 'Custom error',
+            },
+        };
+        peerController.enqueue(new TextEncoder().encode(JSON.stringify(response) + '\n'));
+        await expect(responsePromise).rejects.toMatchObject({
+            code: -32000,
+            message: 'Custom error',
+        });
+    });
+});
+//# sourceMappingURL=connection.test.js.map

package/dist/src/zed-integration/connection.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"connection.test.js","sourceRoot":"","sources":["../../../src/zed-integration/connection.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC3D,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAEjE,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,IAAI,MAAkC,CAAC;IACvC,IAAI,QAAoC,CAAC;IACzC,IAAI,cAA2D,CAAC;IAChE,IAAI,cAAc,GAAa,EAAE,CAAC;IAClC,IAAI,UAAsB,CAAC;IAC3B,IAAI,OAAiC,CAAC;IAEtC,UAAU,CAAC,GAAG,EAAE;QACd,cAAc,GAAG,EAAE,CAAC;QACpB,MAAM,GAAG,IAAI,cAAc,CAAC;YAC1B,KAAK,CAAC,KAAK;gBACT,MAAM,GAAG,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC5C,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC3B,CAAC;SACF,CAAC,CAAC;QAEH,QAAQ,GAAG,IAAI,cAAc,CAAC;YAC5B,KAAK,CAAC,UAAU;gBACd,cAAc,GAAG,UAAU,CAAC;YAC9B,CAAC;SACF,CAAC,CAAC;QAEH,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAClB,UAAU,GAAG,IAAI,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,eAAe,GAAG,UAAU,CAAC,WAAW,CAAC,YAAY,EAAE;YAC3D,GAAG,EAAE,OAAO;SACb,CAAC,CAAC;QAEH,0BAA0B;QAC1B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;YACpB,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YAC5B,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,YAAY;YACpB,MAAM,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE;SACzB,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QAEjC,oBAAoB;QACpB,MAAM,QAAQ,GAAG;YACf,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SAC1B,CAAC;QACF,cAAc,CAAC,OAAO,CACpB,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,CAC1D,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC1C,MAAM,UAAU,CAAC,gBAAgB,CAAC,cAAc,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC;QAEpE,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;YACpB,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QACH,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,CAAC,YAAY,CAAC,CAAC,aAAa,CAAC;YACjC,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,cAAc;YACtB,MAAM,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE;SACzB,CAAC,CAAC;QACH,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,OAAO,CAAC,iBAAiB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAE5C,MAAM,OAAO,GAAG;YACd,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,CAAC;YACL,MAAM,EAAE,gBAAgB;YACxB,MAAM,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE;SACvB,CAAC;QACF,cAAc,CAAC,OAAO,CACpB,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,CACzD,CAAC;QAEF,2DAA2D;QAC3D,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;YACpB,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;YACvE,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QACvE,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,CAAC;YAC7B,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,CAAC;YACL,MAAM,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;SACzB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,YAAY,GAAG;YACnB,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,gBAAgB;YACxB,MAAM,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE;SACvB,CAAC;QACF,cAAc,CAAC,OAAO,CACpB,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,GAAG,IAAI,CAAC,CAC9D,CAAC;QAEF,gCAAgC;QAChC,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;YACpB,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;QACH,qCAAqC;QACrC,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,OAAO,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC;QAEvD,MAAM,OAAO,GAAG;YACd,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,CAAC;YACL,MAAM,EAAE,YAAY;SACrB,CAAC;QACF,cAAc,CAAC,OAAO,CACpB,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,CACzD,CAAC;QAEF,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;YACpB,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QACvE,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,CAAC;YAC7B,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,CAAC;YACL,KAAK,EAAE;gBACL,IAAI,EAAE,CAAC,KAAK;gBACZ,OAAO,EAAE,gBAAgB;gBACzB,IAAI,EAAE,EAAE,OAAO,EAAE,gBAAgB,EAAE;aACpC;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,OAAO,CAAC,iBAAiB,CAAC,YAAY,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC,CAAC;QAEzE,MAAM,OAAO,GAAG;YACd,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,CAAC;YACL,MAAM,EAAE,SAAS;SAClB,CAAC;QACF,cAAc,CAAC,OAAO,CACpB,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,CACzD,CAAC;QAEF,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;YACpB,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QACvE,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,CAAC;YAC7B,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,CAAC;YACL,KAAK,EAAE;gBACL,IAAI,EAAE,CAAC,KAAK;gBACZ,OAAO,EAAE,kBAAkB;gBAC3B,IAAI,EAAE,EAAE,OAAO,EAAE,gBAAgB,EAAE;aACpC;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,eAAe,GAAG,UAAU,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;QAE7D,0BAA0B;QAC1B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;YACpB,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;QAE9C,0BAA0B;QAC1B,MAAM,QAAQ,GAAG;YACf,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,KAAK,EAAE;gBACL,IAAI,EAAE,CAAC,KAAK;gBACZ,OAAO,EAAE,cAAc;aACxB;SACF,CAAC;QACF,cAAc,CAAC,OAAO,CACpB,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,CAC1D,CAAC;QAEF,MAAM,MAAM,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;YAClD,IAAI,EAAE,CAAC,KAAK;YACZ,OAAO,EAAE,cAAc;SACxB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/zed-integration/fileSystemService.d.ts

@@ -0,0 +1,19 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { FileSystemService } from '@google/gemini-cli-core';
+import type * as acp from './acp.js';
+/**
+ * ACP client-based implementation of FileSystemService
+ */
+export declare class AcpFileSystemService implements FileSystemService {
+    private readonly client;
+    private readonly sessionId;
+    private readonly capabilities;
+    private readonly fallback;
+    constructor(client: acp.Client, sessionId: string, capabilities: acp.FileSystemCapability, fallback: FileSystemService);
+    readTextFile(filePath: string): Promise<string>;
+    writeTextFile(filePath: string, content: string): Promise<void>;
+}