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 Psy\Shell;
Chris@13: use Symfony\Component\Console\Application;
Chris@13: use Symfony\Component\Console\Command\Command as BaseCommand;
Chris@13: use Symfony\Component\Console\Helper\Table;
Chris@13: use Symfony\Component\Console\Helper\TableHelper;
Chris@13: use Symfony\Component\Console\Helper\TableStyle;
Chris@13: use Symfony\Component\Console\Output\OutputInterface;
Chris@13: 
Chris@13: /**
Chris@13:  * The Psy Shell base command.
Chris@13:  */
Chris@13: abstract class Command extends BaseCommand
Chris@13: {
Chris@13:     /**
Chris@13:      * Sets the application instance for this command.
Chris@13:      *
Chris@13:      * @param Application $application An Application instance
Chris@13:      *
Chris@13:      * @api
Chris@13:      */
Chris@13:     public function setApplication(Application $application = null)
Chris@13:     {
Chris@13:         if ($application !== null && !$application instanceof Shell) {
Chris@13:             throw new \InvalidArgumentException('PsySH Commands require an instance of Psy\Shell');
Chris@13:         }
Chris@13: 
Chris@13:         return parent::setApplication($application);
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * {@inheritdoc}
Chris@13:      */
Chris@13:     public function asText()
Chris@13:     {
Chris@13:         $messages = [
Chris@13:             '<comment>Usage:</comment>',
Chris@13:             ' ' . $this->getSynopsis(),
Chris@13:             '',
Chris@13:         ];
Chris@13: 
Chris@13:         if ($this->getAliases()) {
Chris@13:             $messages[] = $this->aliasesAsText();
Chris@13:         }
Chris@13: 
Chris@13:         if ($this->getArguments()) {
Chris@13:             $messages[] = $this->argumentsAsText();
Chris@13:         }
Chris@13: 
Chris@13:         if ($this->getOptions()) {
Chris@13:             $messages[] = $this->optionsAsText();
Chris@13:         }
Chris@13: 
Chris@13:         if ($help = $this->getProcessedHelp()) {
Chris@13:             $messages[] = '<comment>Help:</comment>';
Chris@17:             $messages[] = ' ' . \str_replace("\n", "\n ", $help) . "\n";
Chris@13:         }
Chris@13: 
Chris@17:         return \implode("\n", $messages);
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * {@inheritdoc}
Chris@13:      */
Chris@13:     private function getArguments()
Chris@13:     {
Chris@13:         $hidden = $this->getHiddenArguments();
Chris@13: 
Chris@17:         return \array_filter($this->getNativeDefinition()->getArguments(), function ($argument) use ($hidden) {
Chris@17:             return !\in_array($argument->getName(), $hidden);
Chris@13:         });
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * These arguments will be excluded from help output.
Chris@13:      *
Chris@13:      * @return array
Chris@13:      */
Chris@13:     protected function getHiddenArguments()
Chris@13:     {
Chris@13:         return ['command'];
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * {@inheritdoc}
Chris@13:      */
Chris@13:     private function getOptions()
Chris@13:     {
Chris@13:         $hidden = $this->getHiddenOptions();
Chris@13: 
Chris@17:         return \array_filter($this->getNativeDefinition()->getOptions(), function ($option) use ($hidden) {
Chris@17:             return !\in_array($option->getName(), $hidden);
Chris@13:         });
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * These options will be excluded from help output.
Chris@13:      *
Chris@13:      * @return array
Chris@13:      */
Chris@13:     protected function getHiddenOptions()
Chris@13:     {
Chris@13:         return ['verbose'];
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * Format command aliases as text..
Chris@13:      *
Chris@13:      * @return string
Chris@13:      */
Chris@13:     private function aliasesAsText()
Chris@13:     {
Chris@17:         return '<comment>Aliases:</comment> <info>' . \implode(', ', $this->getAliases()) . '</info>' . PHP_EOL;
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * Format command arguments as text.
Chris@13:      *
Chris@13:      * @return string
Chris@13:      */
Chris@13:     private function argumentsAsText()
Chris@13:     {
Chris@13:         $max = $this->getMaxWidth();
Chris@13:         $messages = [];
Chris@13: 
Chris@13:         $arguments = $this->getArguments();
Chris@13:         if (!empty($arguments)) {
Chris@13:             $messages[] = '<comment>Arguments:</comment>';
Chris@13:             foreach ($arguments as $argument) {
Chris@17:                 if (null !== $argument->getDefault() && (!\is_array($argument->getDefault()) || \count($argument->getDefault()))) {
Chris@17:                     $default = \sprintf('<comment> (default: %s)</comment>', $this->formatDefaultValue($argument->getDefault()));
Chris@13:                 } else {
Chris@13:                     $default = '';
Chris@13:                 }
Chris@13: 
Chris@17:                 $description = \str_replace("\n", "\n" . \str_pad('', $max + 2, ' '), $argument->getDescription());
Chris@13: 
Chris@17:                 $messages[] = \sprintf(" <info>%-${max}s</info> %s%s", $argument->getName(), $description, $default);
Chris@13:             }
Chris@13: 
Chris@13:             $messages[] = '';
Chris@13:         }
Chris@13: 
Chris@17:         return \implode(PHP_EOL, $messages);
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * Format options as text.
Chris@13:      *
Chris@13:      * @return string
Chris@13:      */
Chris@13:     private function optionsAsText()
Chris@13:     {
Chris@13:         $max = $this->getMaxWidth();
Chris@13:         $messages = [];
Chris@13: 
Chris@13:         $options = $this->getOptions();
Chris@13:         if ($options) {
Chris@13:             $messages[] = '<comment>Options:</comment>';
Chris@13: 
Chris@13:             foreach ($options as $option) {
Chris@17:                 if ($option->acceptValue() && null !== $option->getDefault() && (!\is_array($option->getDefault()) || \count($option->getDefault()))) {
Chris@17:                     $default = \sprintf('<comment> (default: %s)</comment>', $this->formatDefaultValue($option->getDefault()));
Chris@13:                 } else {
Chris@13:                     $default = '';
Chris@13:                 }
Chris@13: 
Chris@13:                 $multiple = $option->isArray() ? '<comment> (multiple values allowed)</comment>' : '';
Chris@17:                 $description = \str_replace("\n", "\n" . \str_pad('', $max + 2, ' '), $option->getDescription());
Chris@13: 
Chris@17:                 $optionMax = $max - \strlen($option->getName()) - 2;
Chris@17:                 $messages[] = \sprintf(
Chris@13:                     " <info>%s</info> %-${optionMax}s%s%s%s",
Chris@13:                     '--' . $option->getName(),
Chris@17:                     $option->getShortcut() ? \sprintf('(-%s) ', $option->getShortcut()) : '',
Chris@13:                     $description,
Chris@13:                     $default,
Chris@13:                     $multiple
Chris@13:                 );
Chris@13:             }
Chris@13: 
Chris@13:             $messages[] = '';
Chris@13:         }
Chris@13: 
Chris@17:         return \implode(PHP_EOL, $messages);
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * Calculate the maximum padding width for a set of lines.
Chris@13:      *
Chris@13:      * @return int
Chris@13:      */
Chris@13:     private function getMaxWidth()
Chris@13:     {
Chris@13:         $max = 0;
Chris@13: 
Chris@13:         foreach ($this->getOptions() as $option) {
Chris@17:             $nameLength = \strlen($option->getName()) + 2;
Chris@13:             if ($option->getShortcut()) {
Chris@17:                 $nameLength += \strlen($option->getShortcut()) + 3;
Chris@13:             }
Chris@13: 
Chris@17:             $max = \max($max, $nameLength);
Chris@13:         }
Chris@13: 
Chris@13:         foreach ($this->getArguments() as $argument) {
Chris@17:             $max = \max($max, \strlen($argument->getName()));
Chris@13:         }
Chris@13: 
Chris@13:         return ++$max;
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * Format an option default as text.
Chris@13:      *
Chris@13:      * @param mixed $default
Chris@13:      *
Chris@13:      * @return string
Chris@13:      */
Chris@13:     private function formatDefaultValue($default)
Chris@13:     {
Chris@17:         if (\is_array($default) && $default === \array_values($default)) {
Chris@17:             return \sprintf("array('%s')", \implode("', '", $default));
Chris@13:         }
Chris@13: 
Chris@17:         return \str_replace("\n", '', \var_export($default, true));
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * Get a Table instance.
Chris@13:      *
Chris@13:      * Falls back to legacy TableHelper.
Chris@13:      *
Chris@13:      * @return Table|TableHelper
Chris@13:      */
Chris@13:     protected function getTable(OutputInterface $output)
Chris@13:     {
Chris@17:         if (!\class_exists('Symfony\Component\Console\Helper\Table')) {
Chris@13:             return $this->getTableHelper();
Chris@13:         }
Chris@13: 
Chris@13:         $style = new TableStyle();
Chris@13:         $style
Chris@13:             ->setVerticalBorderChar(' ')
Chris@13:             ->setHorizontalBorderChar('')
Chris@13:             ->setCrossingChar('');
Chris@13: 
Chris@13:         $table = new Table($output);
Chris@13: 
Chris@13:         return $table
Chris@13:             ->setRows([])
Chris@13:             ->setStyle($style);
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * Legacy fallback for getTable.
Chris@13:      *
Chris@13:      * @return TableHelper
Chris@13:      */
Chris@13:     protected function getTableHelper()
Chris@13:     {
Chris@13:         $table = $this->getApplication()->getHelperSet()->get('table');
Chris@13: 
Chris@13:         return $table
Chris@13:             ->setRows([])
Chris@13:             ->setLayout(TableHelper::LAYOUT_BORDERLESS)
Chris@13:             ->setHorizontalBorderChar('')
Chris@13:             ->setCrossingChar('');
Chris@13:     }
Chris@13: }