Fix the color wrap function

martynsmith · Sep 3, 2013 · 58580f4 · 58580f4
1 parent 6108735
commit 58580f4
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lib/colors.js b/lib/colors.js
@@ -20,8 +20,11 @@ codes = {
 exports.codes = codes;
 
 function wrap(color, text, reset_color) {
-    reset_color = typeof(a) != 'undefined' ? a : 'reset';
-    return codes[color] + text + codes[reset_color];
+    if(codes[color]) {
+        text = codes[color] + text;
+        text += (codes[reset_color]) ? codes[reset_color] : codes['reset'];
+    }
+    return text;
 };
 exports.wrap = wrap;