feat(inbox): keep selected cases selected after action

Author: ChibiBlasphem

packages/app-builder/src/components/Cases/Inbox/BatchActions.tsx

@@ -1,6 +1,7 @@
 import { User } from '@app-builder/models';
 import { Case } from '@app-builder/models/cases';
 import { Inbox } from '@app-builder/models/inbox';
+import { useState } from 'react';
 import { useTranslation } from 'react-i18next';
 import * as R from 'remeda';
 import { Avatar, Button, MenuCommand } from 'ui-design-system';
@@ -26,12 +27,13 @@ export const BatchActions = ({
   inboxes,
   selectedCases,
 }: BatchActionsProps) => {
+  const [open, setOpen] = useState(false);
   const { t } = useTranslation(['common', 'cases']);
   const canReopen = selectedCases.some(({ status }) => status === 'closed');
   const canClose = selectedCases.some(({ status }) => status !== 'closed');
 
   return (
-    <MenuCommand.Menu>
+    <MenuCommand.Menu open={open} onOpenChange={setOpen}>
       <MenuCommand.Trigger>
         <Button variant="secondary">
           <Icon icon="checked" className="size-4" />

packages/app-builder/src/components/Cases/InboxPage.tsx

@@ -23,7 +23,6 @@ import { getRoute } from '@app-builder/utils/routes';
 import { fromUUIDtoSUUID } from '@app-builder/utils/short-uuid';
 import { useEffect, useMemo, useRef } from 'react';
 import { useTranslation } from 'react-i18next';
-import * as R from 'remeda';
 import { omit } from 'remeda';
 import shortUUID from 'short-uuid';
 import { Button, cn } from 'ui-design-system';
@@ -62,14 +61,8 @@ export const InboxPage = ({
   const { orgUsers } = useOrganizationUsers();
 
   const excludedFilters = !inboxId ? ['excludeAssigned'] : undefined;
-  const {
-    hasSelectedRows,
-    rowSelection,
-    getSelectedRows,
-    selectionProps,
-    tableProps,
-    setRowSelection,
-  } = useListSelection<Case>(cases, (row) => row.id);
+  const { hasSelectedRows, rowSelection, getSelectedRows, selectionProps, tableProps } =
+    useListSelection<Case>(cases, (row) => row.id);
   const massUpdateCasesMutation = useMassUpdateCasesMutation();
 
   const paginationSentinelRef = useRef<HTMLDivElement>(null);
@@ -87,7 +80,6 @@ export const InboxPage = ({
   const onMassUpdateCases: MassUpdateCasesFn = (params) => {
     const selectedCaseIds = getSelectedRows().map((row) => row.id);
     massUpdateCasesMutation.mutateAsync({ caseIds: selectedCaseIds, ...params }).then((res) => {
-      setRowSelection((rows) => R.omit(rows, selectedCaseIds));
       revalidate();
     });
   };