datapeek 0.1.6 → 0.1.8

package/dist/cli/chunk-5X2YZYYM.js

@@ -0,0 +1,137 @@
+#!/usr/bin/env node
+
+
+// src/server/db/mssql.ts
+import sql from "mssql";
+var pool = null;
+function parseConnectionString(connectionString) {
+  const config = {
+    server: "",
+    database: "",
+    options: {
+      encrypt: true,
+      trustServerCertificate: false,
+      enableArithAbort: true
+    }
+  };
+  const parts = connectionString.split(";");
+  for (const part of parts) {
+    const [key, ...valueParts] = part.split("=");
+    const value = valueParts.join("=").trim();
+    const keyLower = key.trim().toLowerCase();
+    switch (keyLower) {
+      case "server":
+      case "data source":
+        let serverValue = value;
+        if (serverValue.startsWith("tcp:")) {
+          serverValue = serverValue.substring(4);
+        }
+        const [serverHost, serverPort] = serverValue.split(",");
+        config.server = serverHost.trim();
+        if (serverPort) {
+          config.port = parseInt(serverPort.trim(), 10);
+        }
+        break;
+      case "database":
+      case "initial catalog":
+        config.database = value;
+        break;
+      case "user id":
+      case "userid":
+      case "uid":
+        config.user = value;
+        break;
+      case "password":
+      case "pwd":
+        config.password = value;
+        break;
+      case "port":
+        config.port = parseInt(value, 10);
+        break;
+      case "encrypt":
+        config.options.encrypt = value.toLowerCase() === "true";
+        break;
+      case "trustservercertificate":
+      case "trust server certificate":
+        config.options.trustServerCertificate = value.toLowerCase() === "true";
+        break;
+    }
+  }
+  return config;
+}
+async function testConnection(config) {
+  const testPool = typeof config === "string" ? new sql.ConnectionPool(parseConnectionString(config)) : new sql.ConnectionPool(config);
+  try {
+    await testPool.connect();
+    await testPool.close();
+  } catch (error) {
+    throw error;
+  }
+}
+async function connect(config) {
+  await disconnect();
+  const connectionConfig = typeof config === "string" ? parseConnectionString(config) : config;
+  pool = new sql.ConnectionPool(connectionConfig);
+  try {
+    await pool.connect();
+  } catch (error) {
+    pool = null;
+    throw error;
+  }
+}
+async function disconnect() {
+  if (pool) {
+    try {
+      await pool.close();
+    } catch (error) {
+    }
+    pool = null;
+  }
+}
+function getConnection() {
+  return pool;
+}
+async function executeQuery(query, parameters) {
+  if (!pool || !pool.connected) {
+    throw new Error("Not connected to database");
+  }
+  const request = pool.request();
+  if (parameters) {
+    for (const param of parameters) {
+      if (param.type) {
+        request.input(param.name, param.type, param.value);
+      } else {
+        request.input(param.name, param.value);
+      }
+    }
+  }
+  const result = await request.query(query);
+  return result.recordset || [];
+}
+async function executeQueryMultiple(query, parameters) {
+  if (!pool || !pool.connected) {
+    throw new Error("Not connected to database");
+  }
+  const request = pool.request();
+  if (parameters) {
+    for (const param of parameters) {
+      if (param.type) {
+        request.input(param.name, param.type, param.value);
+      } else {
+        request.input(param.name, param.value);
+      }
+    }
+  }
+  const result = await request.query(query);
+  return result.recordsets || [];
+}
+
+export {
+  parseConnectionString,
+  testConnection,
+  connect,
+  disconnect,
+  getConnection,
+  executeQuery,
+  executeQueryMultiple
+};

package/dist/cli/dev.js

@@ -4,9 +4,10 @@ import {
   connect,
   disconnect,
   executeQuery,
+  executeQueryMultiple,
   getConnection,
   testConnection
-} from "./chunk-PTI2CUG6.js";
+} from "./chunk-5X2YZYYM.js";
 
 // src/server/index.ts
 import express from "express";
