@gaunt-sloth/agent 0.1.8 → 2.0.0-alpha.0

package/dist/tools/GthDevToolkit.js

@@ -0,0 +1,189 @@
+/**
+ * @module GthDevToolkit
+ */
+import { BaseToolkit, tool } from '@langchain/core/tools';
+import { z } from 'zod';
+import { spawn } from 'child_process';
+import path from 'node:path';
+import { displayInfo, displayError } from '@gaunt-sloth/core/utils/consoleUtils.js';
+import { stdout } from '@gaunt-sloth/core/utils/systemUtils.js';
+// Helper function to create a tool with dev type
+function createGthTool(fn, config, gthDevType) {
+    const toolInstance = tool(fn, config);
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    toolInstance.gthDevType = gthDevType;
+    return toolInstance;
+}
+// Schema definitions for built-in tools
+const RunTestsArgsSchema = z.object({});
+const RunLintArgsSchema = z.object({});
+const RunBuildArgsSchema = z.object({});
+const RunSingleTestArgsSchema = z.object({
+    testPath: z.string().describe('Relative path to the test file to run'),
+});
+const TEST_PATH_PLACEHOLDER = '${testPath}';
+export default class GthDevToolkit extends BaseToolkit {
+    tools;
+    commands;
+    constructor(commands = {}) {
+        super();
+        this.commands = commands;
+        this.tools = this.createTools();
+    }
+    /**
+     * Get tools filtered by operation type
+     */
+    getFilteredTools(allowedOperations) {
+        return this.tools.filter((tool) => {
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const toolType = tool.gthDevType;
+            return allowedOperations.includes(toolType);
+        });
+    }
+    /**
+     * Validate parameter value to prevent security issues
+     */
+    validateParameterValue(paramValue, paramName) {
+        // Check for absolute paths
+        if (path.isAbsolute(paramValue)) {
+            throw new Error(`Absolute paths are not allowed for parameter '${paramName}'`);
+        }
+        // Check for directory traversal attempts
+        if (paramValue.includes('..') || paramValue.includes('\\..\\') || paramValue.includes('/../')) {
+            throw new Error(`Directory traversal attempts are not allowed in parameter '${paramName}'`);
+        }
+        // Check for pipe attempts and other shell injection
+        if (paramValue.includes('|') ||
+            paramValue.includes('&') ||
+            paramValue.includes(';') ||
+            paramValue.includes('`') ||
+            paramValue.includes('$') ||
+            paramValue.includes('$(') ||
+            paramValue.includes('\n') ||
+            paramValue.includes('\r')) {
+            throw new Error(`Shell injection attempts are not allowed in parameter '${paramName}'`);
+        }
+        // Check for null bytes
+        if (paramValue.includes('\0')) {
+            throw new Error(`Null bytes are not allowed in parameter '${paramName}'`);
+        }
+        // Normalize the path to remove any redundant separators
+        const normalizedValue = path.normalize(paramValue);
+        // Double-check after normalization
+        if (normalizedValue.includes('..')) {
+            throw new Error(`Directory traversal attempts are not allowed in parameter '${paramName}'`);
+        }
+        return normalizedValue;
+    }
+    /**
+     * Build the command for running a single test file
+     */
+    buildSingleTestCommand(testPath) {
+        if (this.commands.run_single_test) {
+            if (this.commands.run_single_test.includes(TEST_PATH_PLACEHOLDER)) {
+                // Interpolate if placeholder is available
+                return this.commands.run_single_test.replace(TEST_PATH_PLACEHOLDER, testPath);
+            }
+            else {
+                // Concatenate if no placeholder
+                return `${this.commands.run_single_test} ${testPath}`;
+            }
+        }
+        else {
+            throw new Error('No test command configured');
+        }
+    }
+    async executeCommand(command, toolName) {
+        displayInfo(`\n🔧 Executing ${toolName}: ${command}`);
+        return new Promise((resolve, reject) => {
+            const child = spawn(command, {
+                shell: true,
+            });
+            let output = '';
+            // Capture output if available (when stdio is not 'inherit')
+            if (child.stdout) {
+                child.stdout.on('data', (data) => {
+                    const chunk = data.toString();
+                    stdout.write(chunk);
+                    output += chunk;
+                });
+            }
+            if (child.stderr) {
+                child.stderr.on('data', (data) => {
+                    const chunk = data.toString();
+                    stdout.write(chunk);
+                    output += chunk;
+                });
+            }
+            child.on('close', (code) => {
+                if (code === 0) {
+                    resolve(`Executing '${command}'...\n\n` +
+                        `<COMMAND_OUTPUT>\n` +
+                        output +
+                        `</COMMAND_OUTPUT>\n` +
+                        `\n\nCommand '${command}' completed successfully`);
+                }
+                else {
+                    resolve(`Executing '${command}'...\n\n` +
+                        `<COMMAND_OUTPUT>\n` +
+                        output +
+                        `</COMMAND_OUTPUT>\n` +
+                        `\n\nCommand '${command}' exited with code ${code}`);
+                }
+            });
+            child.on('error', (error) => {
+                const errorMsg = `Failed to execute command '${command}': ${error.message}`;
+                displayError(errorMsg);
+                reject(new Error(errorMsg));
+            });
+        });
+    }
+    createTools() {
+        const tools = [];
+        if (this.commands.run_tests) {
+            tools.push(createGthTool(async (_args) => {
+                return await this.executeCommand(this.commands.run_tests, 'run_tests');
+            }, {
+                name: 'run_tests',
+                description: 'Execute the test suite for this project. Runs the configured test command and returns the output.' +
+                    `\nThe configured command is [${this.commands.run_tests}].`,
+                schema: RunTestsArgsSchema,
+            }, 'execute'));
+        }
+        if (this.commands.run_single_test) {
+            tools.push(createGthTool(async (args) => {
+                const validatedPath = this.validateParameterValue(args.testPath, 'testPath');
+                const command = this.buildSingleTestCommand(validatedPath);
+                return await this.executeCommand(command, 'run_single_test');
+            }, {
+                name: 'run_single_test',
+                description: 'Execute a single test file. Runs the configured test command with the specified test file path. ' +
+                    'The test path must be relative and cannot contain directory traversal attempts or shell injection. ' +
+                    `\nThe base command is [${this.commands.run_single_test}].`,
+                schema: RunSingleTestArgsSchema,
+            }, 'execute'));
+        }
+        if (this.commands.run_lint) {
+            tools.push(createGthTool(async (_args) => {
+                return await this.executeCommand(this.commands.run_lint, 'run_lint');
+            }, {
+                name: 'run_lint',
+                description: 'Run the linter on the project code. Executes the configured lint command and returns any linting errors or warnings.' +
+                    `\nThe configured command is [${this.commands.run_lint}].`,
+                schema: RunLintArgsSchema,
+            }, 'execute'));
+        }
+        if (this.commands.run_build) {
+            tools.push(createGthTool(async (_args) => {
+                return await this.executeCommand(this.commands.run_build, 'run_build');
+            }, {
+                name: 'run_build',
+                description: 'Build the project. Executes the configured build command and returns the build output.' +
+                    `\nThe configured command is [${this.commands.run_build}].`,
+                schema: RunBuildArgsSchema,
+            }, 'execute'));
+        }
+        return tools;
+    }
+}
+//# sourceMappingURL=GthDevToolkit.js.map

