Attach session for run all button

C2DH · Jul 26, 2024 · 1d75da3 · 1d75da3
1 parent 103c05b
commit 1d75da3
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/components/ArticleV3/ArticleThebeSession.js b/src/components/ArticleV3/ArticleThebeSession.js
@@ -1,4 +1,4 @@
-import React from 'react'
+import React, { useEffect } from 'react'
 import { useArticleThebe } from './ArticleThebeProvider'
 import ArticleThebeSessionButton, {
   StatusReady,
@@ -8,8 +8,9 @@ import ArticleThebeSessionButton, {
 import { useExecutionScope } from './ExecutionScope'
 
 const ArticleThebeSession = ({ debug = false, kernelName }) => {
-  const { starting, ready, connectAndStart, shutdown, restart } = useArticleThebe()
+  const { starting, ready, connectAndStart, shutdown, restart, session } = useArticleThebe()
   const executeAll = useExecutionScope((state) => state.executeAll)
+  const attachSession = useExecutionScope((state) => state.attachSession);
   let status = StatusIdle
 
   if (ready) {
@@ -18,6 +19,11 @@ const ArticleThebeSession = ({ debug = false, kernelName }) => {
     status = StatusPreparing
   }
 
+  useEffect(() => {
+    if (ready)
+      attachSession(session);
+  }, [ready]);
+
   const handleSession = () => {
     if (status === StatusIdle) {
       connectAndStart()