From c27b3c23ecdd53acdbfece747b9bbdb39bf4dae9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Roman=20Smr=C5=BE?= <roman.smrz@seznam.cz>
Date: Sun, 27 Aug 2023 18:33:16 +0200
Subject: Replace storedStorage usage with MonadHead

---
 src/Network.hs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'src/Network.hs')

diff --git a/src/Network.hs b/src/Network.hs
index 3614de0..96f8527 100644
--- a/src/Network.hs
+++ b/src/Network.hs
@@ -22,6 +22,7 @@ import Control.Concurrent.STM
 import Control.Exception
 import Control.Monad
 import Control.Monad.Except
+import Control.Monad.Reader
 import Control.Monad.State
 
 import qualified Data.ByteString.Char8 as BC
@@ -527,7 +528,7 @@ withPeerIdentity peer act = liftIO $ atomically $ readTVar (peerIdentityVar peer
 
 setupChannel :: UnifiedIdentity -> Peer -> UnifiedIdentity -> WaitingRefCallback
 setupChannel identity peer upid = do
-    req <- createChannelRequest (peerStorage peer) identity upid
+    req <- flip runReaderT (peerStorage peer) $ createChannelRequest identity upid
     let reqref = refDigest $ storedRef req
     let hitems =
             [ TrChannelRequest reqref
@@ -544,7 +545,7 @@ setupChannel identity peer upid = do
 handleChannelRequest :: Peer -> UnifiedIdentity -> Ref -> WaitingRefCallback
 handleChannelRequest peer identity req = do
     withPeerIdentity peer $ \upid -> do
-        (acc, ch) <- acceptChannelRequest identity upid (wrappedLoad req)
+        (acc, ch) <- flip runReaderT (peerStorage peer) $ acceptChannelRequest identity upid (wrappedLoad req)
         liftIO $ atomically $ do
             getPeerChannel peer >>= \case
                 ChannelPeerRequest wr | wrDigest wr == refDigest req -> do
-- 
cgit v1.2.3