@push.rocks/smartmongo 2.2.0 → 4.0.0

package/ts/{congodb → tsmdb}/engine/TransactionEngine.ts

@@ -1,7 +1,7 @@
-import * as plugins from '../congodb.plugins.js';
+import * as plugins from '../tsmdb.plugins.js';
 import type { IStorageAdapter } from '../storage/IStorageAdapter.js';
 import type { Document, IStoredDocument, ITransactionOptions } from '../types/interfaces.js';
-import { CongoTransactionError, CongoWriteConflictError } from '../errors/CongoErrors.js';
+import { TsmdbTransactionError, TsmdbWriteConflictError } from '../errors/TsmdbErrors.js';
 
 /**
  * Transaction state
@@ -70,7 +70,7 @@ export class TransactionEngine {
   async getSnapshot(txnId: string, dbName: string, collName: string): Promise<IStoredDocument[]> {
     const txn = this.transactions.get(txnId);
     if (!txn || txn.status !== 'active') {
-      throw new CongoTransactionError('Transaction is not active');
+      throw new TsmdbTransactionError('Transaction is not active');
     }
 
     const ns = `${dbName}.${collName}`;
@@ -148,7 +148,7 @@ export class TransactionEngine {
   recordInsert(txnId: string, dbName: string, collName: string, doc: IStoredDocument): void {
     const txn = this.transactions.get(txnId);
     if (!txn || txn.status !== 'active') {
-      throw new CongoTransactionError('Transaction is not active');
+      throw new TsmdbTransactionError('Transaction is not active');
     }
 
     const ns = `${dbName}.${collName}`;
@@ -174,7 +174,7 @@ export class TransactionEngine {
   ): void {
     const txn = this.transactions.get(txnId);
     if (!txn || txn.status !== 'active') {
-      throw new CongoTransactionError('Transaction is not active');
+      throw new TsmdbTransactionError('Transaction is not active');
     }
 
     const ns = `${dbName}.${collName}`;
@@ -203,7 +203,7 @@ export class TransactionEngine {
   recordDelete(txnId: string, dbName: string, collName: string, doc: IStoredDocument): void {
     const txn = this.transactions.get(txnId);
     if (!txn || txn.status !== 'active') {
-      throw new CongoTransactionError('Transaction is not active');
+      throw new TsmdbTransactionError('Transaction is not active');
     }
 
     const ns = `${dbName}.${collName}`;
@@ -231,10 +231,10 @@ export class TransactionEngine {
   async commitTransaction(txnId: string): Promise<void> {
     const txn = this.transactions.get(txnId);
     if (!txn) {
-      throw new CongoTransactionError('Transaction not found');
+      throw new TsmdbTransactionError('Transaction not found');
     }
     if (txn.status !== 'active') {
-      throw new CongoTransactionError(`Cannot commit transaction in state: ${txn.status}`);
+      throw new TsmdbTransactionError(`Cannot commit transaction in state: ${txn.status}`);
     }
 
     // Check for write conflicts
@@ -245,7 +245,7 @@ export class TransactionEngine {
       const hasConflicts = await this.storage.hasConflicts(dbName, collName, ids, txn.startTime);
       if (hasConflicts) {
         txn.status = 'aborted';
-        throw new CongoWriteConflictError();
+        throw new TsmdbWriteConflictError();
       }
     }
 
@@ -281,7 +281,7 @@ export class TransactionEngine {
   async abortTransaction(txnId: string): Promise<void> {
     const txn = this.transactions.get(txnId);
     if (!txn) {
-      throw new CongoTransactionError('Transaction not found');
+      throw new TsmdbTransactionError('Transaction not found');
     }
     if (txn.status !== 'active') {
       // Already committed or aborted, just return
@@ -336,7 +336,7 @@ export class TransactionEngine {
         await this.abortTransaction(txnId);
         this.endTransaction(txnId);
 
-        if (error instanceof CongoWriteConflictError && attempt < maxRetries - 1) {
+        if (error instanceof TsmdbWriteConflictError && attempt < maxRetries - 1) {
           // Retry on write conflict
           lastError = error;
           continue;
@@ -346,6 +346,6 @@ export class TransactionEngine {
       }
     }
 
-    throw lastError || new CongoTransactionError('Transaction failed after max retries');
+    throw lastError || new TsmdbTransactionError('Transaction failed after max retries');
   }
 }

package/ts/{congodb → tsmdb}/engine/UpdateEngine.ts

@@ -1,4 +1,4 @@
-import * as plugins from '../congodb.plugins.js';
+import * as plugins from '../tsmdb.plugins.js';
 import type { Document, IStoredDocument } from '../types/interfaces.js';
 import { QueryEngine } from './QueryEngine.js';
 

package/ts/{congodb/errors/CongoErrors.ts → tsmdb/errors/TsmdbErrors.ts}

@@ -1,14 +1,14 @@
 /**
- * Base error class for all CongoDB errors
+ * Base error class for all TsmDB errors
  * Mirrors MongoDB driver error hierarchy
  */
