wip

Author: noituri

ts/smelter-web-client/src/hooks/useSmelter.ts

@@ -1,19 +1,58 @@
-import { useEffect, useRef, useState } from 'react';
+import { useEffect, useState } from 'react';
 import Smelter from '../smelter/live';
 import type { SmelterInstanceOptions } from '../manager';
 
+export function useSmelter2(options: SmelterInstanceOptions): Smelter | undefined {
+  const [smelter, setSmelter] = useState<Smelter>();
+  const [_, setPromiseQueue] = useState<Promise<void>>(Promise.resolve());
+
+  useEffect(() => {
+    const smelter = new Smelter(options);
+
+    // TODO(noituri): Restart smelter instance
+    let cancel = false;
+    setPromiseQueue((promise) => promise.finally(async () => {
+      await smelter.init();
+      await smelter.start();
+      if (!cancel) {
+        setSmelter(smelter);
+      }
+    }));
+
+    return () => {
+      cancel = true;
+      setPromiseQueue((promise) => promise.finally(async () => {
+        await smelter.terminate();
+      }));
+    }
+  }, [options]);
+
+  return smelter;
+}
+
 export function useSmelter(options: SmelterInstanceOptions): Smelter | undefined {
   const [smelter, setSmelter] = useState<Smelter>();
-  const cleanUpPromise = useRef<Promise<void>>();
+  const [cleanupPromise, setCleanupPromise] = useState<Promise<void>>();
+  const [instanceOptions, setInstanceOptions] = useState<SmelterInstanceOptions>();
 
   useEffect(() => {
+    void (async () => {
+      await cleanupPromise;
+      setInstanceOptions(options);
+    })();
+
+  }, [options, cleanupPromise]);
+
+  useEffect(() => {
+    if (!instanceOptions) {
+      return;
+    }
+
     const smelter = new Smelter(options);
 
+    // TODO(noituri): Restart smelter instance
     let cancel = false;
-    // TODO(noituri): Add smelter.restart()
     const promise = (async () => {
-      await cleanUpPromise.current;
-
       await smelter.init();
       await smelter.start();
       if (!cancel) {
@@ -23,12 +62,13 @@ export function useSmelter(options: SmelterInstanceOptions): Smelter | undefined
 
     return () => {
       cancel = true;
-      cleanUpPromise.current = (async () => {
-        await promise.catch(() => {});
+      setCleanupPromise((prevCleanup) => (async () => {
+        await prevCleanup;
+        await promise.catch(() => { });
         await smelter.terminate();
-      })();
-    };
-  }, [options]);
+      })());
+    }
+  }, [instanceOptions]);
 
   return smelter;
 }