genaicode 0.14.2 → 1.0.0

package/dist/prompt/steps/step-iterate/handlers/container-task/container-commands-types.js

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

package/dist/prompt/steps/step-iterate/handlers/container-task/container-commands-types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"container-commands-types.js","sourceRoot":"","sources":["../../../../../../src/prompt/steps/step-iterate/handlers/container-task/container-commands-types.ts"],"names":[],"mappings":""}

package/dist/prompt/steps/step-iterate/handlers/container-task/container-task-orchestrator.d.ts

@@ -0,0 +1,2 @@
+import { ActionHandlerProps, ActionResult } from '../../step-iterate-types.js';
+export declare function runContainerTaskOrchestrator({ iterateCall, prompt, generateContentFn, options, waitIfPaused, }: ActionHandlerProps): Promise<ActionResult>;

package/dist/prompt/steps/step-iterate/handlers/container-task/container-task-orchestrator.js

@@ -0,0 +1,202 @@
+import Docker from 'dockerode';
+import { ModelType } from '../../../../../ai-service/common-types.js';
+import { putAssistantMessage, putContainerLog, putSystemMessage, putUserMessage, } from '../../../../../main/common/content-bus.js';
+import { askUserForConfirmationWithAnswer } from '../../../../../main/common/user-actions.js';
+import { abortController } from '../../../../../main/common/abort-controller.js';
+import { runContainerTaskDef } from '../../../../function-defs/run-container-task.js';
+import { cleanupOrphanedContainers, createAndStartContainer, executeCommand, pullImage, stopContainer, } from './utils/docker-utils.js';
+import { commandExecutionLoop } from './command-execution-loop.js';
+import { getFunctionDefs } from '../../../../function-calling.js';
+export async function runContainerTaskOrchestrator({ iterateCall, prompt, generateContentFn, options, waitIfPaused, }) {
+    const docker = new Docker({ socketPath: '/var/run/docker.sock' });
+    await cleanupOrphanedContainers(docker);
+    if (abortController?.signal.aborted) {
+        putContainerLog('warn', 'Container task aborted before start.');
+        putSystemMessage('Container task aborted before start.');
+        return { breakLoop: false, items: [] };
+    }
+    try {
+        putContainerLog('info', 'Container task: generating proper task request');
+        const request = [
+            [
+                ...prompt,
+                {
+                    type: 'assistant',
+                    text: iterateCall.args.message,
+                },
+                {
+                    type: 'user',
+                    text: 'I understand that you want to run a container task. Please provide the Docker image and task description.',
+                },
+            ],
+            {
+                functionDefs: getFunctionDefs(),
+                requiredFunctionName: runContainerTaskDef.name,
+                temperature: 0.7,
+                modelType: ModelType.CHEAP,
+                expectedResponseType: {
+                    text: false,
+                    functionCall: true,
+                    media: false,
+                },
+            },
+            options,
+        ];
+        const response = await generateContentFn(...request);
+        const runContainerTaskCall = response
+            .filter((item) => item.type === 'functionCall')
+            .map((item) => item.functionCall).find((call) => call.name === 'runContainerTask');
+        if (!runContainerTaskCall?.args?.image || !runContainerTaskCall.args.taskDescription) {
+            putContainerLog('error', 'Failed to get valid runContainerTask request');
+            putSystemMessage('❌ Failed to get valid runContainerTask request');
+            prompt.push({
+                type: 'assistant',
+                text: iterateCall.args.message,
+            }, {
+                type: 'user',
+                text: 'Failed to get valid runContainerTask request',
+            });
+            return { breakLoop: false, items: [] };
+        }
+        putAssistantMessage(`I would like to run a container task with image: ${runContainerTaskCall.args.image}`, runContainerTaskCall.args);
+        const confirmation = await askUserForConfirmationWithAnswer(`Do you want to run the task?`, 'Run task', 'Reject', true, options);
+        if (!confirmation.confirmed) {
+            putContainerLog('warn', 'Container task cancelled by user.');
+            putSystemMessage('Container task cancelled by user.');
+            if (confirmation.answer) {
+                putUserMessage(confirmation.answer);
+            }
+            prompt.push({
+                type: 'assistant',
+                text: iterateCall.args.message,
+            }, {
+                type: 'user',
+                text: 'I reject running the container task.' + (confirmation.answer ? ` ${confirmation.answer}` : ''),
+            });
+            return { breakLoop: false, items: [] };
+        }
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args.message,
+        }, {
+            type: 'user',
+            text: 'I accept running the container task.' + (confirmation.answer ? ` ${confirmation.answer}` : ''),
+        });
+        const { image, taskDescription, workingDir } = runContainerTaskCall.args;
+        putContainerLog('info', `Starting container task with image: ${image}`);
+        putSystemMessage(`🐳 Starting container task with image: ${image}`);
+        if (confirmation.answer) {
+            putUserMessage(confirmation.answer);
+        }
+        if (abortController?.signal.aborted) {
+            putContainerLog('warn', 'Container task aborted before pulling image.');
+            putSystemMessage('Container task aborted by user.');
+            return { breakLoop: false, items: [] };
+        }
+        try {
+            await pullImage(docker, image);
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            putSystemMessage('❌ Failed to pull Docker image', { error: errorMessage });
+            // The pullImage function already logs the detailed error to the terminal.
+            prompt.push({
+                type: 'assistant',
+                text: iterateCall.args.message,
+                functionCalls: [runContainerTaskCall],
+            }, {
+                type: 'user',
+                functionResponses: [
+                    {
+                        name: 'runContainerTask',
+                        call_id: runContainerTaskCall.id || undefined,
+                        content: `Failed to pull Docker image: ${errorMessage}`,
+                    },
+                ],
+            });
+            return { breakLoop: false, items: [] };
+        }
+        let container;
+        let onAbort;
+        try {
+            container = await createAndStartContainer(docker, image);
+            onAbort = () => {
+                putContainerLog('warn', 'Execution interrupted by user. Stopping container...', undefined, 'docker');
+                // Best-effort stop; stopContainer is idempotent/safe via try/catch
+                if (container) {
+                    stopContainer(container).catch(() => { });
+                }
+            };
+            abortController?.signal.addEventListener('abort', onAbort);
+            putContainerLog('info', 'Ensuring working directory exists.');
+            await executeCommand(container, '/bin/sh', `mkdir -p ${workingDir}`, '', '/');
+            putContainerLog('info', 'Entering command execution loop.');
+            const { success, summary } = await commandExecutionLoop(runContainerTaskCall.args, container, taskDescription, generateContentFn, options, waitIfPaused);
+            const finalMessage = `✅ Task finished with status: ${success ? 'Success' : 'Failed'}.
+
+**Summary:**
+${summary}`;
+            putContainerLog(success ? 'success' : 'error', `Task finished. Summary: ${summary}`);
+            putSystemMessage(finalMessage);
+            prompt.push({
+                type: 'assistant',
+                text: iterateCall.args.message,
+                functionCalls: [runContainerTaskCall],
+            }, {
+                type: 'user',
+                functionResponses: [
+                    {
+                        name: 'runContainerTask',
+                        call_id: runContainerTaskCall.id || undefined,
+                        content: finalMessage,
+                    },
+                ],
+            });
+            return { breakLoop: false, items: [] };
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            putContainerLog('error', 'An error occurred during the container task', { error: errorMessage });
+            putSystemMessage('❌ An error occurred during the container task', {
+                error: errorMessage,
+            });
+            prompt.push({
+                type: 'assistant',
+                text: iterateCall.args.message,
+                functionCalls: [runContainerTaskCall],
+            }, {
+                type: 'user',
+                functionResponses: [
+                    {
+                        name: 'runContainerTask',
+                        call_id: runContainerTaskCall.id || undefined,
+                        content: `An error occurred during the container task: ${errorMessage}`,
+                    },
+                ],
+            });
+            return { breakLoop: false, items: [] };
+        }
+        finally {
+            if (container) {
+                await stopContainer(container);
+                if (onAbort) {
+                    abortController?.signal.removeEventListener('abort', onAbort);
+                }
+            }
+        }
+    }
+    catch (error) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        putContainerLog('error', 'Error during container task initialization', { error: errorMessage });
+        putSystemMessage('Error during container task initialization', { error: errorMessage });
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args.message,
+        }, {
+            type: 'user',
+            text: `Error during container task initialization: ${errorMessage}`,
+        });
+        return { breakLoop: false, items: [] };
+    }
+}
+//# sourceMappingURL=container-task-orchestrator.js.map

