genaicode 0.6.1 → 0.7.0

package/dist/main/codegen.ai-services.test.js

@@ -0,0 +1,198 @@
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { runCodegen } from './codegen.js';
+import * as vertexAi from '../ai-service/vertex-ai.js';
+import * as chatGpt from '../ai-service/chat-gpt.js';
+import * as anthropic from '../ai-service/anthropic.js';
+import * as vertexAiClaude from '../ai-service/vertex-ai-claude.js';
+import * as updateFiles from '../files/update-files.js';
+import * as cliParams from '../cli/cli-params.js';
+import '../files/find-files.js';
+import '../files/cache-file.js';
+import './config.js';
+import { createMockPlanningResponse, createMockCodegenSummary, createMockFileUpdate, createMockResponseSequence, } from './codegen.test-utils.js';
+// Mock all required modules
+vi.mock('../cli/cli-params.js', () => ({
+    interactive: false,
+    ui: false,
+    uiPort: 1337,
+    disableExplanations: true,
+    explicitPrompt: false,
+    allowFileCreate: false,
+    allowFileDelete: false,
+    allowDirectoryCreate: false,
+    allowFileMove: false,
+    verbosePrompt: false,
+    vertexAiClaude: false,
+    helpRequested: false,
+    aiStudio: false,
+    vision: false,
+    imagen: false,
+    temperature: 0.7,
+    cheap: false,
+    taskFile: undefined,
+    disableInitialLint: undefined,
+    askQuestion: false,
+    disableContextOptimization: true,
+    geminiBlockNone: undefined,
+    contentMask: undefined,
+    ignorePatterns: [],
+    disableCache: undefined,
+    disableAiServiceFallback: undefined,
+    disableHistory: true,
+    disableConversationSummary: true,
+    aiService: undefined,
+    dryRun: false,
+}));
+vi.mock('../ai-service/vertex-ai.js', () => ({ generateContent: vi.fn() }));
+vi.mock('../ai-service/chat-gpt.js', () => ({ generateContent: vi.fn() }));
+vi.mock('../ai-service/anthropic.js', () => ({ generateContent: vi.fn() }));
+vi.mock('../ai-service/vertex-ai-claude.js', () => ({ generateContent: vi.fn() }));
+vi.mock('../files/update-files.js');
+vi.mock('../files/cache-file.js');
+vi.mock('../files/find-files.js', () => ({
+    getSourceFiles: () => [],
+    getImageAssetFiles: () => [],
+    refreshFiles: () => null,
+}));
+vi.mock('./config.js', () => ({
+    rootDir: '/mocked/root/dir',
+    rcConfig: {
+        rootDir: '/mocked/root/dir',
+        extensions: ['.js', '.ts', '.tsx', '.jsx'],
+    },
+    importantContext: {},
+}));
+describe('AI Services Integration', () => {
+    beforeEach(() => {
+        vi.resetAllMocks();
+        vi.mocked(cliParams).aiService = undefined;
+        vi.mocked(cliParams).dryRun = false;
+        vi.mocked(cliParams).helpRequested = false;
+        vi.mocked(cliParams).vision = false;
+        vi.mocked(cliParams).imagen = undefined;
+        vi.mocked(cliParams).cheap = false;
+        vi.mocked(cliParams).temperature = 0.7;
+        vi.mocked(cliParams).explicitPrompt = 'test';
+    });
+    describe('Vertex AI', () => {
+        it('should run codegen with Vertex AI by default', async () => {
+            vi.mocked(cliParams).aiService = 'vertex-ai';
+            const mockPlanning = createMockPlanningResponse('Test analysis for Vertex AI', 'Update test.js file', [
+                { filePath: '/mocked/root/dir/test.js', reason: 'Test update' },
+            ]);
+            const mockSummary = createMockCodegenSummary([{ filePath: '/mocked/root/dir/test.js', updateToolName: 'updateFile', prompt: 'Update file content' }], [], 'Test file update');
+            const mockUpdate = createMockFileUpdate('updateFile', {
+                filePath: '/mocked/root/dir/test.js',
+                newContent: 'console.log("Hello");',
+            });
+            const mockSequence = createMockResponseSequence(mockPlanning, mockSummary, [mockUpdate]);
+            mockSequence.forEach((response) => {
+                vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(response);
+            });
+            await runCodegen();
+            // Verify all phases were called
+            expect(vertexAi.generateContent).toHaveBeenCalledTimes(3);
+            expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockUpdate, expect.anything());
+        });
+    });
+    describe('ChatGPT', () => {
+        it('should run codegen with ChatGPT when specified', async () => {
+            vi.mocked(cliParams).aiService = 'chat-gpt';
+            const mockPlanning = createMockPlanningResponse('Test analysis for ChatGPT', 'Create new.js file', [
+                { filePath: '/mocked/root/dir/new.js', reason: 'New file creation' },
+            ]);
+            const mockSummary = createMockCodegenSummary([{ filePath: '/mocked/root/dir/new.js', updateToolName: 'createFile', prompt: 'Create new file' }], [], 'Create new test file');
+            const mockCreate = createMockFileUpdate('createFile', {
+                filePath: '/mocked/root/dir/new.js',
+                newContent: 'const x = 5;',
+            });
+            const mockSequence = createMockResponseSequence(mockPlanning, mockSummary, [mockCreate]);
+            mockSequence.forEach((response) => {
+                vi.mocked(chatGpt.generateContent).mockResolvedValueOnce(response);
+            });
+            await runCodegen();
+            expect(chatGpt.generateContent).toHaveBeenCalledTimes(3);
+            expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockCreate, expect.anything());
+        });
+    });
+    describe('Anthropic', () => {
+        it('should run codegen with Anthropic when specified', async () => {
+            vi.mocked(cliParams).aiService = 'anthropic';
+            const mockPlanning = createMockPlanningResponse('Test analysis for Anthropic', 'Delete obsolete.js file', [
+                { filePath: '/mocked/root/dir/obsolete.js', reason: 'File removal' },
+            ]);
+            const mockSummary = createMockCodegenSummary([{ filePath: '/mocked/root/dir/obsolete.js', updateToolName: 'deleteFile', prompt: 'Remove obsolete file' }], [], 'Remove test file');
+            const mockDelete = createMockFileUpdate('deleteFile', {
+                filePath: '/mocked/root/dir/obsolete.js',
+            });
+            const mockSequence = createMockResponseSequence(mockPlanning, mockSummary, [mockDelete]);
+            mockSequence.forEach((response) => {
+                vi.mocked(anthropic.generateContent).mockResolvedValueOnce(response);
+            });
+            await runCodegen();
+            expect(anthropic.generateContent).toHaveBeenCalledTimes(3);
+            expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockDelete, expect.anything());
+        });
+    });
+    describe('Vertex AI Claude', () => {
+        it('should run codegen with Vertex AI Claude when specified', async () => {
+            vi.mocked(cliParams).aiService = 'vertex-ai-claude';
+            const mockPlanning = createMockPlanningResponse('Test analysis for Vertex AI Claude', 'Update test.js with Claude', [{ filePath: '/mocked/root/dir/test.js', reason: 'Claude test update' }]);
+            const mockSummary = createMockCodegenSummary([{ filePath: '/mocked/root/dir/test.js', updateToolName: 'updateFile', prompt: 'Update with Claude' }], [], 'Claude test update');
+            const mockUpdate = createMockFileUpdate('updateFile', {
+                filePath: '/mocked/root/dir/test.js',
+                newContent: 'console.log("Hello from Claude");',
+            });
+            const mockSequence = createMockResponseSequence(mockPlanning, mockSummary, [mockUpdate]);
+            mockSequence.forEach((response) => {
+                vi.mocked(vertexAiClaude.generateContent).mockResolvedValueOnce(response);
+            });
+            await runCodegen();
+            expect(vertexAiClaude.generateContent).toHaveBeenCalledTimes(3);
+            expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockUpdate, expect.anything());
+        });
+    });
+    describe('AI Service Parameters', () => {
+        it('should pass the temperature parameter to the AI service', async () => {
+            vi.mocked(cliParams).aiService = 'vertex-ai';
+            vi.mocked(cliParams).temperature = 0.5;
+            const mockPlanning = createMockPlanningResponse();
+            const mockSummary = createMockCodegenSummary();
+            const mockUpdate = createMockFileUpdate('updateFile', {
+                filePath: 'test.js',
+                newContent: 'console.log("Temperature test");',
+            });
+            const mockSequence = createMockResponseSequence(mockPlanning, mockSummary, [mockUpdate]);
+            mockSequence.forEach((response) => {
+                vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(response);
+            });
+            await runCodegen();
+            // Verify temperature parameter in each call
+            const calls = vi.mocked(vertexAi.generateContent).mock.calls;
+            calls.forEach((call) => {
+                expect(call[3]).toBe(0.5); // temperature parameter
+            });
+        });
+        it('should pass the cheap parameter to the AI service when --cheap flag is true', async () => {
+            vi.mocked(cliParams).aiService = 'vertex-ai';
+            vi.mocked(cliParams).cheap = true;
+            const mockPlanning = createMockPlanningResponse();
+            const mockSummary = createMockCodegenSummary();
+            const mockUpdate = createMockFileUpdate('updateFile', {
+                filePath: 'test.js',
+                newContent: 'console.log("Cheap test");',
+            });
+            const mockSequence = createMockResponseSequence(mockPlanning, mockSummary, [mockUpdate]);
+            mockSequence.forEach((response) => {
+                vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(response);
+            });
+            await runCodegen();
+            // Verify cheap parameter in each call
+            const calls = vi.mocked(vertexAi.generateContent).mock.calls;
+            calls.forEach((call) => {
+                expect(call[4]).toBe(true); // cheap parameter
+            });
+        });
+    });
+});
+//# sourceMappingURL=codegen.ai-services.test.js.map

