@didim365/agent-cli-core 0.1.0

package/dist/src/code_assist/server.test.js

@@ -0,0 +1,453 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { beforeEach, describe, it, expect, vi, afterEach } from 'vitest';
+import { CodeAssistServer } from './server.js';
+import { OAuth2Client } from 'google-auth-library';
+import { UserTierId, ActionStatus } from './types.js';
+import { FinishReason } from '@google/genai';
+vi.mock('google-auth-library');
+function createTestServer(headers = {}) {
+    const mockRequest = vi.fn();
+    const client = { request: mockRequest };
+    const server = new CodeAssistServer(client, 'test-project', { headers }, 'test-session', UserTierId.FREE);
+    return { server, mockRequest, client };
+}
+describe('CodeAssistServer', () => {
+    beforeEach(() => {
+        vi.resetAllMocks();
+    });
+    it('should be able to be constructed', () => {
+        const auth = new OAuth2Client();
+        const server = new CodeAssistServer(auth, 'test-project', {}, 'test-session', UserTierId.FREE);
+        expect(server).toBeInstanceOf(CodeAssistServer);
+    });
+    it('should call the generateContent endpoint', async () => {
+        const { server, mockRequest } = createTestServer({
+            'x-custom-header': 'test-value',
+        });
+        const mockResponseData = {
+            response: {
+                candidates: [
+                    {
+                        index: 0,
+                        content: {
+                            role: 'model',
+                            parts: [{ text: 'response' }],
+                        },
+                        finishReason: FinishReason.STOP,
+                        safetyRatings: [],
+                    },
+                ],
+            },
+        };
+        mockRequest.mockResolvedValue({ data: mockResponseData });
+        const response = await server.generateContent({
+            model: 'test-model',
+            contents: [{ role: 'user', parts: [{ text: 'request' }] }],
+        }, 'user-prompt-id');
+        expect(mockRequest).toHaveBeenCalledWith({
+            url: expect.stringContaining(':generateContent'),
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                'x-custom-header': 'test-value',
+            },
+            responseType: 'json',
+            body: expect.any(String),
+            signal: undefined,
+        });
+        const requestBody = JSON.parse(mockRequest.mock.calls[0][0].body);
+        expect(requestBody.user_prompt_id).toBe('user-prompt-id');
+        expect(requestBody.project).toBe('test-project');
+        expect(response.candidates?.[0]?.content?.parts?.[0]?.text).toBe('response');
+    });
+    it('should detect error in generateContent response', async () => {
+        const { server, mockRequest } = createTestServer();
+        const mockResponseData = {
+            traceId: 'test-trace-id',
+            response: {
+                candidates: [
+                    {
+                        index: 0,
+                        content: {
+                            role: 'model',
+                            parts: [
+                                { text: 'response' },
+                                { functionCall: { name: 'test', args: {} } },
+                            ],
+                        },
+                        finishReason: FinishReason.SAFETY,
+                        safetyRatings: [],
+                    },
+                ],
+            },
+        };
+        mockRequest.mockResolvedValue({ data: mockResponseData });
+        const recordConversationOfferedSpy = vi.spyOn(server, 'recordConversationOffered');
+        await server.generateContent({
+            model: 'test-model',
+            contents: [{ role: 'user', parts: [{ text: 'request' }] }],
+        }, 'user-prompt-id');
+        expect(recordConversationOfferedSpy).toHaveBeenCalledWith(expect.objectContaining({
+            status: ActionStatus.ACTION_STATUS_ERROR_UNKNOWN,
+        }));
+    });
+    it('should record conversation offered on successful generateContent', async () => {
+        const { server, mockRequest } = createTestServer();
+        const mockResponseData = {
+            traceId: 'test-trace-id',
+            response: {
+                candidates: [
+                    {
+                        index: 0,
+                        content: {
+                            role: 'model',
+                            parts: [
+                                { text: 'response' },
+                                { functionCall: { name: 'test', args: {} } },
+                            ],
+                        },
+                        finishReason: FinishReason.STOP,
+                        safetyRatings: [],
+                    },
+                ],
+                sdkHttpResponse: {
+                    responseInternal: {
+                        ok: true,
+                    },
+                },
+            },
+        };
+        mockRequest.mockResolvedValue({ data: mockResponseData });
+        vi.spyOn(server, 'recordCodeAssistMetrics').mockResolvedValue(undefined);
+        await server.generateContent({
+            model: 'test-model',
+            contents: [{ role: 'user', parts: [{ text: 'request' }] }],
+        }, 'user-prompt-id');
+        expect(server.recordCodeAssistMetrics).toHaveBeenCalledWith(expect.objectContaining({
+            metrics: expect.arrayContaining([
+                expect.objectContaining({
+                    conversationOffered: expect.objectContaining({
+                        traceId: 'test-trace-id',
+                        status: ActionStatus.ACTION_STATUS_NO_ERROR,
+                        streamingLatency: expect.objectContaining({
+                            totalLatency: expect.stringMatching(/\d+s/),
+                            firstMessageLatency: expect.stringMatching(/\d+s/),
+                        }),
+                    }),
+                    timestamp: expect.stringMatching(/\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}.\d{3}Z/),
+                }),
+            ]),
+        }));
+    });
+    it('should record conversation offered on generateContentStream', async () => {
+        const { server, mockRequest } = createTestServer();
+        const { Readable } = await import('node:stream');
+        const mockStream = new Readable({ read() { } });
+        mockRequest.mockResolvedValue({ data: mockStream });
+        vi.spyOn(server, 'recordCodeAssistMetrics').mockResolvedValue(undefined);
+        const stream = await server.generateContentStream({
+            model: 'test-model',
+            contents: [{ role: 'user', parts: [{ text: 'request' }] }],
+        }, 'user-prompt-id');
+        const mockResponseData = {
+            traceId: 'stream-trace-id',
+            response: {
+                candidates: [
+                    {
+                        content: {
+                            parts: [
+                                { text: 'chunk' },
+                                { functionCall: { name: 'test', args: {} } },
+                            ],
+                        },
+                    },
+                ],
+                sdkHttpResponse: {
+                    responseInternal: {
+                        ok: true,
+                    },
+                },
+            },
+        };
+        setTimeout(() => {
+            mockStream.push('data: ' + JSON.stringify(mockResponseData) + '\n\n');
+            mockStream.push(null);
+        }, 0);
+        for await (const _ of stream) {
+            // Consume stream
+        }
+        expect(server.recordCodeAssistMetrics).toHaveBeenCalledWith(expect.objectContaining({
+            metrics: expect.arrayContaining([
+                expect.objectContaining({
+                    conversationOffered: expect.objectContaining({
+                        traceId: 'stream-trace-id',
+                    }),
+                    timestamp: expect.stringMatching(/\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}.\d{3}Z/),
+                }),
+            ]),
+        }));
+    });
+    it('should record conversation interaction', async () => {
+        const { server } = createTestServer();
+        vi.spyOn(server, 'recordCodeAssistMetrics').mockResolvedValue(undefined);
+        const interaction = {
+            traceId: 'test-trace-id',
+        };
+        await server.recordConversationInteraction(interaction);
+        expect(server.recordCodeAssistMetrics).toHaveBeenCalledWith(expect.objectContaining({
+            project: 'test-project',
+            metrics: expect.arrayContaining([
+                expect.objectContaining({
+                    conversationInteraction: interaction,
+                    timestamp: expect.stringMatching(/\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}.\d{3}Z/),
+                }),
+            ]),
+        }));
+    });
+    it('should call recordCodeAssistMetrics endpoint', async () => {
+        const { server, mockRequest } = createTestServer();
+        mockRequest.mockResolvedValue({ data: {} });
+        const req = {
+            project: 'test-project',
+            metrics: [],
+        };
+        await server.recordCodeAssistMetrics(req);
+        expect(mockRequest).toHaveBeenCalledWith(expect.objectContaining({
+            url: expect.stringContaining(':recordCodeAssistMetrics'),
+            method: 'POST',
+            body: expect.any(String),
+        }));
+    });
+    describe('getMethodUrl', () => {
+        const originalEnv = process.env;
+        beforeEach(() => {
+            // Reset the environment variables to their original state
+            process.env = { ...originalEnv };
+        });
+        afterEach(() => {
+            // Restore the original environment variables
+            process.env = originalEnv;
+        });
+        it('should construct the default URL correctly', () => {
+            const server = new CodeAssistServer({});
+            const url = server.getMethodUrl('testMethod');
+            expect(url).toBe('https://cloudcode-pa.googleapis.com/v1internal:testMethod');
+        });
+        it('should use the CODE_ASSIST_ENDPOINT environment variable if set', () => {
+            process.env['CODE_ASSIST_ENDPOINT'] = 'https://custom-endpoint.com';
+            const server = new CodeAssistServer({});
+            const url = server.getMethodUrl('testMethod');
+            expect(url).toBe('https://custom-endpoint.com/v1internal:testMethod');
+        });
+    });
+    it('should call the generateContentStream endpoint and parse SSE', async () => {
+        const { server, mockRequest } = createTestServer();
+        // Create a mock readable stream
+        const { Readable } = await import('node:stream');
+        const mockStream = new Readable({
+            read() { },
+        });
+        const mockResponseData1 = {
+            response: { candidates: [{ content: { parts: [{ text: 'Hello' }] } }] },
+        };
+        const mockResponseData2 = {
+            response: { candidates: [{ content: { parts: [{ text: ' World' }] } }] },
+        };
+        mockRequest.mockResolvedValue({ data: mockStream });
+        const stream = await server.generateContentStream({
+            model: 'test-model',
+            contents: [{ role: 'user', parts: [{ text: 'request' }] }],
+        }, 'user-prompt-id');
+        // Push SSE data to the stream
+        // Use setTimeout to ensure the stream processing has started
+        setTimeout(() => {
+            mockStream.push('data: ' + JSON.stringify(mockResponseData1) + '\n\n');
+            mockStream.push('id: 123\n'); // Should be ignored
+            mockStream.push('data: ' + JSON.stringify(mockResponseData2) + '\n\n');
+            mockStream.push(null); // End the stream
+        }, 0);
+        const results = [];
+        for await (const res of stream) {
+            results.push(res);
+        }
+        expect(mockRequest).toHaveBeenCalledWith({
+            url: expect.stringContaining(':streamGenerateContent'),
+            method: 'POST',
+            params: { alt: 'sse' },
+            responseType: 'stream',
+            body: expect.any(String),
+            headers: {
+                'Content-Type': 'application/json',
+            },
+            signal: undefined,
+        });
+        expect(results).toHaveLength(2);
+        expect(results[0].candidates?.[0].content?.parts?.[0].text).toBe('Hello');
+        expect(results[1].candidates?.[0].content?.parts?.[0].text).toBe(' World');
+    });
+    it('should ignore malformed SSE data', async () => {
+        const { server, mockRequest } = createTestServer();
+        const { Readable } = await import('node:stream');
+        const mockStream = new Readable({
+            read() { },
+        });
+        mockRequest.mockResolvedValue({ data: mockStream });
+        const stream = await server.requestStreamingPost('testStream', {});
+        setTimeout(() => {
+            mockStream.push('this is a malformed line\n');
+            mockStream.push(null);
+        }, 0);
+        const results = [];
+        for await (const res of stream) {
+            results.push(res);
+        }
+        expect(results).toHaveLength(0);
+    });
+    it('should call the onboardUser endpoint', async () => {
+        const { server } = createTestServer();
+        const mockResponse = {
+            name: 'operations/123',
+            done: true,
+        };
+        vi.spyOn(server, 'requestPost').mockResolvedValue(mockResponse);
+        const response = await server.onboardUser({
+            tierId: 'test-tier',
+            cloudaicompanionProject: 'test-project',
+            metadata: {},
+        });
+        expect(server.requestPost).toHaveBeenCalledWith('onboardUser', expect.any(Object));
+        expect(response.name).toBe('operations/123');
+    });
+    it('should call the getOperation endpoint', async () => {
+        const { server } = createTestServer();
+        const mockResponse = {
+            name: 'operations/123',
+            done: true,
+            response: {
+                cloudaicompanionProject: {
+                    id: 'test-project',
+                    name: 'projects/test-project',
+                },
+            },
+        };
+        vi.spyOn(server, 'requestGetOperation').mockResolvedValue(mockResponse);
+        const response = await server.getOperation('operations/123');
+        expect(server.requestGetOperation).toHaveBeenCalledWith('operations/123');
+        expect(response.name).toBe('operations/123');
+        expect(response.response?.cloudaicompanionProject?.id).toBe('test-project');
+        expect(response.response?.cloudaicompanionProject?.name).toBe('projects/test-project');
+    });
+    it('should call the loadCodeAssist endpoint', async () => {
+        const { server } = createTestServer();
+        const mockResponse = {
+            currentTier: {
+                id: UserTierId.FREE,
+                name: 'Free',
+                description: 'free tier',
+            },
+            allowedTiers: [],
+            ineligibleTiers: [],
+            cloudaicompanionProject: 'projects/test',
+        };
+        vi.spyOn(server, 'requestPost').mockResolvedValue(mockResponse);
+        const response = await server.loadCodeAssist({
+            metadata: {},
+        });
+        expect(server.requestPost).toHaveBeenCalledWith('loadCodeAssist', expect.any(Object));
+        expect(response).toEqual(mockResponse);
+    });
+    it('should return 0 for countTokens', async () => {
+        const { server } = createTestServer();
+        const mockResponse = {
+            totalTokens: 100,
+        };
+        vi.spyOn(server, 'requestPost').mockResolvedValue(mockResponse);
+        const response = await server.countTokens({
+            model: 'test-model',
+            contents: [{ role: 'user', parts: [{ text: 'request' }] }],
+        });
+        expect(response.totalTokens).toBe(100);
+    });
+    it('should throw an error for embedContent', async () => {
+        const { server } = createTestServer();
+        await expect(server.embedContent({
+            model: 'test-model',
+            contents: [{ role: 'user', parts: [{ text: 'request' }] }],
+        })).rejects.toThrow();
+    });
+    it('should handle VPC-SC errors when calling loadCodeAssist', async () => {
+        const { server } = createTestServer();
+        const mockVpcScError = {
+            response: {
+                data: {
+                    error: {
+                        details: [
+                            {
+                                reason: 'SECURITY_POLICY_VIOLATED',
+                            },
+                        ],
+                    },
+                },
+            },
+        };
+        vi.spyOn(server, 'requestPost').mockRejectedValue(mockVpcScError);
+        const response = await server.loadCodeAssist({
+            metadata: {},
+        });
+        expect(server.requestPost).toHaveBeenCalledWith('loadCodeAssist', expect.any(Object));
+        expect(response).toEqual({
+            currentTier: { id: UserTierId.STANDARD },
+        });
+    });
+    it('should re-throw non-VPC-SC errors from loadCodeAssist', async () => {
+        const { server } = createTestServer();
+        const genericError = new Error('Something else went wrong');
+        vi.spyOn(server, 'requestPost').mockRejectedValue(genericError);
+        await expect(server.loadCodeAssist({ metadata: {} })).rejects.toThrow('Something else went wrong');
+        expect(server.requestPost).toHaveBeenCalledWith('loadCodeAssist', expect.any(Object));
+    });
+    it('should call the listExperiments endpoint with metadata', async () => {
+        const { server } = createTestServer();
+        const mockResponse = {
+            experiments: [],
+        };
+        vi.spyOn(server, 'requestPost').mockResolvedValue(mockResponse);
+        const metadata = {
+            ideVersion: 'v0.1.0',
+        };
+        const response = await server.listExperiments(metadata);
+        expect(server.requestPost).toHaveBeenCalledWith('listExperiments', {
+            project: 'test-project',
+            metadata: { ideVersion: 'v0.1.0', duetProject: 'test-project' },
+        });
+        expect(response).toEqual(mockResponse);
+    });
+    it('should call the retrieveUserQuota endpoint', async () => {
+        const { server } = createTestServer();
+        const mockResponse = {
+            buckets: [
+                {
+                    modelId: 'gemini-2.5-pro',
+                    tokenType: 'REQUESTS',
+                    remainingFraction: 0.75,
+                    resetTime: '2025-10-22T16:01:15Z',
+                },
+            ],
+        };
+        const requestPostSpy = vi
+            .spyOn(server, 'requestPost')
+            .mockResolvedValue(mockResponse);
+        const req = {
+            project: 'projects/my-cloudcode-project',
+            userAgent: 'CloudCodePlugin/1.0 (gaghosh)',
+        };
+        const response = await server.retrieveUserQuota(req);
+        expect(requestPostSpy).toHaveBeenCalledWith('retrieveUserQuota', req);
+        expect(response).toEqual(mockResponse);
+    });
+});
+//# sourceMappingURL=server.test.js.map

