client/reader/data/site-subscriptions/cache.ts - trunk - githubmirror/wp-calypso

client/reader/data/site-subscriptions/cache.ts

60 строк2 KB

Gabriel Caires

Reader: Migrate Reader follows layer to React Query (#111267)

03 июн 2026, 12:06

03 июн 2026, 12:06fbb8787

import { prepareComparableUrl, type ReadSiteResponse } from '@automattic/api-core';
import { getSiteSubscriptionsQueryKey, readSiteQuery } from '@automattic/api-queries';
import type { QueryClient } from '@tanstack/react-query';

const FOLLOW_SENSITIVE_QUERY_KEYS = [
	getSiteSubscriptionsQueryKey(),
	[ 'read', 'stream', 'following' ],
	[ 'read', 'stream', 'infinite', 'following' ],
	[ 'read', 'subscriptions-count' ],
] as const;

const getNumericId = ( id: number | string ): number | undefined => {
	const numericId = typeof id === 'string' ? Number( id ) : id;

	return typeof numericId === 'number' && Number.isFinite( numericId ) && numericId > 0
		? numericId
		: undefined;
};

export const patchReadSiteFollowStatus = (
	queryClient: QueryClient,
	feedUrl: string,
	isFollowing: boolean
) => {
	const comparableFeedUrl = prepareComparableUrl( feedUrl );
	if ( ! comparableFeedUrl ) {
		return;
	}

	for ( const [ queryKey, site ] of queryClient.getQueriesData< ReadSiteResponse >( {
		queryKey: [ 'read', 'sites' ],
	} ) ) {
		if ( site && prepareComparableUrl( site.feed_URL ) === comparableFeedUrl ) {
			queryClient.setQueryData< ReadSiteResponse >( queryKey, {
				...site,
				is_following: isFollowing,
			} );
		}
	}
};

export const patchReadSiteFollowStatusByBlogId = (
	queryClient: QueryClient,
	blogId: number | string,
	isFollowing: boolean
) => {
	const numericId = getNumericId( blogId );
	if ( ! numericId ) {
		return;
	}

	queryClient.setQueryData< ReadSiteResponse >( readSiteQuery( numericId ).queryKey, ( site ) =>
		site ? { ...site, is_following: isFollowing } : site
	);
};

export const invalidateFollowSensitiveCaches = ( queryClient: QueryClient ) =>
	Promise.all(
		FOLLOW_SENSITIVE_QUERY_KEYS.map( ( queryKey ) => queryClient.invalidateQueries( { queryKey } ) )
	);