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 where D: Deserializer<'de>,