prolog-trace-viz 1.1.3 → 2.1.0

package/dist/variable-tracker.js

@@ -0,0 +1,216 @@
+/**
+ * Variable Binding Tracker
+ *
+ * Tracks variable bindings through recursive Prolog execution by processing
+ * trace events in order and maintaining accumulated state.
+ */
+export class VariableBindingTracker {
+    bindings = new Map();
+    topLevelQuery;
+    queryVarName;
+    constructor(originalQuery) {
+        this.topLevelQuery = originalQuery;
+        // Extract the query variable name (e.g., "X" from "append([1,2], [3,4], X)")
+        this.queryVarName = this.extractQueryVarName(originalQuery);
+    }
+    /**
+     * Extract the query variable name from the original query
+     * e.g., "append([1,2], [3,4], X)" -> "X"
+     * e.g., "factorial(3, Result)" -> "Result"
+     */
+    extractQueryVarName(query) {
+        const match = query.match(/^([^(]+)\((.*)\)$/);
+        if (!match)
+            return 'X'; // fallback
+        const args = this.splitArguments(match[2]);
+        const lastArg = args[args.length - 1]?.trim();
+        // Check if last argument is a variable (starts with uppercase or underscore)
+        if (lastArg && /^[A-Z_]/.test(lastArg)) {
+            return lastArg;
+        }
+        return 'X'; // fallback
+    }
+    /**
+     * Process a trace event and update bindings
+     */
+    processEvent(event) {
+        if (event.port === 'call') {
+            this.processCall(event);
+        }
+        else if (event.port === 'exit') {
+            this.processExit(event);
+        }
+    }
+    /**
+     * Process CALL event - extract variable relationships from parent_info
+     *
+     * Key insight: parent_info.goal shows how the PARENT sees its own result variable
+     * at the moment it's calling this child.
+     */
+    processCall(event) {
+        // Extract the result variable from this call's goal
+        const resultVar = this.extractResultVariable(event.goal);
+        if (!resultVar)
+            return;
+        // Initialize binding for this level
+        this.bindings.set(event.level, {
+            level: event.level,
+            resultVar,
+            resultPattern: resultVar, // Initially just the variable name
+            isResolved: false,
+        });
+        // If there's parent_info, it tells us how the parent sees ITS OWN result
+        if (event.parent_info && event.parent_info.goal) {
+            const parentGoal = event.parent_info.goal;
+            // Skip meta-calls and test predicates
+            if (parentGoal.includes('<meta-call>') || parentGoal.includes('test_')) {
+                return;
+            }
+            // Extract what the parent's result looks like from parent's perspective
+            const parentResultPattern = this.extractResultVariable(parentGoal);
+            if (!parentResultPattern)
+                return;
+            const parentLevel = event.parent_info.level;
+            const parentBinding = this.bindings.get(parentLevel);
+            if (parentBinding) {
+                // Save the parent's OLD result variable before updating
+                const parentOldResultVar = parentBinding.resultVar;
+                // Update parent's pattern - this is how parent sees its result NOW
+                parentBinding.resultPattern = parentResultPattern;
+                // Propagate: substitute the parent's OLD result variable with its NEW pattern in all ancestors
+                // e.g., if parent was "_79854" and now has pattern "[2|_79774]",
+                // then ancestors with "[1|_79854]" should become "[1|[2|_79774]]"
+                if (parentOldResultVar !== parentResultPattern) {
+                    this.propagateBinding(parentLevel, parentOldResultVar, parentResultPattern);
+                }
+            }
+        }
+    }
+    /**
+     * Process EXIT event - we now know the final value
+     */
+    processExit(event) {
+        const binding = this.bindings.get(event.level);
+        if (!binding)
+            return;
+        // Extract the resolved value from EXIT goal
+        const resolvedValue = this.extractResultVariable(event.goal);
+        if (!resolvedValue)
+            return;
+        // Mark as resolved and update pattern
+        binding.isResolved = true;
+        binding.resultPattern = resolvedValue;
+        // Propagate this binding up to all parents that reference this variable
+        this.propagateBinding(event.level, binding.resultVar, resolvedValue);
+    }
+    /**
+     * Propagate a binding up through parent levels
+     */
+    propagateBinding(fromLevel, varName, value) {
+        // Find all levels that reference this variable and substitute
+        for (const [level, binding] of this.bindings.entries()) {
+            if (level < fromLevel && binding.resultPattern.includes(varName)) {
+                binding.resultPattern = this.substitute(binding.resultPattern, varName, value);
+            }
+        }
+    }
+    /**
+     * Get the accumulated query variable state for a given level
+     * This shows what the top-level query variable looks like at this point
+     */
+    getQueryVarState(level) {
+        // Find the top-level binding (lowest level number)
+        let topLevel = Infinity;
+        for (const lvl of this.bindings.keys()) {
+            if (lvl < topLevel) {
+                topLevel = lvl;
+            }
+        }
+        if (topLevel === Infinity)
+            return null;
+        const topBinding = this.bindings.get(topLevel);
+        if (!topBinding)
+            return null;
+        // Clean up the pattern to show holes
+        const cleaned = this.cleanupPattern(topBinding.resultPattern);
+        // Always show the state (whether partial or complete)
+        return `${this.queryVarName} = ${cleaned}`;
+    }
+    /**
+     * Extract the result variable/pattern from a goal
+     * For "append([1,2],[3,4],_79984)", returns "_79984"
+     * For "append([1,2],[3,4],[1|_79854])", returns "[1|_79854]"
+     */
+    extractResultVariable(goal) {
+        const match = goal.match(/^([^(]+)\((.*)\)$/);
+        if (!match)
+            return null;
+        const args = this.splitArguments(match[2]);
+        // Return the last argument (typically the result in Prolog predicates)
+        return args[args.length - 1]?.trim() || null;
+    }
+    /**
+     * Substitute a variable with a value in a pattern
+     * e.g., substitute("[1|_79854]", "_79854", "[2|_79774]") -> "[1|[2|_79774]]"
+     */
+    substitute(pattern, varName, value) {
+        // Simple string replacement for now
+        // TODO: Handle nested structures more carefully
+        return pattern.replace(new RegExp(`\\b${this.escapeRegex(varName)}\\b`, 'g'), value);
+    }
+    /**
+     * Clean up a pattern to show holes
+     * Replace unbound variables with ? and simplify nested lists
+     */
+    cleanupPattern(pattern) {
+        // Replace internal variables (_NNNN) with ?
+        let cleaned = pattern.replace(/_\d+/g, '?');
+        // Replace remaining unbound variables (uppercase identifiers)
+        cleaned = cleaned.replace(/\b[A-Z][A-Za-z0-9_]*\b/g, '?');
+        // Simplify nested list structures: [1|[2|[3,4]]] -> [1,2,3,4]
+        // Keep doing this until no more simplifications possible
+        let prev = '';
+        while (prev !== cleaned) {
+            prev = cleaned;
+            // Pattern: [X|[Y,...]] -> [X,Y,...]
+            cleaned = cleaned.replace(/\[([^\[\]|]+)\|\[([^\[\]]+)\]\]/g, '[$1,$2]');
+        }
+        return cleaned;
+    }
+    /**
+     * Split arguments respecting parentheses and brackets
+     */
+    splitArguments(argsStr) {
+        const args = [];
+        let current = '';
+        let depth = 0;
+        for (const char of argsStr) {
+            if (char === '(' || char === '[') {
+                depth++;
+                current += char;
+            }
+            else if (char === ')' || char === ']') {
+                depth--;
+                current += char;
+            }
+            else if (char === ',' && depth === 0) {
+                args.push(current.trim());
+                current = '';
+            }
+            else {
+                current += char;
+            }
+        }
+        if (current.trim()) {
+            args.push(current.trim());
+        }
+        return args;
+    }
+    /**
+     * Escape special regex characters
+     */
+    escapeRegex(str) {
+        return str.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+    }
+}
+//# sourceMappingURL=variable-tracker.js.map

package/dist/variable-tracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"variable-tracker.js","sourceRoot":"","sources":["../src/variable-tracker.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAWH,MAAM,OAAO,sBAAsB;IACzB,QAAQ,GAA8B,IAAI,GAAG,EAAE,CAAC;IAChD,aAAa,CAAS;IACtB,YAAY,CAAS;IAE7B,YAAY,aAAqB;QAC/B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,6EAA6E;QAC7E,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,aAAa,CAAC,CAAC;IAC9D,CAAC;IAED;;;;OAIG;IACK,mBAAmB,CAAC,KAAa;QACvC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK;YAAE,OAAO,GAAG,CAAC,CAAC,WAAW;QAEnC,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC;QAE9C,6EAA6E;QAC7E,IAAI,OAAO,IAAI,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;YACvC,OAAO,OAAO,CAAC;QACjB,CAAC;QAED,OAAO,GAAG,CAAC,CAAC,WAAW;IACzB,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,KAAiB;QAC5B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACjC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACK,WAAW,CAAC,KAAiB;QACnC,oDAAoD;QACpD,MAAM,SAAS,GAAG,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACzD,IAAI,CAAC,SAAS;YAAE,OAAO;QAEvB,oCAAoC;QACpC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE;YAC7B,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,SAAS;YACT,aAAa,EAAE,SAAS,EAAE,mCAAmC;YAC7D,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;QAEH,yEAAyE;QACzE,IAAI,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC;YAChD,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC;YAE1C,sCAAsC;YACtC,IAAI,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBACvE,OAAO;YACT,CAAC;YAED,wEAAwE;YACxE,MAAM,mBAAmB,GAAG,IAAI,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;YACnE,IAAI,CAAC,mBAAmB;gBAAE,OAAO;YAEjC,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC;YAC5C,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;YAErD,IAAI,aAAa,EAAE,CAAC;gBAClB,wDAAwD;gBACxD,MAAM,kBAAkB,GAAG,aAAa,CAAC,SAAS,CAAC;gBAEnD,mEAAmE;gBACnE,aAAa,CAAC,aAAa,GAAG,mBAAmB,CAAC;gBAElD,+FAA+F;gBAC/F,iEAAiE;gBACjE,kEAAkE;gBAClE,IAAI,kBAAkB,KAAK,mBAAmB,EAAE,CAAC;oBAC/C,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE,kBAAkB,EAAE,mBAAmB,CAAC,CAAC;gBAC9E,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,KAAiB;QACnC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO;YAAE,OAAO;QAErB,4CAA4C;QAC5C,MAAM,aAAa,GAAG,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC7D,IAAI,CAAC,aAAa;YAAE,OAAO;QAE3B,sCAAsC;QACtC,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;QAC1B,OAAO,CAAC,aAAa,GAAG,aAAa,CAAC;QAEtC,wEAAwE;QACxE,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,KAAK,EAAE,OAAO,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;IACvE,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,SAAiB,EAAE,OAAe,EAAE,KAAa;QACxE,8DAA8D;QAC9D,KAAK,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC;YACvD,IAAI,KAAK,GAAG,SAAS,IAAI,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBACjE,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC,UAAU,CACrC,OAAO,CAAC,aAAa,EACrB,OAAO,EACP,KAAK,CACN,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,gBAAgB,CAAC,KAAa;QAC5B,mDAAmD;QACnD,IAAI,QAAQ,GAAG,QAAQ,CAAC;QACxB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC;YACvC,IAAI,GAAG,GAAG,QAAQ,EAAE,CAAC;gBACnB,QAAQ,GAAG,GAAG,CAAC;YACjB,CAAC;QACH,CAAC;QAED,IAAI,QAAQ,KAAK,QAAQ;YAAE,OAAO,IAAI,CAAC;QAEvC,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC/C,IAAI,CAAC,UAAU;YAAE,OAAO,IAAI,CAAC;QAE7B,qCAAqC;QACrC,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;QAE9D,sDAAsD;QACtD,OAAO,GAAG,IAAI,CAAC,YAAY,MAAM,OAAO,EAAE,CAAC;IAC7C,CAAC;IAED;;;;OAIG;IACK,qBAAqB,CAAC,IAAY;QACxC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;QAC9C,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAExB,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3C,uEAAuE;QACvE,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,IAAI,CAAC;IAC/C,CAAC;IAED;;;OAGG;IACK,UAAU,CAAC,OAAe,EAAE,OAAe,EAAE,KAAa;QAChE,oCAAoC;QACpC,gDAAgD;QAChD,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,KAAK,CAAC,CAAC;IACvF,CAAC;IAED;;;OAGG;IACK,cAAc,CAAC,OAAe;QACpC,4CAA4C;QAC5C,IAAI,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QAC5C,8DAA8D;QAC9D,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;QAE1D,8DAA8D;QAC9D,yDAAyD;QACzD,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,OAAO,IAAI,KAAK,OAAO,EAAE,CAAC;YACxB,IAAI,GAAG,OAAO,CAAC;YACf,oCAAoC;YACpC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,kCAAkC,EAAE,SAAS,CAAC,CAAC;QAC3E,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,OAAe;QACpC,MAAM,IAAI,GAAa,EAAE,CAAC;QAC1B,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;YAC3B,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACjC,KAAK,EAAE,CAAC;gBACR,OAAO,IAAI,IAAI,CAAC;YAClB,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACxC,KAAK,EAAE,CAAC;gBACR,OAAO,IAAI,IAAI,CAAC;YAClB,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;gBACvC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC1B,OAAO,GAAG,EAAE,CAAC;YACf,CAAC;iBAAM,CAAC;gBACN,OAAO,IAAI,IAAI,CAAC;YAClB,CAAC;QACH,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;YACnB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAC5B,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,GAAW;QAC7B,OAAO,GAAG,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC;IACpD,CAAC;CACF"}

package/dist/wrapper.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"wrapper.d.ts","sourceRoot":"","sources":["../src/wrapper.ts"],"names":[],"mappings":"AAIA,MAAM,WAAW,aAAa;IAC5B,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CAC9B;AAED;;;;;;;;GAQG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,aAAa,GAAG,MAAM,CA+B7D;AAED;;;;GAIG;AACH,wBAAgB,mBAAmB,CAAC,aAAa,EAAE,MAAM,GAAG,MAAM,CASjE;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CAAC,WAAW,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,MAAM,CAwBzF;AACD,wBAAsB,iBAAiB,CAAC,MAAM,EAAE,aAAa,GAAG,OAAO,CAAC,QAAQ,CAAC,CAiBhF;AAED;;;GAGG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,aAAa,GAAG,IAAI,CA2BlE"}
+{"version":3,"file":"wrapper.d.ts","sourceRoot":"","sources":["../src/wrapper.ts"],"names":[],"mappings":"AAIA,MAAM,WAAW,aAAa;IAC5B,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CAC9B;AAED;;;;;;;;GAQG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,aAAa,GAAG,MAAM,CAgC7D;AAED;;;;GAIG;AACH,wBAAgB,mBAAmB,CAAC,aAAa,EAAE,MAAM,GAAG,MAAM,CASjE;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CAAC,WAAW,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,MAAM,CAMzF;AACD,wBAAsB,iBAAiB,CAAC,MAAM,EAAE,aAAa,GAAG,OAAO,CAAC,QAAQ,CAAC,CAiBhF;AAED;;;GAGG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,aAAa,GAAG,IAAI,CA2BlE"}

package/dist/wrapper.js

@@ -11,7 +11,8 @@ import * as os from 'node:os';
  * - Tracer cleanup
  */
 export function generateWrapper(config) {
-    const { prologContent, query, tracerPath } = config;
+    const { prologContent, query, depth, tracerPath } = config;
+    const maxDepth = depth || 100; // Default to 100 if not specified
     const lines = [
         `% Load custom tracer`,
         `:- ['${tracerPath}'].`,
@@ -25,7 +26,7 @@ export function generateWrapper(config) {
     lines.push('');
     lines.push('% Run trace with error handling');
     lines.push('run_trace :-');
-    lines.push('    install_tracer,');
+    lines.push(`    install_tracer(${maxDepth}),`);
     lines.push('    catch(');
     lines.push(`        (${query.trim()}, export_trace_json('trace.json')),`);
     lines.push('        Error,');
@@ -57,24 +58,9 @@ export function calculateLineOffset(prologContent) {
  */
 export function mapWrapperLineToSource(wrapperLine, prologContent) {
     const offset = calculateLineOffset(prologContent);
-    const sourceLineInWrapper = wrapperLine - offset;
-    // Now we need to find which actual source line this corresponds to
-    // by accounting for empty lines and comments that were preserved
-    const sourceLines = prologContent.split('\n');
-    let actualSourceLine = 0;
-    let nonEmptyLineCount = 0;
-    for (let i = 0; i < sourceLines.length; i++) {
-        actualSourceLine = i + 1; // 1-based line numbers
-        const line = sourceLines[i].trim();
-        // Skip empty lines and comments when counting
-        if (line.length > 0 && !line.startsWith('%') && !line.startsWith('/*')) {
-            nonEmptyLineCount++;
-            if (nonEmptyLineCount === sourceLineInWrapper) {
-                return actualSourceLine;
-            }
-        }
-    }
-    return actualSourceLine; // Fallback to last line
+    // Simple mapping: wrapper line - offset = source line
+    // The wrapper includes all source lines as-is, so no need to skip empty lines or comments
+    return wrapperLine - offset;
 }
 export async function createTempWrapper(config) {
     const content = generateWrapper(config);

package/dist/wrapper.js.map

@@ -1 +1 @@
-{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../src/wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAc9B;;;;;;;;GAQG;AACH,MAAM,UAAU,eAAe,CAAC,MAAqB;IACnD,MAAM,EAAE,aAAa,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,MAAM,CAAC;IAEpD,MAAM,KAAK,GAAa;QACtB,sBAAsB;QACtB,QAAQ,UAAU,KAAK;QACvB,EAAE;QACF,2CAA2C;KAC5C,CAAC;IAEF,0CAA0C;IAC1C,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACnD,MAAM,gBAAgB,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,0BAA0B;IAErE,KAAK,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;IACzB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;IAC9C,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC3B,KAAK,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;IAClC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACzB,KAAK,CAAC,IAAI,CAAC,YAAY,KAAK,CAAC,IAAI,EAAE,qCAAqC,CAAC,CAAC;IAC1E,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAC7B,KAAK,CAAC,IAAI,CAAC,2EAA2E,CAAC,CAAC;IACxF,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACrB,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IACjC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAC5B,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACvB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CAAC,aAAqB;IACvD,4BAA4B;IAC5B,+BAA+B;IAC/B,6BAA6B;IAC7B,kBAAkB;IAClB,oDAAoD;IACpD,oCAAoC;IAEpC,OAAO,CAAC,CAAC,CAAC,2DAA2D;AACvE,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,sBAAsB,CAAC,WAAmB,EAAE,aAAqB;IAC/E,MAAM,MAAM,GAAG,mBAAmB,CAAC,aAAa,CAAC,CAAC;IAClD,MAAM,mBAAmB,GAAG,WAAW,GAAG,MAAM,CAAC;IAEjD,mEAAmE;IACnE,iEAAiE;IACjE,MAAM,WAAW,GAAG,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC9C,IAAI,gBAAgB,GAAG,CAAC,CAAC;IACzB,IAAI,iBAAiB,GAAG,CAAC,CAAC;IAE1B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5C,gBAAgB,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,uBAAuB;QACjD,MAAM,IAAI,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAEnC,8CAA8C;QAC9C,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YACvE,iBAAiB,EAAE,CAAC;YACpB,IAAI,iBAAiB,KAAK,mBAAmB,EAAE,CAAC;gBAC9C,OAAO,gBAAgB,CAAC;YAC1B,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,gBAAgB,CAAC,CAAC,wBAAwB;AACnD,CAAC;AACD,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,MAAqB;IAC3D,MAAM,OAAO,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;IACxC,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;IAC9E,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;IAErD,MAAM,EAAE,CAAC,SAAS,CAAC,WAAW,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IAElD,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC;gBACH,MAAM,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACzD,CAAC;YAAC,MAAM,CAAC;gBACP,wBAAwB;YAC1B,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,YAAY,CAAC,OAAe;IAC1C,sBAAsB;IACtB,MAAM,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC1D,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,UAAU,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;IAElC,8FAA8F;IAC9F,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,6CAA6C,CAAC,CAAC;IAChF,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAEnC,oEAAoE;IACpE,MAAM,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC,sFAAsF,CAAC,CAAC;IAC3H,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,aAAa,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAE7C,OAAO;QACL,aAAa;QACb,KAAK;QACL,UAAU;KACX,CAAC;AACJ,CAAC"}
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../src/wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAc9B;;;;;;;;GAQG;AACH,MAAM,UAAU,eAAe,CAAC,MAAqB;IACnD,MAAM,EAAE,aAAa,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,MAAM,CAAC;IAC3D,MAAM,QAAQ,GAAG,KAAK,IAAI,GAAG,CAAC,CAAC,kCAAkC;IAEjE,MAAM,KAAK,GAAa;QACtB,sBAAsB;QACtB,QAAQ,UAAU,KAAK;QACvB,EAAE;QACF,2CAA2C;KAC5C,CAAC;IAEF,0CAA0C;IAC1C,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACnD,MAAM,gBAAgB,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,0BAA0B;IAErE,KAAK,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;IACzB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;IAC9C,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC3B,KAAK,CAAC,IAAI,CAAC,sBAAsB,QAAQ,IAAI,CAAC,CAAC;IAC/C,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACzB,KAAK,CAAC,IAAI,CAAC,YAAY,KAAK,CAAC,IAAI,EAAE,qCAAqC,CAAC,CAAC;IAC1E,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAC7B,KAAK,CAAC,IAAI,CAAC,2EAA2E,CAAC,CAAC;IACxF,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACrB,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IACjC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAC5B,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACvB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CAAC,aAAqB;IACvD,4BAA4B;IAC5B,+BAA+B;IAC/B,6BAA6B;IAC7B,kBAAkB;IAClB,oDAAoD;IACpD,oCAAoC;IAEpC,OAAO,CAAC,CAAC,CAAC,2DAA2D;AACvE,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,sBAAsB,CAAC,WAAmB,EAAE,aAAqB;IAC/E,MAAM,MAAM,GAAG,mBAAmB,CAAC,aAAa,CAAC,CAAC;IAElD,sDAAsD;IACtD,0FAA0F;IAC1F,OAAO,WAAW,GAAG,MAAM,CAAC;AAC9B,CAAC;AACD,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,MAAqB;IAC3D,MAAM,OAAO,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;IACxC,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;IAC9E,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;IAErD,MAAM,EAAE,CAAC,SAAS,CAAC,WAAW,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IAElD,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC;gBACH,MAAM,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACzD,CAAC;YAAC,MAAM,CAAC;gBACP,wBAAwB;YAC1B,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,YAAY,CAAC,OAAe;IAC1C,sBAAsB;IACtB,MAAM,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC1D,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,UAAU,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;IAElC,8FAA8F;IAC9F,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,6CAA6C,CAAC,CAAC;IAChF,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAEnC,oEAAoE;IACpE,MAAM,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC,sFAAsF,CAAC,CAAC;IAC3H,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,aAAa,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAE7C,OAAO;QACL,aAAa;QACb,KAAK;QACL,UAAU;KACX,CAAC;AACJ,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "prolog-trace-viz",
-  "version": "1.1.3",
+  "version": "2.1.0",
   "description": "Generate enhanced visual trace diagrams for Prolog query execution",
   "type": "module",
   "main": "dist/index.js",

package/tracer.pl

@@ -3,10 +3,17 @@
 
 :- dynamic trace_event/1.
 :- dynamic trace_active/0.
+:- dynamic max_trace_depth/1.
 
 %% install_tracer/0
-%  Install the trace interception hook
+%  Install the trace interception hook with default depth
 install_tracer :-
+    install_tracer(100).
+
+%% install_tracer/1
+%  Install the trace interception hook with specified max depth
+install_tracer(MaxDepth) :-
+    asserta(max_trace_depth(MaxDepth)),
     asserta(trace_active),
     trace.
 
@@ -15,6 +22,7 @@ install_tracer :-
 remove_tracer :-
     notrace,
     retractall(trace_active),
+    retractall(max_trace_depth(_)),
     clear_trace.
 
 %% clear_trace/0
@@ -26,14 +34,47 @@ clear_trace :-
 %  Hook predicate that intercepts trace events
 user:prolog_trace_interception(Port, Frame, _Choice, continue) :-
     trace_active,
+    % Check depth limit
+    prolog_frame_attribute(Frame, level, Level),
+    max_trace_depth(MaxDepth),
+    Level =< MaxDepth,
+    % Don't trace our own operations
+    prolog_frame_attribute(Frame, goal, Goal),
+    \+ is_tracer_goal(Goal),
     !,
     catch(
         capture_trace_event(Port, Frame),
         Error,
         handle_trace_error(Error, Port, Frame)
     ).
+user:prolog_trace_interception(Port, Frame, _Choice, continue) :-
+    trace_active,
+    % Exceeded depth limit - record truncation marker once
+    prolog_frame_attribute(Frame, level, Level),
+    max_trace_depth(MaxDepth),
+    Level > MaxDepth,
+    \+ trace_event(truncated(MaxDepth)),
+    !,
+    assertz(trace_event(truncated(MaxDepth))).
 user:prolog_trace_interception(_, _, _, continue).
 
+%% is_tracer_goal(+Goal)
+%  Check if a goal is part of the tracer infrastructure
+is_tracer_goal(Goal) :-
+    functor(Goal, Functor, _),
+    tracer_predicate(Functor).
+
+tracer_predicate(trace_event).
+tracer_predicate(call_goal).
+tracer_predicate(assertz).
+tracer_predicate(retract).
+tracer_predicate(retractall).
+tracer_predicate(findall).
+tracer_predicate(format).
+tracer_predicate(write).
+tracer_predicate(open).
+tracer_predicate(close).
+
 %% handle_trace_error(+Error, +Port, +Frame)
 %  Handle errors during trace event capture
 handle_trace_error(Error, Port, Frame) :-
@@ -61,11 +102,14 @@ capture_trace_event(Port, Frame) :-
     ;   Arguments = []
     ),
     
-    % Extract clause information
-    extract_clause_info(Frame, Goal, ClauseInfo),
+    % Extract clause information with enhanced data
+    extract_clause_info_enhanced(Frame, Goal, Port, ClauseInfo),
     
-    % Record the event
-    assertz(trace_event(event(Port, Level, Goal, Arguments, ClauseInfo, Predicate))).
+    % Extract parent information
+    extract_parent_info(Frame, ParentInfo),
+    
+    % Record the event with enhanced information
+    assertz(trace_event(event(Port, Level, Goal, Arguments, ClauseInfo, Predicate, ParentInfo))).
 
 %% extract_frame_arguments(+Frame, +Arity, -Arguments)
 %  Extract actual argument values from a frame
@@ -94,6 +138,27 @@ extract_clause_info(Frame, Goal, clause(Head, Body, Line)) :-
     !.
 extract_clause_info(_, _, no_clause).
 
+%% extract_clause_info_enhanced(+Frame, +Goal, +Port, -ClauseInfo)
+%  Extract clause information - keep it simple, TypeScript will handle variable names
+extract_clause_info_enhanced(Frame, Goal, _, ClauseInfo) :-
+    extract_clause_info(Frame, Goal, ClauseInfo).
+
+%% extract_parent_info(+Frame, -ParentInfo)
+%  Extract information about the parent frame
+extract_parent_info(Frame, parent(ParentLevel, ParentGoal)) :-
+    catch(
+        (
+            prolog_frame_attribute(Frame, parent, ParentFrame),
+            ParentFrame \= 0,
+            prolog_frame_attribute(ParentFrame, level, ParentLevel),
+            prolog_frame_attribute(ParentFrame, goal, ParentGoal)
+        ),
+        _,
+        fail
+    ),
+    !.
+extract_parent_info(_, no_parent).
+
 %% export_trace_json(+File)
 %  Export all trace events as JSON
 export_trace_json(File) :-
@@ -115,16 +180,24 @@ write_events_list(_, []).
 write_events_list(Stream, [Event]) :-
     !,
     write(Stream, '  '),
-    write_json_event(Stream, Event).
+    write_json_event_or_marker(Stream, Event).
 write_events_list(Stream, [Event|Rest]) :-
     write(Stream, '  '),
-    write_json_event(Stream, Event),
+    write_json_event_or_marker(Stream, Event),
     write(Stream, ',\n'),
     write_events_list(Stream, Rest).
 
+%% write_json_event_or_marker(+Stream, +EventOrMarker)
+%  Write either a regular event or a truncation marker
+write_json_event_or_marker(Stream, truncated(MaxDepth)) :-
+    !,
+    format(Stream, '{"truncated": true, "max_depth": ~w}', [MaxDepth]).
+write_json_event_or_marker(Stream, Event) :-
+    write_json_event(Stream, Event).
+
 %% write_json_event(+Stream, +Event)
 %  Write a single event as JSON object
-write_json_event(Stream, event(Port, Level, Goal, Arguments, ClauseInfo, Predicate)) :-
+write_json_event(Stream, event(Port, Level, Goal, Arguments, ClauseInfo, Predicate, ParentInfo)) :-
     format(Stream, '{', []),
     format(Stream, '"port": "~w"', [Port]),
     format(Stream, ', "level": ~w', [Level]),
@@ -140,18 +213,56 @@ write_json_event(Stream, event(Port, Level, Goal, Arguments, ClauseInfo, Predica
     ),
     
     % Write clause info if present
-    (   ClauseInfo = clause(Head, Body, Line)
-    ->  format(Stream, ', "clause": {', []),
-        format(Stream, '"head": ', []),
-        write_json_term(Stream, Head),
-        format(Stream, ', "body": ', []),
-        write_json_term(Stream, Body),
-        format(Stream, ', "line": ~w', [Line]),
-        format(Stream, '}', [])
+    write_clause_info_json(Stream, ClauseInfo),
+    
+    % Write parent info if present
+    write_parent_info_json(Stream, ParentInfo),
+    
+    format(Stream, '}', []).
+% Backward compatibility: handle old event format without ParentInfo
+write_json_event(Stream, event(Port, Level, Goal, Arguments, ClauseInfo, Predicate)) :-
+    format(Stream, '{', []),
+    format(Stream, '"port": "~w"', [Port]),
+    format(Stream, ', "level": ~w', [Level]),
+    format(Stream, ', "goal": ', []),
+    write_json_term(Stream, Goal),
+    format(Stream, ', "predicate": "~w"', [Predicate]),
+    
+    % Write arguments if present (exit port)
+    (   Arguments \= []
+    ->  format(Stream, ', "arguments": ', []),
+        write_json_list(Stream, Arguments)
     ;   true
     ),
     
+    % Write clause info if present
+    write_clause_info_json(Stream, ClauseInfo),
+    
+    format(Stream, '}', []).
+
+%% write_clause_info_json(+Stream, +ClauseInfo)
+%  Write clause information as JSON
+write_clause_info_json(Stream, clause(Head, Body, Line)) :-
+    !,
+    format(Stream, ', "clause": {', []),
+    format(Stream, '"head": ', []),
+    write_json_term(Stream, Head),
+    format(Stream, ', "body": ', []),
+    write_json_term(Stream, Body),
+    format(Stream, ', "line": ~w', [Line]),
+    format(Stream, '}', []).
+write_clause_info_json(_, no_clause).
+
+%% write_parent_info_json(+Stream, +ParentInfo)
+%  Write parent information as JSON
+write_parent_info_json(Stream, parent(ParentLevel, ParentGoal)) :-
+    !,
+    format(Stream, ', "parent_info": {', []),
+    format(Stream, '"level": ~w', [ParentLevel]),
+    format(Stream, ', "goal": ', []),
+    write_json_term(Stream, ParentGoal),
     format(Stream, '}', []).
+write_parent_info_json(_, no_parent).
 
 %% write_json_term(+Stream, +Term)
 %  Write a Prolog term as JSON string