twenty/front/src/modules/ui/data/view-bar/hooks/useRemoveView.ts

import { useContext } from 'react';
import { useRecoilCallback } from 'recoil';

import { useRecoilScopeId } from '@/ui/utilities/recoil-scope/hooks/useRecoilScopeId';

import { ViewBarContext } from '../contexts/ViewBarContext';
import { currentViewIdScopedState } from '../states/currentViewIdScopedState';
import { viewsScopedState } from '../states/viewsScopedState';

export const useRemoveView = () => {
  const { onViewRemove, ViewBarRecoilScopeContext } =
    useContext(ViewBarContext);

  const recoilScopeId = useRecoilScopeId(ViewBarRecoilScopeContext);

  const removeView = useRecoilCallback(
    ({ set, snapshot }) =>
      async (viewId: string) => {
        const currentViewId = await snapshot.getPromise(
          currentViewIdScopedState(recoilScopeId),
        );

        if (currentViewId === viewId)
          set(currentViewIdScopedState(recoilScopeId), undefined);

        set(viewsScopedState(recoilScopeId), (previousViews) =>
          previousViews.filter((view) => view.id !== viewId),
        );
        await onViewRemove?.(viewId);
      },
    [onViewRemove, recoilScopeId],
  );

  return { removeView };
};