From 126994720f7410d308c8bdff34a083d859de818b Mon Sep 17 00:00:00 2001
From: Dmitry Zuikov <dzuikov@gmail.com>
Date: Wed, 19 Jul 2023 08:30:25 +0300
Subject: [PATCH] wip, refchanrequest events

---
 hbs2-core/lib/HBS2/Net/Proto/RefChan.hs | 41 ++++++++++++++++++-------
 hbs2-peer/app/RefChan.hs                |  4 +++
 2 files changed, 34 insertions(+), 11 deletions(-)

diff --git a/hbs2-core/lib/HBS2/Net/Proto/RefChan.hs b/hbs2-core/lib/HBS2/Net/Proto/RefChan.hs
index 75a938a7..b78043de 100644
--- a/hbs2-core/lib/HBS2/Net/Proto/RefChan.hs
+++ b/hbs2-core/lib/HBS2/Net/Proto/RefChan.hs
@@ -77,7 +77,20 @@ type ForRefChans e = ( Serialise ( PubKey 'Sign (Encryption e))
 instance ForRefChans e => Serialise (RefChanHeadBlock e)
 instance ForRefChans e => Serialise (SignedBox p e)
 
+type instance SessionData e (RefChanHeadBlock e) = RefChanHeadBlock e
 
+newtype instance SessionKey e  (RefChanHeadBlock e) =
+  RefChanHeadBlockKey (RefChanHeadKey (Encryption e))
+
+deriving newtype instance ForRefChans L4Proto
+  => Hashable (SessionKey L4Proto (RefChanHeadBlock L4Proto))
+
+deriving stock instance ForRefChans L4Proto
+  => Eq (SessionKey L4Proto (RefChanHeadBlock L4Proto))
+
+-- TODO: define-expiration-time
+instance Expires (SessionKey L4Proto (RefChanHeadBlock L4Proto)) where
+  expiresIn = const (Just defCookieTimeoutSec)
 
 newtype RefChanHeadKey s = RefChanHeadKey (PubKey 'Sign s)
 
@@ -213,24 +226,28 @@ instance ForRefChans e => Serialise (RefChanUpdate e)
 data RefChanRequest e =
     RefChanRequest  (RefChanId e)
   | RefChanResponse (RefChanId e) HashRef
-  deriving stock (Generic)
+  deriving stock (Generic,Typeable)
 
 instance ForRefChans e => Serialise (RefChanRequest e)
 
-type instance SessionData e (RefChanHeadBlock e) = RefChanHeadBlock e
+data instance EventKey e (RefChanRequest e) =
+  RefChanRequestEventKey
+  deriving (Generic,Typeable,Eq)
 
-newtype instance SessionKey e  (RefChanHeadBlock e) =
-  RefChanHeadBlockKey (RefChanHeadKey (Encryption e))
+data instance Event e (RefChanRequest e) =
+  RefChanRequestEvent (RefChanId e) HashRef
+  deriving (Typeable,Generic)
 
-deriving newtype instance ForRefChans L4Proto
-  => Hashable (SessionKey L4Proto (RefChanHeadBlock L4Proto))
+instance EventType ( Event e (RefChanRequest e) ) where
+  isPersistent = True
 
-deriving stock instance ForRefChans L4Proto
-  => Eq (SessionKey L4Proto (RefChanHeadBlock L4Proto))
+instance Expires (EventKey e (RefChanRequest e)) where
+  expiresIn = const Nothing
 
--- TODO: define-expiration-time
-instance Expires (SessionKey L4Proto (RefChanHeadBlock L4Proto)) where
-  expiresIn = const (Just defCookieTimeoutSec)
+instance Typeable (RefChanRequest e) => Hashable (EventKey e (RefChanRequest e)) where
+  hashWithSalt salt _ = hashWithSalt salt (someTypeRep p)
+    where
+      p = Proxy @(RefChanRequest e)
 
 -- FIXME: rename
 data RefChanAdapter e m =
@@ -555,6 +572,7 @@ refChanRequestProto :: forall e s m . ( MonadIO m
                                      , Pretty (Peer e)
                                      , Sessions e (KnownPeer e) m
                                      , Sessions e (RefChanHeadBlock e) m
+                                     , EventEmitter e (RefChanRequest e) m
                                      , HasStorage m
                                      , Signatures s
                                      , IsRefPubKey s
@@ -596,6 +614,7 @@ refChanRequestProto self adapter msg = do
 
         guard $ ppk `HashMap.member` view refChanHeadPeers hd
 
+        lift $ emit RefChanRequestEventKey (RefChanRequestEvent @e chan val)
         debug $ "RefChanResponse" <+> pretty peer <+> pretty (AsBase58 chan) <+> pretty val
 
   where
diff --git a/hbs2-peer/app/RefChan.hs b/hbs2-peer/app/RefChan.hs
index 8aa626ce..17504cdb 100644
--- a/hbs2-peer/app/RefChan.hs
+++ b/hbs2-peer/app/RefChan.hs
@@ -139,6 +139,7 @@ refChanWorker :: forall e s m . ( MonadIO m
                                 , Pretty (AsBase58 (PubKey 'Sign s))
                                 , ForRefChans e
                                 , EventListener e (RefChanRound e) m
+                                , EventListener e (RefChanRequest e) m
                                 , Sessions e (RefChanRound e) m
                                 , m ~ PeerM e IO
                                 )
@@ -162,6 +163,9 @@ refChanWorker env brains = do
 
   cleanup1 <- async cleanupRounds
 
+  subscribe @e RefChanRequestEventKey $ \(RefChanRequestEvent chan val) -> do
+    debug $ "RefChanRequestEvent" <+> pretty (AsBase58 chan) <+> pretty val
+
   forever do
    pause @'Seconds 10
    debug "I'm refchan worker"