summaryrefslogtreecommitdiff
path: root/apps/web/src/hooks/use-editor.ts
blob: ff3e2db13bbbc2ef55434da2d6bd6d20b5391fac (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
import { useCallback, useMemo, useRef, useSyncExternalStore } from "react";
import { EditorCore } from "@/core";

export function useEditor(): EditorCore {
	const editor = useMemo(() => EditorCore.getInstance(), []);
	const versionRef = useRef(0);

	const subscribe = useCallback(
		(onStoreChange: () => void) => {
			const handleStoreChange = () => {
				versionRef.current += 1;
				onStoreChange();
			};

			const unsubscribers = [
				editor.playback.subscribe(handleStoreChange),
				editor.timeline.subscribe(handleStoreChange),
				editor.scenes.subscribe(handleStoreChange),
				editor.project.subscribe(handleStoreChange),
				editor.media.subscribe(handleStoreChange),
				editor.renderer.subscribe(handleStoreChange),
				editor.selection.subscribe(handleStoreChange),
			];

			return () => {
				for (const unsubscribe of unsubscribers) {
					unsubscribe();
				}
			};
		},
		[editor],
	);

	const getSnapshot = useCallback(() => versionRef.current, []);

	useSyncExternalStore(subscribe, getSnapshot, getSnapshot);

	return editor;
}