Add shatter state change to context menu

districtr · Sep 24, 2024 · fb1d9e4 · fb1d9e4
1 parent d79cb20
commit fb1d9e4
Showing 1 changed file with 32 additions and 20 deletions.
diff --git a/app/src/app/components/ContextMenu.tsx b/app/src/app/components/ContextMenu.tsx
@@ -8,28 +8,32 @@ import {
   BLOCK_LAYER_ID,
   BLOCK_LAYER_ID_CHILD,
 } from "@constants/layers";
+import { ShatterResult } from "../api/apiHandlers";
 
 export const MapContextMenu: React.FC = () => {
-  const { mapRef, mapDocument, contextMenu } = useMapStore((state) => {
-    return {
-      mapRef: state.mapRef,
-      mapDocument: state.mapDocument,
-      contextMenu: state.contextMenu,
-    };
-  });
-  const patchShatter = useMutation({
+  const { mapRef, mapDocument, contextMenu, shatterIds, setShatterIds } =
+    useMapStore((state) => {
+      return {
+        mapRef: state.mapRef,
+        mapDocument: state.mapDocument,
+        contextMenu: state.contextMenu,
+        shatterIds: state.shatterIds,
+        setShatterIds: state.setShatterIds,
+      };
+    });
+  const patchShatter = useMutation<ShatterResult>({
     mutationFn: patchShatterParents,
     onMutate: ({ document_id, geoids }) => {
       console.log(
-        `Shattering parents for ${geoids} in document ${document_id}...`,
+        `Shattering parents for ${geoids} in document ${document_id}...`
       );
     },
     onError: (error) => {
       console.log("Error updating assignments: ", error);
     },
     onSuccess: (data) => {
       console.log(
-        `Successfully shattered parents into ${data.children.length} children`,
+        `Successfully shattered parents into ${data.children.length} children`
       );
       // mapRef?.current?.setFilter(BLOCK_LAYER_ID_CHILD, [
       //   "match",
@@ -52,24 +56,32 @@ export const MapContextMenu: React.FC = () => {
       //     },
       //   );
       // });
-      console.log(data.parents.geoids);
+
       data.parents.geoids.forEach((parent) =>
         mapRef?.current?.setFeatureState(
           {
             source: BLOCK_SOURCE_ID,
             id: parent,
             sourceLayer: BLOCK_LAYER_ID,
           },
-          { selected: false, zone: null },
-        ),
+          { selected: false, zone: null }
+        )
       );
-      mapRef?.current?.setFilter(BLOCK_LAYER_ID, [
-        "match",
-        ["get", "path"],
-        data.parents.geoids, // will need to add existing filters
-        false,
-        true,
-      ]);
+
+      setShatterIds({
+        parents: [...shatterIds.parents, ...data.parents.geoids],
+        children: [
+          ...shatterIds.children,
+          ...data.children.map((child) => child.geo_id),
+        ],
+      });
+      // mapRef?.current?.setFilter(BLOCK_LAYER_ID, [
+      //   "match",
+      //   ["get", "path"],
+      //   data.parents.geoids, // will need to add existing filters
+      //   false,
+      //   true,
+      // ]);
     },
   });