From 0ade6bfde525c14aefddaf0f8675e48a2bb7a0ee Mon Sep 17 00:00:00 2001
From: Thomas Eizinger <thomas@eizinger.io>
Date: Mon, 18 Sep 2023 14:32:08 +1000
Subject: [PATCH] Add display impl for MessageClass

---
 src/message.rs | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/message.rs b/src/message.rs
index d064137..9a4bc3f 100644
--- a/src/message.rs
+++ b/src/message.rs
@@ -8,7 +8,7 @@ use bytecodec::bytes::{BytesEncoder, CopyableBytesDecoder};
 use bytecodec::combinator::{Collect, Length, Peekable, PreEncode, Repeat};
 use bytecodec::fixnum::{U16beDecoder, U16beEncoder, U32beDecoder, U32beEncoder};
 use bytecodec::{ByteCount, Decode, Encode, Eos, Error, ErrorKind, Result, SizedEncode};
-use std::vec;
+use std::{fmt, vec};
 use trackable::error::ErrorKindExt;
 
 /// Message decoded by [`MessageDecoder`].
@@ -35,6 +35,17 @@ impl MessageClass {
     }
 }
 
+impl fmt::Display for MessageClass {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        match self {
+            MessageClass::Request => write!(f, "request"),
+            MessageClass::Indication => write!(f, "indication"),
+            MessageClass::SuccessResponse => write!(f, "success response"),
+            MessageClass::ErrorResponse => write!(f, "error response"),
+        }
+    }
+}
+
 /// STUN message.
 ///
 /// # NOTE: Binary Format of STUN Messages