@push.rocks/smartmongo 2.0.14 → 2.1.0

package/ts/congodb/engine/UpdateEngine.ts

@@ -0,0 +1,506 @@
+import * as plugins from '../congodb.plugins.js';
+import type { Document, IStoredDocument } from '../types/interfaces.js';
+import { QueryEngine } from './QueryEngine.js';
+
+/**
+ * Update engine for MongoDB-compatible update operations
+ */
+export class UpdateEngine {
+  /**
+   * Apply an update specification to a document
+   * Returns the updated document or null if no update was applied
+   */
+  static applyUpdate(document: IStoredDocument, update: Document, arrayFilters?: Document[]): IStoredDocument {
+    // Check if this is an aggregation pipeline update
+    if (Array.isArray(update)) {
+      // Aggregation pipeline updates are not yet supported
+      throw new Error('Aggregation pipeline updates are not yet supported');
+    }
+
+    // Check if this is a replacement (no $ operators at top level)
+    const hasOperators = Object.keys(update).some(k => k.startsWith('$'));
+
+    if (!hasOperators) {
+      // This is a replacement - preserve _id
+      return {
+        _id: document._id,
+        ...update,
+      };
+    }
+
+    // Apply update operators
+    const result = this.deepClone(document);
+
+    for (const [operator, operand] of Object.entries(update)) {
+      switch (operator) {
+        case '$set':
+          this.applySet(result, operand);
+          break;
+        case '$unset':
+          this.applyUnset(result, operand);
+          break;
+        case '$inc':
+          this.applyInc(result, operand);
+          break;
+        case '$mul':
+          this.applyMul(result, operand);
+          break;
+        case '$min':
+          this.applyMin(result, operand);
+          break;
+        case '$max':
+          this.applyMax(result, operand);
+          break;
+        case '$rename':
+          this.applyRename(result, operand);
+          break;
+        case '$currentDate':
+          this.applyCurrentDate(result, operand);
+          break;
+        case '$setOnInsert':
+          // Only applied during upsert insert, handled elsewhere
+          break;
+        case '$push':
+          this.applyPush(result, operand, arrayFilters);
+          break;
+        case '$pop':
+          this.applyPop(result, operand);
+          break;
+        case '$pull':
+          this.applyPull(result, operand, arrayFilters);
+          break;
+        case '$pullAll':
+          this.applyPullAll(result, operand);
+          break;
+        case '$addToSet':
+          this.applyAddToSet(result, operand);
+          break;
+        case '$bit':
+          this.applyBit(result, operand);
+          break;
+        default:
+          throw new Error(`Unknown update operator: ${operator}`);
+      }
+    }
+
+    return result;
+  }
+
+  /**
+   * Apply $setOnInsert for upsert operations
+   */
+  static applySetOnInsert(document: IStoredDocument, setOnInsert: Document): IStoredDocument {
+    const result = this.deepClone(document);
+    this.applySet(result, setOnInsert);
+    return result;
+  }
+
+  /**
+   * Deep clone a document
+   */
+  private static deepClone(obj: any): any {
+    if (obj === null || typeof obj !== 'object') {
+      return obj;
+    }
+
+    if (obj instanceof plugins.bson.ObjectId) {
+      return new plugins.bson.ObjectId(obj.toHexString());
+    }
+
+    if (obj instanceof Date) {
+      return new Date(obj.getTime());
+    }
+
+    if (obj instanceof plugins.bson.Timestamp) {
+      return new plugins.bson.Timestamp({ t: obj.high, i: obj.low });
+    }
+
+    if (Array.isArray(obj)) {
+      return obj.map(item => this.deepClone(item));
+    }
+
+    const cloned: any = {};
+    for (const key of Object.keys(obj)) {
+      cloned[key] = this.deepClone(obj[key]);
+    }
+    return cloned;
+  }
+
+  /**
+   * Set a nested value
+   */
+  private static setNestedValue(obj: any, path: string, value: any): void {
+    const parts = path.split('.');
+    let current = obj;
+
+    for (let i = 0; i < parts.length - 1; i++) {
+      const part = parts[i];
+
+      // Handle array index notation
+      const arrayMatch = part.match(/^(\w+)\[(\d+)\]$/);
+      if (arrayMatch) {
+        const [, fieldName, indexStr] = arrayMatch;
+        const index = parseInt(indexStr, 10);
+        if (!(fieldName in current)) {
+          current[fieldName] = [];
+        }
+        if (!current[fieldName][index]) {
+          current[fieldName][index] = {};
+        }
+        current = current[fieldName][index];
+        continue;
+      }
+
+      // Handle numeric index (array positional)
+      const numIndex = parseInt(part, 10);
+      if (!isNaN(numIndex) && Array.isArray(current)) {
+        if (!current[numIndex]) {
+          current[numIndex] = {};
+        }
+        current = current[numIndex];
+        continue;
+      }
+
+      if (!(part in current) || current[part] === null) {
+        current[part] = {};
+      }
+      current = current[part];
+    }
+
+    const lastPart = parts[parts.length - 1];
+    const numIndex = parseInt(lastPart, 10);
+    if (!isNaN(numIndex) && Array.isArray(current)) {
+      current[numIndex] = value;
+    } else {
+      current[lastPart] = value;
+    }
+  }
+
+  /**
+   * Get a nested value
+   */
+  private static getNestedValue(obj: any, path: string): any {
+    return QueryEngine.getNestedValue(obj, path);
+  }
+
+  /**
+   * Delete a nested value
+   */
+  private static deleteNestedValue(obj: any, path: string): void {
+    const parts = path.split('.');
+    let current = obj;
+
+    for (let i = 0; i < parts.length - 1; i++) {
+      const part = parts[i];
+      if (!(part in current)) {
+        return;
+      }
+      current = current[part];
+    }
+
+    delete current[parts[parts.length - 1]];
+  }
+
+  // ============================================================================
+  // Field Update Operators
+  // ============================================================================
+
+  private static applySet(doc: any, fields: Document): void {
+    for (const [path, value] of Object.entries(fields)) {
+      this.setNestedValue(doc, path, this.deepClone(value));
+    }
+  }
+
+  private static applyUnset(doc: any, fields: Document): void {
+    for (const path of Object.keys(fields)) {
+      this.deleteNestedValue(doc, path);
+    }
+  }
+
+  private static applyInc(doc: any, fields: Document): void {
+    for (const [path, value] of Object.entries(fields)) {
+      const current = this.getNestedValue(doc, path) || 0;
+      if (typeof current !== 'number') {
+        throw new Error(`Cannot apply $inc to non-numeric field: ${path}`);
+      }
+      this.setNestedValue(doc, path, current + (value as number));
+    }
+  }
+
+  private static applyMul(doc: any, fields: Document): void {
+    for (const [path, value] of Object.entries(fields)) {
+      const current = this.getNestedValue(doc, path) || 0;
+      if (typeof current !== 'number') {
+        throw new Error(`Cannot apply $mul to non-numeric field: ${path}`);
+      }
+      this.setNestedValue(doc, path, current * (value as number));
+    }
+  }
+
+  private static applyMin(doc: any, fields: Document): void {
+    for (const [path, value] of Object.entries(fields)) {
+      const current = this.getNestedValue(doc, path);
+      if (current === undefined || this.compareValues(value, current) < 0) {
+        this.setNestedValue(doc, path, this.deepClone(value));
+      }
+    }
+  }
+
+  private static applyMax(doc: any, fields: Document): void {
+    for (const [path, value] of Object.entries(fields)) {
+      const current = this.getNestedValue(doc, path);
+      if (current === undefined || this.compareValues(value, current) > 0) {
+        this.setNestedValue(doc, path, this.deepClone(value));
+      }
+    }
+  }
+
+  private static applyRename(doc: any, fields: Document): void {
+    for (const [oldPath, newPath] of Object.entries(fields)) {
+      const value = this.getNestedValue(doc, oldPath);
+      if (value !== undefined) {
+        this.deleteNestedValue(doc, oldPath);
+        this.setNestedValue(doc, newPath as string, value);
+      }
+    }
+  }
+
+  private static applyCurrentDate(doc: any, fields: Document): void {
+    for (const [path, spec] of Object.entries(fields)) {
+      if (spec === true) {
+        this.setNestedValue(doc, path, new Date());
+      } else if (typeof spec === 'object' && spec.$type === 'date') {
+        this.setNestedValue(doc, path, new Date());
+      } else if (typeof spec === 'object' && spec.$type === 'timestamp') {
+        this.setNestedValue(doc, path, new plugins.bson.Timestamp({ t: Math.floor(Date.now() / 1000), i: 0 }));
+      }
+    }
+  }
+
+  // ============================================================================
+  // Array Update Operators
+  // ============================================================================
+
+  private static applyPush(doc: any, fields: Document, arrayFilters?: Document[]): void {
+    for (const [path, spec] of Object.entries(fields)) {
+      let arr = this.getNestedValue(doc, path);
+      if (arr === undefined) {
+        arr = [];
+        this.setNestedValue(doc, path, arr);
+      }
+      if (!Array.isArray(arr)) {
+        throw new Error(`Cannot apply $push to non-array field: ${path}`);
+      }
+
+      if (spec && typeof spec === 'object' && '$each' in spec) {
+        // $push with modifiers
+        let elements = (spec.$each as any[]).map(e => this.deepClone(e));
+        const position = spec.$position as number | undefined;
+        const slice = spec.$slice as number | undefined;
+        const sortSpec = spec.$sort;
+
+        if (position !== undefined) {
+          arr.splice(position, 0, ...elements);
+        } else {
+          arr.push(...elements);
+        }
+
+        if (sortSpec !== undefined) {
+          if (typeof sortSpec === 'number') {
+            arr.sort((a, b) => (a - b) * sortSpec);
+          } else {
+            // Sort by field(s)
+            const entries = Object.entries(sortSpec as Document);
+            arr.sort((a, b) => {
+              for (const [field, dir] of entries) {
+                const av = this.getNestedValue(a, field);
+                const bv = this.getNestedValue(b, field);
+                const cmp = this.compareValues(av, bv) * (dir as number);
+                if (cmp !== 0) return cmp;
+              }
+              return 0;
+            });
+          }
+        }
+
+        if (slice !== undefined) {
+          if (slice >= 0) {
+            arr.splice(slice);
+          } else {
+            arr.splice(0, arr.length + slice);
+          }
+        }
+      } else {
+        // Simple push
+        arr.push(this.deepClone(spec));
+      }
+    }
+  }
+
+  private static applyPop(doc: any, fields: Document): void {
+    for (const [path, direction] of Object.entries(fields)) {
+      const arr = this.getNestedValue(doc, path);
+      if (!Array.isArray(arr)) {
+        throw new Error(`Cannot apply $pop to non-array field: ${path}`);
+      }
+
+      if ((direction as number) === 1) {
+        arr.pop();
+      } else {
+        arr.shift();
+      }
+    }
+  }
+
+  private static applyPull(doc: any, fields: Document, arrayFilters?: Document[]): void {
+    for (const [path, condition] of Object.entries(fields)) {
+      const arr = this.getNestedValue(doc, path);
+      if (!Array.isArray(arr)) {
+        continue; // Skip if not an array
+      }
+
+      if (typeof condition === 'object' && condition !== null && !Array.isArray(condition)) {
+        // Condition is a query filter
+        const hasOperators = Object.keys(condition).some(k => k.startsWith('$'));
+        if (hasOperators) {
+          // Filter using query operators
+          const remaining = arr.filter(item => !QueryEngine.matches(item, condition));
+          arr.length = 0;
+          arr.push(...remaining);
+        } else {
+          // Match documents with all specified fields
+          const remaining = arr.filter(item => {
+            if (typeof item !== 'object' || item === null) {
+              return true;
+            }
+            return !Object.entries(condition).every(([k, v]) => {
+              const itemVal = this.getNestedValue(item, k);
+              return this.valuesEqual(itemVal, v);
+            });
+          });
+          arr.length = 0;
+          arr.push(...remaining);
+        }
+      } else {
+        // Direct value match
+        const remaining = arr.filter(item => !this.valuesEqual(item, condition));
+        arr.length = 0;
+        arr.push(...remaining);
+      }
+    }
+  }
+
+  private static applyPullAll(doc: any, fields: Document): void {
+    for (const [path, values] of Object.entries(fields)) {
+      const arr = this.getNestedValue(doc, path);
+      if (!Array.isArray(arr)) {
+        continue;
+      }
+      if (!Array.isArray(values)) {
+        throw new Error(`$pullAll requires an array argument`);
+      }
+
+      const valueSet = new Set(values.map(v => JSON.stringify(v)));
+      const remaining = arr.filter(item => !valueSet.has(JSON.stringify(item)));
+      arr.length = 0;
+      arr.push(...remaining);
+    }
+  }
+
+  private static applyAddToSet(doc: any, fields: Document): void {
+    for (const [path, spec] of Object.entries(fields)) {
+      let arr = this.getNestedValue(doc, path);
+      if (arr === undefined) {
+        arr = [];
+        this.setNestedValue(doc, path, arr);
+      }
+      if (!Array.isArray(arr)) {
+        throw new Error(`Cannot apply $addToSet to non-array field: ${path}`);
+      }
+
+      const existingSet = new Set(arr.map(v => JSON.stringify(v)));
+
+      if (spec && typeof spec === 'object' && '$each' in spec) {
+        for (const item of spec.$each as any[]) {
+          const key = JSON.stringify(item);
+          if (!existingSet.has(key)) {
+            arr.push(this.deepClone(item));
+            existingSet.add(key);
+          }
+        }
+      } else {
+        const key = JSON.stringify(spec);
+        if (!existingSet.has(key)) {
+          arr.push(this.deepClone(spec));
+        }
+      }
+    }
+  }
+
+  private static applyBit(doc: any, fields: Document): void {
+    for (const [path, operations] of Object.entries(fields)) {
+      let current = this.getNestedValue(doc, path) || 0;
+      if (typeof current !== 'number') {
+        throw new Error(`Cannot apply $bit to non-numeric field: ${path}`);
+      }
+
+      for (const [op, value] of Object.entries(operations as Document)) {
+        switch (op) {
+          case 'and':
+            current = current & (value as number);
+            break;
+          case 'or':
+            current = current | (value as number);
+            break;
+          case 'xor':
+            current = current ^ (value as number);
+            break;
+        }
+      }
+
+      this.setNestedValue(doc, path, current);
+    }
+  }
+
+  // ============================================================================
+  // Helper Methods
+  // ============================================================================
+
+  private static compareValues(a: any, b: any): number {
+    if (a === b) return 0;
+    if (a === null || a === undefined) return -1;
+    if (b === null || b === undefined) return 1;
+
+    if (typeof a === 'number' && typeof b === 'number') {
+      return a - b;
+    }
+
+    if (a instanceof Date && b instanceof Date) {
+      return a.getTime() - b.getTime();
+    }
+
+    if (typeof a === 'string' && typeof b === 'string') {
+      return a.localeCompare(b);
+    }
+
+    return String(a).localeCompare(String(b));
+  }
+
+  private static valuesEqual(a: any, b: any): boolean {
+    if (a === b) return true;
+
+    if (a instanceof plugins.bson.ObjectId && b instanceof plugins.bson.ObjectId) {
+      return a.equals(b);
+    }
+
+    if (a instanceof Date && b instanceof Date) {
+      return a.getTime() === b.getTime();
+    }
+
+    if (typeof a === 'object' && typeof b === 'object' && a !== null && b !== null) {
+      return JSON.stringify(a) === JSON.stringify(b);
+    }
+
+    return false;
+  }
+}