package/dist/tools/GthDevToolkit.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"GthDevToolkit.js","sourceRoot":"","sources":["../../src/tools/GthDevToolkit.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,WAAW,EAA2B,IAAI,EAAE,MAAM,uBAAuB,CAAC;AACnF,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AACtC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,yCAAyC,CAAC;AAEpF,OAAO,EAAE,MAAM,EAAE,MAAM,wCAAwC,CAAC;AAEhE,iDAAiD;AACjD,SAAS,aAAa,CACpB,EAAyC,EACzC,MAIC,EACD,UAAqB;IAErB,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;IACtC,8DAA8D;IAC7D,YAAoB,CAAC,UAAU,GAAG,UAAU,CAAC;IAC9C,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,wCAAwC;AACxC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;AACxC,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;AACvC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;AACxC,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uCAAuC,CAAC;CACvE,CAAC,CAAC;AAEH,MAAM,qBAAqB,GAAG,aAAa,CAAC;AAE5C,MAAM,CAAC,OAAO,OAAO,aAAc,SAAQ,WAAW;IACpD,KAAK,CAA4B;IACzB,QAAQ,CAAoB;IAEpC,YAAY,WAA8B,EAAE;QAC1C,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,iBAA8B;QAC7C,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;YAChC,8DAA8D;YAC9D,MAAM,QAAQ,GAAI,IAAY,CAAC,UAAU,CAAC;YAC1C,OAAO,iBAAiB,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,sBAAsB,CAAC,UAAkB,EAAE,SAAiB;QAC1D,2BAA2B;QAC3B,IAAI,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,iDAAiD,SAAS,GAAG,CAAC,CAAC;QACjF,CAAC;QAED,yCAAyC;QACzC,IAAI,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9F,MAAM,IAAI,KAAK,CAAC,8DAA8D,SAAS,GAAG,CAAC,CAAC;QAC9F,CAAC;QAED,oDAAoD;QACpD,IACE,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC;YACxB,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC;YACxB,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC;YACxB,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC;YACxB,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC;YACxB,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC;YACzB,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC;YACzB,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,EACzB,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,0DAA0D,SAAS,GAAG,CAAC,CAAC;QAC1F,CAAC;QAED,uBAAuB;QACvB,IAAI,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,4CAA4C,SAAS,GAAG,CAAC,CAAC;QAC5E,CAAC;QAED,wDAAwD;QACxD,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAEnD,mCAAmC;QACnC,IAAI,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CAAC,8DAA8D,SAAS,GAAG,CAAC,CAAC;QAC9F,CAAC;QAED,OAAO,eAAe,CAAC;IACzB,CAAC;IAED;;OAEG;IACK,sBAAsB,CAAC,QAAgB;QAC7C,IAAI,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC;YAClC,IAAI,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,QAAQ,CAAC,qBAAqB,CAAC,EAAE,CAAC;gBAClE,0CAA0C;gBAC1C,OAAO,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,OAAO,CAAC,qBAAqB,EAAE,QAAQ,CAAC,CAAC;YAChF,CAAC;iBAAM,CAAC;gBACN,gCAAgC;gBAChC,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,eAAe,IAAI,QAAQ,EAAE,CAAC;YACxD,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,OAAe,EAAE,QAAgB;QAC5D,WAAW,CAAC,kBAAkB,QAAQ,KAAK,OAAO,EAAE,CAAC,CAAC;QAEtD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE;gBAC3B,KAAK,EAAE,IAAI;aACZ,CAAC,CAAC;YAEH,IAAI,MAAM,GAAG,EAAE,CAAC;YAEhB,4DAA4D;YAC5D,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;gBACjB,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;oBAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;oBAC9B,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBACpB,MAAM,IAAI,KAAK,CAAC;gBAClB,CAAC,CAAC,CAAC;YACL,CAAC;YAED,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;gBACjB,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;oBAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;oBAC9B,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBACpB,MAAM,IAAI,KAAK,CAAC;gBAClB,CAAC,CAAC,CAAC;YACL,CAAC;YAED,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;gBACzB,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;oBACf,OAAO,CACL,cAAc,OAAO,UAAU;wBAC7B,oBAAoB;wBACpB,MAAM;wBACN,qBAAqB;wBACrB,gBAAgB,OAAO,0BAA0B,CACpD,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,OAAO,CACL,cAAc,OAAO,UAAU;wBAC7B,oBAAoB;wBACpB,MAAM;wBACN,qBAAqB;wBACrB,gBAAgB,OAAO,sBAAsB,IAAI,EAAE,CACtD,CAAC;gBACJ,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBAC1B,MAAM,QAAQ,GAAG,8BAA8B,OAAO,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;gBAC5E,YAAY,CAAC,QAAQ,CAAC,CAAC;gBACvB,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;YAC9B,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,WAAW;QACjB,MAAM,KAAK,GAA8B,EAAE,CAAC;QAE5C,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;YAC5B,KAAK,CAAC,IAAI,CACR,aAAa,CACX,KAAK,EAAE,KAAyC,EAAmB,EAAE;gBACnE,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAU,EAAE,WAAW,CAAC,CAAC;YAC1E,CAAC,EACD;gBACE,IAAI,EAAE,WAAW;gBACjB,WAAW,EACT,mGAAmG;oBACnG,gCAAgC,IAAI,CAAC,QAAQ,CAAC,SAAU,IAAI;gBAC9D,MAAM,EAAE,kBAAkB;aAC3B,EACD,SAAS,CACV,CACF,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC;YAClC,KAAK,CAAC,IAAI,CACR,aAAa,CACX,KAAK,EAAE,IAA6C,EAAmB,EAAE;gBACvE,MAAM,aAAa,GAAG,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;gBAC7E,MAAM,OAAO,GAAG,IAAI,CAAC,sBAAsB,CAAC,aAAa,CAAC,CAAC;gBAC3D,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,iBAAiB,CAAC,CAAC;YAC/D,CAAC,EACD;gBACE,IAAI,EAAE,iBAAiB;gBACvB,WAAW,EACT,kGAAkG;oBAClG,qGAAqG;oBACrG,0BAA0B,IAAI,CAAC,QAAQ,CAAC,eAAe,IAAI;gBAC7D,MAAM,EAAE,uBAAuB;aAChC,EACD,SAAS,CACV,CACF,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;YAC3B,KAAK,CAAC,IAAI,CACR,aAAa,CACX,KAAK,EAAE,KAAwC,EAAmB,EAAE;gBAClE,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAS,EAAE,UAAU,CAAC,CAAC;YACxE,CAAC,EACD;gBACE,IAAI,EAAE,UAAU;gBAChB,WAAW,EACT,sHAAsH;oBACtH,gCAAgC,IAAI,CAAC,QAAQ,CAAC,QAAS,IAAI;gBAC7D,MAAM,EAAE,iBAAiB;aAC1B,EACD,SAAS,CACV,CACF,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;YAC5B,KAAK,CAAC,IAAI,CACR,aAAa,CACX,KAAK,EAAE,KAAyC,EAAmB,EAAE;gBACnE,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAU,EAAE,WAAW,CAAC,CAAC;YAC1E,CAAC,EACD;gBACE,IAAI,EAAE,WAAW;gBACjB,WAAW,EACT,wFAAwF;oBACxF,gCAAgC,IAAI,CAAC,QAAQ,CAAC,SAAU,IAAI;gBAC9D,MAAM,EAAE,kBAAkB;aAC3B,EACD,SAAS,CACV,CACF,CAAC;QACJ,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;CACF"}

