1 files changed, 48 insertions, 0 deletions
diff --git a/rust/crates/commands/src/scene/toggle_bookmark.rs b/rust/crates/commands/src/scene/toggle_bookmark.rs
new file mode 100644
index 00000000..054e1c78
--- /dev/null
+++ b/rust/crates/commands/src/scene/toggle_bookmark.rs
@@ -0,0 +1,48 @@
+use scene::Project;
+
+use crate::bookmarks::{get_frame_time, toggle_bookmark_in_array};
+use crate::command::{Command, CommandError, CommandResult};
+use crate::project_ext::active_scene_mut;
+use time::MediaTime;
+
+pub struct ToggleBookmarkCommand {
+    time: MediaTime,
+    frame_time: MediaTime,
+    saved_bookmarks: Option<Vec<scene::Bookmark>>,
+}
+
+impl ToggleBookmarkCommand {
+    pub fn new(time: MediaTime) -> Self {
+        Self {
+            time,
+            frame_time: MediaTime::ZERO,
+            saved_bookmarks: None,
+        }
+    }
+}
+
+impl Command for ToggleBookmarkCommand {
+    fn execute(
+        &mut self,
+        project: &mut Project,
+    ) -> Result<Option<CommandResult>, CommandError> {
+        let fps = project.settings.fps;
+        let Some(scene) = active_scene_mut(project) else {
+            return Ok(None);
+        };
+
+        self.saved_bookmarks = Some(scene.bookmarks.clone());
+        self.frame_time = get_frame_time(self.time, fps);
+        scene.bookmarks = toggle_bookmark_in_array(&scene.bookmarks, self.frame_time);
+        Ok(None)
+    }
+
+    fn undo(&mut self, project: &mut Project) {
+        let Some(saved) = &self.saved_bookmarks else {
+            return;
+        };
+        if let Some(scene) = active_scene_mut(project) {
+            scene.bookmarks = saved.clone();
+        }
+    }
+}