From 196b3bf0c564eb46d73231533115c98fcfa0f9c5 Mon Sep 17 00:00:00 2001
From: unknown <bartnikjack@gmail.com>
Date: Sun, 24 May 2020 14:22:41 +0100
Subject: [PATCH] added custom words button

---
 public/index.html   |  3 ++-
 public/js/script.js | 17 ++++++++++++++---
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/public/index.html b/public/index.html
index 6c1a7f301..dfaf6d00d 100644
--- a/public/index.html
+++ b/public/index.html
@@ -131,6 +131,7 @@
             <div class="button active" wordCount="50" tabindex="2">50</div>
             <div class="button" wordCount="100" tabindex="2">100</div>
             <div class="button" wordCount="200" tabindex="2">200</div>
+            <div class="button" wordCount="custom" tabindex="2"><i class="fas fa-tools"></i></div>
           </div>
         </div>
         <div class="group time hidden">
@@ -140,7 +141,7 @@
             <div class="button active" timeConfig="30" tabindex="2">30</div>
             <div class="button" timeConfig="60" tabindex="2">60</div>
             <div class="button" timeConfig="120" tabindex="2">120</div>
-            <div class="button" timeConfig="custom" tabindex="2"><i class="far fa-clock"></i></div>
+            <div class="button" timeConfig="custom" tabindex="2"><i class="fas fa-tools"></i></div>
           </div>
         </div>
         <div class="group customText hidden">
diff --git a/public/js/script.js b/public/js/script.js
index 8ed0eae0b..39deb1f7d 100644
--- a/public/js/script.js
+++ b/public/js/script.js
@@ -657,9 +657,13 @@ function changeCustomText() {
 }
 
 function changeWordCount(wordCount) {
+  wordCount = parseInt(wordCount);
   changeMode("words");
-  config.words = parseInt(wordCount);
+  config.words = wordCount;
   $("#top .config .wordCount .button").removeClass("active");
+  if(![10,25,50,100,200].includes(wordCount)){
+    wordCount = "custom";
+  }
   $("#top .config .wordCount .button[wordCount='" + wordCount + "']").addClass(
     "active"
   );
@@ -861,8 +865,15 @@ $(document).on("click", "#top .logo", (e) => {
 });
 
 $(document).on("click", "#top .config .wordCount .button", (e) => {
-  wrd = e.currentTarget.innerHTML;
-  changeWordCount(wrd);
+  wrd = $(e.currentTarget).attr('wordCount');
+  if(wrd == "custom"){
+    let newWrd = prompt('Custom word amount');
+    if(newWrd !== null && !isNaN(newWrd) && newWrd > 0){
+      changeWordCount(newWrd);
+    }
+  }else{
+    changeWordCount(wrd);
+  }
 });
 
 $(document).on("click", "#top .config .time .button", (e) => {