From 98b3d510e4802f7ef043f5ef30940eb6254fb37c Mon Sep 17 00:00:00 2001
From: kozabrada123 <59031733+kozabrada123@users.noreply.github.com>
Date: Tue, 29 Aug 2023 14:44:47 +0200
Subject: [PATCH] Test error observer

---
 src/errors.rs  | 7 ++++++-
 src/gateway.rs | 9 ++++++---
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/src/errors.rs b/src/errors.rs
index eebfae3..26cc6b2 100644
--- a/src/errors.rs
+++ b/src/errors.rs
@@ -2,6 +2,8 @@
 use custom_error::custom_error;
 use reqwest::Error;
 
+use crate::types::WebSocketEvent;
+
 custom_error! {
     #[derive(PartialEq, Eq)]
     pub RegistrationError
@@ -54,9 +56,10 @@ custom_error! {
     /// Supposed to be sent as numbers, though they are sent as string most of the time?
     ///
     /// Also includes errors when initiating a connection and unexpected opcodes
-    #[derive(PartialEq, Eq)]
+    #[derive(PartialEq, Eq, Default)]
     pub GatewayError
     // Errors we have received from the gateway
+    #[default]
     Unknown = "We're not sure what went wrong. Try reconnecting?",
     UnknownOpcode = "You sent an invalid Gateway opcode or an invalid payload for an opcode",
     Decode = "Gateway server couldn't decode payload",
@@ -79,3 +82,5 @@ custom_error! {
     // Other misc errors
     UnexpectedOpcodeReceived{opcode: u8} = "Received an opcode we weren't expecting to receive: {opcode}",
 }
+
+impl WebSocketEvent for GatewayError {}
diff --git a/src/gateway.rs b/src/gateway.rs
index 68205b5..6913e6a 100644
--- a/src/gateway.rs
+++ b/src/gateway.rs
@@ -481,13 +481,15 @@ impl Gateway {
             return;
         }
 
-        // Todo: handle errors in a good way, maybe observers like events?
         if msg.is_error() {
-            warn!("GW: Received error, connection will close..");
+            let error = msg.error().unwrap();
 
-            let _error = msg.error();
+            warn!("GW: Received error {:?}, connection will close..", error);
 
             self.close().await;
+
+            self.events.lock().await.error.notify(error).await;
+
             return;
         }
 
@@ -937,6 +939,7 @@ mod events {
         pub webhooks: Webhooks,
         pub gateway_identify_payload: GatewayEvent<types::GatewayIdentifyPayload>,
         pub gateway_resume: GatewayEvent<types::GatewayResume>,
+        pub error: GatewayEvent<GatewayError>,
     }
 
     #[derive(Default, Debug)]