genaicode 0.5.1 → 0.6.1

package/dist/examples/genaicode_plugins/grok_ai_service.js

@@ -0,0 +1,153 @@
+import OpenAI, { APIError } from 'openai';
+import assert from 'node:assert';
+import { printTokenUsageAndCost, processFunctionCalls, } from '../../src/ai-service/common.js';
+import { abortController } from '../../src/main/interactive/codegen-worker.js';
+import { modelOverrides } from '../../src/main/config.js';
+const grokAiService = {
+    name: 'grok-ai-service',
+    aiServices: {
+        'grok-ai-service': generateContent,
+    },
+};
+async function generateContent(prompt, functionDefs, requiredFunctionName, temperature, cheap = false) {
+    const openai = new OpenAI({
+        apiKey: process.env.GROK_OPENAI_API_KEY,
+        baseURL: 'https://api.x.ai/v1',
+    });
+    const messages = prompt
+        .map((item) => {
+        if (item.type === 'systemPrompt') {
+            return {
+                role: 'system',
+                content: item.systemPrompt,
+            };
+        }
+        else if (item.type === 'user') {
+            const messages = [];
+            if (item.functionResponses) {
+                messages.push(...item.functionResponses.map((response) => ({
+                    role: 'tool',
+                    name: response.name,
+                    content: response.content ?? '',
+                    tool_call_id: response.name,
+                })));
+            }
+            if ((item.images?.length ?? 0) > 0) {
+                messages.push({
+                    role: 'user',
+                    content: [
+                        ...item.images.map((image) => ({
+                            type: 'image_url',
+                            image_url: {
+                                url: 'data:' + image.mediaType + ';base64,' + image.base64url,
+                            },
+                        })),
+                        {
+                            type: 'text',
+                            text: item.text,
+                        },
+                    ],
+                });
+            }
+            else {
+                messages.push({
+                    role: 'user',
+                    content: item.text,
+                });
+            }
+            return messages;
+        }
+        else {
+            assert(item.type === 'assistant');
+            const message = {
+                role: 'assistant',
+                ...(item.text ? { content: item.text } : {}),
+                tool_calls: item.functionCalls?.map((call) => ({
+                    type: 'function',
+                    function: { name: call.name, arguments: JSON.stringify(call.args ?? {}) },
+                    id: call.name,
+                })),
+            };
+            return message;
+        }
+    })
+        .flat();
+    const defaultModel = cheap ? 'grok-2-mini' : 'grok-beta';
+    const model = cheap
+        ? (modelOverrides.chatGpt?.cheap ?? defaultModel)
+        : (modelOverrides.chatGpt?.default ?? defaultModel);
+    console.log(`Using OpenAI model: ${model}`);
+    assert(process.env.OPENAI_API_KEY, 'OPENAI_API_KEY environment variable is not set');
+    let retryCount = 0;
+    const maxRetries = 3;
+    let response = undefined;
+    while (retryCount < maxRetries) {
+        try {
+            response = await openai.chat.completions.create({
+                model: model,
+                messages,
+                tools: functionDefs.map((funDef) => ({ type: 'function', function: funDef })),
+                tool_choice: requiredFunctionName
+                    ? { type: 'function', function: { name: requiredFunctionName } }
+                    : 'required',
+                temperature: temperature,
+            }, { signal: abortController?.signal });
+            break; // Exit loop if successful
+        }
+        catch (error) {
+            if (error instanceof APIError && error.status === 429) {
+                // Rate limit error
+                let retryAfter = error.headers?.['retry-after'] ? parseInt(error.headers['retry-after'], 10) : 60;
+                retryAfter = Math.min(retryAfter, 30);
+                console.log(`Rate limited. Retrying after ${retryAfter} seconds. Attempt ${retryCount + 1} of ${maxRetries}.`);
+                await new Promise((resolve) => setTimeout(resolve, retryAfter * 1000));
+                retryCount++;
+            }
+            else {
+                console.error('An error occurred:', error);
+                throw new Error('API request failed. Operation aborted.');
+            }
+        }
+    }
+    if (retryCount === maxRetries) {
+        console.error(`Failed to complete request after ${maxRetries} attempts due to rate limiting.`);
+        throw new Error('Rate limit exceeded. Operation aborted.');
+    }
+    assert(response);
+    // Print token usage for chat gpt
+    const usage = {
+        inputTokens: response.usage.prompt_tokens - (response.usage?.prompt_tokens_details?.cached_tokens ?? 0),
+        outputTokens: response.usage.completion_tokens,
+        totalTokens: response.usage.total_tokens,
+        cacheReadTokens: response.usage?.prompt_tokens_details?.cached_tokens,
+    };
+    printTokenUsageAndCost({
+        aiService: 'chat-gpt',
+        usage,
+        inputCostPerToken: 0.000005,
+        outputCostPerToken: 0.000015,
+        cheap,
+    });
+    const responseMessage = response.choices[0].message;
+    if (responseMessage.content) {
+        console.log('Message', responseMessage.content);
+    }
+    const toolCalls = responseMessage.tool_calls;
+    if (toolCalls) {
+        const functionCalls = toolCalls.map((call) => {
+            const name = call.function.name;
+            const args = JSON.parse(call.function.arguments);
+            return {
+                id: call.id,
+                name,
+                args,
+            };
+        });
+        return processFunctionCalls(functionCalls, functionDefs);
+    }
+    else {
+        throw new Error('No tool calls found in response');
+    }
+}
+export default grokAiService;
+//# sourceMappingURL=grok_ai_service.js.map

