From e08ffd653e45407b1452402306a66d5b30ff472b Mon Sep 17 00:00:00 2001
From: Dmitry Zuikov <dzuikov@gmail.com>
Date: Tue, 17 Jan 2023 14:41:39 +0300
Subject: [PATCH] wtf

---
 hbs2-core/lib/HBS2/Net/Peer.hs | 39 +++++++++++++++++-----------------
 hbs2-tests/test/Main.hs        |  2 +-
 2 files changed, 21 insertions(+), 20 deletions(-)

diff --git a/hbs2-core/lib/HBS2/Net/Peer.hs b/hbs2-core/lib/HBS2/Net/Peer.hs
index ba9ecf15..b6977313 100644
--- a/hbs2-core/lib/HBS2/Net/Peer.hs
+++ b/hbs2-core/lib/HBS2/Net/Peer.hs
@@ -26,20 +26,21 @@ data AnyMessage e = AnyMessage Integer (Encoded e)
 
 instance Serialise (Encoded e) => Serialise (AnyMessage e)
 
-newtype EngineM e m a = EngineM { fromEngine :: ReaderT (EngineEnv e) m a }
-                        deriving ( Functor
-                                 , Applicative
-                                 , Monad
-                                 , MonadIO
-                                 , MonadReader (EngineEnv e)
-                                 , MonadTrans
-                                )
+newtype EngineM e m a = EngineM { fromEngine :: ReaderT (EngineEnv e m) m a }
+                        deriving newtype ( Functor
+                                         , Applicative
+                                         , Monad
+                                         , MonadIO
+                                         , MonadReader (EngineEnv e m)
+                                         )
 
+instance MonadTrans (EngineM (EngineEnv e m)) where
+  lift = lift
 
-data EngineEnv e  = forall bus m . ( Messaging bus e ByteString
-                                  , Serialise (Encoded e)
-                                  , MonadIO m
-                                  )  =>
+data EngineEnv e m  = forall bus  . ( Messaging bus e ByteString
+                                    , Serialise (Encoded e)
+                                    , MonadIO m
+                                    )  =>
   EngineEnv
   { _peer   :: Maybe (Peer e)
   , _self   :: Peer e
@@ -72,7 +73,7 @@ makeResponse h = AnyProtocol { myProtoId  = natVal (Proxy @(ProtocolId p))
                              }
 
 
-runEngineM :: EngineEnv e -> EngineM e m a -> m a
+runEngineM :: EngineEnv e m -> EngineM e m a -> m a
 runEngineM e f = runReaderT (fromEngine f) e
 
 
@@ -88,9 +89,7 @@ instance (MonadIO m, HasProtocol e p) => Response e p (EngineM e m) where
 
   deferred _ m = do
     e@(EngineEnv { defer = d }) <- ask
-    lift $ runEngineM e m
-    pure ()
-    -- lift $ addJob d (lift $ runEngineM e m)
+    addJob d (runEngineM e m)
 
   response resp = do
     env <- ask
@@ -112,15 +111,15 @@ newEnv :: forall e bus m . ( Monad m
                            )
        => Peer e
        -> bus
-       -> m (EngineEnv e)
+       -> m (EngineEnv e m)
 
 newEnv p pipe = do
-  de <- liftIO $ newPipeline defPipelineSize
+  de <- newPipeline defPipelineSize
   pure $ EngineEnv Nothing p pipe de
 
 runPeer :: forall e m a . ( MonadIO m
                           )
-        => EngineEnv e
+        => EngineEnv e m
         -> [AnyProtocol e (EngineM e m)]
         -> m a
 
@@ -132,6 +131,8 @@ runPeer env@(EngineEnv {bus = pipe, defer = d}) hh = do
 
   let disp = Map.fromList resp
 
+  let q = runPipeline d
+
   -- let q = liftIO $ runPipeline d
 
   runEngineM env $ do
diff --git a/hbs2-tests/test/Main.hs b/hbs2-tests/test/Main.hs
index 7ccf377a..7c91e658 100644
--- a/hbs2-tests/test/Main.hs
+++ b/hbs2-tests/test/Main.hs
@@ -98,7 +98,7 @@ main = do
   --     ]
 
 
-runFakePeer :: EngineEnv Fake -> IO ()
+runFakePeer :: EngineEnv Fake IO -> IO ()
 runFakePeer env = do
 
   let pid = fromIntegral (hash (env ^. self)) :: Word8