nothumanallowed 9.7.2 → 9.8.0

package/src/services/conversations.mjs

@@ -0,0 +1,277 @@
+/**
+ * Multi-conversation manager for NHA Chat.
+ *
+ * Stores conversations in ~/.nha/conversations/ as JSON files.
+ * Each conversation has an ID, auto-generated title, and message history.
+ *
+ * Zero npm dependencies.
+ */
+
+import fs from 'fs';
+import path from 'path';
+import crypto from 'crypto';
+import { NHA_DIR } from '../constants.mjs';
+
+const CONVERSATIONS_DIR = path.join(NHA_DIR, 'conversations');
+const ACTIVE_FILE = path.join(CONVERSATIONS_DIR, '.active');
+const MAX_TITLE_LENGTH = 60;
+const MAX_CONVERSATIONS = 100;
+
+// ── Helpers ──────────────────────────────────────────────────────────────────
+
+function ensureDir() {
+  fs.mkdirSync(CONVERSATIONS_DIR, { recursive: true });
+}
+
+function convPath(id) {
+  return path.join(CONVERSATIONS_DIR, `${id}.json`);
+}
+
+function generateId() {
+  return crypto.randomUUID().slice(0, 8);
+}
+
+/**
+ * Auto-generate a title from the first user message.
+ * Takes the first ~50 chars, trims to last word boundary.
+ */
+function autoTitle(firstMessage) {
+  if (!firstMessage) return 'New Chat';
+  let title = firstMessage.replace(/\s+/g, ' ').trim();
+  if (title.length > MAX_TITLE_LENGTH) {
+    title = title.slice(0, MAX_TITLE_LENGTH);
+    const lastSpace = title.lastIndexOf(' ');
+    if (lastSpace > 20) title = title.slice(0, lastSpace);
+    title += '...';
+  }
+  return title;
+}
+
+// ── CRUD ─────────────────────────────────────────────────────────────────────
+
+/**
+ * Create a new conversation and set it as active.
+ * @returns {{ id: string, title: string, messages: Array, createdAt: string, updatedAt: string }}
+ */
+export function createConversation(title = '') {
+  ensureDir();
+  const id = generateId();
+  const now = new Date().toISOString();
+  const conv = {
+    id,
+    title: title || 'New Chat',
+    messages: [],
+    createdAt: now,
+    updatedAt: now,
+  };
+  fs.writeFileSync(convPath(id), JSON.stringify(conv, null, 2) + '\n', 'utf-8');
+  setActiveId(id);
+  return conv;
+}
+
+/**
+ * Load a conversation by ID.
+ * @returns {object|null}
+ */
+export function loadConversation(id) {
+  try {
+    const data = fs.readFileSync(convPath(id), 'utf-8');
+    return JSON.parse(data);
+  } catch {
+    return null;
+  }
+}
+
+/**
+ * Save a conversation (full overwrite).
+ */
+export function saveConversation(conv) {
+  ensureDir();
+  conv.updatedAt = new Date().toISOString();
+  fs.writeFileSync(convPath(conv.id), JSON.stringify(conv, null, 2) + '\n', 'utf-8');
+}
+
+/**
+ * Delete a conversation by ID.
+ * If it was active, clears active state.
+ * @returns {boolean} true if deleted
+ */
+export function deleteConversation(id) {
+  const filePath = convPath(id);
+  if (!fs.existsSync(filePath)) return false;
+  fs.unlinkSync(filePath);
+  if (getActiveId() === id) {
+    clearActiveId();
+  }
+  return true;
+}
+
+/**
+ * List all conversations, sorted by updatedAt (newest first).
+ * Returns summary objects (no messages).
+ * @returns {Array<{ id: string, title: string, messageCount: number, createdAt: string, updatedAt: string }>}
+ */
+export function listConversations() {
+  ensureDir();
+  const files = fs.readdirSync(CONVERSATIONS_DIR)
+    .filter(f => f.endsWith('.json') && !f.startsWith('.'));
+
+  const convs = [];
+  for (const f of files) {
+    try {
+      const data = JSON.parse(fs.readFileSync(path.join(CONVERSATIONS_DIR, f), 'utf-8'));
+      convs.push({
+        id: data.id,
+        title: data.title || 'Untitled',
+        messageCount: (data.messages || []).length,
+        createdAt: data.createdAt,
+        updatedAt: data.updatedAt,
+      });
+    } catch { /* skip corrupt files */ }
+  }
+
+  convs.sort((a, b) => new Date(b.updatedAt) - new Date(a.updatedAt));
+
+  // Auto-prune old conversations beyond limit
+  if (convs.length > MAX_CONVERSATIONS) {
+    for (const old of convs.slice(MAX_CONVERSATIONS)) {
+      try { fs.unlinkSync(convPath(old.id)); } catch {}
+    }
+    return convs.slice(0, MAX_CONVERSATIONS);
+  }
+
+  return convs;
+}
+
+// ── Active Conversation ──────────────────────────────────────────────────────
+
+/**
+ * Get the active conversation ID.
+ * @returns {string|null}
+ */
+export function getActiveId() {
+  try {
+    return fs.readFileSync(ACTIVE_FILE, 'utf-8').trim() || null;
+  } catch {
+    return null;
+  }
+}
+
+/**
+ * Set the active conversation ID.
+ */
+export function setActiveId(id) {
+  ensureDir();
+  fs.writeFileSync(ACTIVE_FILE, id, 'utf-8');
+}
+
+/**
+ * Clear the active conversation.
+ */
+export function clearActiveId() {
+  try { fs.unlinkSync(ACTIVE_FILE); } catch {}
+}
+
+/**
+ * Get or create the active conversation.
+ * If none exists, creates a new one.
+ * @returns {object} conversation object
+ */
+export function getOrCreateActive() {
+  const activeId = getActiveId();
+  if (activeId) {
+    const conv = loadConversation(activeId);
+    if (conv) return conv;
+  }
+  return createConversation();
+}
+
+/**
+ * Add a message pair (user + assistant) to a conversation.
+ * Auto-titles the conversation from the first user message.
+ */
+export function addMessages(conv, userContent, assistantContent) {
+  conv.messages.push({ role: 'user', content: userContent });
+  conv.messages.push({ role: 'assistant', content: assistantContent });
+
+  // Auto-title from first user message
+  if (conv.title === 'New Chat' && conv.messages.length === 2) {
+    conv.title = autoTitle(userContent);
+  }
+
+  saveConversation(conv);
+}
+
+/**
+ * Get the message history from a conversation, capped at maxTurns pairs.
+ * @returns {Array<{role: string, content: string}>}
+ */
+export function getHistory(conv, maxTurns = 20) {
+  const messages = conv.messages || [];
+  return messages.slice(-(maxTurns * 2));
+}
+
+// ── Export ────────────────────────────────────────────────────────────────────
+
+/**
+ * Export conversation as Markdown.
+ */
+export function exportAsMarkdown(conv) {
+  const lines = [
+    `# ${conv.title}`,
+    `*Created: ${new Date(conv.createdAt).toLocaleString()}*`,
+    `*Messages: ${conv.messages.length}*`,
+    '',
+    '---',
+    '',
+  ];
+
+  for (const msg of conv.messages) {
+    if (msg.role === 'user') {
+      lines.push(`### You`);
+      lines.push(msg.content);
+    } else {
+      lines.push(`### NHA`);
+      lines.push(msg.content);
+    }
+    lines.push('');
+  }
+
+  return lines.join('\n');
+}
+
+/**
+ * Export conversation as JSON.
+ */
+export function exportAsJson(conv) {
+  return JSON.stringify({
+    id: conv.id,
+    title: conv.title,
+    createdAt: conv.createdAt,
+    updatedAt: conv.updatedAt,
+    messages: conv.messages,
+  }, null, 2);
+}
+
+// ── Migration ────────────────────────────────────────────────────────────────
+
+/**
+ * Migrate old single-file chat history to multi-conversation format.
+ * Called once on first run if old history exists.
+ */
+export function migrateOldHistory() {
+  const oldFile = path.join(NHA_DIR, 'memory', 'chat-history.json');
+  if (!fs.existsSync(oldFile)) return;
+
+  try {
+    const messages = JSON.parse(fs.readFileSync(oldFile, 'utf-8'));
+    if (!Array.isArray(messages) || messages.length === 0) return;
+
+    const conv = createConversation('Previous Chat');
+    conv.messages = messages;
+    saveConversation(conv);
+
+    // Rename old file to avoid re-migration
+    fs.renameSync(oldFile, oldFile + '.migrated');
+  } catch { /* migration failed — non-critical */ }
+}