package/dist/prompt/steps/step-iterate/handlers/container-task/container-task-orchestrator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"container-task-orchestrator.js","sourceRoot":"","sources":["../../../../../../src/prompt/steps/step-iterate/handlers/container-task/container-task-orchestrator.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,WAAW,CAAC;AAC/B,OAAO,EAAqC,SAAS,EAAE,MAAM,2CAA2C,CAAC;AACzG,OAAO,EACL,mBAAmB,EACnB,eAAe,EACf,gBAAgB,EAChB,cAAc,GACf,MAAM,2CAA2C,CAAC;AACnD,OAAO,EAAE,gCAAgC,EAAE,MAAM,4CAA4C,CAAC;AAC9F,OAAO,EAAE,eAAe,EAAE,MAAM,gDAAgD,CAAC;AACjF,OAAO,EAAwB,mBAAmB,EAAE,MAAM,iDAAiD,CAAC;AAC5G,OAAO,EACL,yBAAyB,EACzB,uBAAuB,EACvB,cAAc,EACd,SAAS,EACT,aAAa,GACd,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAElE,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAAC,EACjD,WAAW,EACX,MAAM,EACN,iBAAiB,EACjB,OAAO,EACP,YAAY,GACO;IACnB,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,UAAU,EAAE,sBAAsB,EAAE,CAAC,CAAC;IAClE,MAAM,yBAAyB,CAAC,MAAM,CAAC,CAAC;IAExC,IAAI,eAAe,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;QACpC,eAAe,CAAC,MAAM,EAAE,sCAAsC,CAAC,CAAC;QAChE,gBAAgB,CAAC,sCAAsC,CAAC,CAAC;QACzD,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;IAED,IAAI,CAAC;QACH,eAAe,CAAC,MAAM,EAAE,gDAAgD,CAAC,CAAC;QAE1E,MAAM,OAAO,GAAwB;YACnC;gBACE,GAAG,MAAM;gBACT;oBACE,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;iBAChC;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,2GAA2G;iBAClH;aACF;YACD;gBACE,YAAY,EAAE,eAAe,EAAE;gBAC/B,oBAAoB,EAAE,mBAAmB,CAAC,IAAI;gBAC9C,WAAW,EAAE,GAAG;gBAChB,SAAS,EAAE,SAAS,CAAC,KAAK;gBAC1B,oBAAoB,EAAE;oBACpB,IAAI,EAAE,KAAK;oBACX,YAAY,EAAE,IAAI;oBAClB,KAAK,EAAE,KAAK;iBACb;aACF;YACD,OAAO;SACR,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CAAC,GAAG,OAAO,CAAC,CAAC;QACrD,MAAM,oBAAoB,GACxB,QAAQ;aACL,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;aAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CACnC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,kBAAkB,CAAC,CAAC;QAEnD,IAAI,CAAC,oBAAoB,EAAE,IAAI,EAAE,KAAK,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YACrF,eAAe,CAAC,OAAO,EAAE,8CAA8C,CAAC,CAAC;YACzE,gBAAgB,CAAC,gDAAgD,CAAC,CAAC;YAEnE,MAAM,CAAC,IAAI,CACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;aAChC,EACD;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,8CAA8C;aACrD,CACF,CAAC;YACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;QAED,mBAAmB,CACjB,oDAAoD,oBAAoB,CAAC,IAAI,CAAC,KAAK,EAAE,EACrF,oBAAoB,CAAC,IAAI,CAC1B,CAAC;QAEF,MAAM,YAAY,GAAG,MAAM,gCAAgC,CACzD,8BAA8B,EAC9B,UAAU,EACV,QAAQ,EACR,IAAI,EACJ,OAAO,CACR,CAAC;QACF,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC;YAC5B,eAAe,CAAC,MAAM,EAAE,mCAAmC,CAAC,CAAC;YAC7D,gBAAgB,CAAC,mCAAmC,CAAC,CAAC;YACtD,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC;gBACxB,cAAc,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YACtC,CAAC;YAED,MAAM,CAAC,IAAI,CACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;aAChC,EACD;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,sCAAsC,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACtG,CACF,CAAC;YACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;QAED,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;SAChC,EACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,sCAAsC,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACtG,CACF,CAAC;QAEF,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,GAAG,oBAAoB,CAAC,IAAI,CAAC;QAEzE,eAAe,CAAC,MAAM,EAAE,uCAAuC,KAAK,EAAE,CAAC,CAAC;QACxE,gBAAgB,CAAC,0CAA0C,KAAK,EAAE,CAAC,CAAC;QAEpE,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC;YACxB,cAAc,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC;QAED,IAAI,eAAe,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;YACpC,eAAe,CAAC,MAAM,EAAE,8CAA8C,CAAC,CAAC;YACxE,gBAAgB,CAAC,iCAAiC,CAAC,CAAC;YACpD,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;QAED,IAAI,CAAC;YACH,MAAM,SAAS,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,gBAAgB,CAAC,+BAA+B,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;YAC3E,0EAA0E;YAE1E,MAAM,CAAC,IAAI,CACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;gBAC/B,aAAa,EAAE,CAAC,oBAAoB,CAAC;aACtC,EACD;gBACE,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE;oBACjB;wBACE,IAAI,EAAE,kBAAkB;wBACxB,OAAO,EAAE,oBAAoB,CAAC,EAAE,IAAI,SAAS;wBAC7C,OAAO,EAAE,gCAAgC,YAAY,EAAE;qBACxD;iBACF;aACF,CACF,CAAC;YACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;QAED,IAAI,SAAuC,CAAC;QAC5C,IAAI,OAAiC,CAAC;QACtC,IAAI,CAAC;YACH,SAAS,GAAG,MAAM,uBAAuB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAEzD,OAAO,GAAG,GAAG,EAAE;gBACb,eAAe,CAAC,MAAM,EAAE,sDAAsD,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;gBACrG,mEAAmE;gBACnE,IAAI,SAAS,EAAE,CAAC;oBACd,aAAa,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;gBAC3C,CAAC;YACH,CAAC,CAAC;YACF,eAAe,EAAE,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAE3D,eAAe,CAAC,MAAM,EAAE,oCAAoC,CAAC,CAAC;YAC9D,MAAM,cAAc,CAAC,SAAS,EAAE,SAAS,EAAE,YAAY,UAAU,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;YAE9E,eAAe,CAAC,MAAM,EAAE,kCAAkC,CAAC,CAAC;YAC5D,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,MAAM,oBAAoB,CACrD,oBAAoB,CAAC,IAAI,EACzB,SAAS,EACT,eAAe,EACf,iBAAiB,EACjB,OAAO,EACP,YAAY,CACb,CAAC;YAEF,MAAM,YAAY,GAAG,gCAAgC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ;;;EAGvF,OAAO,EAAE,CAAC;YAEN,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,EAAE,2BAA2B,OAAO,EAAE,CAAC,CAAC;YACrF,gBAAgB,CAAC,YAAY,CAAC,CAAC;YAE/B,MAAM,CAAC,IAAI,CACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;gBAC/B,aAAa,EAAE,CAAC,oBAAoB,CAAC;aACtC,EACD;gBACE,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE;oBACjB;wBACE,IAAI,EAAE,kBAAkB;wBACxB,OAAO,EAAE,oBAAoB,CAAC,EAAE,IAAI,SAAS;wBAC7C,OAAO,EAAE,YAAY;qBACtB;iBACF;aACF,CACF,CAAC;YACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,eAAe,CAAC,OAAO,EAAE,6CAA6C,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;YACjG,gBAAgB,CAAC,+CAA+C,EAAE;gBAChE,KAAK,EAAE,YAAY;aACpB,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;gBAC/B,aAAa,EAAE,CAAC,oBAAoB,CAAC;aACtC,EACD;gBACE,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE;oBACjB;wBACE,IAAI,EAAE,kBAAkB;wBACxB,OAAO,EAAE,oBAAoB,CAAC,EAAE,IAAI,SAAS;wBAC7C,OAAO,EAAE,gDAAgD,YAAY,EAAE;qBACxE;iBACF;aACF,CACF,CAAC;YACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACzC,CAAC;gBAAS,CAAC;YACT,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,aAAa,CAAC,SAAS,CAAC,CAAC;gBAC/B,IAAI,OAAO,EAAE,CAAC;oBACZ,eAAe,EAAE,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBAChE,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,eAAe,CAAC,OAAO,EAAE,4CAA4C,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;QAChG,gBAAgB,CAAC,4CAA4C,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;QAExF,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAK,CAAC,OAAO;SAChC,EACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,+CAA+C,YAAY,EAAE;SACpE,CACF,CAAC;QAEF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;AACH,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/container-task/utils/docker-check.d.ts

@@ -0,0 +1,4 @@
+/**
+ * Check if Docker is available
+ */
+export declare function checkDockerAvailability(): Promise<boolean>;

package/dist/prompt/steps/step-iterate/handlers/container-task/utils/docker-check.js

@@ -0,0 +1,15 @@
+import Docker from 'dockerode';
+/**
+ * Check if Docker is available
+ */
+export async function checkDockerAvailability() {
+    try {
+        const docker = new Docker({ socketPath: '/var/run/docker.sock' });
+        await docker.ping();
+        return true;
+    }
+    catch (error) {
+        return false;
+    }
+}
+//# sourceMappingURL=docker-check.js.map

package/dist/prompt/steps/step-iterate/handlers/container-task/utils/docker-check.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"docker-check.js","sourceRoot":"","sources":["../../../../../../../src/prompt/steps/step-iterate/handlers/container-task/utils/docker-check.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,WAAW,CAAC;AAE/B;;GAEG;AAEH,MAAM,CAAC,KAAK,UAAU,uBAAuB;IAC3C,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,UAAU,EAAE,sBAAsB,EAAE,CAAC,CAAC;QAClE,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/container-task/utils/docker-utils.d.ts

@@ -0,0 +1,50 @@
+import Docker from 'dockerode';
+import { AllowedDockerImage } from '../../../../../function-defs/run-container-task.js';
+/**
+ * Pull a Docker image
+ */
+export declare function pullImage(docker: Docker, image: AllowedDockerImage): Promise<void>;
+/**
+ * Create and start a Docker container
+ */
+export declare function createAndStartContainer(docker: Docker, image: AllowedDockerImage): Promise<Docker.Container>;
+/**
+ * Execute a command in a Docker container
+ */
+export declare function executeCommand(container: Docker.Container, shell: '/bin/sh' | '/bin/bash', command: string, stdin: string | undefined, workingDir: string, abortSignal?: AbortSignal): Promise<{
+    output: string;
+    exitCode: number;
+}>;
+/**
+ * Stop and clean up a Docker container
+ */
+export declare function stopContainer(container: Docker.Container): Promise<void>;
+/**
+ * Clean up orphaned containers left over from previous runs.
+ */
+export declare function cleanupOrphanedContainers(docker: Docker): Promise<void>;
+/**
+ * Copy a file or directory from the host to the container.
+ */
+export declare function copyToContainer(container: Docker.Container, hostPath: string, containerPath: string): Promise<void>;
+export interface ArchiveEntry {
+    name: string;
+    type: 'file' | 'directory';
+}
+/**
+ * Lists the files within a container archive without extracting them.
+ */
+export declare function listFilesInContainerArchive(container: Docker.Container, containerPath: string): Promise<ArchiveEntry[]>;
+/**
+ * Extracts a tar stream to a specified directory, with path validation.
+ * @param tarStream The readable tar stream.
+ * @param destinationHostPath The absolute path on the host to extract files to.
+ * @param pathValidator A function to validate each file path before writing.
+ */
+export declare function extractTarStreamToDirectory(tarStream: NodeJS.ReadableStream, destinationHostPath: string, pathValidator: (filePath: string) => boolean): Promise<void>;
+/**
+ * Copy a file or directory from the container to the host.
+ */
+export declare function copyFromContainer(container: Docker.Container, containerPath: string, hostPath: string): Promise<void>;
+export declare function checkPathExistsInContainer(container: Docker.Container, containerPath: string): Promise<boolean>;
+export declare function getFileContentFromContainer(container: Docker.Container, filePath: string): Promise<string>;

package/dist/prompt/steps/step-iterate/handlers/container-task/utils/docker-utils.js

@@ -0,0 +1,358 @@
+import path from 'path';
+import fs from 'fs';
+import tar from 'tar-stream';
+import { putContainerLog, putSystemMessage } from '../../../../../../main/common/content-bus.js';
+import { rcConfig } from '../../../../../../main/config.js';
+import { cacheContainerId, clearCachedContainerIds, getCachedContainerIds, removeCachedContainerId, } from '../../../../../../files/cache-file.js';
+import { isProjectPath } from '../../../../../../files/path-utils.js';
+import { isIgnoredPath } from '../../../../../../files/find-files.js';
+/**
+ * Pull a Docker image
+ */
+export async function pullImage(docker, image) {
+    putContainerLog('info', `Pulling Docker image: ${image}`, undefined, 'docker');
+    return new Promise((resolve, reject) => {
+        docker
+            .pull(image)
+            .then((stream) => {
+            docker.modem.followProgress(stream, (err) => {
+                if (err) {
+                    putContainerLog('error', `Failed to pull Docker image: ${err.message}`, undefined, 'docker');
+                    reject(err);
+                }
+                else {
+                    putContainerLog('success', `Successfully pulled Docker image: ${image}`, undefined, 'docker');
+                    putSystemMessage(`✅ Successfully pulled Docker image: ${image}`);
+                    resolve();
+                }
+            });
+        })
+            .catch((error) => {
+            putContainerLog('error', `Failed to pull Docker image: ${error.message}`, undefined, 'docker');
+            putSystemMessage(`❌ Failed to pull Docker image: ${error.message}`);
+            reject(error);
+        });
+    });
+}
+/**
+ * Create and start a Docker container
+ */
+export async function createAndStartContainer(docker, image) {
+    const container = await docker.createContainer({
+        Image: image,
+        Tty: true,
+        HostConfig: {
+            AutoRemove: true,
+        },
+    });
+    await container.start();
+    putContainerLog('info', `Container started (ID: ${container.id.substring(0, 12)})`, { image }, 'docker');
+    cacheContainerId(container.id);
+    putSystemMessage(`✅ Container started successfully (ID: ${container.id.substring(0, 12)})`);
+    return container;
+}
+/**
+ * Execute a command in a Docker container
+ */
+export async function executeCommand(container, shell, command, stdin, workingDir, abortSignal) {
+    putContainerLog('info', `Executing command in ${workingDir}: ${command}`, { stdin: !!stdin }, 'command');
+    const exec = await container.exec({
+        Cmd: [shell, '-c', command],
+        WorkingDir: workingDir,
+        AttachStdin: true,
+        Tty: false,
+        AttachStdout: true,
+        AttachStderr: true,
+    });
+    const stream = await exec.start({ hijack: true, stdin: true });
+    if (stdin) {
+        stream.write(stdin);
+    }
+    stream.end();
+    abortSignal?.addEventListener('abort', async () => {
+        const killExec = await container.exec({
+            Cmd: ['pkill', '-9', '-f', command],
+            AttachStdout: true,
+            AttachStderr: true,
+        });
+        await killExec.start({});
+        stream.destroy();
+    });
+    let output = '';
+    // Collect output from the stream
+    await new Promise((resolve) => {
+        stream.on('data', (chunk) => {
+            // Docker stream format: first 8 bytes are header, rest is data
+            const data = chunk.slice(8).toString();
+            output += data;
+        });
+        stream.on('end', () => {
+            resolve();
+        });
+        stream.on('error', () => {
+            resolve();
+        });
+        abortSignal?.addEventListener('abort', () => {
+            resolve();
+        });
+    });
+    if (abortSignal?.aborted) {
+        output += '\n\nAborted command execution';
+    }
+    const inspect = await exec.inspect();
+    putContainerLog('info', `Command finished with exit code: ${inspect.ExitCode || 0}`, { command }, 'command');
+    return { output: output.trim(), exitCode: inspect.ExitCode || 0 };
+}
+/**
+ * Stop and clean up a Docker container
+ */
+export async function stopContainer(container) {
+    try {
+        await container.stop();
+        putContainerLog('info', 'Container stopped successfully', undefined, 'docker');
+        putSystemMessage('🛑 Container stopped successfully');
+    }
+    catch (error) {
+        putContainerLog('info', 'Container cleanup completed', undefined, 'docker');
+        putSystemMessage('🛑 Container cleanup completed');
+    }
+    finally {
+        removeCachedContainerId(container.id);
+    }
+}
+/**
+ * Clean up orphaned containers left over from previous runs.
+ */
+export async function cleanupOrphanedContainers(docker) {
+    const containerIds = getCachedContainerIds();
+    if (containerIds.length === 0) {
+        return;
+    }
+    putContainerLog('info', `Found ${containerIds.length} potentially orphaned containers. Cleaning up...`);
+    putSystemMessage(`🧹 Found ${containerIds.length} potentially orphaned containers. Cleaning up...`);
+    const cleanupPromises = containerIds.map(async (id) => {
+        try {
+            const container = docker.getContainer(id);
+            const inspectInfo = await container.inspect();
+            if (inspectInfo.State.Running) {
+                await container.stop();
+                putContainerLog('info', `Stopped orphaned container ${id.substring(0, 12)}`);
+                putSystemMessage(`🛑 Stopped orphaned container ${id.substring(0, 12)}`);
+            }
+        }
+        catch (error) {
+            // Ignore errors (e.g., container not found)
+        }
+    });
+    await Promise.all(cleanupPromises);
+    clearCachedContainerIds();
+    putContainerLog('success', 'Orphaned container cleanup complete.');
+    putSystemMessage('✅ Orphaned container cleanup complete.');
+}
+/**
+ * Validates that the host path is within the project root directory.
+ * @param hostPath The path on the host machine. Can be absolute or relative to the project root.
+ * @returns The absolute, validated path.
+ */
+function validateHostPath(hostPath) {
+    const absoluteHostPath = path.resolve(rcConfig.rootDir, hostPath);
+    if (!absoluteHostPath.startsWith(rcConfig.rootDir)) {
+        throw new Error(`Invalid host path: ${hostPath} is outside the project root directory.`);
+    }
+    return absoluteHostPath;
+}
+/**
+ * Recursively adds files and directories to a tar pack stream.
+ * @param pack The tar-stream pack instance.
+ * @param directoryPath The absolute path of the directory to add.
+ * @param relativeTo The base path for entries in the tar archive.
+ */
+function addDirectoryToPack(pack, directoryPath, relativeTo) {
+    if (isIgnoredPath(directoryPath)) {
+        return;
+    }
+    const items = fs.readdirSync(directoryPath);
+    const dirStats = fs.lstatSync(directoryPath);
+    pack.entry({
+        name: relativeTo.endsWith('/') ? relativeTo : relativeTo + '/',
+        type: 'directory',
+        mode: dirStats.mode & 0o777,
+        mtime: dirStats.mtime,
+    }, () => { });
+    for (const item of items) {
+        const fullPath = path.join(directoryPath, item);
+        const tarPath = path.join(relativeTo, item);
+        const stats = fs.lstatSync(fullPath);
+        if (stats.isDirectory()) {
+            addDirectoryToPack(pack, fullPath, tarPath);
+        }
+        else if (isIgnoredPath(fullPath)) {
+            // Skip ignored files
+        }
+        else if (stats.isSymbolicLink()) {
+            const linkname = fs.readlinkSync(fullPath);
+            pack.entry({
+                name: tarPath,
+                type: 'symlink',
+                linkname,
+                mode: 0o777,
+                mtime: stats.mtime,
+            }, () => { });
+        }
+        else if (stats.isFile()) {
+            const data = fs.readFileSync(fullPath);
+            pack.entry({
+                name: tarPath,
+                type: 'file',
+                mode: stats.mode & 0o777,
+                size: data.length,
+                mtime: stats.mtime,
+            }, data);
+        }
+    }
+}
+/**
+ * Copy a file or directory from the host to the container.
+ */
+export async function copyToContainer(container, hostPath, containerPath) {
+    try {
+        const absoluteHostPath = validateHostPath(hostPath);
+        putContainerLog('info', `Copying from host:${absoluteHostPath} to container:${containerPath}`, undefined, 'copy');
+        putSystemMessage(`📦 Copying from host:${absoluteHostPath} to container:${containerPath}`);
+        const pack = tar.pack();
+        const stats = fs.statSync(absoluteHostPath);
+        if (stats.isDirectory()) {
+            addDirectoryToPack(pack, absoluteHostPath, '');
+        }
+        else if (!isIgnoredPath(absoluteHostPath)) {
+            pack.entry({ name: path.basename(absoluteHostPath) }, fs.readFileSync(absoluteHostPath));
+        }
+        pack.finalize();
+        await container.putArchive(pack, { path: containerPath });
+        putContainerLog('success', `Successfully copied to container.`, undefined, 'copy');
+        putSystemMessage(`✅ Successfully copied to container.`);
+    }
+    catch (error) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        putContainerLog('error', `Failed to copy to container: ${errorMessage}`, undefined, 'copy');
+        putSystemMessage(`❌ Failed to copy to container: ${errorMessage}`);
+        throw error;
+    }
+}
+/**
+ * Lists the files within a container archive without extracting them.
+ */
+export async function listFilesInContainerArchive(container, containerPath) {
+    const entries = [];
+    try {
+        const stream = await container.getArchive({ path: containerPath });
+        const extract = tar.extract();
+        extract.on('entry', (header, stream, next) => {
+            if (header.type === 'file' || header.type === 'directory') {
+                entries.push({
+                    name: header.name,
+                    type: header.type,
+                });
+            }
+            stream.on('end', () => next());
+            stream.resume(); // Discard the stream's data
+        });
+        await new Promise((resolve, reject) => {
+            extract.on('finish', resolve);
+            extract.on('error', reject);
+            stream.pipe(extract);
+        });
+        putContainerLog('debug', `Listed ${entries.length} entries in archive at: ${containerPath}`, undefined, 'copy');
+        return entries;
+    }
+    catch (error) {
+        // If the path doesn't exist, getArchive throws an error. Return an empty list.
+        return [];
+    }
+}
+/**
+ * Extracts a tar stream to a specified directory, with path validation.
+ * @param tarStream The readable tar stream.
+ * @param destinationHostPath The absolute path on the host to extract files to.
+ * @param pathValidator A function to validate each file path before writing.
+ */
+export async function extractTarStreamToDirectory(tarStream, destinationHostPath, pathValidator) {
+    const extract = tar.extract();
+    extract.on('entry', (header, stream, next) => {
+        const filePath = path.join(destinationHostPath, header.name);
+        const dir = path.dirname(filePath);
+        if (!pathValidator(filePath)) {
+            const errorMessage = `Refusing to write outside project root: ${filePath}`;
+            putContainerLog('error', errorMessage, undefined, 'copy');
+            // Pass an error to next() to stop the extraction process
+            stream.on('end', () => next(new Error(errorMessage)));
+            stream.resume();
+            return;
+        }
+        if (header.type === 'directory') {
+            if (!fs.existsSync(filePath)) {
+                fs.mkdirSync(filePath, { recursive: true });
+            }
+            stream.on('end', () => next());
+            stream.resume();
+            return;
+        }
+        // Ensure directory exists for the file
+        if (!fs.existsSync(dir)) {
+            fs.mkdirSync(dir, { recursive: true });
+        }
+        const writeStream = fs.createWriteStream(filePath);
+        stream.pipe(writeStream);
+        writeStream.on('finish', () => next());
+        writeStream.on('error', (err) => {
+            console.error(`Error writing file ${filePath}:`, err);
+            next(err); // Pass error to tar-stream
+        });
+        stream.on('error', (err) => {
+            console.error(`Error in tar stream for ${filePath}:`, err);
+            next(err); // Pass error to tar-stream
+        });
+    });
+    return new Promise((resolve, reject) => {
+        tarStream.on('error', reject);
+        extract.on('finish', resolve);
+        extract.on('error', reject);
+        tarStream.pipe(extract);
+    });
+}
+/**
+ * Copy a file or directory from the container to the host.
+ */
+export async function copyFromContainer(container, containerPath, hostPath) {
+    try {
+        const absoluteHostPath = validateHostPath(hostPath);
+        putContainerLog('info', `Copying from container:${containerPath} to host:${absoluteHostPath}`, undefined, 'copy');
+        putSystemMessage(`📦 Copying from container:${containerPath} to host:${absoluteHostPath}`);
+        const stream = await container.getArchive({ path: containerPath });
+        await extractTarStreamToDirectory(stream, absoluteHostPath, isProjectPath);
+        putContainerLog('success', 'Successfully copied from container.', undefined, 'copy');
+        putSystemMessage(`✅ Successfully copied from container.`);
+    }
+    catch (error) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        putContainerLog('error', `Failed to copy from container: ${errorMessage}`, undefined, 'copy');
+        putSystemMessage(`❌ Failed to copy from container: ${errorMessage}`);
+        throw error;
+    }
+}
+export async function checkPathExistsInContainer(container, containerPath) {
+    const check = await executeCommand(container, '/bin/sh', `ls ${containerPath}`, '', '/');
+    return check.exitCode === 0;
+}
+export async function getFileContentFromContainer(container, filePath) {
+    const { output, exitCode } = await executeCommand(container, '/bin/sh', `cat "${filePath}"`, '', '/');
+    if (exitCode !== 0) {
+        // If cat fails, it might be because the file doesn't exist, which is a valid case for patching (creating a new file).
+        if (output.includes('No such file or directory')) {
+            return '';
+        }
+        throw new Error(`Failed to read file content: ${output}`);
+    }
+    return output;
+}
+//# sourceMappingURL=docker-utils.js.map