fix(ecs): make optional relationships possible

1 files changed, 68 insertions, 1 deletions

diff --git a/ecs/src/relationship.rs b/ecs/src/relationship.rs
index d136db4..143b589 100644
--- a/ecs/src/relationship.rs
+++ b/ecs/src/relationship.rs
@@ -10,7 +10,7 @@ use crate::component::{
     FromOptional as FromOptionalComponent,
     FromOptionalMut as FromOptionalMutComponent,
 };
-use crate::lock::{Error as LockError, Lock, ReadGuard};
+use crate::lock::{Error as LockError, Lock, ReadGuard, WriteGuard};
 use crate::system::{ComponentRef, ComponentRefMut};
 use crate::uid::{Kind as UidKind, Uid};
 use crate::World;
@@ -104,6 +104,22 @@ where
     }
 }
 
+impl<'rel_comp, Kind, ComponentT> FromOptionalMutComponent<'rel_comp>
+    for Option<RelationMut<'rel_comp, Kind, ComponentT>>
+where
+    ComponentT: Component,
+{
+    fn from_optional_mut_component(
+        optional_component: Option<WriteGuard<'rel_comp, Box<dyn Component>>>,
+        world: &'rel_comp World,
+    ) -> Self
+    {
+        optional_component.map(|component| {
+            RelationMut::from_optional_mut_component(Some(component), world)
+        })
+    }
+}
+
 impl<'rel_comp, Kind, ComponentT> FromLockedOptionalComponent<'rel_comp>
     for RelationMut<'rel_comp, Kind, ComponentT>
 where
@@ -123,6 +139,24 @@ where
     }
 }
 
+impl<'rel_comp, Kind, ComponentT> FromLockedOptionalComponent<'rel_comp>
+    for Option<RelationMut<'rel_comp, Kind, ComponentT>>
+where
+    ComponentT: Component,
+{
+    fn from_locked_optional_component(
+        optional_component: Option<&'rel_comp Lock<Box<dyn Component>>>,
+        world: &'rel_comp World,
+    ) -> Result<Self, crate::lock::Error>
+    {
+        optional_component
+            .map(|component| {
+                RelationMut::from_locked_optional_component(Some(component), world)
+            })
+            .transpose()
+    }
+}
+
 impl<'rel_comp, Kind, ComponentT> RelationMut<'rel_comp, Kind, ComponentT>
 where
     ComponentT: Component,
@@ -327,6 +361,21 @@ where
     }
 }
 
+impl<'rel_comp, Kind, ComponentT> FromOptionalComponent<'rel_comp>
+    for Option<Relation<'rel_comp, Kind, ComponentT>>
+where
+    ComponentT: Component,
+{
+    fn from_optional_component(
+        optional_component: Option<ReadGuard<'rel_comp, Box<dyn Component>>>,
+        world: &'rel_comp World,
+    ) -> Self
+    {
+        optional_component
+            .map(|component| Relation::from_optional_component(Some(component), world))
+    }
+}
+
 impl<'rel_comp, Kind, ComponentT> FromLockedOptionalComponent<'rel_comp>
     for Relation<'rel_comp, Kind, ComponentT>
 where
@@ -346,6 +395,24 @@ where
     }
 }
 
+impl<'rel_comp, Kind, ComponentT> FromLockedOptionalComponent<'rel_comp>
+    for Option<Relation<'rel_comp, Kind, ComponentT>>
+where
+    ComponentT: Component,
+{
+    fn from_locked_optional_component(
+        optional_component: Option<&'rel_comp Lock<Box<dyn Component>>>,
+        world: &'rel_comp World,
+    ) -> Result<Self, crate::lock::Error>
+    {
+        optional_component
+            .map(|component| {
+                Relation::from_locked_optional_component(Some(component), world)
+            })
+            .transpose()
+    }
+}
+
 impl<'rel_comp, Kind, ComponentT> Relation<'rel_comp, Kind, ComponentT>
 where
     ComponentT: Component,