231 lines
		
	
	
		
			6.3 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			231 lines
		
	
	
		
			6.3 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
| <?php
 | |
| 
 | |
| /*
 | |
|  * This file is part of the Symfony package.
 | |
|  *
 | |
|  * (c) Fabien Potencier <fabien@symfony.com>
 | |
|  *
 | |
|  * For the full copyright and license information, please view the LICENSE
 | |
|  * file that was distributed with this source code.
 | |
|  */
 | |
| 
 | |
| namespace Symfony\Component\Console;
 | |
| 
 | |
| use Symfony\Component\Console\Application;
 | |
| use Symfony\Component\Console\Input\StringInput;
 | |
| use Symfony\Component\Console\Output\ConsoleOutput;
 | |
| use Symfony\Component\Process\ProcessBuilder;
 | |
| use Symfony\Component\Process\PhpExecutableFinder;
 | |
| 
 | |
| /**
 | |
|  * A Shell wraps an Application to add shell capabilities to it.
 | |
|  *
 | |
|  * Support for history and completion only works with a PHP compiled
 | |
|  * with readline support (either --with-readline or --with-libedit)
 | |
|  *
 | |
|  * @author Fabien Potencier <fabien@symfony.com>
 | |
|  * @author Martin Hasoň <martin.hason@gmail.com>
 | |
|  */
 | |
| class Shell
 | |
