1 files changed, 18 insertions, 0 deletions
diff --git a/frontend/enhancers.js b/frontend/enhancers.js
new file mode 100644
index 0000000..d86ca6d
--- /dev/null
+++ b/frontend/enhancers.js
@@ -0,0 +1,18 @@
+import { REMOVE_FROM_CART_MUTATION, CURRENT_USER_QUERY } from './queries';
+
+import { graphql } from 'react-apollo';
+
+export const removeFromCartEnhancer = graphql(REMOVE_FROM_CART_MUTATION, {
+  name: 'removeFromCart',
+  options: {
+    update: (proxy, payload) => {
+      const data = proxy.readQuery({ query: CURRENT_USER_QUERY });
+      const cartItemId = payload.data.removeFromCartItems.cartItem.id;
+      data.user.cart = data.user.cart.filter(item => item.id !== cartItemId);
+      proxy.writeQuery({ query: CURRENT_USER_QUERY, data });
+    },
+  },
+});
+
+// Current User Query
+export const userEnhancer = graphql(CURRENT_USER_QUERY, { name: 'currentUserQuery' });