@bytebase/dbhub 0.0.8 → 0.1.1

package/dist/resources/employee-sqlite/object.sql

@@ -0,0 +1,74 @@
+-- SQLite implementation of views and functions
+-- This is simplified compared to the MySQL version
+
+-- Drop views if they exist
+DROP VIEW IF EXISTS v_full_employee;
+DROP VIEW IF EXISTS v_full_department;
+DROP VIEW IF EXISTS emp_dept_current;
+
+-- Create helper view to get current department for employees
+CREATE VIEW emp_dept_current AS
+SELECT 
+    e.emp_no,
+    de.dept_no
+FROM 
+    employee e
+JOIN 
+    dept_emp de ON e.emp_no = de.emp_no
+JOIN (
+    SELECT 
+        emp_no, 
+        MAX(from_date) AS max_from_date
+    FROM 
+        dept_emp
+    GROUP BY 
+        emp_no
+) latest ON de.emp_no = latest.emp_no AND de.from_date = latest.max_from_date;
+
+-- View that shows employee with their current department name
+CREATE VIEW v_full_employee AS
+SELECT
+    e.emp_no,
+    e.first_name, 
+    e.last_name,
+    e.birth_date, 
+    e.gender,
+    e.hire_date,
+    d.dept_name AS department
+FROM
+    employee e
+LEFT JOIN
+    emp_dept_current edc ON e.emp_no = edc.emp_no
+LEFT JOIN
+    department d ON edc.dept_no = d.dept_no;
+
+-- View to get current managers for departments
+CREATE VIEW current_managers AS
+SELECT
+    d.dept_no,
+    d.dept_name,
+    e.first_name || ' ' || e.last_name AS manager
+FROM
+    department d
+LEFT JOIN
+    dept_manager dm ON d.dept_no = dm.dept_no
+JOIN (
+    SELECT
+        dept_no,
+        MAX(from_date) AS max_from_date
+    FROM
+        dept_manager
+    GROUP BY
+        dept_no
+) latest ON dm.dept_no = latest.dept_no AND dm.from_date = latest.max_from_date
+LEFT JOIN
+    employee e ON dm.emp_no = e.emp_no;
+
+-- Create a view showing departments with their managers
+CREATE VIEW v_full_department AS
+SELECT
+    dept_no,
+    dept_name,
+    manager
+FROM
+    current_managers;

package/dist/resources/employee-sqlite/show_elapsed.sql

@@ -0,0 +1,4 @@
+-- SQLite doesn't have information_schema like MySQL
+-- This is a simpler version that just shows when the script was run
+
+SELECT 'Database loaded at ' || datetime('now', 'localtime') AS completion_time;

package/dist/resources/employee-sqlite/test_employee_md5.sql