package/dist/tools/GthFileSystemToolkit.d.ts

@@ -0,0 +1,36 @@
+import { BaseToolkit, StructuredToolInterface } from '@langchain/core/tools';
+import type { BinaryFormatConfig } from '@gaunt-sloth/core/config.js';
+export interface GthFileSystemToolkitOptions {
+    allowedDirectories?: string[];
+    aiignoreConfig?: {
+        enabled?: boolean;
+        patterns?: string[];
+    };
+    binaryFormats?: false | BinaryFormatConfig[];
+}
+export default class GthFileSystemToolkit extends BaseToolkit {
+    tools: StructuredToolInterface[];
+    private allowedDirectories;
+    private aiignoreConfig?;
+    private binaryFormats?;
+    constructor(options?: GthFileSystemToolkitOptions);
+    /**
+     * Get tools filtered by operation type
+     */
+    getFilteredTools(allowedOperations: ('read' | 'write')[]): StructuredToolInterface[];
+    private normalizePath;
+    private isProtectedDirectory;
+    private expandHome;
+    private sanitizeRequestedPath;
+    private validatePath;
+    private getFileStats;
+    private searchFiles;
+    private normalizeLineEndings;
+    private createUnifiedDiff;
+    private applyFileEdits;
+    private formatSize;
+    private tailFile;
+    private headFile;
+    private createReadBinaryTool;
+    private createTools;
+}