From 564230f043cd8c99d3144248bb5538ed6d9020f2 Mon Sep 17 00:00:00 2001
From: Mike Dilger <mike@mikedilger.com>
Date: Thu, 15 Feb 2024 10:18:40 +1300
Subject: [PATCH] Handle websocket related errors

---
 src/error.rs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/error.rs b/src/error.rs
index 00d1d1a..454ac7a 100644
--- a/src/error.rs
+++ b/src/error.rs
@@ -74,6 +74,10 @@ pub enum Error {
     #[error("TLS: {0}")]
     Rustls(#[from] tokio_rustls::rustls::Error),
 
+    // Tunstenite
+    #[error("Websocket: {0}")]
+    Tungstenite(#[from] hyper_tungstenite::tungstenite::error::Error),
+
     // Filter is underspecified
     #[error("Filter is underspecified. Scrapers are not allowed")]
     Scraper,
@@ -85,4 +89,8 @@ pub enum Error {
     // UTF-8
     #[error("UTF-8 error")]
     Utf8Error,
+
+    // Tunstenite Protocol
+    #[error("Websocket Protocol: {0}")]
+    WebsocketProtocol(#[from] hyper_tungstenite::tungstenite::error::ProtocolError),
 }