chore(Customer): refetch on add subscription

getlago · Mar 10, 2023 · aa35d90 · aa35d90
1 parent 76edf62
commit aa35d90
Showing 1 changed file with 1 addition and 29 deletions.
diff --git a/src/hooks/customer/useAddSubscription.tsx b/src/hooks/customer/useAddSubscription.tsx
@@ -12,8 +12,6 @@ import {
   LagoApiError,
   CreateSubscriptionInput,
   CustomerSubscriptionFragmentDoc,
-  CustomerDetailsFragmentDoc,
-  CustomerDetailsFragment,
 } from '~/generated/graphql'
 import { SubscriptionUpdateInfo, addToast, hasDefinedGQLError } from '~/core/apolloClient'
 import { ComboBoxProps } from '~/components/form'
@@ -91,6 +89,7 @@ export const useAddSubscription: UseAddSubscription = ({
         })
       }
     },
+    refetchQueries: ['getCustomer'],
   })
 
   const selectedPlan = useMemo(() => {
@@ -176,33 +175,6 @@ export const useAddSubscription: UseAddSubscription = ({
             ...values,
           },
         },
-        update(cache, { data: createData }) {
-          if (!createData?.createSubscription) return
-
-          const cacheId = `Customer:${customerId}`
-
-          const previousData: CustomerDetailsFragment | null = cache.readFragment({
-            id: cacheId,
-            fragment: CustomerDetailsFragmentDoc,
-            fragmentName: 'CustomerDetails',
-          })
-
-          cache.writeFragment({
-            id: cacheId,
-            fragment: CustomerDetailsFragmentDoc,
-            fragmentName: 'CustomerDetails',
-            data: {
-              ...previousData,
-              subscriptions: [
-                createData?.createSubscription,
-                ...(previousData?.subscriptions || []).map((s) => ({
-                  ...s,
-                  __typename: 'Subscription', // The query has nested fragment and the typename is removed - we need to re-add it for it to work
-                })),
-              ],
-            },
-          })
-        },
       })
 
       if (!hasDefinedGQLError('CurrenciesDoesNotMatch', errors)) {