1 files changed, 33 insertions, 29 deletions
diff --git a/apps/web/src/timeline/index.ts b/apps/web/src/timeline/index.ts
index 5ee079b1..21944561 100644
--- a/apps/web/src/timeline/index.ts
+++ b/apps/web/src/timeline/index.ts
@@ -1,29 +1,33 @@
-import type { SceneTracks } from "./types";
-
-export * from "./types";
-export * from "./drag";
-export * from "./track-capabilities";
-export * from "./track-element-update";
-export * from "./element-utils";
-export * from "./audio-separation";
-export * from "./zoom-utils";
-export * from "./ruler-utils";
-export * from "./pixel-utils";
-
-export function calculateTotalDuration({
-	tracks,
-}: {
-	tracks: SceneTracks;
-}): number {
-	const orderedTracks = [...tracks.overlay, tracks.main, ...tracks.audio];
-	if (orderedTracks.length === 0) return 0;
-
-	const trackEndTimes = orderedTracks.map((track) =>
-		track.elements.reduce((maxEnd, element) => {
-			const elementEnd = element.startTime + element.duration;
-			return Math.max(maxEnd, elementEnd);
-		}, 0),
-	);
-
-	return Math.max(...trackEndTimes, 0);
-}
+import { addMediaTime, type MediaTime, ZERO_MEDIA_TIME } from "@/wasm";
+import type { SceneTracks } from "./types";
+
+export * from "./types";
+export * from "./drag";
+export * from "./track-capabilities";
+export * from "./track-element-update";
+export * from "./element-utils";
+export * from "./audio-separation";
+export * from "./zoom-utils";
+export * from "./ruler-utils";
+export * from "./pixel-utils";
+
+export function calculateTotalDuration({
+	tracks,
+}: {
+	tracks: SceneTracks;
+}): MediaTime {
+	const orderedTracks = [...tracks.overlay, tracks.main, ...tracks.audio];
+	if (orderedTracks.length === 0) return ZERO_MEDIA_TIME;
+
+	let maxEnd: MediaTime = ZERO_MEDIA_TIME;
+	for (const track of orderedTracks) {
+		for (const element of track.elements) {
+			const elementEnd = addMediaTime({
+				a: element.startTime,
+				b: element.duration,
+			});
+			if (elementEnd > maxEnd) maxEnd = elementEnd;
+		}
+	}
+	return maxEnd;
+}