[generator] Add DrupalStyle. (#332)

hechoendrupal · Mar 21, 2018 · dcbe52a · dcbe52a
1 parent af8b3e4
commit dcbe52a
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/Generator/Generator.php b/src/Generator/Generator.php
@@ -11,6 +11,7 @@
 use Drupal\Console\Core\Utils\FileQueue;
 use Drupal\Console\Core\Utils\CountCodeLines;
 use Drupal\Console\Core\Utils\DrupalFinder;
+use Drupal\Console\Core\Style\DrupalStyle;
 
 /**
  * Class Generator
@@ -39,6 +40,11 @@ abstract class Generator
      */
     protected $drupalFinder;
 
+    /**
+     * @var DrupalStyle
+     */
+    protected $io;
+
     /**
      * @param $renderer
      */
@@ -71,6 +77,20 @@ public function setDrupalFinder($drupalFinder)
         $this->drupalFinder = $drupalFinder;
     }
 
+    /**
+     * @return \Drupal\Console\Core\Style\DrupalStyle
+     */
+    public function getIo() {
+        return $this->io;
+    }
+
+    /**
+     * @param \Drupal\Console\Core\Style\DrupalStyle $io
+     */
+    public function setIo($io) {
+        $this->io = $io;
+    }
+
     /**
      * @param string $template
      * @param string $target