Add MongoDB scripts support with updateMany/updateOne and getSiblingDB

- Add updateMany and updateOne operations for MongoDB
- Add getSiblingDB() support for cross-database operations
- Add script examples (ref-scripts, tenants-scripts) with test coverage
- Handle new Date() conversion to BSON date format in update operations
- Update README with supported operations and cross-database usage notes

Author: lukaszbudnik

README.md

@@ -577,6 +577,14 @@ MongoDB uses database-based multi-tenancy (similar to MySQL). Migrations are Jav
 **Supported operations:**
 - `db.collection.insertOne({...})` - Insert a document
 - `db.collection.createIndex({...}, {...})` - Create an index
+- `db.collection.updateOne({filter}, {update})` - Update a single document
+- `db.collection.updateMany({filter}, {update})` - Update multiple documents
+- `db.getSiblingDB('dbname').collection.operation(...)` - Access different database
+
+**Supported JavaScript features:**
+- `new Date()` - Automatically converted to current timestamp
+
+**Note:** Single scripts execute in their source directory database (e.g., `ref-scripts`). To access collections in other databases, use `db.getSiblingDB('dbname')`.
 
 Sample MongoDB configuration:
 
@@ -585,9 +593,14 @@ baseLocation: migrations
 driver: mongodb
 dataSource: "mongodb://localhost:27017"
 singleMigrations:
-  - admin
+  - ref
+  - configuration
+singleScripts:
+  - ref-scripts
 tenantMigrations:
   - tenants
+tenantScripts:
+  - tenants-scripts
 ```
 
 Sample migration file (`001_create_users.js`):

db/db_mongodb.go

@@ -530,11 +530,41 @@ func (mc *mongoDBConnector) executeMigration(migration types.Migration, dbName s
 func (mc *mongoDBConnector) executeMongoDBCommand(targetDB *mongo.Database, command string) error {
 	command = strings.TrimSpace(command)
 
-	// Match pattern: db.collectionName.operation(...)
+	// Match pattern: db.collectionName.operation(...) or db.getSiblingDB('dbname').collectionName.operation(...)
 	if !strings.HasPrefix(command, "db.") {
 		return nil // Skip non-db commands
 	}
 
+	// Check for getSiblingDB pattern
+	if strings.HasPrefix(command, "db.getSiblingDB(") {
+		// Extract database name from getSiblingDB('dbname')
+		start := strings.Index(command, "'")
+		if start == -1 {
+			start = strings.Index(command, "\"")
+		}
+		if start == -1 {
+			return fmt.Errorf("invalid getSiblingDB syntax: %s", command)
+		}
+
+		end := strings.Index(command[start+1:], "'")
+		if end == -1 {
+			end = strings.Index(command[start+1:], "\"")
+		}
+		if end == -1 {
+			return fmt.Errorf("invalid getSiblingDB syntax: %s", command)
+		}
+
+		dbName := command[start+1 : start+1+end]
+		targetDB = mc.client.Database(dbName)
+
+		// Find the collection part after getSiblingDB('dbname').
+		dotAfterDB := strings.Index(command[start+1+end:], ".")
+		if dotAfterDB == -1 {
+			return fmt.Errorf("invalid getSiblingDB syntax: %s", command)
+		}
+		command = "db." + command[start+1+end+dotAfterDB+1:]
+	}
+
 	// Extract collection name and operation
 	parts := strings.SplitN(command[3:], ".", 2) // Remove "db." prefix
 	if len(parts) < 2 {
@@ -559,6 +589,10 @@ func (mc *mongoDBConnector) executeMongoDBCommand(targetDB *mongo.Database, comm
 		return mc.handleInsertOne(col, rest[opEnd:])
 	case "createIndex":
 		return mc.handleCreateIndex(col, rest[opEnd:])
+	case "updateMany":
+		return mc.handleUpdateMany(col, rest[opEnd:])
+	case "updateOne":
+		return mc.handleUpdateOne(col, rest[opEnd:])
 	default:
 		common.LogWarn(mc.ctx, "Unsupported operation: %s", operation)
 		return nil
@@ -653,6 +687,104 @@ func (mc *mongoDBConnector) handleCreateIndex(col *mongo.Collection, args string
 	return err
 }
 
+// handleUpdateMany executes updateMany operation
+func (mc *mongoDBConnector) handleUpdateMany(col *mongo.Collection, args string) error {
+	return mc.handleUpdate(col, args, true)
+}
+
+// handleUpdateOne executes updateOne operation
+func (mc *mongoDBConnector) handleUpdateOne(col *mongo.Collection, args string) error {
+	return mc.handleUpdate(col, args, false)
+}
+
+// handleUpdate executes update operations (updateOne or updateMany)
+func (mc *mongoDBConnector) handleUpdate(col *mongo.Collection, args string, many bool) error {
+	// Extract filter and update documents from updateMany({filter}, {update}, {options})
+	start := strings.Index(args, "{")
+	if start == -1 {
+		return fmt.Errorf("invalid update syntax")
+	}
+
+	// Find the matching closing brace for the first argument (filter)
+	braceCount := 0
+	firstArgEnd := -1
+	for i := start; i < len(args); i++ {
+		if args[i] == '{' {
+			braceCount++
+		} else if args[i] == '}' {
+			braceCount--
+			if braceCount == 0 {
+				firstArgEnd = i
+				break
+			}
+		}
+	}
+
+	if firstArgEnd == -1 {
+		return fmt.Errorf("invalid update syntax")
+	}
+
+	filterJSON := args[start : firstArgEnd+1]
+	filterJSON = mc.jsToJSON(filterJSON)
+
+	// Parse filter
+	var filter bson.M
+	if err := bson.UnmarshalExtJSON([]byte(filterJSON), false, &filter); err != nil {
+		return fmt.Errorf("failed to parse filter: %v", err)
+	}
+
+	// Find second argument (update document)
+	remaining := strings.TrimSpace(args[firstArgEnd+1:])
+	if !strings.HasPrefix(remaining, ",") {
+		return fmt.Errorf("missing update document")
+	}
+	remaining = strings.TrimSpace(remaining[1:])
+
+	updateStart := strings.Index(remaining, "{")
+	if updateStart == -1 {
+		return fmt.Errorf("invalid update document")
+	}
+
+	// Find the matching closing brace for the update document
+	braceCount = 0
+	updateEnd := -1
+	for i := updateStart; i < len(remaining); i++ {
+		if remaining[i] == '{' {
+			braceCount++
+		} else if remaining[i] == '}' {
+			braceCount--
+			if braceCount == 0 {
+				updateEnd = i
+				break
+			}
+		}
+	}
+
+	if updateEnd == -1 {
+		return fmt.Errorf("invalid update document")
+	}
+
+	updateJSON := remaining[updateStart : updateEnd+1]
+	updateJSON = mc.jsToJSON(updateJSON)
+
+	// Handle new Date() - replace with current time in extended JSON format
+	updateJSON = strings.ReplaceAll(updateJSON, "new Date()", fmt.Sprintf("{\"$date\":\"%s\"}", time.Now().Format(time.RFC3339Nano)))
+
+	// Parse update document
+	var update bson.M
+	if err := bson.UnmarshalExtJSON([]byte(updateJSON), false, &update); err != nil {
+		return fmt.Errorf("failed to parse update document: %v", err)
+	}
+
+	// Execute update
+	if many {
+		_, err := col.UpdateMany(mc.ctx, filter, update)
+		return err
+	}
+	_, err := col.UpdateOne(mc.ctx, filter, update)
+	return err
+}
+
 // jsToJSON converts JavaScript object notation to proper JSON
 // Handles: {key: value} -> {"key": value}, {key: 'value'} -> {"key": "value"}
 func (mc *mongoDBConnector) jsToJSON(js string) string {

db/db_mongodb_integration_test.go

@@ -86,6 +86,43 @@ func TestMongoDBCreateVersion(t *testing.T) {
 	assert.Equal(t, lenBefore+int(results.MigrationsGrandTotal), lenAfter)
 }
 
+func TestMongoDBScripts(t *testing.T) {
+	configFile := "../test/migrator-mongodb.yaml"
+	config, err := config.FromFile(configFile)
+	assert.Nil(t, err)
+
+	connector := New(newTestContext(), config)
+	defer connector.Dispose()
+
+	// Create test tenant with some initial data
+	testTenant := fmt.Sprintf("scripttenant%d", time.Now().UnixNano())
+	m1 := time.Now().UnixNano()
+	tenantMigration := types.Migration{Name: fmt.Sprintf("%v.js", m1), SourceDir: "tenants", File: fmt.Sprintf("tenants/%v.js", m1), MigrationType: types.MigrationTypeTenantMigration, Contents: "db.settings.insertOne({k: 999, v: '999'})"}
+	connector.CreateTenant(testTenant, "test-tenant-scripts", types.ActionApply, []types.Migration{tenantMigration}, false)
+
+	tenants := connector.GetTenants()
+	noOfTenants := len(tenants)
+
+	s1 := time.Now().UnixNano()
+	ts1 := time.Now().UnixNano()
+
+	// Single script - runs every time, uses getSiblingDB to access ref database
+	singleScript := types.Migration{Name: fmt.Sprintf("%v.js", s1), SourceDir: "ref-scripts", File: fmt.Sprintf("ref-scripts/%v.js", s1), MigrationType: types.MigrationTypeSingleScript, Contents: "db.getSiblingDB('ref').modules.updateMany({}, {$set: {script_updated: new Date()}})"}
+
+	// Tenant script - runs every time for each tenant, uses updateMany
+	tenantScript := types.Migration{Name: fmt.Sprintf("%v.js", ts1), SourceDir: "tenants-scripts", File: fmt.Sprintf("tenants-scripts/%v.js", ts1), MigrationType: types.MigrationTypeTenantScript, Contents: "db.settings.updateMany({}, {$set: {script_run_at: new Date()}})"}
+
+	scriptsToApply := []types.Migration{singleScript, tenantScript}
+
+	results, version := connector.CreateVersion("test-scripts", types.ActionApply, scriptsToApply, false)
+
+	assert.NotNil(t, version)
+	assert.Equal(t, int32(1), results.SingleScripts)
+	assert.Equal(t, int32(1), results.TenantScripts)
+	assert.Equal(t, int32(noOfTenants), results.TenantScriptsTotal)
+	assert.Equal(t, int32(1+noOfTenants), results.ScriptsGrandTotal)
+}
+
 func TestMongoDBCreateTenant(t *testing.T) {
 	configFile := "../test/migrator-mongodb.yaml"
 	config, err := config.FromFile(configFile)

test/migrations-mongodb/ref-scripts/update_stats.js

@@ -0,0 +1,5 @@
+// This script runs every time to update statistics in ref database
+db.getSiblingDB('ref').modules.updateMany(
+  {},
+  { $set: { last_updated: new Date() } }
+);

test/migrations-mongodb/tenants-scripts/refresh_cache.js

@@ -0,0 +1,5 @@
+// This script runs every time for each tenant to refresh cache
+db.settings.updateMany(
+  {},
+  { $set: { cached_at: new Date() } }
+);

test/migrator-mongodb.yaml

@@ -4,5 +4,9 @@ dataSource: "mongodb://localhost:27017"
 singleMigrations:
   - ref
   - configuration
+singleScripts:
+  - ref-scripts
 tenantMigrations:
   - tenants
+tenantScripts:
+  - tenants-scripts