feat: OnDepsChange::Rerun for use_animation

marc2332 · Jan 12, 2025 · ebeffe1 · ebeffe1
1 parent d96f0b4
commit ebeffe1
Showing 1 changed file with 29 additions and 2 deletions.
diff --git a/crates/hooks/src/use_animation.rs b/crates/hooks/src/use_animation.rs
@@ -497,6 +497,7 @@ pub enum OnDepsChange {
     #[default]
     Reset,
     Finish,
+    Rerun,
 }
 
 /// Animate your elements. Use [`use_animation`] to use this.
@@ -776,6 +777,23 @@ pub fn use_animation<Animated: AnimatedValue>(
         }
     });
 
+    use_memo(move || {
+        let context = context.read();
+        if *has_run_yet.peek()
+        {
+            match context.conf.on_deps_change {
+                OnDepsChange::Finish => {
+                    animation.finish()
+                }
+                OnDepsChange::Rerun => {
+                    let last_direction = *animation.last_direction.peek();
+                    animation.run(last_direction);
+                }
+                _ => {}
+            }
+        }
+    });
+
     animation
 }
 
@@ -810,8 +828,17 @@ where
 
     use_memo(move || {
         let context = context.read();
-        if *has_run_yet.peek() && context.conf.on_deps_change == OnDepsChange::Finish {
-            animation.finish()
+        if *has_run_yet.peek()
+        {
+            match context.conf.on_deps_change {
+                OnDepsChange::Finish => {
+                    animation.finish()
+                }
+                OnDepsChange::Rerun => {
+                    animation.run(*animation.last_direction.peek());
+                }
+                _ => {}
+            }
         }
     });