@@ -87,15 +88,20 @@ connectionRoutes.delete("/", async (req, res) => {
 });
 connectionRoutes.get("/status", async (req, res) => {
   try {
-    const { getConnection: getConnection2, executeQuery: executeQuery2 } = await import("./mssql-5VFXLOGV.js");
+    const { getConnection: getConnection2, executeQuery: executeQuery3 } = await import("./mssql-D7NDIUG7.js");
     const pool = getConnection2();
     if (pool && pool.connected) {
       try {
-        const result = await executeQuery2("SELECT DB_NAME() as databaseName");
+        const result = await executeQuery3("SELECT DB_NAME() as databaseName");
         const databaseName = result[0]?.databaseName || null;
         res.json({ connected: true, databaseName });
-      } catch {
-        res.json({ connected: true, databaseName: null });
+      } catch (error) {
+        const errorMessage = error.message || "";
+        if (errorMessage.includes("Login failed") || errorMessage.includes("authentication")) {
+          const { disconnect: disconnect2 } = await import("./mssql-D7NDIUG7.js");
+          await disconnect2();
+        }
+        res.json({ connected: false });
       }
     } else {
       res.json({ connected: false });
@@ -126,6 +132,11 @@ tableRoutes.get("/", async (req, res) => {
     const result = await executeQuery(query);
     res.json(result);
   } catch (error) {
+    const errorMessage = error.message || "";
+    if (errorMessage.includes("Login failed") || errorMessage.includes("authentication")) {
+      const { disconnect: disconnect2 } = await import("./mssql-D7NDIUG7.js");
+      await disconnect2();
+    }
     res.status(500).json({ error: error.message || "Failed to fetch tables" });
   }
 });
@@ -164,6 +175,11 @@ tableRoutes.get("/:schema/:table", async (req, res) => {
     ]);
     res.json(result);
   } catch (error) {
+    const errorMessage = error.message || "";
+    if (errorMessage.includes("Login failed") || errorMessage.includes("authentication")) {
+      const { disconnect: disconnect2 } = await import("./mssql-D7NDIUG7.js");
+      await disconnect2();
+    }
     res.status(500).json({ error: error.message || "Failed to fetch table structure" });
   }
 });
