fix: useEffect ran twice in StrictMode

refs: #14
maraisr · Sep 20, 2022 · 3b9fda6 · 3b9fda6
1 parent ad55c6f
commit 3b9fda6
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/components/Tools.tsx b/src/components/Tools.tsx
@@ -59,8 +59,15 @@ export const Tools = () => {
 
 	// Avoid some "getting ready" states
 	useEffect(() => {
-		api.once(STORY_RENDERED, () => void setReady(true));
-	}, []);
+		const handler = () => {
+			api.off(STORY_RENDERED, handler); // api.once doesnt work in strict mode here
+			setReady(true);
+		};
+
+		api.on(STORY_RENDERED, handler);
+
+		return () => api.off(STORY_RENDERED, handler);
+	}, [api]);
 
 	const disabled =
 		typeof parameters.disable === 'boolean' ? parameters.disable : false;