fix: Delete sites from project slice on update (#149)

techmatters · Oct 30, 2023 · 3e2683f · 3e2683f
1 parent ca1f283
commit 3e2683f
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/site/siteSlice.ts b/src/site/siteSlice.ts
@@ -19,6 +19,7 @@ import { createSlice, PayloadAction } from '@reduxjs/toolkit';
 import {
   SiteAddMutationInput,
   SiteTransferMutationInput,
+  SiteUpdateMutationInput,
 } from 'terraso-client-shared/graphqlSchema/graphql';
 import {
   addSiteToProject,
@@ -72,9 +73,18 @@ export const addSite = createAsyncThunk<Site, SiteAddMutationInput>(
   },
 );
 
-export const updateSite = createAsyncThunk(
+export const updateSite = createAsyncThunk<Site, SiteUpdateMutationInput>(
   'site/updateSite',
-  siteService.updateSite,
+  async (input, _currentUser, { dispatch }) => {
+    const result = await siteService.updateSite(input);
+    dispatch(removeSiteFromAllProjects(result.id));
+    if (result.projectId) {
+      dispatch(
+        addSiteToProject({ projectId: result.projectId, siteId: input.id }),
+      );
+    }
+    return result;
+  },
 );
 
 export const deleteSite = createAsyncThunk<string, Site>(