@hesed/mysql 0.2.1 → 0.3.0

package/dist/mysql/mysql-utils.js

@@ -1,44 +1,25 @@
-import { encode } from '@toon-format/toon';
 import mysql from 'mysql2/promise';
 import { getMySQLConnectionOptions } from './config-loader.js';
+import { FORMATTERS } from './formatters.js';
 import { analyzeQuery, applyDefaultLimit, checkBlacklist, getQueryType, requiresConfirmation } from './query-validator.js';
-/**
- * MySQL Database Utility
- * Provides core database operations with safety validation and formatting
- */
 export class MySQLUtil {
     config;
-    connectionPool;
+    connections;
     constructor(config) {
         this.config = config;
-        this.connectionPool = new Map();
+        this.connections = new Map();
     }
-    /**
-     * Close all connections
-     */
     async closeAll() {
-        await Promise.all([...this.connectionPool.values()].map((conn) => conn.end()));
-        this.connectionPool.clear();
+        const entries = [...this.connections.values()];
+        this.connections.clear();
+        await Promise.allSettled(entries.map(async (connPromise) => (await connPromise).end()));
     }
-    /**
-     * Describe table structure
-     */
     async describeTable(profileName, table, format = 'table') {
         try {
             const connection = await this.getConnection(profileName);
             const [rows, fields] = await connection.query(`DESCRIBE ${table}`);
-            let result = '';
-            if (format === 'json') {
-                result += this.formatAsJson(rows);
-            }
-            else if (format === 'toon') {
-                result += this.formatAsToon(rows);
-            }
-            else {
-                result += this.formatAsTable(rows, fields);
-            }
             return {
-                result,
+                result: this.formatRows(rows, fields, format),
                 structure: rows,
                 success: true,
             };
@@ -51,9 +32,6 @@ export class MySQLUtil {
             };
         }
     }
-    /**
-     * Validate and execute a SQL query
-     */
     async executeQuery(profileName, query, format = 'table', skipConfirmation = false) {
         const blacklistCheck = checkBlacklist(query, this.config.safety.blacklistedOperations);
         if (!blacklistCheck.allowed) {
@@ -118,26 +96,13 @@ export class MySQLUtil {
             };
         }
     }
-    /**
-     * Explain query execution plan
-     */
     async explainQuery(profileName, query, format = 'table') {
         try {
             const connection = await this.getConnection(profileName);
             const [rows, fields] = await connection.query(`EXPLAIN ${query}`);
-            let result = '';
-            if (format === 'json') {
-                result += this.formatAsJson(rows);
-            }
-            else if (format === 'toon') {
-                result += this.formatAsToon(rows);
-            }
-            else {
-                result += this.formatAsTable(rows, fields);
-            }
             return {
                 plan: rows,
-                result,
+                result: this.formatRows(rows, fields, format),
                 success: true,
             };
         }
@@ -149,90 +114,6 @@ export class MySQLUtil {
             };
         }
     }
