Tickables extension

Author: Guillemsc

Runtime/Di/Extensions/DiTickablesExtensions.cs

@@ -1,6 +1,7 @@
 using Juce.Core.Di.Builder;
 using Juce.Core.Tickable;
 using Juce.CoreUnity.Tickables;
+using System;
 
 namespace JuceUnity.Core.Di.Extensions
 {
@@ -25,5 +26,30 @@ public static IDiBindingActionBuilder<T> LinkToTickablesService<T>(this IDiBindi
 
             return actionBuilder;
         }
+
+        public static IDiBindingActionBuilder<T> LinkToTickablesService<T>(this IDiBindingActionBuilder<T> actionBuilder, Func<T, Action> func)
+        {
+            CallbackTickable callbackTickable = null;
+
+            actionBuilder.WhenInit((c, o) =>
+            {
+                Action action = func.Invoke(o);
+
+                callbackTickable = new CallbackTickable(action);
+
+                ITickablesService tickablesService = c.Resolve<ITickablesService>();
+
+                tickablesService.Add(callbackTickable);
+            });
+
+            actionBuilder.WhenDispose((c, o) =>
+            {
+                ITickablesService tickablesService = c.Resolve<ITickablesService>();
+
+                tickablesService.Remove(callbackTickable);
+            });
+
+            return actionBuilder;
+        }
     }
 }