db-mcp 1.1.1 → 2.0.0

package/dist/{sqlite-26V3Y4MK.js → sqlite-U5KSYQXK.js}

@@ -1,9 +1,10 @@
-import { getCoreTools, getJsonHelperTools, getJsonOperationTools, getTextTools, getStatsTools, getVirtualTools, getVectorTools, getGeoTools, getAdminTools, getIntrospectionTools, getMigrationTools, getCodeModeTools, isDDL, getResourceDefinitions, getPromptDefinitions, applyCommonPragmas, autoEnableWal, detectAndSetJsonbSupport, SchemaManager, normalizeSqliteParams } from './chunk-TVIZ3XJH.js';
-export { getCoreTools } from './chunk-TVIZ3XJH.js';
-import { DatabaseAdapter } from './chunk-Z2GFQU3G.js';
-import { getToolGroupIcon, VERSION } from './chunk-FW7UCRLN.js';
-import { ConnectionError, ConfigurationError, QueryError } from './chunk-4IA3DB5C.js';
-import { createModuleLogger, ERROR_CODES, DbMcpError } from './chunk-AOUL5SHS.js';
+import { getCoreTools, getJsonHelperTools, getJsonOperationTools, getTextTools, getStatsTools, getVirtualTools, getVectorTools, getGeoTools, getAdminTools, getIntrospectionTools, getMigrationTools, getCodeModeTools, executeRead, executeWrite, isDDL, executeGeneral, getResourceDefinitions, getPromptDefinitions, applyCommonPragmas, autoEnableWal, detectAndSetJsonbSupport, SchemaManager } from './chunk-5Y42NPBP.js';
+export { getCoreTools } from './chunk-5Y42NPBP.js';
+import { DatabaseAdapter } from './chunk-Z7C2TM4L.js';
+import { getToolGroupIcon, VERSION } from './chunk-VIDSICEL.js';
+import './chunk-OKOVZ5QE.js';
+import { ConnectionError, ConfigurationError } from './chunk-X3MUUOWM.js';
+import { createModuleLogger, ERROR_CODES, DbMcpError } from './chunk-645ZEFLA.js';
 import initSqlJs from 'sql.js';
 
 // src/adapters/sqlite/tools/index.ts
