Close the SSE connection whe stopping the process client side

moshell-lang · Nov 30, 2023 · 093bb80 · 093bb80
1 parent d78b0f3
commit 093bb80
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/frontend/src/playground.tsx b/frontend/src/playground.tsx
@@ -49,6 +49,15 @@ export function Playground() {
     });
   };
 
+  const handleStopClick = () =>
+    setSSE((current) => {
+      if (current) {
+        current.close();
+        appendOutputText('Process terminated by user');
+      }
+      return null;
+    });
+
   useEffect(() => {
     localStorage.setItem('editorText', editorText);
   }, [editorText]);
@@ -57,7 +66,7 @@ export function Playground() {
     <>
       <Navbar
         isRunning={sse !== null}
-        onRunClick={handleRunClick}
+        onRunClick={sse === null ? handleRunClick : handleStopClick}
       />
       <Split
         class="d-flex"