| {
 | |
|     private $application;
 | |
|     private $history;
 | |
|     private $output;
 | |
|     private $hasReadline;
 | |
|     private $processIsolation;
 | |
| 
 | |
|     /**
 | |
|      * Constructor.
 | |
|      *
 | |
|      * If there is no readline support for the current PHP executable
 | |
|      * a \RuntimeException exception is thrown.
 | |
|      *
 | |
|      * @param Application $application An application instance
 | |
|      */
 | |
|     public function __construct(Application $application)
 | |
|     {
 | |
|         $this->hasReadline = function_exists('readline');
 | |
|         $this->application = $application;
 | |
|         $this->history = getenv('HOME').'/.history_'.$application->getName();
 | |
|         $this->output = new ConsoleOutput();
 | |
|         $this->processIsolation = false;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Runs the shell.
 | |
|      */
 | |
|     public function run()
 | |
|     {
 | |
|         $this->application->setAutoExit(false);
 | |
|         $this->application->setCatchExceptions(true);
 | |
| 
 | |
|         if ($this->hasReadline) {
 | |
|             readline_read_history($this->history);
 | |
|             readline_completion_function(array($this, 'autocompleter'));
 | |
|         }
 | |
| 
 | |
|         $this->output->writeln($this->getHeader());
 | |
|         $php = null;
 | |
|         if ($this->processIsolation) {
 | |
|             $finder = new PhpExecutableFinder();
 | |
|             $php = $finder->find();
 | |
|             $this->output->writeln(<<<EOF
 | |
| <info>Running with process isolation, you should consider this:</info>
 | |
|   * each command is executed as separate process,
 | |
|   * commands don't support interactivity, all params must be passed explicitly,
 | |
|   * commands output is not colorized.
 | |
| 
 | |
| EOF
 | |
|             );
 | |
|         }
 | |
| 
 | |
|         while (true) {
 | |
|             $command = $this->readline();
 | |
| 
 | |
|             if (false === $command) {
 | |
|                 $this->output->writeln("\n");
 | |
| 
 | |
|                 break;
 | |
|             }
 | |
| 
 | |
|             if ($this->hasReadline) {
 | |
|                 readline_add_history($command);
 | |
|                 readline_write_history($this->history);
 | |
|             }
 | |
| 
 | |
|             if ($this->processIsolation) {
 | |
|                 $pb = new ProcessBuilder();
 | |
| 
 | |
|                 $process = $pb
 | |
|                     ->add($php)
 | |
|                     ->add($_SERVER['argv'][0])
 | |
|                     ->add($command)
 | |
|                     ->inheritEnvironmentVariables(true)
 | |
|                     ->getProcess()
 | |
|                 ;
 | |
| 
 | |
|                 $output = $this->output;
 | |
|                 $process->run(function($type, $data) use ($output) {
 | |
|                     $output->writeln($data);
 | |
|                 });
 | |
| 
 | |
|                 $ret = $process->getExitCode();
 | |
|             } else {
 | |
|                 $ret = $this->application->run(new StringInput($command), $this->output);
 | |
|             }
 | |
| 
 | |
|             if (0 !== $ret) {
 | |
|                 $this->output->writeln(sprintf('<error>The command terminated with an error status (%s)</error>', $ret));
 | |
|             }
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Returns the shell header.
 | |
|      *
 | |
|      * @return string The header string
 | |
|      */
 | |
|     protected function getHeader()
 | |
|     {
 | |
|         return <<<EOF
 | |
| 
 | |
| Welcome to the <info>{$this->application->getName()}</info> shell (<comment>{$this->application->getVersion()}</comment>).
 | |
| 
 | |
| At the prompt, type <comment>help</comment> for some help,
 | |
| or <comment>list</comment> to get a list of available commands.
 | |
| 
 | |
| To exit the shell, type <comment>^D</comment>.
 | |
| 
 | |
| EOF;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Renders a prompt.
 | |
|      *
 | |
|      * @return string The prompt
 | |
|      */
 | |
|     protected function getPrompt()
 | |
|     {
 | |
|         // using the formatter here is required when using readline
 | |
|         return $this->output->getFormatter()->format($this->application->getName().' > ');
 | |
|     }
 | |
| 
 | |
|     protected function getOutput()
 | |
|     {
 | |
|         return $this->output;
 | |
|     }
 | |
| 
 | |
|     protected function getApplication()
 | |
|     {
 | |
|         return $this->application;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Tries to return autocompletion for the current entered text.
 | |
|      *
 | |
|      * @param string $text The last segment of the entered text
 | |
|      *
 | |
|      * @return Boolean|array A list of guessed strings or true
 | |
|      */
 | |
|     private function autocompleter($text)
 | |
|     {
 | |
|         $info = readline_info();
 | |
|         $text = substr($info['line_buffer'], 0, $info['end']);
 | |
| 
 | |
|         if ($info['point'] !== $info['end']) {
 | |
|             return true;
 | |
|         }
 | |
| 
 | |
|         // task name?
 | |
|         if (false === strpos($text, ' ') || !$text) {
 | |
|             return array_keys($this->application->all());
 | |
|         }
 | |
| 
 | |
|         // options and arguments?
 | |
|         try {
 | |
|             $command = $this->application->find(substr($text, 0, strpos($text, ' ')));
 | |
|         } catch (\Exception $e) {
 | |
|             return true;
 | |
|         }
 | |
| 
 | |
|         $list = array('--help');
 | |
|         foreach ($command->getDefinition()->getOptions() as $option) {
 | |
|             $list[] = '--'.$option->getName();
 | |
|         }
 | |
| 
 | |
|         return $list;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Reads a single line from standard input.
 | |
|      *
 | |
|      * @return string The single line from standard input
 | |
|      */
 | |
|     private function readline()
 | |
|     {
 | |
|         if ($this->hasReadline) {
 | |
|             $line = readline($this->getPrompt());
 | |
|         } else {
 | |
|             $this->output->write($this->getPrompt());
 | |
|             $line = fgets(STDIN, 1024);
 | |
|             $line = (!$line && strlen($line) == 0) ? false : rtrim($line);
 | |
|         }
 | |
| 
 | |
|         return $line;
 | |
|     }
 | |
| 
 | |
|     public function getProcessIsolation()
 | |
|     {
 | |
|         return $this->processIsolation;
 | |
|     }
 | |
| 
 | |
|     public function setProcessIsolation($processIsolation)
 | |
|     {
 | |
|         $this->processIsolation = (Boolean) $processIsolation;
 | |
| 
 | |
|         if ($this->processIsolation && !class_exists('Symfony\\Component\\Process\\Process')) {
 | |
|             throw new \RuntimeException('Unable to isolate processes as the Symfony Process Component is not installed.');
 | |
|         }
 | |
|     }
 | |
| }
 |