From 071f0665eb21da3c1d74e9fb266b91a33175fc2f Mon Sep 17 00:00:00 2001
From: Alaister Young <a@alaisteryoung.com>
Date: Wed, 30 Aug 2023 15:45:58 +1000
Subject: [PATCH] hotfix: invalidate permissions on profile creation

---
 studio/data/profile/profile-create-mutation.ts | 8 +++++++-
 studio/lib/profile.tsx                         | 3 ---
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/studio/data/profile/profile-create-mutation.ts b/studio/data/profile/profile-create-mutation.ts
index a6a3f358a4c..37ff0d23424 100644
--- a/studio/data/profile/profile-create-mutation.ts
+++ b/studio/data/profile/profile-create-mutation.ts
@@ -1,6 +1,8 @@
 import { useMutation, UseMutationOptions, useQueryClient } from '@tanstack/react-query'
 import { toast } from 'react-hot-toast'
 
+import { organizationKeys } from 'data/organizations/keys'
+import { permissionKeys } from 'data/permissions/keys'
 import { post } from 'lib/common/fetch'
 import { API_URL } from 'lib/constants'
 import { ResponseError } from 'types'
@@ -29,7 +31,11 @@ export const useProfileCreateMutation = ({
 
   return useMutation<ProfileCreateData, ResponseError, void>(() => createProfile(), {
     async onSuccess(data, variables, context) {
-      await queryClient.invalidateQueries(profileKeys.profile())
+      await Promise.all([
+        queryClient.invalidateQueries(profileKeys.profile()),
+        queryClient.invalidateQueries(organizationKeys.list()),
+        queryClient.invalidateQueries(permissionKeys.list()),
+      ])
       await onSuccess?.(data, variables, context)
     },
     async onError(data, variables, context) {
diff --git a/studio/lib/profile.tsx b/studio/lib/profile.tsx
index d8feec8dcbc..5810bce5301 100644
--- a/studio/lib/profile.tsx
+++ b/studio/lib/profile.tsx
@@ -3,7 +3,6 @@ import { useRouter } from 'next/router'
 import { PropsWithChildren, createContext, useContext, useMemo } from 'react'
 
 import { useTelemetryProps } from 'common'
-import { invalidateOrganizationsQuery } from 'data/organizations/organizations-query'
 import { usePermissionsQuery } from 'data/permissions/permissions-query'
 import { useProfileCreateMutation } from 'data/profile/profile-create-mutation'
 import { useProfileQuery } from 'data/profile/profile-query'
@@ -41,8 +40,6 @@ export const ProfileProvider = ({ children }: PropsWithChildren<{}>) => {
         telemetryProps,
         router
       )
-
-      await invalidateOrganizationsQuery(queryClient)
     },
     onError() {
       ui.setNotification({