<?php 
namespace JmesPath; 
 
/** 
 * Provides CLI debugging information for the AST and Compiler runtimes. 
 */ 
class DebugRuntime 
{ 
    private $runtime; 
    private $out; 
    private $lexer; 
    private $parser; 
 
    public function __construct(callable $runtime, $output = null) 
    { 
        $this->runtime = $runtime; 
        $this->out = $output ?: STDOUT; 
        $this->lexer = new Lexer(); 
        $this->parser = new Parser($this->lexer); 
    } 
 
    public function __invoke($expression, $data) 
    { 
        if ($this->runtime instanceof CompilerRuntime) { 
            return $this->debugCompiled($expression, $data); 
        } 
 
        return $this->debugInterpreted($expression, $data); 
    } 
 
    private function debugInterpreted($expression, $data) 
    { 
        return $this->debugCallback( 
            function () use ($expression, $data) { 
                $runtime = $this->runtime; 
                return $runtime($expression, $data); 
            }, 
            $expression, 
            $data 
        ); 
    } 
 
    private function debugCompiled($expression, $data) 
    { 
        $result = $this->debugCallback( 
            function () use ($expression, $data) { 
                $runtime = $this->runtime; 
                return $runtime($expression, $data); 
            }, 
            $expression, 
            $data 
        ); 
        $this->dumpCompiledCode($expression); 
 
        return $result; 
    } 
 
    private function dumpTokens($expression) 
    { 
        $lexer = new Lexer(); 
        fwrite($this->out, "Tokens\n======\n\n"); 
        $tokens = $lexer->tokenize($expression); 
 
        foreach ($tokens as $t) { 
            fprintf( 
                $this->out, 
                "%3d  %-13s  %s\n", $t['pos'], $t['type'], 
                json_encode($t['value']) 
            ); 
        } 
 
        fwrite($this->out, "\n"); 
    } 
 
    private function dumpAst($expression) 
    { 
        $parser = new Parser(); 
        $ast = $parser->parse($expression); 
        fwrite($this->out, "AST\n========\n\n"); 
        fwrite($this->out, json_encode($ast, JSON_PRETTY_PRINT) . "\n"); 
    } 
 
    private function dumpCompiledCode($expression) 
    { 
        fwrite($this->out, "Code\n========\n\n"); 
        $dir = sys_get_temp_dir(); 
        $hash = md5($expression); 
        $functionName = "jmespath_{$hash}"; 
        $filename = "{$dir}/{$functionName}.php"; 
        fwrite($this->out, "File: {$filename}\n\n"); 
        fprintf($this->out, file_get_contents($filename)); 
    } 
 
    private function debugCallback(callable $debugFn, $expression, $data) 
    { 
        fprintf($this->out, "Expression\n==========\n\n%s\n\n", $expression); 
        $this->dumpTokens($expression); 
        $this->dumpAst($expression); 
        fprintf($this->out, "\nData\n====\n\n%s\n\n", json_encode($data, JSON_PRETTY_PRINT)); 
        $startTime = microtime(true); 
        $result = $debugFn(); 
        $total = microtime(true) - $startTime; 
        fprintf($this->out, "\nResult\n======\n\n%s\n\n", json_encode($result, JSON_PRETTY_PRINT)); 
        fwrite($this->out, "Time\n====\n\n"); 
        fprintf($this->out, "Total time:     %f ms\n\n", $total); 
 
        return $result; 
    } 
} 
 
 |