fix: correct cell width resolution logic (fixes #88, #9)

gajus · Feb 10, 2019 · ce01874 · ce01874
1 parent 6e167c0
commit ce01874
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src/calculateCellWidthIndex.js b/src/calculateCellWidthIndex.js
@@ -8,6 +8,10 @@ import stringWidth from 'string-width';
  */
 export default (cells) => {
   return cells.map((value) => {
-    return stringWidth(value);
+    return Math.max(
+      ...value.split('\n').map((line) => {
+        return stringWidth(line);
+      })
+    );
   });
 };
diff --git a/test/calculateCellWidthIndex.js b/test/calculateCellWidthIndex.js
@@ -17,4 +17,13 @@ describe('calculateCellWidthIndex', () => {
       expect(cellWidthIndex[2]).to.equal(6, 'third column');
     });
   });
+  context('cell contains newline characters', () => {
+    it('picks the longest line length', () => {
+      const cellWidthIndex = calculateCellWidthIndex([
+        'aaaa\naa'
+      ]);
+
+      expect(cellWidthIndex[0]).to.equal(4);
+    });
+  });
 });