From 8e9b778f75e56af4d28db9d5d8b3651c125f87b8 Mon Sep 17 00:00:00 2001
From: lemonsh <lemon@lemonsh.moe>
Date: Sun, 14 May 2023 00:06:28 +0200
Subject: [PATCH] impl Display for PortForwardProtocol

---
 connectbox/src/models.rs | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/connectbox/src/models.rs b/connectbox/src/models.rs
index 0f787b9..a64d8a7 100644
--- a/connectbox/src/models.rs
+++ b/connectbox/src/models.rs
@@ -1,3 +1,4 @@
+use std::fmt::Display;
 use std::net::Ipv4Addr;
 use std::time::Duration;
 
@@ -93,6 +94,16 @@ impl<'de> Deserialize<'de> for PortForwardProtocol {
     }
 }
 
+impl Display for PortForwardProtocol {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        f.write_str(match self {
+            PortForwardProtocol::Tcp => "TCP",
+            PortForwardProtocol::Udp => "UDP",
+            PortForwardProtocol::Both => "Both",
+        })
+    }
+}
+
 fn bool_from_int<'de, D>(deserializer: D) -> Result<bool, D::Error>
 where
     D: Deserializer<'de>,