forge-sql-orm 2.1.14 → 2.1.16

package/src/lib/drizzle/extensions/additionalActions.ts

@@ -536,6 +536,96 @@ async function handleNonCachedQuery(
   }
 }
 
+/**
+ * Creates a select query builder with field aliasing and optional caching support.
+ *
+ * @param db - The database instance
+ * @param fields - The fields to select with aliases
+ * @param selectFn - Function to create the base select query
+ * @param useCache - Whether to enable caching for this query
+ * @param options - ForgeSQL ORM options
+ * @param cacheTtl - Optional cache TTL override
+ * @returns Select query builder with aliasing and optional caching
+ */
+/**
+ * Creates a catch handler for Promise-like objects
+ */
+function createCatchHandler(receiver: any): (onrejected: any) => Promise<any> {
+  return (onrejected: any) => receiver.then(undefined, onrejected);
+}
+
+/**
+ * Creates a finally handler for Promise-like objects
+ */
+function createFinallyHandler(receiver: any): (onfinally: any) => Promise<any> {
+  return (onfinally: any) => {
+    const handleFinally = (value: any) => Promise.resolve(value).finally(onfinally);
+    const handleReject = (reason: any) => Promise.reject(reason).finally(onfinally);
+    return receiver.then(handleFinally, handleReject);
+  };
+}
+
+/**
+ * Creates a then handler for cached queries
+ */
+function createCachedThenHandler(
+  target: any,
+  options: ForgeSqlOrmOptions,
+  cacheTtl: number | undefined,
+  selections: any,
+  aliasMap: any,
+): (onfulfilled?: any, onrejected?: any) => Promise<any> {
+  return (onfulfilled?: any, onrejected?: any) => {
+    const ttl = cacheTtl ?? options.cacheTTL ?? 120;
+    return handleCachedQuery(target, options, ttl, selections, aliasMap, onfulfilled, onrejected);
+  };
+}
+
+/**
+ * Creates a then handler for non-cached queries
+ */
+function createNonCachedThenHandler(
+  target: any,
+  options: ForgeSqlOrmOptions,
+  selections: any,
+  aliasMap: any,
+): (onfulfilled?: any, onrejected?: any) => Promise<any> {
+  return (onfulfilled?: any, onrejected?: any) => {
+    return handleNonCachedQuery(target, options, selections, aliasMap, onfulfilled, onrejected);
+  };
+}
+
+/**
+ * Creates an execute handler that transforms results
+ */
+function createExecuteHandler(
+  target: any,
+  selections: any,
+  aliasMap: any,
+): (...args: any[]) => Promise<any> {
+  return async (...args: any[]) => {
+    const rows = await target.execute(...args);
+    return applyFromDriverTransform(rows, selections, aliasMap);
+  };
+}
+
+/**
+ * Creates a function call handler that wraps results
+ */
+function createFunctionCallHandler(
+  value: Function,
+  target: any,
+  wrapBuilder: (rawBuilder: any) => any,
+): (...args: any[]) => any {
+  return (...args: any[]) => {
+    const result = value.apply(target, args);
+    if (typeof result === "object" && result !== null && "execute" in result) {
+      return wrapBuilder(result);
+    }
+    return result;
+  };
+}
+
 /**
  * Creates a select query builder with field aliasing and optional caching support.
  *
@@ -562,61 +652,27 @@ function createAliasedSelectBuilder<TSelection extends SelectedFields>(
     return new Proxy(rawBuilder, {
       get(target, prop, receiver) {
         if (prop === "execute") {
-          return async (...args: any[]) => {
-            const rows = await target.execute(...args);
-            return applyFromDriverTransform(rows, selections, aliasMap);
-          };
+          return createExecuteHandler(target, selections, aliasMap);
         }
 
         if (prop === "then") {
-          return (onfulfilled?: any, onrejected?: any) => {
-            if (useCache) {
-              const ttl = cacheTtl ?? options.cacheTTL ?? 120;
-              return handleCachedQuery(
-                target,
-                options,
-                ttl,
-                selections,
-                aliasMap,
-                onfulfilled,
-                onrejected,
-              );
-            } else {
-              return handleNonCachedQuery(
-                target,
-                options,
-                selections,
-                aliasMap,
-                onfulfilled,
-                onrejected,
-              );
-            }
-          };
+          return useCache
+            ? createCachedThenHandler(target, options, cacheTtl, selections, aliasMap)
+            : createNonCachedThenHandler(target, options, selections, aliasMap);
         }
+
         if (prop === "catch") {
-          return (onrejected: any) => (receiver as any).then(undefined, onrejected);
+          return createCatchHandler(receiver);
         }
 
         if (prop === "finally") {
-          return (onfinally: any) =>
-            (receiver as any).then(
-              (value: any) => Promise.resolve(value).finally(onfinally),
-              (reason: any) => Promise.reject(reason).finally(onfinally),
-            );
+          return createFinallyHandler(receiver);
         }
 
         const value = Reflect.get(target, prop, receiver);
 
         if (typeof value === "function") {
-          return (...args: any[]) => {
-            const result = value.apply(target, args);
-
-            if (typeof result === "object" && result !== null && "execute" in result) {
-              return wrapBuilder(result);
-            }
-
-            return result;
-          };
+          return createFunctionCallHandler(value, target, wrapBuilder);
         }
 
         return value;

package/src/utils/cacheTableUtils.ts

@@ -0,0 +1,511 @@
+import { Parser } from "node-sql-parser";
+import { ForgeSqlOrmOptions } from "../core/ForgeSQLQueryBuilder";
+
+/**
+ * Extracts table name from object value.
+ */
+function extractTableNameFromObject(value: any, context?: string): string | null {
+  // If it's an array, skip it (not a table name)
+  if (Array.isArray(value)) {
+    return null;
+  }
+  // Handle backticks_quote_string type only for node.table context
+  if (
+    context?.includes("node.table") &&
+    value.type === "backticks_quote_string" &&
+    typeof value.value === "string"
+  ) {
+    return value.value === "dual" ? null : value.value.toLowerCase();
+  }
+  // Try value.name first (most common)
+  if (typeof value.name === "string") {
+    return value.name === "dual" ? null : value.name.toLowerCase();
+  }
+  // Try value.table if it's a nested structure
+  if (value.table) {
+    return normalizeTableName(value.table, context);
+  }
+  // Log when we encounter an object that we can't extract table name from
+  // eslint-disable-next-line no-console
+  console.warn(
+    `[cacheTableUtils] Unable to extract table name from object:`,
+    JSON.stringify(value, null, 2),
+    context ? `(context: ${context})` : "",
+  );
+  return null;
+}
+
+/**
+ * Helper function to safely convert to string and lowercase.
+ */
+function normalizeTableName(value: any, context?: string): string | null {
+  if (!value) {
+    return null;
+  }
+  // If it's already a string, use it
+  if (typeof value === "string") {
+    return value === "dual" ? null : value.toLowerCase();
+  }
+  // If it's an object, try to extract name from various properties
+  if (typeof value === "object") {
+    return extractTableNameFromObject(value, context);
+  }
+  // For other types (number, boolean, etc.), log and don't treat as table name
+  // eslint-disable-next-line no-console
+  console.warn(
+    `[cacheTableUtils] Unexpected table name type:`,
+    typeof value,
+    value,
+    context ? `(context: ${context})` : "",
+  );
+  return null;
+}
+
+/**
+ * Checks if a node is a column reference alias.
+ */
+function isColumnRefAlias(node: any): boolean {
+  return node.type === "column_ref" && !node.table;
+}
+
+/**
+ * Checks if a node is an explicit alias (has 'as' property but is not a table node).
+ */
+function isExplicitAlias(node: any): boolean {
+  return Boolean(node.as && node.type !== "table" && node.type !== "dual" && !node.table);
+}
+
+/**
+ * Checks if a node has a short name that is likely an alias.
+ */
+function isShortNameAlias(node: any): boolean {
+  if (!node.name || node.table || node.type === "table" || node.type === "dual") {
+    return false;
+  }
+  const nameStr = typeof node.name === "string" ? node.name : node.name?.name || node.name?.value;
+  return typeof nameStr === "string" && nameStr.length <= 2;
+}
+
+/**
+ * Checks if a node is likely an alias (not a real table).
+ */
+function isLikelyAlias(node: any): boolean {
+  return isColumnRefAlias(node) || isExplicitAlias(node) || isShortNameAlias(node);
+}
+
+/**
+ * Extracts table name from table node.
+ *
+ * @param node - AST node with table information
+ * @returns Table name in lowercase or null if not applicable
+ */
+function extractTableName(node: any): string | null {
+  if (!node) {
+    return null;
+  }
+
+  // Early return for likely aliases
+  if (isLikelyAlias(node)) {
+    return null;
+  }
+
+  // Handle table node directly
+  if (node.type === "table" || node.type === "dual") {
+    const fromTable = node.table
+      ? normalizeTableName(node.table, `node.type=${node.type}, node.table`)
+      : null;
+    if (fromTable) {
+      return fromTable;
+    }
+    const fromName = node.name
+      ? normalizeTableName(node.name, `node.type=${node.type}, node.name`)
+      : null;
+    if (fromName) {
+      return fromName;
+    }
+    return null;
+  }
+
+  // Handle table reference in different formats
+  if (node.table) {
+    const tableName = normalizeTableName(node.table, `node.table (type: ${node.type})`);
+    if (tableName) {
+      return tableName;
+    }
+  }
+
+  return null;
+}
+
+/**
+ * Processes and adds table name to the set if valid.
+ */
+function processTableName(node: any, tableName: string, tables: Set<string>): void {
+  // Filter out a_ prefixed names (field aliases)
+  if (tableName.startsWith("a_")) {
+    return;
+  }
+
+  // Filter out short names that are likely table aliases (u, us, o, oi, etc.)
+  // Only filter if it's not a real table node (type === "table" or "dual")
+  const isRealTableNode = node.type === "table" || node.type === "dual";
+  if (!isRealTableNode && tableName.length <= 2) {
+    return;
+  }
+
+  tables.add(tableName);
+}
+
+/**
+ * Extracts table name from node and adds to set if valid.
+ */
+function extractAndAddTableName(node: any, tables: Set<string>): void {
+  const tableName = extractTableName(node);
+  if (tableName && tableName.length > 0) {
+    processTableName(node, tableName, tables);
+  }
+}
+
+/**
+ * Processes CTE (Common Table Expressions) - WITH clause.
+ */
+function processCTE(node: any, tables: Set<string>): void {
+  if (!node.with && !node.with_list) {
+    return;
+  }
+  const withClauses = node.with_list || (Array.isArray(node.with) ? node.with : [node.with]);
+  withClauses.forEach((cte: any) => {
+    if (cte?.stmt) {
+      extractTablesFromNode(cte.stmt, tables);
+    }
+    if (cte?.as?.stmt) {
+      extractTablesFromNode(cte.as.stmt, tables);
+    }
+  });
+}
+
+/**
+ * Processes FROM and JOIN clauses.
+ */
+function processFromAndJoin(node: any, tables: Set<string>): void {
+  if (node.from) {
+    if (Array.isArray(node.from)) {
+      node.from.forEach((item: any) => extractTablesFromNode(item, tables));
+    } else {
+      extractTablesFromNode(node.from, tables);
+    }
+  }
+
+  if (node.join) {
+    if (Array.isArray(node.join)) {
+      node.join.forEach((item: any) => extractTablesFromNode(item, tables));
+    } else {
+      extractTablesFromNode(node.join, tables);
+    }
+  }
+}
+
+/**
+ * Processes SELECT columns that may contain subqueries.
+ */
+function processSelectColumns(node: any, tables: Set<string>): void {
+  const columns = node.columns || node.select;
+  if (!columns) {
+    return;
+  }
+
+  if (Array.isArray(columns)) {
+    columns.forEach((col: any) => {
+      if (!col) return;
+
+      // If the column itself is a subquery
+      if (col.type === "subquery" || col.type === "select") {
+        extractTablesFromNode(col, tables);
+      }
+
+      // Process expression (may contain subqueries)
+      if (col.expr) {
+        extractTablesFromNode(col.expr, tables);
+      }
+
+      // Process AST (alternative structure for subqueries)
+      if (col.ast) {
+        extractTablesFromNode(col.ast, tables);
+      }
+    });
+  } else if (typeof columns === "object") {
+    extractTablesFromNode(columns, tables);
+  }
+}
+
+/**
+ * Processes ORDER BY or GROUP BY clause.
+ */
+function processOrderByOrGroupBy(clause: any, tables: Set<string>): void {
+  if (!clause) {
+    return;
+  }
+  if (Array.isArray(clause)) {
+    clause.forEach((item: any) => {
+      if (item?.expr) {
+        extractTablesFromNode(item.expr, tables);
+      }
+      extractTablesFromNode(item, tables);
+    });
+  } else {
+    extractTablesFromNode(clause, tables);
+  }
+}
+
+/**
+ * Processes UNION operations.
+ */
+function processUnionNode(unionNode: any, tables: Set<string>): void {
+  if (!unionNode) {
+    return;
+  }
+
+  const isUnionType =
+    unionNode.type === "select" ||
+    unionNode.type === "union" ||
+    unionNode.type === "union_all" ||
+    unionNode.type === "union_distinct" ||
+    unionNode.type === "intersect" ||
+    unionNode.type === "except" ||
+    unionNode.type === "minus";
+
+  if (isUnionType) {
+    extractTablesFromNode(unionNode, tables);
+  } else if (unionNode.select) {
+    extractTablesFromNode(unionNode.select, tables);
+  } else if (unionNode.ast) {
+    extractTablesFromNode(unionNode.ast, tables);
+  } else {
+    extractTablesFromNode(unionNode, tables);
+  }
+}
+
+/**
+ * Processes UNION/UNION ALL/UNION DISTINCT/INTERSECT/EXCEPT/MINUS clauses.
+ */
+function processUnion(node: any, tables: Set<string>): void {
+  if (!node.union) {
+    return;
+  }
+
+  if (Array.isArray(node.union)) {
+    node.union.forEach((unionNode: any) => processUnionNode(unionNode, tables));
+  } else if (typeof node.union === "object") {
+    processUnionNode(node.union, tables);
+  }
+}
+
+/**
+ * Processes UNION/INTERSECT/EXCEPT operation nodes.
+ */
+function processUnionOperation(node: any, tables: Set<string>): void {
+  const isUnionOperation =
+    node.type === "union" ||
+    node.type === "union_all" ||
+    node.type === "union_distinct" ||
+    node.type === "intersect" ||
+    node.type === "except" ||
+    node.type === "minus";
+
+  if (!isUnionOperation) {
+    return;
+  }
+
+  if (node.left) {
+    extractTablesFromNode(node.left, tables);
+  }
+  if (node.right) {
+    extractTablesFromNode(node.right, tables);
+  }
+  extractTablesFromNode(node, tables);
+}
+
+/**
+ * Processes _next property (alternative UNION structure).
+ */
+function processNext(node: any, tables: Set<string>): void {
+  if (!node._next) {
+    return;
+  }
+  if (Array.isArray(node._next)) {
+    node._next.forEach((nextNode: any) => extractTablesFromNode(nextNode, tables));
+  } else {
+    extractTablesFromNode(node._next, tables);
+  }
+}
+
+/**
+ * Recursively processes all object properties for any remaining nested structures.
+ */
+function processRecursively(node: any, tables: Set<string>): void {
+  const isLikelyAlias =
+    (node.type === "column_ref" && !node.table) ||
+    (node.name &&
+      !node.table &&
+      node.type !== "table" &&
+      node.type !== "dual" &&
+      node.name.length <= 2);
+
+  if (isLikelyAlias || Array.isArray(node)) {
+    return;
+  }
+
+  Object.values(node).forEach((value) => {
+    if (value && typeof value === "object") {
+      if (Array.isArray(value)) {
+        value.forEach((item: any) => {
+          if (item && typeof item === "object") {
+            extractTablesFromNode(item, tables);
+          }
+        });
+      } else {
+        extractTablesFromNode(value, tables);
+      }
+    }
+  });
+}
+
+/**
+ * Recursively extracts table names from SQL AST node.
+ * Handles regular tables, CTEs, subqueries, and complex query structures.
+ *
+ * @param node - AST node to extract tables from
+ * @param tables - Accumulator set for table names
+ */
+function extractTablesFromNode(node: any, tables: Set<string>): void {
+  if (!node || typeof node !== "object") {
+    return;
+  }
+
+  // Extract table name if node is a table type
+  extractAndAddTableName(node, tables);
+
+  // Handle CTE (Common Table Expressions) - WITH clause
+  processCTE(node, tables);
+
+  // Extract tables from FROM and JOIN clauses
+  processFromAndJoin(node, tables);
+
+  // Handle subqueries explicitly
+  if (node.type === "subquery" || node.type === "select") {
+    if (node.ast) {
+      extractTablesFromNode(node.ast, tables);
+    }
+    if (node.from) {
+      extractTablesFromNode(node.from, tables);
+    }
+  }
+
+  // Extract tables from WHERE clause (may contain subqueries)
+  if (node.where) {
+    extractTablesFromNode(node.where, tables);
+  }
+
+  // Extract tables from SELECT columns (may contain subqueries)
+  processSelectColumns(node, tables);
+
+  // Extract tables from HAVING clause (may contain subqueries)
+  if (node.having) {
+    extractTablesFromNode(node.having, tables);
+  }
+
+  // Extract tables from ORDER BY clause (may contain subqueries)
+  processOrderByOrGroupBy(node.orderby || node.order_by, tables);
+
+  // Extract tables from GROUP BY clause (may contain subqueries)
+  processOrderByOrGroupBy(node.groupby || node.group_by, tables);
+
+  // Extract tables from UPDATE statement
+  if (node.type === "update" && node.table) {
+    extractTablesFromNode(node.table, tables);
+  }
+
+  // Extract tables from INSERT statement
+  if (node.type === "insert" && node.table) {
+    extractTablesFromNode(node.table, tables);
+  }
+
+  // Extract tables from DELETE statement
+  if (node.type === "delete" && node.from) {
+    extractTablesFromNode(node.from, tables);
+  }
+
+  // Extract tables from UNION operations
+  processUnion(node, tables);
+
+  // Handle node types for UNION/INTERSECT/EXCEPT operations
+  processUnionOperation(node, tables);
+
+  // Handle _next property (alternative UNION structure)
+  processNext(node, tables);
+
+  // Recursively process all object properties
+  processRecursively(node, tables);
+}
+
+/**
+ * Extracts all table names from SQL query using node-sql-parser, with regex fallback.
+ * Returns them as comma-separated string in format `table1`,`table2`.
+ *
+ * @param sql - SQL query string
+ * @param options - ForgeSQL ORM options for logging
+ * @returns Comma-separated string of unique table names in backticks
+ */
+export function extractBacktickedValues(sql: string, options: ForgeSqlOrmOptions): string {
+  // Try to use node-sql-parser first
+  try {
+    const parser = new Parser();
+    const ast = parser.astify(sql.trim());
+
+    const tables = new Set<string>();
+
+    // Handle both single statement and multiple statements
+    const statements = Array.isArray(ast) ? ast : [ast];
+    statements.forEach((statement) => {
+      extractTablesFromNode(statement, tables);
+    });
+
+    if (tables.size > 0) {
+      // Sort to ensure consistent order for the same input
+      const backtickedValues = Array.from(tables)
+        .sort((a, b) => a.localeCompare(b, undefined, { sensitivity: "base", numeric: true }))
+        .map((table) => `\`${table}\``)
+        .join(",");
+      if (options.logCache) {
+        // eslint-disable-next-line no-console
+        console.warn(`Extracted backticked values: ${backtickedValues}`);
+      }
+      return backtickedValues;
+    }
+  } catch (error) {
+    if (options.logCache) {
+      // eslint-disable-next-line no-console
+      console.error(
+        `Error extracting backticked values: ${error}. Using regex-based extraction instead.`,
+      );
+    }
+    // If parsing fails, fall back to regex-based extraction
+    // This handles cases where node-sql-parser doesn't support the SQL syntax
+  }
+
+  // Fallback to regex-based extraction (original logic)
+  const regex = /`([^`]+)`/g;
+  const matches = new Set<string>();
+  let match;
+
+  while ((match = regex.exec(sql.toLowerCase())) !== null) {
+    if (!match[1].startsWith("a_")) {
+      matches.add(`\`${match[1]}\``);
+    }
+  }
+
+  // Sort to ensure consistent order for the same input
+  return Array.from(matches)
+    .sort((a, b) => a.localeCompare(b, undefined, { sensitivity: "base", numeric: true }))
+    .join(",");
+}