Add option to not remove whitespace - fixes #9 (#17)

Author: SamVerschueren

index.js

@@ -111,9 +111,16 @@ const exec = (str, cols, opts) => {
 		const i = item[0];
 		const word = item[1];
 
+		rows[rows.length - 1] = options.trim === false ? rows[rows.length - 1] : rows[rows.length - 1].trim();
 		let rowLength = stringWidth(rows[rows.length - 1]);
 
-		if (rowLength) {
+		if (rowLength || word === '') {
+			if (rowLength === cols && options.wordWrap === false) {
+				// If we start with a new word but the current row length equals the length of the columns, add a new row
+				rows.push('');
+				rowLength = 0;
+			}
+
 			rows[rows.length - 1] += ' ';
 			rowLength++;
 		}
@@ -145,7 +152,7 @@ const exec = (str, cols, opts) => {
 		rows[rows.length - 1] += word;
 	}
 
-	pre = rows.map(x => x.trim()).join('\n');
+	pre = rows.map(r => options.trim === false ? r : r.trim()).join('\n');
 
 	for (const item of Array.from(pre).entries()) {
 		const i = item[0];

readme.md

@@ -61,6 +61,13 @@ Default: `true`
 
 By default, an attempt is made to split words at spaces, ensuring that they don't extend past the configured columns. If wordWrap is `false`, each column will instead be completely filled splitting words as necessary.
 
+##### trim
+
+Type: `boolean`<br>
+Default: `true`
+
+Whitespace on all lines is removed by default. Set this option to `false` if you don't want to trim.
+
 
 ## Related
 

test.js

@@ -86,6 +86,14 @@ test('no word-wrapping', t => {
 	t.is(res3, 'The q\nuick\nbrown\n\u001B[31mfox j\u001B[39m\n\u001B[31mumped\u001B[39m\n\u001B[31mover\u001B[39m\n\u001B[31m\u001B[39mthe l\nazy \u001B[32md\u001B[39m\n\u001B[32mog an\u001B[39m\n\u001B[32md the\u001B[39m\n\u001B[32mn ran\u001B[39m\n\u001B[32maway\u001B[39m\n\u001B[32mwith\u001B[39m\n\u001B[32mthe u\u001B[39m\n\u001B[32mnicor\u001B[39m\n\u001B[32mn.\u001B[39m');
 });
 
+test('no word-wrapping and no trimming', t => {
+	const res = m(fixture3, 13, {wordWrap: false, trim: false});
+	t.is(res, '12345678\n901234567890 \n12345');
+
+	const res2 = m(fixture, 5, {wordWrap: false, trim: false});
+	t.is(res2, 'The q\nuick \nbrown\n \u001B[31mfox \u001B[39m\njumpe\nd ove\nr \u001B[39mthe\n lazy\n \u001B[32mdog \u001B[39m\nand t\nhen r\nan aw\nay wi\nth th\ne uni\ncorn.\u001B[39m');
+});
+
 test('supports fullwidth characters', t => {
 	t.is(m('안녕하세', 4, {hard: true}), '안녕\n하세');
 });