1 files changed, 133 insertions, 19 deletions
diff --git a/apps/web/src/lib/timeline/track-element-update.ts b/apps/web/src/lib/timeline/track-element-update.ts
index 19765923..f1c2028e 100644
--- a/apps/web/src/lib/timeline/track-element-update.ts
+++ b/apps/web/src/lib/timeline/track-element-update.ts
@@ -1,33 +1,147 @@
-import type { TimelineElement, TimelineTrack } from "@/lib/timeline";
+import type { SceneTracks, TimelineElement, TimelineTrack } from "@/lib/timeline";
 
-export function updateElementInTracks({
+export function findTrackInSceneTracks({
 	tracks,
 	trackId,
+}: {
+	tracks: SceneTracks;
+	trackId: string;
+}): TimelineTrack | null {
+	if (tracks.main.id === trackId) {
+		return tracks.main;
+	}
+
+	return (
+		tracks.overlay.find((track) => track.id === trackId) ??
+		tracks.audio.find((track) => track.id === trackId) ??
+		null
+	);
+}
+
+export function updateTrackInSceneTracks({
+	tracks,
+	trackId,
+	update,
+}: {
+	tracks: SceneTracks;
+	trackId: string;
+	update: <TTrack extends TimelineTrack>(track: TTrack) => TTrack;
+}): SceneTracks {
+	if (tracks.main.id === trackId) {
+		return {
+			...tracks,
+			main: update(tracks.main),
+		};
+	}
+
+	const overlayTrackIndex = tracks.overlay.findIndex((track) => track.id === trackId);
+	if (overlayTrackIndex >= 0) {
+		return {
+			...tracks,
+			overlay: tracks.overlay.map((track, index) =>
+				index === overlayTrackIndex ? update(track) : track,
+			),
+		};
+	}
+
+	const audioTrackIndex = tracks.audio.findIndex((track) => track.id === trackId);
+	if (audioTrackIndex >= 0) {
+		return {
+			...tracks,
+			audio: tracks.audio.map((track, index) =>
+				index === audioTrackIndex ? update(track) : track,
+			),
+		};
+	}
+
+	return tracks;
+}
+
+function updateElementInTrack<TTrack extends TimelineTrack>({
+	track,
 	elementId,
 	update,
 	elementPredicate,
 }: {
-	tracks: TimelineTrack[];
-	trackId: string;
+	track: TTrack;
 	elementId: string;
 	update: (element: TimelineElement) => TimelineElement;
 	elementPredicate?: (element: TimelineElement) => boolean;
-}): TimelineTrack[] {
-	return tracks.map((track) => {
-		if (track.id !== trackId) {
-			return track;
+}): TTrack {
+	const nextElements = track.elements.map((element) => {
+		if (element.id !== elementId) {
+			return element;
+		}
+		if (elementPredicate && !elementPredicate(element)) {
+			return element;
 		}
+		return update(element);
+	});
 
-		const nextElements = track.elements.map((element) => {
-			if (element.id !== elementId) {
-				return element;
-			}
-			if (elementPredicate && !elementPredicate(element)) {
-				return element;
-			}
-			return update(element);
-		});
+	return {
+		...track,
+		elements: nextElements,
+	} as TTrack;
+}
 
-		return { ...track, elements: nextElements } as TimelineTrack;
-	});
+export function updateElementInSceneTracks({
+	tracks,
+	trackId,
+	elementId,
+	update,
+	elementPredicate,
+}: {
+	tracks: SceneTracks;
+	trackId: string;
+	elementId: string;
+	update: (element: TimelineElement) => TimelineElement;
+	elementPredicate?: (element: TimelineElement) => boolean;
+}): SceneTracks {
+	if (tracks.main.id === trackId) {
+		return {
+			...tracks,
+			main: updateElementInTrack({
+				track: tracks.main,
+				elementId,
+				update,
+				elementPredicate,
+			}),
+		};
+	}
+
+	const overlayTrackIndex = tracks.overlay.findIndex((track) => track.id === trackId);
+	if (overlayTrackIndex >= 0) {
+		return {
+			...tracks,
+			overlay: tracks.overlay.map((track, index) =>
+				index === overlayTrackIndex
+					? updateElementInTrack({
+							track,
+							elementId,
+							update,
+							elementPredicate,
+						})
+					: track,
+			),
+		};
+	}
+
+	const audioTrackIndex = tracks.audio.findIndex((track) => track.id === trackId);
+	if (audioTrackIndex >= 0) {
+		return {
+			...tracks,
+			audio: tracks.audio.map((track, index) =>
+				index === audioTrackIndex
+					? updateElementInTrack({
+							track,
+							elementId,
+							update,
+							elementPredicate,
+						})
+					: track,
+			),
+		};
+	}
+
+	return tracks;
 }