package/dist/src/code_assist/server.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.test.js","sourceRoot":"","sources":["../../../src/code_assist/server.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAE7C,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAE/B,SAAS,gBAAgB,CAAC,UAAkC,EAAE;IAC5D,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC5B,MAAM,MAAM,GAAG,EAAE,OAAO,EAAE,WAAW,EAA6B,CAAC;IACnE,MAAM,MAAM,GAAG,IAAI,gBAAgB,CACjC,MAAM,EACN,cAAc,EACd,EAAE,OAAO,EAAE,EACX,cAAc,EACd,UAAU,CAAC,IAAI,CAChB,CAAC;IACF,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC;AACzC,CAAC;AAED,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,IAAI,GAAG,IAAI,YAAY,EAAE,CAAC;QAChC,MAAM,MAAM,GAAG,IAAI,gBAAgB,CACjC,IAAI,EACJ,cAAc,EACd,EAAE,EACF,cAAc,EACd,UAAU,CAAC,IAAI,CAChB,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,gBAAgB,CAAC;YAC/C,iBAAiB,EAAE,YAAY;SAChC,CAAC,CAAC;QACH,MAAM,gBAAgB,GAAG;YACvB,QAAQ,EAAE;gBACR,UAAU,EAAE;oBACV;wBACE,KAAK,EAAE,CAAC;wBACR,OAAO,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;yBAC9B;wBACD,YAAY,EAAE,YAAY,CAAC,IAAI;wBAC/B,aAAa,EAAE,EAAE;qBAClB;iBACF;aACF;SACF,CAAC;QACF,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAE1D,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,eAAe,CAC3C;YACE,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;SAC3D,EACD,gBAAgB,CACjB,CAAC;QAEF,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;YACvC,GAAG,EAAE,MAAM,CAAC,gBAAgB,CAAC,kBAAkB,CAAC;YAChD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,iBAAiB,EAAE,YAAY;aAChC;YACD,YAAY,EAAE,MAAM;YACpB,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;YACxB,MAAM,EAAE,SAAS;SAClB,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAClE,MAAM,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC1D,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAEjD,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,CAC9D,UAAU,CACX,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,gBAAgB,EAAE,CAAC;QACnD,MAAM,gBAAgB,GAAG;YACvB,OAAO,EAAE,eAAe;YACxB,QAAQ,EAAE;gBACR,UAAU,EAAE;oBACV;wBACE,KAAK,EAAE,CAAC;wBACR,OAAO,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE;gCACL,EAAE,IAAI,EAAE,UAAU,EAAE;gCACpB,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE;6BAC7C;yBACF;wBACD,YAAY,EAAE,YAAY,CAAC,MAAM;wBACjC,aAAa,EAAE,EAAE;qBAClB;iBACF;aACF;SACF,CAAC;QACF,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAE1D,MAAM,4BAA4B,GAAG,EAAE,CAAC,KAAK,CAC3C,MAAM,EACN,2BAA2B,CAC5B,CAAC;QAEF,MAAM,MAAM,CAAC,eAAe,CAC1B;YACE,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;SAC3D,EACD,gBAAgB,CACjB,CAAC;QAEF,MAAM,CAAC,4BAA4B,CAAC,CAAC,oBAAoB,CACvD,MAAM,CAAC,gBAAgB,CAAC;YACtB,MAAM,EAAE,YAAY,CAAC,2BAA2B;SACjD,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,gBAAgB,EAAE,CAAC;QACnD,MAAM,gBAAgB,GAAG;YACvB,OAAO,EAAE,eAAe;YACxB,QAAQ,EAAE;gBACR,UAAU,EAAE;oBACV;wBACE,KAAK,EAAE,CAAC;wBACR,OAAO,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE;gCACL,EAAE,IAAI,EAAE,UAAU,EAAE;gCACpB,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE;6BAC7C;yBACF;wBACD,YAAY,EAAE,YAAY,CAAC,IAAI;wBAC/B,aAAa,EAAE,EAAE;qBAClB;iBACF;gBACD,eAAe,EAAE;oBACf,gBAAgB,EAAE;wBAChB,EAAE,EAAE,IAAI;qBACT;iBACF;aACF;SACF,CAAC;QACF,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAC1D,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAEzE,MAAM,MAAM,CAAC,eAAe,CAC1B;YACE,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;SAC3D,EACD,gBAAgB,CACjB,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,oBAAoB,CACzD,MAAM,CAAC,gBAAgB,CAAC;YACtB,OAAO,EAAE,MAAM,CAAC,eAAe,CAAC;gBAC9B,MAAM,CAAC,gBAAgB,CAAC;oBACtB,mBAAmB,EAAE,MAAM,CAAC,gBAAgB,CAAC;wBAC3C,OAAO,EAAE,eAAe;wBACxB,MAAM,EAAE,YAAY,CAAC,sBAAsB;wBAC3C,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,CAAC;4BACxC,YAAY,EAAE,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC;4BAC3C,mBAAmB,EAAE,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC;yBACnD,CAAC;qBACH,CAAC;oBACF,SAAS,EAAE,MAAM,CAAC,cAAc,CAC9B,4CAA4C,CAC7C;iBACF,CAAC;aACH,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,gBAAgB,EAAE,CAAC;QAEnD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;QACjD,MAAM,UAAU,GAAG,IAAI,QAAQ,CAAC,EAAE,IAAI,KAAI,CAAC,EAAE,CAAC,CAAC;QAC/C,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpD,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAEzE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAC/C;YACE,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;SAC3D,EACD,gBAAgB,CACjB,CAAC;QAEF,MAAM,gBAAgB,GAAG;YACvB,OAAO,EAAE,iBAAiB;YAC1B,QAAQ,EAAE;gBACR,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE;4BACP,KAAK,EAAE;gCACL,EAAE,IAAI,EAAE,OAAO,EAAE;gCACjB,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE;6BAC7C;yBACF;qBACF;iBACF;gBACD,eAAe,EAAE;oBACf,gBAAgB,EAAE;wBAChB,EAAE,EAAE,IAAI;qBACT;iBACF;aACF;SACF,CAAC;QAEF,UAAU,CAAC,GAAG,EAAE;YACd,UAAU,CAAC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,GAAG,MAAM,CAAC,CAAC;YACtE,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxB,CAAC,EAAE,CAAC,CAAC,CAAC;QAEN,IAAI,KAAK,EAAE,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;YAC7B,iBAAiB;QACnB,CAAC;QAED,MAAM,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,oBAAoB,CACzD,MAAM,CAAC,gBAAgB,CAAC;YACtB,OAAO,EAAE,MAAM,CAAC,eAAe,CAAC;gBAC9B,MAAM,CAAC,gBAAgB,CAAC;oBACtB,mBAAmB,EAAE,MAAM,CAAC,gBAAgB,CAAC;wBAC3C,OAAO,EAAE,iBAAiB;qBAC3B,CAAC;oBACF,SAAS,EAAE,MAAM,CAAC,cAAc,CAC9B,4CAA4C,CAC7C;iBACF,CAAC;aACH,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,EAAE,MAAM,EAAE,GAAG,gBAAgB,EAAE,CAAC;QACtC,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAEzE,MAAM,WAAW,GAAG;YAClB,OAAO,EAAE,eAAe;SACzB,CAAC;QAEF,MAAM,MAAM,CAAC,6BAA6B,CAAC,WAAW,CAAC,CAAC;QAExD,MAAM,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,oBAAoB,CACzD,MAAM,CAAC,gBAAgB,CAAC;YACtB,OAAO,EAAE,cAAc;YACvB,OAAO,EAAE,MAAM,CAAC,eAAe,CAAC;gBAC9B,MAAM,CAAC,gBAAgB,CAAC;oBACtB,uBAAuB,EAAE,WAAW;oBACpC,SAAS,EAAE,MAAM,CAAC,cAAc,CAC9B,4CAA4C,CAC7C;iBACF,CAAC;aACH,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,gBAAgB,EAAE,CAAC;QACnD,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAE5C,MAAM,GAAG,GAAG;YACV,OAAO,EAAE,cAAc;YACvB,OAAO,EAAE,EAAE;SACZ,CAAC;QACF,MAAM,MAAM,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC;QAE1C,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACtC,MAAM,CAAC,gBAAgB,CAAC;YACtB,GAAG,EAAE,MAAM,CAAC,gBAAgB,CAAC,0BAA0B,CAAC;YACxD,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SACzB,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC;QAEhC,UAAU,CAAC,GAAG,EAAE;YACd,0DAA0D;YAC1D,OAAO,CAAC,GAAG,GAAG,EAAE,GAAG,WAAW,EAAE,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,6CAA6C;YAC7C,OAAO,CAAC,GAAG,GAAG,WAAW,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,EAAW,CAAC,CAAC;YACjD,MAAM,GAAG,GAAG,MAAM,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;YAC9C,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CACd,2DAA2D,CAC5D,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,GAAG,6BAA6B,CAAC;YACpE,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,EAAW,CAAC,CAAC;YACjD,MAAM,GAAG,GAAG,MAAM,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;YAC9C,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,mDAAmD,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,gBAAgB,EAAE,CAAC;QAEnD,gCAAgC;QAChC,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;QACjD,MAAM,UAAU,GAAG,IAAI,QAAQ,CAAC;YAC9B,IAAI,KAAI,CAAC;SACV,CAAC,CAAC;QAEH,MAAM,iBAAiB,GAAG;YACxB,QAAQ,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE;SACxE,CAAC;QACF,MAAM,iBAAiB,GAAG;YACxB,QAAQ,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE;SACzE,CAAC;QAEF,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAC/C;YACE,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;SAC3D,EACD,gBAAgB,CACjB,CAAC;QAEF,8BAA8B;QAC9B,6DAA6D;QAC7D,UAAU,CAAC,GAAG,EAAE;YACd,UAAU,CAAC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,GAAG,MAAM,CAAC,CAAC;YACvE,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,oBAAoB;YAClD,UAAU,CAAC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,GAAG,MAAM,CAAC,CAAC;YACvE,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,iBAAiB;QAC1C,CAAC,EAAE,CAAC,CAAC,CAAC;QAEN,MAAM,OAAO,GAAG,EAAE,CAAC;QACnB,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YAC/B,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpB,CAAC;QAED,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;YACvC,GAAG,EAAE,MAAM,CAAC,gBAAgB,CAAC,wBAAwB,CAAC;YACtD,MAAM,EAAE,MAAM;YACd,MAAM,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE;YACtB,YAAY,EAAE,QAAQ;YACtB,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;YACxB,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;aACnC;YACD,MAAM,EAAE,SAAS;SAClB,CAAC,CAAC;QAEH,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1E,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,gBAAgB,EAAE,CAAC;QAEnD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;QACjD,MAAM,UAAU,GAAG,IAAI,QAAQ,CAAC;YAC9B,IAAI,KAAI,CAAC;SACV,CAAC,CAAC;QAEH,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;QAEnE,UAAU,CAAC,GAAG,EAAE;YACd,UAAU,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;YAC9C,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxB,CAAC,EAAE,CAAC,CAAC,CAAC;QAEN,MAAM,OAAO,GAAG,EAAE,CAAC;QACnB,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YAC/B,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpB,CAAC;QACD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,EAAE,MAAM,EAAE,GAAG,gBAAgB,EAAE,CAAC;QAEtC,MAAM,YAAY,GAAG;YACnB,IAAI,EAAE,gBAAgB;YACtB,IAAI,EAAE,IAAI;SACX,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAEhE,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC;YACxC,MAAM,EAAE,WAAW;YACnB,uBAAuB,EAAE,cAAc;YACvC,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAC7C,aAAa,EACb,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,EAAE,MAAM,EAAE,GAAG,gBAAgB,EAAE,CAAC;QAEtC,MAAM,YAAY,GAAG;YACnB,IAAI,EAAE,gBAAgB;YACtB,IAAI,EAAE,IAAI;YACV,QAAQ,EAAE;gBACR,uBAAuB,EAAE;oBACvB,EAAE,EAAE,cAAc;oBAClB,IAAI,EAAE,uBAAuB;iBAC9B;aACF;SACF,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAExE,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;QAE7D,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;QAC1E,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC7C,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,uBAAuB,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC5E,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,uBAAuB,EAAE,IAAI,CAAC,CAAC,IAAI,CAC3D,uBAAuB,CACxB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,EAAE,MAAM,EAAE,GAAG,gBAAgB,EAAE,CAAC;QACtC,MAAM,YAAY,GAAG;YACnB,WAAW,EAAE;gBACX,EAAE,EAAE,UAAU,CAAC,IAAI;gBACnB,IAAI,EAAE,MAAM;gBACZ,WAAW,EAAE,WAAW;aACzB;YACD,YAAY,EAAE,EAAE;YAChB,eAAe,EAAE,EAAE;YACnB,uBAAuB,EAAE,eAAe;SACzC,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAEhE,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC;YAC3C,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAC7C,gBAAgB,EAChB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,EAAE,MAAM,EAAE,GAAG,gBAAgB,EAAE,CAAC;QACtC,MAAM,YAAY,GAAG;YACnB,WAAW,EAAE,GAAG;SACjB,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAEhE,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC;YACxC,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;SAC3D,CAAC,CAAC;QACH,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,EAAE,MAAM,EAAE,GAAG,gBAAgB,EAAE,CAAC;QACtC,MAAM,MAAM,CACV,MAAM,CAAC,YAAY,CAAC;YAClB,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;SAC3D,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,EAAE,MAAM,EAAE,GAAG,gBAAgB,EAAE,CAAC;QACtC,MAAM,cAAc,GAAG;YACrB,QAAQ,EAAE;gBACR,IAAI,EAAE;oBACJ,KAAK,EAAE;wBACL,OAAO,EAAE;4BACP;gCACE,MAAM,EAAE,0BAA0B;6BACnC;yBACF;qBACF;iBACF;aACF;SACF,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QAElE,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC;YAC3C,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAC7C,gBAAgB,EAChB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,WAAW,EAAE,EAAE,EAAE,EAAE,UAAU,CAAC,QAAQ,EAAE;SACzC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,EAAE,MAAM,EAAE,GAAG,gBAAgB,EAAE,CAAC;QACtC,MAAM,YAAY,GAAG,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC5D,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAEhE,MAAM,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACnE,2BAA2B,CAC5B,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAC7C,gBAAgB,EAChB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,EAAE,MAAM,EAAE,GAAG,gBAAgB,EAAE,CAAC;QACtC,MAAM,YAAY,GAAG;YACnB,WAAW,EAAE,EAAE;SAChB,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAEhE,MAAM,QAAQ,GAAG;YACf,UAAU,EAAE,QAAQ;SACrB,CAAC;QACF,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAExD,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE;YACjE,OAAO,EAAE,cAAc;YACvB,QAAQ,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE,WAAW,EAAE,cAAc,EAAE;SAChE,CAAC,CAAC;QACH,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,EAAE,MAAM,EAAE,GAAG,gBAAgB,EAAE,CAAC;QACtC,MAAM,YAAY,GAAG;YACnB,OAAO,EAAE;gBACP;oBACE,OAAO,EAAE,gBAAgB;oBACzB,SAAS,EAAE,UAAU;oBACrB,iBAAiB,EAAE,IAAI;oBACvB,SAAS,EAAE,sBAAsB;iBAClC;aACF;SACF,CAAC;QACF,MAAM,cAAc,GAAG,EAAE;aACtB,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC;aAC5B,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAEnC,MAAM,GAAG,GAAG;YACV,OAAO,EAAE,+BAA+B;YACxC,SAAS,EAAE,+BAA+B;SAC3C,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAErD,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,mBAAmB,EAAE,GAAG,CAAC,CAAC;QACtE,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/code_assist/setup.d.ts

