Fixed SymfonyQuestionHelper multi-choice with defaults

sstok · sstok · commit bcb4f0932fc3 · 2016-06-18T18:47:33.000+02:00
diff --git a/src/Symfony/Component/Console/Helper/SymfonyQuestionHelper.php b/src/Symfony/Component/Console/Helper/SymfonyQuestionHelper.php
@@ -67,6 +67,20 @@ protected function writePrompt(OutputInterface $output, Question $question)
 
                 break;
 
+            case $question instanceof ChoiceQuestion && $question->isMultiSelect():
+                $choices = $question->getChoices();
+                $default = explode(',', $default);
+
+                foreach ($default as &$defaultVal) {
+                    $defaultVal = $choices[$defaultVal];
+                }
+
+                unset($defaultVal);
+
+                $text = sprintf(' <info>%s</info> [<comment>%s</comment>]:', $text, implode(', ', $default));
+
+                break;
+
             case $question instanceof ChoiceQuestion:
                 $choices = $question->getChoices();
                 $text = sprintf(' <info>%s</info> [<comment>%s</comment>]:', $text, $choices[$default]);
diff --git a/src/Symfony/Component/Console/Question/ChoiceQuestion.php b/src/Symfony/Component/Console/Question/ChoiceQuestion.php
@@ -68,6 +68,16 @@ public function setMultiselect($multiselect)
         return $this;
     }
 
+    /**
+     * Returns whether the choices are multiselect.
+     *
+     * @return bool
+     */
+    public function isMultiselect()
+    {
+        return $this->multiselect;
+    }
+
     /**
      * Gets the prompt for choices.
      *