package/dist/main/codegen.ai-services.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codegen.ai-services.test.js","sourceRoot":"","sources":["../../src/main/codegen.ai-services.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,KAAK,QAAQ,MAAM,4BAA4B,CAAC;AACvD,OAAO,KAAK,OAAO,MAAM,2BAA2B,CAAC;AACrD,OAAO,KAAK,SAAS,MAAM,4BAA4B,CAAC;AACxD,OAAO,KAAK,cAAc,MAAM,mCAAmC,CAAC;AACpE,OAAO,KAAK,WAAW,MAAM,0BAA0B,CAAC;AACxD,OAAO,KAAK,SAAS,MAAM,sBAAsB,CAAC;AAClD,OAAO,wBAAwB,CAAC;AAChC,OAAO,wBAAwB,CAAC;AAChC,OAAO,aAAa,CAAC;AAErB,OAAO,EACL,0BAA0B,EAC1B,wBAAwB,EACxB,oBAAoB,EACpB,0BAA0B,GAC3B,MAAM,yBAAyB,CAAC;AAEjC,4BAA4B;AAC5B,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,WAAW,EAAE,KAAK;IAClB,EAAE,EAAE,KAAK;IACT,MAAM,EAAE,IAAI;IACZ,mBAAmB,EAAE,IAAI;IACzB,cAAc,EAAE,KAAK;IACrB,eAAe,EAAE,KAAK;IACtB,eAAe,EAAE,KAAK;IACtB,oBAAoB,EAAE,KAAK;IAC3B,aAAa,EAAE,KAAK;IACpB,aAAa,EAAE,KAAK;IACpB,cAAc,EAAE,KAAK;IACrB,aAAa,EAAE,KAAK;IACpB,QAAQ,EAAE,KAAK;IACf,MAAM,EAAE,KAAK;IACb,MAAM,EAAE,KAAK;IACb,WAAW,EAAE,GAAG;IAChB,KAAK,EAAE,KAAK;IACZ,QAAQ,EAAE,SAAS;IACnB,kBAAkB,EAAE,SAAS;IAC7B,WAAW,EAAE,KAAK;IAClB,0BAA0B,EAAE,IAAI;IAChC,eAAe,EAAE,SAAS;IAC1B,WAAW,EAAE,SAAS;IACtB,cAAc,EAAE,EAAE;IAClB,YAAY,EAAE,SAAS;IACvB,wBAAwB,EAAE,SAAS;IACnC,cAAc,EAAE,IAAI;IACpB,0BAA0B,EAAE,IAAI;IAChC,SAAS,EAAE,SAA+B;IAC1C,MAAM,EAAE,KAAK;CACd,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,4BAA4B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAC5E,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3E,EAAE,CAAC,IAAI,CAAC,4BAA4B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAC5E,EAAE,CAAC,IAAI,CAAC,mCAAmC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACnF,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;AACpC,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;AAClC,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,cAAc,EAAE,GAAG,EAAE,CAAC,EAAE;IACxB,kBAAkB,EAAE,GAAG,EAAE,CAAC,EAAE;IAC5B,YAAY,EAAE,GAAG,EAAE,CAAC,IAAI;CACzB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5B,OAAO,EAAE,kBAAkB;IAC3B,QAAQ,EAAE;QACR,OAAO,EAAE,kBAAkB;QAC3B,UAAU,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;KAC3C;IACD,gBAAgB,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC;QACpC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC;QACpC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,SAAS,CAAC;QACxC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,KAAK,GAAG,KAAK,CAAC;QACnC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,GAAG,GAAG,CAAC;QACvC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,GAAG,MAAM,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;YAE7C,MAAM,YAAY,GAAG,0BAA0B,CAAC,6BAA6B,EAAE,qBAAqB,EAAE;gBACpG,EAAE,QAAQ,EAAE,0BAA0B,EAAE,MAAM,EAAE,aAAa,EAAE;aAChE,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,wBAAwB,CAC1C,CAAC,EAAE,QAAQ,EAAE,0BAA0B,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,EAAE,qBAAqB,EAAE,CAAC,EACvG,EAAE,EACF,kBAAkB,CACnB,CAAC;YAEF,MAAM,UAAU,GAAG,oBAAoB,CAAC,YAAY,EAAE;gBACpD,QAAQ,EAAE,0BAA0B;gBACpC,UAAU,EAAE,uBAAuB;aACpC,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,0BAA0B,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YACzF,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACtE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,EAAE,CAAC;YAEnB,gCAAgC;YAChC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1D,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,UAAU,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,UAAU,CAAC;YAE5C,MAAM,YAAY,GAAG,0BAA0B,CAAC,2BAA2B,EAAE,oBAAoB,EAAE;gBACjG,EAAE,QAAQ,EAAE,yBAAyB,EAAE,MAAM,EAAE,mBAAmB,EAAE;aACrE,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,wBAAwB,CAC1C,CAAC,EAAE,QAAQ,EAAE,yBAAyB,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC,EAClG,EAAE,EACF,sBAAsB,CACvB,CAAC;YAEF,MAAM,UAAU,GAAG,oBAAoB,CAAC,YAAY,EAAE;gBACpD,QAAQ,EAAE,yBAAyB;gBACnC,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,0BAA0B,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YACzF,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChC,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACrE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,EAAE,CAAC;YAEnB,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACzD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,UAAU,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;YAE7C,MAAM,YAAY,GAAG,0BAA0B,CAAC,6BAA6B,EAAE,yBAAyB,EAAE;gBACxG,EAAE,QAAQ,EAAE,8BAA8B,EAAE,MAAM,EAAE,cAAc,EAAE;aACrE,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,wBAAwB,CAC1C,CAAC,EAAE,QAAQ,EAAE,8BAA8B,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,EAAE,sBAAsB,EAAE,CAAC,EAC5G,EAAE,EACF,kBAAkB,CACnB,CAAC;YAEF,MAAM,UAAU,GAAG,oBAAoB,CAAC,YAAY,EAAE;gBACpD,QAAQ,EAAE,8BAA8B;aACzC,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,0BAA0B,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YACzF,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,EAAE,CAAC;YAEnB,MAAM,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,UAAU,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,kBAAkB,CAAC;YAEpD,MAAM,YAAY,GAAG,0BAA0B,CAC7C,oCAAoC,EACpC,4BAA4B,EAC5B,CAAC,EAAE,QAAQ,EAAE,0BAA0B,EAAE,MAAM,EAAE,oBAAoB,EAAE,CAAC,CACzE,CAAC;YAEF,MAAM,WAAW,GAAG,wBAAwB,CAC1C,CAAC,EAAE,QAAQ,EAAE,0BAA0B,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,EAAE,oBAAoB,EAAE,CAAC,EACtG,EAAE,EACF,oBAAoB,CACrB,CAAC;YAEF,MAAM,UAAU,GAAG,oBAAoB,CAAC,YAAY,EAAE;gBACpD,QAAQ,EAAE,0BAA0B;gBACpC,UAAU,EAAE,mCAAmC;aAChD,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,0BAA0B,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YACzF,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChC,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YAC5E,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,EAAE,CAAC;YAEnB,MAAM,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAChE,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,UAAU,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;YAC7C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,GAAG,GAAG,CAAC;YAEvC,MAAM,YAAY,GAAG,0BAA0B,EAAE,CAAC;YAClD,MAAM,WAAW,GAAG,wBAAwB,EAAE,CAAC;YAC/C,MAAM,UAAU,GAAG,oBAAoB,CAAC,YAAY,EAAE;gBACpD,QAAQ,EAAE,SAAS;gBACnB,UAAU,EAAE,kCAAkC;aAC/C,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,0BAA0B,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YACzF,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACtE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,EAAE,CAAC;YAEnB,4CAA4C;YAC5C,MAAM,KAAK,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;YAC7D,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBACrB,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,wBAAwB;YACrD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;YAC3F,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;YAC7C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC;YAElC,MAAM,YAAY,GAAG,0BAA0B,EAAE,CAAC;YAClD,MAAM,WAAW,GAAG,wBAAwB,EAAE,CAAC;YAC/C,MAAM,UAAU,GAAG,oBAAoB,CAAC,YAAY,EAAE;gBACpD,QAAQ,EAAE,SAAS;gBACnB,UAAU,EAAE,4BAA4B;aACzC,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,0BAA0B,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YACzF,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACtE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,EAAE,CAAC;YAEnB,sCAAsC;YACtC,MAAM,KAAK,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;YAC7D,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBACrB,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,kBAAkB;YAChD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/main/codegen.cli-options.test.d.ts