@@ -0,0 +1,119 @@
+--  Sample employee database 
+--  See changelog table for details
+--  Copyright (C) 2007,2008, MySQL AB
+--  
+--  Original data created by Fusheng Wang and Carlo Zaniolo
+--  http://www.cs.aau.dk/TimeCenter/software.htm
+--  http://www.cs.aau.dk/TimeCenter/Data/employeeTemporalDataSet.zip
+-- 
+--  Current schema by Giuseppe Maxia 
+--  Data conversion from XML to relational by Patrick Crews
+--  SQLite adaptation by Claude Code
+-- 
+-- This work is licensed under the 
+-- Creative Commons Attribution-Share Alike 3.0 Unported License. 
+-- To view a copy of this license, visit 
+-- http://creativecommons.org/licenses/by-sa/3.0/ or send a letter to 
+-- Creative Commons, 171 Second Street, Suite 300, San Francisco, 
+-- California, 94105, USA.
+-- 
+--  DISCLAIMER
+--  To the best of our knowledge, this data is fabricated, and
+--  it does not correspond to real people. 
+--  Any similarity to existing people is purely coincidental.
+-- 
+
+SELECT 'TESTING INSTALLATION' as 'INFO';
+
+DROP TABLE IF EXISTS expected_value;
+DROP TABLE IF EXISTS found_value;
+
+CREATE TABLE expected_value (
+    table_name TEXT NOT NULL PRIMARY KEY,
+    recs INTEGER NOT NULL,
+    crc_md5 TEXT NOT NULL
+);
+
+CREATE TABLE found_value (
+    table_name TEXT NOT NULL PRIMARY KEY,
+    recs INTEGER NOT NULL,
+    crc_md5 TEXT NOT NULL
+);
+
+INSERT INTO expected_value VALUES 
+('employee',    1000, '595460127fb609c2b110b1796083e242'),
+('department',     9, 'd1af5e170d2d1591d776d5638d71fc5f'),
+('dept_manager',  16, '8ff425d5ad6dc56975998d1893b8dca9'),
+('dept_emp',    1103, 'e302aa5b56a69b49e40eb0d60674addc'),
+('title',       1470, 'ba77dd331ce00f76c1643a7d73cdcee6'),
+('salary',      9488, '61f22cfece4d34f5bb94c9f05a3da3ef');
+
+SELECT table_name, recs AS expected_record, crc_md5 AS expected_crc FROM expected_value;
+
+DROP TABLE IF EXISTS tchecksum;
+CREATE TABLE tchecksum (chk TEXT);
+
+-- For SQLite, we need to use a different approach for MD5 calculation
+-- Insert employee checksums
+INSERT INTO found_value
+SELECT 'employee', COUNT(*), 
+       (SELECT hex(md5(group_concat(emp_no||birth_date||first_name||last_name||gender||hire_date, '#')))
+        FROM (SELECT * FROM employee ORDER BY emp_no))
+FROM employee;
+
+-- Insert department checksums
+INSERT INTO found_value
+SELECT 'department', COUNT(*),
+       (SELECT hex(md5(group_concat(dept_no||dept_name, '#')))
+        FROM (SELECT * FROM department ORDER BY dept_no))
+FROM department;
+
+-- Insert dept_manager checksums
+INSERT INTO found_value
+SELECT 'dept_manager', COUNT(*),
+       (SELECT hex(md5(group_concat(dept_no||emp_no||from_date||to_date, '#')))
+        FROM (SELECT * FROM dept_manager ORDER BY dept_no, emp_no))
+FROM dept_manager;
+
+-- Insert dept_emp checksums
+INSERT INTO found_value
+SELECT 'dept_emp', COUNT(*),
+       (SELECT hex(md5(group_concat(dept_no||emp_no||from_date||to_date, '#')))
+        FROM (SELECT * FROM dept_emp ORDER BY dept_no, emp_no))
+FROM dept_emp;
+
+-- Insert title checksums
+INSERT INTO found_value
+SELECT 'title', COUNT(*),
+       (SELECT hex(md5(group_concat(emp_no||title||from_date||IFNULL(to_date,''), '#')))
+        FROM (SELECT * FROM title ORDER BY emp_no, title, from_date))
+FROM title;
+
+-- Insert salary checksums
+INSERT INTO found_value
+SELECT 'salary', COUNT(*),
+       (SELECT hex(md5(group_concat(emp_no||amount||from_date||to_date, '#')))
+        FROM (SELECT * FROM salary ORDER BY emp_no, from_date, to_date))
+FROM salary;
+
+SELECT table_name, recs as 'found_records', crc_md5 as found_crc FROM found_value;
+
+-- Compare expected vs found
+SELECT  
+    e.table_name, 
+    CASE WHEN e.recs=f.recs THEN 'OK' ELSE 'not ok' END AS records_match, 
+    CASE WHEN e.crc_md5=f.crc_md5 THEN 'ok' ELSE 'not ok' END AS crc_match 
+FROM 
+    expected_value e 
+    JOIN found_value f USING (table_name);
+
+-- Check for failures
+SELECT 
+    'CRC' as summary, 
+    CASE WHEN (SELECT COUNT(*) FROM expected_value e JOIN found_value f USING(table_name) WHERE f.crc_md5 != e.crc_md5) = 0 
+         THEN 'OK' ELSE 'FAIL' END as 'result'
+UNION ALL
+SELECT 
+    'count', 
+    CASE WHEN (SELECT COUNT(*) FROM expected_value e JOIN found_value f USING(table_name) WHERE f.recs != e.recs) = 0 
+         THEN 'OK' ELSE 'FAIL' END;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bytebase/dbhub",
-  "version": "0.0.8",
+  "version": "0.1.1",
   "description": "Universal Database MCP Server",
   "main": "dist/index.js",
   "type": "module",
