Chris@0: <?php
Chris@0: 
Chris@0: namespace DrupalCodeGenerator\Helper;
Chris@0: 
Chris@0: use Symfony\Component\Console\Question\ChoiceQuestion;
Chris@0: use Symfony\Component\Console\Question\Question;
Chris@0: 
Chris@0: /**
Chris@0:  * Implements missing Question setters.
Chris@0:  */
Chris@0: trait QuestionSettersTrait {
Chris@0: 
Chris@0:   /**
Chris@0:    * Sets question text.
Chris@0:    *
Chris@0:    * @param \Symfony\Component\Console\Question\Question $question
Chris@0:    *   The question to update.
Chris@0:    * @param mixed $question_text
Chris@0:    *   The question text.
Chris@0:    */
Chris@0:   protected function setQuestionText(Question $question, $question_text) {
Chris@0:     // Choice question has a different constructor signature.
Chris@0:     if ($question instanceof ChoiceQuestion) {
Chris@0:       $question->__construct($question_text, $question->getChoices(), $question->getDefault());
Chris@0:     }
Chris@0:     else {
Chris@0:       $question->__construct($question_text, $question->getDefault());
Chris@0:     }
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Sets question default value.
Chris@0:    *
Chris@0:    * @param \Symfony\Component\Console\Question\Question $question
Chris@0:    *   The question to update.
Chris@0:    * @param mixed $default_value
Chris@0:    *   Default value for the question.
Chris@0:    */
Chris@0:   protected function setQuestionDefault(Question $question, $default_value) {
Chris@0:     if ($question instanceof ChoiceQuestion) {
Chris@0:       $question->__construct($question->getQuestion(), $question->getChoices(), $default_value);
Chris@0:     }
Chris@0:     else {
Chris@0:       $question->__construct($question->getQuestion(), $default_value);
Chris@0:     }
Chris@0:   }
Chris@0: 
Chris@0: }