codebase overhaul (#697)

3 files changed, 148 insertions, 0 deletions

diff --git a/apps/web/src/lib/commands/media/add-media-asset.ts b/apps/web/src/lib/commands/media/add-media-asset.ts
new file mode 100644
index 00000000..d947cb97
--- /dev/null
+++ b/apps/web/src/lib/commands/media/add-media-asset.ts
@@ -0,0 +1,61 @@
+import { Command } from "@/lib/commands/base-command";
+import { EditorCore } from "@/core";
+import type { MediaAsset } from "@/types/assets";
+import { generateUUID } from "@/utils/id";
+import { storageService } from "@/services/storage/storage-service";
+
+export class AddMediaAssetCommand extends Command {
+	private assetId: string;
+	private savedAssets: MediaAsset[] | null = null;
+	private createdAsset: MediaAsset | null = null;
+
+	constructor(
+		private projectId: string,
+		private asset: Omit<MediaAsset, "id">,
+	) {
+		super();
+		this.assetId = generateUUID();
+	}
+
+	execute(): void {
+		const editor = EditorCore.getInstance();
+		this.savedAssets = [...editor.media.getAssets()];
+
+		this.createdAsset = {
+			...this.asset,
+			id: this.assetId,
+		};
+
+		editor.media.setAssets({
+			assets: [...this.savedAssets, this.createdAsset],
+		});
+
+		storageService
+			.saveMediaAsset({
+				projectId: this.projectId,
+				mediaAsset: this.createdAsset,
+			})
+			.catch((error) => {
+				console.error("Failed to save media item:", error);
+			});
+	}
+
+	undo(): void {
+		if (this.savedAssets) {
+			const editor = EditorCore.getInstance();
+			editor.media.setAssets({ assets: this.savedAssets });
+
+			if (this.createdAsset) {
+				storageService
+					.deleteMediaAsset({ projectId: this.projectId, id: this.assetId })
+					.catch((error) => {
+						console.error("Failed to delete media item on undo:", error);
+					});
+			}
+		}
+	}
+
+	getAssetId(): string {
+		return this.assetId;
+	}
+}
diff --git a/apps/web/src/lib/commands/media/index.ts b/apps/web/src/lib/commands/media/index.ts
new file mode 100644
index 00000000..466222f2
--- /dev/null
+++ b/apps/web/src/lib/commands/media/index.ts
@@ -0,0 +1,2 @@
+export { AddMediaAssetCommand } from "./add-media-asset";
+export { RemoveMediaAssetCommand } from "./remove-media-asset";
diff --git a/apps/web/src/lib/commands/media/remove-media-asset.ts b/apps/web/src/lib/commands/media/remove-media-asset.ts
new file mode 100644
index 00000000..aa1a5775
--- /dev/null
+++ b/apps/web/src/lib/commands/media/remove-media-asset.ts
@@ -0,0 +1,85 @@
+import { Command } from "@/lib/commands/base-command";
+import { EditorCore } from "@/core";
+import type { MediaAsset } from "@/types/assets";
+import { storageService } from "@/services/storage/storage-service";
+import { videoCache } from "@/services/media/video-cache";
+import { hasMediaId } from "@/lib/timeline/element-utils";
+import type { TimelineTrack } from "@/types/timeline";
+
+export class RemoveMediaAssetCommand extends Command {
+	private savedAssets: MediaAsset[] | null = null;
+	private savedTracks: TimelineTrack[] | null = null;
+	private removedAsset: MediaAsset | null = null;
+
+	constructor(
+		private projectId: string,
+		private assetId: string,
+	) {
+		super();
+	}
+
+	execute(): void {
+		const editor = EditorCore.getInstance();
+		const assets = editor.media.getAssets();
+
+		this.savedAssets = [...assets];
+		this.savedTracks = editor.timeline.getTracks();
+
+		this.removedAsset =
+			assets.find((media) => media.id === this.assetId) ?? null;
+
+		if (!this.removedAsset) {
+			console.error("Media asset not found:", this.assetId);
+			return;
+		}
+
+		videoCache.clearVideo({ mediaId: this.assetId });
+
+		editor.media.setAssets({
+			assets: assets.filter((media) => media.id !== this.assetId),
+		});
+
+		const elementsToRemove: Array<{ trackId: string; elementId: string }> = [];
+
+		for (const track of this.savedTracks) {
+			for (const element of track.elements) {
+				if (hasMediaId(element) && element.mediaId === this.assetId) {
+					elementsToRemove.push({ trackId: track.id, elementId: element.id });
+				}
+			}
+		}
+
+		if (elementsToRemove.length > 0) {
+			editor.timeline.deleteElements({ elements: elementsToRemove });
+		}
+
+		storageService
+			.deleteMediaAsset({ projectId: this.projectId, id: this.assetId })
+			.catch((error) => {
+				console.error("Failed to delete media item:", error);
+			});
+	}
+
+	undo(): void {
+		const editor = EditorCore.getInstance();
+
+		if (this.savedAssets) {
+			editor.media.setAssets({ assets: this.savedAssets });
+		}
+
+		if (this.savedTracks) {
+			editor.timeline.updateTracks(this.savedTracks);
+		}
+
+		if (this.removedAsset) {
+			storageService
+				.saveMediaAsset({
+					projectId: this.projectId,
+					mediaAsset: this.removedAsset,
+				})
+				.catch((error) => {
+					console.error("Failed to restore media item on undo:", error);
+				});
+		}
+	}
+}