From d1ae298161660e5fd2ac9a0dcbc3c3b3305a05e5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Roman=20Smr=C5=BE?= <roman.smrz@seznam.cz>
Date: Sat, 15 Aug 2026 23:11:25 +0200
Subject: Track local head cache in main/test commands

---
 main/Main.hs | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'main/Main.hs')

diff --git a/main/Main.hs b/main/Main.hs
index 654c04b..5d5df63 100644
--- a/main/Main.hs
+++ b/main/Main.hs
@@ -51,6 +51,7 @@ import Erebos.Set
 import Erebos.State
 import Erebos.Storable
 import Erebos.Storage
+import Erebos.Storage.Head
 import Erebos.Storage.Merge
 import Erebos.Sync
 import Erebos.TextFormat
@@ -635,6 +636,11 @@ instance MonadHead LocalState CommandM where
         modify $ \s -> s { csHead = h' }
         return x
 
+    getLocalHeadCache _ = do
+        h <- maybe (fail "failed to reload head") return =<< reloadHead =<< gets csHead
+        modify $ \s -> s { csHead = h }
+        return $ headCache h
+
 type Command = CommandM ()
 
 getSelectedPeer :: CommandM Peer
-- 
cgit v1.2.3