feat: add caller location logging for SQL queries with Rails-style indicator

Author: Dylan Dietz

src/logger.ts

@@ -13,6 +13,7 @@ import {
   createTruncateForLog,
   getQueryType,
   formatParams,
+  getCallerLocation,
 } from './utils.js';
 import { createSqlFormatter } from './sqlFormatter.js';
 
@@ -123,8 +124,8 @@ export class EnhancedLogger {
 
         // Handle SQL query logs - Winston spreads data directly on info, not in message
         if (level === 'query' && 'query' in info && 'type' in info) {
-          const queryInfo = info as unknown as { type: string; query: string; params?: string; duration: string; timestamp: string };
-          const { query, params = '', duration, type } = queryInfo;
+          const queryInfo = info as unknown as { type: string; query: string; params?: string; duration: string; timestamp: string; caller?: string };
+          const { query, params = '', duration, type, caller } = queryInfo;
 
           // Determine query type for color coding
           const queryType = type.toUpperCase();
@@ -212,7 +213,7 @@ export class EnhancedLogger {
           if (this.config.enableColors) {
             switch (queryType) {
               case 'SELECT':
-                logLine = chalk.blue(logLine);
+                logLine = chalk.cyan(logLine);  // Light blue for all SELECT queries
                 break;
               case 'UPDATE':
                 logLine = chalk.yellow(logLine);
@@ -234,6 +235,12 @@ export class EnhancedLogger {
             }
           }
 
+          // Add caller location if available (Rails-style ↳ indicator)
+          if (caller) {
+            const callerLine = this.config.enableColors ? chalk.dim(`  ${caller}`) : `  ${caller}`;
+            logLine += `\n${callerLine}`;
+          }
+          
           return logLine;
         }
 
@@ -406,6 +413,9 @@ export class EnhancedLogger {
       const queryType = getQueryType(e.query);
       const formattedParams = formatParams(e.params);
       const duration = Number(e.duration);
+      
+      // Capture caller location
+      const caller = getCallerLocation();
 
       if (duration > this.config.slowQueryThreshold) {
         // Truncate both query and params for slow queries
@@ -414,6 +424,9 @@ export class EnhancedLogger {
 
         // Rails-style slow query logging
         this.warn(`  ${queryType} (${duration}ms)  ${truncatedQuery}  ${truncatedParams}`);
+        if (caller) {
+          this.warn(`  ${caller}`);
+        }
         this.warn(`  Slow query detected`);
       } else {
         // Use logger.query() with proper QueryLogData format for Rails-style formatting
@@ -422,6 +435,7 @@ export class EnhancedLogger {
           query: e.query,
           params: formattedParams,
           duration: `${duration}ms`,
+          caller, // Add caller location
         });
       }
     });

src/sqlFormatter.ts

@@ -8,10 +8,6 @@ export const createSqlFormatter = (config: LoggerConfig) => {
   const truncateForLog = createTruncateForLog(config);
   const enableColors = config.enableColors ?? true;
   const maxStringLength = config.maxStringLength ?? 100;
-  
-  // For chalk v4: Just use the default instance when colors enabled
-  // When disabled, we'll just pass through without coloring
-  const chalkInstance = chalk;
 
   const formatSqlQuery = (query: string, params: string): string => {
     // If custom formatter is provided, use it

src/types.ts

@@ -74,6 +74,8 @@ export interface QueryLogData {
   query: string;
   params: string;
   duration: string;
+  /** Optional caller location (Rails-style ↳ indicator) */
+  caller?: string;
 }
 
 export interface RequestLogData {

src/utils.ts

@@ -253,3 +253,80 @@ export function formatParams(params: unknown): string {
     return String(params);
   }
 }
+
+/**
+ * Get caller location from stack trace (Rails-style ↳ indicator)
+ * Returns formatted location like "↳ app/controllers/users.ts:42:in `getUser`"
+ */
+export function getCallerLocation(): string {
+  const oldLimit = Error.stackTraceLimit;
+  Error.stackTraceLimit = 50; // Capture enough frames
+  
+  const stack = new Error().stack || '';
+  Error.stackTraceLimit = oldLimit;
+  
+  const lines = stack.split('\n');
+  
+  // Start from frame 1 (skip "Error" line)
+  for (let i = 1; i < lines.length; i++) {
+    const line = lines[i];
+    
+    if (!line) continue;
+    
+    // Skip internal frames we want to ignore
+    if (
+      line.includes('node_modules') ||
+      line.includes('node:internal') ||
+      line.includes('node:diagnostics_channel') ||
+      line.includes('node:async_hooks') ||
+      line.includes('/dist/') ||         // Our compiled library code
+      line.includes('logger.cjs') ||
+      line.includes('logger.js') ||
+      line.includes('utils.cjs') ||
+      line.includes('utils.js') ||
+      line.includes('sqlFormatter') ||
+      line.includes('EnhancedLogger') ||
+      line.includes('setupPrismaLogging') ||
+      line.includes('getCallerLocation') ||
+      line.includes('winston') ||
+      line.includes('prisma/client') ||
+      line.includes('prisma-client') ||
+      line.includes('@prisma') ||
+      line.includes('TracingChannel') ||
+      line.includes('Module._compile') ||
+      line.includes('Module.load') ||
+      line.includes('Function._load') ||
+      line.includes('_triggerQuery') ||  // Mock Prisma trigger
+      line.includes('.$on(') ||           // Prisma event registration
+      line.includes('PrismaClient')       // Prisma client internals
+    ) {
+      continue;
+    }
+    
+    // Try to extract file location
+    // Format: "    at functionName (file:line:column)" or "    at file:line:column"
+    const match = line.match(/at\s+(?:([^\s]+)\s+\()?([^)]+):(\d+):(\d+)\)?/);
+    
+    if (match) {
+      const functionName = match[1] || 'anonymous';
+      let filePath = match[2] || '';
+      const lineNum = match[3];
+      
+      if (!filePath) continue;
+      
+      // Clean up file path - remove file:// protocol
+      filePath = filePath.replace('file://', '');
+      
+      // Try to make it relative to current working directory
+      const cwd = process.cwd();
+      if (filePath.startsWith(cwd)) {
+        filePath = filePath.substring(cwd.length + 1);
+      }
+      
+      // Format like Rails: ↳ path/to/file.ts:line:in `function`
+      return `↳ ${filePath}:${lineNum}:in \`${functionName}\``;
+    }
+  }
+  
+  return ''; // No suitable caller found
+}