diff --git a/swarm/src/toggle.rs b/swarm/src/toggle.rs index efd61776..2dbd5268 100644 --- a/swarm/src/toggle.rs +++ b/swarm/src/toggle.rs @@ -51,6 +51,16 @@ impl Toggle { pub fn is_enabled(&self) -> bool { self.inner.is_some() } + + /// Returns a reference to the inner `NetworkBehaviour`. + pub fn as_ref(&self) -> Option<&TBehaviour> { + self.inner.as_ref() + } + + /// Returns a mutable reference to the inner `NetworkBehaviour`. + pub fn as_mut(&mut self) -> Option<&mut TBehaviour> { + self.inner.as_mut() + } } impl From> for Toggle {