Primo Committ
This commit is contained in:
paoloar77
331
vendor/psy/psysh/src/Input/ShellInput.php
vendored
Normal file
331
vendor/psy/psysh/src/Input/ShellInput.php
vendored
Normal file
@@ -0,0 +1,331 @@
|
||||
<?php
|
||||
|
||||
/*
|
||||
* This file is part of Psy Shell.
|
||||
*
|
||||
* (c) 2012-2022 Justin Hileman
|
||||
*
|
||||
* For the full copyright and license information, please view the LICENSE
|
||||
* file that was distributed with this source code.
|
||||
*/
|
||||
|
||||
namespace Psy\Input;
|
||||
|
||||
use Symfony\Component\Console\Input\InputDefinition;
|
||||
use Symfony\Component\Console\Input\StringInput;
|
||||
|
||||
/**
|
||||
* A StringInput subclass specialized for code arguments.
|
||||
*/
|
||||
class ShellInput extends StringInput
|
||||
{
|
||||
private $hasCodeArgument = false;
|
||||
|
||||
/**
|
||||
* Unlike the parent implementation's tokens, this contains an array of
|
||||
* token/rest pairs, so that code arguments can be handled while parsing.
|
||||
*/
|
||||
private $tokenPairs;
|
||||
private $parsed;
|
||||
|
||||
/**
|
||||
* Constructor.
|
||||
*
|
||||
* @param string $input An array of parameters from the CLI (in the argv format)
|
||||
*/
|
||||
public function __construct(string $input)
|
||||
{
|
||||
parent::__construct($input);
|
||||
|
||||
$this->tokenPairs = $this->tokenize($input);
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*
|
||||
* @throws \InvalidArgumentException if $definition has CodeArgument before the final argument position
|
||||
*/
|
||||
public function bind(InputDefinition $definition)
|
||||
{
|
||||
$hasCodeArgument = false;
|
||||
|
||||
if ($definition->getArgumentCount() > 0) {
|
||||
$args = $definition->getArguments();
|
||||
$lastArg = \array_pop($args);
|
||||
foreach ($args as $arg) {
|
||||
if ($arg instanceof CodeArgument) {
|
||||
$msg = \sprintf('Unexpected CodeArgument before the final position: %s', $arg->getName());
|
||||
throw new \InvalidArgumentException($msg);
|
||||
}
|
||||
}
|
||||
|
||||
if ($lastArg instanceof CodeArgument) {
|
||||
$hasCodeArgument = true;
|
||||
}
|
||||
}
|
||||
|
||||
$this->hasCodeArgument = $hasCodeArgument;
|
||||
|
||||
return parent::bind($definition);
|
||||
}
|
||||
|
||||
/**
|
||||
* Tokenizes a string.
|
||||
*
|
||||
* The version of this on StringInput is good, but doesn't handle code
|
||||
* arguments if they're at all complicated. This does :)
|
||||
*
|
||||
* @param string $input The input to tokenize
|
||||
*
|
||||
* @return array An array of token/rest pairs
|
||||
*
|
||||
* @throws \InvalidArgumentException When unable to parse input (should never happen)
|
||||
*/
|
||||
private function tokenize(string $input): array
|
||||
{
|
||||
$tokens = [];
|
||||
$length = \strlen($input);
|
||||
$cursor = 0;
|
||||
while ($cursor < $length) {
|
||||
if (\preg_match('/\s+/A', $input, $match, 0, $cursor)) {
|
||||
} elseif (\preg_match('/([^="\'\s]+?)(=?)('.StringInput::REGEX_QUOTED_STRING.'+)/A', $input, $match, 0, $cursor)) {
|
||||
$tokens[] = [
|
||||
$match[1].$match[2].\stripcslashes(\str_replace(['"\'', '\'"', '\'\'', '""'], '', \substr($match[3], 1, \strlen($match[3]) - 2))),
|
||||
\stripcslashes(\substr($input, $cursor)),
|
||||
];
|
||||
} elseif (\preg_match('/'.StringInput::REGEX_QUOTED_STRING.'/A', $input, $match, 0, $cursor)) {
|
||||
$tokens[] = [
|
||||
\stripcslashes(\substr($match[0], 1, \strlen($match[0]) - 2)),
|
||||
\stripcslashes(\substr($input, $cursor)),
|
||||
];
|
||||
} elseif (\preg_match('/'.StringInput::REGEX_STRING.'/A', $input, $match, 0, $cursor)) {
|
||||
$tokens[] = [
|
||||
\stripcslashes($match[1]),
|
||||
\stripcslashes(\substr($input, $cursor)),
|
||||
];
|
||||
} else {
|
||||
// should never happen
|
||||
// @codeCoverageIgnoreStart
|
||||
throw new \InvalidArgumentException(\sprintf('Unable to parse input near "... %s ..."', \substr($input, $cursor, 10)));
|
||||
// @codeCoverageIgnoreEnd
|
||||
}
|
||||
|
||||
$cursor += \strlen($match[0]);
|
||||
}
|
||||
|
||||
return $tokens;
|
||||
}
|
||||
|
||||
/**
|
||||
* Same as parent, but with some bonus handling for code arguments.
|
||||
*/
|
||||
protected function parse()
|
||||
{
|
||||
$parseOptions = true;
|
||||
$this->parsed = $this->tokenPairs;
|
||||
while (null !== $tokenPair = \array_shift($this->parsed)) {
|
||||
// token is what you'd expect. rest is the remainder of the input
|
||||
// string, including token, and will be used if this is a code arg.
|
||||
list($token, $rest) = $tokenPair;
|
||||
|
||||
if ($parseOptions && '' === $token) {
|
||||
$this->parseShellArgument($token, $rest);
|
||||
} elseif ($parseOptions && '--' === $token) {
|
||||
$parseOptions = false;
|
||||
} elseif ($parseOptions && 0 === \strpos($token, '--')) {
|
||||
$this->parseLongOption($token);
|
||||
} elseif ($parseOptions && '-' === $token[0] && '-' !== $token) {
|
||||
$this->parseShortOption($token);
|
||||
} else {
|
||||
$this->parseShellArgument($token, $rest);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Parses an argument, with bonus handling for code arguments.
|
||||
*
|
||||
* @param string $token The current token
|
||||
* @param string $rest The remaining unparsed input, including the current token
|
||||
*
|
||||
* @throws \RuntimeException When too many arguments are given
|
||||
*/
|
||||
private function parseShellArgument(string $token, string $rest)
|
||||
{
|
||||
$c = \count($this->arguments);
|
||||
|
||||
// if input is expecting another argument, add it
|
||||
if ($this->definition->hasArgument($c)) {
|
||||
$arg = $this->definition->getArgument($c);
|
||||
|
||||
if ($arg instanceof CodeArgument) {
|
||||
// When we find a code argument, we're done parsing. Add the
|
||||
// remaining input to the current argument and call it a day.
|
||||
$this->parsed = [];
|
||||
$this->arguments[$arg->getName()] = $rest;
|
||||
} else {
|
||||
$this->arguments[$arg->getName()] = $arg->isArray() ? [$token] : $token;
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
// (copypasta)
|
||||
//
|
||||
// @codeCoverageIgnoreStart
|
||||
|
||||
// if last argument isArray(), append token to last argument
|
||||
if ($this->definition->hasArgument($c - 1) && $this->definition->getArgument($c - 1)->isArray()) {
|
||||
$arg = $this->definition->getArgument($c - 1);
|
||||
$this->arguments[$arg->getName()][] = $token;
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
// unexpected argument
|
||||
$all = $this->definition->getArguments();
|
||||
if (\count($all)) {
|
||||
throw new \RuntimeException(\sprintf('Too many arguments, expected arguments "%s".', \implode('" "', \array_keys($all))));
|
||||
}
|
||||
|
||||
throw new \RuntimeException(\sprintf('No arguments expected, got "%s".', $token));
|
||||
// @codeCoverageIgnoreEnd
|
||||
}
|
||||
|
||||
// Everything below this is copypasta from ArgvInput private methods
|
||||
// @codeCoverageIgnoreStart
|
||||
|
||||
/**
|
||||
* Parses a short option.
|
||||
*
|
||||
* @param string $token The current token
|
||||
*/
|
||||
private function parseShortOption(string $token)
|
||||
{
|
||||
$name = \substr($token, 1);
|
||||
|
||||
if (\strlen($name) > 1) {
|
||||
if ($this->definition->hasShortcut($name[0]) && $this->definition->getOptionForShortcut($name[0])->acceptValue()) {
|
||||
// an option with a value (with no space)
|
||||
$this->addShortOption($name[0], \substr($name, 1));
|
||||
} else {
|
||||
$this->parseShortOptionSet($name);
|
||||
}
|
||||
} else {
|
||||
$this->addShortOption($name, null);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Parses a short option set.
|
||||
*
|
||||
* @param string $name The current token
|
||||
*
|
||||
* @throws \RuntimeException When option given doesn't exist
|
||||
*/
|
||||
private function parseShortOptionSet(string $name)
|
||||
{
|
||||
$len = \strlen($name);
|
||||
for ($i = 0; $i < $len; $i++) {
|
||||
if (!$this->definition->hasShortcut($name[$i])) {
|
||||
throw new \RuntimeException(\sprintf('The "-%s" option does not exist.', $name[$i]));
|
||||
}
|
||||
|
||||
$option = $this->definition->getOptionForShortcut($name[$i]);
|
||||
if ($option->acceptValue()) {
|
||||
$this->addLongOption($option->getName(), $i === $len - 1 ? null : \substr($name, $i + 1));
|
||||
|
||||
break;
|
||||
} else {
|
||||
$this->addLongOption($option->getName(), null);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Parses a long option.
|
||||
*
|
||||
* @param string $token The current token
|
||||
*/
|
||||
private function parseLongOption(string $token)
|
||||
{
|
||||
$name = \substr($token, 2);
|
||||
|
||||
if (false !== $pos = \strpos($name, '=')) {
|
||||
if (($value = \substr($name, $pos + 1)) === '') {
|
||||
\array_unshift($this->parsed, [$value, null]);
|
||||
}
|
||||
$this->addLongOption(\substr($name, 0, $pos), $value);
|
||||
} else {
|
||||
$this->addLongOption($name, null);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Adds a short option value.
|
||||
*
|
||||
* @param string $shortcut The short option key
|
||||
* @param mixed $value The value for the option
|
||||
*
|
||||
* @throws \RuntimeException When option given doesn't exist
|
||||
*/
|
||||
private function addShortOption(string $shortcut, $value)
|
||||
{
|
||||
if (!$this->definition->hasShortcut($shortcut)) {
|
||||
throw new \RuntimeException(\sprintf('The "-%s" option does not exist.', $shortcut));
|
||||
}
|
||||
|
||||
$this->addLongOption($this->definition->getOptionForShortcut($shortcut)->getName(), $value);
|
||||
}
|
||||
|
||||
/**
|
||||
* Adds a long option value.
|
||||
*
|
||||
* @param string $name The long option key
|
||||
* @param mixed $value The value for the option
|
||||
*
|
||||
* @throws \RuntimeException When option given doesn't exist
|
||||
*/
|
||||
private function addLongOption(string $name, $value)
|
||||
{
|
||||
if (!$this->definition->hasOption($name)) {
|
||||
throw new \RuntimeException(\sprintf('The "--%s" option does not exist.', $name));
|
||||
}
|
||||
|
||||
$option = $this->definition->getOption($name);
|
||||
|
||||
if (null !== $value && !$option->acceptValue()) {
|
||||
throw new \RuntimeException(\sprintf('The "--%s" option does not accept a value.', $name));
|
||||
}
|
||||
|
||||
if (\in_array($value, ['', null], true) && $option->acceptValue() && \count($this->parsed)) {
|
||||
// if option accepts an optional or mandatory argument
|
||||
// let's see if there is one provided
|
||||
$next = \array_shift($this->parsed);
|
||||
$nextToken = $next[0];
|
||||
if ((isset($nextToken[0]) && '-' !== $nextToken[0]) || \in_array($nextToken, ['', null], true)) {
|
||||
$value = $nextToken;
|
||||
} else {
|
||||
\array_unshift($this->parsed, $next);
|
||||
}
|
||||
}
|
||||
|
||||
if ($value === null) {
|
||||
if ($option->isValueRequired()) {
|
||||
throw new \RuntimeException(\sprintf('The "--%s" option requires a value.', $name));
|
||||
}
|
||||
|
||||
if (!$option->isArray() && !$option->isValueOptional()) {
|
||||
$value = true;
|
||||
}
|
||||
}
|
||||
|
||||
if ($option->isArray()) {
|
||||
$this->options[$name][] = $value;
|
||||
} else {
|
||||
$this->options[$name] = $value;
|
||||
}
|
||||
}
|
||||
|
||||
// @codeCoverageIgnoreEnd
|
||||
}
|
||||
Reference in New Issue
Block a user