From 76fa5100532b340ac09c130a406c4646a13c7fbf Mon Sep 17 00:00:00 2001
From: Ramon Wenger <ramon.wenger@iterativ.ch>
Date: Wed, 12 Feb 2020 20:28:46 +0100
Subject: [PATCH] Update local school class cache after joining a class

---
 client/src/graphql/gql/mutations/joinClass.gql |  4 ++++
 client/src/pages/joinClass.vue                 | 13 +++++++++++++
 2 files changed, 17 insertions(+)

diff --git a/client/src/graphql/gql/mutations/joinClass.gql b/client/src/graphql/gql/mutations/joinClass.gql
index 5386a337..025ff5b7 100644
--- a/client/src/graphql/gql/mutations/joinClass.gql
+++ b/client/src/graphql/gql/mutations/joinClass.gql
@@ -1,5 +1,9 @@
 mutation JoinClass($input: JoinClassInput!) {
   joinClass(input: $input) {
     success
+    schoolClass {
+      id
+      name
+    }
   }
 }
diff --git a/client/src/pages/joinClass.vue b/client/src/pages/joinClass.vue
index 26456e62..d34eaca0 100644
--- a/client/src/pages/joinClass.vue
+++ b/client/src/pages/joinClass.vue
@@ -21,6 +21,7 @@
 
 <script>
   import JOIN_CLASS_MUTATION from '@/graphql/gql/mutations/joinClass.gql';
+  import ME_QUERY from '@/graphql/gql/meQuery.gql';
 
   export default {
     data: () => ({
@@ -40,6 +41,18 @@
             input: {
               code
             }
+          },
+          update(store, {data: {joinClass: {schoolClass}}}) {
+            if (schoolClass) {
+              const data = store.readQuery({query: ME_QUERY});
+              if (data) {
+                data.me.schoolClasses.edges.push({
+                  node: schoolClass,
+                  __typename: 'SchoolClassNode'
+                });
+                store.writeQuery({query: ME_QUERY, data});
+              }
+            }
           }
         }).catch(e => {
           console.debug(e);