@@ -17,22 +17,23 @@
   "license": "MIT",
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.6.1",
-    "@types/sqlite3": "^5.1.0",
+    "better-sqlite3": "^11.9.0",
     "dotenv": "^16.4.7",
     "express": "^4.18.2",
     "mssql": "^11.0.1",
     "mysql2": "^3.13.0",
     "pg": "^8.13.3",
-    "sqlite3": "^5.1.7",
     "zod": "^3.24.2"
   },
   "devDependencies": {
+    "@types/better-sqlite3": "^7.6.12",
     "@types/express": "^4.17.21",
     "@types/mssql": "^9.1.7",
     "@types/node": "^22.13.10",
     "@types/pg": "^8.11.11",
     "cross-env": "^7.0.3",
     "ts-node": "^10.9.2",
+    "tsup": "^8.4.0",
     "tsx": "^4.19.3",
     "typescript": "^5.8.2"
   },
@@ -49,7 +50,7 @@
     "src/**/*"
   ],
   "scripts": {
-    "build": "tsc",
+    "build": "tsup",
     "start": "node dist/index.js",
     "dev": "NODE_ENV=development tsx src/index.ts",
     "crossdev": "cross-env NODE_ENV=development tsx src/index.ts"

package/dist/config/demo-loader.js

@@ -1,45 +0,0 @@
-/**
- * Demo data loader for SQLite in-memory database
- *
- * This module loads the sample employee database into the SQLite in-memory database
- * when the --demo flag is specified.
- */
-import fs from 'fs';
-import path from 'path';
-import { fileURLToPath } from 'url';
-// Create __dirname equivalent for ES modules
-const __filename = fileURLToPath(import.meta.url);
-const __dirname = path.dirname(__filename);
-// Path to sample data files
-const DEMO_DATA_DIR = path.join(__dirname, '..', '..', 'resources', 'employee-sqlite');
-/**
- * Load SQL file contents
- */
-export function loadSqlFile(fileName) {
-    const filePath = path.join(DEMO_DATA_DIR, fileName);
-    return fs.readFileSync(filePath, 'utf8');
-}
-/**
- * Get SQLite DSN for in-memory database
- */
-export function getInMemorySqliteDSN() {
-    return 'sqlite::memory:';
-}
-/**
- * Load SQL files sequentially
- */
-export function getSqliteInMemorySetupSql() {
-    // First, load the schema
-    let sql = loadSqlFile('employee.sql');
-    // Replace .read directives with the actual file contents
-    // This is necessary because in-memory SQLite can't use .read
-    const readRegex = /\.read\s+([a-zA-Z0-9_]+\.sql)/g;
-    let match;
-    while ((match = readRegex.exec(sql)) !== null) {
-        const includePath = match[1];
-        const includeContent = loadSqlFile(includePath);
-        // Replace the .read line with the file contents
-        sql = sql.replace(match[0], includeContent);
-    }
-    return sql;
-}

package/dist/config/env.js

@@ -1,146 +0,0 @@
-import dotenv from "dotenv";
-import path from 'path';
-import fs from 'fs';
-import { fileURLToPath } from 'url';
-// Create __dirname equivalent for ES modules
-const __filename = fileURLToPath(import.meta.url);
-const __dirname = path.dirname(__filename);
-// Parse command line arguments
-export function parseCommandLineArgs() {
-    // Check if any args start with '--' (the way tsx passes them)
-    const args = process.argv.slice(2);
-    const parsedManually = {};
-    for (let i = 0; i < args.length; i++) {
-        const arg = args[i];
-        if (arg.startsWith('--')) {
-            const [key, value] = arg.substring(2).split('=');
-            if (value) {
-                // Handle --key=value format
-                parsedManually[key] = value;
-            }
-            else if (i + 1 < args.length && !args[i + 1].startsWith('--')) {
-                // Handle --key value format
-                parsedManually[key] = args[i + 1];
-                i++; // Skip the next argument as it's the value
-            }
-            else {
-                // Handle --key format (boolean flag)
-                parsedManually[key] = 'true';
-            }
-        }
-    }
-    // Just use the manually parsed args - removed parseArgs dependency for Node.js <18.3.0 compatibility
-    return parsedManually;
-}
-/**
- * Load environment files from various locations
- * Returns the name of the file that was loaded, or null if none was found
- */
-export function loadEnvFiles() {
-    // Determine if we're in development or production mode
-    const isDevelopment = process.env.NODE_ENV === 'development' || process.argv[1]?.includes('tsx');
-    // Select environment file names based on environment
-    const envFileNames = isDevelopment
-        ? ['.env.local', '.env'] // In development, try .env.local first, then .env
-        : ['.env']; // In production, only look for .env
-    // Build paths to check for environment files
-    const envPaths = [];
-    for (const fileName of envFileNames) {
-        envPaths.push(fileName, // Current working directory
-        path.join(__dirname, '..', '..', fileName), // Two levels up (src/config -> src -> root)
-        path.join(process.cwd(), fileName) // Explicit current working directory
-        );
-    }
-    // Try to load the first env file found from the prioritized locations
-    for (const envPath of envPaths) {
-        console.error(`Checking for env file: ${envPath}`);
-        if (fs.existsSync(envPath)) {
-            dotenv.config({ path: envPath });
-            // Return the name of the file that was loaded
-            return path.basename(envPath);
-        }
-    }
-    return null;
-}
-/**
- * Check if demo mode is enabled from command line args
- * Returns true if --demo flag is provided
- */
-export function isDemoMode() {
-    const args = parseCommandLineArgs();
-    return args.demo === 'true';
-}
-/**
- * Resolve DSN from command line args, environment variables, or .env files
- * Returns the DSN and its source, or null if not found
- */
-export function resolveDSN() {
-    // Get command line arguments
-    const args = parseCommandLineArgs();
-    // Check for demo mode first (highest priority)
-    if (isDemoMode()) {
-        // Will use in-memory SQLite with demo data
-        return {
-            dsn: 'sqlite::memory:',
-            source: 'demo mode',
-            isDemo: true
-        };
-    }
-    // 1. Check command line arguments 
-    if (args.dsn) {
-        return { dsn: args.dsn, source: 'command line argument' };
-    }
-    // 2. Check environment variables before loading .env
-    if (process.env.DSN) {
-        return { dsn: process.env.DSN, source: 'environment variable' };
-    }
-    // 3. Try loading from .env files
-    const loadedEnvFile = loadEnvFiles();
-    if (loadedEnvFile && process.env.DSN) {
-        return { dsn: process.env.DSN, source: `${loadedEnvFile} file` };
-    }
-    return null;
-}
-/**
- * Resolve transport type from command line args or environment variables
- * Returns 'stdio' or 'sse', with 'stdio' as the default
- */
-export function resolveTransport() {
-    // Get command line arguments
-    const args = parseCommandLineArgs();
-    // 1. Check command line arguments first (highest priority)
-    if (args.transport) {
-        const type = args.transport === 'sse' ? 'sse' : 'stdio';
-        return { type, source: 'command line argument' };
-    }
-    // 2. Check environment variables
-    if (process.env.TRANSPORT) {
-        const type = process.env.TRANSPORT === 'sse' ? 'sse' : 'stdio';
-        return { type, source: 'environment variable' };
-    }
-    // 3. Default to stdio
-    return { type: 'stdio', source: 'default' };
-}
-/**
- * Resolve port from command line args or environment variables
- * Returns port number with 8080 as the default
- *
- * Note: The port option is only applicable when using --transport=sse
- * as it controls the HTTP server port for SSE connections.
- */
-export function resolvePort() {
-    // Get command line arguments
-    const args = parseCommandLineArgs();
-    // 1. Check command line arguments first (highest priority)
-    if (args.port) {
-        const port = parseInt(args.port, 10);
-        return { port, source: 'command line argument' };
-    }
-    // 2. Check environment variables
-    if (process.env.PORT) {
-        const port = parseInt(process.env.PORT, 10);
-        return { port, source: 'environment variable' };
-    }
-    // 3. Default to 8080
-    return { port: 8080, source: 'default' };
-}

package/dist/connectors/interface.js

@@ -1,55 +0,0 @@
-/**
- * Registry for available database connectors
- */
-export class ConnectorRegistry {
-    /**
-     * Register a new connector
-     */
-    static register(connector) {
-        ConnectorRegistry.connectors.set(connector.id, connector);
-    }
-    /**
-     * Get a connector by ID
-     */
-    static getConnector(id) {
-        return ConnectorRegistry.connectors.get(id) || null;
-    }
-    /**
-     * Get connector for a DSN string
-     * Tries to find a connector that can handle the given DSN format
-     */
-    static getConnectorForDSN(dsn) {
-        for (const connector of ConnectorRegistry.connectors.values()) {
-            if (connector.dsnParser.isValidDSN(dsn)) {
-                return connector;
-            }
-        }
-        return null;
-    }
-    /**
-     * Get all available connector IDs
-     */
-    static getAvailableConnectors() {
-        return Array.from(ConnectorRegistry.connectors.keys());
-    }
-    /**
-     * Get sample DSN for a specific connector
-     */
-    static getSampleDSN(connectorId) {
-        const connector = ConnectorRegistry.getConnector(connectorId);
-        if (!connector)
-            return null;
-        return connector.dsnParser.getSampleDSN();
-    }
-    /**
-     * Get all available sample DSNs
-     */
-    static getAllSampleDSNs() {
-        const samples = {};
-        for (const [id, connector] of ConnectorRegistry.connectors.entries()) {
-            samples[id] = connector.dsnParser.getSampleDSN();
-        }
-        return samples;
-    }
-}
-ConnectorRegistry.connectors = new Map();

package/dist/connectors/manager.js

@@ -1,91 +0,0 @@
-import { ConnectorRegistry } from './interface.js';
-// Singleton instance for global access
-let managerInstance = null;
-/**
- * Manages database connectors and provides a unified interface to work with them
- */
-export class ConnectorManager {
-    constructor() {
-        this.activeConnector = null;
-        this.connected = false;
-        if (!managerInstance) {
-            managerInstance = this;
-        }
-    }
-    /**
-     * Initialize and connect to the database using a DSN
-     */
-    async connectWithDSN(dsn, initScript) {
-        // First try to find a connector that can handle this DSN
-        let connector = ConnectorRegistry.getConnectorForDSN(dsn);
-        if (!connector) {
-            throw new Error(`No connector found that can handle the DSN: ${dsn}`);
-        }
-        this.activeConnector = connector;
-        // Connect to the database
-        await this.activeConnector.connect(dsn, initScript);
-        this.connected = true;
-    }
-    /**
-     * Initialize and connect to the database using a specific connector type
-     */
-    async connectWithType(connectorType, dsn) {
-        // Get the connector from the registry
-        const connector = ConnectorRegistry.getConnector(connectorType);
-        if (!connector) {
-            throw new Error(`Connector "${connectorType}" not found`);
-        }
-        this.activeConnector = connector;
-        // Use provided DSN or get sample DSN
-        const connectionString = dsn || connector.dsnParser.getSampleDSN();
-        // Connect to the database
-        await this.activeConnector.connect(connectionString);
-        this.connected = true;
-    }
-    /**
-     * Close the database connection
-     */
-    async disconnect() {
-        if (this.activeConnector && this.connected) {
-            await this.activeConnector.disconnect();
-            this.connected = false;
-        }
-    }
-    /**
-     * Get the active connector
-     */
-    getConnector() {
-        if (!this.activeConnector) {
-            throw new Error('No active connector. Call connectWithDSN() or connectWithType() first.');
-        }
-        return this.activeConnector;
-    }
-    /**
-     * Check if there's an active connection
-     */
-    isConnected() {
-        return this.connected;
-    }
-    /**
-     * Get all available connector types
-     */
-    static getAvailableConnectors() {
-        return ConnectorRegistry.getAvailableConnectors();
-    }
-    /**
-     * Get sample DSNs for all available connectors
-     */
-    static getAllSampleDSNs() {
-        return ConnectorRegistry.getAllSampleDSNs();
-    }
-    /**
-     * Get the current active connector instance
-     * This is used by resource and tool handlers
-     */
-    static getCurrentConnector() {
-        if (!managerInstance) {
-            throw new Error('ConnectorManager not initialized');
-        }
-        return managerInstance.getConnector();
-    }
-}