minor

DiTso · Nov 1, 2016 · 55e6f92 · 55e6f92
1 parent 1ee7f9f
commit 55e6f92
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/dist/shave.js b/dist/shave.js
@@ -34,10 +34,11 @@ function shave(target, maxHeight, opts) {
     if (el.offsetHeight < maxHeight) continue;
 
     var fullText = el.textContent;
-    var words = fullText.split(' ');
-    if (!spaces) words = el.textContent;
+    var words = spaces ? fullText.split(' ') : fullText;
+
     // If 0 or 1 words, we're done
     if (words.length < 2) continue;
+
     // Binary search for number of words which can fit in allotted height
     var max = words.length - 1;
     var min = 0;
@@ -68,4 +69,4 @@ if (plugin) {
 
 return shave;
 
-})));
+})));
diff --git a/src/shave.js b/src/shave.js
@@ -29,10 +29,11 @@ export default function shave(target, maxHeight, opts) {
     if (el.offsetHeight < maxHeight) continue;
 
     const fullText = el.textContent;
-    let words = fullText.split(' ');
-    if (!spaces) words = el.textContent;
+    const words = spaces ? fullText.split(' ') : fullText;
+
     // If 0 or 1 words, we're done
     if (words.length < 2) continue;
+
     // Binary search for number of words which can fit in allotted height
     let max = words.length - 1;
     let min = 0;