-export class CongoError extends Error {
+export class TsmdbError extends Error {
   public code?: number;
   public codeName?: string;
 
   constructor(message: string, code?: number, codeName?: string) {
     super(message);
-    this.name = 'CongoError';
+    this.name = 'TsmdbError';
     this.code = code;
     this.codeName = codeName;
     Object.setPrototypeOf(this, new.target.prototype);
@@ -18,33 +18,33 @@ export class CongoError extends Error {
 /**
  * Error thrown during connection issues
  */
-export class CongoConnectionError extends CongoError {
+export class TsmdbConnectionError extends TsmdbError {
   constructor(message: string) {
     super(message);
-    this.name = 'CongoConnectionError';
+    this.name = 'TsmdbConnectionError';
   }
 }
 
 /**
  * Error thrown when an operation times out
  */
-export class CongoTimeoutError extends CongoError {
+export class TsmdbTimeoutError extends TsmdbError {
   constructor(message: string) {
     super(message, 50, 'MaxTimeMSExpired');
-    this.name = 'CongoTimeoutError';
+    this.name = 'TsmdbTimeoutError';
   }
 }
 
 /**
  * Error thrown during write operations
  */
-export class CongoWriteError extends CongoError {
+export class TsmdbWriteError extends TsmdbError {
   public writeErrors?: IWriteError[];
   public result?: any;
 
   constructor(message: string, code?: number, writeErrors?: IWriteError[]) {
     super(message, code);
-    this.name = 'CongoWriteError';
+    this.name = 'TsmdbWriteError';
     this.writeErrors = writeErrors;
   }
 }
@@ -52,13 +52,13 @@ export class CongoWriteError extends CongoError {
 /**
  * Error thrown for duplicate key violations
  */
-export class CongoDuplicateKeyError extends CongoWriteError {
+export class TsmdbDuplicateKeyError extends TsmdbWriteError {
   public keyPattern?: Record<string, 1>;
   public keyValue?: Record<string, any>;
 
   constructor(message: string, keyPattern?: Record<string, 1>, keyValue?: Record<string, any>) {
     super(message, 11000);
-    this.name = 'CongoDuplicateKeyError';
+    this.name = 'TsmdbDuplicateKeyError';
     this.codeName = 'DuplicateKey';
     this.keyPattern = keyPattern;
     this.keyValue = keyValue;
@@ -68,13 +68,13 @@ export class CongoDuplicateKeyError extends CongoWriteError {
 /**
  * Error thrown for bulk write failures
  */
-export class CongoBulkWriteError extends CongoError {
+export class TsmdbBulkWriteError extends TsmdbError {
   public writeErrors: IWriteError[];
   public result: any;
 
   constructor(message: string, writeErrors: IWriteError[], result: any) {
     super(message, 65);
-    this.name = 'CongoBulkWriteError';
+    this.name = 'TsmdbBulkWriteError';
     this.writeErrors = writeErrors;
     this.result = result;
   }
@@ -83,20 +83,20 @@ export class CongoBulkWriteError extends CongoError {
 /**
  * Error thrown during transaction operations
  */
-export class CongoTransactionError extends CongoError {
+export class TsmdbTransactionError extends TsmdbError {
   constructor(message: string, code?: number) {
     super(message, code);
-    this.name = 'CongoTransactionError';
+    this.name = 'TsmdbTransactionError';
   }
 }
 
 /**
  * Error thrown when a transaction is aborted due to conflict
  */
-export class CongoWriteConflictError extends CongoTransactionError {
+export class TsmdbWriteConflictError extends TsmdbTransactionError {
   constructor(message: string = 'Write conflict during transaction') {
     super(message, 112);
-    this.name = 'CongoWriteConflictError';
+    this.name = 'TsmdbWriteConflictError';
     this.codeName = 'WriteConflict';
   }
 }
@@ -104,20 +104,20 @@ export class CongoWriteConflictError extends CongoTransactionError {
 /**
  * Error thrown for invalid arguments
  */
-export class CongoArgumentError extends CongoError {
+export class TsmdbArgumentError extends TsmdbError {
   constructor(message: string) {
     super(message);
-    this.name = 'CongoArgumentError';
+    this.name = 'TsmdbArgumentError';
   }
 }
 
 /**
  * Error thrown when an operation is not supported
  */
-export class CongoNotSupportedError extends CongoError {
+export class TsmdbNotSupportedError extends TsmdbError {
   constructor(message: string) {
     super(message, 115);
-    this.name = 'CongoNotSupportedError';
+    this.name = 'TsmdbNotSupportedError';
     this.codeName = 'CommandNotSupported';
   }
 }
@@ -125,20 +125,20 @@ export class CongoNotSupportedError extends CongoError {
 /**
  * Error thrown when cursor is exhausted or closed
  */
-export class CongoCursorError extends CongoError {
+export class TsmdbCursorError extends TsmdbError {
   constructor(message: string) {
     super(message);
-    this.name = 'CongoCursorError';
+    this.name = 'TsmdbCursorError';
   }
 }
 
 /**
  * Error thrown when a namespace (database.collection) is invalid
  */
-export class CongoNamespaceError extends CongoError {
+export class TsmdbNamespaceError extends TsmdbError {
   constructor(message: string) {
     super(message, 73);
-    this.name = 'CongoNamespaceError';
+    this.name = 'TsmdbNamespaceError';
     this.codeName = 'InvalidNamespace';
   }
 }
@@ -146,10 +146,10 @@ export class CongoNamespaceError extends CongoError {
 /**
  * Error thrown when an index operation fails
  */
-export class CongoIndexError extends CongoError {
+export class TsmdbIndexError extends TsmdbError {
   constructor(message: string, code?: number) {
     super(message, code || 86);
-    this.name = 'CongoIndexError';
+    this.name = 'TsmdbIndexError';
   }
 }
 
@@ -164,18 +164,18 @@ export interface IWriteError {
 }
 
 /**
- * Convert any error to a CongoError
+ * Convert any error to a TsmdbError
  */
-export function toCongoError(error: any): CongoError {
-  if (error instanceof CongoError) {
+export function toTsmdbError(error: any): TsmdbError {
+  if (error instanceof TsmdbError) {
     return error;
   }
-  const congoError = new CongoError(error.message || String(error));
+  const tsmdbError = new TsmdbError(error.message || String(error));
   if (error.code) {
-    congoError.code = error.code;
+    tsmdbError.code = error.code;
   }
   if (error.codeName) {
-    congoError.codeName = error.codeName;
+    tsmdbError.codeName = error.codeName;
   }
-  return congoError;
+  return tsmdbError;
 }

package/ts/{congodb → tsmdb}/index.ts

@@ -1,8 +1,8 @@
-// CongoDB - MongoDB Wire Protocol compatible in-memory database server
-// Use the official MongoDB driver to connect to CongoServer
+// TsmDB - MongoDB Wire Protocol compatible in-memory database server
+// Use the official MongoDB driver to connect to TsmdbServer
 
 // Re-export plugins for external use
-import * as plugins from './congodb.plugins.js';
+import * as plugins from './tsmdb.plugins.js';
 export { plugins };
 
 // Export BSON types for convenience
@@ -12,13 +12,15 @@ export { ObjectId, Binary, Timestamp, Long, Decimal128, UUID } from 'bson';
 export * from './types/interfaces.js';
 
 // Export errors
-export * from './errors/CongoErrors.js';
+export * from './errors/TsmdbErrors.js';
 
 // Export storage adapters
 export type { IStorageAdapter } from './storage/IStorageAdapter.js';
 export { MemoryStorageAdapter } from './storage/MemoryStorageAdapter.js';
 export { FileStorageAdapter } from './storage/FileStorageAdapter.js';
 export { OpLog } from './storage/OpLog.js';
+export { WAL } from './storage/WAL.js';
+export type { IWalEntry, TWalOperation } from './storage/WAL.js';
 
 // Export engines
 export { QueryEngine } from './engine/QueryEngine.js';
@@ -26,12 +28,19 @@ export { UpdateEngine } from './engine/UpdateEngine.js';
 export { AggregationEngine } from './engine/AggregationEngine.js';
 export { IndexEngine } from './engine/IndexEngine.js';
 export { TransactionEngine } from './engine/TransactionEngine.js';
+export { QueryPlanner } from './engine/QueryPlanner.js';
+export type { IQueryPlan, TQueryPlanType } from './engine/QueryPlanner.js';
+export { SessionEngine } from './engine/SessionEngine.js';
+export type { ISession, ISessionEngineOptions } from './engine/SessionEngine.js';
 
-// Export server (the main entry point for using CongoDB)
-export { CongoServer } from './server/CongoServer.js';
-export type { ICongoServerOptions } from './server/CongoServer.js';
+// Export server (the main entry point for using TsmDB)
+export { TsmdbServer } from './server/TsmdbServer.js';
+export type { ITsmdbServerOptions } from './server/TsmdbServer.js';
 
 // Export wire protocol utilities (for advanced usage)
 export { WireProtocol } from './server/WireProtocol.js';
 export { CommandRouter } from './server/CommandRouter.js';
 export type { ICommandHandler, IHandlerContext, ICursorState } from './server/CommandRouter.js';
+
+// Export utilities
+export * from './utils/checksum.js';

package/ts/{congodb → tsmdb}/server/CommandRouter.ts

@@ -1,7 +1,10 @@
-import * as plugins from '../congodb.plugins.js';
+import * as plugins from '../tsmdb.plugins.js';
 import type { IStorageAdapter } from '../storage/IStorageAdapter.js';
 import type { IParsedCommand } from './WireProtocol.js';
-import type { CongoServer } from './CongoServer.js';
+import type { TsmdbServer } from './TsmdbServer.js';
+import { IndexEngine } from '../engine/IndexEngine.js';
+import { TransactionEngine } from '../engine/TransactionEngine.js';
+import { SessionEngine } from '../engine/SessionEngine.js';
 
 // Import handlers
 import { HelloHandler } from './handlers/HelloHandler.js';
@@ -18,10 +21,20 @@ import { AdminHandler } from './handlers/AdminHandler.js';
  */
 export interface IHandlerContext {
   storage: IStorageAdapter;
-  server: CongoServer;
+  server: TsmdbServer;
   database: string;
   command: plugins.bson.Document;
   documentSequences?: Map<string, plugins.bson.Document[]>;
+  /** Get or create an IndexEngine for a collection */
+  getIndexEngine: (collName: string) => IndexEngine;
+  /** Transaction engine instance */
+  transactionEngine: TransactionEngine;
+  /** Current transaction ID (if in a transaction) */
+  txnId?: string;
+  /** Session ID (from lsid) */
+  sessionId?: string;
+  /** Session engine instance */
+  sessionEngine: SessionEngine;
 }
 
 /**
@@ -36,19 +49,61 @@ export interface ICommandHandler {
  */
 export class CommandRouter {
   private storage: IStorageAdapter;
-  private server: CongoServer;
+  private server: TsmdbServer;
   private handlers: Map<string, ICommandHandler> = new Map();
 
   // Cursor state for getMore operations
   private cursors: Map<bigint, ICursorState> = new Map();
   private cursorIdCounter: bigint = BigInt(1);
 
-  constructor(storage: IStorageAdapter, server: CongoServer) {
+  // Index engine cache: db.collection -> IndexEngine
+  private indexEngines: Map<string, IndexEngine> = new Map();
+
+  // Transaction engine (shared across all handlers)
+  private transactionEngine: TransactionEngine;
+
+  // Session engine (shared across all handlers)
+  private sessionEngine: SessionEngine;
+
+  constructor(storage: IStorageAdapter, server: TsmdbServer) {
     this.storage = storage;
     this.server = server;
+    this.transactionEngine = new TransactionEngine(storage);
+    this.sessionEngine = new SessionEngine();
+    // Link session engine to transaction engine for auto-abort on session expiry
+    this.sessionEngine.setTransactionEngine(this.transactionEngine);
     this.registerHandlers();
   }
 
+  /**
+   * Get or create an IndexEngine for a database.collection
+   */
+  getIndexEngine(dbName: string, collName: string): IndexEngine {
+    const key = `${dbName}.${collName}`;
+    let engine = this.indexEngines.get(key);
+    if (!engine) {
+      engine = new IndexEngine(dbName, collName, this.storage);
+      this.indexEngines.set(key, engine);
+    }
+    return engine;
+  }
+
+  /**
+   * Clear index engine cache for a collection (used when collection is dropped)
+   */
+  clearIndexEngineCache(dbName: string, collName?: string): void {
+    if (collName) {
+      this.indexEngines.delete(`${dbName}.${collName}`);
+    } else {
+      // Clear all engines for the database
+      for (const key of this.indexEngines.keys()) {
+        if (key.startsWith(`${dbName}.`)) {
+          this.indexEngines.delete(key);
+        }
+      }
+    }
+  }
+
   /**
    * Register all command handlers
    */
@@ -120,6 +175,29 @@ export class CommandRouter {
   async route(parsedCommand: IParsedCommand): Promise<plugins.bson.Document> {
     const { commandName, command, database, documentSequences } = parsedCommand;
 
+    // Extract session ID from lsid using SessionEngine helper
+    let sessionId = SessionEngine.extractSessionId(command.lsid);
+    let txnId: string | undefined;
+
+    // If we have a session ID, register/touch the session
+    if (sessionId) {
+      this.sessionEngine.getOrCreateSession(sessionId);
+    }
+
+    // Check if this starts a new transaction
+    if (command.startTransaction && sessionId) {
+      txnId = this.transactionEngine.startTransaction(sessionId);
+      this.sessionEngine.startTransaction(sessionId, txnId, command.txnNumber);
+    } else if (sessionId && this.sessionEngine.isInTransaction(sessionId)) {
+      // Continue existing transaction
+      txnId = this.sessionEngine.getTransactionId(sessionId);
+      // Verify transaction is still active
+      if (txnId && !this.transactionEngine.isActive(txnId)) {
+        this.sessionEngine.endTransaction(sessionId);
+        txnId = undefined;
+      }
+    }
+
     // Create handler context
     const context: IHandlerContext = {
       storage: this.storage,
@@ -127,6 +205,11 @@ export class CommandRouter {
       database,
       command,
       documentSequences,
+      getIndexEngine: (collName: string) => this.getIndexEngine(database, collName),
+      transactionEngine: this.transactionEngine,
+      sessionEngine: this.sessionEngine,
+      txnId,
+      sessionId,
     };
 
     // Find handler
@@ -164,6 +247,32 @@ export class CommandRouter {
       };
     }
   }
+
+  /**
+   * Close the command router and cleanup resources
+   */
+  close(): void {
+    // Close session engine (stops cleanup interval, clears sessions)
+    this.sessionEngine.close();
+    // Clear cursors
+    this.cursors.clear();
+    // Clear index engine cache
+    this.indexEngines.clear();
+  }
+
+  /**
+   * Get session engine (for administrative purposes)
+   */
+  getSessionEngine(): SessionEngine {
+    return this.sessionEngine;
+  }
+
+  /**
+   * Get transaction engine (for administrative purposes)
+   */
+  getTransactionEngine(): TransactionEngine {
+    return this.transactionEngine;
+  }
 }
 
 /**

package/ts/{congodb/server/CongoServer.ts → tsmdb/server/TsmdbServer.ts}

@@ -1,5 +1,5 @@
 import * as net from 'net';
-import * as plugins from '../congodb.plugins.js';
+import * as plugins from '../tsmdb.plugins.js';
 import { WireProtocol, OP_QUERY } from './WireProtocol.js';
 import { CommandRouter } from './CommandRouter.js';
 import { MemoryStorageAdapter } from '../storage/MemoryStorageAdapter.js';
@@ -9,7 +9,7 @@ import type { IStorageAdapter } from '../storage/IStorageAdapter.js';
 /**
  * Server configuration options
  */
-export interface ICongoServerOptions {
+export interface ITsmdbServerOptions {
   /** Port to listen on (default: 27017) */
   port?: number;
   /** Host to bind to (default: 127.0.0.1) */
@@ -36,25 +36,25 @@ interface IConnectionState {
 }
 
 /**
- * CongoServer - MongoDB Wire Protocol compatible server
+ * TsmdbServer - MongoDB Wire Protocol compatible server
  *
  * This server implements the MongoDB wire protocol (OP_MSG) to allow
  * official MongoDB drivers to connect and perform operations.
  *
  * @example
  * ```typescript
- * import { CongoServer } from '@push.rocks/smartmongo/congodb';
+ * import { TsmdbServer } from '@push.rocks/smartmongo/tsmdb';
  * import { MongoClient } from 'mongodb';
  *
- * const server = new CongoServer({ port: 27017 });
+ * const server = new TsmdbServer({ port: 27017 });
  * await server.start();
  *
  * const client = new MongoClient('mongodb://127.0.0.1:27017');
  * await client.connect();
  * ```
  */
-export class CongoServer {
-  private options: Required<ICongoServerOptions>;
+export class TsmdbServer {
+  private options: Required<ITsmdbServerOptions>;
   private server: net.Server | null = null;
   private storage: IStorageAdapter;
   private commandRouter: CommandRouter;
@@ -63,7 +63,7 @@ export class CongoServer {
   private isRunning = false;
   private startTime: Date = new Date();
 
-  constructor(options: ICongoServerOptions = {}) {
+  constructor(options: ITsmdbServerOptions = {}) {
     this.options = {
       port: options.port ?? 27017,
       host: options.host ?? '127.0.0.1',
@@ -154,6 +154,9 @@ export class CongoServer {
     }
     this.connections.clear();
 
+    // Close command router (cleans up session engine, cursors, etc.)
+    this.commandRouter.close();
+
     // Close storage
     await this.storage.close();
 

package/ts/{congodb → tsmdb}/server/WireProtocol.ts

@@ -1,4 +1,4 @@
-import * as plugins from '../congodb.plugins.js';
+import * as plugins from '../tsmdb.plugins.js';
 
 /**
  * MongoDB Wire Protocol Implementation