@bytebase/dbhub 0.2.2 → 0.2.3

package/README.md

@@ -36,14 +36,14 @@ https://demo.dbhub.ai/sse connects a [sample employee database](https://github.c
 
 ### Database Resources
 
-| Resource Name                 | URI Format                                                  | PostgreSQL | MySQL | SQL Server | SQLite |
-| ----------------------------- | ----------------------------------------------------------- | :--------: | :---: | :--------: | :----: |
-| schemas                       | `db://schemas`                                              |     ✅     |  ✅   |     ✅     |   ✅   |
-| tables_in_schema              | `db://schemas/{schemaName}/tables`                          |     ✅     |  ✅   |     ✅     |   ✅   |
-| table_structure_in_schema     | `db://schemas/{schemaName}/tables/{tableName}`              |     ✅     |  ✅   |     ✅     |   ✅   |
-| indexes_in_table              | `db://schemas/{schemaName}/tables/{tableName}/indexes`      |     ✅     |  ✅   |     ✅     |   ✅   |
-| procedures_in_schema          | `db://schemas/{schemaName}/procedures`                      |     ✅     |  ✅   |     ✅     |   ❌   |
-| procedure_details_in_schema   | `db://schemas/{schemaName}/procedures/{procedureName}`      |     ✅     |  ✅   |     ✅     |   ❌   |
+| Resource Name               | URI Format                                             | PostgreSQL | MySQL | SQL Server | SQLite |
+| --------------------------- | ------------------------------------------------------ | :--------: | :---: | :--------: | :----: |
+| schemas                     | `db://schemas`                                         |     ✅     |  ✅   |     ✅     |   ✅   |
+| tables_in_schema            | `db://schemas/{schemaName}/tables`                     |     ✅     |  ✅   |     ✅     |   ✅   |
+| table_structure_in_schema   | `db://schemas/{schemaName}/tables/{tableName}`         |     ✅     |  ✅   |     ✅     |   ✅   |
+| indexes_in_table            | `db://schemas/{schemaName}/tables/{tableName}/indexes` |     ✅     |  ✅   |     ✅     |   ✅   |
+| procedures_in_schema        | `db://schemas/{schemaName}/procedures`                 |     ✅     |  ✅   |     ✅     |   ❌   |
+| procedure_details_in_schema | `db://schemas/{schemaName}/procedures/{procedureName}` |     ✅     |  ✅   |     ✅     |   ❌   |
 
 ### Database Tools
 
@@ -54,10 +54,10 @@ https://demo.dbhub.ai/sse connects a [sample employee database](https://github.c
 
 ### Prompt Capabilities
 
-| Prompt              | Command Name    | PostgreSQL | MySQL | SQL Server | SQLite |
-| ------------------- | --------------- | :--------: | :---: | :--------: | :----: |
-| Generate SQL        | `generate_sql`  |     ✅     |  ✅   |     ✅     |   ✅   |
-| Explain DB Elements | `explain_db`    |     ✅     |  ✅   |     ✅     |   ✅   |
+| Prompt              | Command Name   | PostgreSQL | MySQL | SQL Server | SQLite |
+| ------------------- | -------------- | :--------: | :---: | :--------: | :----: |
+| Generate SQL        | `generate_sql` |     ✅     |  ✅   |     ✅     |   ✅   |
+| Explain DB Elements | `explain_db`   |     ✅     |  ✅   |     ✅     |   ✅   |
 
 ## Installation
 
@@ -181,14 +181,17 @@ For real databases, a Database Source Name (DSN) is required. You can provide th
   DSN=postgres://user:password@localhost:5432/dbname?sslmode=disable
   ```
 
+> [!WARNING]
+> When running in Docker, use `host.docker.internal` instead of `localhost` to connect to databases running on your host machine. For example: `mysql://user:password@host.docker.internal:3306/dbname`
+
 DBHub supports the following database connection string formats:
 
