From 36a0282256dbb942a4852a2797e403f1352399c9 Mon Sep 17 00:00:00 2001
From: Dmitry Zuikov <dzuikov@gmail.com>
Date: Fri, 14 Jul 2023 16:00:11 +0300
Subject: [PATCH] wip, posting refchan head transaction

---
 hbs2-core/lib/HBS2/Net/Proto/RefChan.hs |  8 ++++---
 hbs2-peer/app/PeerMain.hs               | 31 ++++++++++++++++++++++---
 2 files changed, 33 insertions(+), 6 deletions(-)

diff --git a/hbs2-core/lib/HBS2/Net/Proto/RefChan.hs b/hbs2-core/lib/HBS2/Net/Proto/RefChan.hs
index 50e716d1..2cdcf5a0 100644
--- a/hbs2-core/lib/HBS2/Net/Proto/RefChan.hs
+++ b/hbs2-core/lib/HBS2/Net/Proto/RefChan.hs
@@ -16,7 +16,7 @@ import HBS2.Data.Types.Refs
 
 import Data.Config.Suckless
 
--- import HBS2.System.Logger.Simple
+import HBS2.System.Logger.Simple
 
 import Codec.Serialise
 import Control.Monad.Identity
@@ -91,15 +91,17 @@ refChanHeadProto :: forall e s m . ( MonadIO m
                                    , Pretty (AsBase58 (PubKey 'Sign s))
                                    , s ~ Encryption e
                                    )
-                  => RefChanHeadAdapter e m
+                  => Bool
+                  -> RefChanHeadAdapter e m
                   -> RefChanHead e
                   -> m ()
 
-refChanHeadProto adapter msg = do
+refChanHeadProto self adapter msg = do
   -- авторизовать пира
 
   case msg of
     RefChanHead pkt _ -> do
+      trace $ "RefChanHead" <+> pretty self
       pure ()
 
     RefChanGetHead _ -> do
diff --git a/hbs2-peer/app/PeerMain.hs b/hbs2-peer/app/PeerMain.hs
index a056b140..6affe0fb 100644
--- a/hbs2-peer/app/PeerMain.hs
+++ b/hbs2-peer/app/PeerMain.hs
@@ -5,13 +5,15 @@
 {-# Language MultiWayIf #-}
 module Main where
 
+import HBS2.Prelude.Plated
+
 import HBS2.Actors.Peer
 import HBS2.Base58
 import HBS2.Clock
 import HBS2.Defaults
 import HBS2.Events
 import HBS2.Hash
-import HBS2.Data.Types.Refs (RefLogKey(..))
+import HBS2.Data.Types.Refs
 import HBS2.Net.Auth.Credentials
 import HBS2.Net.IP.Addr
 import HBS2.Net.Messaging.UDP
@@ -24,10 +26,11 @@ import HBS2.Net.Proto.PeerAnnounce
 import HBS2.Net.Proto.PeerExchange
 import HBS2.Net.Proto.PeerMeta
 import HBS2.Net.Proto.RefLog
+import HBS2.Net.Proto.RefChan
 import HBS2.Net.Proto.Sessions
 import HBS2.OrDie
-import HBS2.Prelude.Plated
 import HBS2.Storage.Simple
+import HBS2.Data.Detect
 
 import HBS2.System.Logger.Simple hiding (info)
 import HBS2.System.Logger.Simple qualified as Log
@@ -89,6 +92,8 @@ import UnliftIO.Exception qualified as U
 import UnliftIO.Async as U
 import Control.Monad.Trans.Resource
 
+import Streaming.Prelude qualified as S
+import Streaming qualified as S
 
 -- TODO: write-workers-to-config
 defStorageThreads :: Integral a => a
@@ -553,6 +558,10 @@ runPeer opts = U.handle (\e -> myException e
     pause @'Seconds 600
     liftIO $ Cache.purgeExpired nbcache
 
+  let refChanHeadAdapter = RefChanHeadAdapter
+                           { _refChanHeadOnHead = dontHandle
+                           }
+
   let pexFilt pips = do
         tcpex <- listTCPPexCandidates @e brains <&> HashSet.fromList
         fset  <- forM pips $ \p -> do
@@ -959,7 +968,23 @@ runPeer opts = U.handle (\e -> myException e
 
   let refChanHeadSendAction h = do
         trace $ "refChanHeadSendAction" <+> pretty h
-        pure ()
+        void $ liftIO $ async $ withPeerM penv $ do
+          me <- ownPeer @e
+          sto <- getStorage
+
+          chunks <- S.toList_ $ do
+            deepScan ScanDeep (const none) h (liftIO . getBlock sto) $ \ha -> do
+              unless (ha == h) do
+                blk <- liftIO $ getBlock sto ha
+                maybe1 blk none S.yield
+
+          let box = deserialiseOrFail @(SignedBox (RefChanHeadBlock e) e) (LBS.concat chunks)
+
+          case box of
+            Left{} -> err $ "can't read head block" <+> pretty h
+            Right (SignedBox k _ _) -> do
+              let msg = RefChanHead k (RefChanHeadBlockTran (HashRef h))
+              runResponseM me $ refChanHeadProto @e True refChanHeadAdapter msg
 
   let arpc = RpcAdapter pokeAction
                         dieAction