Fixes badge in non-default color theme.

Signed-off-by: Eric Wang <skygragon@gmail.com>

Author: skygragon

lib/chalk.js

@@ -26,8 +26,7 @@ const DEFAULT = {
 };
 
 chalk.setTheme = function(name) {
-  const theme = this.themes.get(name) || this.themes.get('default') || {};
-  this.theme = _.extendOwn(DEFAULT, theme);
+  this.theme = this.themes.get(name) || this.themes.get('default');
 };
 
 chalk.sprint = function(s, hex) {
@@ -55,7 +54,8 @@ chalk.init = function() {
   const h = require('./helper');
   for (let f of h.getCodeDirData('colors')) {
     const theme = {};
-    _.pairs(f.data).forEach(function(x) {
+    const data = _.extendOwn({}, DEFAULT, f.data);
+    for (let x of _.pairs(data)) {
       const k = x[0];
       const v = x[1];
       const bgK = bgName(k);
@@ -67,7 +67,7 @@ chalk.init = function() {
         theme[k] = style.color.ansi.hex(v);
         theme[bgK] = style.bgColor.ansi.hex(v);
       }
-    });
+    }
     chalk.themes.set(f.name, theme);
   }