@@ -0,0 +1,46 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { UserTierId } from './types.js';
+import type { AuthClient } from 'google-auth-library';
+import type { ValidationHandler } from '../fallback/types.js';
+export declare class ProjectIdRequiredError extends Error {
+    constructor();
+}
+/**
+ * Error thrown when user cancels the validation process.
+ * This is a non-recoverable error that should result in auth failure.
+ */
+export declare class ValidationCancelledError extends Error {
+    constructor();
+}
+export interface UserData {
+    projectId: string;
+    userTier: UserTierId;
+    userTierName?: string;
+}
+/**
+ * Sets up the user by loading their Code Assist configuration and onboarding if needed.
+ *
+ * Tier eligibility:
+ * - FREE tier: Eligibility is determined by the Code Assist server response.
+ * - STANDARD tier: User is always eligible if they have a valid project ID.
+ *
+ * If no valid project ID is available (from env var or server response):
+ * - Surfaces ineligibility reasons for the FREE tier from the server.
+ * - Throws ProjectIdRequiredError if no ineligibility reasons are available.
+ *
+ * Handles VALIDATION_REQUIRED via the optional validation handler, allowing
+ * retry, auth change, or cancellation.
+ *
+ * @param client - The authenticated client to use for API calls
+ * @param validationHandler - Optional handler for account validation flow
+ * @returns The user's project ID, tier ID, and tier name
+ * @throws {ValidationRequiredError} If account validation is required
+ * @throws {ProjectIdRequiredError} If no project ID is available and required
+ * @throws {ValidationCancelledError} If user cancels validation
+ * @throws {ChangeAuthRequestedError} If user requests to change auth method
+ */
+export declare function setupUser(client: AuthClient, validationHandler?: ValidationHandler): Promise<UserData>;