-| Database   | DSN Format                                               | Example                                                           |
-| ---------- | -------------------------------------------------------- | ----------------------------------------------------------------- |
-| PostgreSQL | `postgres://[user]:[password]@[host]:[port]/[database]`  | `postgres://user:password@localhost:5432/dbname?sslmode=disable`  |
-| 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`                     |
+| Database   | DSN Format                                               | Example                                                          |
+| ---------- | -------------------------------------------------------- | ---------------------------------------------------------------- |
+| PostgreSQL | `postgres://[user]:[password]@[host]:[port]/[database]`  | `postgres://user:password@localhost:5432/dbname?sslmode=disable` |
+| 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`                    |
 
 ### Transport
 

package/dist/index.js

@@ -974,11 +974,11 @@ var MySQLConnector = class {
     }
     try {
       const [rows] = await this.pool.query(`
-        SELECT schema_name 
-        FROM information_schema.schemata
-        ORDER BY schema_name
+        SELECT SCHEMA_NAME 
+        FROM INFORMATION_SCHEMA.SCHEMATA
+        ORDER BY SCHEMA_NAME
       `);
-      return rows.map((row) => row.schema_name);
+      return rows.map((row) => row.SCHEMA_NAME);
     } catch (error) {
       console.error("Error getting schemas:", error);
       throw error;
@@ -989,15 +989,15 @@ var MySQLConnector = class {
       throw new Error("Not connected to database");
     }
     try {
-      const schemaClause = schema ? "WHERE table_schema = ?" : "WHERE table_schema = DATABASE()";
+      const schemaClause = schema ? "WHERE TABLE_SCHEMA = ?" : "WHERE TABLE_SCHEMA = DATABASE()";
       const queryParams = schema ? [schema] : [];
       const [rows] = await this.pool.query(`
-        SELECT table_name 
-        FROM information_schema.tables 
+        SELECT TABLE_NAME 
+        FROM INFORMATION_SCHEMA.TABLES 
         ${schemaClause}
-        ORDER BY table_name
+        ORDER BY TABLE_NAME
       `, queryParams);
-      return rows.map((row) => row.table_name);
+      return rows.map((row) => row.TABLE_NAME);
     } catch (error) {
       console.error("Error getting tables:", error);
       throw error;
@@ -1008,15 +1008,15 @@ var MySQLConnector = class {
       throw new Error("Not connected to database");
     }
     try {
-      const schemaClause = schema ? "WHERE table_schema = ?" : "WHERE table_schema = DATABASE()";
+      const schemaClause = schema ? "WHERE TABLE_SCHEMA = ?" : "WHERE TABLE_SCHEMA = DATABASE()";
       const queryParams = schema ? [schema, tableName] : [tableName];
       const [rows] = await this.pool.query(`
-        SELECT COUNT(*) as count
-        FROM information_schema.tables 
+        SELECT COUNT(*) AS COUNT
+        FROM INFORMATION_SCHEMA.TABLES 
         ${schemaClause} 
-        AND table_name = ?
+        AND TABLE_NAME = ?
       `, queryParams);
-      return rows[0].count > 0;
+      return rows[0].COUNT > 0;
     } catch (error) {
       console.error("Error checking if table exists:", error);
       throw error;
@@ -1036,7 +1036,7 @@ var MySQLConnector = class {
           NON_UNIQUE,
           SEQ_IN_INDEX
         FROM 
-          information_schema.STATISTICS 
+          INFORMATION_SCHEMA.STATISTICS 
         WHERE 
           ${schemaClause}
           AND TABLE_NAME = ? 
@@ -1080,18 +1080,18 @@ var MySQLConnector = class {
       throw new Error("Not connected to database");
     }
     try {
-      const schemaClause = schema ? "WHERE table_schema = ?" : "WHERE table_schema = DATABASE()";
+      const schemaClause = schema ? "WHERE TABLE_SCHEMA = ?" : "WHERE TABLE_SCHEMA = DATABASE()";
       const queryParams = schema ? [schema, tableName] : [tableName];
       const [rows] = await this.pool.query(`
         SELECT 