-    /**
-     * Format query results as CSV
-     */
-    formatAsCsv(rows, fields) {
-        if (!rows || rows.length === 0) {
-            return '';
-        }
-        const columnNames = fields.map((f) => f.name);
-        let csv = columnNames.join(',') + '\n';
-        for (const row of rows) {
-            const values = columnNames.map((name) => {
-                const value = row[name] ?? '';
-                const str = String(value);
-                if (str.includes(',') || str.includes('"') || str.includes('\n')) {
-                    return '"' + str.replaceAll('"', '""') + '"';
-                }
-                return str;
-            });
-            csv += values.join(',') + '\n';
-        }
-        return csv;
-    }
-    /**
-     * Format query results as JSON
-     */
-    formatAsJson(rows) {
-        return JSON.stringify(rows, null, 2);
-    }
-    /**
-     * Format query results as table
-     */
-    formatAsTable(rows, fields) {
-        if (!rows || rows.length === 0) {
-            return 'No results';
-        }
-        const columnNames = fields.map((f) => f.name);
-        const columnWidths = columnNames.map((name) => {
-            const dataWidth = Math.max(...rows.map((row) => String(row[name] ?? '').length));
-            return Math.max(name.length, dataWidth, 3);
-        });
-        let table = '┌' + columnWidths.map((w) => '─'.repeat(w + 2)).join('┬') + '┐\n';
-        table += '│ ' + columnNames.map((name, i) => name.padEnd(columnWidths[i])).join(' │ ') + ' │\n';
-        table += '├' + columnWidths.map((w) => '─'.repeat(w + 2)).join('┼') + '┤\n';
-        for (const row of rows) {
-            table +=
-                '│ ' +
-                    columnNames
-                        .map((name, i) => {
-                        const value = row[name] ?? 'NULL';
-                        return String(value).padEnd(columnWidths[i]);
-                    })
-                        .join(' │ ') +
-                    ' │\n';
-        }
-        table += '└' + columnWidths.map((w) => '─'.repeat(w + 2)).join('┴') + '┘';
-        return table;
-    }
-    /**
-     * Format query results as TOON
-     */
-    formatAsToon(rows) {
-        if (!rows || rows.length === 0) {
-            return '';
-        }
-        const serializedRows = rows.map((row) => {
-            const serialized = {};
-            for (const [key, value] of Object.entries(row)) {
-                if (value instanceof Date) {
-                    serialized[key] = Number.isNaN(value.getTime()) ? null : value.toISOString();
-                }
-                else if (Buffer.isBuffer(value)) {
-                    serialized[key] = value.toString('base64');
-                }
-                else {
-                    serialized[key] = value;
-                }
-            }
-            return serialized;
-        });
-        return encode(serializedRows);
-    }
-    /**
-     * List all databases
-     */
     async listDatabases(profileName) {
         try {
             const connection = await this.getConnection(profileName);
@@ -252,9 +133,6 @@ export class MySQLUtil {
             };
         }
     }
-    /**
-     * List all tables in current database
-     */
     async listTables(profileName) {
         try {
             const connection = await this.getConnection(profileName);
@@ -276,26 +154,13 @@ export class MySQLUtil {
             };
         }
     }
-    /**
-     * Show table indexes
-     */
     async showIndexes(profileName, table, format = 'table') {
         try {
             const connection = await this.getConnection(profileName);
             const [rows, fields] = await connection.query(`SHOW INDEXES FROM ${table}`);
-            let result = '';
-            if (format === 'json') {
-                result += this.formatAsJson(rows);
-            }
-            else if (format === 'toon') {
-                result += this.formatAsToon(rows);
-            }
-            else {
-                result += this.formatAsTable(rows, fields);
-            }
             return {
                 indexes: rows,
-                result,
+                result: this.formatRows(rows, fields, format),
                 success: true,
             };
         }
@@ -307,9 +172,6 @@ export class MySQLUtil {
             };
         }
     }
-    /**
-     * Test database connection
-     */
     async testConnection(profileName) {
         try {
             const connection = await this.getConnection(profileName);
@@ -330,41 +192,33 @@ export class MySQLUtil {
             };
         }
     }
