saveinme 1.4.0 → 2.0.0

package/src/ai.js

@@ -0,0 +1,447 @@
+import { GoogleGenerativeAI } from '@google/generative-ai';
+import { input, select, confirm } from '@inquirer/prompts';
+import chalk from 'chalk';
+import { execSync, exec, spawnSync } from 'child_process';
+import { getConfig, saveConfig, getAllNotes } from './store.js';
+import { showHeader, showSuccess, showError, showInfo } from './display.js';
+
+/**
+ * Retrieves the Gemini API Key from environment variables or saved configuration.
+ */
+function getApiKey() {
+  return process.env.GEMINI_API_KEY || getConfig().geminiApiKey || null;
+}
+
+/**
+ * Checks if local Ollama command is installed and in PATH.
+ */
+function isOllamaInstalled() {
+  try {
+    execSync('ollama --version', { stdio: 'ignore' });
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+/**
+ * Opens a URL in the user's default browser.
+ */
+function openUrl(url) {
+  const start = process.platform === 'darwin' ? 'open' : process.platform === 'win32' ? 'start' : 'xdg-open';
+  exec(`${start} ${url}`);
+}
+
+/**
+ * Prompts the user to set up and save their preferred AI Provider (Gemini or Ollama).
+ */
+export async function setupApiKey() {
+  showHeader();
+  console.log(chalk.bold.yellow('\n🤖 AI Copilot Setup'));
+
+  const providerChoice = await select({
+    message: 'Select AI Provider:',
+    choices: [
+      { name: '🌐 Google Gemini (Cloud API)', value: 'gemini' },
+      { name: '📴 Ollama (Local Offline LLM)', value: 'ollama' },
+    ],
+  });
+
+  if (providerChoice === 'gemini') {
+    console.log(chalk.dim('\n  To use Gemini, you need a Gemini API Key.'));
+    console.log(chalk.dim('  You can get one for FREE from Google AI Studio at:'));
+    console.log(chalk.cyan('  https://aistudio.google.com/\n'));
+
+    const key = await input({
+      message: 'Enter your Gemini API Key:',
+      validate: v => v.trim().length > 0 || 'API Key cannot be empty.',
+    });
+
+    saveConfig({
+      aiProvider: 'gemini',
+      geminiApiKey: key.trim(),
+      ollamaUrl: undefined,
+      ollamaModel: undefined,
+    });
+    showSuccess('API Key saved successfully! You can now use "sim -ai".');
+    return true;
+  } else {
+    if (!isOllamaInstalled()) {
+      console.log(chalk.yellow('\n❌ Ollama is not installed or not found in your PATH.'));
+      const download = await confirm({
+        message: 'Would you like to open the Ollama website to download and install it?',
+        default: true,
+      });
+
+      if (download) {
+        showInfo('Opening https://ollama.com/ in your browser...');
+        openUrl('https://ollama.com/');
+        showInfo('Once installed, start the Ollama application and run:  sim -ai --setup');
+        return false;
+      } else {
+        showInfo('Proceeding with manual setup (useful if your Ollama server is hosted on a remote machine).');
+      }
+    }
+
+    const url = await input({
+      message: 'Ollama Endpoint URL:',
+      default: getConfig().ollamaUrl || 'http://localhost:11434',
+    });
+
+    let detectedModels = [];
+    let connectionFailed = false;
+    showInfo('Fetching locally downloaded models from Ollama...');
+    try {
+      const controller = new AbortController();
+      const timeoutId = setTimeout(() => controller.abort(), 2000); // 2 second timeout
+      const res = await fetch(`${url.trim()}/api/tags`, { signal: controller.signal });
+      clearTimeout(timeoutId);
+
+      if (res.ok) {
+        const data = await res.json();
+        if (data.models && data.models.length > 0) {
+          detectedModels = data.models.map(m => m.name);
+        }
+      }
+    } catch {
+      connectionFailed = true;
+    }
+
+    let model = '';
+    if (connectionFailed) {
+      console.log(chalk.yellow('\n⚠️  Could not connect to Ollama at that URL.'));
+      console.log(chalk.dim('   Make sure you run "ollama serve" in a separate terminal window.\n'));
+      model = await input({
+        message: 'Ollama Model Name (e.g. llama3, gemma2, phi3):',
+        default: getConfig().ollamaModel || 'llama3',
+        validate: v => v.trim().length > 0 || 'Model name cannot be empty.',
+      });
+    } else if (detectedModels.length > 0) {
+      console.log(chalk.green(`\n✔ Connected to Ollama! Found ${detectedModels.length} local model(s).`));
+      const choices = [
+        ...detectedModels.map(m => ({ name: `🦙 ${m}`, value: m })),
+        { name: chalk.cyan('+ Use another model (not listed)…'), value: '__other__' }
+      ];
+
+      const modelChoice = await select({
+        message: 'Select a locally installed model:',
+        choices,
+      });
+
+      if (modelChoice === '__other__') {
+        model = await input({
+          message: 'Ollama Model Name (e.g. llama3, gemma2):',
+          validate: v => v.trim().length > 0 || 'Model name cannot be empty.',
+        });
+      } else {
+        model = modelChoice;
+      }
+    } else {
+      console.log(chalk.yellow('\n⚠️  No local models found on your Ollama server.'));
+      
+      const downloadChoice = await select({
+        message: 'Select a popular model to download (pull):',
+        choices: [
+          { name: '🦙 llama3 (8B - Smart & Balanced)', value: 'llama3' },
+          { name: '💎 gemma2 (9B - Google\'s local LLM)', value: 'gemma2' },
+          { name: '⚡ phi3 (3.8B - Lightweight & Fast)', value: 'phi3' },
+          { name: '🌪️ mistral (7B - Great coder/conversational)', value: 'mistral' },
+          { name: '✍️ Enter model name manually', value: '__custom__' },
+        ],
+      });
+
+      let modelToPull = downloadChoice;
+      if (downloadChoice === '__custom__') {
+        modelToPull = await input({
+          message: 'Ollama Model Name to pull:',
+          validate: v => v.trim().length > 0 || 'Model name cannot be empty.',
+        });
+      }
+
+      model = modelToPull.trim();
+
+      const shouldPull = await confirm({
+        message: `Would you like to download (pull) "${model}" now?`,
+        default: true,
+      });
+
+      if (shouldPull) {
+        console.log(chalk.cyan(`\n📥 Downloading model "${model}" via Ollama...`));
+        console.log(chalk.dim('   This can take a few minutes depending on your internet connection.\n'));
+
+        // Run ollama pull directly in the terminal, sharing stdout/stderr
+        const pullProcess = spawnSync('ollama', ['pull', model], { stdio: 'inherit' });
+
+        if (pullProcess.status === 0) {
+          showSuccess(`Model "${model}" downloaded successfully!`);
+        } else {
+          showError(`Failed to download model "${model}". We will save the configuration anyway, but you may need to run "ollama pull ${model}" manually.`);
+        }
+      } else {
+        showInfo(`Skipped download. Make sure to run "ollama pull ${model}" before querying.`);
+      }
+    }
+
+    saveConfig({
+      aiProvider: 'ollama',
+      ollamaUrl: url.trim(),
+      ollamaModel: model.trim(),
+      geminiApiKey: undefined,
+    });
+    showSuccess(`Ollama local provider saved successfully! Model: "${model.trim()}".`);
+    return true;
+  }
+}
+
+/**
+ * Formats notes list as a clean context string for the AI.
+ */
+function buildNotesContext() {
+  const notes = getAllNotes();
+  if (notes.length === 0) {
+    return 'The user does not have any notes saved yet.';
+  }
+
+  let context = 'Here is the user\'s local terminal notes database:\n\n';
+  notes.forEach((note, index) => {
+    context += `--- NOTE #${index + 1} ---\n`;
+    context += `Title: ${note.title}\n`;
+    if (note.category) context += `Notebook/Category: ${note.category}\n`;
+    if (note.tags && note.tags.length > 0) context += `Tags: ${note.tags.join(', ')}\n`;
+    context += `Priority: ${note.priority || 'medium'}\n`;
+    context += `Pinned: ${note.pinned ? 'Yes' : 'No'}\n`;
+    context += `Last Updated: ${note.updatedAt}\n`;
+    context += `Content:\n${note.content}\n`;
+    context += `-----------------\n\n`;
+  });
+  return context;
+}
+
+/**
+ * Helper to stream response from a local Ollama server.
+ */
+async function streamOllama(prompt) {
+  const config = getConfig();
+  const url = config.ollamaUrl || 'http://localhost:11434';
+  const model = config.ollamaModel || 'llama3';
+
+  try {
+    const res = await fetch(`${url}/api/generate`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({
+        model: model,
+        prompt: prompt,
+        stream: true,
+      }),
+    });
+
+    if (!res.ok) {
+      let errMsg = '';
+      try {
+        // Try parsing JSON error from Ollama
+        const errJson = await res.json();
+        errMsg = errJson.error || '';
+      } catch {}
+
+      showError(`\nOllama server returned status ${res.status}${errMsg ? ': ' + errMsg : ''}`);
+      
+      if (res.status === 404 || errMsg.toLowerCase().includes('not found') || errMsg.toLowerCase().includes('pull')) {
+        console.log(chalk.bold.yellow('\n💡 How to fix:'));
+        console.log(`  1. Make sure you have pulled the model locally. Run this in a separate terminal:`);
+        console.log(chalk.cyan(`     ollama pull ${model}`));
+        console.log(`  2. To see your downloaded models, run:`);
+        console.log(chalk.cyan(`     ollama list`));
+        console.log(`  3. If you want to use a different model in saveinme, run:`);
+        console.log(chalk.cyan(`     sim -ai --setup\n`));
+      }
+      return;
+    }
+
+    if (res.body.getReader) {
+      const reader = res.body.getReader();
+      const decoder = new TextDecoder();
+      let buffer = '';
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        buffer += decoder.decode(value, { stream: true });
+        const lines = buffer.split('\n');
+        buffer = lines.pop() || '';
+        for (const line of lines) {
+          if (line.trim()) {
+            try {
+              const parsed = JSON.parse(line);
+              if (parsed.response) process.stdout.write(parsed.response);
+            } catch {}
+          }
+        }
+      }
+    } else {
+      let buffer = '';
+      for await (const chunk of res.body) {
+        buffer += chunk.toString();
+        const lines = buffer.split('\n');
+        buffer = lines.pop() || '';
+        for (const line of lines) {
+          if (line.trim()) {
+            try {
+              const parsed = JSON.parse(line);
+              if (parsed.response) process.stdout.write(parsed.response);
+            } catch {}
+          }
+        }
+      }
+    }
+  } catch (err) {
+    showError(`\nOllama Connection Error: ${err.message}`);
+    console.log(chalk.bold.yellow('\n💡 How to fix:'));
+    console.log(`  1. Make sure the Ollama application is running on your system.`);
+    console.log(`  2. Run the service in a separate terminal if it's not active:`);
+    console.log(chalk.cyan('     ollama serve'));
+    console.log(`  3. Check if Ollama is running at the configured endpoint: ${url}`);
+    console.log(`  4. To configure a different URL or model, run:`);
+    console.log(chalk.cyan('     sim -ai --setup\n'));
+  }
+}
+
+/**
+ * Runs a query against the configured AI model.
+ * @param {string} question The question or prompt
+ */
+export async function queryAI(question) {
+  const config = getConfig();
+  const provider = config.aiProvider || (getApiKey() ? 'gemini' : null);
+
+  if (!provider) {
+    showError('AI Copilot is not configured.');
+    const ok = await setupApiKey();
+    if (!ok) return;
+    return queryAI(question);
+  }
+
+  const notesContext = buildNotesContext();
+  const prompt = `You are "saveinme (sim)", a smart personal terminal notebook assistant.
+You help the user summarize, extract, or search information from their notes database.
+Keep your responses helpful, very concise, and clean for terminal output (use bolding, lists, and markdown spacing appropriately).
+
+${notesContext}
+
+User Question: ${question}
+Answer:`;
+
+  if (provider === 'gemini') {
+    const apiKey = getApiKey();
+    if (!apiKey) {
+      showError('Gemini API key is not configured.');
+      await setupApiKey();
+      return queryAI(question);
+    }
+
+    const genAI = new GoogleGenerativeAI(apiKey);
+    const model = genAI.getGenerativeModel({ model: 'gemini-1.5-flash' });
+
+    showInfo('Thinking (Gemini)...');
+    try {
+      const result = await model.generateContentStream(prompt);
+      console.log(chalk.bold.cyan('\n🤖 saveinme AI:'));
+      for await (const chunk of result.stream) {
+        process.stdout.write(chunk.text());
+      }
+      console.log('\n');
+    } catch (err) {
+      showError(`Gemini API Error: ${err.message}`);
+    }
+  } else {
+    showInfo(`Thinking (Ollama: ${config.ollamaModel || 'llama3'})...`);
+    console.log(chalk.bold.cyan('\n🤖 saveinme AI:'));
+    await streamOllama(prompt);
+    console.log('\n');
+  }
+}
+
+/**
+ * Launches an interactive chat session with the AI Copilot.
+ */
+export async function chatAI() {
+  let config = getConfig();
+  let provider = config.aiProvider || (getApiKey() ? 'gemini' : null);
+
+  if (!provider) {
+    showError('AI Copilot is not configured.');
+    const ok = await setupApiKey();
+    if (!ok) return;
+    config = getConfig();
+    provider = config.aiProvider || (getApiKey() ? 'gemini' : null);
+    if (!provider) return;
+  }
+
+  showHeader();
+  let modelName = provider === 'gemini' ? 'gemini-1.5-flash' : (config.ollamaModel || 'llama3');
+  console.log(chalk.bold.yellow(`\n💬 saveinme AI Copilot Chat Session`));
+  console.log(chalk.dim(`  Provider: ${provider === 'gemini' ? 'Gemini' : 'Ollama'} (${modelName})`));
+  console.log(chalk.dim('  Ask me anything about your notes. I will read your live database.'));
+  console.log(chalk.dim('  Type "/setup" to change the model/provider.'));
+  console.log(chalk.dim('  Type "exit" or "quit" to leave the chat.\n'));
+
+  while (true) {
+    const question = await input({
+      message: chalk.cyan('You:'),
+    });
+
+    const trimmed = question.trim();
+    if (trimmed.toLowerCase() === 'exit' || trimmed.toLowerCase() === 'quit') {
+      showInfo('AI Chat session closed.');
+      break;
+    }
+
+    if (trimmed.toLowerCase() === '/setup' || trimmed.toLowerCase() === '/model') {
+      await setupApiKey();
+      config = getConfig();
+      provider = config.aiProvider || (getApiKey() ? 'gemini' : null);
+      modelName = provider === 'gemini' ? 'gemini-1.5-flash' : (config.ollamaModel || 'llama3');
+      console.log(chalk.bold.yellow(`\n💬 AI Configuration Updated`));
+      console.log(chalk.dim(`  Current Provider: ${provider === 'gemini' ? 'Gemini' : 'Ollama'} (${modelName})\n`));
+      continue;
+    }
+
+    if (!trimmed) continue;
+
+    const notesContext = buildNotesContext();
+    const prompt = `You are "saveinme (sim)", a smart personal terminal notebook assistant.
+You help the user summarize, search, or extract information from their notes database.
+Keep your responses helpful, concise, and formatted beautifully for terminal display.
+
+${notesContext}
+
+User Question: ${trimmed}
+Answer:`;
+
+    console.log(chalk.bold.magenta('\n🤖 saveinme AI:'));
+    if (provider === 'gemini') {
+      const apiKey = getApiKey();
+      if (!apiKey) {
+        showError('Gemini API key is not configured.');
+        await setupApiKey();
+        config = getConfig();
+        provider = config.aiProvider || (getApiKey() ? 'gemini' : null);
+        if (!provider) return;
+        continue;
+      }
+      const genAI = new GoogleGenerativeAI(apiKey);
+      const model = genAI.getGenerativeModel({ model: 'gemini-1.5-flash' });
+      try {
+        const result = await model.generateContentStream(prompt);
+        for await (const chunk of result.stream) {
+          process.stdout.write(chunk.text());
+        }
+        console.log('\n');
+      } catch (err) {
+        showError(`Gemini API Error: ${err.message}`);
+      }
+    } else {
+      await streamOllama(prompt);
+      console.log('\n');
+    }
+  }
+}

