prolog-trace-viz 1.1.2 → 2.0.0

package/dist/tree-formatter.js

@@ -0,0 +1,136 @@
+/**
+ * Tree Formatter - Generates Mermaid diagram from tree structure
+ */
+/**
+ * Format tree as Mermaid diagram
+ */
+export function formatTreeAsMermaid(root) {
+    if (!root) {
+        return 'graph TD\n  A["No execution tree"]';
+    }
+    const lines = [];
+    lines.push('graph TD');
+    lines.push('');
+    // Generate nodes
+    const nodes = [];
+    const edges = [];
+    const styles = [];
+    collectNodesAndEdges(root, nodes, edges, styles);
+    // Add nodes section
+    lines.push('%% Nodes');
+    lines.push(...nodes);
+    lines.push('');
+    // Add edges section
+    lines.push('%% Edges');
+    lines.push(...edges);
+    lines.push('');
+    // Add styles section
+    lines.push('%% Styles');
+    lines.push(...styles);
+    return lines.join('\n');
+}
+/**
+ * Collect nodes, edges, and styles recursively
+ */
+function collectNodesAndEdges(node, nodes, edges, styles, isRoot = true) {
+    // Generate node definition
+    const nodeLabel = formatNodeLabel(node);
+    nodes.push(`${node.id}[${nodeLabel}]`);
+    // Generate style
+    const style = getNodeStyle(node, isRoot);
+    if (style) {
+        styles.push(style);
+    }
+    // Generate edges to children
+    for (let i = 0; i < node.children.length; i++) {
+        const child = node.children[i];
+        // Use actual subgoal content if available, otherwise generic label
+        let edgeLabel = `subgoal ${i + 1}`;
+        if (node.subgoals && i < node.subgoals.length) {
+            edgeLabel = node.subgoals[i].goal;
+        }
+        edges.push(`${node.id} -->|"${edgeLabel}"| ${child.id}`);
+        // Recursively process child
+        collectNodesAndEdges(child, nodes, edges, styles, false);
+    }
+}
+/**
+ * Format node label with circled numbers and clause info
+ */
+function formatNodeLabel(node) {
+    const parts = [];
+    // Use source clause head if available, otherwise use goal
+    const displayGoal = node.clauseHead || node.goal;
+    // Add call step with circled number
+    parts.push(`${toCircledNumber(node.callStep)} ${displayGoal}`);
+    // Add clause number if available
+    if (node.clauseNumber) {
+        parts.push(`clause ${node.clauseNumber}`);
+    }
+    // Add exit step and binding if available
+    if (node.exitStep) {
+        parts.push(`${toCircledNumber(node.exitStep)} EXIT${node.finalBinding ? ': ' + node.finalBinding : ''}`);
+    }
+    return `"${parts.join('<br/>')}"`;
+}
+/**
+ * Convert number to circled Unicode character
+ */
+function toCircledNumber(n) {
+    if (n >= 1 && n <= 20) {
+        // Unicode circled numbers 1-20
+        return String.fromCharCode(0x2460 + n - 1);
+    }
+    else if (n >= 21 && n <= 35) {
+        // Unicode circled numbers 21-35
+        return String.fromCharCode(0x3251 + n - 21);
+    }
+    else if (n >= 36 && n <= 50) {
+        // Unicode circled numbers 36-50
+        return String.fromCharCode(0x32B1 + n - 36);
+    }
+    else {
+        // Fallback for numbers > 50
+        return `(${n})`;
+    }
+}
+/**
+ * Get Mermaid style for node based on status
+ */
+function getNodeStyle(node, isRoot) {
+    if (isRoot) {
+        // Blue for root query
+        return `style ${node.id} fill:#e1f5ff,stroke:#01579b,stroke-width:3px`;
+    }
+    switch (node.status) {
+        case 'success':
+            // Green for success
+            return `style ${node.id} fill:#c8e6c9,stroke:#388e3c`;
+        case 'failure':
+            // Red for failure
+            return `style ${node.id} fill:#ffcdd2,stroke:#c62828`;
+        case 'pending':
+            // Yellow for pending
+            return `style ${node.id} fill:#fff9c4,stroke:#f57f17`;
+        default:
+            return null;
+    }
+}
+/**
+ * Handle backtracking visualization in tree
+ */
+export function markBacktrackingPaths(root) {
+    // Walk the tree and mark failed attempts
+    function markNode(node) {
+        // If this node failed, mark it
+        if (node.status === 'failure') {
+            // Already marked by status
+        }
+        // Process children
+        for (const child of node.children) {
+            markNode(child);
+        }
+    }
+    markNode(root);
+}
+//# sourceMappingURL=tree-formatter.js.map