-          column_name, 
-          data_type, 
-          is_nullable,
-          column_default
-        FROM information_schema.columns
+          COLUMN_NAME as column_name, 
+          DATA_TYPE as data_type, 
+          IS_NULLABLE as is_nullable,
+          COLUMN_DEFAULT as column_default
+        FROM INFORMATION_SCHEMA.COLUMNS
         ${schemaClause}
-        AND table_name = ?
-        ORDER BY ordinal_position
+        AND TABLE_NAME = ?
+        ORDER BY ORDINAL_POSITION
       `, queryParams);
       return rows;
     } catch (error) {
@@ -1104,15 +1104,15 @@ var MySQLConnector = class {
       throw new Error("Not connected to database");
     }
     try {
-      const schemaClause = schema ? "WHERE routine_schema = ?" : "WHERE routine_schema = DATABASE()";
+      const schemaClause = schema ? "WHERE ROUTINE_SCHEMA = ?" : "WHERE ROUTINE_SCHEMA = DATABASE()";
       const queryParams = schema ? [schema] : [];
       const [rows] = await this.pool.query(`
-        SELECT routine_name
-        FROM information_schema.routines
+        SELECT ROUTINE_NAME
+        FROM INFORMATION_SCHEMA.ROUTINES
         ${schemaClause}
-        ORDER BY routine_name
+        ORDER BY ROUTINE_NAME
       `, queryParams);
-      return rows.map((row) => row.routine_name);
+      return rows.map((row) => row.ROUTINE_NAME);
     } catch (error) {
       console.error("Error getting stored procedures:", error);
       throw error;
@@ -1123,39 +1123,39 @@ var MySQLConnector = class {
       throw new Error("Not connected to database");
     }
     try {
-      const schemaClause = schema ? "WHERE r.routine_schema = ?" : "WHERE r.routine_schema = DATABASE()";
+      const schemaClause = schema ? "WHERE r.ROUTINE_SCHEMA = ?" : "WHERE r.ROUTINE_SCHEMA = DATABASE()";
       const queryParams = schema ? [schema, procedureName] : [procedureName];
       const [rows] = await this.pool.query(`
         SELECT 
-          r.routine_name AS procedure_name,
+          r.ROUTINE_NAME AS procedure_name,
           CASE 
-            WHEN r.routine_type = 'PROCEDURE' THEN 'procedure'
+            WHEN r.ROUTINE_TYPE = 'PROCEDURE' THEN 'procedure'
             ELSE 'function'
           END AS procedure_type,
-          LOWER(r.routine_type) AS routine_type,
-          r.routine_definition,
-          r.dtd_identifier AS return_type,
+          LOWER(r.ROUTINE_TYPE) AS routine_type,
+          r.ROUTINE_DEFINITION,
+          r.DTD_IDENTIFIER AS return_type,
           (
             SELECT GROUP_CONCAT(
-              CONCAT(p.parameter_name, ' ', p.parameter_mode, ' ', p.data_type)
-              ORDER BY p.ordinal_position
+              CONCAT(p.PARAMETER_NAME, ' ', p.PARAMETER_MODE, ' ', p.DATA_TYPE)
+              ORDER BY p.ORDINAL_POSITION
               SEPARATOR ', '
             )
-            FROM information_schema.parameters p
-            WHERE p.specific_schema = r.routine_schema
-            AND p.specific_name = r.routine_name
-            AND p.parameter_name IS NOT NULL
+            FROM INFORMATION_SCHEMA.PARAMETERS p
+            WHERE p.SPECIFIC_SCHEMA = r.ROUTINE_SCHEMA
+            AND p.SPECIFIC_NAME = r.ROUTINE_NAME
+            AND p.PARAMETER_NAME IS NOT NULL
           ) AS parameter_list
-        FROM information_schema.routines r
+        FROM INFORMATION_SCHEMA.ROUTINES r
         ${schemaClause}