package/dist/examples/genaicode_plugins/grok_ai_service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"grok_ai_service.js","sourceRoot":"","sources":["../../../examples/genaicode_plugins/grok_ai_service.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,EAAE,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAC1C,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EACL,sBAAsB,EACtB,oBAAoB,GAKrB,MAAM,gCAAgC,CAAC;AAGxC,OAAO,EAAE,eAAe,EAAE,MAAM,8CAA8C,CAAC;AAC/E,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAE1D,MAAM,aAAa,GAAW;IAC5B,IAAI,EAAE,iBAAiB;IACvB,UAAU,EAAE;QACV,iBAAiB,EAAE,eAAe;KACnC;CACF,CAAC;AAEF,KAAK,UAAU,eAAe,CAC5B,MAAoB,EACpB,YAA2B,EAC3B,oBAAmC,EACnC,WAAmB,EACnB,KAAK,GAAG,KAAK;IAEb,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;QACxB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,mBAAmB;QACvC,OAAO,EAAE,qBAAqB;KAC/B,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAsC,MAAM;SACvD,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACZ,IAAI,IAAI,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;YACjC,OAAO;gBACL,IAAI,EAAE,QAAiB;gBACvB,OAAO,EAAE,IAAI,CAAC,YAAa;aAC5B,CAAC;QACJ,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAChC,MAAM,QAAQ,GAAiC,EAAE,CAAC;YAClD,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3B,QAAQ,CAAC,IAAI,CACX,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;oBAC3C,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,EAAE;oBAC/B,YAAY,EAAE,QAAQ,CAAC,IAAI;iBAC5B,CAAC,CAAC,CACJ,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC;gBACnC,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE;wBACP,GAAG,IAAI,CAAC,MAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;4BAC9B,IAAI,EAAE,WAAoB;4BAC1B,SAAS,EAAE;gCACT,GAAG,EAAE,OAAO,GAAG,KAAK,CAAC,SAAS,GAAG,UAAU,GAAG,KAAK,CAAC,SAAS;6BAC9D;yBACF,CAAC,CAAC;wBACH;4BACE,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,IAAI,CAAC,IAAK;yBACjB;qBACF;iBACF,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,IAAI,CAAC,IAAK;iBACpB,CAAC,CAAC;YACL,CAAC;YACD,OAAO,QAAQ,CAAC;QAClB,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;YAClC,MAAM,OAAO,GAA+B;gBAC1C,IAAI,EAAE,WAAoB;gBAC1B,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC5C,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBAC7C,IAAI,EAAE,UAAmB;oBACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE;oBACzE,EAAE,EAAE,IAAI,CAAC,IAAI;iBACd,CAAC,CAAC;aACJ,CAAC;YACF,OAAO,OAAO,CAAC;QACjB,CAAC;IACH,CAAC,CAAC;SACD,IAAI,EAAE,CAAC;IAEV,MAAM,YAAY,GAAG,KAAK,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,WAAW,CAAC;IACzD,MAAM,KAAK,GAAG,KAAK;QACjB,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,EAAE,KAAK,IAAI,YAAY,CAAC;QACjD,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,EAAE,OAAO,IAAI,YAAY,CAAC,CAAC;IACtD,OAAO,CAAC,GAAG,CAAC,uBAAuB,KAAK,EAAE,CAAC,CAAC;IAC5C,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,gDAAgD,CAAC,CAAC;IAErF,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,MAAM,UAAU,GAAG,CAAC,CAAC;IACrB,IAAI,QAAQ,GAAuD,SAAS,CAAC;IAE7E,OAAO,UAAU,GAAG,UAAU,EAAE,CAAC;QAC/B,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAC7C;gBACE,KAAK,EAAE,KAAK;gBACZ,QAAQ;gBACR,KAAK,EAAE,YAAY,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,UAAmB,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;gBACtF,WAAW,EAAE,oBAAoB;oBAC/B,CAAC,CAAC,EAAE,IAAI,EAAE,UAAmB,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,oBAAoB,EAAE,EAAE;oBACzE,CAAC,CAAC,UAAU;gBACd,WAAW,EAAE,WAAW;aACzB,EACD,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,EAAE,CACpC,CAAC;YACF,MAAM,CAAC,0BAA0B;QACnC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,QAAQ,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACtD,mBAAmB;gBACnB,IAAI,UAAU,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAClG,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;gBACtC,OAAO,CAAC,GAAG,CAAC,gCAAgC,UAAU,qBAAqB,UAAU,GAAG,CAAC,OAAO,UAAU,GAAG,CAAC,CAAC;gBAC/G,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC;gBACvE,UAAU,EAAE,CAAC;YACf,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,KAAK,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;gBAC3C,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,UAAU,KAAK,UAAU,EAAE,CAAC;QAC9B,OAAO,CAAC,KAAK,CAAC,oCAAoC,UAAU,iCAAiC,CAAC,CAAC;QAC/F,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;IAC7D,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,CAAC;IAEjB,iCAAiC;IACjC,MAAM,KAAK,GAAe;QACxB,WAAW,EAAE,QAAQ,CAAC,KAAM,CAAC,aAAa,GAAG,CAAC,QAAQ,CAAC,KAAK,EAAE,qBAAqB,EAAE,aAAa,IAAI,CAAC,CAAC;QACxG,YAAY,EAAE,QAAQ,CAAC,KAAM,CAAC,iBAAiB;QAC/C,WAAW,EAAE,QAAQ,CAAC,KAAM,CAAC,YAAY;QACzC,eAAe,EAAE,QAAQ,CAAC,KAAK,EAAE,qBAAqB,EAAE,aAAa;KACtE,CAAC;IACF,sBAAsB,CAAC;QACrB,SAAS,EAAE,UAAU;QACrB,KAAK;QACL,iBAAiB,EAAE,QAAQ;QAC3B,kBAAkB,EAAE,QAAQ;QAC5B,KAAK;KACN,CAAC,CAAC;IAEH,MAAM,eAAe,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;IAEpD,IAAI,eAAe,CAAC,OAAO,EAAE,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,eAAe,CAAC,OAAO,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,SAAS,GAAG,eAAe,CAAC,UAAU,CAAC;IAC7C,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,aAAa,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YAC3C,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YAChC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YAEjD,OAAO;gBACL,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,IAAI;gBACJ,IAAI;aACL,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO,oBAAoB,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;IAC3D,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;IACrD,CAAC;AACH,CAAC;AAED,eAAe,aAAa,CAAC"}