package/dist/tree-formatter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tree-formatter.js","sourceRoot":"","sources":["../src/tree-formatter.ts"],"names":[],"mappings":"AAAA;;GAEG;AAIH;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,IAAqB;IACvD,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,oCAAoC,CAAC;IAC9C,CAAC;IAED,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACvB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,iBAAiB;IACjB,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,MAAM,MAAM,GAAa,EAAE,CAAC;IAE5B,oBAAoB,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;IAEjD,oBAAoB;IACpB,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACvB,KAAK,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;IACrB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,oBAAoB;IACpB,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACvB,KAAK,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;IACrB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,qBAAqB;IACrB,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACxB,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC;IAEtB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED;;GAEG;AACH,SAAS,oBAAoB,CAC3B,IAAc,EACd,KAAe,EACf,KAAe,EACf,MAAgB,EAChB,SAAkB,IAAI;IAEtB,2BAA2B;IAC3B,MAAM,SAAS,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;IACxC,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,EAAE,IAAI,SAAS,GAAG,CAAC,CAAC;IAEvC,iBAAiB;IACjB,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACzC,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrB,CAAC;IAED,6BAA6B;IAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAE/B,mEAAmE;QACnE,IAAI,SAAS,GAAG,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC;QACnC,IAAI,IAAI,CAAC,QAAQ,IAAI,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YAC9C,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACpC,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,EAAE,SAAS,SAAS,MAAM,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;QAEzD,4BAA4B;QAC5B,oBAAoB,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;IAC3D,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,IAAc;IACrC,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,0DAA0D;IAC1D,MAAM,WAAW,GAAG,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,IAAI,CAAC;IAEjD,oCAAoC;IACpC,KAAK,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,WAAW,EAAE,CAAC,CAAC;IAE/D,iCAAiC;IACjC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;QACtB,KAAK,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;IAC5C,CAAC;IAED,yCAAyC;IACzC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;QAClB,KAAK,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAC3G,CAAC;IAED,OAAO,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC;AACpC,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,CAAS;IAChC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;QACtB,+BAA+B;QAC/B,OAAO,MAAM,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IAC7C,CAAC;SAAM,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;QAC9B,gCAAgC;QAChC,OAAO,MAAM,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC;IAC9C,CAAC;SAAM,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;QAC9B,gCAAgC;QAChC,OAAO,MAAM,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC;IAC9C,CAAC;SAAM,CAAC;QACN,4BAA4B;QAC5B,OAAO,IAAI,CAAC,GAAG,CAAC;IAClB,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,YAAY,CAAC,IAAc,EAAE,MAAe;IACnD,IAAI,MAAM,EAAE,CAAC;QACX,sBAAsB;QACtB,OAAO,SAAS,IAAI,CAAC,EAAE,+CAA+C,CAAC;IACzE,CAAC;IAED,QAAQ,IAAI,CAAC,MAAM,EAAE,CAAC;QACpB,KAAK,SAAS;YACZ,oBAAoB;YACpB,OAAO,SAAS,IAAI,CAAC,EAAE,8BAA8B,CAAC;QACxD,KAAK,SAAS;YACZ,kBAAkB;YAClB,OAAO,SAAS,IAAI,CAAC,EAAE,8BAA8B,CAAC;QACxD,KAAK,SAAS;YACZ,qBAAqB;YACrB,OAAO,SAAS,IAAI,CAAC,EAAE,8BAA8B,CAAC;QACxD;YACE,OAAO,IAAI,CAAC;IAChB,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,IAAc;IAClD,yCAAyC;IACzC,SAAS,QAAQ,CAAC,IAAc;QAC9B,+BAA+B;QAC/B,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC9B,2BAA2B;QAC7B,CAAC;QAED,mBAAmB;QACnB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,IAAI,CAAC,CAAC;AACjB,CAAC"}

package/dist/tree.d.ts