@@ -0,0 +1,3 @@
+import '../files/find-files.js';
+import '../files/cache-file.js';
+import './config.js';

package/dist/main/codegen.cli-options.test.js

@@ -0,0 +1,171 @@
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { runCodegen } from './codegen.js';
+import * as vertexAi from '../ai-service/vertex-ai.js';
+import * as updateFiles from '../files/update-files.js';
+import * as cliParams from '../cli/cli-params.js';
+import * as cliOptions from '../cli/cli-options.js';
+import '../files/find-files.js';
+import '../files/cache-file.js';
+import './config.js';
+import { createMockPlanningResponse, createMockCodegenSummary, createMockFileUpdate, createMockResponseSequence, } from './codegen.test-utils.js';
+import { rcConfig } from './config.js';
+// Mock all required modules
+vi.mock('../cli/cli-params.js', () => ({
+    interactive: false,
+    ui: false,
+    uiPort: 1337,
+    disableExplanations: true,
+    explicitPrompt: false,
+    allowFileCreate: false,
+    allowFileDelete: false,
+    allowDirectoryCreate: false,
+    allowFileMove: false,
+    verbosePrompt: false,
+    vertexAiClaude: false,
+    helpRequested: false,
+    aiStudio: false,
+    vision: false,
+    imagen: false,
+    temperature: 0.7,
+    cheap: false,
+    taskFile: undefined,
+    disableInitialLint: undefined,
+    askQuestion: false,
+    disableContextOptimization: true,
+    geminiBlockNone: undefined,
+    contentMask: undefined,
+    ignorePatterns: [],
+    disableCache: undefined,
+    disableAiServiceFallback: undefined,
+    disableHistory: true,
+    disableConversationSummary: true,
+    aiService: undefined,
+    dryRun: false,
+}));
+vi.mock('../ai-service/vertex-ai.js', () => ({ generateContent: vi.fn() }));
+vi.mock('../files/update-files.js');
+vi.mock('../files/cache-file.js');
+vi.mock('../cli/cli-options.js', () => ({
+    printHelpMessage: vi.fn(),
+}));
+vi.mock('../files/find-files.js', () => ({
+    getSourceFiles: () => [],
+    getImageAssetFiles: () => [],
+    refreshFiles: () => null,
+}));
+vi.mock('./config.js', () => ({
+    rootDir: '/mocked/root/dir',
+    rcConfig: {
+        rootDir: '/mocked/root/dir',
+        extensions: ['.js', '.ts', '.tsx', '.jsx'],
+        lintCommand: 'echo "mock lint"',
+    },
+    importantContext: {},
+}));
+describe('CLI Options', () => {
+    beforeEach(() => {
+        vi.resetAllMocks();
+        // Reset CLI parameters to default values
+        vi.mocked(cliParams).aiService = undefined;
+        vi.mocked(cliParams).disableInitialLint = true;
+        vi.mocked(cliParams).dryRun = false;
+        vi.mocked(cliParams).helpRequested = false;
+        vi.mocked(cliParams).vision = false;
+        vi.mocked(cliParams).imagen = undefined;
+        vi.mocked(cliParams).cheap = false;
+        vi.mocked(cliParams).temperature = 0.7;
+        vi.mocked(cliParams).explicitPrompt = 'test';
+    });
+    describe('Help Option', () => {
+        it('should print help message and not run codegen when --help option is provided', async () => {
+            vi.mocked(cliParams).helpRequested = true;
+            await runCodegen();
+            expect(cliOptions.printHelpMessage).toHaveBeenCalled();
+            expect(vertexAi.generateContent).not.toHaveBeenCalled();
+            expect(updateFiles.updateFiles).not.toHaveBeenCalled();
+        });
+    });
+    describe('Dry Run Mode', () => {
+        it('should not update files in dry run mode but should still plan and analyze', async () => {
+            vi.mocked(cliParams).dryRun = true;
+            vi.mocked(cliParams).aiService = 'vertex-ai';
+            const mockPlanning = createMockPlanningResponse('Test analysis for dry run', 'Update test.js file', [
+                { filePath: '/mocked/root/dir/test.js', reason: 'Test update' },
+            ]);
+            const mockSummary = createMockCodegenSummary([{ filePath: '/mocked/root/dir/test.js', updateToolName: 'updateFile', prompt: 'Update file content' }], [], 'Test file update');
+            const mockUpdate = createMockFileUpdate('updateFile', {
+                filePath: '/mocked/root/dir/test.js',
+                newContent: 'console.log("Dry run");',
+            });
+            const mockSequence = createMockResponseSequence(mockPlanning, mockSummary, [mockUpdate]);
+            mockSequence.forEach((response) => {
+                vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(response);
+            });
+            await runCodegen();
+            expect(vertexAi.generateContent).toHaveBeenCalledTimes(3);
+            expect(updateFiles.updateFiles).not.toHaveBeenCalled();
+        });
+    });
+    describe('Temperature Setting', () => {
+        it('should use custom temperature value when specified', async () => {
+            vi.mocked(cliParams).temperature = 0.3;
+            vi.mocked(cliParams).aiService = 'vertex-ai';
+            const mockPlanning = createMockPlanningResponse('Test analysis with custom temperature', 'Update test.js file', [
+                { filePath: '/mocked/root/dir/test.js', reason: 'Temperature test' },
+            ]);
+            const mockSummary = createMockCodegenSummary([
+                {
+                    filePath: '/mocked/root/dir/test.js',
+                    updateToolName: 'updateFile',
+                    prompt: 'Update with custom temperature',
+                },
+            ], [], 'Temperature test update');
+            const mockUpdate = createMockFileUpdate('updateFile', {
+                filePath: '/mocked/root/dir/test.js',
+                newContent: 'console.log("Custom temperature");',
+            });
+            const mockSequence = createMockResponseSequence(mockPlanning, mockSummary, [mockUpdate]);
+            mockSequence.forEach((response) => {
+                vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(response);
+            });
+            await runCodegen();
+            const calls = vi.mocked(vertexAi.generateContent).mock.calls;
+            calls.forEach((call) => {
+                expect(call[3]).toBe(0.3); // Check temperature parameter
+            });
+        });
+    });
+    describe('Context Optimization', () => {
+        it('should respect disableContextOptimization flag', async () => {
+            vi.mocked(cliParams).disableContextOptimization = false;
+            vi.mocked(cliParams).aiService = 'vertex-ai';
+            vi.mocked(rcConfig).lintCommand = undefined;
+            const mockPlanning = createMockPlanningResponse('Test analysis with context optimization', 'Update test.js file', [{ filePath: '/mocked/root/dir/test.js', reason: 'Context optimization test' }]);
+            const mockSummary = createMockCodegenSummary([{ filePath: '/mocked/root/dir/test.js', updateToolName: 'updateFile', prompt: 'Update with optimization' }], ['/mocked/root/dir/context.js'], // Additional context paths
+            'Context optimization test');
+            const mockUpdate = createMockFileUpdate('updateFile', {
+                filePath: '/mocked/root/dir/test.js',
+                newContent: 'console.log("Optimized context");',
+            });
+            const mockSequence = createMockResponseSequence(mockPlanning, mockSummary, [mockUpdate]);
+            mockSequence.forEach((response) => {
+                vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(response);
+            });
+            await runCodegen();
+            expect(vertexAi.generateContent).toHaveBeenCalledTimes(3);
+            // Check if context paths were included in the call
+            const calls = vi.mocked(vertexAi.generateContent).mock.calls;
+            expect(calls[1][0]).toEqual(expect.arrayContaining([
+                expect.objectContaining({
+                    type: 'user',
+                    functionResponses: expect.arrayContaining([
+                        expect.objectContaining({
+                            name: 'getSourceCode',
+                        }),
+                    ]),
+                }),
+            ]));
+        });
+    });
+});
+//# sourceMappingURL=codegen.cli-options.test.js.map