@@ -175,12 +191,63 @@ tableRoutes.get("/:schema/:table/data", async (req, res) => {
     const sortColumn = req.query.sortColumn;
     const sortDirection = req.query.sortDirection || "asc";
     const offset = (page - 1) * pageSize;
+    const filters = {};
+    Object.keys(req.query).forEach((key) => {
+      const match = key.match(/^filter\[(.+)\]$/);
+      if (match && req.query[key] && String(req.query[key]).trim()) {
+        filters[match[1]] = String(req.query[key]).trim();
+      }
+    });
+    console.log("Received filters from query:", filters);
+    console.log("All query params:", req.query);
     const pool = getConnection();
     if (!pool || !pool.connected) {
       return res.status(400).json({ error: "Not connected to database" });
     }
-    const countQuery = `SELECT COUNT(*) as total FROM [${schema}].[${table}]`;
-    const countResult = await executeQuery(countQuery);
+    const filterColumns = [];
+    if (Object.keys(filters).length > 0) {
+      try {
+        const columnNames = Object.keys(filters);
+        const placeholders = columnNames.map((_, i) => `@col${i}`).join(", ");
+        const validateQuery = `
+          SELECT COLUMN_NAME 
+          FROM INFORMATION_SCHEMA.COLUMNS 
+          WHERE TABLE_SCHEMA = @schema AND TABLE_NAME = @table AND COLUMN_NAME IN (${placeholders})
+        `;
+        const validateParams = [
+          { name: "schema", value: schema, type: sql.NVarChar },
+          { name: "table", value: table, type: sql.NVarChar },
+          ...columnNames.map((col, i) => ({ name: `col${i}`, value: col, type: sql.NVarChar }))
+        ];
+        const validateResult = await executeQuery(validateQuery, validateParams);
+        filterColumns.push(...validateResult.map((r) => r.COLUMN_NAME));
+        console.log("Validated filter columns:", filterColumns, "from filters:", filters);
+      } catch (e) {
+        console.error("Error validating filter columns:", e);
+        console.error("Filters that failed validation:", filters);
+      }
+    }
+    let whereClause = "";
+    const filterParams = [];
+    if (filterColumns.length > 0) {
+      const validFilters = filterColumns.filter((col) => {
+        const filterValue = filters[col];
+        return filterValue && filterValue.trim() !== "";
+      });
+      if (validFilters.length > 0) {
+        const whereConditions = validFilters.map((col, index) => {
+          const filterValue = filters[col];
+          filterParams.push({ name: `filter${index}`, value: `%${filterValue.trim()}%`, type: sql.NVarChar });
+          return `[${col}] LIKE @filter${index}`;
+        });
+        whereClause = `WHERE ${whereConditions.join(" AND ")}`;
+        console.log("Applying WHERE clause:", whereClause);
+        console.log("Filter params:", filterParams);
+        console.log("Valid filters:", validFilters);
+      }
+    }
+    const countQuery = `SELECT COUNT(*) as total FROM [${schema}].[${table}]${whereClause ? " " + whereClause : ""}`;
+    const countResult = await executeQuery(countQuery, filterParams.length > 0 ? filterParams : []);
     const total = countResult[0]?.total || 0;
     let orderByColumn = sortColumn || "";
     let orderByDirection = sortDirection?.toUpperCase() === "DESC" ? "DESC" : "ASC";
@@ -224,24 +291,26 @@ tableRoutes.get("/:schema/:table/data", async (req, res) => {
     let data;
     let generatedQuery = "";
     if (orderByColumn) {
-      generatedQuery = `SELECT * FROM [${schema}].[${table}]
+      generatedQuery = `SELECT * FROM [${schema}].[${table}]${whereClause ? "\n" + whereClause : ""}
 ORDER BY [${orderByColumn}] ${orderByDirection}
 OFFSET ${offset} ROWS
 FETCH NEXT ${pageSize} ROWS ONLY`;
       const dataQuery = `
         SELECT * FROM [${schema}].[${table}]
+        ${whereClause}
         ORDER BY [${orderByColumn}] ${orderByDirection}
         OFFSET @offset ROWS
         FETCH NEXT @pageSize ROWS ONLY
       `;
       data = await executeQuery(dataQuery, [
         { name: "offset", value: offset, type: sql.Int },
-        { name: "pageSize", value: pageSize, type: sql.Int }
+        { name: "pageSize", value: pageSize, type: sql.Int },
+        ...filterParams
       ]);
     } else {
       generatedQuery = `SELECT * FROM (
   SELECT *, ROW_NUMBER() OVER (ORDER BY (SELECT NULL)) as rn
-  FROM [${schema}].[${table}]
+  FROM [${schema}].[${table}]${whereClause ? "\n  " + whereClause : ""}
 ) t
 WHERE rn > ${offset} AND rn <= ${offset + pageSize}
 ORDER BY rn`;
@@ -249,13 +318,15 @@ ORDER BY rn`;
         SELECT * FROM (
           SELECT *, ROW_NUMBER() OVER (ORDER BY (SELECT NULL)) as rn
           FROM [${schema}].[${table}]
+          ${whereClause}
         ) t
         WHERE rn > @offset AND rn <= @offset + @pageSize
         ORDER BY rn
       `;
       data = await executeQuery(dataQuery, [
         { name: "offset", value: offset, type: sql.Int },
-        { name: "pageSize", value: pageSize, type: sql.Int }
+        { name: "pageSize", value: pageSize, type: sql.Int },
+        ...filterParams
       ]);
       data = data.map((row) => {
         const { rn, ...rest } = row;
@@ -274,7 +345,11 @@ ORDER BY rn`;
     });
   } catch (error) {
     console.error("Error fetching table data:", error);
-    const errorMessage = error.message || "Failed to fetch table data";
+    const errorMessage = error.message || "";
+    if (errorMessage.includes("Login failed") || errorMessage.includes("authentication")) {
+      const { disconnect: disconnect2 } = await import("./mssql-D7NDIUG7.js");
+      await disconnect2();
+    }
     const errorDetails = error.originalError?.message || error.originalError?.info?.message || "";
     res.status(500).json({
       error: errorMessage,
@@ -299,17 +374,29 @@ queryRoutes.post("/", async (req, res) => {
       });
     }
     const dangerousKeywords = ["DROP", "DELETE", "INSERT", "UPDATE", "ALTER", "CREATE", "TRUNCATE", "EXEC", "EXECUTE"];
-    const hasDangerousKeyword = dangerousKeywords.some(
-      (keyword) => trimmedQuery.includes(keyword)
-    );
+    const hasDangerousKeyword = dangerousKeywords.some((keyword) => {
+      const regex = new RegExp(`\\b${keyword}\\b`, "i");
+      return regex.test(sqlQuery);
+    });
     if (hasDangerousKeyword) {
       return res.status(400).json({
         error: "Query contains prohibited keywords. Only SELECT queries are allowed."
       });
     }
-    const result = await executeQuery(sqlQuery);
-    res.json({ data: result });
+    const startTime = Date.now();
+    const resultSets = await executeQueryMultiple(sqlQuery);
+    const executionTime = Date.now() - startTime;
+    res.json({
+      data: resultSets[0] || [],
+      resultSets: resultSets.length > 0 ? resultSets : [],
+      executionTime
+    });
   } catch (error) {
+    const errorMessage = error.message || "";
+    if (errorMessage.includes("Login failed") || errorMessage.includes("authentication")) {
+      const { disconnect: disconnect2 } = await import("./mssql-D7NDIUG7.js");
+      await disconnect2();
+    }
     res.status(500).json({
       error: error.message || "Query execution failed",
       details: error.originalError?.message

package/dist/cli/index.js

@@ -4,9 +4,10 @@ import {
   connect,
   disconnect,
   executeQuery,
+  executeQueryMultiple,
   getConnection,
   testConnection
-} from "./chunk-PTI2CUG6.js";
+} from "./chunk-5X2YZYYM.js";
 
 // src/cli/index.ts
 import { Command } from "commander";
@@ -90,15 +91,20 @@ connectionRoutes.delete("/", async (req, res) => {
 });
 connectionRoutes.get("/status", async (req, res) => {
   try {
-    const { getConnection: getConnection2, executeQuery: executeQuery2 } = await import("./mssql-5VFXLOGV.js");
+    const { getConnection: getConnection2, executeQuery: executeQuery3 } = await import("./mssql-D7NDIUG7.js");
     const pool = getConnection2();
     if (pool && pool.connected) {
       try {
-        const result = await executeQuery2("SELECT DB_NAME() as databaseName");
+        const result = await executeQuery3("SELECT DB_NAME() as databaseName");
         const databaseName = result[0]?.databaseName || null;
         res.json({ connected: true, databaseName });
-      } catch {
-        res.json({ connected: true, databaseName: null });
+      } catch (error) {
+        const errorMessage = error.message || "";
+        if (errorMessage.includes("Login failed") || errorMessage.includes("authentication")) {
+          const { disconnect: disconnect2 } = await import("./mssql-D7NDIUG7.js");
+          await disconnect2();
+        }
+        res.json({ connected: false });
       }
     } else {
       res.json({ connected: false });
@@ -129,6 +135,11 @@ tableRoutes.get("/", async (req, res) => {
     const result = await executeQuery(query);
     res.json(result);
   } catch (error) {
+    const errorMessage = error.message || "";
+    if (errorMessage.includes("Login failed") || errorMessage.includes("authentication")) {
+      const { disconnect: disconnect2 } = await import("./mssql-D7NDIUG7.js");
+      await disconnect2();
+    }
     res.status(500).json({ error: error.message || "Failed to fetch tables" });
   }
 });
@@ -167,6 +178,11 @@ tableRoutes.get("/:schema/:table", async (req, res) => {
     ]);
     res.json(result);
   } catch (error) {
+    const errorMessage = error.message || "";
+    if (errorMessage.includes("Login failed") || errorMessage.includes("authentication")) {
+      const { disconnect: disconnect2 } = await import("./mssql-D7NDIUG7.js");
+      await disconnect2();
+    }
     res.status(500).json({ error: error.message || "Failed to fetch table structure" });
   }
 });
@@ -178,12 +194,63 @@ tableRoutes.get("/:schema/:table/data", async (req, res) => {
     const sortColumn = req.query.sortColumn;
     const sortDirection = req.query.sortDirection || "asc";
     const offset = (page - 1) * pageSize;
+    const filters = {};
+    Object.keys(req.query).forEach((key) => {
+      const match = key.match(/^filter\[(.+)\]$/);
+      if (match && req.query[key] && String(req.query[key]).trim()) {
+        filters[match[1]] = String(req.query[key]).trim();
+      }
+    });
+    console.log("Received filters from query:", filters);
+    console.log("All query params:", req.query);
     const pool = getConnection();
     if (!pool || !pool.connected) {
       return res.status(400).json({ error: "Not connected to database" });
     }
-    const countQuery = `SELECT COUNT(*) as total FROM [${schema}].[${table}]`;
-    const countResult = await executeQuery(countQuery);
+    const filterColumns = [];
+    if (Object.keys(filters).length > 0) {
+      try {
+        const columnNames = Object.keys(filters);
+        const placeholders = columnNames.map((_, i) => `@col${i}`).join(", ");
+        const validateQuery = `
+          SELECT COLUMN_NAME 
+          FROM INFORMATION_SCHEMA.COLUMNS 
+          WHERE TABLE_SCHEMA = @schema AND TABLE_NAME = @table AND COLUMN_NAME IN (${placeholders})
+        `;
+        const validateParams = [
+          { name: "schema", value: schema, type: sql.NVarChar },
+          { name: "table", value: table, type: sql.NVarChar },
+          ...columnNames.map((col, i) => ({ name: `col${i}`, value: col, type: sql.NVarChar }))
+        ];
+        const validateResult = await executeQuery(validateQuery, validateParams);
+        filterColumns.push(...validateResult.map((r) => r.COLUMN_NAME));
+        console.log("Validated filter columns:", filterColumns, "from filters:", filters);
+      } catch (e) {
+        console.error("Error validating filter columns:", e);
+        console.error("Filters that failed validation:", filters);
+      }
+    }
+    let whereClause = "";
+    const filterParams = [];
+    if (filterColumns.length > 0) {
+      const validFilters = filterColumns.filter((col) => {
+        const filterValue = filters[col];
+        return filterValue && filterValue.trim() !== "";
+      });
+      if (validFilters.length > 0) {
+        const whereConditions = validFilters.map((col, index) => {
+          const filterValue = filters[col];
+          filterParams.push({ name: `filter${index}`, value: `%${filterValue.trim()}%`, type: sql.NVarChar });
+          return `[${col}] LIKE @filter${index}`;
+        });
+        whereClause = `WHERE ${whereConditions.join(" AND ")}`;
+        console.log("Applying WHERE clause:", whereClause);
+        console.log("Filter params:", filterParams);
+        console.log("Valid filters:", validFilters);
+      }
+    }
+    const countQuery = `SELECT COUNT(*) as total FROM [${schema}].[${table}]${whereClause ? " " + whereClause : ""}`;
+    const countResult = await executeQuery(countQuery, filterParams.length > 0 ? filterParams : []);
     const total = countResult[0]?.total || 0;
     let orderByColumn = sortColumn || "";
     let orderByDirection = sortDirection?.toUpperCase() === "DESC" ? "DESC" : "ASC";
@@ -227,24 +294,26 @@ tableRoutes.get("/:schema/:table/data", async (req, res) => {
     let data;
     let generatedQuery = "";
     if (orderByColumn) {
-      generatedQuery = `SELECT * FROM [${schema}].[${table}]
+      generatedQuery = `SELECT * FROM [${schema}].[${table}]${whereClause ? "\n" + whereClause : ""}
 ORDER BY [${orderByColumn}] ${orderByDirection}
 OFFSET ${offset} ROWS
 FETCH NEXT ${pageSize} ROWS ONLY`;
       const dataQuery = `
         SELECT * FROM [${schema}].[${table}]
+        ${whereClause}
         ORDER BY [${orderByColumn}] ${orderByDirection}
         OFFSET @offset ROWS
         FETCH NEXT @pageSize ROWS ONLY
       `;
       data = await executeQuery(dataQuery, [
         { name: "offset", value: offset, type: sql.Int },
-        { name: "pageSize", value: pageSize, type: sql.Int }
+        { name: "pageSize", value: pageSize, type: sql.Int },
+        ...filterParams
       ]);
     } else {
       generatedQuery = `SELECT * FROM (
   SELECT *, ROW_NUMBER() OVER (ORDER BY (SELECT NULL)) as rn
-  FROM [${schema}].[${table}]
+  FROM [${schema}].[${table}]${whereClause ? "\n  " + whereClause : ""}
 ) t
 WHERE rn > ${offset} AND rn <= ${offset + pageSize}
 ORDER BY rn`;
@@ -252,13 +321,15 @@ ORDER BY rn`;
         SELECT * FROM (
           SELECT *, ROW_NUMBER() OVER (ORDER BY (SELECT NULL)) as rn
           FROM [${schema}].[${table}]
+          ${whereClause}
         ) t
         WHERE rn > @offset AND rn <= @offset + @pageSize
         ORDER BY rn
       `;
       data = await executeQuery(dataQuery, [
         { name: "offset", value: offset, type: sql.Int },
-        { name: "pageSize", value: pageSize, type: sql.Int }
+        { name: "pageSize", value: pageSize, type: sql.Int },
+        ...filterParams
       ]);
       data = data.map((row) => {
         const { rn, ...rest } = row;
@@ -277,7 +348,11 @@ ORDER BY rn`;
     });
   } catch (error) {
     console.error("Error fetching table data:", error);
-    const errorMessage = error.message || "Failed to fetch table data";
+    const errorMessage = error.message || "";
+    if (errorMessage.includes("Login failed") || errorMessage.includes("authentication")) {
+      const { disconnect: disconnect2 } = await import("./mssql-D7NDIUG7.js");
+      await disconnect2();
+    }
     const errorDetails = error.originalError?.message || error.originalError?.info?.message || "";
     res.status(500).json({
       error: errorMessage,
@@ -302,17 +377,29 @@ queryRoutes.post("/", async (req, res) => {
       });
     }
     const dangerousKeywords = ["DROP", "DELETE", "INSERT", "UPDATE", "ALTER", "CREATE", "TRUNCATE", "EXEC", "EXECUTE"];
-    const hasDangerousKeyword = dangerousKeywords.some(
-      (keyword) => trimmedQuery.includes(keyword)
-    );
+    const hasDangerousKeyword = dangerousKeywords.some((keyword) => {
+      const regex = new RegExp(`\\b${keyword}\\b`, "i");
+      return regex.test(sqlQuery);
+    });
     if (hasDangerousKeyword) {
       return res.status(400).json({
         error: "Query contains prohibited keywords. Only SELECT queries are allowed."
       });
     }
-    const result = await executeQuery(sqlQuery);
-    res.json({ data: result });
+    const startTime = Date.now();
+    const resultSets = await executeQueryMultiple(sqlQuery);
+    const executionTime = Date.now() - startTime;
+    res.json({
+      data: resultSets[0] || [],
+      resultSets: resultSets.length > 0 ? resultSets : [],
+      executionTime
+    });
   } catch (error) {
+    const errorMessage = error.message || "";
+    if (errorMessage.includes("Login failed") || errorMessage.includes("authentication")) {
+      const { disconnect: disconnect2 } = await import("./mssql-D7NDIUG7.js");
+      await disconnect2();
+    }
     res.status(500).json({
       error: error.message || "Query execution failed",
       details: error.originalError?.message

package/dist/cli/mssql-D7NDIUG7.js

@@ -0,0 +1,20 @@
+#!/usr/bin/env node
+
+import {
+  connect,
+  disconnect,
+  executeQuery,
+  executeQueryMultiple,
+  getConnection,
+  parseConnectionString,
+  testConnection
+} from "./chunk-5X2YZYYM.js";
+export {
+  connect,
+  disconnect,
+  executeQuery,
+  executeQueryMultiple,
+  getConnection,
+  parseConnectionString,
+  testConnection
+};