-        AND r.routine_name = ?
+        AND r.ROUTINE_NAME = ?
       `, queryParams);
       if (rows.length === 0) {
         const schemaName = schema || "current schema";
         throw new Error(`Stored procedure '${procedureName}' not found in ${schemaName}`);
       }
       const procedure = rows[0];
-      let definition = procedure.routine_definition;
+      let definition = procedure.ROUTINE_DEFINITION;
       try {
         const schemaValue = schema || await this.getCurrentSchema();
         if (procedure.procedure_type === "procedure") {
@@ -1183,15 +1183,15 @@ var MySQLConnector = class {
         }
         if (!definition) {
           const [bodyRows] = await this.pool.query(`
-            SELECT routine_definition, routine_body 
-            FROM information_schema.routines
-            WHERE routine_schema = ? AND routine_name = ?
+            SELECT ROUTINE_DEFINITION, ROUTINE_BODY 
+            FROM INFORMATION_SCHEMA.ROUTINES
+            WHERE ROUTINE_SCHEMA = ? AND ROUTINE_NAME = ?
           `, [schemaValue, procedureName]);
           if (bodyRows && bodyRows.length > 0) {
-            if (bodyRows[0].routine_definition) {
-              definition = bodyRows[0].routine_definition;
-            } else if (bodyRows[0].routine_body) {
-              definition = bodyRows[0].routine_body;
+            if (bodyRows[0].ROUTINE_DEFINITION) {
+              definition = bodyRows[0].ROUTINE_DEFINITION;
+            } else if (bodyRows[0].ROUTINE_BODY) {
+              definition = bodyRows[0].ROUTINE_BODY;
             }
           }
         }
@@ -1214,8 +1214,8 @@ var MySQLConnector = class {
   }
   // Helper method to get current schema (database) name
   async getCurrentSchema() {
-    const [rows] = await this.pool.query("SELECT DATABASE() as db");
-    return rows[0].db;
+    const [rows] = await this.pool.query("SELECT DATABASE() AS DB");
+    return rows[0].DB;
   }
   async executeQuery(query) {
     if (!this.pool) {
@@ -1595,7 +1595,7 @@ async function tablesResourceHandler(uri, variables, _extra) {
 }
 
 // src/resources/schema.ts
-async function schemaResourceHandler(uri, variables, _extra) {
+async function tableStructureResourceHandler(uri, variables, _extra) {
   const connector = ConnectorManager.getCurrentConnector();
   const tableName = Array.isArray(variables.tableName) ? variables.tableName[0] : variables.tableName;
   const schemaName = variables.schemaName ? Array.isArray(variables.schemaName) ? variables.schemaName[0] : variables.schemaName : void 0;
@@ -1796,7 +1796,7 @@ function registerResources(server) {
   server.resource(
     "table_structure_in_schema",
     new ResourceTemplate("db://schemas/{schemaName}/tables/{tableName}", { list: void 0 }),
-    schemaResourceHandler
+    tableStructureResourceHandler
   );
   server.resource(
     "indexes_in_table",
@@ -1846,7 +1846,6 @@ async function runQueryToolHandler({ query }, _extra) {
 
 // src/tools/list-connectors.ts
 async function listConnectorsToolHandler(_args, _extra) {
-  const connectors = ConnectorManager.getAvailableConnectors();
   const samples = ConnectorRegistry.getAllSampleDSNs();
   let activeConnectorId = null;
   try {
@@ -2169,6 +2168,10 @@ ${possibleTableMatches.join("\n")}`
       "EXPLANATION_ERROR"
     );
   }
+  return formatPromptErrorResponse(
+    `Unable to process request for schema: ${schema}, table: ${table}`,
+    "UNHANDLED_REQUEST"
+  );
 }
 function determineTablePurpose(tableName, columns) {
   const lowerTableName = tableName.toLowerCase();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bytebase/dbhub",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "description": "Universal Database MCP Server",
   "main": "dist/index.js",
   "type": "module",