Added concept of conversation threads that are persistent

Author: mmiscool

src/llmCall.js

@@ -10,22 +10,36 @@ import {
 } from "./terminalHelpers.js";
 import Anthropic from '@anthropic-ai/sdk';
 import cliProgress from 'cli-progress';
-import { spawn } from 'child_process';``
+import { spawn } from 'child_process'; ``
 
 let throttleTime = 20;
 let lastCallTime = 0;
 
 export class conversation {
-    constructor() {
+    constructor(id = null, targetFile = null) {
         this.messages = [];
+        this.title = '';
+        this.targetFile = targetFile;
+
+        if (id) {
+            this.id = id;
+            this.loadConversation(id);
+        } else {
+            //generate a unique id for the conversation based on the current time in the format
+            // of yyyy-mm-dd-hh-mm-ss
+            this.id = new Date().toISOString().replace(/[-:.]/g, '').replace('T', '_').split('.')[0];
+        }
+
     }
 
     async addMessage(role, content) {
         this.messages.push({ role, content });
+        this.storeConversation();
     }
 
     async addFileMessage(role, filePath, description = '') {
         this.messages.push({ role, content: filePath, filePath, description });
+        this.storeConversation();
     }
 
     async lastMessage() {
@@ -42,9 +56,70 @@ export class conversation {
     async getMessages() {
         return this.messages;
     }
+
+    async getConversation() {
+        return {
+            messages: this.messages,
+            title: this.title,
+            id: this.id,
+            targetFile: this.targetFile
+        }
+    }
+
+    async clearMessages() {
+        this.messages = [];
+        this.storeConversation();
+    }
+
+    async storeConversation(id = this.id) {
+        // write the conversation to a file
+        const conversationObject = {
+            messages: this.messages,
+            title: this.title,
+            id: this.id,
+            targetFile: this.targetFile
+        };
+        const conversationJSON = JSON.stringify(conversationObject);
+        const filePath = `./.aiCoder/conversations/${id}.json`;
+        await writeFile(filePath, conversationJSON);
+
+    }
+
+    async loadConversation(id = this.id) {
+        // load the conversation from a file  
+        const filePath = `./.aiCoder/conversations/${id}.json`;
+
+        const conversationJSON = await readFile(filePath);
+        const conversationObject = JSON.parse(conversationJSON);
+        this.messages = conversationObject.messages;
+        this.title = conversationObject.title;
+        this.id = conversationObject.id;
+        this.targetFile = conversationObject.targetFile;
+    }
 }
 
 
+
+export async function listConversations() {
+    // load all the conversations from the conversation folder
+    const conversationFolder = './.aiCoder/conversations';
+    if (!fs.existsSync
+        (conversationFolder)) {
+        fs.mkdirSync(conversationFolder);
+    }
+
+    const conversationFiles = fs.readdirSync(conversationFolder);
+    for (const file of conversationFiles) {
+        const conversationId = file.split('.')[0];
+        const newConversation = new conversation(conversationId);
+        this.conversations.push(newConversation);
+    }
+
+    return this.conversations;
+}
+
+
+
 async function throttle() {
     // check if the current time is greater than the last call time + throttle time and if not wait until it is
     // if it needs use the printAndPause function to show a message to the user and wait the remaining time
@@ -185,14 +260,14 @@ export async function selectModel(overwrite = false) {
     const llmModelFileName = `./.aiCoder/${await selectAIservice()}-model.txt`;
     if (readFile(llmModelFileName) && !overwrite) {
         return readFile(llmModelFileName);
-    } 
+    }
 }
 
 
 export async function selectAIservice(overwrite = false) {
     if (fs.existsSync('./.aiCoder/ai-service.txt') && !overwrite) {
         return fs.readFileSync('./.aiCoder/ai-service.txt', 'utf8');
-    } 
+    }
 }
 
 // ollama related functions -----------------------------------------------------------------------------------------------

src/main.js

@@ -1,6 +1,6 @@
 #!/usr/bin/env node
 import {
-    printAndPause,
+  printAndPause,
 } from './terminalHelpers.js';
 
 import './gitnoteSetup.js';
@@ -13,8 +13,8 @@ export const debugMode = false;
 
 // graceful shutdown
 process.on('SIGINT', () => {
-    printAndPause("\nExiting gracefully...");
-    process.exit(0); // Exit with a success code
+  printAndPause("\nExiting gracefully...");
+  process.exit(0); // Exit with a success code
 });
 
 
@@ -26,42 +26,14 @@ ctx.skipApprovingChanges = false;
 
 
 async function appStart(params) {
-    if (!ctx.targetFile) ctx.targetFile = await getFilePath();
-    setupServer();
+  if (!ctx.targetFile) ctx.targetFile = await getFilePath();
+  setupServer();
 }
 
 // Determine file path based on argument or interactive selection
 export async function getFilePath(newFilePathArg = null) {
-    const filePath = newFilePathArg;
-    newFilePathArg = await fileSelector({
-      message: 'Select a file:',
-      pageSize: await selectListHeight() - 2,
-      filter: (filePath) => {
-        printDebugMessage(filePath);
-        //return true;
-        if (filePath.path.includes(
-          'node_modules') ||
-          filePath.path.includes('.git') ||
-          filePath.path.includes('.vscode')
-        ) {
-          return false;
-        }
-  
-        // check if the file starts with a dot
-        if (filePath.path.includes('/.')) {
-          return false;
-        }
-  
-        return true;
-      }
-  
-    });
-  
-    //convert to relative path taking into account the git root
-    newFilePathArg = await path.relative(process.cwd(), newFilePathArg);
-  
-    return `./${newFilePathArg}`;
-  }
-  
+  return `./`;
+}
+
 
 appStart();