genaicode 0.0.37 → 0.2.0

package/dist/cli/validate-cli-params.test.js

@@ -0,0 +1,106 @@
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { validateCliParams, getCliParamValue, hasCliParam } from './validate-cli-params.js';
+describe('validateCliParams', () => {
+    const originalArgv = process.argv;
+    beforeEach(() => {
+        process.argv = ['node', 'script.js'];
+    });
+    afterEach(() => {
+        process.argv = originalArgv;
+    });
+    it('should not throw for valid parameters', () => {
+        process.argv.push('--dry-run', '--consider-all-files');
+        expect(() => validateCliParams()).not.toThrow();
+    });
+    it('should throw for invalid parameters', () => {
+        process.argv.push('--invalid-param');
+        expect(() => validateCliParams()).toThrow();
+    });
+    it('should throw for parameters without --', () => {
+        process.argv.push('invalid-param');
+        expect(() => validateCliParams()).toThrow();
+    });
+    it('should not throw for valid --temperature parameter', () => {
+        process.argv.push('--temperature=0.5');
+        expect(() => validateCliParams()).not.toThrow();
+    });
+    it('should throw for invalid --temperature parameter', () => {
+        process.argv.push('--temperature=invalid');
+        expect(() => validateCliParams()).toThrow();
+    });
+    it('should throw an error when both vision and vertexAi flags are true', () => {
+        process.argv.push('--vision');
+        process.argv.push('--vertex-ai');
+        expect(() => validateCliParams()).toThrow('--vision and --vertex-ai are currently not supported together.');
+    });
+    // New tests for --imagen parameter
+    it('should not throw for valid --imagen parameter with vertex-ai', () => {
+        process.argv.push('--imagen=vertex-ai');
+        expect(() => validateCliParams()).not.toThrow();
+    });
+    it('should not throw for valid --imagen parameter with dall-e', () => {
+        process.argv.push('--imagen=dall-e');
+        expect(() => validateCliParams()).not.toThrow();
+    });
+    it('should throw for invalid --imagen parameter value', () => {
+        process.argv.push('--imagen=invalid-service');
+        expect(() => validateCliParams()).toThrow('Invalid --imagen value. It must be either "vertex-ai" or "dall-e".');
+    });
+    it('should throw for --imagen parameter without value', () => {
+        process.argv.push('--imagen=');
+        expect(() => validateCliParams()).toThrow('Invalid --imagen value. It must be either "vertex-ai" or "dall-e".');
+    });
+});
+describe('getCliParamValue', () => {
+    const originalArgv = process.argv;
+    beforeEach(() => {
+        process.argv = ['node', 'script.js'];
+    });
+    afterEach(() => {
+        process.argv = originalArgv;
+    });
+    it('should return the value for a parameter with a value', () => {
+        process.argv.push('--explicit-prompt=Test prompt');
+        expect(getCliParamValue('--explicit-prompt')).toBe('Test prompt');
+    });
+    it('should return null for a parameter without a value', () => {
+        process.argv.push('--dry-run');
+        expect(getCliParamValue('--dry-run')).toBeNull();
+    });
+    it('should return null for a non-existent parameter', () => {
+        expect(getCliParamValue('--non-existent')).toBeNull();
+    });
+    it('should return the value for --temperature parameter', () => {
+        process.argv.push('--temperature=0.5');
+        expect(getCliParamValue('--temperature')).toBe('0.5');
+    });
+    it('should return the value for --imagen parameter', () => {
+        process.argv.push('--imagen=vertex-ai');
+        expect(getCliParamValue('--imagen')).toBe('vertex-ai');
+    });
+});
+describe('hasCliParam', () => {
+    const originalArgv = process.argv;
+    beforeEach(() => {
+        process.argv = ['node', 'script.js'];
+    });
+    afterEach(() => {
+        process.argv = originalArgv;
+    });
+    it('should return true for an existing parameter', () => {
+        process.argv.push('--dry-run');
+        expect(hasCliParam('--dry-run')).toBe(true);
+    });
+    it('should return false for a non-existent parameter', () => {
+        expect(hasCliParam('--non-existent')).toBe(false);
+    });
+    it('should return true for --temperature parameter', () => {
+        process.argv.push('--temperature=0.5');
+        expect(hasCliParam('--temperature')).toBe(true);
+    });
+    it('should return true for --imagen parameter', () => {
+        process.argv.push('--imagen=dall-e');
+        expect(hasCliParam('--imagen')).toBe(true);
+    });
+});
+//# sourceMappingURL=validate-cli-params.test.js.map

