prepopulate cache with newly added links

Automattic · Nov 14, 2023 · 441292c · 441292c
1 parent 7e349d8
commit 441292c
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/client/me/profile-links/data/use-add-profile-link-mutation.ts b/client/me/profile-links/data/use-add-profile-link-mutation.ts
@@ -6,6 +6,7 @@ import type {
 	AddProfileLinksPayload,
 	AddProfileLinksResponse,
 	AddProfileLinksVariables,
+	ProfileLink,
 } from './types';
 
 export const useAddProfileLinkMutation = ( options: AddProfileLinksMutationOptions = {} ) => {
@@ -16,6 +17,18 @@ export const useAddProfileLinkMutation = ( options: AddProfileLinksMutationOptio
 		},
 		...options,
 		onSuccess: ( ...args ) => {
+			const [ data ] = args;
+
+			queryClient.setQueryData(
+				[ 'profile-links' ],
+				( cachedProfileLinks: ProfileLink[] | undefined ) => {
+					if ( ! cachedProfileLinks || ! data.added ) {
+						return cachedProfileLinks;
+					}
+					return [ ...cachedProfileLinks, ...data.added ];
+				}
+			);
+
 			queryClient.invalidateQueries( [ 'profile-links' ] );
 			options.onSuccess?.( ...args );
 		},