feat: add Azure Auth

Author: tianzhou

README.md

@@ -194,6 +194,14 @@ DBHub supports the following database connection string formats:
 | SQL Server | `sqlserver://[user]:[password]@[host]:[port]/[database]` | `sqlserver://user:password@localhost:1433/dbname`                |
 | SQLite     | `sqlite:///[path/to/file]` or `sqlite::memory:`          | `sqlite:///path/to/database.db` or `sqlite::memory:`             |
 
+#### SQL Server
+
+Extra query parameters:
+
+#### authentication
+
+- `authentication=azure-active-directory-access-token`. Only applicable when running from Azure. See [DefaultAzureCredential](https://learn.microsoft.com/en-us/azure/developer/javascript/sdk/authentication/credential-chains#use-defaultazurecredential-for-flexibility).
+
 ### Transport
 
 - **stdio** (default) - for direct integration with tools like Claude Desktop:

src/connectors/interface.ts

@@ -43,7 +43,7 @@ export interface DSNParser {
    * - MySQL: "mysql://user:password@localhost:3306/dbname"
    * - SQLite: "sqlite:///path/to/database.db" or "sqlite::memory:"
    */
-  parse(dsn: string): any;
+  parse(dsn: string): Promise<any>;
 
   /**
    * Generate a sample DSN string for this connector type

src/connectors/mariadb/index.ts

@@ -6,7 +6,7 @@ import { Connector, ConnectorRegistry, DSNParser, QueryResult, TableColumn, Tabl
  * Handles DSN strings like: mariadb://user:password@localhost:3306/dbname
  */
 class MariadbDSNParser implements DSNParser {
-  parse(dsn: string): mariadb.ConnectionConfig {
+  async parse(dsn: string): Promise<mariadb.ConnectionConfig> {
     // Basic validation
     if (!this.isValidDSN(dsn)) {
       throw new Error(`Invalid MariaDB DSN: ${dsn}`);
@@ -63,7 +63,7 @@ export class MariaDBConnector implements Connector {
 
   async connect(dsn: string): Promise<void> {
     try {
-      const config = this.dsnParser.parse(dsn);
+      const config = await this.dsnParser.parse(dsn);
 
       this.pool = mariadb.createPool(config);
 

src/connectors/mysql/index.ts

@@ -6,7 +6,7 @@ import { Connector, ConnectorRegistry, DSNParser, QueryResult, TableColumn, Tabl
  * Handles DSN strings like: mysql://user:password@localhost:3306/dbname
  */
 class MySQLDSNParser implements DSNParser {
-  parse(dsn: string): mysql.ConnectionOptions {
+  async parse(dsn: string): Promise<mysql.ConnectionOptions> {
     // Basic validation
     if (!this.isValidDSN(dsn)) {
       throw new Error(`Invalid MySQL DSN: ${dsn}`);
@@ -63,7 +63,7 @@ export class MySQLConnector implements Connector {
 
   async connect(dsn: string): Promise<void> {
     try {
-      const config = this.dsnParser.parse(dsn);
+      const config = await this.dsnParser.parse(dsn);
       this.pool = mysql.createPool(config);
 
       // Test the connection

src/connectors/postgres/index.ts

@@ -7,7 +7,7 @@ import { Connector, ConnectorRegistry, DSNParser, QueryResult, TableColumn, Tabl
  * Handles DSN strings like: postgres://user:password@localhost:5432/dbname?sslmode=disable
  */
 class PostgresDSNParser implements DSNParser {
-  parse(dsn: string): pg.PoolConfig {
+  async parse(dsn: string): Promise<pg.PoolConfig> {
     // Basic validation
     if (!this.isValidDSN(dsn)) {
       throw new Error(`Invalid PostgreSQL DSN: ${dsn}`);
@@ -66,7 +66,7 @@ export class PostgresConnector implements Connector {
 
   async connect(dsn: string): Promise<void> {
     try {
-      const config = this.dsnParser.parse(dsn);
+      const config = await this.dsnParser.parse(dsn);
       this.pool = new Pool(config);
 
       // Test the connection

src/connectors/sqlite/index.ts

@@ -18,7 +18,7 @@ import Database from 'better-sqlite3';
  * - sqlite::memory: (in-memory database)
  */
 class SQLiteDSNParser implements DSNParser {
-  parse(dsn: string): { dbPath: string } {
+  async parse(dsn: string): Promise<{ dbPath: string }> {
     // Basic validation
     if (!this.isValidDSN(dsn)) {
       throw new Error(`Invalid SQLite DSN: ${dsn}`);
@@ -85,7 +85,7 @@ export class SQLiteConnector implements Connector {
   private dbPath: string = ':memory:'; // Default to in-memory database
 
   async connect(dsn: string, initScript?: string): Promise<void> {
-    const config = this.dsnParser.parse(dsn);
+    const config = await this.dsnParser.parse(dsn);
     this.dbPath = config.dbPath;
 
     try {

src/connectors/sqlserver/index.ts

@@ -7,7 +7,7 @@ import {DefaultAzureCredential} from "@azure/identity";
  * Expected format: mssql://username:password@host:port/database
  */
 export class SQLServerDSNParser implements DSNParser {
-  parse(dsn: string): sql.config {
+  async parse(dsn: string): Promise<sql.config> {
     // Remove the protocol prefix
     if (!this.isValidDSN(dsn)) {
       throw new Error('Invalid SQL Server DSN format. Expected: sqlserver://username:password@host:port/database');
@@ -21,8 +21,6 @@ export class SQLServerDSNParser implements DSNParser {
     const user = url.username;
     const password = url.password ? decodeURIComponent(url.password) : '';
 
-    let authenticationType: string | undefined;
-
     // Parse additional options from query parameters
     const options: Record<string, any> = {};
     for (const [key, value] of url.searchParams.entries()) {
@@ -54,19 +52,26 @@ export class SQLServerDSNParser implements DSNParser {
       },
     };
 
-
-    const credential = new DefaultAzureCredential();
-    const accessToken = credential.getToken('https://database.windows.net/.default');
-
-
-    if (authenticationType) {
-      const credential = new DefaultAzureCredential();
-      config.authentication = {
-        type: authenticationType,
-        options: {
-          token: credential.getToken('https://database.windows.net/.default'),
-        },
-      };
+    // Handle Azure Active Directory authentication with access token
+    if (options.authentication === "azure-active-directory-access-token") {
+      try {        
+        // Create a credential instance
+        const credential = new DefaultAzureCredential();
+        
+        // Get token for SQL Server resource
+        const token = await credential.getToken("https://database.windows.net/");
+        
+        // Set the token in the config
+        config.authentication = {
+          type: "azure-active-directory-access-token",
+          options: {
+            token: token.token
+          }
+        };
+      } catch (error: unknown) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        throw new Error(`Failed to get Azure AD token: ${errorMessage}`);
+      }
     }
 
     return config;
@@ -99,7 +104,7 @@ export class SQLServerConnector implements Connector {
 
   async connect(dsn: string): Promise<void> {
     try {
-      this.config = this.dsnParser.parse(dsn);
+      this.config = await this.dsnParser.parse(dsn);
 
       if (!this.config.options) {
         this.config.options = {};