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\Command;
Chris@13: 
Chris@13: use PhpParser\Node\Arg;
Chris@16: use PhpParser\Node\Expr\New_;
Chris@13: use PhpParser\Node\Expr\StaticCall;
Chris@13: use PhpParser\Node\Expr\Variable;
Chris@13: use PhpParser\Node\Name\FullyQualified as FullyQualifiedName;
Chris@16: use PhpParser\Node\Scalar\String_;
Chris@13: use PhpParser\Node\Stmt\Throw_;
Chris@13: use PhpParser\PrettyPrinter\Standard as Printer;
Chris@13: use Psy\Context;
Chris@13: use Psy\ContextAware;
Chris@13: use Psy\Input\CodeArgument;
Chris@13: use Psy\ParserFactory;
Chris@13: use Symfony\Component\Console\Input\InputInterface;
Chris@13: use Symfony\Component\Console\Output\OutputInterface;
Chris@13: 
Chris@13: /**
Chris@13:  * Throw an exception or error out of the Psy Shell.
Chris@13:  */
Chris@13: class ThrowUpCommand extends Command implements ContextAware
Chris@13: {
Chris@13:     const THROW_CLASS = 'Psy\Exception\ThrowUpException';
Chris@13: 
Chris@13:     private $parser;
Chris@13:     private $printer;
Chris@13: 
Chris@13:     /**
Chris@13:      * Context instance (for ContextAware interface).
Chris@13:      *
Chris@13:      * @var Context
Chris@13:      */
Chris@13:     protected $context;
Chris@13: 
Chris@13:     /**
Chris@13:      * {@inheritdoc}
Chris@13:      */
Chris@13:     public function __construct($name = null)
Chris@13:     {
Chris@13:         $parserFactory = new ParserFactory();
Chris@13: 
Chris@13:         $this->parser  = $parserFactory->createParser();
Chris@13:         $this->printer = new Printer();
Chris@13: 
Chris@13:         parent::__construct($name);
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * ContextAware interface.
Chris@13:      *
Chris@13:      * @param Context $context
Chris@13:      */
Chris@13:     public function setContext(Context $context)
Chris@13:     {
Chris@13:         $this->context = $context;
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * {@inheritdoc}
Chris@13:      */
Chris@13:     protected function configure()
Chris@13:     {
Chris@13:         $this
Chris@13:             ->setName('throw-up')
Chris@13:             ->setDefinition([
Chris@13:                 new CodeArgument('exception', CodeArgument::OPTIONAL, 'Exception or Error to throw.'),
Chris@13:             ])
Chris@13:             ->setDescription('Throw an exception or error out of the Psy Shell.')
Chris@13:             ->setHelp(
Chris@13:                 <<<'HELP'
Chris@13: Throws an exception or error out of the current the Psy Shell instance.
Chris@13: 
Chris@13: By default it throws the most recent exception.
Chris@13: 
Chris@13: e.g.
Chris@13: <return>>>> throw-up</return>
Chris@13: <return>>>> throw-up $e</return>
Chris@13: <return>>>> throw-up new Exception('WHEEEEEE!')</return>
Chris@16: <return>>>> throw-up "bye!"</return>
Chris@13: HELP
Chris@13:             );
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * {@inheritdoc}
Chris@13:      *
Chris@13:      * @throws InvalidArgumentException if there is no exception to throw
Chris@13:      */
Chris@13:     protected function execute(InputInterface $input, OutputInterface $output)
Chris@13:     {
Chris@13:         $args = $this->prepareArgs($input->getArgument('exception'));
Chris@13:         $throwStmt = new Throw_(new StaticCall(new FullyQualifiedName(self::THROW_CLASS), 'fromThrowable', $args));
Chris@13:         $throwCode = $this->printer->prettyPrint([$throwStmt]);
Chris@13: 
Chris@13:         $shell = $this->getApplication();
Chris@13:         $shell->addCode($throwCode, !$shell->hasCode());
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * Parse the supplied command argument.
Chris@13:      *
Chris@13:      * If no argument was given, this falls back to `$_e`
Chris@13:      *
Chris@13:      * @throws InvalidArgumentException if there is no exception to throw
Chris@13:      *
Chris@13:      * @param string $code
Chris@13:      *
Chris@13:      * @return Arg[]
Chris@13:      */
Chris@13:     private function prepareArgs($code = null)
Chris@13:     {
Chris@13:         if (!$code) {
Chris@13:             // Default to last exception if nothing else was supplied
Chris@13:             return [new Arg(new Variable('_e'))];
Chris@13:         }
Chris@13: 
Chris@17:         if (\strpos('<?', $code) === false) {
Chris@13:             $code = '<?php ' . $code;
Chris@13:         }
Chris@13: 
Chris@16:         $nodes = $this->parse($code);
Chris@17:         if (\count($nodes) !== 1) {
Chris@13:             throw new \InvalidArgumentException('No idea how to throw this');
Chris@13:         }
Chris@13: 
Chris@16:         $node = $nodes[0];
Chris@16: 
Chris@16:         // Make this work for PHP Parser v3.x
Chris@16:         $expr = isset($node->expr) ? $node->expr : $node;
Chris@16: 
Chris@16:         $args = [new Arg($expr, false, false, $node->getAttributes())];
Chris@16: 
Chris@16:         // Allow throwing via a string, e.g. `throw-up "SUP"`
Chris@16:         if ($expr instanceof String_) {
Chris@16:             return [new New_(new FullyQualifiedName('Exception'), $args)];
Chris@16:         }
Chris@16: 
Chris@16:         return $args;
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * Lex and parse a string of code into statements.
Chris@13:      *
Chris@13:      * @param string $code
Chris@13:      *
Chris@13:      * @return array Statements
Chris@13:      */
Chris@13:     private function parse($code)
Chris@13:     {
Chris@13:         try {
Chris@13:             return $this->parser->parse($code);
Chris@13:         } catch (\PhpParser\Error $e) {
Chris@17:             if (\strpos($e->getMessage(), 'unexpected EOF') === false) {
Chris@13:                 throw $e;
Chris@13:             }
Chris@13: 
Chris@13:             // If we got an unexpected EOF, let's try it again with a semicolon.
Chris@13:             return $this->parser->parse($code . ';');
Chris@13:         }
Chris@13:     }
Chris@13: }