From c01d4bf9ef447785ae53b480e7e266a926f7aca3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Wojciech=20Pi=C4=85tkowski?= <chmurson@gmail.com>
Date: Tue, 2 Dec 2025 18:20:23 +0100
Subject: [PATCH] fix: werid UI states when leader disconnects upon unlocking

---
 fe/src/stores/boxStore/joinLockedBoxStore.ts | 1 +
 fe/src/stores/boxStore/openLockedBoxStore.ts | 8 ++++++++
 2 files changed, 9 insertions(+)

diff --git a/fe/src/stores/boxStore/joinLockedBoxStore.ts b/fe/src/stores/boxStore/joinLockedBoxStore.ts
index 0d67bc1..c8534a7 100644
--- a/fe/src/stores/boxStore/joinLockedBoxStore.ts
+++ b/fe/src/stores/boxStore/joinLockedBoxStore.ts
@@ -215,6 +215,7 @@ export const useJoinLockedBoxStore = create<JoinLockedBoxState>()(
 
           return {
             ...newState,
+            unlockingStartDate: null,
             state: "connecting",
             connecting: true,
           };
diff --git a/fe/src/stores/boxStore/openLockedBoxStore.ts b/fe/src/stores/boxStore/openLockedBoxStore.ts
index 37da760..235379d 100644
--- a/fe/src/stores/boxStore/openLockedBoxStore.ts
+++ b/fe/src/stores/boxStore/openLockedBoxStore.ts
@@ -181,6 +181,14 @@ export const useOpenLockedBoxStore = create<OpenLockedBoxState>()(
           const onlineKeyHoldersCount = newOnlineKeyHolders.length;
 
           const newState = {
+            shareAccessKeyByKeyHolderId: {
+              ...state.shareAccessKeyByKeyHolderId,
+              [disconnectedKeyHolderId]: false,
+            },
+            shareAccessKeyMapByKeyHolderId: {
+              ...state.shareAccessKeyMapByKeyHolderId,
+              [disconnectedKeyHolderId]: {},
+            },
             onlineKeyHolders: newOnlineKeyHolders,
             offLineKeyHolders: [
               disconnectedKeyHolder,