package/dist/examples/genaicode_plugins/nonsense_operation.d.ts

@@ -0,0 +1,3 @@
+import { Plugin } from '../../src/main/codegen-types';
+declare const fakeAiService: Plugin;
+export default fakeAiService;

package/dist/examples/genaicode_plugins/nonsense_operation.js

@@ -0,0 +1,23 @@
+const fakeAiService = {
+    name: 'nonsense-operation',
+    operations: {
+        'nonsense-operation': {
+            executor: async (args) => {
+                console.log('NONSE OPERATION EXECUTED', args);
+            },
+            def: {
+                name: 'nonsense',
+                description: 'nonsense operation',
+                parameters: {
+                    type: 'object',
+                    properties: {
+                        input: { type: 'string' },
+                    },
+                    required: ['input'],
+                },
+            },
+        },
+    },
+};
+export default fakeAiService;
+//# sourceMappingURL=nonsense_operation.js.map

package/dist/examples/genaicode_plugins/nonsense_operation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"nonsense_operation.js","sourceRoot":"","sources":["../../../examples/genaicode_plugins/nonsense_operation.ts"],"names":[],"mappings":"AAEA,MAAM,aAAa,GAAW;IAC5B,IAAI,EAAE,oBAAoB;IAC1B,UAAU,EAAE;QACV,oBAAoB,EAAE;YACpB,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;gBACvB,OAAO,CAAC,GAAG,CAAC,0BAA0B,EAAE,IAAI,CAAC,CAAC;YAChD,CAAC;YACD,GAAG,EAAE;gBACH,IAAI,EAAE,UAAU;gBAChB,WAAW,EAAE,oBAAoB;gBACjC,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;qBAC1B;oBACD,QAAQ,EAAE,CAAC,OAAO,CAAC;iBACpB;aACF;SACF;KACF;CACF,CAAC;AAEF,eAAe,aAAa,CAAC"}

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

