@bytebase/dbhub 0.9.0 → 0.11.0

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.9.0",
+  "version": "0.11.0",
   "description": "Universal Database MCP Server",
   "main": "dist/index.js",
   "type": "module",
@@ -18,12 +18,14 @@
   "dependencies": {
     "@azure/identity": "^4.8.0",
     "@modelcontextprotocol/sdk": "^1.12.1",
+    "better-sqlite3": "^11.9.0",
     "dotenv": "^16.4.7",
     "express": "^4.18.2",
     "mariadb": "^3.4.0",
     "mssql": "^11.0.1",
     "mysql2": "^3.13.0",
     "pg": "^8.13.3",
+    "ssh2": "^1.16.0",
     "zod": "^3.24.2"
   },
   "devDependencies": {
@@ -31,10 +33,12 @@
     "@testcontainers/mssqlserver": "^11.0.3",
     "@testcontainers/mysql": "^11.0.3",
     "@testcontainers/postgresql": "^11.0.3",
+    "@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",
+    "@types/ssh2": "^1.15.5",
     "cross-env": "^7.0.3",
     "husky": "^9.0.11",
     "lint-staged": "^15.2.2",
@@ -59,7 +63,7 @@
     "src/**/*"
   ],
   "lint-staged": {
-    "*.{js,ts}": "pnpm run test:pre-commit"
+    "*.{js,ts}": "vitest related --run"
   },
   "scripts": {
     "build": "tsup",
@@ -69,7 +73,6 @@
     "test": "vitest run",
     "test:watch": "vitest",
     "test:integration": "vitest run --testNamePattern='Integration Tests'",
-    "test:pre-commit": "vitest related --run --exclude '**/sqlserver.integration.test.ts'",
     "pre-commit": "lint-staged"
   }
 }