-    /**
-     * Format rows for SELECT/SHOW/DESCRIBE/EXPLAIN query result
-     */
+    formatRows(rows, fields, format) {
+        return FORMATTERS[format](rows, fields);
+    }
     formatSelectResult(rows, fields, format) {
         const rowCount = Array.isArray(rows) ? rows.length : 0;
-        let result = `Query executed successfully. Rows returned: ${rowCount}\n\n`;
-        switch (format) {
-            case 'csv': {
-                result += this.formatAsCsv(rows, fields);
-                break;
-            }
-            case 'json': {
-                result += this.formatAsJson(rows);
-                break;
-            }
-            case 'toon': {
-                result += this.formatAsToon(rows);
-                break;
+        return `Query executed successfully. Rows returned: ${rowCount}\n\n` + this.formatRows(rows, fields, format);
+    }
+    async getConnection(profileName) {
+        const existing = this.connections.get(profileName);
+        if (existing) {
+            try {
+                const conn = await existing;
+                await conn.ping();
+                return conn;
             }
-            default: {
-                result += this.formatAsTable(rows, fields);
+            catch {
+                this.connections.delete(profileName);
             }
         }
-        return result;
-    }
-    /**
-     * Get or create MySQL connection for a profile
-     */
-    async getConnection(profileName) {
-        if (this.connectionPool.has(profileName)) {
-            return this.connectionPool.get(profileName);
+        const connPromise = mysql.createConnection(getMySQLConnectionOptions(this.config, profileName));
+        this.connections.set(profileName, connPromise);
+        try {
+            return await connPromise;
+        }
+        catch (error) {
+            this.connections.delete(profileName);
+            throw error;
         }
-        const options = getMySQLConnectionOptions(this.config, profileName);
-        const connection = await mysql.createConnection(options);
-        this.connectionPool.set(profileName, connection);
-        return connection;
     }
 }

package/dist/mysql/query-validator.d.ts

@@ -1,7 +1,3 @@
-/**
- * Query Validation and Safety Module
- * Provides SQL query analysis and safety checks
- */
 interface BlacklistCheckResult {
     allowed: boolean;
     reason?: string;
@@ -15,24 +11,9 @@ interface QueryWarning {
     message: string;
     suggestion: string;
 }
-/**
- * Check if query contains blacklisted operations
- */
 export declare function checkBlacklist(query: string, blacklistedOperations: string[]): BlacklistCheckResult;
-/**
- * Check if query requires user confirmation
- */
 export declare function requiresConfirmation(query: string, confirmationOperations: string[]): ConfirmationCheckResult;
-/**
- * Get query type (SELECT, INSERT, UPDATE, etc.)
- */
 export declare function getQueryType(query: string): string;
-/**
- * Analyze query for potential issues and provide warnings
- */
 export declare function analyzeQuery(query: string): QueryWarning[];
-/**
- * Apply default LIMIT to SELECT queries if not present
- */
 export declare function applyDefaultLimit(query: string, defaultLimit: number): string;
 export {};

package/dist/mysql/query-validator.js

@@ -1,10 +1,3 @@
-/**
- * Query Validation and Safety Module
- * Provides SQL query analysis and safety checks
- */
-/**
- * Check if query contains blacklisted operations
- */
 export function checkBlacklist(query, blacklistedOperations) {
     const normalizedQuery = query.trim().toUpperCase();
     for (const operation of blacklistedOperations) {
@@ -18,9 +11,6 @@ export function checkBlacklist(query, blacklistedOperations) {
     }
     return { allowed: true };
 }
-/**
- * Check if query requires user confirmation
- */
 export function requiresConfirmation(query, confirmationOperations) {
     const normalizedQuery = query.trim().toUpperCase();
     for (const operation of confirmationOperations) {
@@ -34,9 +24,6 @@ export function requiresConfirmation(query, confirmationOperations) {
     }
     return { required: false };
 }
-/**
- * Get query type (SELECT, INSERT, UPDATE, etc.)
- */
 export function getQueryType(query) {
     const normalizedQuery = query.trim().toUpperCase();
     const firstWord = normalizedQuery.split(/\s+/)[0];
@@ -58,9 +45,6 @@ export function getQueryType(query) {
     }
     return 'UNKNOWN';
 }
-/**
- * Analyze query for potential issues and provide warnings
- */
 export function analyzeQuery(query) {
     const warnings = [];
     const normalizedQuery = query.trim().toUpperCase();
@@ -91,9 +75,6 @@ export function analyzeQuery(query) {
     }
     return warnings;
 }
-/**
- * Apply default LIMIT to SELECT queries if not present
- */
 export function applyDefaultLimit(query, defaultLimit) {
     const normalizedQuery = query.trim().toUpperCase();
     if (normalizedQuery.startsWith('SELECT') && !normalizedQuery.includes('LIMIT')) {