@semalt-ai/code 1.7.0 → 1.8.0

package/lib/storage.js

@@ -0,0 +1,96 @@
+'use strict';
+
+const crypto = require('crypto');
+const fs = require('fs');
+const os = require('os');
+const path = require('path');
+
+const SESSIONS_DIR = path.join(os.homedir(), '.semalt-ai', 'sessions');
+
+class SessionStorage {
+  constructor() {
+    fs.mkdirSync(SESSIONS_DIR, { recursive: true });
+  }
+
+  generateId() {
+    return crypto.randomBytes(4).toString('hex');
+  }
+
+  save(session) {
+    const filename = `${session.created_at}-${session.id}.json`;
+    fs.writeFileSync(path.join(SESSIONS_DIR, filename), JSON.stringify(session, null, 2));
+  }
+
+  list() {
+    let files;
+    try {
+      files = fs.readdirSync(SESSIONS_DIR).filter((f) => f.endsWith('.json'));
+    } catch {
+      return [];
+    }
+    const sessions = [];
+    for (const file of files) {
+      try {
+        const data = JSON.parse(fs.readFileSync(path.join(SESSIONS_DIR, file), 'utf8'));
+        sessions.push({
+          id: data.id,
+          created_at: data.created_at,
+          model: data.model,
+          message_count: Array.isArray(data.messages)
+            ? data.messages.filter((m) => m.role !== 'system').length
+            : 0,
+        });
+      } catch {
+        // skip corrupt files
+      }
+    }
+    return sessions.sort((a, b) => b.created_at - a.created_at);
+  }
+
+  _findFile(id) {
+    let files;
+    try {
+      files = fs.readdirSync(SESSIONS_DIR).filter((f) => f.endsWith('.json'));
+    } catch {
+      return null;
+    }
+    return files.find((f) => f.includes(id)) || null;
+  }
+
+  load(id) {
+    const file = this._findFile(id);
+    if (!file) return null;
+    try {
+      return JSON.parse(fs.readFileSync(path.join(SESSIONS_DIR, file), 'utf8'));
+    } catch {
+      return null;
+    }
+  }
+
+  delete(id) {
+    const file = this._findFile(id);
+    if (file) fs.unlinkSync(path.join(SESSIONS_DIR, file));
+  }
+
+  pruneOlderThan(days) {
+    const cutoff = Date.now() - days * 86400000;
+    let files;
+    try {
+      files = fs.readdirSync(SESSIONS_DIR).filter((f) => f.endsWith('.json'));
+    } catch {
+      return;
+    }
+    for (const file of files) {
+      try {
+        const data = JSON.parse(fs.readFileSync(path.join(SESSIONS_DIR, file), 'utf8'));
+        if (data.created_at < cutoff) {
+          fs.unlinkSync(path.join(SESSIONS_DIR, file));
+        }
+      } catch {
+        // skip
+      }
+    }
+  }
+}
+
+module.exports = { SessionStorage };