package/dist/main/codegen.cli-options.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codegen.cli-options.test.js","sourceRoot":"","sources":["../../src/main/codegen.cli-options.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,KAAK,QAAQ,MAAM,4BAA4B,CAAC;AACvD,OAAO,KAAK,WAAW,MAAM,0BAA0B,CAAC;AACxD,OAAO,KAAK,SAAS,MAAM,sBAAsB,CAAC;AAClD,OAAO,KAAK,UAAU,MAAM,uBAAuB,CAAC;AACpD,OAAO,wBAAwB,CAAC;AAChC,OAAO,wBAAwB,CAAC;AAChC,OAAO,aAAa,CAAC;AAErB,OAAO,EACL,0BAA0B,EAC1B,wBAAwB,EACxB,oBAAoB,EACpB,0BAA0B,GAC3B,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAEvC,4BAA4B;AAC5B,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,WAAW,EAAE,KAAK;IAClB,EAAE,EAAE,KAAK;IACT,MAAM,EAAE,IAAI;IACZ,mBAAmB,EAAE,IAAI;IACzB,cAAc,EAAE,KAAK;IACrB,eAAe,EAAE,KAAK;IACtB,eAAe,EAAE,KAAK;IACtB,oBAAoB,EAAE,KAAK;IAC3B,aAAa,EAAE,KAAK;IACpB,aAAa,EAAE,KAAK;IACpB,cAAc,EAAE,KAAK;IACrB,aAAa,EAAE,KAAK;IACpB,QAAQ,EAAE,KAAK;IACf,MAAM,EAAE,KAAK;IACb,MAAM,EAAE,KAAK;IACb,WAAW,EAAE,GAAG;IAChB,KAAK,EAAE,KAAK;IACZ,QAAQ,EAAE,SAAS;IACnB,kBAAkB,EAAE,SAAS;IAC7B,WAAW,EAAE,KAAK;IAClB,0BAA0B,EAAE,IAAI;IAChC,eAAe,EAAE,SAAS;IAC1B,WAAW,EAAE,SAAS;IACtB,cAAc,EAAE,EAAE;IAClB,YAAY,EAAE,SAAS;IACvB,wBAAwB,EAAE,SAAS;IACnC,cAAc,EAAE,IAAI;IACpB,0BAA0B,EAAE,IAAI;IAChC,SAAS,EAAE,SAA+B;IAC1C,MAAM,EAAE,KAAK;CACd,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,4BAA4B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAC5E,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;AACpC,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;AAClC,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,cAAc,EAAE,GAAG,EAAE,CAAC,EAAE;IACxB,kBAAkB,EAAE,GAAG,EAAE,CAAC,EAAE;IAC5B,YAAY,EAAE,GAAG,EAAE,CAAC,IAAI;CACzB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5B,OAAO,EAAE,kBAAkB;IAC3B,QAAQ,EAAE;QACR,OAAO,EAAE,kBAAkB;QAC3B,UAAU,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;QAC1C,WAAW,EAAE,kBAAkB;KAChC;IACD,gBAAgB,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,yCAAyC;QACzC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,kBAAkB,GAAG,IAAI,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC;QACpC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC;QACpC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,SAAS,CAAC;QACxC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,KAAK,GAAG,KAAK,CAAC;QACnC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,GAAG,GAAG,CAAC;QACvC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,GAAG,MAAM,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;YAC5F,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,GAAG,IAAI,CAAC;YAE1C,MAAM,UAAU,EAAE,CAAC;YAEnB,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACvD,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACxD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;YACzF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC;YACnC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;YAE7C,MAAM,YAAY,GAAG,0BAA0B,CAAC,2BAA2B,EAAE,qBAAqB,EAAE;gBAClG,EAAE,QAAQ,EAAE,0BAA0B,EAAE,MAAM,EAAE,aAAa,EAAE;aAChE,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,wBAAwB,CAC1C,CAAC,EAAE,QAAQ,EAAE,0BAA0B,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,EAAE,qBAAqB,EAAE,CAAC,EACvG,EAAE,EACF,kBAAkB,CACnB,CAAC;YAEF,MAAM,UAAU,GAAG,oBAAoB,CAAC,YAAY,EAAE;gBACpD,QAAQ,EAAE,0BAA0B;gBACpC,UAAU,EAAE,yBAAyB;aACtC,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,0BAA0B,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YACzF,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACtE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,EAAE,CAAC;YAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1D,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,GAAG,GAAG,CAAC;YACvC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;YAE7C,MAAM,YAAY,GAAG,0BAA0B,CAAC,uCAAuC,EAAE,qBAAqB,EAAE;gBAC9G,EAAE,QAAQ,EAAE,0BAA0B,EAAE,MAAM,EAAE,kBAAkB,EAAE;aACrE,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,wBAAwB,CAC1C;gBACE;oBACE,QAAQ,EAAE,0BAA0B;oBACpC,cAAc,EAAE,YAAY;oBAC5B,MAAM,EAAE,gCAAgC;iBACzC;aACF,EACD,EAAE,EACF,yBAAyB,CAC1B,CAAC;YAEF,MAAM,UAAU,GAAG,oBAAoB,CAAC,YAAY,EAAE;gBACpD,QAAQ,EAAE,0BAA0B;gBACpC,UAAU,EAAE,oCAAoC;aACjD,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,0BAA0B,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YACzF,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACtE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,EAAE,CAAC;YAEnB,MAAM,KAAK,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;YAC7D,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBACrB,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,8BAA8B;YAC3D,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,0BAA0B,GAAG,KAAK,CAAC;YACxD,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;YAC7C,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,GAAG,SAAS,CAAC;YAE5C,MAAM,YAAY,GAAG,0BAA0B,CAC7C,yCAAyC,EACzC,qBAAqB,EACrB,CAAC,EAAE,QAAQ,EAAE,0BAA0B,EAAE,MAAM,EAAE,2BAA2B,EAAE,CAAC,CAChF,CAAC;YAEF,MAAM,WAAW,GAAG,wBAAwB,CAC1C,CAAC,EAAE,QAAQ,EAAE,0BAA0B,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,EAAE,0BAA0B,EAAE,CAAC,EAC5G,CAAC,6BAA6B,CAAC,EAAE,2BAA2B;YAC5D,2BAA2B,CAC5B,CAAC;YAEF,MAAM,UAAU,GAAG,oBAAoB,CAAC,YAAY,EAAE;gBACpD,QAAQ,EAAE,0BAA0B;gBACpC,UAAU,EAAE,mCAAmC;aAChD,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,0BAA0B,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YACzF,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACtE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,EAAE,CAAC;YAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1D,mDAAmD;YACnD,MAAM,KAAK,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;YAC7D,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CACzB,MAAM,CAAC,eAAe,CAAC;gBACrB,MAAM,CAAC,gBAAgB,CAAC;oBACtB,IAAI,EAAE,MAAM;oBACZ,iBAAiB,EAAE,MAAM,CAAC,eAAe,CAAC;wBACxC,MAAM,CAAC,gBAAgB,CAAC;4BACtB,IAAI,EAAE,eAAe;yBACtB,CAAC;qBACH,CAAC;iBACH,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/main/codegen.image-generation.test.d.ts

