Add stop on demand

FluffyLabs · Nov 13, 2024 · 167be91 · 167be91
1 parent a2313c6
commit 167be91
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 12 deletions.
diff --git a/src/components/DebuggerControlls/index.tsx b/src/components/DebuggerControlls/index.tsx
@@ -1,4 +1,4 @@
-import { RefreshCcw, Play, StepForward } from "lucide-react";
+import { RefreshCcw, Play, StepForward, Pause } from "lucide-react";
 import { LoadingSpinner } from "../LoadingSpinner";
 import { ProgramLoader } from "../ProgramLoader";
 import { Button } from "../ui/button";
@@ -82,6 +82,10 @@ export const DebuggerControlls = () => {
     dispatch(refreshPageAllWorkers());
   };
 
+  const handlePauseProgram = async () => {
+    dispatch(setIsRunMode(false));
+  };
+
   return (
     <div className="col-span-12 md:col-span-6 max-sm:order-2 flex align-middle max-sm:justify-between mb-3">
       <div className="md:mr-3">
@@ -104,18 +108,32 @@ export const DebuggerControlls = () => {
         <RefreshCcw className="w-3.5 md:mr-1.5" />
         <span className="hidden md:block">Reset</span>
       </Button>
-      <Button
-        className="md:mr-3"
-        onClick={handleRunProgram}
-        disabled={isDebugFinished || !pvmInitialized || isProgramEditMode || isLoading || !!error}
-      >
-        {isLoading ? <LoadingSpinner className="w-3.5 md:mr-1.5" size={20} /> : <Play className="w-3.5 md:mr-1.5" />}
-        <span className="hidden md:block">Run</span>
-      </Button>
+      {!isDebugFinished && isRunMode ? (
+        <Button className="md:mr-3" onClick={handlePauseProgram}>
+          <Pause className="w-3.5 md:mr-1.5" />
+          <span className="hidden md:block">Stop</span>
+        </Button>
+      ) : (
+        <Button
+          className="md:mr-3"
+          onClick={handleRunProgram}
+          disabled={isDebugFinished || !pvmInitialized || isProgramEditMode || isLoading || !!error}
+        >
+          {isLoading ? <LoadingSpinner className="w-3.5 md:mr-1.5" size={20} /> : <Play className="w-3.5 md:mr-1.5" />}
+          <span className="hidden md:block">Run</span>
+        </Button>
+      )}
       <Button
         className="md:mr-3"
         onClick={onNext}
-        disabled={isDebugFinished || !pvmInitialized || isProgramEditMode || isLoading || !!error}
+        disabled={
+          (!isDebugFinished && isRunMode) ||
+          isDebugFinished ||
+          !pvmInitialized ||
+          isProgramEditMode ||
+          isLoading ||
+          !!error
+        }
       >
         {isLoading ? (
           <LoadingSpinner className="w-3.5 md:mr-1.5" size={20} />

diff --git a/src/store/workers/workersSlice.ts b/src/store/workers/workersSlice.ts
@@ -1,7 +1,7 @@
 import { createSlice, createAsyncThunk, isRejected } from "@reduxjs/toolkit";
 import { RootState } from "@/store";
 import { CurrentInstruction, ExpectedState } from "@/types/pvm.ts";
-import { setIsDebugFinished, setIsStepMode } from "@/store/debugger/debuggerSlice.ts";
+import { setIsDebugFinished, setIsRunMode, setIsStepMode } from "@/store/debugger/debuggerSlice.ts";
 import PvmWorker from "@/packages/web-worker/worker?worker&inline";
 import { SupportedLangs } from "@/packages/web-worker/utils.ts";
 import { virtualTrapInstruction } from "@/utils/virtualTrapInstruction.ts";
@@ -256,7 +256,11 @@ export const continueAllWorkers = createAsyncThunk("workers/continueAllWorkers",
       dispatch(setIsDebugFinished(true));
     }
 
-    if (allSame && !allFinished && allRunning && !anyBreakpoint) {
+    if (!allSame) {
+      dispatch(setIsRunMode(false));
+    }
+
+    if (debuggerState.isRunMode && allSame && !allFinished && allRunning && !anyBreakpoint) {
       await stepAllWorkersAgain();
     }
   };