@@ -0,0 +1,75 @@
+/**
+ * Tree Builder - Constructs hierarchical call tree from trace events
+ */
+import { TraceEvent } from './timeline.js';
+import { SourceClauseMap } from './clauses.js';
+export interface TreeNode {
+    id: string;
+    goal: string;
+    clauseHead?: string;
+    clauseNumber?: string;
+    callStep: number;
+    exitStep?: number;
+    status: 'success' | 'failure' | 'pending';
+    children: TreeNode[];
+    finalBinding?: string;
+    subgoals?: Array<{
+        label: string;
+        goal: string;
+    }>;
+}
+/**
+ * Tree Builder class - constructs call tree from trace events
+ */
+export declare class TreeBuilder {
+    private events;
+    private sourceClauseMap?;
+    private nodes;
+    private nodeCounter;
+    private callStack;
+    private stepToNode;
+    private stepCounter;
+    constructor(events: TraceEvent[], sourceClauseMap?: SourceClauseMap | undefined);
+    /**
+     * Check if a predicate is part of tracer infrastructure and should be filtered
+     */
+    private isTracerPredicate;
+    /**
+     * Build the complete tree from trace events
+     */
+    build(): TreeNode | null;
+    /**
+     * Backfill clause information from EXIT events to nodes
+     * CALL events don't have clause info, only EXIT events do
+     */
+    private backfillClauseInfo;
+    /**
+     * Generate node ID (A-Z, AA-AZ, BA-BZ, ...)
+     */
+    private generateNodeId;
+    /**
+     * Process CALL event - create new node
+     */
+    private processCall;
+    /**
+     * Extract subgoals from a clause body
+     */
+    private extractSubgoals;
+    /**
+     * Process EXIT event - mark node as successful
+     */
+    private processExit;
+    /**
+     * Process FAIL event - mark node as failed
+     */
+    private processFail;
+    /**
+     * Extract binding by comparing CALL and EXIT goals
+     */
+    private extractBinding;
+    /**
+     * Split arguments respecting parentheses and brackets
+     */
+    private splitArguments;
+}
+//# sourceMappingURL=tree.d.ts.map

package/dist/tree.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tree.d.ts","sourceRoot":"","sources":["../src/tree.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAC3C,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAE/C,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,SAAS,GAAG,SAAS,GAAG,SAAS,CAAC;IAC1C,QAAQ,EAAE,QAAQ,EAAE,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,KAAK,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACnD;AAED;;GAEG;AACH,qBAAa,WAAW;IAOV,OAAO,CAAC,MAAM;IAAgB,OAAO,CAAC,eAAe,CAAC;IANlE,OAAO,CAAC,KAAK,CAAkB;IAC/B,OAAO,CAAC,WAAW,CAAK;IACxB,OAAO,CAAC,SAAS,CAAoC;IACrD,OAAO,CAAC,UAAU,CAAoC;IACtD,OAAO,CAAC,WAAW,CAAK;gBAEJ,MAAM,EAAE,UAAU,EAAE,EAAU,eAAe,CAAC,EAAE,eAAe,YAAA;IAEnF;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAWzB;;OAEG;IACH,KAAK,IAAI,QAAQ,GAAG,IAAI;IAsCxB;;;OAGG;IACH,OAAO,CAAC,kBAAkB;IAsB1B;;OAEG;IACH,OAAO,CAAC,cAAc;IAatB;;OAEG;IACH,OAAO,CAAC,WAAW;IA+CnB;;OAEG;IACH,OAAO,CAAC,eAAe;IAgCvB;;OAEG;IACH,OAAO,CAAC,WAAW;IAwBnB;;OAEG;IACH,OAAO,CAAC,WAAW;IAWnB;;OAEG;IACH,OAAO,CAAC,cAAc;IAyBtB;;OAEG;IACH,OAAO,CAAC,cAAc;CA0BvB"}

package/dist/tree.js

