Chris@13: <?php
Chris@13: 
Chris@13: /*
Chris@13:  * This file is part of Psy Shell.
Chris@13:  *
Chris@13:  * (c) 2012-2018 Justin Hileman
Chris@13:  *
Chris@13:  * For the full copyright and license information, please view the LICENSE
Chris@13:  * file that was distributed with this source code.
Chris@13:  */
Chris@13: 
Chris@13: namespace Psy\CodeCleaner;
Chris@13: 
Chris@13: use PhpParser\Node;
Chris@13: use PhpParser\Node\Expr\FuncCall;
Chris@13: use PhpParser\Node\Name;
Chris@13: use PhpParser\Node\Scalar\MagicConst\Dir;
Chris@13: use PhpParser\Node\Scalar\MagicConst\File;
Chris@13: use PhpParser\Node\Scalar\String_;
Chris@13: 
Chris@13: /**
Chris@13:  * Swap out __DIR__ and __FILE__ magic constants with our best guess?
Chris@13:  */
Chris@13: class MagicConstantsPass extends CodeCleanerPass
Chris@13: {
Chris@13:     /**
Chris@13:      * Swap out __DIR__ and __FILE__ constants, because the default ones when
Chris@13:      * calling eval() don't make sense.
Chris@13:      *
Chris@13:      * @param Node $node
Chris@13:      *
Chris@13:      * @return null|FuncCall|String_
Chris@13:      */
Chris@13:     public function enterNode(Node $node)
Chris@13:     {
Chris@13:         if ($node instanceof Dir) {
Chris@13:             return new FuncCall(new Name('getcwd'), [], $node->getAttributes());
Chris@13:         } elseif ($node instanceof File) {
Chris@13:             return new String_('', $node->getAttributes());
Chris@13:         }
Chris@13:     }
Chris@13: }