From c5795b7325b46abf6eead5ffbf0cb4069f13313d Mon Sep 17 00:00:00 2001
From: Lucas Bordeau <bordeau.lucas@gmail.com>
Date: Mon, 3 Mar 2025 11:25:48 +0100
Subject: [PATCH] Fixed board total count on view picker button (#10612)

This PR adds a missing call to `setRecordIndexEntityCount` during board
data loading.
---
 .../hooks/useSetRecordIdsForColumn.ts             | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/packages/twenty-front/src/modules/object-record/record-board/hooks/useSetRecordIdsForColumn.ts b/packages/twenty-front/src/modules/object-record/record-board/hooks/useSetRecordIdsForColumn.ts
index 1d1bf6ab9..c68bd51f4 100644
--- a/packages/twenty-front/src/modules/object-record/record-board/hooks/useSetRecordIdsForColumn.ts
+++ b/packages/twenty-front/src/modules/object-record/record-board/hooks/useSetRecordIdsForColumn.ts
@@ -2,6 +2,7 @@ import { useRecoilCallback } from 'recoil';
 
 import { recordGroupDefinitionFamilyState } from '@/object-record/record-group/states/recordGroupDefinitionFamilyState';
 import { recordGroupFieldMetadataComponentState } from '@/object-record/record-group/states/recordGroupFieldMetadataComponentState';
+import { useSetRecordIndexEntityCount } from '@/object-record/record-index/hooks/useSetRecordIndexEntityCount';
 import { recordIndexRecordIdsByGroupComponentFamilyState } from '@/object-record/record-index/states/recordIndexRecordIdsByGroupComponentFamilyState';
 import { ObjectRecord } from '@/object-record/types/ObjectRecord';
 import { useRecoilComponentCallbackStateV2 } from '@/ui/utilities/state/component-state/hooks/useRecoilComponentCallbackStateV2';
@@ -21,6 +22,9 @@ export const useSetRecordIdsForColumn = (recordBoardId?: string) => {
       recordBoardId,
     );
 
+  const { setRecordIndexEntityCount } =
+    useSetRecordIndexEntityCount(recordBoardId);
+
   const setRecordIdsForColumn = useRecoilCallback(
     ({ set, snapshot }) =>
       (currentRecordGroupId: string, records: ObjectRecord[]) => {
@@ -55,9 +59,18 @@ export const useSetRecordIdsForColumn = (recordBoardId?: string) => {
             recordIndexRecordIdsByGroupFamilyState(currentRecordGroupId),
             recordGroupRowIds,
           );
+
+          setRecordIndexEntityCount(
+            recordGroupRowIds.length,
+            currentRecordGroupId,
+          );
         }
       },
-    [recordIndexRecordIdsByGroupFamilyState, recordGroupFieldMetadataState],
+    [
+      recordIndexRecordIdsByGroupFamilyState,
+      recordGroupFieldMetadataState,
+      setRecordIndexEntityCount,
+    ],
   );
 
   return {