Ouroboros: add local signatures

hydra-node/src/Hydra/Network/Ouroboros.hs

@@ -51,12 +51,7 @@ import Hydra.Network.Ouroboros.Type (
 import Network.Mux.Compat (
   WithMuxBearer (..),
  )
-import Network.Socket (
-  AddrInfo (addrAddress),
-  SockAddr,
-  defaultHints,
-  getAddrInfo,
- )
+import Network.Socket (AddrInfo (addrAddress), SockAddr, Socket, defaultHints, getAddrInfo)
 import Network.TypedProtocol.Codec (
   AnyMessageAndAgency (..),
  )
@@ -69,7 +64,7 @@ import Ouroboros.Network.ErrorPolicy (
   WithAddr (WithAddr),
   nullErrorPolicies,
  )
-import Ouroboros.Network.IOManager (withIOManager)
+import Ouroboros.Network.IOManager (IOManager, withIOManager)
 import Ouroboros.Network.Mux (
   MiniProtocol (
     MiniProtocol,
@@ -139,12 +134,26 @@ withOuroborosNetwork tracer localHost remoteHosts networkCallback between = do
             { broadcast = atomically . writeTChan bchan
             }
  where
+  resolveSockAddr :: Host -> IO SockAddr
   resolveSockAddr Host{hostname, port} = do
     is <- getAddrInfo (Just defaultHints) (Just $ toString hostname) (Just $ show port)
     case is of
       (info : _) -> pure $ addrAddress info
       _ -> error "getAdrrInfo failed.. do proper error handling"
 
+  connect ::
+    IOManager ->
+    IO t ->
+    ( t ->
+      OuroborosApplicationWithMinimalCtx
+        InitiatorMode
+        SockAddr
+        LByteString
+        IO
+        ()
+        Void
+    ) ->
+    IO Void
   connect iomgr newBroadcastChannel app = do
     -- REVIEW(SN): move outside to have this information available?
     networkState <- newNetworkMutableState
@@ -160,6 +169,10 @@ withOuroborosNetwork tracer localHost remoteHosts networkCallback between = do
       (subscriptionParams localAddr remoteAddrs)
       (actualConnect iomgr newBroadcastChannel app)
 
+  subscriptionParams ::
+    SockAddr ->
+    [SockAddr] ->
+    SubscriptionParams a IPSubscriptionTarget
   subscriptionParams localAddr remoteAddrs =
     SubscriptionParams
       { spLocalAddresses = LocalAddresses (Just localAddr) Nothing Nothing
@@ -168,6 +181,12 @@ withOuroborosNetwork tracer localHost remoteHosts networkCallback between = do
       , spSubscriptionTarget = IPSubscriptionTarget remoteAddrs (length remoteAddrs)
       }
 
+  actualConnect ::
+    IOManager ->
+    IO t ->
+    (t -> OuroborosApplicationWithMinimalCtx 'InitiatorMode SockAddr LByteString IO () Void) ->
+    Socket ->
+    IO ()
   actualConnect iomgr newBroadcastChannel app sn = do
     chan <- newBroadcastChannel
     connectToNodeSocket
@@ -180,12 +199,18 @@ withOuroborosNetwork tracer localHost remoteHosts networkCallback between = do
       (unversionedProtocol (app chan))
       sn
    where
+    networkConnectTracers :: NetworkConnectTracers a v
     networkConnectTracers =
       NetworkConnectTracers
         { nctMuxTracer = nullTracer
         , nctHandshakeTracer = nullTracer
         }
 
+  withServerListening ::
+    IOManager ->
+    OuroborosApplicationWithMinimalCtx 'ResponderMode SockAddr LByteString IO a b ->
+    IO b ->
+    IO ()
   withServerListening iomgr app continuation = do
     networkState <- newNetworkMutableState
     localAddr <- resolveSockAddr localHost
@@ -208,8 +233,10 @@ withOuroborosNetwork tracer localHost remoteHosts networkCallback between = do
       $ \_addr serverAsync -> do
         race_ (wait serverAsync) continuation
    where
+    notConfigureSocket :: a -> b -> IO ()
     notConfigureSocket _ _ = pure ()
 
+    networkServerTracers :: NetworkServerTracers SockAddr v
     networkServerTracers =
       NetworkServerTracers
         { nstMuxTracer = nullTracer
@@ -218,6 +245,7 @@ withOuroborosNetwork tracer localHost remoteHosts networkCallback between = do
         , nstAcceptPolicyTracer = contramap (WithHost localHost . TraceAcceptPolicy) tracer
         }
 
+    onIOException :: IOException -> IO ()
     onIOException ioException =
       throwIO $
         NetworkServerListenException