@@ -0,0 +1 @@
+export declare function isProjectPath(filePath: string): boolean;

package/dist/files/path-utils.js

@@ -0,0 +1,12 @@
+import path from 'path';
+import { getSourceFiles } from './find-files.js';
+import { rcConfig } from '../main/config.js';
+import { isAncestorDirectory } from './file-utils.js';
+export 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=path-utils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"path-utils.js","sourceRoot":"","sources":["../../src/files/path-utils.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAEtD,MAAM,UAAU,aAAa,CAAC,QAAgB;IAC5C,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/files/update-files.d.ts

@@ -1,5 +1,5 @@
-import { FunctionCall } from '../ai-service/common.js';
 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
  */

package/dist/files/update-files.js

@@ -1,194 +1,19 @@
-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';
+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) {
-    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);
+        const executor = getOperationExecutor(name);
+        if (!executor) {
+            console.warn(`Unknown operation: ${name}`);
         }
-        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 (args === undefined) {
+            console.warn(`Operation ${name} has no arguments`);
         }
-        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, 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;
-            }
+        else {
+            await executor(args, options);
         }
     }
 }
-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

@@ -1 +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,CAAC,QAAQ,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,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"}
+{"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/index.d.ts

@@ -1,3 +1,8 @@
-export { type CodegenOptions } from './main/codegen-types.js';
+export * from './main/codegen-types.js';
 export * from './main/codegen.js';
 export * from './prompt/prompt-service.js';
+export * from './ai-service/common.js';
+export * from './prompt/steps/step-ask-question/step-ask-question-types.js';
+export * from './main/common/content-bus.js';
+export * from './prompt/steps/steps-types.js';
+export * from './main/common/user-actions.js';

package/dist/index.js

@@ -1,3 +1,9 @@
+export * from './main/codegen-types.js';
 export * from './main/codegen.js';
 export * from './prompt/prompt-service.js';
+export * from './ai-service/common.js';
+export * from './prompt/steps/step-ask-question/step-ask-question-types.js';
+export * from './main/common/content-bus.js';
+export * from './prompt/steps/steps-types.js';
+export * from './main/common/user-actions.js';
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,cAAc,mBAAmB,CAAC;AAClC,cAAc,4BAA4B,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,yBAAyB,CAAC;AACxC,cAAc,mBAAmB,CAAC;AAClC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,wBAAwB,CAAC;AACvC,cAAc,6DAA6D,CAAC;AAC5E,cAAc,8BAA8B,CAAC;AAC7C,cAAc,+BAA+B,CAAC;AAC9C,cAAc,+BAA+B,CAAC"}

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

@@ -1,4 +1,10 @@
-export type AiServiceType = 'vertex-ai' | 'ai-studio' | 'vertex-ai-claude' | 'chat-gpt' | 'anthropic';
+import { FunctionDef, GenerateContentFunction } from '../ai-service/common';
+import { ActionHandler } from '../prompt/steps/step-ask-question/step-ask-question-types';
+/** Example: {@link ../../examples/genaicode_plugins/grok_ai_service.ts} */
+export type PluginAiServiceType = `plugin:${string}`;
+/** Example: {@link ../../examples/genaicode_plugins/nonsense_action_handlers.ts} */
+export type PluginActionType = `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;
@@ -40,3 +46,27 @@ export interface CodegenOptions {
     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>;
+    actionHandlers?: Record<string, {
+        /** The action handler implementation */
+        handler: ActionHandler;
+        /**
+         * Description of what this action handler does.
+         * This description will be included in the askQuestion function definition
+         * to help the AI understand when to use this action.
+         */
+        description: string;
+    }>;
+}
+export {};

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

@@ -0,0 +1,3 @@
+import { AiServiceType } from './codegen-types';
+export declare function getSupportedAiServices(): AiServiceType[];
+export declare function stringToAiServiceType(aiService: string | undefined | null): AiServiceType;

package/dist/main/codegen-utils.js

@@ -0,0 +1,16 @@
+import { getRegisteredAiServices } from './plugin-loader.js';
+export function getSupportedAiServices() {
+    return ['vertex-ai', 'ai-studio', 'vertex-ai-claude', 'chat-gpt', 'anthropic', ...getRegisteredAiServices().keys()];
+}
+export function stringToAiServiceType(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/main/codegen-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codegen-utils.js","sourceRoot":"","sources":["../../src/main/codegen-utils.ts"],"names":[],"mappings":"AACA,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,qBAAqB,CAAC,SAAoC;IACxE,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/main/codegen.js

@@ -1,6 +1,5 @@
 import { exec } from 'child_process';
 import util from 'util';
-import assert from 'node:assert';
 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';
@@ -20,6 +19,8 @@ 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 { stringToAiServiceType } from './codegen-utils.js';
 /** Executes codegen */
 export async function runCodegen(isDev = false) {
     // Print to console the received parameters
@@ -39,7 +40,7 @@ export async function runCodegen(isDev = false) {
         allowFileDelete: cliParams.allowFileDelete,
         allowDirectoryCreate: cliParams.allowDirectoryCreate,
         allowFileMove: cliParams.allowFileMove,
-        aiService: cliParamToAiService(),
+        aiService: stringToAiServiceType(cliParams.aiService),
         vision: cliParams.vision,
         imagen: cliParams.imagen,
         disableContextOptimization: cliParams.disableContextOptimization,
@@ -103,7 +104,7 @@ export async function runCodegenIteration(options, abortSignal, waitIfPaused = (
     await waitIfPaused();
     putSystemMessage('Generating response');
     try {
-        const functionCalls = await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt(options), waitIfPaused);
+        const functionCalls = await promptService(getGenerateContentFunctions(), GENERATE_IMAGE_FNS, getCodeGenPrompt(options), waitIfPaused);
         console.log('Received function calls:', functionCalls);
         await waitIfPaused();
         if (functionCalls.length === 0) {
@@ -133,7 +134,7 @@ export async function runCodegenIteration(options, abortSignal, waitIfPaused = (
                     const firstLintError = error;
                     const lintErrorPrompt = getLintFixPrompt(rcConfig.lintCommand, options, firstLintError.stdout, firstLintError.stderr);
                     putSystemMessage('Generating response for lint fixes');
-                    const lintFixFunctionCalls = (await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, {
+                    const lintFixFunctionCalls = (await promptService(getGenerateContentFunctions(), GENERATE_IMAGE_FNS, {
                         prompt: lintErrorPrompt,
                         options: { ...options, considerAllFiles: true },
                     }, waitIfPaused));
@@ -183,30 +184,18 @@ export async function runCodegenIteration(options, abortSignal, waitIfPaused = (
 }
 // helper functions and consts
 const execPromise = util.promisify(exec);
-const GENERATE_CONTENT_FNS = {
-    'vertex-ai-claude': generateContentVertexAiClaude,
-    'vertex-ai': generateContentVertexAi,
-    'ai-studio': generateContentAiStudio,
-    anthropic: generateContentAnthropic,
-    'chat-gpt': generateContentGPT,
-};
+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,
 };
-function cliParamToAiService() {
-    const result = cliParams.vertexAi
-        ? 'vertex-ai'
-        : cliParams.aiStudio
-            ? 'ai-studio'
-            : cliParams.vertexAiClaude
-                ? 'vertex-ai-claude'
-                : cliParams.chatGpt
-                    ? 'chat-gpt'
-                    : cliParams.anthropic
-                        ? 'anthropic'
-                        : undefined;
-    assert(result, 'Please specify which AI service should be used');
-    return result;
-}
 //# sourceMappingURL=codegen.js.map