Refactor disableRunButton function

Aaron Imming · Aaron Imming · commit 4363e587c0de · 2019-07-09T11:50:26.000+08:00
diff --git a/src/botPage/view/blockly/blocks/shared.js b/src/botPage/view/blockly/blocks/shared.js
@@ -410,6 +410,21 @@ export const getPredictionForContracts = (contracts, selectedContractType) => {
     return predictionRange;
 };
 
-export const disableRunButton = isDisabled => {
-    $('#runButton, #summaryRunButton').attr('disabled', isDisabled);
+export const disableRunButton = should_disable => {
+    const el_run_buttons = document.querySelectorAll('#runButton, #summaryRunButton');
+    const is_running = globalObserver.getState('is_running');
+
+    el_run_buttons.forEach(el_run_button => {
+        if (is_running) {
+            if (should_disable) {
+                el_run_button.setAttributeNode(document.createAttribute('disabled'));
+            } else {
+                // Do not enable. The bot is running.
+            }
+        } else if (should_disable) {
+            el_run_button.setAttributeNode(document.createAttribute('disabled'));
+        } else {
+            el_run_button.removeAttribute('disabled');
+        }
+    });
 };
