Chris@0: <?php
Chris@0: 
Chris@0: /*
Chris@0:  * This file is part of Psy Shell.
Chris@0:  *
Chris@0:  * (c) 2012-2017 Justin Hileman
Chris@0:  *
Chris@0:  * For the full copyright and license information, please view the LICENSE
Chris@0:  * file that was distributed with this source code.
Chris@0:  */
Chris@0: 
Chris@0: namespace Psy\CodeCleaner;
Chris@0: 
Chris@0: use PhpParser\Node;
Chris@0: use PhpParser\Node\Stmt\Class_;
Chris@0: use Psy\Exception\FatalErrorException;
Chris@0: 
Chris@0: /**
Chris@0:  * The final class pass handles final classes.
Chris@0:  */
Chris@0: class FinalClassPass extends CodeCleanerPass
Chris@0: {
Chris@0:     private $finalClasses;
Chris@0: 
Chris@0:     /**
Chris@0:      * @param array $nodes
Chris@0:      */
Chris@0:     public function beforeTraverse(array $nodes)
Chris@0:     {
Chris@0:         $this->finalClasses = array();
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * @throws RuntimeException if the node is a class that extends a final class
Chris@0:      *
Chris@0:      * @param Node $node
Chris@0:      */
Chris@0:     public function enterNode(Node $node)
Chris@0:     {
Chris@0:         if ($node instanceof Class_) {
Chris@0:             if ($node->extends) {
Chris@0:                 $extends = (string) $node->extends;
Chris@0:                 if ($this->isFinalClass($extends)) {
Chris@0:                     $msg = sprintf('Class %s may not inherit from final class (%s)', $node->name, $extends);
Chris@0:                     throw new FatalErrorException($msg, 0, E_ERROR, null, $node->getLine());
Chris@0:                 }
Chris@0:             }
Chris@0: 
Chris@0:             if ($node->isFinal()) {
Chris@0:                 $this->finalClasses[strtolower($node->name)] = true;
Chris@0:             }
Chris@0:         }
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * @param string $name Class name
Chris@0:      *
Chris@0:      * @return bool
Chris@0:      */
Chris@0:     private function isFinalClass($name)
Chris@0:     {
Chris@0:         if (!class_exists($name)) {
Chris@0:             return isset($this->finalClasses[strtolower($name)]);
Chris@0:         }
Chris@0: 
Chris@0:         $refl = new \ReflectionClass($name);
Chris@0: 
Chris@0:         return $refl->isFinal();
Chris@0:     }
Chris@0: }