[console] Improve DrupalStyle class. (#136)

hechoendrupal · Jan 19, 2017 · 96dd25e · 96dd25e
1 parent d72c7b6
commit 96dd25e
Showing 1 changed file with 46 additions and 0 deletions.
diff --git a/src/Style/DrupalStyle.php b/src/Style/DrupalStyle.php
@@ -220,4 +220,50 @@ public function text($message)
         $message = sprintf('// %s', $message);
         parent::text($message);
     }
+
+    public function successLite($message, $newLine = false) {
+        $message = sprintf('<info>✔</info> %s', $message);
+        parent::text($message);
+        if ($newLine){
+            parent::newLine();
+        }
+    }
+
+    public function errorLite($message, $newLine = false) {
+        $message = sprintf('<fg=red>✘</> %s', $message);
+        parent::text($message);
+        if ($newLine){
+            parent::newLine();
+        }
+    }
+
+    public function warningLite($message, $newLine = false) {
+        $message = sprintf('<comment>!</comment> %s', $message);
+        parent::text($message);
+        if ($newLine){
+            parent::newLine();
+        }
+    }
+
+    public function customLite($message, $prefix = '*', $style = '', $newLine = false) {
+        if ($style) {
+            $message = sprintf(
+                '<%s>%s</%s> %s',
+                $style,
+                $prefix,
+                $style,
+                $message
+            );
+        } else {
+            $message = sprintf(
+                '%s %s',
+                $prefix,
+                $message
+            );
+        }
+        parent::text($message);
+        if ($newLine){
+            parent::newLine();
+        }
+    }
 }