Chris@2: <?php
Chris@2: /*
Chris@2:  * This file is part of code-unit-reverse-lookup.
Chris@2:  *
Chris@2:  * (c) Sebastian Bergmann <sebastian@phpunit.de>
Chris@2:  *
Chris@2:  * For the full copyright and license information, please view the LICENSE
Chris@2:  * file that was distributed with this source code.
Chris@2:  */
Chris@2: 
Chris@2: namespace SebastianBergmann\CodeUnitReverseLookup;
Chris@2: 
Chris@2: /**
Chris@2:  * @since Class available since Release 1.0.0
Chris@2:  */
Chris@2: class Wizard
Chris@2: {
Chris@2:     /**
Chris@2:      * @var array
Chris@2:      */
Chris@2:     private $lookupTable = [];
Chris@2: 
Chris@2:     /**
Chris@2:      * @var array
Chris@2:      */
Chris@2:     private $processedClasses = [];
Chris@2: 
Chris@2:     /**
Chris@2:      * @var array
Chris@2:      */
Chris@2:     private $processedFunctions = [];
Chris@2: 
Chris@2:     /**
Chris@2:      * @param string $filename
Chris@2:      * @param int    $lineNumber
Chris@2:      *
Chris@2:      * @return string
Chris@2:      */
Chris@2:     public function lookup($filename, $lineNumber)
Chris@2:     {
Chris@2:         if (!isset($this->lookupTable[$filename][$lineNumber])) {
Chris@2:             $this->updateLookupTable();
Chris@2:         }
Chris@2: 
Chris@2:         if (isset($this->lookupTable[$filename][$lineNumber])) {
Chris@2:             return $this->lookupTable[$filename][$lineNumber];
Chris@2:         } else {
Chris@2:             return $filename . ':' . $lineNumber;
Chris@2:         }
Chris@2:     }
Chris@2: 
Chris@2:     private function updateLookupTable()
Chris@2:     {
Chris@2:         $this->processClassesAndTraits();
Chris@2:         $this->processFunctions();
Chris@2:     }
Chris@2: 
Chris@2:     private function processClassesAndTraits()
Chris@2:     {
Chris@2:         foreach (array_merge(get_declared_classes(), get_declared_traits()) as $classOrTrait) {
Chris@2:             if (isset($this->processedClasses[$classOrTrait])) {
Chris@2:                 continue;
Chris@2:             }
Chris@2: 
Chris@2:             $reflector = new \ReflectionClass($classOrTrait);
Chris@2: 
Chris@2:             foreach ($reflector->getMethods() as $method) {
Chris@2:                 $this->processFunctionOrMethod($method);
Chris@2:             }
Chris@2: 
Chris@2:             $this->processedClasses[$classOrTrait] = true;
Chris@2:         }
Chris@2:     }
Chris@2: 
Chris@2:     private function processFunctions()
Chris@2:     {
Chris@2:         foreach (get_defined_functions()['user'] as $function) {
Chris@2:             if (isset($this->processedFunctions[$function])) {
Chris@2:                 continue;
Chris@2:             }
Chris@2: 
Chris@2:             $this->processFunctionOrMethod(new \ReflectionFunction($function));
Chris@2: 
Chris@2:             $this->processedFunctions[$function] = true;
Chris@2:         }
Chris@2:     }
Chris@2: 
Chris@2:     /**
Chris@2:      * @param \ReflectionFunctionAbstract $functionOrMethod
Chris@2:      */
Chris@2:     private function processFunctionOrMethod(\ReflectionFunctionAbstract $functionOrMethod)
Chris@2:     {
Chris@2:         if ($functionOrMethod->isInternal()) {
Chris@2:             return;
Chris@2:         }
Chris@2: 
Chris@2:         $name = $functionOrMethod->getName();
Chris@2: 
Chris@2:         if ($functionOrMethod instanceof \ReflectionMethod) {
Chris@2:             $name = $functionOrMethod->getDeclaringClass()->getName() . '::' . $name;
Chris@2:         }
Chris@2: 
Chris@2:         if (!isset($this->lookupTable[$functionOrMethod->getFileName()])) {
Chris@2:             $this->lookupTable[$functionOrMethod->getFileName()] = [];
Chris@2:         }
Chris@2: 
Chris@2:         foreach (range($functionOrMethod->getStartLine(), $functionOrMethod->getEndLine()) as $line) {
Chris@2:             $this->lookupTable[$functionOrMethod->getFileName()][$line] = $name;
Chris@2:         }
Chris@2:     }
Chris@2: }