From 8bcaaf5524504f01bd5643d74848cec625370aa9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Roman=20Smr=C5=BE?= <roman.smrz@seznam.cz>
Date: Mon, 28 Jul 2025 16:48:15 +0200
Subject: Peer address input in ServiceHandler

---
 main/Main.hs | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

(limited to 'main/Main.hs')

diff --git a/main/Main.hs b/main/Main.hs
index e2e585a..6a6c565 100644
--- a/main/Main.hs
+++ b/main/Main.hs
@@ -172,17 +172,16 @@ options =
             return sopt { soptService = service }
         return opts { optServices = services' }
 
-    provideTunnelFun :: Maybe String -> Writer [ String ] (Peer -> Bool)
-    provideTunnelFun Nothing = return $ const True
-    provideTunnelFun (Just "all") = return $ const True
-    provideTunnelFun (Just "none") = return $ const False
-    provideTunnelFun (Just "websocket") = return $ \peer ->
-        case peerAddress peer of
-            CustomPeerAddress addr | Just WebSocketAddress {} <- cast addr -> True
-            _ -> False
+    provideTunnelFun :: Maybe String -> Writer [ String ] (Peer -> PeerAddress -> Bool)
+    provideTunnelFun Nothing = return $ \_ _ -> True
+    provideTunnelFun (Just "all") = return $ \_ _ -> True
+    provideTunnelFun (Just "none") = return $ \_ _ -> False
+    provideTunnelFun (Just "websocket") = return $ \_ -> \case
+        CustomPeerAddress addr | Just WebSocketAddress {} <- cast addr -> True
+        _ -> False
     provideTunnelFun (Just name) = do
         tell [ "Invalid value of --discovery-tunnel: ‘" <> name <> "’\n" ]
-        return $ const False
+        return $ \_ _ -> False
 
 servicesOptions :: [ OptDescr (Options -> Writer [ String ] Options) ]
 servicesOptions = concatMap helper $ "all" : map soptName availableServices
-- 
cgit v1.2.3