1 files changed, 6 insertions, 1 deletions
diff --git a/lampada/src/lib.rs b/lampada/src/lib.rs
index c61c596..a01ba06 100644
--- a/lampada/src/lib.rs
+++ b/lampada/src/lib.rs
@@ -15,6 +15,7 @@ use stanza::client::{
     iq::{self, Iq, IqType},
     Stanza,
 };
+use stanza::stream::Error as StreamError;
 use tokio::{
     sync::{mpsc, oneshot, Mutex},
     task::JoinSet,
@@ -59,12 +60,16 @@ pub trait Logic {
         connection: Connected,
     ) -> impl std::future::Future<Output = ()> + Send;
 
+    fn handle_stream_error(
+        self,
+        stream_error: StreamError,
+    ) -> impl std::future::Future<Output = ()> + Send;
+
     /// run to handle an incoming xmpp stanza
     fn handle_stanza(
         self,
         stanza: Stanza,
         connection: Connected,
-        supervisor: SupervisorSender,
     ) -> impl std::future::Future<Output = ()> + std::marker::Send;
 
     /// run to handle a command message when a connection is currently established