1 files changed, 58 insertions, 0 deletions

diff --git a/src/mqtt/connect_code.rs b/src/mqtt/connect_code.rs
new file mode 100644
index 0000000..570ce0f
--- /dev/null
+++ b/src/mqtt/connect_code.rs
@@ -0,0 +1,58 @@
+use super::protocol;
+
+#[derive(Debug, Clone, Copy, PartialEq, Eq)]
+pub enum ConnectCode {
+    ConnectionAccepted,
+    UnacceptableProtocolVersion,
+    IdentifierRejected,
+    ServerUnavailable,
+    BadUsernamePassword,
+    NotAuthorized,
+    Unknown(u8),
+}
+
+impl core::fmt::Display for ConnectCode {
+    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
+        match self {
+            ConnectCode::ConnectionAccepted => write!(f, "Connection Accepted"),
+            ConnectCode::UnacceptableProtocolVersion => write!(f, "Unacceptable Protocol Version"),
+            ConnectCode::IdentifierRejected => write!(f, "Identifier Rejected"),
+            ConnectCode::ServerUnavailable => write!(f, "Server Unavailable"),
+            ConnectCode::BadUsernamePassword => write!(f, "Bad Username or Password"),
+            ConnectCode::NotAuthorized => write!(f, "Not Authorized"),
+            ConnectCode::Unknown(code) => write!(f, "Unknown({})", code),
+        }
+    }
+}
+
+impl From<u8> for ConnectCode {
+    fn from(value: u8) -> Self {
+        match value {
+            protocol::CONNACK_CODE_ACCEPTED => ConnectCode::ConnectionAccepted,
+            protocol::CONNACK_CODE_UNACCEPTABLE_PROTOCOL_VERSION => {
+                ConnectCode::UnacceptableProtocolVersion
+            }
+            protocol::CONNACK_CODE_IDENTIFIER_REJECTED => ConnectCode::IdentifierRejected,
+            protocol::CONNACK_CODE_SERVER_UNAVAILABLE => ConnectCode::ServerUnavailable,
+            protocol::CONNACK_CODE_BAD_USERNAME_PASSWORD => ConnectCode::BadUsernamePassword,
+            protocol::CONNACK_CODE_NOT_AUTHORIZED => ConnectCode::NotAuthorized,
+            code => ConnectCode::Unknown(code),
+        }
+    }
+}
+
+impl From<ConnectCode> for u8 {
+    fn from(value: ConnectCode) -> Self {
+        match value {
+            ConnectCode::ConnectionAccepted => protocol::CONNACK_CODE_ACCEPTED,
+            ConnectCode::UnacceptableProtocolVersion => {
+                protocol::CONNACK_CODE_UNACCEPTABLE_PROTOCOL_VERSION
+            }
+            ConnectCode::IdentifierRejected => protocol::CONNACK_CODE_IDENTIFIER_REJECTED,
+            ConnectCode::ServerUnavailable => protocol::CONNACK_CODE_SERVER_UNAVAILABLE,
+            ConnectCode::BadUsernamePassword => protocol::CONNACK_CODE_BAD_USERNAME_PASSWORD,
+            ConnectCode::NotAuthorized => protocol::CONNACK_CODE_NOT_AUTHORIZED,
+            ConnectCode::Unknown(code) => code,
+        }
+    }
+}