fix: deselect item on delete

packages/core/reducer/__tests__/state.spec.tsx

@@ -1,6 +1,11 @@
 import { defaultAppState } from "../../components/Puck/context";
 import { rootDroppableId } from "../../lib/root-droppable-id";
-import { DuplicateAction, SetUiAction, createReducer } from "../../reducer";
+import {
+  DuplicateAction,
+  RemoveAction,
+  SetUiAction,
+  createReducer,
+} from "../../reducer";
 import { AppState, Config, Data, UiState } from "../../types";
 
 type Props = {
@@ -65,4 +70,29 @@ describe("State reducer", () => {
       expect(newState.ui.itemSelector?.zone).toBe(rootDroppableId);
     });
   });
+
+  describe("remove action", () => {
+    it("should deselect the item", () => {
+      const state: AppState = {
+        ui: defaultUi,
+        data: {
+          ...defaultData,
+          content: [
+            {
+              type: "Comp",
+              props: { id: "sampleId", prop: "Some example data" },
+            },
+          ],
+        },
+      };
+      const action: RemoveAction = {
+        type: "remove",
+        index: 0,
+        zone: rootDroppableId,
+      };
+
+      const newState = reducer(state, action);
+      expect(newState.ui.itemSelector).toBeNull();
+    });
+  });
 });

packages/core/reducer/state.ts

@@ -23,5 +23,12 @@ export const reduceUi = (ui: UiState, action: PuckAction): UiState => {
     };
   }
 
+  if (action.type === "remove") {
+    return {
+      ...ui,
+      itemSelector: null,
+    };
+  }
+
   return ui;
 };