Chris@0: <?php
Chris@0: 
Chris@0: /*
Chris@0:  * This file is part of Psy Shell.
Chris@0:  *
Chris@0:  * (c) 2012-2018 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\Command;
Chris@0: 
Chris@0: use PhpParser\NodeTraverser;
Chris@0: use PhpParser\PrettyPrinter\Standard as Printer;
Chris@0: use Psy\Input\CodeArgument;
Chris@0: use Psy\ParserFactory;
Chris@0: use Psy\Readline\Readline;
Chris@0: use Psy\Sudo\SudoVisitor;
Chris@0: use Symfony\Component\Console\Input\InputInterface;
Chris@0: use Symfony\Component\Console\Output\OutputInterface;
Chris@0: 
Chris@0: /**
Chris@0:  * Evaluate PHP code, bypassing visibility restrictions.
Chris@0:  */
Chris@0: class SudoCommand extends Command
Chris@0: {
Chris@0:     private $readline;
Chris@0:     private $parser;
Chris@0:     private $traverser;
Chris@0:     private $printer;
Chris@0: 
Chris@0:     /**
Chris@0:      * {@inheritdoc}
Chris@0:      */
Chris@0:     public function __construct($name = null)
Chris@0:     {
Chris@0:         $parserFactory = new ParserFactory();
Chris@0:         $this->parser = $parserFactory->createParser();
Chris@0: 
Chris@0:         $this->traverser = new NodeTraverser();
Chris@0:         $this->traverser->addVisitor(new SudoVisitor());
Chris@0: 
Chris@0:         $this->printer = new Printer();
Chris@0: 
Chris@0:         parent::__construct($name);
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * Set the Shell's Readline service.
Chris@0:      *
Chris@0:      * @param Readline $readline
Chris@0:      */
Chris@0:     public function setReadline(Readline $readline)
Chris@0:     {
Chris@0:         $this->readline = $readline;
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * {@inheritdoc}
Chris@0:      */
Chris@0:     protected function configure()
Chris@0:     {
Chris@0:         $this
Chris@0:             ->setName('sudo')
Chris@0:             ->setDefinition([
Chris@0:                 new CodeArgument('code', CodeArgument::REQUIRED, 'Code to execute.'),
Chris@0:             ])
Chris@0:             ->setDescription('Evaluate PHP code, bypassing visibility restrictions.')
Chris@0:             ->setHelp(
Chris@0:                 <<<'HELP'
Chris@0: Evaluate PHP code, bypassing visibility restrictions.
Chris@0: 
Chris@0: e.g.
Chris@0: <return>>>> $sekret->whisper("hi")</return>
Chris@0: <return>PHP error:  Call to private method Sekret::whisper() from context '' on line 1</return>
Chris@0: 
Chris@0: <return>>>> sudo $sekret->whisper("hi")</return>
Chris@0: <return>=> "hi"</return>
Chris@0: 
Chris@0: <return>>>> $sekret->word</return>
Chris@0: <return>PHP error:  Cannot access private property Sekret::$word on line 1</return>
Chris@0: 
Chris@0: <return>>>> sudo $sekret->word</return>
Chris@0: <return>=> "hi"</return>
Chris@0: 
Chris@0: <return>>>> $sekret->word = "please"</return>
Chris@0: <return>PHP error:  Cannot access private property Sekret::$word on line 1</return>
Chris@0: 
Chris@0: <return>>>> sudo $sekret->word = "please"</return>
Chris@0: <return>=> "please"</return>
Chris@0: HELP
Chris@0:             );
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * {@inheritdoc}
Chris@0:      */
Chris@0:     protected function execute(InputInterface $input, OutputInterface $output)
Chris@0:     {
Chris@0:         $code = $input->getArgument('code');
Chris@0: 
Chris@0:         // special case for !!
Chris@0:         if ($code === '!!') {
Chris@0:             $history = $this->readline->listHistory();
Chris@4:             if (\count($history) < 2) {
Chris@0:                 throw new \InvalidArgumentException('No previous command to replay');
Chris@0:             }
Chris@4:             $code = $history[\count($history) - 2];
Chris@0:         }
Chris@0: 
Chris@4:         if (\strpos('<?', $code) === false) {
Chris@0:             $code = '<?php ' . $code;
Chris@0:         }
Chris@0: 
Chris@0:         $nodes = $this->traverser->traverse($this->parse($code));
Chris@0: 
Chris@0:         $sudoCode = $this->printer->prettyPrint($nodes);
Chris@0:         $shell = $this->getApplication();
Chris@0:         $shell->addCode($sudoCode, !$shell->hasCode());
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * Lex and parse a string of code into statements.
Chris@0:      *
Chris@0:      * @param string $code
Chris@0:      *
Chris@0:      * @return array Statements
Chris@0:      */
Chris@0:     private function parse($code)
Chris@0:     {
Chris@0:         try {
Chris@0:             return $this->parser->parse($code);
Chris@0:         } catch (\PhpParser\Error $e) {
Chris@4:             if (\strpos($e->getMessage(), 'unexpected EOF') === false) {
Chris@0:                 throw $e;
Chris@0:             }
Chris@0: 
Chris@0:             // If we got an unexpected EOF, let's try it again with a semicolon.
Chris@0:             return $this->parser->parse($code . ';');
Chris@0:         }
Chris@0:     }
Chris@0: }