@@ -0,0 +1,3 @@
+import '../files/find-files.js';
+import '../files/cache-file.js';
+import './config.js';

package/dist/main/codegen.image-generation.test.js

@@ -0,0 +1,187 @@
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { runCodegen } from './codegen.js';
+import * as vertexAi from '../ai-service/vertex-ai.js';
+import * as chatGpt from '../ai-service/chat-gpt.js';
+import * as vertexAiImagen from '../ai-service/vertex-ai-imagen.js';
+import * as dallE from '../ai-service/dall-e.js';
+import * as updateFiles from '../files/update-files.js';
+import * as cliParams from '../cli/cli-params.js';
+import '../files/find-files.js';
+import '../files/cache-file.js';
+import './config.js';
+import { createMockPlanningResponse, createMockCodegenSummary, createMockImageGenerationResponse, createMockResponseSequence, } from './codegen.test-utils.js';
+// Mock all required modules
+vi.mock('../cli/cli-params.js', () => ({
+    interactive: false,
+    ui: false,
+    uiPort: 1337,
+    disableExplanations: true,
+    explicitPrompt: false,
+    allowFileCreate: false,
+    allowFileDelete: false,
+    allowDirectoryCreate: false,
+    allowFileMove: false,
+    verbosePrompt: false,
+    vertexAiClaude: false,
+    helpRequested: false,
+    aiStudio: false,
+    vision: false,
+    imagen: false,
+    temperature: 0.7,
+    cheap: false,
+    taskFile: undefined,
+    disableInitialLint: undefined,
+    askQuestion: false,
+    disableContextOptimization: true,
+    geminiBlockNone: undefined,
+    contentMask: undefined,
+    ignorePatterns: [],
+    disableCache: undefined,
+    disableAiServiceFallback: undefined,
+    disableHistory: true,
+    disableConversationSummary: true,
+    aiService: undefined,
+    dryRun: false,
+}));
+vi.mock('../ai-service/vertex-ai.js', () => ({ generateContent: vi.fn() }));
+vi.mock('../ai-service/chat-gpt.js', () => ({ generateContent: vi.fn() }));
+vi.mock('../ai-service/vertex-ai-imagen.js', () => ({ generateImage: vi.fn() }));
+vi.mock('../ai-service/dall-e.js', () => ({ generateImage: vi.fn() }));
+vi.mock('../files/update-files.js');
+vi.mock('../files/cache-file.js');
+vi.mock('../files/find-files.js', () => ({
+    getSourceFiles: () => [],
+    getImageAssetFiles: () => [],
+    refreshFiles: () => null,
+}));
+vi.mock('./config.js', () => ({
+    rootDir: '/mocked/root/dir',
+    rcConfig: {
+        rootDir: '/mocked/root/dir',
+        extensions: ['.js', '.ts', '.tsx', '.jsx'],
+    },
+    importantContext: {},
+}));
+describe('Image Generation', () => {
+    beforeEach(() => {
+        vi.resetAllMocks();
+        // Reset CLI parameters to default values
+        vi.mocked(cliParams).aiService = undefined;
+        vi.mocked(cliParams).dryRun = false;
+        vi.mocked(cliParams).helpRequested = false;
+        vi.mocked(cliParams).vision = false;
+        vi.mocked(cliParams).imagen = undefined;
+        vi.mocked(cliParams).cheap = false;
+        vi.mocked(cliParams).temperature = 0.7;
+        vi.mocked(cliParams).explicitPrompt = 'test';
+    });
+    describe('Vertex AI Imagen', () => {
+        it('should use Vertex AI Imagen when imagen flag is set to vertex-ai', async () => {
+            vi.mocked(cliParams).imagen = 'vertex-ai';
+            vi.mocked(cliParams).aiService = 'vertex-ai';
+            const mockPlanning = createMockPlanningResponse('Test analysis for Vertex AI Imagen', 'Generate landscape image', [{ filePath: '/mocked/root/dir/landscape.png', reason: 'Generate test image' }]);
+            const mockSummary = createMockCodegenSummary([
+                {
+                    filePath: '/mocked/root/dir/landscape.png',
+                    updateToolName: 'generateImage',
+                    prompt: 'Generate image',
+                },
+            ], [], 'Mock summary with image generation');
+            const mockGenerate = createMockImageGenerationResponse('A beautiful landscape', '/mocked/root/dir/landscape.png', 512, 512, 'Generate a beautiful landscape image');
+            const mockSequence = createMockResponseSequence(mockPlanning, mockSummary, [mockGenerate]);
+            mockSequence.forEach((response) => {
+                vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(response);
+            });
+            vi.mocked(vertexAiImagen.generateImage).mockResolvedValueOnce('mocked-image-data');
+            await runCodegen();
+            expect(vertexAi.generateContent).toHaveBeenCalledTimes(3);
+            expect(vertexAiImagen.generateImage).toHaveBeenCalledWith('A beautiful landscape', undefined, {
+                height: 512,
+                width: 512,
+            }, false);
+            expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockGenerate, expect.anything());
+        });
+        it('should pass the cheap parameter to Vertex AI Imagen when --cheap flag is true', async () => {
+            vi.mocked(cliParams).imagen = 'vertex-ai';
+            vi.mocked(cliParams).aiService = 'vertex-ai';
+            vi.mocked(cliParams).cheap = true;
+            const mockPlanning = createMockPlanningResponse('Test analysis for cheap Vertex AI Imagen', 'Generate small landscape image', [{ filePath: '/mocked/root/dir/landscape.png', reason: 'Generate small test image' }]);
+            const mockSummary = createMockCodegenSummary([
+                {
+                    filePath: '/mocked/root/dir/landscape.png',
+                    updateToolName: 'generateImage',
+                    prompt: 'Generate small image',
+                },
+            ], [], 'Mock summary with cheap image generation');
+            const mockGenerate = createMockImageGenerationResponse('A simple landscape', '/mocked/root/dir/landscape.png', 256, 256, 'Generate a simple landscape image');
+            const mockSequence = createMockResponseSequence(mockPlanning, mockSummary, [mockGenerate]);
+            mockSequence.forEach((response) => {
+                vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(response);
+            });
+            vi.mocked(vertexAiImagen.generateImage).mockResolvedValueOnce('mocked-cheap-image-data');
+            await runCodegen();
+            expect(vertexAiImagen.generateImage).toHaveBeenCalledWith('A simple landscape', undefined, {
+                height: 256,
+                width: 256,
+            }, false);
+        });
+    });
+    describe('DALL-E', () => {
+        it('should use DALL-E when imagen flag is set to dall-e', async () => {
+            vi.mocked(cliParams).imagen = 'dall-e';
+            vi.mocked(cliParams).aiService = 'chat-gpt';
+            const mockPlanning = createMockPlanningResponse('Test analysis for DALL-E', 'Generate city image', [
+                { filePath: '/mocked/root/dir/city.png', reason: 'Generate test image' },
+            ]);
+            const mockSummary = createMockCodegenSummary([
+                {
+                    filePath: '/mocked/root/dir/city.png',
+                    updateToolName: 'generateImage',
+                    prompt: 'Generate image',
+                },
+            ], [], 'Mock summary with image generation');
+            const mockGenerate = createMockImageGenerationResponse('A futuristic city', '/mocked/root/dir/city.png', 1024, 1024, 'Generate a futuristic city image');
+            const mockSequence = createMockResponseSequence(mockPlanning, mockSummary, [mockGenerate]);
+            mockSequence.forEach((response) => {
+                vi.mocked(chatGpt.generateContent).mockResolvedValueOnce(response);
+            });
+            vi.mocked(dallE.generateImage).mockResolvedValueOnce('mocked-image-data');
+            await runCodegen();
+            expect(chatGpt.generateContent).toHaveBeenCalledTimes(3);
+            expect(dallE.generateImage).toHaveBeenCalledWith('A futuristic city', undefined, {
+                width: 1024,
+                height: 1024,
+            }, false);
+            expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockGenerate, expect.anything());
+        });
+    });
+    describe('Error Handling', () => {
+        it('should throw an error when imagen flag is set but no AI service is specified', async () => {
+            vi.mocked(cliParams).imagen = 'vertex-ai';
+            vi.mocked(cliParams).aiService = undefined;
+            await expect(runCodegen()).rejects.toThrow('Please specify which AI service should be used');
+        });
+        it('should handle image generation errors gracefully', async () => {
+            vi.mocked(cliParams).imagen = 'vertex-ai';
+            vi.mocked(cliParams).aiService = 'vertex-ai';
+            const mockPlanning = createMockPlanningResponse();
+            const mockSummary = createMockCodegenSummary([
+                {
+                    filePath: '/mocked/root/dir/error.png',
+                    updateToolName: 'generateImage',
+                    prompt: 'Generate error image',
+                },
+            ], [], 'Mock summary with error');
+            const mockGenerate = createMockImageGenerationResponse('Error image', '/mocked/root/dir/error.png');
+            const mockSequence = createMockResponseSequence(mockPlanning, mockSummary, [mockGenerate]);
+            mockSequence.forEach((response) => {
+                vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(response);
+            });
+            vi.mocked(vertexAiImagen.generateImage).mockRejectedValueOnce(new Error('Generation failed'));
+            await runCodegen();
+            expect(vertexAiImagen.generateImage).toHaveBeenCalled();
+            // The error should be caught and handled without breaking the test
+        });
+    });
+});
+//# sourceMappingURL=codegen.image-generation.test.js.map

