Merge pull request #2036 from JoinColony/feature/2005-structured-work…

…-aside Add structured work to colony home aside
JoinColony · Feb 20, 2020 · 1876981 · 1876981
2 parents 0c296cc + 897997c
commit 1876981
Show file tree

Hide file tree

Showing 11 changed files with 696 additions and 8 deletions.
diff --git a/src/data/cacheUpdates.ts b/src/data/cacheUpdates.ts
@@ -22,11 +22,50 @@ import {
   UserQuery,
   UserQueryVariables,
   UserDocument,
+  CreateProgramMutationResult,
+  ColonyProgramsQuery,
+  ColonyProgramsQueryVariables,
+  ColonyProgramsDocument,
 } from './generated';
 
 type Cache = typeof apolloCache;
 
 const cacheUpdates = {
+  createProgram(colonyAddress: Address) {
+    return (cache: Cache, { data }: CreateProgramMutationResult) => {
+      try {
+        const cacheData = cache.readQuery<
+          ColonyProgramsQuery,
+          ColonyProgramsQueryVariables
+        >({
+          query: ColonyProgramsDocument,
+          variables: {
+            address: colonyAddress,
+          },
+        });
+        const createProgramData = data && data.createProgram;
+        if (cacheData && createProgramData) {
+          const programs = cacheData.colony.programs || [];
+          programs.push(createProgramData);
+          cache.writeQuery<ColonyProgramsQuery, ColonyProgramsQueryVariables>({
+            data: {
+              colony: {
+                ...cacheData.colony,
+                programs,
+              },
+            },
+            query: ColonyProgramsDocument,
+            variables: {
+              address: colonyAddress,
+            },
+          });
+        }
+      } catch (e) {
+        log.verbose(e);
+        log.verbose('Not updating store - colony programs not loaded yet');
+      }
+    };
+  },
   createTask(colonyAddress: Address) {
     return (cache: Cache, { data }: CreateTaskMutationResult) => {
       try {