package/dist/cli/validate-cli-params.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"validate-cli-params.test.js","sourceRoot":"","sources":["../../src/cli/validate-cli-params.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAE5F,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,MAAM,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAElC,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,CAAC,IAAI,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,CAAC,IAAI,GAAG,YAAY,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,sBAAsB,CAAC,CAAC;QACvD,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACvC,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;QAC5E,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC9B,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAEjC,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC,OAAO,CAAC,gEAAgE,CAAC,CAAC;IAC9G,CAAC,CAAC,CAAC;IAEH,mCAAmC;IACnC,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACxC,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QAC9C,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC,OAAO,CAAC,oEAAoE,CAAC,CAAC;IAClH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC/B,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC,OAAO,CAAC,oEAAoE,CAAC,CAAC;IAClH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,MAAM,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAElC,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,CAAC,IAAI,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,CAAC,IAAI,GAAG,YAAY,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;QACnD,MAAM,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC/B,MAAM,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACvC,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACxC,MAAM,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,MAAM,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAElC,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,CAAC,IAAI,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,CAAC,IAAI,GAAG,YAAY,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC/B,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,CAAC,WAAW,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACvC,MAAM,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACrC,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/files/file-utils.d.ts

@@ -0,0 +1,2 @@
+/** Check if directory is ancestor of given directory */
+export declare function isAncestorDirectory(parent: string, dir: string): boolean;

package/dist/files/file-utils.js

@@ -0,0 +1,7 @@
+import path from 'path';
+/** Check if directory is ancestor of given directory */
+export function isAncestorDirectory(parent, dir) {
+    const relative = path.relative(parent, dir);
+    return parent === dir || (relative ? !relative.startsWith('..') && !path.isAbsolute(relative) : false);
+}
+//# sourceMappingURL=file-utils.js.map

package/dist/files/file-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"file-utils.js","sourceRoot":"","sources":["../../src/files/file-utils.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,MAAM,CAAC;AAExB,wDAAwD;AACxD,MAAM,UAAU,mBAAmB,CAAC,MAAc,EAAE,GAAW;IAC7D,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC5C,OAAO,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;AACzG,CAAC"}

package/dist/files/file-utils.test.d.ts

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

package/dist/files/file-utils.test.js

@@ -0,0 +1,18 @@
+import { vi, describe, beforeEach, it, expect } from 'vitest';
+import { isAncestorDirectory } from './file-utils.js';
+// Test for isAncestorDirectory
+describe('isAncestorDirectory', () => {
+    beforeEach(() => {
+        vi.resetAllMocks();
+    });
+    it('should return true if directories are the same', () => {
+        expect(isAncestorDirectory('/project', '/project')).toBe(true);
+    });
+    it('should return true if parent is ancestor of dir', () => {
+        expect(isAncestorDirectory('/project', '/project/src')).toBe(true);
+    });
+    it('should return false if parent is not ancestor of dir', () => {
+        expect(isAncestorDirectory('/project', '/other')).toBe(false);
+    });
+});
+//# sourceMappingURL=file-utils.test.js.map

package/dist/files/file-utils.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"file-utils.test.js","sourceRoot":"","sources":["../../src/files/file-utils.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAEtD,+BAA+B;AAC/B,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,CAAC,mBAAmB,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,CAAC,mBAAmB,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,CAAC,mBAAmB,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/files/find-files.d.ts

@@ -0,0 +1,8 @@
+type DependencyList = string[];
+/** Generates a dependency list for given file */
+export declare function getDependencyList(entryFile: string): DependencyList;
+/** Get source files of the application */
+export declare function getSourceFiles(): string[];
+/** Get source files of the application */
+export declare function getImageAssetFiles(): string[];
+export {};

package/dist/files/find-files.js

@@ -0,0 +1,76 @@
+import fs from 'fs';
+import path from 'path';
+import { rcConfig, sourceExtensions, IMAGE_ASSET_EXTENSIONS, ignorePaths } from '../main/config.js';
+// List of possible extensions for dependency resolution
+const POSSIBLE_DEPENDENCY_EXTENSIONS = ['.ts', '.js', '.tsx', '.jsx'];
+function findFiles(dir, recursive, extensions) {
+    const files = [];
+    const items = fs.readdirSync(dir);
+    for (const item of items) {
+        const fullPath = path.join(dir, item);
+        if (ignorePaths.some((ignorePath) => fullPath.endsWith(ignorePath))) {
+            continue;
+        }
+        if (fs.statSync(fullPath).isDirectory()) {
+            if (recursive) {
+                files.push(...findFiles(fullPath, true, extensions));
+            }
+        }
+        else if (extensions.includes(path.extname(fullPath))) {
+            files.push(fullPath);
+        }
+    }
+    return files;
+}
+function getDependencies(filePath) {
+    const content = fs.readFileSync(filePath, 'utf-8');
+    const dependencyRegex = /import\s+.+?\s+from\s+['"](.+?\/?[^'"]+)['"]/g;
+    const dependencies = [];
+    let match;
+    while ((match = dependencyRegex.exec(content)) !== null) {
+        const dependencyPath = match[1];
+        // Resolve relative paths from the file's directory
+        const resolvedPath = path.resolve(path.dirname(filePath), dependencyPath);
+        // Only add the dependency if it's a local file and not a module
+        if (fs.existsSync(resolvedPath)) {
+            dependencies.push(resolvedPath);
+        }
+        else {
+            const possibleExtensions = POSSIBLE_DEPENDENCY_EXTENSIONS;
+            for (const ext of possibleExtensions) {
+                const extendedPath = resolvedPath + ext;
+                if (fs.existsSync(extendedPath)) {
+                    dependencies.push(extendedPath);
+                }
+            }
+        }
+    }
+    return dependencies;
+}
+/** Generates a dependency list for given file */
+export function getDependencyList(entryFile) {
+    const visitedFiles = new Set();
+    const result = new Set();
+    function traverse(file) {
+        if (visitedFiles.has(file))
+            return;
+        visitedFiles.add(file);
+        const dependencies = getDependencies(file);
+        dependencies.forEach((dependency) => result.add(dependency));
+        dependencies.forEach(traverse);
+    }
+    result.add(path.resolve(entryFile));
+    traverse(entryFile);
+    return Array.from(result);
+}
+const sourceFiles = findFiles(rcConfig.rootDir, true, sourceExtensions);
+/** Get source files of the application */
+export function getSourceFiles() {
+    return [...sourceFiles];
+}
+const imageAssetFiles = findFiles(rcConfig.rootDir, true, IMAGE_ASSET_EXTENSIONS);
+/** Get source files of the application */
+export function getImageAssetFiles() {
+    return [...imageAssetFiles];
+}
+//# sourceMappingURL=find-files.js.map

package/dist/files/find-files.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"find-files.js","sourceRoot":"","sources":["../../src/files/find-files.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAEpG,wDAAwD;AACxD,MAAM,8BAA8B,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;AAMtE,SAAS,SAAS,CAAC,GAAW,EAAE,SAAkB,EAAE,UAA0B;IAC5E,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IAClC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAEtC,IAAI,WAAW,CAAC,IAAI,CAAC,CAAC,UAAkB,EAAE,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC;YAC5E,SAAS;QACX,CAAC;QAED,IAAI,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC;YACxC,IAAI,SAAS,EAAE,CAAC;gBACd,KAAK,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;aAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC;YACvD,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,eAAe,CAAC,QAAgB;IACvC,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACnD,MAAM,eAAe,GAAG,+CAA+C,CAAC;IACxE,MAAM,YAAY,GAAmB,EAAE,CAAC;IACxC,IAAI,KAAK,CAAC;IACV,OAAO,CAAC,KAAK,GAAG,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACxD,MAAM,cAAc,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAChC,mDAAmD;QACnD,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,cAAc,CAAC,CAAC;QAE1E,gEAAgE;QAChE,IAAI,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;YAChC,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAClC,CAAC;aAAM,CAAC;YACN,MAAM,kBAAkB,GAAG,8BAA8B,CAAC;YAC1D,KAAK,MAAM,GAAG,IAAI,kBAAkB,EAAE,CAAC;gBACrC,MAAM,YAAY,GAAG,YAAY,GAAG,GAAG,CAAC;gBACxC,IAAI,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;oBAChC,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBAClC,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,iDAAiD;AACjD,MAAM,UAAU,iBAAiB,CAAC,SAAiB;IACjD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAU,CAAC;IACvC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC;IAEjC,SAAS,QAAQ,CAAC,IAAY;QAC5B,IAAI,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC;YAAE,OAAO;QACnC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACvB,MAAM,YAAY,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;QAC3C,YAAY,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;QAC7D,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACjC,CAAC;IAED,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;IACpC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAEpB,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAC5B,CAAC;AAED,MAAM,WAAW,GAAG,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,EAAE,gBAAgB,CAAC,CAAC;AAExE,0CAA0C;AAC1C,MAAM,UAAU,cAAc;IAC5B,OAAO,CAAC,GAAG,WAAW,CAAC,CAAC;AAC1B,CAAC;AAED,MAAM,eAAe,GAAG,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,EAAE,sBAAsB,CAAC,CAAC;AAElF,0CAA0C;AAC1C,MAAM,UAAU,kBAAkB;IAChC,OAAO,CAAC,GAAG,eAAe,CAAC,CAAC;AAC9B,CAAC"}

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

@@ -0,0 +1,18 @@
+import { CodegenOptions } from '../main/codegen-types.js';
+type SourceCodeMap = Record<string, {
+    content: string | null;
+} | undefined>;
+type ImageAssetsMap = Record<string, {
+    mimeType: string | false;
+    width: number;
+    height: number;
+}>;
+/** Print source code of all source files */
+export declare function getSourceCode({ filterPaths, taskFile, forceAll, }: {
+    filterPaths?: string[];
+    taskFile: string | undefined;
+    forceAll?: boolean;
+}, options: CodegenOptions): SourceCodeMap;
+/** Get image asset files summary */
+export declare function getImageAssets(): ImageAssetsMap;
+export {};

package/dist/files/read-files.js

@@ -0,0 +1,66 @@
+import fs from 'fs';
+import mime from 'mime-types';
+import sizeOf from 'image-size';
+import path from 'path';
+import globRegex from 'glob-regex';
+import { getSourceFiles, getImageAssetFiles } from './find-files.js';
+import { rcConfig, importantContext } from '../main/config.js';
+import { verifySourceCodeLimit } from '../prompt/limits.js';
+/**
+ * Read contents of source files and create a map with file path as key and file content as value
+ */
+function readSourceFiles({ contentMask, ignorePatterns }, filterPaths, forceAll = false) {
+    const sourceCode = {};
+    const importantFiles = new Set(importantContext.files || []);
+    for (const file of getSourceFiles()) {
+        if (!filterPaths || filterPaths.includes(file) || importantFiles.has(file)) {
+            // Always include important files
+            if (importantFiles.has(file)) {
+                sourceCode[file] = { content: fs.readFileSync(file, 'utf-8') };
+                continue;
+            }
+            // Apply content mask filter if it's set
+            if (!filterPaths && contentMask && !forceAll) {
+                const relativePath = path.relative(rcConfig.rootDir, file);
+                if (!relativePath.startsWith(contentMask)) {
+                    sourceCode[file] = { content: null }; // Include the file path but set content to null
+                    continue;
+                }
+            }
+            if (!forceAll && ignorePatterns?.some((pattern) => globRegex.default(pattern).test(file))) {
+                sourceCode[file] = { content: null };
+            }
+            else {
+                const content = fs.readFileSync(file, 'utf-8');
+                sourceCode[file] = { content };
+            }
+        }
+    }
+    return sourceCode;
+}
+/** Print source code of all source files */
+export function getSourceCode({ filterPaths, taskFile, forceAll, }, options) {
+    const sourceCode = readSourceFiles(options, filterPaths, forceAll);
+    if (taskFile && !sourceCode[taskFile]) {
+        sourceCode[taskFile] = {
+            content: fs.readFileSync(taskFile, 'utf-8'),
+        };
+    }
+    verifySourceCodeLimit(JSON.stringify(sourceCode));
+    return sourceCode;
+}
+/** Get image asset files summary */
+export function getImageAssets() {
+    const imageAssets = {};
+    for (const file of getImageAssetFiles()) {
+        const dimensions = sizeOf.default(file);
+        imageAssets[file] = {
+            mimeType: mime.lookup(file),
+            width: dimensions.width,
+            height: dimensions.height,
+        };
+    }
+    verifySourceCodeLimit(JSON.stringify(imageAssets));
+    return imageAssets;
+}
+//# sourceMappingURL=read-files.js.map

package/dist/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;AAa5D;;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,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,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,gDAAgD;oBACtF,SAAS;gBACX,CAAC;YACH,CAAC;YACD,IAAI,CAAC,QAAQ,IAAI,cAAc,EAAE,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;gBAC1F,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACvC,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,CAAC;YACjC,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,OAAO,CAAC,IAAI,CAAC,CAAC;QACxC,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/files/read-files.test.d.ts

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

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

@@ -0,0 +1,88 @@
+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,
+    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).default.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/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,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,OAAO,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;YAElF,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/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/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/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/files/update-files.d.ts

@@ -0,0 +1,6 @@
+import { FunctionCall } from '../ai-service/common.js';
+import { CodegenOptions } from '../main/codegen-types.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>;