Do not count escape sequence for column width

Author: takamin

lib/index.js

@@ -130,6 +130,14 @@
     DataCell.prototype.setData = function(data) {
         this.data = data;
     };
+    DataCell.prototype.visibleLength = function() {
+        if(typeof(this.data) != "string") {
+            return ("" + this.data).length;
+        }
+        // Remove escape sequences for text style from the string
+        var s = this.data.replace(/\x1b[^m]*m/g, '');
+        return s.length;
+    };
 
     //
     // Column class
@@ -147,11 +155,10 @@
         return this.width;
     };
     Column.prototype.setCellAt = function(row, cell) {
-        var data = cell.getData();
-        var s = "" + data;
-        var width = s.length;
+        var width = cell.visibleLength();
         this.cellAtRow[row] = cell;
         if(this.opt.autoAlign) {
+            var data = cell.getData();
             if(typeof(data) == "number") {
                 this.updateNumWidth(data);
                 width = this.getNumMaxWidth();

package.json

@@ -1,24 +1,34 @@
 {
-    "name": "list-it",
-    "version": "0.3.0",
-    "description":
-        "This module is used to create a preformatted text table.",
-    "repository": {
-        "type": "git",
-        "url": "https://github.com/takamin/list-it"
-	},
-    "directories": {
-        "lib": "./lib"
-    },
-    "main": "./lib/index.js",
-    "scripts": {
-        "test": "node ./test/test.js"
-    },
-    "keywords": [ "console", "pre", "text", "column", "table", "format", "cli" ],
-    "author": "Koji Takami",
-    "license": "MIT",
-    "devDependencies": {
-        "ansi-escape-sequences": "^2.2.2",
-         "chai": "^3.5.0"
-    }
+  "name": "list-it",
+  "version": "0.3.1",
+  "description": "This module is used to create a preformatted text table.",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/takamin/list-it"
+  },
+  "directories": {
+    "lib": "./lib"
+  },
+  "main": "./lib/index.js",
+  "scripts": {
+    "test": "node ./test/test.js"
+  },
+  "keywords": [
+    "console",
+    "pre",
+    "text",
+    "column",
+    "table",
+    "format",
+    "cli"
+  ],
+  "author": "Koji Takami",
+  "license": "MIT",
+  "devDependencies": {
+    "ansi-escape-sequences": "^2.2.2",
+    "chai": "^3.5.0"
+  },
+  "dependencies": {
+    "ansi-escape-sequences": "^2.2.2"
+  }
 }

test/test.js

@@ -250,7 +250,7 @@
                         "C 333.3 D   -4.444");
                 });
             });
-            describe("bidimentional data", function() {
+            describe("bidimensional data", function() {
                 it("no autoAlign", function() {
                     var buffer = new listit.buffer();
                     buffer.d([
@@ -274,4 +274,19 @@
             });
         });
     });
+    describe("Do not count escape sequences for column width,", function() {
+        Object.keys(ansi.style).forEach(function(key) {
+            var style = ansi.style[key];
+            it("style " + key + "#1", function() {
+                var buffer = new listit.buffer();
+                buffer.d([ styled("A", style), "B" ], ["C", "D"]);
+                assert.equal(buffer.toString(), styled("A", style) + " B\n" + "C D");
+            });
+            it("style " + key + "#2", function() {
+                var buffer = new listit.buffer();
+                buffer.d([ "A", styled("B", style) ], ["C", "D"]);
+                assert.equal(buffer.toString(), "A " + styled("B", style) + "\n" + "C D");
+            });
+        });
+    });
 }());