Merge pull request #20 from fetus-hina/plain

Add ansi_to_text method
drudru · Jun 7, 2015 · 9290483 · 9290483
2 parents 1246b37 + 2ae883d
commit 9290483
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 3 deletions.
diff --git a/ansi_up.js b/ansi_up.js
@@ -97,21 +97,29 @@
     };
 
     Ansi_Up.prototype.ansi_to_html = function (txt, options) {
-      var self = this;
+      return this.process(txt, options, true);
+    };
+
+    Ansi_Up.prototype.ansi_to_text = function (txt) {
+      var options = {};
+      return this.process(txt, options, false);
+    };
 
+    Ansi_Up.prototype.process = function (txt, options, markup) {
+      var self = this;
       var raw_text_chunks = txt.split(/\033\[/);
       var first_chunk = raw_text_chunks.shift(); // the first chunk is not the result of the split
 
       var color_chunks = raw_text_chunks.map(function (chunk) {
-        return self.process_chunk(chunk, options);
+        return self.process_chunk(chunk, options, markup);
       });
 
       color_chunks.unshift(first_chunk);
 
       return color_chunks.join('');
     };
 
-    Ansi_Up.prototype.process_chunk = function (text, options) {
+    Ansi_Up.prototype.process_chunk = function (text, options, markup) {
 
       // Are we using classes or styles?
       options = typeof options == 'undefined' ? {} : options;
@@ -144,6 +152,10 @@
         return orig_txt;
       }
 
+      if (!markup) {
+        return orig_txt;
+      }
+
       var self = this;
 
       while (nums.length > 0) {
@@ -284,6 +296,11 @@
         return a2h.ansi_to_html(txt, options);
       },
 
+      ansi_to_text: function (txt) {
+        var a2h = new Ansi_Up();
+        return a2h.ansi_to_text(txt);
+      },
+
       ansi_to_html_obj: function () {
         return new Ansi_Up();
       }

diff --git a/test/ansi_up-test.js b/test/ansi_up-test.js
@@ -481,5 +481,22 @@ describe('ansi_up', function() {
       });
     });
   });
+  describe('ansi to text', function() {
+    it('should remove color sequence', function() {
+      var start = "foo \033[1;32mbar\033[0m baz";
+      var l = ansi_up.ansi_to_text(start);
+      l.should.eql("foo bar baz");
+    });
+    it('should remove unsupported sequence', function() {
+      var start = "foo \033[1Abar";
+      var l = ansi_up.ansi_to_text(start);
+      l.should.eql('foo bar');
+    });
+    it('should keep multiline', function() {
+      var start = "foo \033[1;32mbar\nbaz\033[0m qux";
+      var l = ansi_up.ansi_to_text(start);
+      l.should.eql("foo bar\nbaz qux");
+    });
+  });
 });