[console] Add and initialize io DrupalStyle property. (#259)

hechoendrupal · Dec 31, 2017 · 1caa7a1 · 1caa7a1
1 parent 23383af
commit 1caa7a1
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/src/Command/Command.php b/src/Command/Command.php
@@ -8,7 +8,10 @@
 namespace Drupal\Console\Core\Command;
 
 use Symfony\Component\Console\Command\Command as BaseCommand;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Output\OutputInterface;
 use Drupal\Console\Core\Command\Shared\CommandTrait;
+use Drupal\Console\Core\Style\DrupalStyle;
 
 /**
  * Class Command
@@ -17,4 +20,24 @@
 abstract class Command extends BaseCommand
 {
     use CommandTrait;
+
+    /**
+     * @var DrupalStyle
+     */
+    protected $io;
+
+    /**
+     * {@inheritdoc}
+     */
+    protected function initialize(InputInterface $input, OutputInterface $output)
+    {
+        $this->io = new DrupalStyle($input, $output);
+    }
+
+    /**
+     * @return \Drupal\Console\Core\Style\DrupalStyle
+     */
+    public function getIo() {
+        return $this->io;
+    }
 }
diff --git a/src/Command/ContainerAwareCommand.php b/src/Command/ContainerAwareCommand.php
@@ -7,14 +7,13 @@
 
 namespace Drupal\Console\Core\Command;
 
-use Symfony\Component\Console\Command\Command as BaseCommand;
 use Drupal\Console\Core\Command\Shared\ContainerAwareCommandTrait;
 
 /**
  * Class ContainerAwareCommand
  * @package Drupal\Console\Core\Command
  */
-abstract class ContainerAwareCommand extends BaseCommand
+abstract class ContainerAwareCommand extends Command
 {
     use ContainerAwareCommandTrait;
 }