genaicode 0.5.1 → 0.6.0

package/dist/src/files/read-files.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"read-files.js","sourceRoot":"","sources":["../../../src/files/read-files.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,YAAY,CAAC;AAC9B,OAAO,MAAM,MAAM,YAAY,CAAC;AAChC,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,SAAS,MAAM,YAAY,CAAC;AAEnC,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrE,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAE/D,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAE,MAAM,uCAAuC,CAAC;AAanE;;GAEG;AACH,SAAS,eAAe,CACtB,EAAE,WAAW,EAAE,cAAc,EAAkB,EAC/C,WAAsB,EACtB,QAAQ,GAAG,KAAK;IAEhB,MAAM,UAAU,GAAkB,EAAE,CAAC;IACrC,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,gBAAgB,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IAE7D,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,EAAE,CAAC;QACpC,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3E,iCAAiC;YACjC,IAAI,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC7B,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,CAAC;gBAC/D,SAAS;YACX,CAAC;YAED,MAAM,OAAO,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;YAEjC,wCAAwC;YACxC,IAAI,CAAC,WAAW,IAAI,WAAW,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC7C,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;gBAC3D,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC1C,UAAU,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;oBAChE,SAAS;gBACX,CAAC;YACH,CAAC;YAED,IAAI,CAAC,QAAQ,IAAI,cAAc,EAAE,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;gBAClF,UAAU,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAClE,CAAC;iBAAM,CAAC;gBACN,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAC/C,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,GAAG,OAAO,EAAE,CAAC;YAC7C,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,4CAA4C;AAC5C,MAAM,UAAU,aAAa,CAC3B,EACE,WAAW,EACX,QAAQ,EACR,QAAQ,GAKT,EACD,OAAuB;IAEvB,MAAM,UAAU,GAAG,eAAe,CAAC,OAAO,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC;IAEnE,IAAI,QAAQ,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACtC,UAAU,CAAC,QAAQ,CAAC,GAAG;YACrB,OAAO,EAAE,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC;SAC5C,CAAC;IACJ,CAAC;IAED,qBAAqB,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC;IAClD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,oCAAoC;AACpC,MAAM,UAAU,cAAc;IAC5B,MAAM,WAAW,GAAmB,EAAE,CAAC;IACvC,KAAK,MAAM,IAAI,IAAI,kBAAkB,EAAE,EAAE,CAAC;QACxC,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;QAChC,WAAW,CAAC,IAAI,CAAC,GAAG;YAClB,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YAC3B,KAAK,EAAE,UAAU,CAAC,KAAM;YACxB,MAAM,EAAE,UAAU,CAAC,MAAO;SAC3B,CAAC;IACJ,CAAC;IACD,qBAAqB,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC;IACnD,OAAO,WAAW,CAAC;AACrB,CAAC"}

package/dist/src/files/read-files.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/src/files/read-files.test.js

