From 92ab09b3507cc7a29a531304ab1841fac899507a Mon Sep 17 00:00:00 2001
From: Ashley Ruglys <ashley.ruglys@gmail.com>
Date: Mon, 3 Aug 2020 10:12:59 +0200
Subject: [PATCH] Add as_ref and as_mut functions to Toggle

---
 swarm/src/toggle.rs | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/swarm/src/toggle.rs b/swarm/src/toggle.rs
index efd6177638a..2dbd5268dd5 100644
--- a/swarm/src/toggle.rs
+++ b/swarm/src/toggle.rs
@@ -51,6 +51,16 @@ impl<TBehaviour> Toggle<TBehaviour> {
     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<TBehaviour> From<Option<TBehaviour>> for Toggle<TBehaviour> {