fix goodbye issue and ai's not working

Author: rootedbox

ai.js

@@ -147,7 +147,8 @@ async function main(continuePrompt = true) {
   }
 }
 
-if (import.meta.url === `file://${process.argv[1]}`) {
+
+if (!`file://${process.argv[1]}`.includes('!')) {
   main();
 }
 

aic.js

@@ -51,8 +51,6 @@ async function chatPrompt() {
 }
 
 async function startConversation(client, initialQuestion) {
-  const exitWords = new Set(["exit", "goodbye", "quit"]);
-
   let continueConversation = true;
   let question = initialQuestion;
 
@@ -61,7 +59,7 @@ async function startConversation(client, initialQuestion) {
       question = await chatPrompt();
     }
 
-    if (shouldExitConversation(question, exitWords)) {
+    if (shouldExitConversation(question)) {
       continueConversation = false;
       break;
     }
@@ -74,7 +72,8 @@ async function startConversation(client, initialQuestion) {
   process.exit(0);
 }
 
-function shouldExitConversation(question, exitWords) {
+function shouldExitConversation(question) {
+  const exitWords = new Set(["exit", "goodbye", "quit"]);
   return !question || exitWords.has(question.toLowerCase().trim());
 }
 
@@ -95,7 +94,7 @@ async function main(startConvo = true) {
   }
 }
 
-if (import.meta.url === `file://${process.argv[1]}`) {
+if (!`file://${process.argv[1]}`.includes('!')) {
   main();
 }
 

aiq.js

@@ -429,6 +429,11 @@ async function promptConnectionDetails(existingConfig = {}) {
   };
 }
 
+function shouldExitConversation(question) {
+  const exitWords = new Set(["exit", "goodbye", "quit"]);
+  return !question || exitWords.has(question.toLowerCase().trim());
+}
+
 async function promptForCommands(dbConfigs, connectionNameOrFile, client) {
   let command;
   do {
@@ -441,10 +446,12 @@ async function promptForCommands(dbConfigs, connectionNameOrFile, client) {
     ]);
     command = input.command;
 
-    if (command.toLowerCase() !== "exit") {
-      await processQuery(dbConfigs, connectionNameOrFile, command, client);
+    if (shouldExitConversation(command)) {
+      break;
     }
-  } while (command.toLowerCase() !== "exit");
+    
+    await processQuery(dbConfigs, connectionNameOrFile, command, client);
+  } while (true);
 }
 
 async function handleNoOrSingleParam() {
@@ -487,7 +494,7 @@ async function main(prompt = true) {
   process.exit(0);
 }
 
-if (import.meta.url === `file://${process.argv[1]}`) {
+if (!`file://${process.argv[1]}`.includes('!')) {
   main();
 }