From 9dc6818fd76ac776e3c8693bbd18caf667a26bdb Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Sat, 29 Oct 2022 17:10:46 +0800
Subject: [PATCH] ui: scratchpad: use seperate cooldown for pretest and submit

---
 .../scratchpad/ScratchpadToolbarContainer.jsx | 16 ++++----
 .../components/scratchpad/reducers/ui.ts      | 38 +++++++++++--------
 packages/ui-default/package.json              |  2 +-
 3 files changed, 33 insertions(+), 23 deletions(-)

diff --git a/packages/ui-default/components/scratchpad/ScratchpadToolbarContainer.jsx b/packages/ui-default/components/scratchpad/ScratchpadToolbarContainer.jsx
index a18dd3d5d..e774f1155 100644
--- a/packages/ui-default/components/scratchpad/ScratchpadToolbarContainer.jsx
+++ b/packages/ui-default/components/scratchpad/ScratchpadToolbarContainer.jsx
@@ -18,8 +18,8 @@ const mapStateToProps = (state) => ({
   recordsVisible: state.ui.records.visible,
   isPosting: state.ui.isPosting,
   isRunning: state.pretest.isRunning,
-  isWaiting: state.ui.isWaiting,
-  waitSec: state.ui.waitSec,
+  pretestWaitSec: state.ui.pretestWaitSec,
+  submitWaitSec: state.ui.submitWaitSec,
   editorLang: state.editor.lang,
   editorCode: state.editor.code,
   pretestInput: state.pretest.input,
@@ -96,7 +96,9 @@ export default connect(mapStateToProps, mapDispatchToProps)(class ScratchpadTool
   }
 
   componentDidUpdate() {
-    if (this.props.waitSec > 0) setTimeout(() => this.props.tick(), 1000);
+    if (this.props.pretestWaitSec > 0 || this.props.submitWaitSec > 0) {
+      setTimeout(() => this.props.tick(), 1000);
+    }
   }
 
   render() {
@@ -109,7 +111,7 @@ export default connect(mapStateToProps, mapDispatchToProps)(class ScratchpadTool
       <Toolbar>
         {canUsePretest && (
           <ToolbarButton
-            disabled={this.props.isPosting || this.props.isRunning || this.props.isWaiting}
+            disabled={this.props.isPosting || this.props.isRunning || this.props.pretestWaitSec}
             className="scratchpad__toolbar__pretest"
             onClick={() => this.props.postPretest(this.props)}
             data-global-hotkey="f9"
@@ -119,11 +121,11 @@ export default connect(mapStateToProps, mapDispatchToProps)(class ScratchpadTool
             {' '}
             {i18n('Run Pretest')}
             {' '}
-            {this.props.isWaiting ? `(${this.props.waitSec}s)` : '(F9)'}
+            {this.props.pretestWaitSec ? `(${this.props.pretestWaitSec}s)` : '(F9)'}
           </ToolbarButton>
         )}
         <ToolbarButton
-          disabled={this.props.isPosting || this.props.isWaiting}
+          disabled={this.props.isPosting || this.props.submitWaitSec}
           className="scratchpad__toolbar__submit"
           onClick={() => this.props.postSubmit(this.props)}
           data-global-hotkey="f10"
@@ -133,7 +135,7 @@ export default connect(mapStateToProps, mapDispatchToProps)(class ScratchpadTool
           {' '}
           {i18n('Submit Solution')}
           {' '}
-          {this.props.isWaiting ? `(${this.props.waitSec}s)` : '(F10)'}
+          {this.props.submitWaitSec ? `(${this.props.submitWaitSec}s)` : '(F10)'}
         </ToolbarButton>
         <ToolbarButton
           data-global-hotkey="alt+q"
diff --git a/packages/ui-default/components/scratchpad/reducers/ui.ts b/packages/ui-default/components/scratchpad/reducers/ui.ts
index 4e1022527..b4943a077 100644
--- a/packages/ui-default/components/scratchpad/reducers/ui.ts
+++ b/packages/ui-default/components/scratchpad/reducers/ui.ts
@@ -18,8 +18,9 @@ export default function reducer(state = {
     isLoading: false,
   },
   isPosting: false,
-  waitSec: 0,
-  isWaiting: false,
+  pretestWaitSec: 0,
+  submitWaitSec: 0,
+  lastTick: 0,
   activePage: 'problem',
 }, action: any = {}) {
   switch (action.type) {
@@ -65,15 +66,20 @@ export default function reducer(state = {
     case 'SCRATCHPAD_POST_PRETEST_FULFILLED':
     case 'SCRATCHPAD_POST_SUBMIT_FULFILLED': {
       Notification.success(i18n('Submitted.'));
-      if (action.type === 'SCRATCHPAD_POST_SUBMIT_FULFILLED' && UiContext.canViewRecord) {
-        state.records.visible = true;
-      }
-      return {
-        ...state,
-        isPosting: false,
-        waitSec: 5,
-        isWaiting: true,
-      };
+      return (action.type === 'SCRATCHPAD_POST_SUBMIT_FULFILLED' && UiContext.canViewRecord)
+        ? {
+          ...state,
+          records: {
+            ...state.records,
+            visible: true,
+          },
+          isPosting: false,
+          submitWaitSec: 8,
+        } : {
+          ...state,
+          isPosting: false,
+          pretestWaitSec: 5,
+        };
     }
     case 'SCRATCHPAD_POST_PRETEST_REJECTED':
     case 'SCRATCHPAD_POST_SUBMIT_REJECTED': {
@@ -81,15 +87,17 @@ export default function reducer(state = {
       return {
         ...state,
         isPosting: false,
-        waitSec: 5,
-        isWaiting: true,
+        pretestWaitSec: 3,
+        submitWaitSec: 3,
       };
     }
     case 'SCRATCHPAD_WAITING_TICK': {
+      if (Date.now() - state.lastTick < 950) return state;
       return {
         ...state,
-        waitSec: state.waitSec - 1,
-        isWaiting: state.waitSec > 1,
+        lastTick: Date.now(),
+        pretestWaitSec: Math.max(state.pretestWaitSec - 1, 0),
+        submitWaitSec: Math.max(state.submitWaitSec - 1, 0),
       };
     }
     case 'SCRATCHPAD_RECORDS_LOAD_SUBMISSIONS_PENDING': {
diff --git a/packages/ui-default/package.json b/packages/ui-default/package.json
index ec3611dce..724cff267 100644
--- a/packages/ui-default/package.json
+++ b/packages/ui-default/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@hydrooj/ui-default",
-  "version": "4.42.13",
+  "version": "4.42.14",
   "author": "undefined <i@undefined.moe>",
   "license": "AGPL-3.0",
   "main": "index.ts",