From 44c0da9d03cc9d990d16d08040177d8afed6c068 Mon Sep 17 00:00:00 2001
From: Dmitry Zuikov <dzuikov@gmail.com>
Date: Tue, 10 Sep 2024 13:34:12 +0300
Subject: [PATCH] wip

---
 fixme-new/lib/Fixme/Run.hs          |  3 ++
 fixme-new/lib/Fixme/Run/Internal.hs | 74 +++++++++++++++++++++++------
 fixme-new/lib/Fixme/State.hs        | 38 +++++++++++++++
 3 files changed, 101 insertions(+), 14 deletions(-)

diff --git a/fixme-new/lib/Fixme/Run.hs b/fixme-new/lib/Fixme/Run.hs
index 59f96c29..0b2d77a9 100644
--- a/fixme-new/lib/Fixme/Run.hs
+++ b/fixme-new/lib/Fixme/Run.hs
@@ -423,6 +423,9 @@ runTop forms = do
        entry $ bindMatch "fixme:refchan:export" $ nil_ $ const do
         void $ lift $ refchanExport
 
+       entry $ bindMatch "fixme:refchan:import" $ nil_ $ const do
+        void $ lift $ refchanImport
+
        entry $ bindMatch "git:blobs" $  \_ -> do
         blobs <- lift (listBlobs Nothing)
 
diff --git a/fixme-new/lib/Fixme/Run/Internal.hs b/fixme-new/lib/Fixme/Run/Internal.hs
index 1803cb0d..becdc853 100644
--- a/fixme-new/lib/Fixme/Run/Internal.hs
+++ b/fixme-new/lib/Fixme/Run/Internal.hs
@@ -19,6 +19,8 @@ import HBS2.Merkle
 import HBS2.Data.Types.Refs
 import HBS2.Storage
 import HBS2.Storage.Compact
+import HBS2.Peer.Proto.RefChan
+import HBS2.Peer.RPC.Client.RefChan
 import HBS2.Storage.Operations.ByteString
 import HBS2.System.Dir
 import HBS2.Net.Auth.Credentials
@@ -57,8 +59,8 @@ import Lens.Micro.Platform
 import System.Process.Typed
 import Control.Monad.Trans.Cont
 import Control.Monad.Trans.Maybe
-import System.IO.Temp as Temp
-import System.IO qualified as IO
+import Control.Monad.Except
+import Control.Concurrent.STM (flushTQueue)
 import Data.Time.Clock.POSIX (utcTimeToPOSIXSeconds)
 import System.Directory (getModificationTime)
 
@@ -335,18 +337,6 @@ cat_ hash = do
       liftIO $ action  dict fallback
 
 
-data FixmeExported =
-  FixmeExported
-  { exportedKey    :: FixmeKey
-  , exportedWeight :: Word64
-  , exportedName   :: FixmeAttrName
-  , exportedValue  :: FixmeAttrVal
-  }
-  deriving stock Generic
-
-instance FromRow FixmeExported
-instance ToRow  FixmeExported
-instance Serialise FixmeExported
 
 refchanExport :: FixmePerks m => FixmeM m ()
 refchanExport = do
@@ -382,6 +372,8 @@ refchanExport = do
     let chu  = chunksOf 10000 what
 
     for_ chu $ \x -> do
+
+      -- FIXME: encrypt-tree
       h  <- writeAsMerkle sto (serialise x)
 
       let tx = AnnotatedHashRef Nothing (HashRef h)
@@ -399,3 +391,57 @@ refchanExport = do
       when (isNothing r) do
         err $ red "hbs2-peer rpc calling timeout"
 
+
+refchanImport :: FixmePerks m => FixmeM m ()
+refchanImport = do
+
+  sto <- getStorage
+  rchanAPI <- getClientAPI @RefChanAPI @UNIX
+
+  chan <- asks fixmeEnvRefChan
+           >>= readTVarIO
+           >>= orThrowUser "refchan not set"
+
+
+  ttsmap <- newTVarIO HM.empty
+
+  tq <- newTQueueIO
+
+  walkRefChanTx @UNIX (const $ pure True) chan $ \txh u -> do
+
+    case  u of
+
+      A (AcceptTran (Just ts) _ what) -> do
+        debug $ red "ACCEPT" <+> pretty ts <+> pretty what
+        atomically $ modifyTVar ttsmap (HM.insertWith max what (coerce @_ @Word64 ts))
+
+      A _  -> none
+
+      P orig (ProposeTran _ box) -> void $ runMaybeT do
+        (_, bs) <- unboxSignedBox0 box & toMPlus
+
+        AnnotatedHashRef _ href <- deserialiseOrFail @AnnotatedHashRef (LBS.fromStrict bs)
+                                    & toMPlus . either (const Nothing) Just
+
+        -- FIXME: decrypt-tree
+        what <- runExceptT (readFromMerkle sto (SimpleKey (coerce href)))
+                  <&> either (const Nothing) Just
+                  >>= toMPlus
+
+        exported <- deserialiseOrFail @[FixmeExported] what
+                      & toMPlus
+
+        for_ exported $ \exported -> do
+          atomically $ writeTQueue tq (orig, exported)
+
+  imported <- atomically $ flushTQueue tq
+
+  withState $ transactional do
+    for_ imported $ \(h, i) -> do
+      w <- readTVarIO ttsmap <&> fromMaybe (exportedWeight i) . HM.lookup h
+      let item = i { exportedWeight = w }
+      notice $ "import" <+> pretty (exportedKey item) <+> pretty (exportedWeight item)
+      insertFixmeExported item
+
+
+
diff --git a/fixme-new/lib/Fixme/State.hs b/fixme-new/lib/Fixme/State.hs
index a66ae52d..e4a71336 100644
--- a/fixme-new/lib/Fixme/State.hs
+++ b/fixme-new/lib/Fixme/State.hs
@@ -6,11 +6,13 @@ module Fixme.State
   , cleanupDatabase
   , listFixme
   , insertFixme
+  , insertFixmeExported
   , modifyFixme
   , insertScanned
   , selectIsAlreadyScanned
   , selectFixmeKey
   , getFixme
+  , FixmeExported(..)
   , HasPredicate(..)
   , SelectPredicate(..)
   ) where
@@ -373,3 +375,39 @@ insertFixme fme = do
     lift $ insert sql (o,w,"fixme-text",txt)
 
 
+data FixmeExported =
+  FixmeExported
+  { exportedKey    :: FixmeKey
+  , exportedWeight :: Word64
+  , exportedName   :: FixmeAttrName
+  , exportedValue  :: FixmeAttrVal
+  }
+  deriving stock Generic
+
+instance FromRow FixmeExported
+instance ToRow  FixmeExported
+instance Serialise FixmeExported
+
+
+insertFixmeExported :: FixmePerks m => FixmeExported -> DBPipeM m ()
+insertFixmeExported item = do
+
+  let sql = [qc|
+
+      insert into object (o, w, k, v)
+      values (?, ?, ?, ?)
+      on conflict (o, k)
+      do update set
+        v = case
+              when excluded.w > object.w and (excluded.v <> object.v) then excluded.v
+              else object.v
+            end,
+        w = case
+              when excluded.w > object.w and (excluded.v <> object.v) then excluded.w
+              else object.w
+            end
+  |]
+
+  insert sql item
+
+