@@ -0,0 +1,89 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import fs from 'fs';
+import mime from 'mime-types';
+import sizeOf from 'image-size';
+import { getSourceCode, getImageAssets } from './read-files.js';
+import { getSourceFiles, getImageAssetFiles } from './find-files.js';
+import { verifySourceCodeLimit } from '../prompt/limits.js';
+import * as cliParams from '../cli/cli-params.js';
+import { rcConfig } from '../main/config.js';
+vi.mock('fs');
+vi.mock('mime-types');
+vi.mock('image-size');
+vi.mock('./find-files.js', () => ({
+    getImageAssetFiles: vi.fn(),
+    getSourceFiles: vi.fn(),
+}));
+vi.mock('../prompt/limits.js');
+vi.mock('../cli/cli-params.js', () => ({
+    taskFile: null,
+    contentMask: null,
+    disableExplanations: true,
+    ignorePatterns: [],
+}));
+vi.mock('../main/config.js', () => ({
+    rcConfig: { rootDir: '/home/project' },
+    importantContext: {},
+}));
+describe('read-files', () => {
+    beforeEach(() => {
+        vi.resetAllMocks();
+        vi.mocked(cliParams).contentMask = undefined;
+    });
+    afterEach(() => {
+        vi.clearAllMocks();
+    });
+    describe('getSourceCode', () => {
+        it('should return source code for all files', () => {
+            const mockFiles = ['/home/project/file1.js', '/home/project/file2.js'];
+            vi.mocked(getSourceFiles).mockReturnValue(mockFiles);
+            vi.mocked(fs).readFileSync.mockImplementation((file) => `Content of ${file}`);
+            rcConfig.rootDir = '/home/project';
+            const result = getSourceCode({ taskFile: undefined }, { aiService: 'vertex-ai' });
+            expect(result).toEqual({
+                '/home/project/file1.js': { content: 'Content of /home/project/file1.js' },
+                '/home/project/file2.js': { content: 'Content of /home/project/file2.js' },
+            });
+            expect(verifySourceCodeLimit).toHaveBeenCalled();
+        });
+        it('should apply content mask when specified', () => {
+            const mockFiles = ['/home/project/file1.js', '/home/project/subfolder/file2.js'];
+            vi.mocked(getSourceFiles).mockReturnValue(mockFiles);
+            vi.mocked(fs).readFileSync.mockImplementation((file) => `Content of ${file}`);
+            rcConfig.rootDir = '/home/project';
+            const result = getSourceCode({ taskFile: undefined }, { aiService: 'vertex-ai', contentMask: 'subfolder' });
+            expect(result).toEqual({
+                '/home/project/file1.js': { content: null },
+                '/home/project/subfolder/file2.js': {
+                    content: 'Content of /home/project/subfolder/file2.js',
+                },
+            });
+        });
+        it('should include task file when specified', () => {
+            const mockFiles = ['/home/project/file1.js'];
+            vi.mocked(getSourceFiles).mockReturnValue(mockFiles);
+            vi.mocked(fs).readFileSync.mockImplementation((file) => `Content of ${file}`);
+            rcConfig.rootDir = '/home/project';
+            const result = getSourceCode({ taskFile: '/home/project/task.md' }, { aiService: 'vertex-ai' });
+            expect(result).toEqual({
+                '/home/project/file1.js': { content: 'Content of /home/project/file1.js' },
+                '/home/project/task.md': { content: 'Content of /home/project/task.md' },
+            });
+        });
+    });
+    describe('getImageAssets', () => {
+        it('should return image assets information', () => {
+            const mockImageFiles = ['/home/project/image1.png', '/home/project/image2.jpg'];
+            vi.mocked(getImageAssetFiles).mockReturnValue(mockImageFiles);
+            vi.mocked(mime).lookup.mockImplementation((file) => (file.endsWith('.png') ? 'image/png' : 'image/jpeg'));
+            vi.mocked(sizeOf).mockImplementation(() => ({ width: 100, height: 200 }));
+            const result = getImageAssets();
+            expect(result).toEqual({
+                '/home/project/image1.png': { mimeType: 'image/png', width: 100, height: 200 },
+                '/home/project/image2.jpg': { mimeType: 'image/jpeg', width: 100, height: 200 },
+            });
+            expect(verifySourceCodeLimit).toHaveBeenCalled();
+        });
+    });
+});
+//# sourceMappingURL=read-files.test.js.map