@@ -0,0 +1,267 @@
+/**
+ * Tree Builder - Constructs hierarchical call tree from trace events
+ */
+/**
+ * Tree Builder class - constructs call tree from trace events
+ */
+export class TreeBuilder {
+    events;
+    sourceClauseMap;
+    nodes = [];
+    nodeCounter = 0;
+    callStack = new Map(); // level -> node
+    stepToNode = new Map(); // step number -> node
+    stepCounter = 0;
+    constructor(events, sourceClauseMap) {
+        this.events = events;
+        this.sourceClauseMap = sourceClauseMap;
+    }
+    /**
+     * Check if a predicate is part of tracer infrastructure and should be filtered
+     */
+    isTracerPredicate(predicate) {
+        const tracerPredicates = [
+            'catch/3',
+            'export_trace_json/1',
+            'run_trace/0',
+            'install_tracer/1',
+            'remove_tracer/0',
+        ];
+        return tracerPredicates.includes(predicate);
+    }
+    /**
+     * Build the complete tree from trace events
+     */
+    build() {
+        let root = null;
+        for (const event of this.events) {
+            // Filter out tracer infrastructure
+            if (this.isTracerPredicate(event.predicate)) {
+                continue;
+            }
+            this.stepCounter++;
+            switch (event.port) {
+                case 'call':
+                    const node = this.processCall(event, this.stepCounter);
+                    if (!root) {
+                        root = node;
+                    }
+                    break;
+                case 'exit':
+                    this.processExit(event, this.stepCounter);
+                    break;
+                case 'redo':
+                    // REDO doesn't create new nodes, just marks retry
+                    break;
+                case 'fail':
+                    this.processFail(event, this.stepCounter);
+                    break;
+            }
+        }
+        // Backfill clause info from EXIT events to nodes
+        if (root) {
+            this.backfillClauseInfo();
+        }
+        return root;
+    }
+    /**
+     * Backfill clause information from EXIT events to nodes
+     * CALL events don't have clause info, only EXIT events do
+     */
+    backfillClauseInfo() {
+        // For each node, find its EXIT event to get clause info
+        for (const node of this.nodes) {
+            if (!node.clauseHead && node.clauseNumber && this.sourceClauseMap) {
+                const lineNumber = parseInt(node.clauseNumber);
+                const sourceClause = this.sourceClauseMap[lineNumber];
+                if (sourceClause) {
+                    node.clauseHead = sourceClause.head;
+                    // Extract subgoals from clause body
+                    if (sourceClause.body && sourceClause.body !== 'true') {
+                        const subgoalGoals = this.extractSubgoals(sourceClause.body);
+                        node.subgoals = subgoalGoals.map((goal, index) => ({
+                            label: `[${node.callStep}.${index + 1}]`,
+                            goal,
+                        }));
+                    }
+                }
+            }
+        }
+    }
+    /**
+     * Generate node ID (A-Z, AA-AZ, BA-BZ, ...)
+     */
+    generateNodeId() {
+        const index = this.nodeCounter++;
+        if (index < 26) {
+            return String.fromCharCode(65 + index); // A-Z
+        }
+        // After Z: AA, AB, AC, ..., AZ, BA, BB, ...
+        const firstChar = String.fromCharCode(65 + Math.floor(index / 26) - 1);
+        const secondChar = String.fromCharCode(65 + (index % 26));
+        return firstChar + secondChar;
+    }
+    /**
+     * Process CALL event - create new node
+     */
+    processCall(event, stepNumber) {
+        // Get source clause if available
+        let clauseHead;
+        let subgoals;
+        if (event.clause && this.sourceClauseMap) {
+            const sourceClause = this.sourceClauseMap[event.clause.line];
+            if (sourceClause) {
+                clauseHead = sourceClause.head;
+                // Extract subgoals from clause body
+                if (sourceClause.body && sourceClause.body !== 'true') {
+                    const subgoalGoals = this.extractSubgoals(sourceClause.body);
+                    subgoals = subgoalGoals.map((goal, index) => ({
+                        label: `[${stepNumber}.${index + 1}]`,
+                        goal,
+                    }));
+                }
+            }
+        }
+        const node = {
+            id: this.generateNodeId(),
+            goal: event.goal,
+            clauseHead,
+            clauseNumber: event.clause?.line.toString(),
+            callStep: stepNumber,
+            status: 'pending',
+            children: [],
+            subgoals,
+        };
+        // Add as child to parent if there is one
+        const parentLevel = event.level - 1;
+        const parent = this.callStack.get(parentLevel);
+        if (parent) {
+            parent.children.push(node);
+        }
+        // Track in call stack
+        this.callStack.set(event.level, node);
+        this.stepToNode.set(stepNumber, node);
+        this.nodes.push(node);
+        return node;
+    }
+    /**
+     * Extract subgoals from a clause body
+     */
+    extractSubgoals(clauseBody) {
+        if (!clauseBody || clauseBody === 'true') {
+            return [];
+        }
+        // Split on commas, respecting parentheses depth
+        const subgoals = [];
+        let current = '';
+        let depth = 0;
+        for (const char of clauseBody) {
+            if (char === '(') {
+                depth++;
+                current += char;
+            }
+            else if (char === ')') {
+                depth--;
+                current += char;
+            }
+            else if (char === ',' && depth === 0) {
+                subgoals.push(current.trim());
+                current = '';
+            }
+            else {
+                current += char;
+            }
+        }
+        if (current.trim()) {
+            subgoals.push(current.trim());
+        }
+        return subgoals;
+    }
+    /**
+     * Process EXIT event - mark node as successful
+     */
+    processExit(event, stepNumber) {
+        const node = this.callStack.get(event.level);
+        if (node) {
+            node.exitStep = stepNumber;
+            node.status = 'success';
+            // Store clause number from EXIT event
+            if (event.clause && !node.clauseNumber) {
+                node.clauseNumber = event.clause.line.toString();
+            }
+            // Extract final binding if available
+            // Compare CALL goal with EXIT goal to find bindings
+            const callGoal = node.goal;
+            const exitGoal = event.goal;
+            if (callGoal !== exitGoal) {
+                node.finalBinding = this.extractBinding(callGoal, exitGoal);
+            }
+            // Remove from call stack
+            this.callStack.delete(event.level);
+        }
+    }
+    /**
+     * Process FAIL event - mark node as failed
+     */
+    processFail(event, stepNumber) {
+        const node = this.callStack.get(event.level);
+        if (node) {
+            node.exitStep = stepNumber;
+            node.status = 'failure';
+            // Remove from call stack
+            this.callStack.delete(event.level);
+        }
+    }
+    /**
+     * Extract binding by comparing CALL and EXIT goals
+     */
+    extractBinding(callGoal, exitGoal) {
+        // Simple extraction - find the first variable that changed
+        const callMatch = callGoal.match(/^([^(]+)\((.*)\)$/);
+        const exitMatch = exitGoal.match(/^([^(]+)\((.*)\)$/);
+        if (!callMatch || !exitMatch) {
+            return undefined;
+        }
+        const callArgs = this.splitArguments(callMatch[2]);
+        const exitArgs = this.splitArguments(exitMatch[2]);
+        // Find first differing argument where call has variable
+        for (let i = 0; i < Math.min(callArgs.length, exitArgs.length); i++) {
+            const callArg = callArgs[i].trim();
+            const exitArg = exitArgs[i].trim();
+            if (callArg !== exitArg && /^[A-Z_]/.test(callArg)) {
+                return `${callArg}=${exitArg}`;
+            }
+        }
+        return undefined;
+    }
+    /**
+     * 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;
+    }
+}
+//# sourceMappingURL=tree.js.map

package/dist/tree.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tree.js","sourceRoot":"","sources":["../src/tree.ts"],"names":[],"mappings":"AAAA;;GAEG;AAkBH;;GAEG;AACH,MAAM,OAAO,WAAW;IAOF;IAA8B;IAN1C,KAAK,GAAe,EAAE,CAAC;IACvB,WAAW,GAAG,CAAC,CAAC;IAChB,SAAS,GAA0B,IAAI,GAAG,EAAE,CAAC,CAAC,gBAAgB;IAC9D,UAAU,GAA0B,IAAI,GAAG,EAAE,CAAC,CAAC,sBAAsB;IACrE,WAAW,GAAG,CAAC,CAAC;IAExB,YAAoB,MAAoB,EAAU,eAAiC;QAA/D,WAAM,GAAN,MAAM,CAAc;QAAU,oBAAe,GAAf,eAAe,CAAkB;IAAG,CAAC;IAEvF;;OAEG;IACK,iBAAiB,CAAC,SAAiB;QACzC,MAAM,gBAAgB,GAAG;YACvB,SAAS;YACT,qBAAqB;YACrB,aAAa;YACb,kBAAkB;YAClB,iBAAiB;SAClB,CAAC;QACF,OAAO,gBAAgB,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAC9C,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,IAAI,GAAoB,IAAI,CAAC;QAEjC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChC,mCAAmC;YACnC,IAAI,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC5C,SAAS;YACX,CAAC;YAED,IAAI,CAAC,WAAW,EAAE,CAAC;YAEnB,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACnB,KAAK,MAAM;oBACT,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;oBACvD,IAAI,CAAC,IAAI,EAAE,CAAC;wBACV,IAAI,GAAG,IAAI,CAAC;oBACd,CAAC;oBACD,MAAM;gBACR,KAAK,MAAM;oBACT,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;oBAC1C,MAAM;gBACR,KAAK,MAAM;oBACT,kDAAkD;oBAClD,MAAM;gBACR,KAAK,MAAM;oBACT,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;oBAC1C,MAAM;YACV,CAAC;QACH,CAAC;QAED,iDAAiD;QACjD,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC5B,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;OAGG;IACK,kBAAkB;QACxB,wDAAwD;QACxD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;gBAClE,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;gBACtD,IAAI,YAAY,EAAE,CAAC;oBACjB,IAAI,CAAC,UAAU,GAAG,YAAY,CAAC,IAAI,CAAC;oBAEpC,oCAAoC;oBACpC,IAAI,YAAY,CAAC,IAAI,IAAI,YAAY,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBACtD,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;wBAC7D,IAAI,CAAC,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;4BACjD,KAAK,EAAE,IAAI,IAAI,CAAC,QAAQ,IAAI,KAAK,GAAG,CAAC,GAAG;4BACxC,IAAI;yBACL,CAAC,CAAC,CAAC;oBACN,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,cAAc;QACpB,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAEjC,IAAI,KAAK,GAAG,EAAE,EAAE,CAAC;YACf,OAAO,MAAM,CAAC,YAAY,CAAC,EAAE,GAAG,KAAK,CAAC,CAAC,CAAC,MAAM;QAChD,CAAC;QAED,4CAA4C;QAC5C,MAAM,SAAS,GAAG,MAAM,CAAC,YAAY,CAAC,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;QACvE,MAAM,UAAU,GAAG,MAAM,CAAC,YAAY,CAAC,EAAE,GAAG,CAAC,KAAK,GAAG,EAAE,CAAC,CAAC,CAAC;QAC1D,OAAO,SAAS,GAAG,UAAU,CAAC;IAChC,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,KAAiB,EAAE,UAAkB;QACvD,iCAAiC;QACjC,IAAI,UAA8B,CAAC;QACnC,IAAI,QAA4D,CAAC;QAEjE,IAAI,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzC,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC7D,IAAI,YAAY,EAAE,CAAC;gBACjB,UAAU,GAAG,YAAY,CAAC,IAAI,CAAC;gBAE/B,oCAAoC;gBACpC,IAAI,YAAY,CAAC,IAAI,IAAI,YAAY,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACtD,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;oBAC7D,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;wBAC5C,KAAK,EAAE,IAAI,UAAU,IAAI,KAAK,GAAG,CAAC,GAAG;wBACrC,IAAI;qBACL,CAAC,CAAC,CAAC;gBACN,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,IAAI,GAAa;YACrB,EAAE,EAAE,IAAI,CAAC,cAAc,EAAE;YACzB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,UAAU;YACV,YAAY,EAAE,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE;YAC3C,QAAQ,EAAE,UAAU;YACpB,MAAM,EAAE,SAAS;YACjB,QAAQ,EAAE,EAAE;YACZ,QAAQ;SACT,CAAC;QAEF,yCAAyC;QACzC,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC;QACpC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAC/C,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7B,CAAC;QAED,sBAAsB;QACtB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACtC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;QACtC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEtB,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,UAAkB;QACxC,IAAI,CAAC,UAAU,IAAI,UAAU,KAAK,MAAM,EAAE,CAAC;YACzC,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,gDAAgD;QAChD,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;YAC9B,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACjB,KAAK,EAAE,CAAC;gBACR,OAAO,IAAI,IAAI,CAAC;YAClB,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACxB,KAAK,EAAE,CAAC;gBACR,OAAO,IAAI,IAAI,CAAC;YAClB,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;gBACvC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC9B,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,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,KAAiB,EAAE,UAAkB;QACvD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC7C,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,QAAQ,GAAG,UAAU,CAAC;YAC3B,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;YAExB,sCAAsC;YACtC,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;gBACvC,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnD,CAAC;YAED,qCAAqC;YACrC,oDAAoD;YACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;YAC3B,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC;YAC5B,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBAC1B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAC9D,CAAC;YAED,yBAAyB;YACzB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,KAAiB,EAAE,UAAkB;QACvD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC7C,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,QAAQ,GAAG,UAAU,CAAC;YAC3B,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;YAExB,yBAAyB;YACzB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,QAAgB,EAAE,QAAgB;QACvD,2DAA2D;QAC3D,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACtD,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;QAEtD,IAAI,CAAC,SAAS,IAAI,CAAC,SAAS,EAAE,CAAC;YAC7B,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;QAEnD,wDAAwD;QACxD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACpE,MAAM,OAAO,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YACnC,MAAM,OAAO,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAEnC,IAAI,OAAO,KAAK,OAAO,IAAI,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;gBACnD,OAAO,GAAG,OAAO,IAAI,OAAO,EAAE,CAAC;YACjC,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,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;CACF"}

package/dist/wrapper.d.ts

@@ -19,8 +19,15 @@ export interface TempFile {
  */
 export declare function generateWrapper(config: WrapperConfig): string;
 /**
- * Creates a temporary wrapper file and returns its path along with a cleanup function.
+ * Calculates the line offset between wrapper file and original source file.
+ * This is needed because the tracer reports line numbers from the wrapper file,
+ * but we need to map them back to the original source file line numbers.
  */
+export declare function calculateLineOffset(prologContent: string): number;
+/**
+ * Maps a line number from the wrapper file back to the original source file.
+ */
+export declare function mapWrapperLineToSource(wrapperLine: number, prologContent: string): number;
 export declare function createTempWrapper(config: WrapperConfig): Promise<TempFile>;
 /**
  * Serializes a wrapper back to its component parts (for testing round-trips).

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,CA0B7D;AAED;;GAEG;AACH,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,33 +11,57 @@ 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}'].`,
         '',
         `% User's Prolog code (no instrumentation)`,
-        prologContent.trim(),
-        '',
-        '% Run trace with error handling',
-        'run_trace :-',
-        '    install_tracer,',
-        '    catch(',
-        `        (${query.trim()}, export_trace_json('trace.json')),`,
-        '        Error,',
-        `        (format('Error: ~w~n', [Error]), export_trace_json('trace.json'))`,
-        '    ),',
-        '    remove_tracer.',
-        '',
-        ':- run_trace.',
-        ':- halt.',
-        '',
     ];
+    // Add user content and track line mapping
+    const userLines = prologContent.trim().split('\n');
+    const wrapperStartLine = lines.length + 1; // +1 for 1-based indexing
+    lines.push(...userLines);
+    lines.push('');
+    lines.push('% Run trace with error handling');
+    lines.push('run_trace :-');
+    lines.push(`    install_tracer(${maxDepth}),`);
+    lines.push('    catch(');
+    lines.push(`        (${query.trim()}, export_trace_json('trace.json')),`);
+    lines.push('        Error,');
+    lines.push(`        (format('Error: ~w~n', [Error]), export_trace_json('trace.json'))`);
+    lines.push('    ),');
+    lines.push('    remove_tracer.');
+    lines.push('');
+    lines.push(':- run_trace.');
+    lines.push(':- halt.');
+    lines.push('');
     return lines.join('\n');
 }
 /**
- * Creates a temporary wrapper file and returns its path along with a cleanup function.
+ * Calculates the line offset between wrapper file and original source file.
+ * This is needed because the tracer reports line numbers from the wrapper file,
+ * but we need to map them back to the original source file line numbers.
  */
+export function calculateLineOffset(prologContent) {
+    // The wrapper structure is:
+    // Line 1: % Load custom tracer
+    // Line 2: :- ['tracerPath'].
+    // Line 3: (empty)
+    // Line 4: % User's Prolog code (no instrumentation)
+    // Line 5+: User content starts here
+    return 4; // User content starts at line 5 in wrapper, so offset is 4
+}
+/**
+ * Maps a line number from the wrapper file back to the original source file.
+ */
+export function mapWrapperLineToSource(wrapperLine, prologContent) {
+    const offset = calculateLineOffset(prologContent);
+    // 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);
     const tempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'prolog-trace-viz-'));

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;QAC3C,aAAa,CAAC,IAAI,EAAE;QACpB,EAAE;QACF,iCAAiC;QACjC,cAAc;QACd,qBAAqB;QACrB,YAAY;QACZ,YAAY,KAAK,CAAC,IAAI,EAAE,qCAAqC;QAC7D,gBAAgB;QAChB,2EAA2E;QAC3E,QAAQ;QACR,oBAAoB;QACpB,EAAE;QACF,eAAe;QACf,UAAU;QACV,EAAE;KACH,CAAC;IAEF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED;;GAEG;AACH,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.2",
+  "version": "2.0.0",
   "description": "Generate enhanced visual trace diagrams for Prolog query execution",
   "type": "module",
   "main": "dist/index.js",