@saiteja1123/mcp-server 1.1.4 → 1.1.5

package/src/deep-scan/store.js

@@ -0,0 +1,57 @@
+import fs from 'fs/promises';
+import path from 'path';
+import { assertSourceSafePayload, cloneSourceSafe } from './sourceSafe.js';
+import { validateDeepScanState, validateRunIdSegment } from './contracts.js';
+
+export class LocalDeepScanStore {
+  constructor({ rootPath, directory = '.vibesecur/deep-scans' }) {
+    if (!rootPath || typeof rootPath !== 'string') throw new Error('LocalDeepScanStore requires rootPath');
+    if (!directory || typeof directory !== 'string') throw new Error('LocalDeepScanStore requires directory');
+    const normalizedDirectory = path.normalize(directory);
+    if (
+      path.isAbsolute(normalizedDirectory)
+      || normalizedDirectory === '..'
+      || normalizedDirectory.startsWith(`..${path.sep}`)
+    ) {
+      throw new Error('LocalDeepScanStore directory must stay inside rootPath');
+    }
+    this.rootPath = path.resolve(rootPath);
+    this.directory = normalizedDirectory;
+  }
+
+  get baseDir() {
+    return path.join(this.rootPath, this.directory);
+  }
+
+  runPath(runId) {
+    return path.join(this.baseDir, `${validateRunIdSegment(runId, 'runId')}.json`);
+  }
+
+  async saveRun(state) {
+    const safeState = validateDeepScanState(cloneSourceSafe(state, 'deepScanState'));
+    assertSourceSafePayload(safeState, 'deepScanState');
+    await fs.mkdir(this.baseDir, { recursive: true });
+    const target = this.runPath(safeState.runId);
+    const tmp = `${target}.${process.pid}.${Date.now()}.tmp`;
+    await fs.writeFile(tmp, `${JSON.stringify(safeState, null, 2)}\n`, 'utf8');
+    await fs.rename(tmp, target);
+    return safeState;
+  }
+
+  async getRun(runId) {
+    const raw = await fs.readFile(this.runPath(runId), 'utf8');
+    return validateDeepScanState(JSON.parse(raw));
+  }
+
+  async listRuns() {
+    await fs.mkdir(this.baseDir, { recursive: true });
+    const entries = await fs.readdir(this.baseDir);
+    const runs = [];
+    for (const entry of entries) {
+      if (!entry.endsWith('.json')) continue;
+      const runId = entry.slice(0, -5);
+      runs.push(await this.getRun(runId));
+    }
+    return runs.sort((a, b) => String(b.updatedAt).localeCompare(String(a.updatedAt)));
+  }
+}