genaicode 0.0.37 → 0.2.0

package/dist/files/update-files.js

@@ -0,0 +1,197 @@
+import fs from 'fs';
+import path from 'path';
+import assert from 'node:assert';
+import * as diff from 'diff';
+import { getSourceFiles } from './find-files.js';
+import { isAncestorDirectory } from './file-utils.js';
+import { rcConfig } from '../main/config.js';
+import { getTempBuffer } from './temp-buffer.js';
+import { imglyRemoveBackground } from '../images/imgly-remove-background.js';
+import { splitImage } from '../images/split-image.js';
+import { resizeImageFile } from '../images/resize-image.js';
+/**
+ * @param functionCalls Result of the code generation, a map of file paths to new content
+ */
+export async function updateFiles(functionCalls, options) {
+    const { allowDirectoryCreate, allowFileCreate, allowFileDelete, allowFileMove } = options;
+    for (const { name, args } of functionCalls) {
+        let { filePath, newContent, source, destination, inputFilePath, outputFilePath } = args;
+        const { patch, parts, size } = args;
+        // Check if filePath is absolute, if not use rootDir as baseline
+        if (name !== 'moveFile' && name !== 'imglyRemoveBackground' && name !== 'splitImage') {
+            assert(filePath);
+            filePath = path.isAbsolute(filePath) ? filePath : path.join(rcConfig.rootDir, filePath);
+        }
+        else if (name === 'moveFile') {
+            assert(source);
+            source = path.isAbsolute(source) ? source : path.join(rcConfig.rootDir, source);
+            assert(destination);
+            destination = path.isAbsolute(destination) ? destination : path.join(rcConfig.rootDir, destination);
+        }
+        else if (name === 'imglyRemoveBackground' || name === 'splitImage') {
+            assert(inputFilePath);
+            inputFilePath = path.isAbsolute(inputFilePath) ? inputFilePath : path.join(rcConfig.rootDir, inputFilePath);
+            if (name === 'imglyRemoveBackground') {
+                assert(outputFilePath);
+                outputFilePath = path.isAbsolute(outputFilePath) ? outputFilePath : path.join(rcConfig.rootDir, outputFilePath);
+            }
+            else if (name === 'splitImage') {
+                parts?.forEach((part) => (part.outputFilePath = part.outputFilePath
+                    ? path.isAbsolute(part.outputFilePath)
+                        ? part.outputFilePath
+                        : path.join(rcConfig.rootDir, part.outputFilePath)
+                    : undefined));
+            }
+        }
+        // ignore files which are not located inside project directory (sourceFiles)
+        if ((name !== 'moveFile' && name !== 'imglyRemoveBackground' && name !== 'splitImage' && !isProjectPath(filePath)) ||
+            (name === 'moveFile' && (!isProjectPath(source) || !isProjectPath(destination))) ||
+            (name === 'imglyRemoveBackground' && (!isProjectPath(inputFilePath) || !isProjectPath(outputFilePath))) ||
+            (name === 'splitImage' &&
+                (!isProjectPath(inputFilePath) ||
+                    parts.some((part) => part.outputFilePath && !isProjectPath(part.outputFilePath))))) {
+            console.log(`Skipping file: ${filePath || source || inputFilePath}`);
+            throw new Error(`File ${filePath || source || inputFilePath} is not located inside project directory, something is wrong?`);
+        }
+        if (name === 'deleteFile') {
+            assert(filePath);
+            assert(allowFileDelete, 'File delete option was not enabled');
+            console.log(`Removing file: ${filePath}`);
+            fs.unlinkSync(filePath);
+        }
+        else if (name === 'createDirectory') {
+            assert(filePath);
+            assert(allowDirectoryCreate, 'Directory create option was not enabled');
+            console.log(`Creating directory: ${filePath}`);
+            fs.mkdirSync(filePath, { recursive: true });
+        }
+        else if (name === 'updateFile' || name === 'createFile' || name === 'patchFile') {
+            assert(filePath);
+            if (name === 'patchFile') {
+                assert(patch);
+                console.log(`Applying a patch: ${filePath} content`);
+                newContent = diff.applyPatch(fs.readFileSync(filePath, 'utf-8'), patch) || '';
+                assert(!!newContent, 'Patch was not successful');
+            }
+            assert(!!newContent, 'newContent must not be empty');
+            if (name === 'createFile') {
+                console.log(`Creating file: ${filePath}`);
+                assert(allowFileCreate, 'File create option was not enabled');
+                assert(!fs.existsSync(filePath), 'File already exists');
+                if (allowDirectoryCreate) {
+                    fs.mkdirSync(path.dirname(filePath), { recursive: true });
+                }
+            }
+            else {
+                console.log(`Updating file: ${filePath}`);
+                assert(fs.existsSync(filePath), 'File does not exist');
+            }
+            fs.writeFileSync(filePath, options.aiService === 'vertex-ai' || options.aiService === 'ai-studio'
+                ? // Fixing a problem caused by vertex function calling. Possibly not a good fix
+                    newContent.replace(/\\n/g, '\n').replace(/\\'/g, "'")
+                : newContent, 'utf-8');
+        }
+        else if (name === 'moveFile') {
+            console.log(`Moving file from ${source} to ${destination}`);
+            assert(source);
+            assert(destination);
+            assert(fs.existsSync(source), 'Source file does not exist');
+            assert(!fs.existsSync(destination), 'Destination file already exists');
+            assert(allowFileMove, 'File move option was not enabled');
+            if (allowDirectoryCreate) {
+                fs.mkdirSync(path.dirname(destination), { recursive: true });
+            }
+            fs.renameSync(source, destination);
+        }
+        else if (name === 'downloadFile') {
+            console.log(`Downloading image: ${filePath}`);
+            assert(filePath);
+            assert(fs.existsSync(filePath) || allowFileCreate, 'File create option was not enabled and file does not exist');
+            if (allowDirectoryCreate) {
+                fs.mkdirSync(path.dirname(filePath), { recursive: true });
+            }
+            try {
+                const downloadUrl = args?.downloadUrl;
+                assert(downloadUrl, 'image url is not empty');
+                if (downloadUrl.startsWith('temp://')) {
+                    assert(getTempBuffer(downloadUrl), 'Temp buffer not present but expected');
+                    fs.writeFileSync(filePath, getTempBuffer(downloadUrl));
+                }
+                else {
+                    const imageResponse = await fetch(downloadUrl);
+                    const arrayBuffer = await imageResponse.arrayBuffer();
+                    const buffer = Buffer.from(arrayBuffer);
+                    fs.writeFileSync(filePath, buffer);
+                }
+                console.log(`Image download and saved to: ${filePath}`);
+            }
+            catch (error) {
+                console.error('Failed to download image:', error);
+                throw error;
+            }
+        }
+        else if (name === 'imglyRemoveBackground') {
+            console.log(`Removing background from image: ${inputFilePath}`);
+            assert(inputFilePath);
+            assert(outputFilePath);
+            assert(fs.existsSync(inputFilePath), 'Input file does not exist');
+            assert(allowFileCreate || fs.existsSync(outputFilePath), 'File create option was not enabled and output file does not exist');
+            if (allowDirectoryCreate) {
+                fs.mkdirSync(path.dirname(outputFilePath), { recursive: true });
+            }
+            try {
+                await imglyRemoveBackground(inputFilePath, outputFilePath);
+                console.log(`Background removed and image saved to: ${outputFilePath}`);
+            }
+            catch (error) {
+                console.error('Failed to remove background', error);
+                throw error;
+            }
+        }
+        else if (name === 'splitImage') {
+            console.log(`Splitting image: ${inputFilePath}`, parts);
+            assert(inputFilePath);
+            assert(fs.existsSync(inputFilePath), 'Input file does not exist');
+            assert(Array.isArray(parts) && parts.length > 0, 'Parts array must not be empty');
+            for (const part of parts) {
+                assert(part.outputFilePath);
+                assert(allowFileCreate || fs.existsSync(part.outputFilePath), 'File create option was not enabled and output file does not exist');
+                if (allowDirectoryCreate) {
+                    fs.mkdirSync(path.dirname(part.outputFilePath), { recursive: true });
+                }
+            }
+            try {
+                await splitImage(inputFilePath, parts.map((part) => ({ outputFilePath: part.outputFilePath, rect: part.rect })));
+                console.log(`Image split successfully`);
+            }
+            catch (error) {
+                console.error('Failed to split image', error);
+                throw error;
+            }
+        }
+        else if (name === 'resizeImage') {
+            console.log(`Resizing image: ${filePath}`, size);
+            assert(filePath);
+            assert(size);
+            assert(size.width);
+            assert(size.height);
+            assert(fs.existsSync(filePath), 'Input file does not exist');
+            try {
+                await resizeImageFile(filePath, { width: size.width, height: size.height });
+                console.log(`Image resized successfully`);
+            }
+            catch (error) {
+                console.error('Failed to resize image', error);
+                throw error;
+            }
+        }
+    }
+}
+function isProjectPath(filePath) {
+    const sourceFiles = getSourceFiles();
+    return (isAncestorDirectory(rcConfig.rootDir, filePath) ||
+        sourceFiles.includes(filePath) ||
+        !sourceFiles.some((sourceFile) => path.dirname(filePath) === path.dirname(sourceFile) ||
+            isAncestorDirectory(path.dirname(sourceFile), path.dirname(filePath))));
+}
+//# sourceMappingURL=update-files.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"update-files.js","sourceRoot":"","sources":["../../src/files/update-files.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAE7B,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,EAAE,qBAAqB,EAAE,MAAM,sCAAsC,CAAC;AAC7E,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,2BAA2B,CAAC;AAI5D;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,aAA6B,EAAE,OAAuB;IACtF,MAAM,EAAE,oBAAoB,EAAE,eAAe,EAAE,eAAe,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC;IAE1F,KAAK,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,aAAa,EAAE,CAAC;QAC3C,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,cAAc,EAAE,GAAG,IAOlF,CAAC;QACF,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,IAI9B,CAAC;QAEF,gEAAgE;QAChE,IAAI,IAAI,KAAK,UAAU,IAAI,IAAI,KAAK,uBAAuB,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;YACrF,MAAM,CAAC,QAAQ,CAAC,CAAC;YACjB,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC1F,CAAC;aAAM,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC;YACf,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAChF,MAAM,CAAC,WAAW,CAAC,CAAC;YACpB,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QACtG,CAAC;aAAM,IAAI,IAAI,KAAK,uBAAuB,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;YACrE,MAAM,CAAC,aAAa,CAAC,CAAC;YACtB,aAAa,GAAG,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;YAC5G,IAAI,IAAI,KAAK,uBAAuB,EAAE,CAAC;gBACrC,MAAM,CAAC,cAAc,CAAC,CAAC;gBACvB,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;YAClH,CAAC;iBAAM,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;gBACjC,KAAK,EAAE,OAAO,CACZ,CAAC,IAAiC,EAAE,EAAE,CACpC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc;oBACxC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,cAAc,CAAC;wBACpC,CAAC,CAAC,IAAI,CAAC,cAAc;wBACrB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,cAAc,CAAC;oBACpD,CAAC,CAAC,SAAS,CAAC,CACjB,CAAC;YACJ,CAAC;QACH,CAAC;QAED,4EAA4E;QAC5E,IACE,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,KAAK,uBAAuB,IAAI,IAAI,KAAK,YAAY,IAAI,CAAC,aAAa,CAAC,QAAS,CAAC,CAAC;YAC/G,CAAC,IAAI,KAAK,UAAU,IAAI,CAAC,CAAC,aAAa,CAAC,MAAO,CAAC,IAAI,CAAC,aAAa,CAAC,WAAY,CAAC,CAAC,CAAC;YAClF,CAAC,IAAI,KAAK,uBAAuB,IAAI,CAAC,CAAC,aAAa,CAAC,aAAc,CAAC,IAAI,CAAC,aAAa,CAAC,cAAe,CAAC,CAAC,CAAC;YACzG,CAAC,IAAI,KAAK,YAAY;gBACpB,CAAC,CAAC,aAAa,CAAC,aAAc,CAAC;oBAC7B,KAAM,CAAC,IAAI,CACT,CAAC,IAAiC,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAClG,CAAC,CAAC,EACP,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,kBAAkB,QAAQ,IAAI,MAAM,IAAI,aAAa,EAAE,CAAC,CAAC;YACrE,MAAM,IAAI,KAAK,CACb,QAAQ,QAAQ,IAAI,MAAM,IAAI,aAAa,+DAA+D,CAC3G,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;YAC1B,MAAM,CAAC,QAAQ,CAAC,CAAC;YACjB,MAAM,CAAC,eAAe,EAAE,oCAAoC,CAAC,CAAC;YAC9D,OAAO,CAAC,GAAG,CAAC,kBAAkB,QAAQ,EAAE,CAAC,CAAC;YAC1C,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAC1B,CAAC;aAAM,IAAI,IAAI,KAAK,iBAAiB,EAAE,CAAC;YACtC,MAAM,CAAC,QAAQ,CAAC,CAAC;YACjB,MAAM,CAAC,oBAAoB,EAAE,yCAAyC,CAAC,CAAC;YACxE,OAAO,CAAC,GAAG,CAAC,uBAAuB,QAAQ,EAAE,CAAC,CAAC;YAC/C,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9C,CAAC;aAAM,IAAI,IAAI,KAAK,YAAY,IAAI,IAAI,KAAK,YAAY,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;YAClF,MAAM,CAAC,QAAQ,CAAC,CAAC;YACjB,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;gBACzB,MAAM,CAAC,KAAK,CAAC,CAAC;gBACd,OAAO,CAAC,GAAG,CAAC,qBAAqB,QAAQ,UAAU,CAAC,CAAC;gBACrD,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;gBAC9E,MAAM,CAAC,CAAC,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC;YACnD,CAAC;YAED,MAAM,CAAC,CAAC,CAAC,UAAU,EAAE,8BAA8B,CAAC,CAAC;YACrD,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC1B,OAAO,CAAC,GAAG,CAAC,kBAAkB,QAAQ,EAAE,CAAC,CAAC;gBAC1C,MAAM,CAAC,eAAe,EAAE,oCAAoC,CAAC,CAAC;gBAC9D,MAAM,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,qBAAqB,CAAC,CAAC;gBACxD,IAAI,oBAAoB,EAAE,CAAC;oBACzB,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC5D,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,kBAAkB,QAAQ,EAAE,CAAC,CAAC;gBAC1C,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,qBAAqB,CAAC,CAAC;YACzD,CAAC;YACD,EAAE,CAAC,aAAa,CACd,QAAQ,EACR,OAAO,CAAC,SAAS,KAAK,WAAW,IAAI,OAAO,CAAC,SAAS,KAAK,WAAW;gBACpE,CAAC,CAAC,8EAA8E;oBAC9E,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC;gBACvD,CAAC,CAAC,UAAU,EACd,OAAO,CACR,CAAC;QACJ,CAAC;aAAM,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;YAC/B,OAAO,CAAC,GAAG,CAAC,oBAAoB,MAAM,OAAO,WAAW,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,CAAC;YACf,MAAM,CAAC,WAAW,CAAC,CAAC;YACpB,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,4BAA4B,CAAC,CAAC;YAC5D,MAAM,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,iCAAiC,CAAC,CAAC;YACvE,MAAM,CAAC,aAAa,EAAE,kCAAkC,CAAC,CAAC;YAC1D,IAAI,oBAAoB,EAAE,CAAC;gBACzB,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC/D,CAAC;YACD,EAAE,CAAC,UAAU,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QACrC,CAAC;aAAM,IAAI,IAAI,KAAK,cAAc,EAAE,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,sBAAsB,QAAQ,EAAE,CAAC,CAAC;YAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC;YACjB,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,eAAe,EAAE,4DAA4D,CAAC,CAAC;YACjH,IAAI,oBAAoB,EAAE,CAAC;gBACzB,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5D,CAAC;YACD,IAAI,CAAC;gBACH,MAAM,WAAW,GAAG,IAAI,EAAE,WAAiC,CAAC;gBAC5D,MAAM,CAAC,WAAW,EAAE,wBAAwB,CAAC,CAAC;gBAC9C,IAAI,WAAW,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;oBACtC,MAAM,CAAC,aAAa,CAAC,WAAW,CAAC,EAAE,sCAAsC,CAAC,CAAC;oBAC3E,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,aAAa,CAAC,WAAW,CAAE,CAAC,CAAC;gBAC1D,CAAC;qBAAM,CAAC;oBACN,MAAM,aAAa,GAAG,MAAM,KAAK,CAAC,WAAW,CAAC,CAAC;oBAC/C,MAAM,WAAW,GAAG,MAAM,aAAa,CAAC,WAAW,EAAE,CAAC;oBACtD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;oBACxC,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;gBACrC,CAAC;gBACD,OAAO,CAAC,GAAG,CAAC,gCAAgC,QAAQ,EAAE,CAAC,CAAC;YAC1D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;gBAClD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;aAAM,IAAI,IAAI,KAAK,uBAAuB,EAAE,CAAC;YAC5C,OAAO,CAAC,GAAG,CAAC,mCAAmC,aAAa,EAAE,CAAC,CAAC;YAChE,MAAM,CAAC,aAAa,CAAC,CAAC;YACtB,MAAM,CAAC,cAAc,CAAC,CAAC;YACvB,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,2BAA2B,CAAC,CAAC;YAClE,MAAM,CACJ,eAAe,IAAI,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAChD,mEAAmE,CACpE,CAAC;YACF,IAAI,oBAAoB,EAAE,CAAC;gBACzB,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAClE,CAAC;YACD,IAAI,CAAC;gBACH,MAAM,qBAAqB,CAAC,aAAa,EAAE,cAAc,CAAC,CAAC;gBAC3D,OAAO,CAAC,GAAG,CAAC,0CAA0C,cAAc,EAAE,CAAC,CAAC;YAC1E,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;gBACpD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;aAAM,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;YACjC,OAAO,CAAC,GAAG,CAAC,oBAAoB,aAAa,EAAE,EAAE,KAAK,CAAC,CAAC;YACxD,MAAM,CAAC,aAAa,CAAC,CAAC;YACtB,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,2BAA2B,CAAC,CAAC;YAClE,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,+BAA+B,CAAC,CAAC;YAClF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;gBAC5B,MAAM,CACJ,eAAe,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,cAAc,CAAC,EACrD,mEAAmE,CACpE,CAAC;gBACF,IAAI,oBAAoB,EAAE,CAAC;oBACzB,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvE,CAAC;YACH,CAAC;YACD,IAAI,CAAC;gBACH,MAAM,UAAU,CACd,aAAa,EACb,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,cAAe,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CACjF,CAAC;gBACF,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;YAC1C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAC;gBAC9C,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;aAAM,IAAI,IAAI,KAAK,aAAa,EAAE,CAAC;YAClC,OAAO,CAAC,GAAG,CAAC,mBAAmB,QAAQ,EAAE,EAAE,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,QAAQ,CAAC,CAAC;YACjB,MAAM,CAAC,IAAI,CAAC,CAAC;YACb,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpB,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,2BAA2B,CAAC,CAAC;YAE7D,IAAI,CAAC;gBACH,MAAM,eAAe,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC5E,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;YAC5C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;gBAC/C,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,aAAa,CAAC,QAAgB;IACrC,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;IAErC,OAAO,CACL,mBAAmB,CAAC,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC;QAC/C,WAAW,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAC9B,CAAC,WAAW,CAAC,IAAI,CACf,CAAC,UAAU,EAAE,EAAE,CACb,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;YACnD,mBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CACxE,CACF,CAAC;AACJ,CAAC"}

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

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

package/dist/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/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/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/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/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/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/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/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/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/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/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/main/codegen-interactive-utils.d.ts

@@ -0,0 +1,8 @@
+import { AiServiceType, CodegenOptions } from './codegen-types.js';
+export declare const displayWelcome: () => void;
+type UserAction = 'text_prompt' | 'task_file' | 'process_comments' | 'select_ai_service' | 'configure' | 'help' | 'exit';
+export declare const getUserAction: () => Promise<UserAction>;
+export declare const getUserOptions: (options: CodegenOptions) => Promise<CodegenOptions>;
+export declare const checkForCodegenComments: () => boolean;
+export declare const selectAiService: (defaultAiService: AiServiceType | undefined) => Promise<AiServiceType>;
+export {};

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

@@ -0,0 +1,148 @@
+import fs from 'fs';
+import { select, checkbox, Separator, input, confirm } from '@inquirer/prompts';
+import { getSourceFiles } from '../files/find-files.js';
+import { CODEGEN_TRIGGER } from '../prompt/prompt-consts.js';
+import { createRequire } from 'module';
+const require = createRequire(import.meta.url);
+const packageJson = require('../../package.json');
+// Function to display the welcome message
+export const displayWelcome = () => {
+    console.log(`Welcome to Genaicode v${packageJson.version}`);
+};
+// Function to get user action
+export const getUserAction = async () => {
+    const choices = [
+        { name: 'Enter a text prompt', value: 'text_prompt' },
+        { name: 'Select a task file', value: 'task_file' },
+        { name: `Process ${CODEGEN_TRIGGER} comments`, value: 'process_comments' },
+        new Separator(),
+        { name: 'Select AI service', value: 'select_ai_service' },
+        { name: 'Configuration', value: 'configure' },
+        { name: 'Print help', value: 'help' },
+        new Separator(),
+        { name: 'Exit', value: 'exit' },
+    ];
+    const action = await select({
+        message: 'What would you like to do?',
+        pageSize: choices.length,
+        choices,
+    });
+    return action;
+};
+// Function to get user options
+export const getUserOptions = async (options) => {
+    const updatedOptions = { ...options };
+    // General options
+    const generalOptions = await checkbox({
+        message: 'Select general options:',
+        choices: [
+            { name: 'Consider all files', value: 'considerAllFiles', checked: options.considerAllFiles },
+            { name: 'Dry run', value: 'dryRun', checked: options.dryRun },
+            { name: 'Verbose', value: 'verbose', checked: options.verbose },
+            { name: 'Require explanations', value: 'requireExplanations', checked: options.requireExplanations },
+            { name: 'Disable initial lint', value: 'disableInitialLint', checked: options.disableInitialLint },
+            { name: 'Ask questions', value: 'askQuestion', checked: options.askQuestion },
+            { name: 'Disable cache', value: 'disableCache', checked: options.disableCache },
+        ],
+    });
+    updatedOptions.considerAllFiles = generalOptions.includes('considerAllFiles');
+    updatedOptions.dryRun = generalOptions.includes('dryRun');
+    updatedOptions.verbose = generalOptions.includes('verbose');
+    updatedOptions.requireExplanations = generalOptions.includes('requireExplanations');
+    updatedOptions.disableInitialLint = generalOptions.includes('disableInitialLint');
+    updatedOptions.askQuestion = generalOptions.includes('askQuestion');
+    updatedOptions.disableCache = generalOptions.includes('disableCache');
+    // File operation options
+    const fileOptions = await checkbox({
+        message: 'Select file operation options:',
+        choices: [
+            { name: 'Allow file creation', value: 'allowFileCreate', checked: options.allowFileCreate },
+            { name: 'Allow file deletion', value: 'allowFileDelete', checked: options.allowFileDelete },
+            { name: 'Allow directory creation', value: 'allowDirectoryCreate', checked: options.allowDirectoryCreate },
+            { name: 'Allow file moving', value: 'allowFileMove', checked: options.allowFileMove },
+        ],
+    });
+    updatedOptions.allowFileCreate = fileOptions.includes('allowFileCreate');
+    updatedOptions.allowFileDelete = fileOptions.includes('allowFileDelete');
+    updatedOptions.allowDirectoryCreate = fileOptions.includes('allowDirectoryCreate');
+    updatedOptions.allowFileMove = fileOptions.includes('allowFileMove');
+    // AI model options
+    updatedOptions.disableContextOptimization = await confirm({
+        message: 'Disable context optimization?',
+        default: options.disableContextOptimization || false,
+    });
+    updatedOptions.temperature = parseFloat(await input({
+        message: 'Enter temperature (0.0 - 2.0):',
+        default: options.temperature?.toString() || '0.7',
+        validate: (value) => {
+            const num = parseFloat(value);
+            return (num >= 0 && num <= 2) || 'Please enter a number between 0 and 2';
+        },
+    }));
+    updatedOptions.cheap = await confirm({
+        message: 'Use cheaper, faster model?',
+        default: options.cheap || false,
+    });
+    updatedOptions.geminiBlockNone = await confirm({
+        message: 'Disable safety settings for Gemini Pro model?',
+        default: options.geminiBlockNone || false,
+    });
+    // Vision and image generation options
+    updatedOptions.vision = await confirm({
+        message: 'Enable vision capabilities?',
+        default: options.vision || false,
+    });
+    if (await confirm({ message: 'Enable image generation?', default: !!options.imagen })) {
+        updatedOptions.imagen = await select({
+            message: 'Select image generation service:',
+            choices: [
+                { name: 'Vertex AI', value: 'vertex-ai' },
+                { name: 'DALL-E', value: 'dall-e' },
+            ],
+            default: options.imagen || 'vertex-ai',
+        });
+    }
+    else {
+        updatedOptions.imagen = undefined;
+    }
+    // Content mask and ignore patterns
+    updatedOptions.contentMask = await input({
+        message: 'Enter content mask (prefix of path relative to rootDir):',
+        default: options.contentMask || '',
+    });
+    const ignorePatterns = await input({
+        message: 'Enter ignore patterns (comma-separated):',
+        default: options.ignorePatterns?.join(',') || '',
+    });
+    updatedOptions.ignorePatterns = ignorePatterns ? ignorePatterns.split(',').map(p => p.trim()) : undefined;
+    return updatedOptions;
+};
+// Function to check for CODEGEN comments in source files
+export const checkForCodegenComments = () => {
+    const sourceFiles = getSourceFiles();
+    for (const file of sourceFiles) {
+        const content = fs.readFileSync(file, 'utf-8');
+        if (content.includes(CODEGEN_TRIGGER)) {
+            return true;
+        }
+    }
+    return false;
+};
+// Function to select AI service
+export const selectAiService = async (defaultAiService) => {
+    const choices = [
+        { name: 'Vertex AI (Gemini)', value: 'vertex-ai' },
+        { name: 'AI Studio (Gemini)', value: 'ai-studio' },
+        { name: 'ChatGPT', value: 'chat-gpt' },
+        { name: 'Anthropic Claude', value: 'anthropic' },
+        { name: 'Claude via Vertex AI', value: 'vertex-ai-claude' },
+    ];
+    const selectedAiService = await select({
+        message: 'Select the AI model you want to use:',
+        pageSize: choices.length,
+        choices: choices.map((choice) => ({ ...choice, checked: defaultAiService === choice.value })),
+        default: defaultAiService,
+    });
+    return selectedAiService;
+};
+//# sourceMappingURL=codegen-interactive-utils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"codegen-interactive-utils.js","sourceRoot":"","sources":["../../src/main/codegen-interactive-utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAChF,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AAGvC,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC/C,MAAM,WAAW,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAAC;AAElD,0CAA0C;AAC1C,MAAM,CAAC,MAAM,cAAc,GAAG,GAAG,EAAE;IACjC,OAAO,CAAC,GAAG,CAAC,yBAAyB,WAAW,CAAC,OAAO,EAAE,CAAC,CAAC;AAC9D,CAAC,CAAC;AAWF,8BAA8B;AAC9B,MAAM,CAAC,MAAM,aAAa,GAAG,KAAK,IAAyB,EAAE;IAC3D,MAAM,OAAO,GAAG;QACd,EAAE,IAAI,EAAE,qBAAqB,EAAE,KAAK,EAAE,aAAa,EAAE;QACrD,EAAE,IAAI,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE;QAClD,EAAE,IAAI,EAAE,WAAW,eAAe,WAAW,EAAE,KAAK,EAAE,kBAAkB,EAAE;QAC1E,IAAI,SAAS,EAAE;QACf,EAAE,IAAI,EAAE,mBAAmB,EAAE,KAAK,EAAE,mBAAmB,EAAE;QACzD,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,WAAW,EAAE;QAC7C,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE;QACrC,IAAI,SAAS,EAAE;QACf,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE;KACvB,CAAC;IAEX,MAAM,MAAM,GAAG,MAAM,MAAM,CAAa;QACtC,OAAO,EAAE,4BAA4B;QACrC,QAAQ,EAAE,OAAO,CAAC,MAAM;QACxB,OAAO;KACR,CAAC,CAAC;IACH,OAAO,MAAM,CAAC;AAChB,CAAC,CAAC;AAEF,+BAA+B;AAC/B,MAAM,CAAC,MAAM,cAAc,GAAG,KAAK,EAAE,OAAuB,EAA2B,EAAE;IACvF,MAAM,cAAc,GAAG,EAAE,GAAG,OAAO,EAAE,CAAC;IAEtC,kBAAkB;IAClB,MAAM,cAAc,GAAG,MAAM,QAAQ,CAAC;QACpC,OAAO,EAAE,yBAAyB;QAClC,OAAO,EAAE;YACP,EAAE,IAAI,EAAE,oBAAoB,EAAE,KAAK,EAAE,kBAAkB,EAAE,OAAO,EAAE,OAAO,CAAC,gBAAgB,EAAE;YAC5F,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE;YAC7D,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE;YAC/D,EAAE,IAAI,EAAE,sBAAsB,EAAE,KAAK,EAAE,qBAAqB,EAAE,OAAO,EAAE,OAAO,CAAC,mBAAmB,EAAE;YACpG,EAAE,IAAI,EAAE,sBAAsB,EAAE,KAAK,EAAE,oBAAoB,EAAE,OAAO,EAAE,OAAO,CAAC,kBAAkB,EAAE;YAClG,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,CAAC,WAAW,EAAE;YAC7E,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,cAAc,EAAE,OAAO,EAAE,OAAO,CAAC,YAAY,EAAE;SAChF;KACF,CAAC,CAAC;IAEH,cAAc,CAAC,gBAAgB,GAAG,cAAc,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC;IAC9E,cAAc,CAAC,MAAM,GAAG,cAAc,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC1D,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAC5D,cAAc,CAAC,mBAAmB,GAAG,cAAc,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC;IACpF,cAAc,CAAC,kBAAkB,GAAG,cAAc,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC;IAClF,cAAc,CAAC,WAAW,GAAG,cAAc,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IACpE,cAAc,CAAC,YAAY,GAAG,cAAc,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;IAEtE,yBAAyB;IACzB,MAAM,WAAW,GAAG,MAAM,QAAQ,CAAC;QACjC,OAAO,EAAE,gCAAgC;QACzC,OAAO,EAAE;YACP,EAAE,IAAI,EAAE,qBAAqB,EAAE,KAAK,EAAE,iBAAiB,EAAE,OAAO,EAAE,OAAO,CAAC,eAAe,EAAE;YAC3F,EAAE,IAAI,EAAE,qBAAqB,EAAE,KAAK,EAAE,iBAAiB,EAAE,OAAO,EAAE,OAAO,CAAC,eAAe,EAAE;YAC3F,EAAE,IAAI,EAAE,0BAA0B,EAAE,KAAK,EAAE,sBAAsB,EAAE,OAAO,EAAE,OAAO,CAAC,oBAAoB,EAAE;YAC1G,EAAE,IAAI,EAAE,mBAAmB,EAAE,KAAK,EAAE,eAAe,EAAE,OAAO,EAAE,OAAO,CAAC,aAAa,EAAE;SACtF;KACF,CAAC,CAAC;IAEH,cAAc,CAAC,eAAe,GAAG,WAAW,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;IACzE,cAAc,CAAC,eAAe,GAAG,WAAW,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;IACzE,cAAc,CAAC,oBAAoB,GAAG,WAAW,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC;IACnF,cAAc,CAAC,aAAa,GAAG,WAAW,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC;IAErE,mBAAmB;IACnB,cAAc,CAAC,0BAA0B,GAAG,MAAM,OAAO,CAAC;QACxD,OAAO,EAAE,+BAA+B;QACxC,OAAO,EAAE,OAAO,CAAC,0BAA0B,IAAI,KAAK;KACrD,CAAC,CAAC;IAEH,cAAc,CAAC,WAAW,GAAG,UAAU,CAAC,MAAM,KAAK,CAAC;QAClD,OAAO,EAAE,gCAAgC;QACzC,OAAO,EAAE,OAAO,CAAC,WAAW,EAAE,QAAQ,EAAE,IAAI,KAAK;QACjD,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE;YAClB,MAAM,GAAG,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;YAC9B,OAAO,CAAC,GAAG,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,IAAI,uCAAuC,CAAC;QAC3E,CAAC;KACF,CAAC,CAAC,CAAC;IAEJ,cAAc,CAAC,KAAK,GAAG,MAAM,OAAO,CAAC;QACnC,OAAO,EAAE,4BAA4B;QACrC,OAAO,EAAE,OAAO,CAAC,KAAK,IAAI,KAAK;KAChC,CAAC,CAAC;IAEH,cAAc,CAAC,eAAe,GAAG,MAAM,OAAO,CAAC;QAC7C,OAAO,EAAE,+CAA+C;QACxD,OAAO,EAAE,OAAO,CAAC,eAAe,IAAI,KAAK;KAC1C,CAAC,CAAC;IAEH,sCAAsC;IACtC,cAAc,CAAC,MAAM,GAAG,MAAM,OAAO,CAAC;QACpC,OAAO,EAAE,6BAA6B;QACtC,OAAO,EAAE,OAAO,CAAC,MAAM,IAAI,KAAK;KACjC,CAAC,CAAC;IAEH,IAAI,MAAM,OAAO,CAAC,EAAE,OAAO,EAAE,0BAA0B,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;QACtF,cAAc,CAAC,MAAM,GAAG,MAAM,MAAM,CAAa;YAC/C,OAAO,EAAE,kCAAkC;YAC3C,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE;gBACzC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;aACpC;YACD,OAAO,EAAE,OAAO,CAAC,MAAM,IAAI,WAAW;SACvC,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,cAAc,CAAC,MAAM,GAAG,SAAS,CAAC;IACpC,CAAC;IAED,mCAAmC;IACnC,cAAc,CAAC,WAAW,GAAG,MAAM,KAAK,CAAC;QACvC,OAAO,EAAE,0DAA0D;QACnE,OAAO,EAAE,OAAO,CAAC,WAAW,IAAI,EAAE;KACnC,CAAC,CAAC;IAEH,MAAM,cAAc,GAAG,MAAM,KAAK,CAAC;QACjC,OAAO,EAAE,0CAA0C;QACnD,OAAO,EAAE,OAAO,CAAC,cAAc,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE;KACjD,CAAC,CAAC;IACH,cAAc,CAAC,cAAc,GAAG,cAAc,CAAC,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAE1G,OAAO,cAAc,CAAC;AACxB,CAAC,CAAC;AAEF,yDAAyD;AACzD,MAAM,CAAC,MAAM,uBAAuB,GAAG,GAAY,EAAE;IACnD,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;IACrC,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;QAC/B,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC/C,IAAI,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YACtC,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAEF,gCAAgC;AAChC,MAAM,CAAC,MAAM,eAAe,GAAG,KAAK,EAAE,gBAA2C,EAA0B,EAAE;IAC3G,MAAM,OAAO,GAAG;QACd,EAAE,IAAI,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE;QAClD,EAAE,IAAI,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE;QAClD,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE;QACtC,EAAE,IAAI,EAAE,kBAAkB,EAAE,KAAK,EAAE,WAAW,EAAE;QAChD,EAAE,IAAI,EAAE,sBAAsB,EAAE,KAAK,EAAE,kBAAkB,EAAE;KACnD,CAAC;IAEX,MAAM,iBAAiB,GAAG,MAAM,MAAM,CAAgB;QACpD,OAAO,EAAE,sCAAsC;QAC/C,QAAQ,EAAE,OAAO,CAAC,MAAM;QACxB,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,MAAM,EAAE,OAAO,EAAE,gBAAgB,KAAK,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;QAC7F,OAAO,EAAE,gBAAgB;KAC1B,CAAC,CAAC;IAEH,OAAO,iBAAiB,CAAC;AAC3B,CAAC,CAAC"}

package/dist/main/codegen-interactive.d.ts

@@ -0,0 +1,2 @@
+import { CodegenOptions } from './codegen-types.js';
+export declare const runInteractiveMode: (options: CodegenOptions) => Promise<void>;

package/dist/main/codegen-interactive.js

@@ -0,0 +1,68 @@
+import { input } from '@inquirer/prompts';
+import fileSelector from 'inquirer-file-selector';
+import { runCodegenIteration } from './codegen.js';
+import { CODEGEN_TRIGGER } from '../prompt/prompt-consts.js';
+import { displayWelcome, getUserAction, getUserOptions, checkForCodegenComments, selectAiService, } from './codegen-interactive-utils.js';
+import { rcConfig } from './config.js';
+import { printHelpMessage } from '../cli/cli-options.js';
+// Main function for interactive mode
+export const runInteractiveMode = async (options) => {
+    displayWelcome();
+    const runNext = true;
+    while (runNext) {
+        const action = await getUserAction();
+        if (action === 'exit') {
+            console.log('Exiting Genaicode Interactive Mode. Goodbye!');
+            break;
+        }
+        switch (action) {
+            case 'process_comments':
+                await runProcessComments(options);
+                break;
+            case 'text_prompt':
+                await runTextPrompt(options);
+                break;
+            case 'task_file':
+                await runTaskFile(options);
+                break;
+            case 'select_ai_service':
+                options.aiService = await selectAiService(options.aiService);
+                console.log(`Selected AI service: ${options.aiService}`);
+                break;
+            case 'configure':
+                options = await getUserOptions(options);
+                break;
+            case 'help':
+                printHelpMessage();
+                break;
+        }
+        console.log('Task completed. Returning to main menu...\n');
+    }
+};
+async function runProcessComments(options) {
+    const hasCodegenComments = checkForCodegenComments();
+    if (!hasCodegenComments) {
+        console.warn(`Warning: No ${CODEGEN_TRIGGER} comments found in the source code.`);
+    }
+    console.log(`Processing ${CODEGEN_TRIGGER} comments...`);
+    await runCodegenIteration({ ...options, considerAllFiles: false });
+}
+async function runTextPrompt(options) {
+    const prompt = await input({ message: 'Enter your text prompt:' });
+    console.log(`Executing prompt: ${prompt}`);
+    await runCodegenIteration({ ...options, explicitPrompt: prompt });
+}
+async function runTaskFile(options) {
+    const taskFile = await fileSelector({
+        message: 'Select task file:',
+        path: rcConfig.rootDir,
+    });
+    if (taskFile) {
+        console.log(`Selected task file: ${taskFile}`);
+        await runCodegenIteration({ ...options, taskFile });
+    }
+    else {
+        console.log('No task file selected.');
+    }
+}
+//# sourceMappingURL=codegen-interactive.js.map

package/dist/main/codegen-interactive.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codegen-interactive.js","sourceRoot":"","sources":["../../src/main/codegen-interactive.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAC1C,OAAO,YAAY,MAAM,wBAAwB,CAAC;AAClD,OAAO,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AACnD,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAC7D,OAAO,EACL,cAAc,EACd,aAAa,EACb,cAAc,EACd,uBAAuB,EACvB,eAAe,GAChB,MAAM,gCAAgC,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAEvC,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzD,qCAAqC;AACrC,MAAM,CAAC,MAAM,kBAAkB,GAAG,KAAK,EAAE,OAAuB,EAAE,EAAE;IAClE,cAAc,EAAE,CAAC;IAEjB,MAAM,OAAO,GAAG,IAAI,CAAC;IACrB,OAAO,OAAO,EAAE,CAAC;QACf,MAAM,MAAM,GAAG,MAAM,aAAa,EAAE,CAAC;QAErC,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;YACtB,OAAO,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAC;YAC5D,MAAM;QACR,CAAC;QAED,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,kBAAkB;gBACrB,MAAM,kBAAkB,CAAC,OAAO,CAAC,CAAC;gBAClC,MAAM;YACR,KAAK,aAAa;gBAChB,MAAM,aAAa,CAAC,OAAO,CAAC,CAAC;gBAC7B,MAAM;YACR,KAAK,WAAW;gBACd,MAAM,WAAW,CAAC,OAAO,CAAC,CAAC;gBAC3B,MAAM;YACR,KAAK,mBAAmB;gBACtB,OAAO,CAAC,SAAS,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAC7D,OAAO,CAAC,GAAG,CAAC,wBAAwB,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;gBACzD,MAAM;YACR,KAAK,WAAW;gBACd,OAAO,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,CAAC;gBACxC,MAAM;YACR,KAAK,MAAM;gBACT,gBAAgB,EAAE,CAAC;gBACnB,MAAM;QACV,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;IAC7D,CAAC;AACH,CAAC,CAAC;AAEF,KAAK,UAAU,kBAAkB,CAAC,OAAuB;IACvD,MAAM,kBAAkB,GAAG,uBAAuB,EAAE,CAAC;IACrD,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxB,OAAO,CAAC,IAAI,CAAC,eAAe,eAAe,qCAAqC,CAAC,CAAC;IACpF,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,cAAc,eAAe,cAAc,CAAC,CAAC;IACzD,MAAM,mBAAmB,CAAC,EAAE,GAAG,OAAO,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAC;AACrE,CAAC;AAED,KAAK,UAAU,aAAa,CAAC,OAAuB;IAClD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,EAAE,OAAO,EAAE,yBAAyB,EAAE,CAAC,CAAC;IACnE,OAAO,CAAC,GAAG,CAAC,qBAAqB,MAAM,EAAE,CAAC,CAAC;IAC3C,MAAM,mBAAmB,CAAC,EAAE,GAAG,OAAO,EAAE,cAAc,EAAE,MAAM,EAAE,CAAC,CAAC;AACpE,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,OAAuB;IAChD,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC;QAClC,OAAO,EAAE,mBAAmB;QAC5B,IAAI,EAAE,QAAQ,CAAC,OAAO;KACvB,CAAC,CAAC;IACH,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO,CAAC,GAAG,CAAC,uBAAuB,QAAQ,EAAE,CAAC,CAAC;QAC/C,MAAM,mBAAmB,CAAC,EAAE,GAAG,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;IACtD,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;IACxC,CAAC;AACH,CAAC"}

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

@@ -0,0 +1,28 @@
+export type AiServiceType = 'vertex-ai' | 'ai-studio' | 'vertex-ai-claude' | 'chat-gpt' | 'anthropic';
+export type ImagenType = 'vertex-ai' | 'dall-e';
+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;
+    disableCache?: boolean;
+    dependencyTree?: boolean;
+}

package/dist/main/codegen-types.js

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