Effector: Thing -> IAutoUpdate

OnGoTeam · May 10, 2019 · 3b08a57 · 3b08a57
1 parent 3431688
commit 3b08a57
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 12 deletions.
diff --git a/ogtdglib/src/Effects/Effect.cs b/ogtdglib/src/Effects/Effect.cs
@@ -52,7 +52,7 @@ public void Step()
         /// </summary>
         /// <param name="effect"></param>
         /// <returns></returns>
-        public static Thing Apply(Effect effect)
+        public static IAutoUpdate Apply(Effect effect)
         {
             return new Effector(effect);
         }

diff --git a/ogtdglib/src/Effects/Effector.cs b/ogtdglib/src/Effects/Effector.cs
@@ -4,33 +4,26 @@
 namespace OgtDgLib.Effects
 {
     [PublicAPI]
-    internal sealed class Effector : Thing
+    internal sealed class Effector : IAutoUpdate
     {
         public StateBinding EffectBinding = new StateBinding(nameof(ItsEffect));
         public Effect ItsEffect;
 
         internal Effector(Effect itsEffect)
         {
             ItsEffect = itsEffect;
-            owner = ItsEffect.Owner;
-        }
-
-        public override void Initialize()
-        {
-            Step();
+            AutoUpdatables.Add(this);
         }
 
         private void Step()
         {
             ItsEffect.Step();
         }
 
-        public override void Update()
+        public void Update()
         {
             if (!ItsEffect.Decayed)
-                ItsEffect.Step();
-            else
-                Level.Remove(this);
+                Step();
         }
     }
 }