package/dist/src/files/read-files.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"read-files.test.js","sourceRoot":"","sources":["../../../src/files/read-files.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,YAAY,CAAC;AAC9B,OAAO,MAAM,MAAM,YAAY,CAAC;AAChC,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrE,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,KAAK,SAAS,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACd,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;AACtB,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;AACtB,EAAE,CAAC,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,CAAC;IAChC,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;IAC3B,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;CACxB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAC/B,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,QAAQ,EAAE,IAAI;IACd,WAAW,EAAE,IAAI;IACjB,mBAAmB,EAAE,IAAI;IACzB,cAAc,EAAE,EAAE;CACnB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,QAAQ,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE;IACtC,gBAAgB,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,GAAG,SAAS,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,SAAS,GAAG,CAAC,wBAAwB,EAAE,wBAAwB,CAAC,CAAC;YACvE,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YACrD,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;YAC9E,QAAQ,CAAC,OAAO,GAAG,eAAe,CAAC;YAEnC,MAAM,MAAM,GAAG,aAAa,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CAAC;YAElF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,wBAAwB,EAAE,EAAE,OAAO,EAAE,mCAAmC,EAAE;gBAC1E,wBAAwB,EAAE,EAAE,OAAO,EAAE,mCAAmC,EAAE;aAC3E,CAAC,CAAC;YACH,MAAM,CAAC,qBAAqB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,SAAS,GAAG,CAAC,wBAAwB,EAAE,kCAAkC,CAAC,CAAC;YACjF,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YACrD,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;YAC9E,QAAQ,CAAC,OAAO,GAAG,eAAe,CAAC;YAEnC,MAAM,MAAM,GAAG,aAAa,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,CAAC,CAAC;YAE5G,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,wBAAwB,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;gBAC3C,kCAAkC,EAAE;oBAClC,OAAO,EAAE,6CAA6C;iBACvD;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,SAAS,GAAG,CAAC,wBAAwB,CAAC,CAAC;YAC7C,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YACrD,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;YAC9E,QAAQ,CAAC,OAAO,GAAG,eAAe,CAAC;YAEnC,MAAM,MAAM,GAAG,aAAa,CAAC,EAAE,QAAQ,EAAE,uBAAuB,EAAE,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CAAC;YAEhG,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,wBAAwB,EAAE,EAAE,OAAO,EAAE,mCAAmC,EAAE;gBAC1E,uBAAuB,EAAE,EAAE,OAAO,EAAE,kCAAkC,EAAE;aACzE,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,cAAc,GAAG,CAAC,0BAA0B,EAAE,0BAA0B,CAAC,CAAC;YAChF,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YAC9D,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;YAC1G,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;YAE1E,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;YAEhC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,0BAA0B,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;gBAC9E,0BAA0B,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;aAChF,CAAC,CAAC;YACH,MAAM,CAAC,qBAAqB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/files/temp-buffer.d.ts

@@ -0,0 +1,2 @@
+export declare function getTempBuffer(url: string): Buffer | undefined;
+export declare function setTempBuffer(buffer: Buffer): string;

package/dist/src/files/temp-buffer.js

@@ -0,0 +1,12 @@
+/** Temporary storage */
+let tempId = 1;
+const temp = {};
+export function getTempBuffer(url) {
+    return temp[url];
+}
+export function setTempBuffer(buffer) {
+    const imageUrl = 'temp://' + tempId++;
+    temp[imageUrl] = buffer;
+    return imageUrl;
+}
+//# sourceMappingURL=temp-buffer.js.map

package/dist/src/files/temp-buffer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"temp-buffer.js","sourceRoot":"","sources":["../../../src/files/temp-buffer.ts"],"names":[],"mappings":"AAAA,wBAAwB;AACxB,IAAI,MAAM,GAAG,CAAC,CAAC;AACf,MAAM,IAAI,GAA8B,EAAE,CAAC;AAE3C,MAAM,UAAU,aAAa,CAAC,GAAW;IACvC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,MAAc;IAC1C,MAAM,QAAQ,GAAG,SAAS,GAAG,MAAM,EAAE,CAAC;IACtC,IAAI,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC;IACxB,OAAO,QAAQ,CAAC;AAClB,CAAC"}

package/dist/src/files/update-files.d.ts

@@ -0,0 +1,6 @@
+import { CodegenOptions } from '../main/codegen-types.js';
+import { FunctionCall } from '../ai-service/common.js';
+/**
+ * @param functionCalls Result of the code generation, a map of file paths to new content
+ */
+export declare function updateFiles(functionCalls: FunctionCall[], options: CodegenOptions): Promise<void>;

package/dist/src/files/update-files.js

@@ -0,0 +1,19 @@
+import { getOperationExecutor } from '../operations/operations-index.js';
+/**
+ * @param functionCalls Result of the code generation, a map of file paths to new content
+ */
+export async function updateFiles(functionCalls, options) {
+    for (const { name, args } of functionCalls) {
+        const executor = getOperationExecutor(name);
+        if (!executor) {
+            console.warn(`Unknown operation: ${name}`);
+        }
+        else if (args === undefined) {
+            console.warn(`Operation ${name} has no arguments`);
+        }
+        else {
+            await executor(args, options);
+        }
+    }
+}
+//# sourceMappingURL=update-files.js.map

