Merge pull request #18 from Thomas-Matheus/mariadb-support

Add Maria DB Connector

Author: tianzhou

README.md

@@ -49,15 +49,15 @@ https://demo.dbhub.ai/sse connects a [sample employee database](https://github.c
 
 | Tool            | Command Name      | PostgreSQL | MySQL | SQL Server | SQLite |
 | --------------- | ----------------- | :--------: | :---: | :--------: | :----: |
-| Execute Query   | `run_query`       |     ✅     |  ✅   |     ✅     |   ✅   |
-| List Connectors | `list_connectors` |     ✅     |  ✅   |     ✅     |   ✅   |
+| Execute Query   | `run_query`       |     ✅     |  ✅   |     ✅     |   ✅   |   ✅   |
+| List Connectors | `list_connectors` |     ✅     |  ✅   |     ✅     |   ✅   |   ✅   |
 
 ### Prompt Capabilities
 
 | Prompt              | Command Name   | PostgreSQL | MySQL | SQL Server | SQLite |
 | ------------------- | -------------- | :--------: | :---: | :--------: | :----: |
-| Generate SQL        | `generate_sql` |     ✅     |  ✅   |     ✅     |   ✅   |
-| Explain DB Elements | `explain_db`   |     ✅     |  ✅   |     ✅     |   ✅   |
+| Generate SQL        | `generate_sql` |     ✅     |  ✅   |     ✅     |   ✅   |   ✅   |
+| Explain DB Elements | `explain_db`   |     ✅     |  ✅   |     ✅     |   ✅   |   ✅   |
 
 ## Installation
 
@@ -192,6 +192,7 @@ DBHub supports the following database connection string formats:
 | SQLite     | `sqlite:///[path/to/file]` or `sqlite::memory:`          | `sqlite:///path/to/database.db` or `sqlite::memory:`             |
 | SQL Server | `sqlserver://[user]:[password]@[host]:[port]/[database]` | `sqlserver://user:password@localhost:1433/dbname`                |
 | MySQL      | `mysql://[user]:[password]@[host]:[port]/[database]`     | `mysql://user:password@localhost:3306/dbname`                    |
+| MariaDB    | `mariadb://[user]:[password]@[host]:[port]/[database]`   | `mariadb://user:password@localhost:3306/dbname`                  |
 
 ### Transport
 

package.json

@@ -27,6 +27,7 @@
     "dotenv": "^16.4.7",
     "express": "^4.18.2",
     "mssql": "^11.0.1",
+    "mariadb": "^3.4.0",
     "mysql2": "^3.13.0",
     "pg": "^8.13.3",
     "zod": "^3.24.2"

src/config/env.ts

@@ -166,4 +166,28 @@ export function resolvePort(): { port: number; source: string } {
 
   // 3. Default to 8080
   return { port: 8080, source: 'default' };
-}
+}
+
+/**
+ * Redact sensitive information from a DSN string
+ * Replaces the password with asterisks
+ * @param dsn - The DSN string to redact
+ * @returns The sanitized DSN string
+ */
+export function redactDSN(dsn: string): string {
+  try {
+    // Create a URL object to parse the DSN
+    const url = new URL(dsn);
+    
+    // Replace the password with asterisks
+    if (url.password) {
+      url.password = '*******';
+    }
+    
+    // Return the sanitized DSN
+    return url.toString();
+  } catch (error) {
+    // If parsing fails, do basic redaction with regex
+    return dsn.replace(/\/\/([^:]+):([^@]+)@/, '//$1:***@');
+  }
+}

src/connectors/interface.ts

@@ -39,6 +39,7 @@ export interface DSNParser {
    * Parse a connection string into connector-specific configuration
    * Example DSN formats:
    * - PostgreSQL: "postgres://user:password@localhost:5432/dbname?sslmode=disable"
+   * - MariaDB: "mariadb://user:password@localhost:3306/dbname"
    * - MySQL: "mysql://user:password@localhost:3306/dbname"
    * - SQLite: "sqlite:///path/to/database.db" or "sqlite::memory:"
    */