cloving 0.1.19 → 0.1.21

package/dist/cloving_gpt/index.js

@@ -13,6 +13,9 @@ const mistral_1 = require("./adapters/mistral");
 const ollama_1 = require("./adapters/ollama");
 const gemini_1 = require("./adapters/gemini");
 const config_utils_1 = require("../utils/config_utils");
+const fs_1 = __importDefault(require("fs"));
+const os_1 = __importDefault(require("os"));
+const path_1 = __importDefault(require("path"));
 class ClovingGPT {
     constructor(options) {
         const { model: partialModel } = options;
@@ -52,51 +55,53 @@ class ClovingGPT {
     }
     async reviewPrompt(prompt, endpoint) {
         if (this.silent)
-            return;
+            return prompt;
         const tokenCount = Math.ceil(prompt.length / 4).toLocaleString();
         const reviewPrompt = await (0, prompts_1.confirm)({
-            message: `Do you want to review the ~${tokenCount} token prompt before sending it to ${endpoint}?`,
+            message: `Do you want to review/edit the ~${tokenCount} token prompt before sending it to ${endpoint}?`,
             default: true
         });
         if (reviewPrompt) {
-            const less = (0, child_process_1.spawn)('less', ['-R'], { stdio: ['pipe', process_1.default.stdout, process_1.default.stderr] });
-            less.stdin.write(prompt);
-            less.stdin.end();
-            await new Promise((resolve, reject) => {
-                less.on('close', async (code) => {
+            const tempFile = path_1.default.join(os_1.default.tmpdir(), `cloving_prompt_${Date.now()}.txt`);
+            fs_1.default.writeFileSync(tempFile, prompt);
+            const editor = process_1.default.env.EDITOR || 'nano';
+            const editProcess = (0, child_process_1.spawn)(editor, [tempFile], { stdio: 'inherit' });
+            return new Promise((resolve, reject) => {
+                editProcess.on('close', async (code) => {
                     if (code !== 0) {
-                        console.error('less command exited with an error');
-                        reject(new Error('Less command failed'));
+                        console.error(`${editor} exited with code ${code}`);
+                        fs_1.default.unlinkSync(tempFile);
+                        reject(new Error(`Editor exited with non-zero code`));
                         return;
                     }
-                    const confirmContinue = await (0, prompts_1.confirm)({
-                        message: 'Do you want to continue?',
-                        default: true
-                    });
-                    if (!confirmContinue) {
-                        console.error('Operation cancelled');
-                        process_1.default.exit(1);
+                    const editedPrompt = fs_1.default.readFileSync(tempFile, 'utf-8').trim();
+                    fs_1.default.unlinkSync(tempFile);
+                    if (editedPrompt === '') {
+                        console.log('Prompt is empty.');
+                        resolve(null);
+                        return;
                     }
-                    resolve();
+                    resolve(editedPrompt);
                 });
-                less.stdin.on('error', (err) => {
-                    if (isNodeError(err) && err.code === 'EPIPE') {
-                        resolve();
-                    }
-                    else {
-                        console.error('Pipeline error:', err);
-                        reject(err);
-                    }
+                editProcess.on('error', (err) => {
+                    console.error('Error launching editor:', err);
+                    fs_1.default.unlinkSync(tempFile);
+                    reject(err);
                 });
             });
         }
+        return prompt;
     }
     async generateText(request) {
         request.temperature || (request.temperature = this.temperature);
         const endpoint = this.adapter.getEndpoint();
         const payload = this.adapter.getPayload(request);
         const headers = this.adapter.getHeaders(this.apiKey);
-        await this.reviewPrompt(request.prompt, endpoint);
+        const reviewedPrompt = await this.reviewPrompt(request.prompt, endpoint);
+        if (reviewedPrompt === null) {
+            throw new Error('Operation cancelled by user');
+        }
+        request.prompt = reviewedPrompt;
         try {
             const response = await axios_1.default.post(endpoint, payload, { headers });
             return this.adapter.extractResponse(response.data);
@@ -111,7 +116,4 @@ class ClovingGPT {
         }
     }
 }
-function isNodeError(error) {
-    return error instanceof Error && 'code' in error;
-}
 exports.default = ClovingGPT;

package/dist/commands/generate/code.js

@@ -16,7 +16,11 @@ const generateCodePrompt = (prompt, files, contextFiles, previousCode) => {
     const specialFileContents = (0, command_utils_1.collectSpecialFileContents)();
     const specialFiles = Object.keys(specialFileContents).map((file) => `### Contents of ${file}\n\n${JSON.stringify(specialFileContents[file], null, 2)}\n\n`).join('\n');
     const contextFileContents = Object.keys(contextFiles).map((file) => `### Contents of ${file}\n\n${contextFiles[file]}\n\n`).join('\n');
-    let promptText = `### Description of App
+    let promptText = `### Request
+
+Generate code that does the following: ${prompt}
+
+### Description of App
 
 ${JSON.stringify((0, config_utils_1.getClovingConfig)(), null, 2)}
 

package/dist/commands/generate/review.js

@@ -17,7 +17,60 @@ const review = async (options) => {
     let { prompt, files } = options;
     options.silent = (0, config_utils_1.getConfig)(options).globalSilent || false;
     const gpt = new cloving_gpt_1.default(options);
-    let promptText = '';
+    let promptText = `### Prompt
+
+${prompt || 'Please provide a code review.'}
+
+### Example of a well-structured response
+
+    # Code Review: Enhanced Code Generation and File Handling
+
+    ## Changes Overview
+
+    These changes primarily focus on improving the code generation process and adding more robust file handling capabilities. The main modifications include:
+
+    1. Enhancing the code generation prompt with a structured example response.
+    2. Implementing a new function to extract file names and contents from the generated code.
+    3. Adding a new option to save generated code directly to files.
+    4. Improving the copy-to-clipboard functionality to allow copying individual files.
+    5. Removing an unused import in the review.ts file.
+
+    ## Reason for Changes
+
+    The changes aim to enhance the usability and flexibility of the code generation tool by providing clearer prompts, better file management, and more control over the generated code.
+
+    ## Detailed Description
+
+    1. **Enhanced Code Generation Prompt** - The \`generateCodePrompt\` function now includes a structured example of a well-formatted response. This helps guide the AI to produce more consistent and usable output, making it easier to parse and use the generated code.
+    2. **File Extraction Function** - A new function \`extractFilesAndContent\` has been added to parse the raw code command and extract individual file names and their contents. This enables more granular control over the generated code, allowing for file-specific operations.
+    3. **Save to File Option** - A new 'Save Source Code to Files' option has been added to the user action menu. This feature allows users to save generated code directly to files in the appropriate directory structure.
+    4. **Improved Copy to Clipboard** - The 'Copy Source Code to Clipboard' option has been enhanced to allow users to choose specific files to copy, rather than copying all generated code at once.
+    5. **Removed Unused Import** - The unused import of \`execFileSync\` from 'child_process' has been removed from the review.ts file, improving code cleanliness.
+
+    ## Potential Bugs and Recommended Fixes
+
+    ### 1. Possible undefined access in \`extractFilesAndContent\`
+
+    \`\`\`typescript
+    const contentMatch = regex.exec(rawCodeCommand)
+    if (contentMatch) {
+      files.push(fileName)
+      let content = contentMatch[1]
+      // ...
+    }
+    \`\`\`
+
+    There's a potential for \`contentMatch[1]\` to be undefined if the regex doesn't capture any groups. To fix this, add a null check:
+
+    \`\`\`typescript
+    const contentMatch = regex.exec(rawCodeCommand)
+    if (contentMatch && contentMatch[1]) {
+      files.push(fileName)
+      let content = contentMatch[1]
+      // ...
+    }
+    \`\`\`
+`;
     let contextFiles = {};
     try {
         // Define the prompt for analysis
@@ -42,7 +95,8 @@ const review = async (options) => {
                 }
             }
             const contextFileContents = Object.keys(contextFiles).map((file) => `### Contents of ${file}\n\n${contextFiles[file]}\n\n`).join('\n');
-            promptText = `### Description of App
+            promptText += `
+### Description of App
 
 ${JSON.stringify((0, config_utils_1.getClovingConfig)(), null, 2)}
 
@@ -56,57 +110,17 @@ ${contextFileContents}
 ${gitDiff}
 ==== end diff =====`;
         }
-        promptText += `
-
-### Example of a well-structured response
-
-# Code Review: Enhanced Code Generation and File Handling
-
-## Changes Overview
-
-These changes primarily focus on improving the code generation process and adding more robust file handling capabilities. The main modifications include:
-
-1. Enhancing the code generation prompt with a structured example response.
-2. Implementing a new function to extract file names and contents from the generated code.
-3. Adding a new option to save generated code directly to files.
-4. Improving the copy-to-clipboard functionality to allow copying individual files.
-5. Removing an unused import in the review.ts file.
-
-## Reason for Changes
-
-The changes aim to enhance the usability and flexibility of the code generation tool by providing clearer prompts, better file management, and more control over the generated code.
-
-## Detailed Description
-
-1. **Enhanced Code Generation Prompt** - The \`generateCodePrompt\` function now includes a structured example of a well-formatted response. This helps guide the AI to produce more consistent and usable output, making it easier to parse and use the generated code.
-2. **File Extraction Function** - A new function \`extractFilesAndContent\` has been added to parse the raw code command and extract individual file names and their contents. This enables more granular control over the generated code, allowing for file-specific operations.
-3. **Save to File Option** - A new 'Save Source Code to Files' option has been added to the user action menu. This feature allows users to save generated code directly to files in the appropriate directory structure.
-4. **Improved Copy to Clipboard** - The 'Copy Source Code to Clipboard' option has been enhanced to allow users to choose specific files to copy, rather than copying all generated code at once.
-5. **Removed Unused Import** - The unused import of \`execFileSync\` from 'child_process' has been removed from the review.ts file, improving code cleanliness.
-
-## Potential Bugs and Recommended Fixes
-
-### 1. Possible undefined access in \`extractFilesAndContent\`
-
-\`\`\`typescript
-const contentMatch = regex.exec(rawCodeCommand)
-if (contentMatch) {
-  files.push(fileName)
-  let content = contentMatch[1]
-  // ...
-}
-\`\`\`
+        if (files) {
+            promptText += `
 
-There's a potential for \`contentMatch[1]\` to be undefined if the regex doesn't capture any groups. To fix this, add a null check:
+### Request
 
-\`\`\`typescript
-const contentMatch = regex.exec(rawCodeCommand)
-if (contentMatch && contentMatch[1]) {
-  files.push(fileName)
-  let content = contentMatch[1]
-  // ...
-}
-\`\`\`
+I would like you to explain the code and document a description of it.
+List any bugs in the new code as well as recommended fixes for those bugs with code examples.
+Format the output of this code review in Markdown format.`;
+        }
+        else {
+            promptText += `
 
 ### Request
 
@@ -114,10 +128,12 @@ Do not use any data from the example response structure, only use the structure.
 I would like you to explain why these change are being made and document a description of these changes.
 Also list any bugs in the new code as well as recommended fixes for those bugs with code examples.
 Format the output of this code review in Markdown format.`;
+        }
         if (prompt) {
             promptText += `
 
 ### Additional Context
+
 ${prompt}`;
         }
         // get the analysis

package/dist/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cloving",
-  "version": "0.1.19",
+  "version": "0.1.20",
   "packageManager": "yarn@1.22.22",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -51,6 +51,7 @@
     "eslint-plugin-promise": "^7.0.0",
     "ignore": "^5.3.1",
     "ini": "^4.1.3",
+    "isbinaryfile": "^5.0.2",
     "nock": "^13.5.4",
     "vite": "^5.3.5",
     "vite-node": "^2.0.5"

package/dist/utils/command_utils.js

@@ -8,6 +8,7 @@ const child_process_1 = require("child_process");
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const config_utils_1 = require("./config_utils");
+const isbinaryfile_1 = require("isbinaryfile");
 exports.SPECIAL_FILES = [
     'package.json', 'Gemfile', 'requirements.txt', 'Pipfile', 'pyproject.toml', 'pom.xml', 'build.gradle',
     '.csproj', 'packages.config', 'composer.json', 'CMakeLists.txt', 'conanfile.txt', 'conanfile.py',
@@ -37,10 +38,17 @@ const getAllFilesInDirectory = async (dir) => {
     const subdirs = await fs_1.default.promises.readdir(dir);
     const files = await Promise.all(subdirs.map(async (subdir) => {
         const res = path_1.default.resolve(dir, subdir);
-        if (subdir === 'node_modules' || subdir === '.git') {
+        if (subdir === 'node_modules' || subdir === '.git' || subdir === '.DS_Store') {
             return [];
         }
-        return (await fs_1.default.promises.stat(res)).isDirectory() ? (0, exports.getAllFilesInDirectory)(res) : res;
+        const stat = await fs_1.default.promises.stat(res);
+        if (stat.isDirectory()) {
+            return (0, exports.getAllFilesInDirectory)(res);
+        }
+        else {
+            const isBinary = await (0, isbinaryfile_1.isBinaryFile)(res);
+            return isBinary ? [] : res;
+        }
     }));
     return files.flat();
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cloving",
-  "version": "0.1.19",
+  "version": "0.1.21",
   "packageManager": "yarn@1.22.22",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -51,6 +51,7 @@
     "eslint-plugin-promise": "^7.0.0",
     "ignore": "^5.3.1",
     "ini": "^4.1.3",
+    "isbinaryfile": "^5.0.2",
     "nock": "^13.5.4",
     "vite": "^5.3.5",
     "vite-node": "^2.0.5"