package/src/display.js

@@ -142,10 +142,12 @@ export function showNotesList(notes, heading = null) {
   console.log(
     chalk.dim(
       `\n  ${notes.length} note${notes.length !== 1 ? 's' : ''} saved` +
+      `  ·  sim -ui  dashboard` +
+      `  ·  sim -ai  ai copilot` +
       `  ·  sim -c  create` +
       `  ·  sim -cl  paste clip` +
-      `  ·  sim -v <title>  view` +
-      `  ·  sim -f <term>  search` +
+      `  ·  sim <title>  view` +
+      `  ·  sim -rm  remove` +
       `  ·  sim -h  help\n`
     )
   );
@@ -244,13 +246,17 @@ export function showHelp() {
   ${chalk.dim('  * Note: You can use "saveinme" or the shortcut "sim" interchangeably *')}
 
   ${chalk.bold.white('sim')}                         List all saved notes (pinned first)
+  ${chalk.bold.white('sim <title>')}                  Directly view a note by its title
+  ${chalk.bold.white('sim -ui')}                      Launch Split-Pane TUI Dashboard
   ${chalk.bold.white('sim -c')} [${chalk.italic('title')}]             Create/edit a note interactively
   ${chalk.bold.white('sim -cl')} [${chalk.italic('title')}]            Instantly save text from your clipboard
-  ${chalk.bold.white('sim -t')} [${chalk.italic('notebook')}]          Set default target notebook (folder)
-  ${chalk.bold.white('sim -v')} ${chalk.italic('<title>')}             View a note's full content
-  ${chalk.bold.white('sim -d')} ${chalk.italic('<title>')}             Delete a note permanently
+  ${chalk.bold.white('sim -d')} [${chalk.italic('notebook')} | ${chalk.italic('--clear')}] Set default notebook / select from list / clear default
+  ${chalk.bold.white('sim -v')} [${chalk.italic('title')}]             View a note's full content (or pick from list)
+  ${chalk.bold.white('sim -rm')} [${chalk.italic('title')}]            Remove note(s) permanently (or pick from list)
   ${chalk.bold.white('sim -p')} ${chalk.italic('<title>')}             Toggle pin status (stays at top)
   ${chalk.bold.white('sim -f')} ${chalk.italic('<term>')}              Search title, content, category, or tags
+  ${chalk.bold.white('sim -ai')} [${chalk.italic('query')} | ${chalk.italic('--setup')} | ${chalk.italic('--reset')}] Ask/chat with AI Copilot / configure provider / reset settings
+  ${chalk.bold.white('sim --sync')} [${chalk.italic('--setup')}]       Sync notes database with git remote / configure remote repo
   ${chalk.bold.white('sim --path')}                 Show database folder path on disk
   ${chalk.bold.white('sim -h')}                     Show this detailed help manual
 
@@ -282,17 +288,49 @@ export function showHelp() {
 
   ${chalk.bold.cyan('3. Default Target Notebooks')}
     ↳ Targets allow you to automatically group notes by category (like folder routing).
-      - Set active notebook target:   ${chalk.white('sim -t "<name>"')}
-      - View current active target:   ${chalk.white('sim -t')}
-      - Clear target default:         ${chalk.white('sim -t --clear')}
+      - Set active notebook target:   ${chalk.white('sim -d "<name>"')}
+      - Select target from a list:    ${chalk.white('sim -d')}       ${chalk.dim('↳ Shows interactive notebook selector')}
+      - Clear target default:         ${chalk.white('sim -d --clear')}
 
     ${chalk.yellow('Example (Set target to "Work"):')}
-      $ sim -t "Work"
+      $ sim -d "Work"
       $ sim -c "meeting summary"   ${chalk.dim('↳ Automatically saved inside notebook "Work"')}
     ${chalk.yellow('Example (Override target for single command using -n):')}
       $ sim -c "groceries" -n "Personal"  ${chalk.dim('↳ Bypasses default target, saves to "Personal"')}
 
-  ${chalk.bold.cyan('4. Piping Command Outputs (Advanced)')}
+  ${chalk.bold.cyan('4. Viewing Notes')}
+    ↳ You can view the contents of any saved note easily:
+      - View a note directly:         ${chalk.white('sim "My Note"')}
+      - View note via flag:           ${chalk.white('sim -v "My Note"')}
+      - Select note from a list:      ${chalk.white('sim -v')}       ${chalk.dim('↳ Shows interactive note selector')}
+
+  ${chalk.bold.cyan('5. Removing Notes (Single or Group)')}
+    ↳ Permanently delete one or more notes:
+      - Delete a single note:         ${chalk.white('sim -rm "My Note"')}
+      - Delete multiple notes:        ${chalk.white('sim -rm note1 note2')}
+      - Select notes to remove:       ${chalk.white('sim -rm')}      ${chalk.dim('↳ Opens multi-select checkbox menu')}
+
+  ${chalk.bold.cyan('6. Split-Pane TUI Dashboard')}
+    ↳ Run the full-screen terminal user interface to browse, read, and manage notes visually:
+      $ sim -ui
+      - Use Arrow keys (or j/k) to navigate notes on the left.
+      - View wrapped note contents and metadata on the right.
+      - Press [Enter] to edit, [c] to create, [p] to pin, [d] to delete, [/] to search, and [q] to exit.
+
+  ${chalk.bold.cyan('7. AI Copilot (Gemini / Ollama)')}
+    ↳ Query your notes using cloud Gemini or local offline Ollama models:
+      - Set up AI Provider:           ${chalk.white('sim -ai --setup')}
+      - Reset AI settings:            ${chalk.white('sim -ai --reset')}
+      - Query your notes directly:    ${chalk.white('sim -ai "how do I configure the database?"')}
+      - Start interactive AI chat:    ${chalk.white('sim -ai')}
+
+  ${chalk.bold.cyan('8. Git Sync & Backup')}
+    ↳ Keep notes backed up and synced across different machines:
+      - Set up remote git repo:       ${chalk.white('sim --sync --setup')}
+      - Trigger manual sync:          ${chalk.white('sim --sync')}
+      - Auto-Sync: Once remote is set up, notes will automatically sync in the background on save/delete!
+
+  ${chalk.bold.cyan('9. Piping Command Outputs (Advanced)')}
     ↳ Pipe stdout of any command directly into a note. Saves ${chalk.bold.green('instantly')} with ${chalk.bold.red('zero prompts')}!
       - Target: Default target notebook, unless overridden with ${chalk.white('-n')}.
       - Title: Uses custom title if provided, otherwise defaults to: