feat: add MongoDB configurable authSource and fix connection field persistence

datlechin · datlechin · commit 09755c8e8cd2 · 2026-03-11T11:32:19.000+07:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -10,9 +10,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 
 - DuckDB database support — connect to `.duckdb` files, query CSV/Parquet/JSON files via SQL, schema navigation, and DuckDB extension management
+- MongoDB configurable auth database (`authSource`) — authenticate against any database instead of hardcoded `admin`
 
 ### Fixed
 
+- MongoDB Read Preference, Write Concern, and Redis Database not persisted across app restarts
+
 - Result truncation at 100K rows now reported to UI via `PluginQueryResult.isTruncated` instead of being silently discarded
 - DELETE and UPDATE queries using all columns in WHERE clause instead of just the primary key for PostgreSQL, Redshift, MSSQL, and ClickHouse
 - SSL/TLS always being enabled for MongoDB, Redis, and ClickHouse connections due to case mismatch in SSL mode string comparison (#249)
diff --git a/Plugins/MongoDBDriverPlugin/MongoDBConnection.swift b/Plugins/MongoDBDriverPlugin/MongoDBConnection.swift
@@ -59,6 +59,7 @@ final class MongoDBConnection: @unchecked Sendable {
     private let sslMode: String
     private let sslCACertPath: String
     private let sslClientCertPath: String
+    private let authSource: String?
     private let readPreference: String?
     private let writeConcern: String?
 
@@ -111,6 +112,7 @@ final class MongoDBConnection: @unchecked Sendable {
         sslMode: String = "Disabled",
         sslCACertPath: String = "",
         sslClientCertPath: String = "",
+        authSource: String? = nil,
         readPreference: String? = nil,
         writeConcern: String? = nil
     ) {
@@ -122,6 +124,7 @@ final class MongoDBConnection: @unchecked Sendable {
         self.sslMode = sslMode
         self.sslCACertPath = sslCACertPath
         self.sslClientCertPath = sslClientCertPath
+        self.authSource = authSource
         self.readPreference = readPreference
         self.writeConcern = writeConcern
     }
@@ -167,10 +170,11 @@ final class MongoDBConnection: @unchecked Sendable {
         uri += "\(encodedHost):\(port)"
         uri += database.isEmpty ? "/" : "/\(encodedDb)"
 
+        let effectiveAuthSource = authSource.flatMap { $0.isEmpty ? nil : $0 } ?? "admin"
         var params: [String] = [
             "connectTimeoutMS=10000",
             "serverSelectionTimeoutMS=10000",
-            "authSource=admin"
+            "authSource=\(effectiveAuthSource)"
         ]
 
         let sslEnabled = ["Preferred", "Required", "Verify CA", "Verify Identity"].contains(sslMode)
diff --git a/Plugins/MongoDBDriverPlugin/MongoDBPlugin.swift b/Plugins/MongoDBDriverPlugin/MongoDBPlugin.swift
@@ -17,6 +17,7 @@ final class MongoDBPlugin: NSObject, TableProPlugin, DriverPlugin {
     static let iconName = "leaf.fill"
     static let defaultPort = 27017
     static let additionalConnectionFields: [ConnectionField] = [
+        ConnectionField(id: "mongoAuthSource", label: "Auth Database", placeholder: "admin"),
         ConnectionField(id: "mongoReadPreference", label: "Read Preference", placeholder: "primary"),
         ConnectionField(id: "mongoWriteConcern", label: "Write Concern", placeholder: "majority")
     ]
diff --git a/Plugins/MongoDBDriverPlugin/MongoDBPluginDriver.swift b/Plugins/MongoDBDriverPlugin/MongoDBPluginDriver.swift
@@ -40,6 +40,7 @@ final class MongoDBPluginDriver: PluginDatabaseDriver {
             sslMode: config.additionalFields["sslMode"] ?? "Disabled",
             sslCACertPath: config.additionalFields["sslCACertPath"] ?? "",
             sslClientCertPath: config.additionalFields["sslClientCertPath"] ?? "",
+            authSource: config.additionalFields["mongoAuthSource"],
             readPreference: config.additionalFields["mongoReadPreference"],
             writeConcern: config.additionalFields["mongoWriteConcern"]
         )
diff --git a/TablePro/AppDelegate+ConnectionHandler.swift b/TablePro/AppDelegate+ConnectionHandler.swift
@@ -404,6 +404,7 @@ extension AppDelegate {
             sslConfig: sslConfig,
             color: color,
             tagId: tagId,
+            mongoAuthSource: parsed.authSource,
             redisDatabase: parsed.redisDatabase,
             oracleServiceName: parsed.oracleServiceName
         )
diff --git a/TablePro/Core/Database/DatabaseDriver.swift b/TablePro/Core/Database/DatabaseDriver.swift
@@ -346,6 +346,7 @@ enum DatabaseDriverFactory {
         switch connection.type {
         case .mongodb:
             fields["sslCACertPath"] = ssl.caCertificatePath
+            fields["mongoAuthSource"] = connection.mongoAuthSource ?? ""
             fields["mongoReadPreference"] = connection.mongoReadPreference ?? ""
             fields["mongoWriteConcern"] = connection.mongoWriteConcern ?? ""
         case .redis:
diff --git a/TablePro/Core/Storage/ConnectionStorage.swift b/TablePro/Core/Storage/ConnectionStorage.swift
@@ -121,6 +121,7 @@ final class ConnectionStorage {
             groupId: connection.groupId,
             safeModeLevel: connection.safeModeLevel,
             aiPolicy: connection.aiPolicy,
+            mongoAuthSource: connection.mongoAuthSource,
             mongoReadPreference: connection.mongoReadPreference,
             mongoWriteConcern: connection.mongoWriteConcern,
             redisDatabase: connection.redisDatabase,
@@ -368,6 +369,14 @@ private struct StoredConnection: Codable {
     // AI policy
     let aiPolicy: String?
 
+    // MongoDB-specific
+    let mongoAuthSource: String?
+    let mongoReadPreference: String?
+    let mongoWriteConcern: String?
+
+    // Redis-specific
+    let redisDatabase: Int?
+
     // MSSQL schema
     let mssqlSchema: String?
 
@@ -413,6 +422,14 @@ private struct StoredConnection: Codable {
         // AI policy
         self.aiPolicy = connection.aiPolicy?.rawValue
 
+        // MongoDB-specific
+        self.mongoAuthSource = connection.mongoAuthSource
+        self.mongoReadPreference = connection.mongoReadPreference
+        self.mongoWriteConcern = connection.mongoWriteConcern
+
+        // Redis-specific
+        self.redisDatabase = connection.redisDatabase
+
         // MSSQL schema
         self.mssqlSchema = connection.mssqlSchema
 
@@ -431,7 +448,9 @@ private struct StoredConnection: Codable {
         case color, tagId, groupId
         case safeModeLevel
         case isReadOnly // Legacy key for migration reading only
-        case aiPolicy, mssqlSchema, oracleServiceName, startupCommands
+        case aiPolicy
+        case mongoAuthSource, mongoReadPreference, mongoWriteConcern, redisDatabase
+        case mssqlSchema, oracleServiceName, startupCommands
     }
 
     func encode(to encoder: Encoder) throws {
@@ -460,6 +479,10 @@ private struct StoredConnection: Codable {
         try container.encodeIfPresent(groupId, forKey: .groupId)
         try container.encode(safeModeLevel, forKey: .safeModeLevel)
         try container.encodeIfPresent(aiPolicy, forKey: .aiPolicy)
+        try container.encodeIfPresent(mongoAuthSource, forKey: .mongoAuthSource)
+        try container.encodeIfPresent(mongoReadPreference, forKey: .mongoReadPreference)
+        try container.encodeIfPresent(mongoWriteConcern, forKey: .mongoWriteConcern)
+        try container.encodeIfPresent(redisDatabase, forKey: .redisDatabase)
         try container.encodeIfPresent(mssqlSchema, forKey: .mssqlSchema)
         try container.encodeIfPresent(oracleServiceName, forKey: .oracleServiceName)
         try container.encodeIfPresent(startupCommands, forKey: .startupCommands)
@@ -506,6 +529,10 @@ private struct StoredConnection: Codable {
             safeModeLevel = wasReadOnly ? SafeModeLevel.readOnly.rawValue : SafeModeLevel.silent.rawValue
         }
         aiPolicy = try container.decodeIfPresent(String.self, forKey: .aiPolicy)
+        mongoAuthSource = try container.decodeIfPresent(String.self, forKey: .mongoAuthSource)
+        mongoReadPreference = try container.decodeIfPresent(String.self, forKey: .mongoReadPreference)
+        mongoWriteConcern = try container.decodeIfPresent(String.self, forKey: .mongoWriteConcern)
+        redisDatabase = try container.decodeIfPresent(Int.self, forKey: .redisDatabase)
         mssqlSchema = try container.decodeIfPresent(String.self, forKey: .mssqlSchema)
         oracleServiceName = try container.decodeIfPresent(String.self, forKey: .oracleServiceName)
         startupCommands = try container.decodeIfPresent(String.self, forKey: .startupCommands)
@@ -550,6 +577,10 @@ private struct StoredConnection: Codable {
             groupId: parsedGroupId,
             safeModeLevel: SafeModeLevel(rawValue: safeModeLevel) ?? .silent,
             aiPolicy: parsedAIPolicy,
+            mongoAuthSource: mongoAuthSource,
+            mongoReadPreference: mongoReadPreference,
+            mongoWriteConcern: mongoWriteConcern,
+            redisDatabase: redisDatabase,
             mssqlSchema: mssqlSchema,
             oracleServiceName: oracleServiceName,
             startupCommands: startupCommands
diff --git a/TablePro/Models/Connection/DatabaseConnection.swift b/TablePro/Models/Connection/DatabaseConnection.swift
@@ -405,6 +405,7 @@ struct DatabaseConnection: Identifiable, Hashable {
     var groupId: UUID?
     var safeModeLevel: SafeModeLevel
     var aiPolicy: AIConnectionPolicy?
+    var mongoAuthSource: String?
     var mongoReadPreference: String?
     var mongoWriteConcern: String?
     var redisDatabase: Int?
@@ -427,6 +428,7 @@ struct DatabaseConnection: Identifiable, Hashable {
         groupId: UUID? = nil,
         safeModeLevel: SafeModeLevel = .silent,
         aiPolicy: AIConnectionPolicy? = nil,
+        mongoAuthSource: String? = nil,
         mongoReadPreference: String? = nil,
         mongoWriteConcern: String? = nil,
         redisDatabase: Int? = nil,
@@ -448,6 +450,7 @@ struct DatabaseConnection: Identifiable, Hashable {
         self.groupId = groupId
         self.safeModeLevel = safeModeLevel
         self.aiPolicy = aiPolicy
+        self.mongoAuthSource = mongoAuthSource
         self.mongoReadPreference = mongoReadPreference
         self.mongoWriteConcern = mongoWriteConcern
         self.redisDatabase = redisDatabase
diff --git a/TablePro/Views/Connection/ConnectionFormView.swift b/TablePro/Views/Connection/ConnectionFormView.swift
@@ -72,6 +72,7 @@ struct ConnectionFormView: View {
     @State private var aiPolicy: AIConnectionPolicy?
 
     // MongoDB-specific settings
+    @State private var mongoAuthSource: String = ""
     @State private var mongoReadPreference: String = ""
     @State private var mongoWriteConcern: String = ""
 
@@ -572,6 +573,11 @@ struct ConnectionFormView: View {
         Form {
             if type == .mongodb {
                 Section("MongoDB") {
+                    TextField(
+                        String(localized: "Auth Database"),
+                        text: $mongoAuthSource,
+                        prompt: Text("admin")
+                    )
                     Picker(String(localized: "Read Preference"), selection: $mongoReadPreference) {
                         Text(String(localized: "Default")).tag("")
                         Text("Primary").tag("primary")
@@ -813,6 +819,7 @@ struct ConnectionFormView: View {
             aiPolicy = existing.aiPolicy
 
             // Load MongoDB settings
+            mongoAuthSource = existing.mongoAuthSource ?? ""
             mongoReadPreference = existing.mongoReadPreference ?? ""
             mongoWriteConcern = existing.mongoWriteConcern ?? ""
 
@@ -884,6 +891,7 @@ struct ConnectionFormView: View {
             groupId: selectedGroupId,
             safeModeLevel: safeModeLevel,
             aiPolicy: aiPolicy,
+            mongoAuthSource: mongoAuthSource.isEmpty ? nil : mongoAuthSource,
             mongoReadPreference: mongoReadPreference.isEmpty ? nil : mongoReadPreference,
             mongoWriteConcern: mongoWriteConcern.isEmpty ? nil : mongoWriteConcern,
             mssqlSchema: mssqlSchema.isEmpty ? nil : mssqlSchema,
@@ -1015,6 +1023,7 @@ struct ConnectionFormView: View {
             color: connectionColor,
             tagId: selectedTagId,
             groupId: selectedGroupId,
+            mongoAuthSource: mongoAuthSource.isEmpty ? nil : mongoAuthSource,
             mongoReadPreference: mongoReadPreference.isEmpty ? nil : mongoReadPreference,
             mongoWriteConcern: mongoWriteConcern.isEmpty ? nil : mongoWriteConcern,
             mssqlSchema: mssqlSchema.isEmpty ? nil : mssqlSchema,
@@ -1154,6 +1163,9 @@ struct ConnectionFormView: View {
                     applySSHAgentSocketPath(parsed.agentSocket ?? "")
                 }
             }
+            if let authSourceValue = parsed.authSource, !authSourceValue.isEmpty {
+                mongoAuthSource = authSourceValue
+            }
             if let connectionName = parsed.connectionName, !connectionName.isEmpty {
                 name = connectionName
             } else if name.isEmpty {

Original file line number	Diff line number	Diff line change
`@@ -17,6 +17,7 @@ final class MongoDBPlugin: NSObject, TableProPlugin, DriverPlugin {`
`17`	`17`	`static let iconName = "leaf.fill"`
`18`	`18`	`static let defaultPort = 27017`
`19`	`19`	`static let additionalConnectionFields: [ConnectionField] = [`
	`20`	`+ ConnectionField(id: "mongoAuthSource", label: "Auth Database", placeholder: "admin"),`
`20`	`21`	`ConnectionField(id: "mongoReadPreference", label: "Read Preference", placeholder: "primary"),`
`21`	`22`	`ConnectionField(id: "mongoWriteConcern", label: "Write Concern", placeholder: "majority")`
`22`	`23`	`]`
Original file line number	Diff line number	Diff line change
`@@ -40,6 +40,7 @@ final class MongoDBPluginDriver: PluginDatabaseDriver {`
`40`	`40`	`sslMode: config.additionalFields["sslMode"] ?? "Disabled",`
`41`	`41`	`sslCACertPath: config.additionalFields["sslCACertPath"] ?? "",`
`42`	`42`	`sslClientCertPath: config.additionalFields["sslClientCertPath"] ?? "",`
	`43`	`+ authSource: config.additionalFields["mongoAuthSource"],`
`43`	`44`	`readPreference: config.additionalFields["mongoReadPreference"],`
`44`	`45`	`writeConcern: config.additionalFields["mongoWriteConcern"]`
`45`	`46`	`)`
Original file line number	Diff line number	Diff line change
`@@ -404,6 +404,7 @@ extension AppDelegate {`
`404`	`404`	`sslConfig: sslConfig,`
`405`	`405`	`color: color,`
`406`	`406`	`tagId: tagId,`
	`407`	`+ mongoAuthSource: parsed.authSource,`
`407`	`408`	`redisDatabase: parsed.redisDatabase,`
`408`	`409`	`oracleServiceName: parsed.oracleServiceName`
`409`	`410`	`)`