@stbestichhh/term-ai 1.0.0 → 1.1.0

package/index.js

@@ -3,6 +3,9 @@
 import fs from 'node:fs';
 import os from 'node:os';
 import path from 'node:path';
+import readline from 'node:readline';
+import { execSync } from 'node:child_process';
+
 import { GoogleGenAI } from '@google/genai';
 
 const CONFIG_DIR = path.join(process.env.XDG_CONFIG_HOME || path.join(os.homedir(), '.config'), 'tai');
@@ -28,6 +31,20 @@ function saveApiKey(key) {
   console.log(`API key saved to ${CONFIG_FILE}`);
 }
 
+function askConfirmation(query) {
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+  });
+
+  return new Promise((resolve) => {
+    rl.question(query, (answer) => {
+      rl.close();
+      resolve(answer.trim().toLowerCase());
+    });
+  });
+}
+
 const args = process.argv.slice(2);
 
 if (args[0] === 'config' && args[1] === 'set-key') {
@@ -64,15 +81,33 @@ async function run() {
       contents: prompt,
       config: {
         systemInstruction:
-          'You are a command-line assistant. Provide concise, direct terminal commands and short explanations without conversational fluff.',
+          `You are a command-line assistant. Provide concise, direct terminal commands and short explanations without conversational fluff. Wrap command in markdown code block with language as 'bash', 'sh' or 'tai'. In response show no more than one command.`,
       },
     });
 
-    console.log(response.text.trim());
+    const text = response.text.trim();
+    const match = text.match(/```(?:tai|bash|sh)?\s*([\s\S]*?)\s*```/i);
+    const command = match ? match[1].trim() : text.trim();
+
+    const cleanOutput = text
+          .replace(/```(?:tai|bash|sh)?\n?/gi, '')
+          .replace(/```/g, '')
+          .trim();
+
+    console.log(`\n${cleanOutput}\n`);
+
+    const answer = await askConfirmation(`Execute this command? [Y/n] `);
+
+    if (['y', 'yes', ''].includes(answer)) {
+      console.log('\x1b[33mExecuting...\x1b[0m\n');
+      execSync(command, { stdio: 'inherit' });
+    } else {
+      console.log(`Aborted.`)
+    }
   } catch (error) {
     console.error('Error:', error.message);
     process.exit(1);
   }
 }
 
-run();
+void run();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@stbestichhh/term-ai",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "bin": {
     "tai": "./index.js"
   },