190 lines
4.9 KiB
PHP
190 lines
4.9 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\Question;
|
|
|
|
use Symfony\Component\Console\Exception\InvalidArgumentException;
|
|
|
|
/**
|
|
* Represents a choice question.
|
|
*
|
|
* @author Fabien Potencier <fabien@symfony.com>
|
|
*/
|
|
class ChoiceQuestion extends Question
|
|
{
|
|
private $choices;
|
|
private $multiselect = false;
|
|
private $prompt = ' > ';
|
|
private $errorMessage = 'Value "%s" is invalid';
|
|
|
|
/**
|
|
* @param string $question The question to ask to the user
|
|
* @param array $choices The list of available choices
|
|
* @param mixed $default The default answer to return
|
|
*/
|
|
public function __construct($question, array $choices, $default = null)
|
|
{
|
|
if (!$choices) {
|
|
throw new \LogicException('Choice question must have at least 1 choice available.');
|
|
}
|
|
|
|
parent::__construct($question, $default);
|
|
|
|
$this->choices = $choices;
|
|
$this->setValidator($this->getDefaultValidator());
|
|
$this->setAutocompleterValues($choices);
|
|
}
|
|
|
|
/**
|
|
* Returns available choices.
|
|
*
|
|
* @return array
|
|
*/
|
|
public function getChoices()
|
|
{
|
|
return $this->choices;
|
|
}
|
|
|
|
/**
|
|
* Sets multiselect option.
|
|
*
|
|
* When multiselect is set to true, multiple choices can be answered.
|
|
*
|
|
* @param bool $multiselect
|
|
*
|
|
* @return $this
|
|
*/
|
|
public function setMultiselect($multiselect)
|
|
{
|
|
$this->multiselect = $multiselect;
|
|
$this->setValidator($this->getDefaultValidator());
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* Returns whether the choices are multiselect.
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function isMultiselect()
|
|
{
|
|
return $this->multiselect;
|
|
}
|
|
|
|
/**
|
|
* Gets the prompt for choices.
|
|
*
|
|
* @return string
|
|
*/
|
|
public function getPrompt()
|
|
{
|
|
return $this->prompt;
|
|
}
|
|
|
|
/**
|
|
* Sets the prompt for choices.
|
|
*
|
|
* @param string $prompt
|
|
*
|
|
* @return $this
|
|
*/
|
|
public function setPrompt($prompt)
|
|
{
|
|
$this->prompt = $prompt;
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* Sets the error message for invalid values.
|
|
*
|
|
* The error message has a string placeholder (%s) for the invalid value.
|
|
*
|
|
* @param string $errorMessage
|
|
*
|
|
* @return $this
|
|
*/
|
|
public function setErrorMessage($errorMessage)
|
|
{
|
|
$this->errorMessage = $errorMessage;
|
|
$this->setValidator($this->getDefaultValidator());
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* Returns the default answer validator.
|
|
*
|
|
* @return callable
|
|
*/
|
|
private function getDefaultValidator()
|
|
{
|
|
$choices = $this->choices;
|
|
$errorMessage = $this->errorMessage;
|
|
$multiselect = $this->multiselect;
|
|
$isAssoc = $this->isAssoc($choices);
|
|
|
|
return function ($selected) use ($choices, $errorMessage, $multiselect, $isAssoc) {
|
|
// Collapse all spaces.
|
|
$selectedChoices = str_replace(' ', '', $selected);
|
|
|
|
if ($multiselect) {
|
|
// Check for a separated comma values
|
|
if (!preg_match('/^[^,]+(?:,[^,]+)*$/', $selectedChoices, $matches)) {
|
|
throw new InvalidArgumentException(sprintf($errorMessage, $selected));
|
|
}
|
|
$selectedChoices = explode(',', $selectedChoices);
|
|
} else {
|
|
$selectedChoices = array($selected);
|
|
}
|
|
|
|
$multiselectChoices = array();
|
|
foreach ($selectedChoices as $value) {
|
|
$results = array();
|
|
foreach ($choices as $key => $choice) {
|
|
if ($choice === $value) {
|
|
$results[] = $key;
|
|
}
|
|
}
|
|
|
|
if (count($results) > 1) {
|
|
throw new InvalidArgumentException(sprintf('The provided answer is ambiguous. Value should be one of %s.', implode(' or ', $results)));
|
|
}
|
|
|
|
$result = array_search($value, $choices);
|
|
|
|
if (!$isAssoc) {
|
|
if (false !== $result) {
|
|
$result = $choices[$result];
|
|
} elseif (isset($choices[$value])) {
|
|
$result = $choices[$value];
|
|
}
|
|
} elseif (false === $result && isset($choices[$value])) {
|
|
$result = $value;
|
|
}
|
|
|
|
if (false === $result) {
|
|
throw new InvalidArgumentException(sprintf($errorMessage, $value));
|
|
}
|
|
|
|
$multiselectChoices[] = (string) $result;
|
|
}
|
|
|
|
if ($multiselect) {
|
|
return $multiselectChoices;
|
|
}
|
|
|
|
return current($multiselectChoices);
|
|
};
|
|
}
|
|
}
|