package/dist/main/codegen.image-generation.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codegen.image-generation.test.js","sourceRoot":"","sources":["../../src/main/codegen.image-generation.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,KAAK,QAAQ,MAAM,4BAA4B,CAAC;AACvD,OAAO,KAAK,OAAO,MAAM,2BAA2B,CAAC;AACrD,OAAO,KAAK,cAAc,MAAM,mCAAmC,CAAC;AACpE,OAAO,KAAK,KAAK,MAAM,yBAAyB,CAAC;AACjD,OAAO,KAAK,WAAW,MAAM,0BAA0B,CAAC;AACxD,OAAO,KAAK,SAAS,MAAM,sBAAsB,CAAC;AAClD,OAAO,wBAAwB,CAAC;AAChC,OAAO,wBAAwB,CAAC;AAChC,OAAO,aAAa,CAAC;AAErB,OAAO,EACL,0BAA0B,EAC1B,wBAAwB,EACxB,iCAAiC,EACjC,0BAA0B,GAC3B,MAAM,yBAAyB,CAAC;AAEjC,4BAA4B;AAC5B,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,WAAW,EAAE,KAAK;IAClB,EAAE,EAAE,KAAK;IACT,MAAM,EAAE,IAAI;IACZ,mBAAmB,EAAE,IAAI;IACzB,cAAc,EAAE,KAAK;IACrB,eAAe,EAAE,KAAK;IACtB,eAAe,EAAE,KAAK;IACtB,oBAAoB,EAAE,KAAK;IAC3B,aAAa,EAAE,KAAK;IACpB,aAAa,EAAE,KAAK;IACpB,cAAc,EAAE,KAAK;IACrB,aAAa,EAAE,KAAK;IACpB,QAAQ,EAAE,KAAK;IACf,MAAM,EAAE,KAAK;IACb,MAAM,EAAE,KAAK;IACb,WAAW,EAAE,GAAG;IAChB,KAAK,EAAE,KAAK;IACZ,QAAQ,EAAE,SAAS;IACnB,kBAAkB,EAAE,SAAS;IAC7B,WAAW,EAAE,KAAK;IAClB,0BAA0B,EAAE,IAAI;IAChC,eAAe,EAAE,SAAS;IAC1B,WAAW,EAAE,SAAS;IACtB,cAAc,EAAE,EAAE;IAClB,YAAY,EAAE,SAAS;IACvB,wBAAwB,EAAE,SAAS;IACnC,cAAc,EAAE,IAAI;IACpB,0BAA0B,EAAE,IAAI;IAChC,SAAS,EAAE,SAA+B;IAC1C,MAAM,EAAE,KAAK;CACd,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,4BAA4B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAC5E,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3E,EAAE,CAAC,IAAI,CAAC,mCAAmC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACjF,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACvE,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;AACpC,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;AAClC,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,cAAc,EAAE,GAAG,EAAE,CAAC,EAAE;IACxB,kBAAkB,EAAE,GAAG,EAAE,CAAC,EAAE;IAC5B,YAAY,EAAE,GAAG,EAAE,CAAC,IAAI;CACzB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5B,OAAO,EAAE,kBAAkB;IAC3B,QAAQ,EAAE;QACR,OAAO,EAAE,kBAAkB;QAC3B,UAAU,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;KAC3C;IACD,gBAAgB,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,yCAAyC;QACzC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC;QACpC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC;QACpC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,SAAS,CAAC;QACxC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,KAAK,GAAG,KAAK,CAAC;QACnC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,GAAG,GAAG,CAAC;QACvC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,GAAG,MAAM,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,WAAW,CAAC;YAC1C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;YAE7C,MAAM,YAAY,GAAG,0BAA0B,CAC7C,oCAAoC,EACpC,0BAA0B,EAC1B,CAAC,EAAE,QAAQ,EAAE,gCAAgC,EAAE,MAAM,EAAE,qBAAqB,EAAE,CAAC,CAChF,CAAC;YAEF,MAAM,WAAW,GAAG,wBAAwB,CAC1C;gBACE;oBACE,QAAQ,EAAE,gCAAgC;oBAC1C,cAAc,EAAE,eAAe;oBAC/B,MAAM,EAAE,gBAAgB;iBACzB;aACF,EACD,EAAE,EACF,oCAAoC,CACrC,CAAC;YAEF,MAAM,YAAY,GAAG,iCAAiC,CACpD,uBAAuB,EACvB,gCAAgC,EAChC,GAAG,EACH,GAAG,EACH,sCAAsC,CACvC,CAAC;YAEF,MAAM,YAAY,GAAG,0BAA0B,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;YAC3F,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACtE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC;YAEnF,MAAM,UAAU,EAAE,CAAC;YAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1D,MAAM,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACvD,uBAAuB,EACvB,SAAS,EACT;gBACE,MAAM,EAAE,GAAG;gBACX,KAAK,EAAE,GAAG;aACX,EACD,KAAK,CACN,CAAC;YACF,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,YAAY,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QACxF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;YAC7F,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,WAAW,CAAC;YAC1C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;YAC7C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC;YAElC,MAAM,YAAY,GAAG,0BAA0B,CAC7C,0CAA0C,EAC1C,gCAAgC,EAChC,CAAC,EAAE,QAAQ,EAAE,gCAAgC,EAAE,MAAM,EAAE,2BAA2B,EAAE,CAAC,CACtF,CAAC;YAEF,MAAM,WAAW,GAAG,wBAAwB,CAC1C;gBACE;oBACE,QAAQ,EAAE,gCAAgC;oBAC1C,cAAc,EAAE,eAAe;oBAC/B,MAAM,EAAE,sBAAsB;iBAC/B;aACF,EACD,EAAE,EACF,0CAA0C,CAC3C,CAAC;YAEF,MAAM,YAAY,GAAG,iCAAiC,CACpD,oBAAoB,EACpB,gCAAgC,EAChC,GAAG,EACH,GAAG,EACH,mCAAmC,CACpC,CAAC;YAEF,MAAM,YAAY,GAAG,0BAA0B,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;YAC3F,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACtE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,qBAAqB,CAAC,yBAAyB,CAAC,CAAC;YAEzF,MAAM,UAAU,EAAE,CAAC;YAEnB,MAAM,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACvD,oBAAoB,EACpB,SAAS,EACT;gBACE,MAAM,EAAE,GAAG;gBACX,KAAK,EAAE,GAAG;aACX,EACD,KAAK,CACN,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,QAAQ,CAAC;YACvC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,UAAU,CAAC;YAE5C,MAAM,YAAY,GAAG,0BAA0B,CAAC,0BAA0B,EAAE,qBAAqB,EAAE;gBACjG,EAAE,QAAQ,EAAE,2BAA2B,EAAE,MAAM,EAAE,qBAAqB,EAAE;aACzE,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,wBAAwB,CAC1C;gBACE;oBACE,QAAQ,EAAE,2BAA2B;oBACrC,cAAc,EAAE,eAAe;oBAC/B,MAAM,EAAE,gBAAgB;iBACzB;aACF,EACD,EAAE,EACF,oCAAoC,CACrC,CAAC;YAEF,MAAM,YAAY,GAAG,iCAAiC,CACpD,mBAAmB,EACnB,2BAA2B,EAC3B,IAAI,EACJ,IAAI,EACJ,kCAAkC,CACnC,CAAC;YAEF,MAAM,YAAY,GAAG,0BAA0B,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;YAC3F,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChC,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACrE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC;YAE1E,MAAM,UAAU,EAAE,CAAC;YAEnB,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACzD,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAC9C,mBAAmB,EACnB,SAAS,EACT;gBACE,KAAK,EAAE,IAAI;gBACX,MAAM,EAAE,IAAI;aACb,EACD,KAAK,CACN,CAAC;YACF,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,YAAY,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QACxF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;YAC5F,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,WAAW,CAAC;YAC1C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,SAAS,CAAC;YAE3C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,gDAAgD,CAAC,CAAC;QAC/F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,WAAW,CAAC;YAC1C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,WAAW,CAAC;YAE7C,MAAM,YAAY,GAAG,0BAA0B,EAAE,CAAC;YAClD,MAAM,WAAW,GAAG,wBAAwB,CAC1C;gBACE;oBACE,QAAQ,EAAE,4BAA4B;oBACtC,cAAc,EAAE,eAAe;oBAC/B,MAAM,EAAE,sBAAsB;iBAC/B;aACF,EACD,EAAE,EACF,yBAAyB,CAC1B,CAAC;YAEF,MAAM,YAAY,GAAG,iCAAiC,CAAC,aAAa,EAAE,4BAA4B,CAAC,CAAC;YAEpG,MAAM,YAAY,GAAG,0BAA0B,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;YAC3F,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACtE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAC;YAE9F,MAAM,UAAU,EAAE,CAAC;YAEnB,MAAM,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACxD,mEAAmE;QACrE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}