@@ -35,129 +36,7 @@ function getAllToolDefinitions(adapter) {
 function getToolsByGroup(adapter, group) {
   return getAllToolDefinitions(adapter).filter((tool) => tool.group === group);
 }
-
-// src/adapters/sqlite/query-executor.ts
 var log = createModuleLogger("SQLITE");
-function rowsFromSqlJsResult(result) {
-  const numColumns = result.columns.length;
-  const numRows = result.values.length;
-  const rows = new Array(numRows);
-  const cols = result.columns;
-  for (let i = 0; i < numRows; i++) {
-    const rowObj = {};
-    const rowValues = result.values[i];
-    if (rowValues) {
-      for (let j = 0; j < numColumns; j++) {
-        rowObj[cols[j]] = rowValues[j];
-      }
-    }
-    rows[i] = rowObj;
-  }
-  return rows;
-}
-function executeRead(db, sql, params) {
-  const start = Date.now();
-  try {
-    const normalizedParams = normalizeSqliteParams(params);
-    const results = normalizedParams ? db.exec(sql, normalizedParams) : db.exec(sql);
-    if (results.length === 0) {
-      return Promise.resolve({
-        rows: [],
-        executionTimeMs: Date.now() - start
-      });
-    }
-    const firstResult = results[0];
-    if (!firstResult) {
-      return Promise.resolve({
-        rows: [],
-        executionTimeMs: Date.now() - start
-      });
-    }
-    const columns = firstResult.columns.map((name) => ({
-      name,
-      type: "unknown"
-    }));
-    const rows = rowsFromSqlJsResult(firstResult);
-    return Promise.resolve({
-      rows,
-      columns,
-      executionTimeMs: Date.now() - start
-    });
-  } catch (error) {
-    const message = error instanceof Error ? error.message : String(error);
-    log.error(`Query failed: ${message}`, {
-      code: ERROR_CODES.DB.QUERY_FAILED.full
-    });
-    throw new QueryError(
-      `Query execution failed: ${message}`,
-      "DB_QUERY_FAILED",
-      {
-        sql,
-        cause: error instanceof Error ? error : void 0
-      }
-    );
-  }
-}
-function executeWrite(db, sql, params) {
-  const start = Date.now();
-  try {
-    const normalizedParams = normalizeSqliteParams(params);
-    if (normalizedParams) {
-      db.run(sql, normalizedParams);
-    } else {
-      db.run(sql);
-    }
-    const changes = db.getRowsModified();
-    return Promise.resolve({
-      rowsAffected: changes,
-      executionTimeMs: Date.now() - start
-    });
-  } catch (error) {
-    const message = error instanceof Error ? error.message : String(error);
-    log.error(`Write query failed: ${message}`, {
-      code: ERROR_CODES.DB.QUERY_FAILED.full
-    });
-    throw new QueryError(`Write query failed: ${message}`, "DB_WRITE_FAILED", {
-      sql,
-      cause: error instanceof Error ? error : void 0
-    });
-  }
-}
-function executeGeneral(db, sql, params) {
-  const start = Date.now();
-  try {
-    const normalizedParams = normalizeSqliteParams(params);
-    const results = normalizedParams ? db.exec(sql, normalizedParams) : db.exec(sql);
-    if (results.length === 0) {
-      return Promise.resolve({
-        rowsAffected: db.getRowsModified(),
-        executionTimeMs: Date.now() - start
-      });
-    }
-    const firstResult = results[0];
-    if (!firstResult) {
-      return Promise.resolve({
-        rowsAffected: db.getRowsModified(),
-        executionTimeMs: Date.now() - start
-      });
-    }
-    const rows = rowsFromSqlJsResult(firstResult);
-    return Promise.resolve({
-      rows,
-      executionTimeMs: Date.now() - start
-    });
-  } catch (error) {
-    const message = error instanceof Error ? error.message : String(error);
-    log.error(`Query failed: ${message}`, {
-      code: ERROR_CODES.DB.QUERY_FAILED.full
-    });
-    throw new QueryError(`Query failed: ${message}`, "DB_QUERY_FAILED", {
-      sql,
-      cause: error instanceof Error ? error : void 0
-    });
-  }
-}
-var log2 = createModuleLogger("SQLITE");
 async function connectSqliteDatabase(adapter, config) {
   if (config.type !== "sqlite") {
     throw new ConfigurationError(
@@ -172,7 +51,7 @@ async function connectSqliteDatabase(adapter, config) {
     let db;
     if (filePath === ":memory:") {
       db = new sqlJsInstance.Database();
-      log2.info("Connected to in-memory SQLite database", {
+      log.info("Connected to in-memory SQLite database", {
         code: "SQLITE_CONNECT"
       });
     } else {
@@ -181,18 +60,18 @@ async function connectSqliteDatabase(adapter, config) {
         if (fs.existsSync(filePath)) {
           const buffer = await fs.promises.readFile(filePath);
           db = new sqlJsInstance.Database(buffer);
-          log2.info(`Connected to SQLite database: ${filePath}`, {
+          log.info(`Connected to SQLite database: ${filePath}`, {
             code: "SQLITE_CONNECT"
           });
         } else {
           db = new sqlJsInstance.Database();
-          log2.info(`Created new SQLite database: ${filePath}`, {
+          log.info(`Created new SQLite database: ${filePath}`, {
             code: "SQLITE_CONNECT"
           });
         }
       } catch {
         db = new sqlJsInstance.Database();
-        log2.warning("File access unavailable, using in-memory database", {
+        log.warning("File access unavailable, using in-memory database", {
           code: "SQLITE_FALLBACK"
         });
       }
@@ -207,17 +86,28 @@ async function connectSqliteDatabase(adapter, config) {
       { runPragma: (pragma) => db.run(`PRAGMA ${pragma}`) },
       filePath,
       sqliteConfig.options,
-      log2
+      log
     );
     detectAndSetJsonbSupport(() => {
       const versionResult = db.exec("SELECT sqlite_version()");
       return versionResult[0]?.values[0]?.[0] ?? "0.0.0";
-    }, log2);
+    }, log);
+    if (sqliteConfig.initializationSql && Array.isArray(sqliteConfig.initializationSql)) {
+      for (const sql of sqliteConfig.initializationSql) {
+        db.run(sql);
+      }
+      log.info(
+        `Executed ${sqliteConfig.initializationSql.length} initialization SQL statements`,
+        {
+          code: "SQLITE_INIT_SQL"
+        }
+      );
+    }
     const schemaManager = new SchemaManager(adapter);
     return { db, sqlJsInstance, schemaManager };
   } catch (error) {
     const message = error instanceof Error ? error.message : String(error);
-    log2.error(`Failed to connect to SQLite: ${message}`, {
+    log.error(`Failed to connect to SQLite: ${message}`, {
       code: ERROR_CODES.DB.CONNECT_FAILED.full
     });
     throw new ConnectionError(
@@ -235,17 +125,17 @@ async function disconnectSqliteDatabase(db, config) {
       const fs = await import('fs');
       const data = db.export();
       await fs.promises.writeFile(config.filePath, Buffer.from(data));
-      log2.info(`Saved database to: ${config.filePath}`, {
+      log.info(`Saved database to: ${config.filePath}`, {
         code: "SQLITE_DISCONNECT"
       });
     } catch {
-      log2.warning("Could not save database to file", {
+      log.warning("Could not save database to file", {
         code: "SQLITE_SAVE_FAILED"
       });
     }
   }
   db.close();
-  log2.info("Disconnected from SQLite database", {
+  log.info("Disconnected from SQLite database", {
     code: "SQLITE_DISCONNECT"
   });
 }
@@ -541,6 +431,40 @@ var SqliteAdapter = class extends DatabaseAdapter {
   getToolDefinitions() {
     return getAllToolDefinitions(this);
   }
+  /**
+   * Get tools filtered by adapter capabilities
+   * Used by Code Mode to accurately reflect available runtime features
+   */
+  getAvailableToolDefinitions() {
+    const allTools = this.getToolDefinitions();
+    const capabilities = this.getCapabilities();
+    return allTools.filter((tool) => {
+      if (!capabilities.transactions && tool.group === "transactions") {
+        return false;
+      }
+      if (!capabilities.vector && tool.group === "vector") {
+        return false;
+      }
+      if (!capabilities.geospatial) {
+        const spatialiteTools = [
+          "sqlite_geo_load_spatialite",
+          "sqlite_geo_spatial_table",
+          "sqlite_geo_spatial_query",
+          "sqlite_geo_spatial_analysis",
+          "sqlite_geo_spatial_index",
+          "sqlite_geo_transform",
+          "sqlite_geo_import"
+        ];
+        if (spatialiteTools.includes(tool.name)) {
+          return false;
+        }
+      }
+      if (!capabilities.fullTextSearch && tool.name.startsWith("sqlite_fts_")) {
+        return false;
+      }
+      return true;
+    });
+  }
   /**
    * Get resource definitions
    */