From 232fe6b6e2b1c4c236a0914140e2ac7508374e3c Mon Sep 17 00:00:00 2001
From: Noa Aarts <noa@voorwaarts.nl>
Date: Wed, 11 Dec 2024 12:47:40 +0100
Subject: [PATCH] add response for protocol status

---
 src/lib.rs                       |  7 +++++++
 src/protocols/binary_protocol.rs | 17 +++++++++++++++++
 src/protocols/text_protocol.rs   | 17 +++++++++++++++++
 3 files changed, 41 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index 4b35c6f..ec943c7 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -53,6 +53,12 @@ fn increment_counter(amount: u64) {
     COUNTER.fetch_add(amount, std::sync::atomic::Ordering::Relaxed);
 }
 
+#[derive(Debug, PartialEq)]
+pub enum ProtocolStatus {
+    Enabled(&'static str),
+    Disabled(&'static str),
+}
+
 #[derive(Debug, PartialEq)]
 pub enum Protocol {
     Text,
@@ -72,6 +78,7 @@ pub enum Command {
 #[derive(Debug, PartialEq)]
 pub enum Response {
     Help,
+    Protocols(Vec<ProtocolStatus>),
     Size(Coordinate, Coordinate),
     GetPixel(Coordinate, Coordinate, [u8; 3]),
 }
diff --git a/src/protocols/binary_protocol.rs b/src/protocols/binary_protocol.rs
index e505112..b7c0996 100644
--- a/src/protocols/binary_protocol.rs
+++ b/src/protocols/binary_protocol.rs
@@ -106,6 +106,23 @@ To set a pixel using RGB, use ({SET_PX_RGB_BIN:02X}) (u8 canvas) (x as u16_le) (
 );
                 writer.write_all(help_text.as_bytes()).await
             }
+            Response::Protocols(protos) => {
+                for protocol in protos {
+                    match protocol {
+                        crate::ProtocolStatus::Enabled(proto) => {
+                            writer
+                                .write_all(format!("Enabled: {}\n", proto).as_bytes())
+                                .await?;
+                        }
+                        crate::ProtocolStatus::Disabled(proto) => {
+                            writer
+                                .write_all(format!("Disabled: {}\n", proto).as_bytes())
+                                .await?;
+                        }
+                    }
+                }
+                Ok(())
+            }
             Response::Size(x, y) => {
                 writer.write_u16(x).await?;
                 writer.write_u16(y).await
diff --git a/src/protocols/text_protocol.rs b/src/protocols/text_protocol.rs
index 1c711a2..f5e48bf 100644
--- a/src/protocols/text_protocol.rs
+++ b/src/protocols/text_protocol.rs
@@ -146,6 +146,23 @@ impl<W: AsyncWriteExt + std::marker::Unpin> Responder<W> for TextParser {
     async fn unparse(&self, response: Response, writer: &mut W) -> io::Result<()> {
         match response {
             Response::Help => writer.write_all(HELP_TEXT).await,
+            Response::Protocols(protos) => {
+                for protocol in protos {
+                    match protocol {
+                        crate::ProtocolStatus::Enabled(proto) => {
+                            writer
+                                .write_all(format!("Enabled: {}\n", proto).as_bytes())
+                                .await?;
+                        }
+                        crate::ProtocolStatus::Disabled(proto) => {
+                            writer
+                                .write_all(format!("Disabled: {}\n", proto).as_bytes())
+                                .await?;
+                        }
+                    }
+                }
+                Ok(())
+            }
             Response::Size(x, y) => writer.write_all(format!("SIZE {x} {y}\n").as_bytes()).await,
             Response::GetPixel(x, y, color) => {
                 writer