From 49cabbed9593d43beae096f696a571d49f0f10f7 Mon Sep 17 00:00:00 2001
From: Kodi Craft <kodi@kdcf.me>
Date: Mon, 24 Jun 2024 12:26:33 +0200
Subject: [PATCH] Fix server code generation for unix

---
 src/lib.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/lib.rs b/src/lib.rs
index a37a529..2ab9d7e 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -211,6 +211,14 @@ fn derive_protocol(input: proc_macro2::TokenStream) -> proc_macro2::TokenStream
     };
 
     // Create a struct to implement the communication between the server and the client
+    #[cfg(feature = "tcp")]
+    let listener_statement = quote! {
+        let listener = #listener_type::bind(addr).await?;
+    };
+    #[cfg(feature = "unix")]
+    let listener_statement = quote! {
+        let listener = #listener_type::bind(addr.as_ref())?;
+    };
     let sc_struct = quote! {
         #vis struct #server_connection_struct_name<H: #server_trait_name> {
             handler: ::std::sync::Arc<tokio::sync::Mutex<H>>,
@@ -218,7 +226,7 @@ fn derive_protocol(input: proc_macro2::TokenStream) -> proc_macro2::TokenStream
         }
         impl<H: #server_trait_name> #server_connection_struct_name<H> {
             pub async fn bind<S: #stream_addr_trait>(handler: H, addr: S) -> Result<Self, std::io::Error> {
-                let listener = #listener_type::bind(addr).await?;
+                #listener_statement
                 let (stream, _) = listener.accept().await?;
                 Ok(Self {
                     handler: ::std::sync::Arc::new(tokio::sync::Mutex::new(handler)),