package/dist/src/files/update-files.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"update-files.js","sourceRoot":"","sources":["../../../src/files/update-files.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AAEzE;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,aAA6B,EAAE,OAAuB;IACtF,KAAK,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,aAAa,EAAE,CAAC;QAC3C,MAAM,QAAQ,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,CAAC,IAAI,CAAC,sBAAsB,IAAI,EAAE,CAAC,CAAC;QAC7C,CAAC;aAAM,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC,aAAa,IAAI,mBAAmB,CAAC,CAAC;QACrD,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/src/images/ensure-alpha.d.ts

@@ -0,0 +1 @@
+export declare function ensureAlpha(image: string | Buffer): Promise<Buffer>;

package/dist/src/images/ensure-alpha.js

@@ -0,0 +1,5 @@
+import sharp from 'sharp';
+export async function ensureAlpha(image) {
+    return sharp(image).ensureAlpha().toBuffer();
+}
+//# sourceMappingURL=ensure-alpha.js.map

package/dist/src/images/ensure-alpha.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ensure-alpha.js","sourceRoot":"","sources":["../../../src/images/ensure-alpha.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,KAAsB;IACtD,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,QAAQ,EAAE,CAAC;AAC/C,CAAC"}

package/dist/src/images/imgly-remove-background.d.ts

@@ -0,0 +1,2 @@
+/** Converts white color to transparency on a image and saves to destination path */
+export declare function imglyRemoveBackground(inputFilePath: string, outputFilePath: string): Promise<string>;

package/dist/src/images/imgly-remove-background.js

@@ -0,0 +1,21 @@
+import fs from 'fs';
+import { createRequire } from 'node:module';
+import { removeBackground } from '@imgly/background-removal-node';
+/** Converts white color to transparency on a image and saves to destination path */
+export async function imglyRemoveBackground(inputFilePath, outputFilePath) {
+    try {
+        console.log(`Removing background for image: ${inputFilePath}`);
+        const publicPath = 'file://' + createRequire(import.meta.url).resolve('@imgly/background-removal-node');
+        const blob = await removeBackground(inputFilePath, {
+            publicPath,
+        });
+        fs.writeFileSync(outputFilePath, Buffer.from(await blob.arrayBuffer()));
+        console.log(`Background removed successfully. Saved to: ${outputFilePath}`);
+        return outputFilePath;
+    }
+    catch (error) {
+        console.error('Error removing background:', error);
+        throw error;
+    }
+}
+//# sourceMappingURL=imgly-remove-background.js.map

package/dist/src/images/imgly-remove-background.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"imgly-remove-background.js","sourceRoot":"","sources":["../../../src/images/imgly-remove-background.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAElE,oFAAoF;AACpF,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,aAAqB,EAAE,cAAsB;IACvF,IAAI,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,kCAAkC,aAAa,EAAE,CAAC,CAAC;QAE/D,MAAM,UAAU,GAAG,SAAS,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,gCAAgC,CAAC,CAAC;QACxG,MAAM,IAAI,GAAS,MAAM,gBAAgB,CAAC,aAAa,EAAE;YACvD,UAAU;SACX,CAAC,CAAC;QACH,EAAE,CAAC,aAAa,CAAC,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;QAExE,OAAO,CAAC,GAAG,CAAC,8CAA8C,cAAc,EAAE,CAAC,CAAC;QAC5E,OAAO,cAAc,CAAC;IACxB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,4BAA4B,EAAE,KAAK,CAAC,CAAC;QACnD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/src/images/resize-image.d.ts

@@ -0,0 +1,19 @@
+interface ImageSize {
+    width: number;
+    height: number;
+}
+/**
+ * Resize an image buffer to the desired size
+ * @param {Buffer} input - The input image buffer
+ * @param {ImageSize} size - The desired size of the image
+ * @returns {Promise<Buffer>} - The resized image buffer
+ */
+export declare function resizeImageBuffer(input: Buffer, size: ImageSize): Promise<Buffer>;
+/**
+ * Resize an image file to the desired size and save it back to the same file
+ * @param {string} filePath - The path of the image file to resize
+ * @param {ImageSize} size - The desired size of the image
+ * @returns {Promise<void>}
+ */
+export declare function resizeImageFile(filePath: string, size: ImageSize): Promise<void>;
+export {};

package/dist/src/images/resize-image.js

@@ -0,0 +1,22 @@
+import sharp from 'sharp';
+import fs from 'fs';
+/**
+ * Resize an image buffer to the desired size
+ * @param {Buffer} input - The input image buffer
+ * @param {ImageSize} size - The desired size of the image
+ * @returns {Promise<Buffer>} - The resized image buffer
+ */
+export async function resizeImageBuffer(input, size) {
+    return await sharp(input).resize(size.width, size.height).toBuffer();
+}
+/**
+ * Resize an image file to the desired size and save it back to the same file
+ * @param {string} filePath - The path of the image file to resize
+ * @param {ImageSize} size - The desired size of the image
+ * @returns {Promise<void>}
+ */
+export async function resizeImageFile(filePath, size) {
+    const buffer = await resizeImageBuffer(await fs.promises.readFile(filePath), size);
+    await fs.promises.writeFile(filePath, buffer);
+}
+//# sourceMappingURL=resize-image.js.map

package/dist/src/images/resize-image.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"resize-image.js","sourceRoot":"","sources":["../../../src/images/resize-image.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,MAAM,IAAI,CAAC;AAOpB;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,KAAa,EAAE,IAAe;IACpE,OAAO,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;AACvE,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,QAAgB,EAAE,IAAe;IACrE,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,CAAC;IACnF,MAAM,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;AAChD,CAAC"}

package/dist/src/images/split-image.d.ts

@@ -0,0 +1,18 @@
+interface Rect {
+    x: number;
+    y: number;
+    width: number;
+    height: number;
+}
+interface Part {
+    rect: Rect;
+    outputFilePath: string;
+}
+/**
+ * This function opens the input file, and splits its into parts, extracting parts of the input image, and saving to outputFilePaths
+ *
+ * @param {string} inputFilePath
+ * @param {Part[]} parts
+ */
+export declare function splitImage(inputFilePath: string, parts: Part[]): Promise<void>;
+export {};

package/dist/src/images/split-image.js

@@ -0,0 +1,25 @@
+import sharp from 'sharp';
+/**
+ * This function opens the input file, and splits its into parts, extracting parts of the input image, and saving to outputFilePaths
+ *
+ * @param {string} inputFilePath
+ * @param {Part[]} parts
+ */
+export async function splitImage(inputFilePath, parts) {
+    try {
+        // Process each part
+        for (const part of parts) {
+            const { rect, outputFilePath } = part;
+            const { x, y, width, height } = rect;
+            // Extract the specified rectangle from the input image
+            console.log('Extracting', rect, outputFilePath);
+            await sharp(inputFilePath).extract({ left: x, top: y, width, height }).toFile(outputFilePath);
+        }
+        console.log('Image splitting completed successfully.');
+    }
+    catch (error) {
+        console.error('Error splitting image:', error);
+        throw error;
+    }
+}
+//# sourceMappingURL=split-image.js.map

package/dist/src/images/split-image.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"split-image.js","sourceRoot":"","sources":["../../../src/images/split-image.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAc1B;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,aAAqB,EAAE,KAAa;IACnE,IAAI,CAAC;QACH,oBAAoB;QACpB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC;YACtC,MAAM,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;YAErC,uDAAuD;YACvD,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,EAAE,cAAc,CAAC,CAAC;YAChD,MAAM,KAAK,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QAChG,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;IACzD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;QAC/C,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/src/index.d.ts

@@ -0,0 +1,3 @@
+export { type CodegenOptions } from './main/codegen-types.js';
+export * from './main/codegen.js';
+export * from './prompt/prompt-service.js';

package/dist/src/index.js

@@ -0,0 +1,3 @@
+export * from './main/codegen.js';
+export * from './prompt/prompt-service.js';
+//# sourceMappingURL=index.js.map

package/dist/src/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAEA,cAAc,mBAAmB,CAAC;AAClC,cAAc,4BAA4B,CAAC"}

package/dist/src/main/codegen-types.d.ts

@@ -0,0 +1,58 @@
+import { FunctionDef, GenerateContentFunction } from '../ai-service/common';
+export type PluginAiServiceType = `plugin:${string}`;
+export type AiServiceType = 'vertex-ai' | 'ai-studio' | 'vertex-ai-claude' | 'chat-gpt' | 'anthropic' | PluginAiServiceType;
+export type ImagenType = 'vertex-ai' | 'dall-e';
+export interface UploadedImage {
+    base64url: string;
+    mediaType: 'image/jpeg' | 'image/png' | 'image/gif' | 'image/webp';
+    originalName: string;
+}
+export interface CodegenOptions {
+    explicitPrompt?: string;
+    taskFile?: string;
+    considerAllFiles?: boolean;
+    allowFileCreate?: boolean;
+    allowFileDelete?: boolean;
+    allowDirectoryCreate?: boolean;
+    allowFileMove?: boolean;
+    vision?: boolean;
+    imagen?: ImagenType;
+    aiService: AiServiceType;
+    disableContextOptimization?: boolean;
+    temperature?: number;
+    cheap?: boolean;
+    dryRun?: boolean;
+    verbose?: boolean;
+    requireExplanations?: boolean;
+    geminiBlockNone?: boolean;
+    disableInitialLint?: boolean;
+    contentMask?: string;
+    ignorePatterns?: string[];
+    askQuestion?: boolean;
+    interactive?: boolean;
+    ui?: boolean;
+    uiPort?: number;
+    uiFrameAncestors?: string[];
+    disableCache?: boolean;
+    dependencyTree?: boolean;
+    historyEnabled?: boolean;
+    disableAiServiceFallback?: boolean;
+    selfReflectionEnabled?: boolean;
+    conversationSummaryEnabled?: boolean;
+    images?: UploadedImage[];
+    isDev?: boolean;
+}
+interface ExecutorArgs {
+    [key: string]: unknown;
+}
+export type OperationExecutor = (args: ExecutorArgs, options: CodegenOptions) => Promise<void>;
+export type Operation = {
+    executor: OperationExecutor;
+    def: FunctionDef;
+};
+export interface Plugin {
+    name: string;
+    aiServices?: Record<string, GenerateContentFunction>;
+    operations?: Record<string, Operation>;
+}
+export {};

package/dist/src/main/codegen-types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=codegen-types.js.map

package/dist/src/main/codegen-types.js.map

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

package/dist/src/main/codegen-utils.d.ts

@@ -0,0 +1,3 @@
+import { AiServiceType } from './codegen-types';
+export declare function getSupportedAiServices(): AiServiceType[];
+export declare function cliParamToAiService(): AiServiceType;

package/dist/src/main/codegen-utils.js

@@ -0,0 +1,18 @@
+import * as cliParams from '../cli/cli-params.js';
+import { getRegisteredAiServices } from './plugin-loader.js';
+export function getSupportedAiServices() {
+    return ['vertex-ai', 'ai-studio', 'vertex-ai-claude', 'chat-gpt', 'anthropic', ...getRegisteredAiServices().keys()];
+}
+export function cliParamToAiService() {
+    const aiService = cliParams.aiService;
+    if (!aiService) {
+        throw new Error('Please specify which AI service should be used with --ai-service option');
+    }
+    // Validate that the provided AI service is supported
+    const supportedServices = getSupportedAiServices();
+    if (!supportedServices.includes(aiService)) {
+        throw new Error(`Unsupported AI service: ${aiService}. Supported services are: ${supportedServices.join(', ')}`);
+    }
+    return aiService;
+}
+//# sourceMappingURL=codegen-utils.js.map

package/dist/src/main/codegen-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codegen-utils.js","sourceRoot":"","sources":["../../../src/main/codegen-utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,SAAS,MAAM,sBAAsB,CAAC;AAElD,OAAO,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAE7D,MAAM,UAAU,sBAAsB;IACpC,OAAO,CAAC,WAAW,EAAE,WAAW,EAAE,kBAAkB,EAAE,UAAU,EAAE,WAAW,EAAE,GAAG,uBAAuB,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;AACtH,CAAC;AAED,MAAM,UAAU,mBAAmB;IACjC,MAAM,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC;IACtC,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,yEAAyE,CAAC,CAAC;IAC7F,CAAC;IAED,qDAAqD;IACrD,MAAM,iBAAiB,GAAG,sBAAsB,EAAE,CAAC;IACnD,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,SAA0B,CAAC,EAAE,CAAC;QAC5D,MAAM,IAAI,KAAK,CAAC,2BAA2B,SAAS,6BAA6B,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACnH,CAAC;IAED,OAAO,SAA0B,CAAC;AACpC,CAAC"}

package/dist/src/main/codegen.d.ts

@@ -0,0 +1,4 @@
+import { CodegenOptions } from './codegen-types.js';
+/** Executes codegen */
+export declare function runCodegen(isDev?: boolean): Promise<void>;
+export declare function runCodegenIteration(options: CodegenOptions, abortSignal?: AbortSignal, waitIfPaused?: () => Promise<void>): Promise<void>;

package/dist/src/main/codegen.js

@@ -0,0 +1,201 @@
+import { exec } from 'child_process';
+import util from 'util';
+import * as cliParams from '../cli/cli-params.js';
+import { validateCliParams } from '../cli/validate-cli-params.js';
+import { generateContent as generateContentVertexAi } from '../ai-service/vertex-ai.js';
+import { generateContent as generateContentGPT } from '../ai-service/chat-gpt.js';
+import { generateContent as generateContentAnthropic } from '../ai-service/anthropic.js';
+import { generateContent as generateContentVertexAiClaude } from '../ai-service/vertex-ai-claude.js';
+import { generateContent as generateContentAiStudio } from '../ai-service/ai-studio.js';
+import { generateImage as generateImageDallE } from '../ai-service/dall-e.js';
+import { generateImage as generateImageVertexAi } from '../ai-service/vertex-ai-imagen.js';
+import { promptService } from '../prompt/prompt-service.js';
+import { updateFiles } from '../files/update-files.js';
+import { rcConfig } from '../main/config.js';
+import { getLintFixPrompt } from '../prompt/prompt-codegen.js';
+import { printHelpMessage } from '../cli/cli-options.js';
+import { getCodeGenPrompt } from '../prompt/prompt-codegen.js';
+import { runInteractiveMode } from './interactive/codegen-interactive.js';
+import { runCodegenUI } from './ui/codegen-ui.js';
+import { putSystemMessage, putUserMessage, setCurrentIterationId, unsetCurrentIterationId, } from './common/content-bus.js';
+import { refreshFiles } from '../files/find-files.js';
+import { getRegisteredAiServices } from './plugin-loader.js';
+import { cliParamToAiService } from './codegen-utils.js';
+/** Executes codegen */
+export async function runCodegen(isDev = false) {
+    // Print to console the received parameters
+    console.log(`Received parameters: ${process.argv.slice(2).join(' ')}`);
+    validateCliParams();
+    // Handle --help option
+    if (cliParams.helpRequested) {
+        printHelpMessage();
+        return;
+    }
+    const options = {
+        explicitPrompt: cliParams.explicitPrompt,
+        taskFile: cliParams.taskFile,
+        considerAllFiles: cliParams.considerAllFiles,
+        dependencyTree: cliParams.dependencyTree,
+        allowFileCreate: cliParams.allowFileCreate,
+        allowFileDelete: cliParams.allowFileDelete,
+        allowDirectoryCreate: cliParams.allowDirectoryCreate,
+        allowFileMove: cliParams.allowFileMove,
+        aiService: cliParamToAiService(),
+        vision: cliParams.vision,
+        imagen: cliParams.imagen,
+        disableContextOptimization: cliParams.disableContextOptimization,
+        temperature: cliParams.temperature,
+        cheap: cliParams.cheap,
+        dryRun: cliParams.dryRun,
+        verbose: cliParams.verbosePrompt,
+        requireExplanations: !cliParams.disableExplanations,
+        geminiBlockNone: cliParams.geminiBlockNone,
+        disableInitialLint: cliParams.disableInitialLint,
+        contentMask: cliParams.contentMask,
+        ignorePatterns: cliParams.ignorePatterns,
+        askQuestion: cliParams.askQuestion,
+        disableCache: cliParams.disableCache,
+        interactive: cliParams.interactive,
+        ui: cliParams.ui,
+        uiPort: cliParams.uiPort,
+        disableAiServiceFallback: cliParams.disableAiServiceFallback,
+        historyEnabled: !cliParams.disableHistory,
+        selfReflectionEnabled: !cliParams.disableSelfReflection,
+        conversationSummaryEnabled: !cliParams.disableConversationSummary,
+        isDev,
+    };
+    if (cliParams.ui) {
+        await runCodegenUI(options);
+    }
+    else if (cliParams.interactive) {
+        // Handle interactive mode
+        await runInteractiveMode(options);
+    }
+    else {
+        console.log('Executing codegen in non-interactive mode');
+        await runCodegenIteration(options);
+    }
+}
+export async function runCodegenIteration(options, abortSignal, waitIfPaused = () => Promise.resolve()) {
+    refreshFiles();
+    setCurrentIterationId();
+    putUserMessage(options.explicitPrompt ?? options.taskFile ?? 'Run codegen iteration without explicit prompt.');
+    if (rcConfig.lintCommand && !options.disableInitialLint) {
+        try {
+            putSystemMessage(`Executing lint command: ${rcConfig.lintCommand}`);
+            await execPromise(rcConfig.lintCommand, { cwd: rcConfig.rootDir });
+            putSystemMessage('Lint command executed successfully');
+        }
+        catch (error) {
+            const { stderr, stdout } = error;
+            putSystemMessage('Lint command failed. Aborting codegen, please fix lint issues before running codegen, or use --disable-initial-lint');
+            console.log('Lint errors:', stdout, stderr);
+            unsetCurrentIterationId();
+            return;
+        }
+    }
+    else if (rcConfig.lintCommand && options.disableInitialLint) {
+        console.log('Initial lint was skipped.');
+    }
+    if (abortSignal?.aborted) {
+        unsetCurrentIterationId();
+        throw new Error('Codegen iteration aborted');
+    }
+    await waitIfPaused();
+    putSystemMessage('Generating response');
+    try {
+        const functionCalls = await promptService(getGenerateContentFunctions(), GENERATE_IMAGE_FNS, getCodeGenPrompt(options), waitIfPaused);
+        console.log('Received function calls:', functionCalls);
+        await waitIfPaused();
+        if (functionCalls.length === 0) {
+            putSystemMessage('No updates to apply');
+        }
+        else if (options.dryRun) {
+            putSystemMessage('Dry run mode, not updating files');
+        }
+        else {
+            putSystemMessage('Update files');
+            await updateFiles(functionCalls.filter((call) => call.name !== 'explanation' && call.name !== 'getSourceCode'), options);
+            putSystemMessage('Initial updates applied');
+            if (abortSignal?.aborted) {
+                throw new Error('Codegen iteration aborted after initial updates');
+            }
+            await waitIfPaused();
+            // Check if lintCommand is specified in .genaicoderc
+            if (rcConfig.lintCommand && functionCalls.length > 0) {
+                try {
+                    putSystemMessage(`Executing lint command: ${rcConfig.lintCommand}`);
+                    await execPromise(rcConfig.lintCommand, { cwd: rcConfig.rootDir });
+                    putSystemMessage('Lint command executed successfully');
+                }
+                catch (error) {
+                    putSystemMessage('Lint command failed. Attempting to fix issues...');
+                    // Prepare the lint error output for the second pass
+                    const firstLintError = error;
+                    const lintErrorPrompt = getLintFixPrompt(rcConfig.lintCommand, options, firstLintError.stdout, firstLintError.stderr);
+                    putSystemMessage('Generating response for lint fixes');
+                    const lintFixFunctionCalls = (await promptService(getGenerateContentFunctions(), GENERATE_IMAGE_FNS, {
+                        prompt: lintErrorPrompt,
+                        options: { ...options, considerAllFiles: true },
+                    }, waitIfPaused));
+                    console.log('Received function calls for lint fixes:', lintFixFunctionCalls);
+                    await waitIfPaused();
+                    putSystemMessage('Applying lint fixes');
+                    updateFiles(lintFixFunctionCalls.filter((call) => call.name !== 'explanation' && call.name !== 'getSourceCode'), options);
+                    if (abortSignal?.aborted) {
+                        throw new Error('Codegen iteration aborted after lint fixes');
+                    }
+                    // Run lint command again to verify fixes
+                    try {
+                        putSystemMessage(`Re-running lint command: ${rcConfig.lintCommand}`);
+                        await execPromise(rcConfig.lintCommand);
+                        putSystemMessage('Lint command executed successfully after fixes');
+                    }
+                    catch (secondLintError) {
+                        const error = secondLintError;
+                        putSystemMessage('Lint command still failing after fixes. Manual intervention may be required.');
+                        console.log('Lint errors:', error.stdout, error.stderr);
+                    }
+                }
+            }
+            console.log('Done!');
+        }
+    }
+    catch (error) {
+        if (error instanceof Error) {
+            if (error.name === 'AbortError' || error.message.includes('interrupted')) {
+                putSystemMessage('Codegen iteration was interrupted');
+            }
+            else if (error.message.includes('Rate limit exceeded')) {
+                putSystemMessage('Rate limit exceeded. Consider switching to a different AI service or waiting before retrying.');
+            }
+            else {
+                putSystemMessage(`An error occurred during codegen: ${error.message}`);
+            }
+        }
+        else {
+            putSystemMessage('An unknown error occurred during codegen');
+        }
+        console.error('Error details:', error);
+    }
+    finally {
+        unsetCurrentIterationId();
+    }
+}
+// helper functions and consts
+const execPromise = util.promisify(exec);
+function getGenerateContentFunctions() {
+    return {
+        'vertex-ai-claude': generateContentVertexAiClaude,
+        'vertex-ai': generateContentVertexAi,
+        'ai-studio': generateContentAiStudio,
+        anthropic: generateContentAnthropic,
+        'chat-gpt': generateContentGPT,
+        ...Object.fromEntries(getRegisteredAiServices().entries()),
+    };
+}
+const GENERATE_IMAGE_FNS = {
+    'dall-e': generateImageDallE,
+    'vertex-ai': generateImageVertexAi,
+};
+//# sourceMappingURL=codegen.js.map