package/ts/congodb/errors/CongoErrors.ts

@@ -0,0 +1,181 @@
+/**
+ * Base error class for all CongoDB errors
+ * Mirrors MongoDB driver error hierarchy
+ */
+export class CongoError extends Error {
+  public code?: number;
+  public codeName?: string;
+
+  constructor(message: string, code?: number, codeName?: string) {
+    super(message);
+    this.name = 'CongoError';
+    this.code = code;
+    this.codeName = codeName;
+    Object.setPrototypeOf(this, new.target.prototype);
+  }
+}
+
+/**
+ * Error thrown during connection issues
+ */
+export class CongoConnectionError extends CongoError {
+  constructor(message: string) {
+    super(message);
+    this.name = 'CongoConnectionError';
+  }
+}
+
+/**
+ * Error thrown when an operation times out
+ */
+export class CongoTimeoutError extends CongoError {
+  constructor(message: string) {
+    super(message, 50, 'MaxTimeMSExpired');
+    this.name = 'CongoTimeoutError';
+  }
+}
+
+/**
+ * Error thrown during write operations
+ */
+export class CongoWriteError extends CongoError {
+  public writeErrors?: IWriteError[];
+  public result?: any;
+
+  constructor(message: string, code?: number, writeErrors?: IWriteError[]) {
+    super(message, code);
+    this.name = 'CongoWriteError';
+    this.writeErrors = writeErrors;
+  }
+}
+
+/**
+ * Error thrown for duplicate key violations
+ */
+export class CongoDuplicateKeyError extends CongoWriteError {
+  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.codeName = 'DuplicateKey';
+    this.keyPattern = keyPattern;
+    this.keyValue = keyValue;
+  }
+}
+
+/**
+ * Error thrown for bulk write failures
+ */
+export class CongoBulkWriteError extends CongoError {
+  public writeErrors: IWriteError[];
+  public result: any;
+
+  constructor(message: string, writeErrors: IWriteError[], result: any) {
+    super(message, 65);
+    this.name = 'CongoBulkWriteError';
+    this.writeErrors = writeErrors;
+    this.result = result;
+  }
+}
+
+/**
+ * Error thrown during transaction operations
+ */
+export class CongoTransactionError extends CongoError {
+  constructor(message: string, code?: number) {
+    super(message, code);
+    this.name = 'CongoTransactionError';
+  }
+}
+
+/**
+ * Error thrown when a transaction is aborted due to conflict
+ */
+export class CongoWriteConflictError extends CongoTransactionError {
+  constructor(message: string = 'Write conflict during transaction') {
+    super(message, 112);
+    this.name = 'CongoWriteConflictError';
+    this.codeName = 'WriteConflict';
+  }
+}
+
+/**
+ * Error thrown for invalid arguments
+ */
+export class CongoArgumentError extends CongoError {
+  constructor(message: string) {
+    super(message);
+    this.name = 'CongoArgumentError';
+  }
+}
+
+/**
+ * Error thrown when an operation is not supported
+ */
+export class CongoNotSupportedError extends CongoError {
+  constructor(message: string) {
+    super(message, 115);
+    this.name = 'CongoNotSupportedError';
+    this.codeName = 'CommandNotSupported';
+  }
+}
+
+/**
+ * Error thrown when cursor is exhausted or closed
+ */
+export class CongoCursorError extends CongoError {
+  constructor(message: string) {
+    super(message);
+    this.name = 'CongoCursorError';
+  }
+}
+
+/**
+ * Error thrown when a namespace (database.collection) is invalid
+ */
+export class CongoNamespaceError extends CongoError {
+  constructor(message: string) {
+    super(message, 73);
+    this.name = 'CongoNamespaceError';
+    this.codeName = 'InvalidNamespace';
+  }
+}
+
+/**
+ * Error thrown when an index operation fails
+ */
+export class CongoIndexError extends CongoError {
+  constructor(message: string, code?: number) {
+    super(message, code || 86);
+    this.name = 'CongoIndexError';
+  }
+}
+
+/**
+ * Write error detail for bulk operations
+ */
+export interface IWriteError {
+  index: number;
+  code: number;
+  errmsg: string;
+  op: any;
+}
+
+/**
+ * Convert any error to a CongoError
+ */
+export function toCongoError(error: any): CongoError {
+  if (error instanceof CongoError) {
+    return error;
+  }
+  const congoError = new CongoError(error.message || String(error));
+  if (error.code) {
+    congoError.code = error.code;
+  }
+  if (error.codeName) {
+    congoError.codeName = error.codeName;
+  }
+  return congoError;
+}