feat(fancy): support icon field

unjs · May 27, 2018 · 0123bed · 0123bed
1 parent 7a750bf
commit 0123bed
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -159,6 +159,7 @@ Extended fields:
 - `badge`
 - `additional`
 - `additionalStyle` (By default: `grey`)
+- `icon` (Default depends on log type)
 
 ## Integrations
 

diff --git a/src/reporters/fancy.js b/src/reporters/fancy.js
@@ -22,12 +22,11 @@ export default class FancyReporter {
     this.stream = stream || process.stderr
   }
 
-  formatBadge (type, color = 'blue') {
+  formatBadge (type, color = 'blue', icon) {
     return chalk['bg' + startCase(color)].black(` ${type.toUpperCase()} `) + ' '
   }
 
-  formatTag (type, color = 'blue') {
-    const icon = ICONS[type] || ICONS.default
+  formatTag (type, color = 'blue', icon) {
     return chalk[color](`${icon} ${type.toLowerCase()}`) + ' '
   }
 
@@ -48,10 +47,12 @@ export default class FancyReporter {
       this.clear()
     }
 
+    const icon = logObj.icon || ICONS[type] || ICONS.default
+
     if (logObj.badge) {
-      this.stream.write('\n\n' + this.formatBadge(logObj.type, logObj.color) + message + '\n\n')
+      this.stream.write('\n\n' + this.formatBadge(logObj.type, logObj.color, icon) + message + '\n\n')
     } else {
-      this.stream.write(this.formatTag(logObj.type, logObj.color) + message + '\n')
+      this.stream.write(this.formatTag(logObj.type, logObj.color, icon) + message + '\n')
     }
 
     if (logObj.additional) {