@automattic/jetpack-shared-extension-utils 2.0.0 → 2.0.1

package/CHANGELOG.md

@@ -5,6 +5,11 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [2.0.1] - 2026-06-22
+### Changed
+- Move the WordPress.com plans and Jetpack modules data stores to the new `jetpack-shared-stores` package. [#49494]
+- Update package dependencies. [#49631] [#49691] [#49757]
+
 ## [2.0.0] - 2026-06-15
 ### Removed
 - Remove `react-native` export condition. [#49526]
@@ -974,6 +979,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Changed
 - Core: prepare utility for release
 
+[2.0.1]: https://github.com/Automattic/jetpack-shared-extension-utils/compare/2.0.0...2.0.1
 [2.0.0]: https://github.com/Automattic/jetpack-shared-extension-utils/compare/1.5.22...2.0.0
 [1.5.22]: https://github.com/Automattic/jetpack-shared-extension-utils/compare/1.5.21...1.5.22
 [1.5.21]: https://github.com/Automattic/jetpack-shared-extension-utils/compare/1.5.20...1.5.21

package/index.js

@@ -23,7 +23,7 @@ export { default as useRefInterval } from './src/hooks/use-ref-interval';
 export { default as useModuleStatus } from './src/hooks/use-module-status';
 export { getBlockIconComponent, getBlockIconProp } from './src/get-block-icon-from-metadata';
 export { default as getJetpackBlocksVariation } from './src/get-jetpack-blocks-variation';
-export * from './src/modules-state';
+export { store, JETPACK_MODULES_STORE_ID } from '@automattic/jetpack-shared-stores';
 export { default as isMyJetpackAvailable } from './src/is-my-jetpack-available';
 export { default as hasFeatureFlag } from './src/has-feature-flag';
 export * from './src/libs';

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@automattic/jetpack-shared-extension-utils",
-	"version": "2.0.0",
+	"version": "2.0.1",
 	"description": "Utility functions used by the block editor extensions",
 	"homepage": "https://github.com/Automattic/jetpack/tree/HEAD/projects/js-packages/shared-extension-utils/#readme",
 	"bugs": {
@@ -29,33 +29,34 @@
 	"dependencies": {
 		"@automattic/color-studio": "4.1.0",
 		"@automattic/jetpack-analytics": "^1.0.15",
-		"@automattic/jetpack-base-styles": "^1.2.6",
-		"@automattic/jetpack-components": "^1.12.7",
-		"@automattic/jetpack-connection": "^1.4.60",
+		"@automattic/jetpack-base-styles": "^1.2.7",
+		"@automattic/jetpack-components": "^1.12.9",
+		"@automattic/jetpack-connection": "^1.4.62",
 		"@automattic/jetpack-script-data": "^0.6.4",
+		"@automattic/jetpack-shared-stores": "^0.1.0",
 		"@types/jest": "30.0.0",
-		"@wordpress/api-fetch": "7.48.0",
-		"@wordpress/block-editor": "15.21.0",
-		"@wordpress/components": "35.0.0",
-		"@wordpress/compose": "8.1.0",
-		"@wordpress/data": "10.48.0",
-		"@wordpress/dom-ready": "4.48.0",
-		"@wordpress/element": "8.0.0",
-		"@wordpress/hooks": "4.48.0",
-		"@wordpress/i18n": "6.21.0",
-		"@wordpress/plugins": "7.48.0",
-		"@wordpress/primitives": "4.48.0",
+		"@wordpress/api-fetch": "7.48.1",
+		"@wordpress/block-editor": "15.21.1",
+		"@wordpress/components": "35.0.1",
+		"@wordpress/compose": "8.1.1",
+		"@wordpress/data": "10.48.1",
+		"@wordpress/dom-ready": "4.48.1",
+		"@wordpress/element": "8.0.1",
+		"@wordpress/hooks": "4.48.1",
+		"@wordpress/i18n": "6.21.1",
+		"@wordpress/plugins": "7.48.1",
+		"@wordpress/primitives": "4.48.1",
 		"@wordpress/ui": "0.13.0",
-		"@wordpress/url": "4.48.0",
+		"@wordpress/url": "4.48.1",
 		"clsx": "2.1.1",
 		"debug": "4.4.3"
 	},
 	"devDependencies": {
 		"@automattic/jetpack-webpack-config": "workspace:*",
-		"@babel/core": "7.29.0",
-		"@babel/plugin-transform-react-jsx": "7.28.6",
-		"@babel/preset-react": "7.28.5",
-		"@babel/runtime": "7.29.2",
+		"@babel/core": "7.29.7",
+		"@babel/plugin-transform-react-jsx": "7.29.7",
+		"@babel/preset-react": "7.29.7",
+		"@babel/runtime": "7.29.7",
 		"@testing-library/dom": "10.4.1",
 		"@testing-library/react": "16.3.2",
 		"@testing-library/user-event": "14.6.1",

package/src/hooks/use-module-status/index.js

@@ -1,6 +1,6 @@
+import { JETPACK_MODULES_STORE_ID } from '@automattic/jetpack-shared-stores';
 import { useDispatch, useSelect } from '@wordpress/data';
 import { useMemo, useCallback } from '@wordpress/element';
-import { JETPACK_MODULES_STORE_ID } from '../../modules-state';
 
 /**
  * @typedef {object} ModuleStatus

package/src/store/wordpress-com/index.ts

@@ -1,108 +1,10 @@
 /**
- * External dependencies
+ * Back-compat re-export.
+ *
+ * The `wordpress-com/plans` store moved to `@automattic/jetpack-shared-stores`
+ * so it can be externalized into a single bundle and registered only once.
+ * This shim preserves the historical `@automattic/jetpack-shared-extension-utils/store/wordpress-com`
+ * import path for existing consumers.
  */
-import { createReduxStore, register } from '@wordpress/data';
-/**
- * Internal dependencies
- */
-import actions from './actions.ts';
-import reducer from './reducer.ts';
-/**
- * Types
- */
-import type { AiFeatureProps, PlanStateProps } from './types.ts';
-
-const store = 'wordpress-com/plans';
-
-export const selectors = {
-	/*
-	 * Return the plan with the given slug.
-	 *
-	 * @param {Object} state    - The Plans state tree.
-	 * @param {string} planSlug - The plan slug to find.
-	 * @return {Object}           The plan.
-	 */
-	getPlan( state: PlanStateProps, planSlug: string ) {
-		return state.plans.find( plan => plan.product_slug === planSlug );
-	},
-
-	/**
-	 * Return the AI Assistant feature.
-	 *
-	 * @param {PlanStateProps} state - The Plans state tree.
-	 * @return {AiFeatureProps}       The AI Assistant feature data.
-	 */
-	getAiAssistantFeature( state: PlanStateProps ): AiFeatureProps {
-		// Clean up the _meta property.
-		const data = { ...state.features.aiAssistant };
-		delete data._meta;
-
-		return data;
-	},
-
-	/**
-	 * Get the isRequesting flag for the AI Assistant feature.
-	 *
-	 * @param {PlanStateProps} state - The Plans state tree.
-	 * @return {boolean}              The isRequesting flag.
-	 */
-	getIsRequestingAiAssistantFeature( state: PlanStateProps ): boolean {
-		return state.features.aiAssistant?._meta?.isRequesting;
-	},
-
-	getAsyncRequestCountdownValue( state: PlanStateProps ): number {
-		return state.features.aiAssistant?._meta?.asyncRequestCountdown;
-	},
-
-	getAsyncRequestCountdownTimerId( state: PlanStateProps ): number {
-		return state.features.aiAssistant?._meta?.asyncRequestTimerId;
-	},
-};
-
-export const wordpressPlansStore = createReduxStore( store, {
-	actions,
-
-	reducer,
-
-	selectors,
-
-	controls: {
-		FETCH_FROM_API( { url } ) {
-			// We cannot use `@wordpress/api-fetch` here since it unconditionally sends
-			// the `X-WP-Nonce` header, which is disallowed by WordPress.com.
-			// (To reproduce, note that you need to call `apiFetch` with `
-			// `{ credentials: 'same-origin', mode: 'cors' }`, since its defaults are
-			// different from `fetch`'s.)
-			return fetch( url ).then( response => response.json() );
-		},
-	},
-
-	resolvers: {
-		*getPlan() {
-			const url = 'https://public-api.wordpress.com/rest/v1.5/plans';
-			const plans = yield actions.fetchFromAPI( url );
-			return actions.setPlans( plans );
-		},
-
-		getAiAssistantFeature: ( state: PlanStateProps ) => {
-			if ( state?.features?.aiAssistant ) {
-				return;
-			}
-
-			return actions.fetchAiAssistantFeature();
-		},
-	},
-} );
-
-register( wordpressPlansStore );
-
-// Types
-
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-type OmitFirstArg< F > = F extends ( _: any, ...args: infer P ) => infer R
-	? ( ...args: P ) => R
-	: never;
-
-export type WordPressPlansSelectors = {
-	[ key in keyof typeof selectors ]: OmitFirstArg< ( typeof selectors )[ key ] >;
-};
+export { selectors, wordpressPlansStore } from '@automattic/jetpack-shared-stores';
+export type * from '@automattic/jetpack-shared-stores';

package/src/store/wordpress-com/types.ts

@@ -1,161 +1,8 @@
-export type Plan = {
-	product_id: number;
-	product_name: string;
-	product_slug: string;
-};
-// AI Assistant feature props
-export type UpgradeTypeProp = 'vip' | 'default';
-
-export type TierUnlimitedProps = {
-	slug: 'ai-assistant-tier-unlimited';
-	limit: 999999999 | 3000;
-	value: 1;
-	readableLimit: string;
-};
-
-export type TierFreeProps = {
-	slug: 'ai-assistant-tier-free';
-	limit: 20;
-	value: 0;
-};
-
-export type Tier100Props = {
-	slug: 'ai-assistant-tier-100';
-	limit: 100;
-	value: 100;
-};
-
-export type Tier200Props = {
-	slug: 'ai-assistant-tier-200';
-	limit: 200;
-	value: 200;
-};
-
-export type Tier500Props = {
-	slug: 'ai-assistant-tier-500';
-	limit: 500;
-	value: 500;
-};
-
-export type Tier750Props = {
-	slug: 'ai-assistant-tier-750';
-	limit: 750;
-	value: 750;
-};
-
-export type Tier1000Props = {
-	slug: 'ai-assistant-tier-1000';
-	limit: 1000;
-	value: 1000;
-};
-
-export type TierProp = {
-	slug: TierSlugProp;
-	limit: TierLimitProp;
-	value: TierValueProp;
-	readableLimit?: string;
-};
-
-export type TierLimitProp =
-	| TierUnlimitedProps[ 'limit' ]
-	| TierFreeProps[ 'limit' ]
-	| Tier100Props[ 'limit' ]
-	| Tier200Props[ 'limit' ]
-	| Tier500Props[ 'limit' ]
-	| Tier750Props[ 'limit' ]
-	| Tier1000Props[ 'limit' ];
-
-export type TierSlugProp =
-	| TierUnlimitedProps[ 'slug' ]
-	| TierFreeProps[ 'slug' ]
-	| Tier100Props[ 'slug' ]
-	| Tier200Props[ 'slug' ]
-	| Tier500Props[ 'slug' ]
-	| Tier750Props[ 'slug' ]
-	| Tier1000Props[ 'slug' ];
-
-export type TierValueProp =
-	| TierUnlimitedProps[ 'value' ]
-	| TierFreeProps[ 'value' ]
-	| Tier100Props[ 'value' ]
-	| Tier200Props[ 'value' ]
-	| Tier500Props[ 'value' ]
-	| Tier750Props[ 'value' ]
-	| Tier1000Props[ 'value' ];
-
-export type FeatureControl = {
-	enabled: boolean;
-	'min-jetpack-version': string;
-	[ key: string ]: FeatureControl | boolean | string;
-};
-
-export type FeaturesControl = { [ key: string ]: FeatureControl };
-
-export type AiFeatureProps = {
-	hasFeature: boolean;
-	isOverLimit: boolean;
-	requestsCount: number;
-	requestsLimit: number;
-	requireUpgrade: boolean;
-	errorMessage?: string;
-	errorCode?: string;
-	upgradeType: UpgradeTypeProp;
-	currentTier?: TierProp;
-	usagePeriod?: {
-		currentStart: string;
-		nextStart: string;
-		requestsCount: number;
-	};
-	nextTier?: TierProp | null;
-	tierPlansEnabled?: boolean;
-	costs?: {
-		[ key: string ]: {
-			[ key: string ]: number;
-		};
-	};
-	featuresControl?: FeaturesControl;
-};
-
-// Type used in the `wordpress-com/plans` store.
-export type AiFeatureStateProps = AiFeatureProps & {
-	_meta?: {
-		isRequesting: boolean;
-		asyncRequestCountdown: number;
-		asyncRequestTimerId: number;
-	};
-};
-
-export type PlanStateProps = {
-	plans: Array< Plan >;
-	features: {
-		aiAssistant?: AiFeatureStateProps;
-	};
-};
-
-/*
- * `sites/$site/ai-assistant-feature` endpoint response body props
+/**
+ * Back-compat re-export.
+ *
+ * The `wordpress-com/plans` store types moved to `@automattic/jetpack-shared-stores`.
+ * This shim preserves the historical
+ * `@automattic/jetpack-shared-extension-utils/store/wordpress-com/types` import path.
  */
-export type SiteAIAssistantFeatureEndpointResponseProps = {
-	'has-feature': boolean;
-	'is-over-limit': boolean;
-	'requests-count': number;
-	'requests-limit': number;
-	'usage-period': {
-		'current-start': string;
-		'next-start': string;
-		'requests-count': number;
-	};
-	'site-require-upgrade': boolean;
-	'error-message'?: string;
-	'error-code'?: string;
-	'upgrade-type': UpgradeTypeProp;
-	'current-tier': TierProp;
-	'tier-plans': Array< TierProp >;
-	'next-tier'?: TierProp | null;
-	costs?: {
-		[ key: string ]: {
-			[ key: string ]: number;
-		};
-	};
-	'features-control'?: FeaturesControl;
-};
+export type * from '@automattic/jetpack-shared-stores';

package/src/modules-state/actions.js

@@ -1,94 +0,0 @@
-import { isSimpleSite } from '@automattic/jetpack-script-data';
-import { select } from '@wordpress/data';
-import {
-	fetchJetpackModules,
-	updateJetpackModuleStatus as updateJetpackModuleStatusControl,
-} from './controls';
-import { JETPACK_MODULES_STORE_ID } from '.';
-
-export const SET_JETPACK_MODULES = 'SET_JETPACK_MODULES';
-export const SET_MODULE_UPDATING = 'SET_MODULE_UPDATING';
-
-/**
- * Yield actions to update module status
- *
- * @param {object}  settings        - Jetpack module settings.
- * @param {string}  settings.name   - Jetpack module name.
- * @param {boolean} settings.active - If the module is active or not.
- * @yield {object} - an action object.
- * @return {object} - an action object.
- */
-export function* updateJetpackModuleStatus( settings ) {
-	try {
-		yield setIsUpdating( settings.name, true );
-		yield updateJetpackModuleStatusControl( settings );
-		const data = yield fetchJetpackModules();
-		yield setJetpackModules( { data } );
-		return true;
-	} catch {
-		const oldSettings = select( JETPACK_MODULES_STORE_ID ).getJetpackModules();
-		yield setJetpackModules( oldSettings );
-		return false;
-	} finally {
-		yield setIsUpdating( settings.name, false );
-	}
-}
-
-/**
- * Yield actions to update module status
- * @yield {object} - an action object.
- * @return {boolean} - if operation is successful or not.
- */
-export function* fetchModules() {
-	// We don't fetch modules for Simple Site and aknowledge that all modules are active
-	if ( isSimpleSite() ) {
-		return true;
-	}
-	try {
-		yield setIsLoading( true );
-		const data = yield fetchJetpackModules();
-		yield setJetpackModules( { data } );
-		return true;
-	} catch {
-		const oldSettings = select( JETPACK_MODULES_STORE_ID ).getJetpackModules();
-		yield setJetpackModules( oldSettings );
-		return false;
-	} finally {
-		yield setIsLoading( false );
-	}
-}
-
-/**
- * Set modules as loading action
- *
- * @param {boolean} isLoading - If the modules are loading or not.
- * @return {object} - an action object.
- */
-export function setIsLoading( isLoading ) {
-	return setJetpackModules( { isLoading } );
-}
-
-/**
- * Set modules as updating action
- *
- * @param {string}  name       - Name of the module.
- * @param {boolean} isUpdating - If the modules are updating or not.
- * @return {object} - an action object.
- */
-function setIsUpdating( name, isUpdating ) {
-	return { type: SET_MODULE_UPDATING, name, isUpdating };
-}
-
-/**
- * Set Jetpack module action
- *
- * @param {object}  options           - Jetpack settings.
- * @param {object}  options.modules   - Jetpack modules.
- * @param {boolean} options.isLoading - If the modules are loading or not.
- * @return {object} - an action object.
- */
-export function setJetpackModules( options ) {
-	return { type: SET_JETPACK_MODULES, options };
-}
-
-export default { updateJetpackModuleStatus, setJetpackModules, fetchModules };

package/src/modules-state/controls.js

@@ -1,48 +0,0 @@
-import apiFetch from '@wordpress/api-fetch';
-
-export const FETCH_JETPACK_MODULES = 'FETCH_JETPACK_MODULES';
-export const UPDATE_JETPACK_MODULE_STATUS = 'UPDATE_JETPACK_MODULE_STATUS';
-
-/**
- * fetchJetpackModules action
- *
- * @return {object} - an action object.
- */
-export const fetchJetpackModules = () => {
-	return {
-		type: FETCH_JETPACK_MODULES,
-	};
-};
-
-/**
- * Updating single module status action
- *
- * @param           settings        - Jetpack module settings.
- * @param {string}  settings.name   - Jetpack module name.
- * @param {boolean} settings.active - If the module is active or not.
- */
-
-export const updateJetpackModuleStatus = settings => {
-	return {
-		type: UPDATE_JETPACK_MODULE_STATUS,
-		settings,
-	};
-};
-
-export default {
-	[ FETCH_JETPACK_MODULES ]: function () {
-		return apiFetch( {
-			path: `/jetpack/v4/module/all`,
-			method: 'GET',
-		} );
-	},
-	[ UPDATE_JETPACK_MODULE_STATUS ]: function ( { settings } ) {
-		return apiFetch( {
-			path: `/jetpack/v4/module/${ settings.name }/active`,
-			method: 'POST',
-			data: {
-				active: settings.active,
-			},
-		} );
-	},
-};

package/src/modules-state/index.js

@@ -1,30 +0,0 @@
-import { createReduxStore, register, dispatch } from '@wordpress/data';
-import actions from './actions';
-import controls from './controls';
-import reducer from './reducer';
-import resolvers from './resolvers';
-import selectors from './selectors';
-
-export const JETPACK_MODULES_STORE_ID = 'jetpack-modules';
-export const store = createReduxStore( JETPACK_MODULES_STORE_ID, {
-	reducer,
-	actions,
-	controls,
-	resolvers,
-	selectors,
-} );
-
-register( store );
-
-const initialData =
-	window?.Initial_State?.getModules || // Jetpack Dashboard
-	window?.Jetpack_Editor_Initial_State?.modules || // Gutenberg
-	null;
-
-// This is a temporary fix to have store filled properly.
-// TODO: Create a proper solution after fixing initial issue (https://github.com/Automattic/jetpack/issues/34793).
-if ( initialData !== null ) {
-	dispatch( JETPACK_MODULES_STORE_ID ).setJetpackModules( {
-		data: { ...initialData },
-	} );
-}

package/src/modules-state/reducer.js

@@ -1,28 +0,0 @@
-const defaultState = {
-	isLoading: false,
-	isUpdating: {},
-	data: {},
-};
-
-const setModulesData = ( state = defaultState, action ) => {
-	switch ( action.type ) {
-		case 'SET_JETPACK_MODULES':
-			return {
-				...state,
-				...action.options,
-			};
-		case 'SET_MODULE_UPDATING':
-			return {
-				...state,
-				...{
-					isUpdating: {
-						...state.isUpdating,
-						[ action.name ]: action.isUpdating,
-					},
-				},
-			};
-	}
-	return state;
-};
-
-export default setModulesData;

package/src/modules-state/resolvers.js

@@ -1,24 +0,0 @@
-import { setIsLoading, setJetpackModules } from './actions';
-import { fetchJetpackModules } from './controls';
-
-/**
- * Yield actions to get the Jetpack modules.
- *
- * @yield {object} - an action object.
- * @return {object} - an action object.
- */
-export function* getJetpackModules() {
-	try {
-		yield setIsLoading( true );
-		const data = yield fetchJetpackModules();
-		if ( data ) {
-			return setJetpackModules( { data } );
-		}
-	} catch ( e ) {
-		console.error( e ); // eslint-disable-line no-console
-	} finally {
-		yield setIsLoading( false );
-	}
-}
-
-export default { getJetpackModules };

package/src/modules-state/selectors.js

@@ -1,13 +0,0 @@
-import { isSimpleSite } from '@automattic/jetpack-script-data';
-
-const jetpackModulesSelectors = {
-	getJetpackModules: state => state.data,
-	// We consider simple sites to have all modules active
-	// TODO: we would remove this when wrapping logic with hooks
-	isModuleActive: ( state, moduleName ) =>
-		isSimpleSite() || ( state?.data?.[ moduleName ]?.activated ?? false ),
-	areModulesLoading: state => state.isLoading ?? false,
-	isModuleUpdating: ( state, moduleName ) => state?.isUpdating?.[ moduleName ] ?? false,
-};
-
-export default jetpackModulesSelectors;

package/src/store/wordpress-com/actions.ts

@@ -1,194 +0,0 @@
-/**
- * External dependencies
- */
-import apiFetch from '@wordpress/api-fetch';
-/**
- * Types & Constants
- */
-import {
-	ACTION_DECREASE_NEW_ASYNC_REQUEST_COUNTDOWN,
-	ACTION_DEQUEUE_ASYNC_REQUEST,
-	ACTION_ENQUEUE_ASYNC_REQUEST,
-	ACTION_FETCH_FROM_API,
-	ACTION_INCREASE_AI_ASSISTANT_REQUESTS_COUNT,
-	ACTION_REQUEST_AI_ASSISTANT_FEATURE,
-	ACTION_SET_PLANS,
-	ACTION_SET_AI_ASSISTANT_FEATURE_REQUIRE_UPGRADE,
-	ACTION_STORE_AI_ASSISTANT_FEATURE,
-	ENDPOINT_AI_ASSISTANT_FEATURE,
-	NEW_ASYNC_REQUEST_TIMER_INTERVAL,
-	ACTION_SET_TIER_PLANS_ENABLED,
-} from './constants.ts';
-import type { Plan, AiFeatureProps, SiteAIAssistantFeatureEndpointResponseProps } from './types.ts';
-
-/**
- * Map the response from the `sites/$site/ai-assistant-feature`
- * endpoint to the AI Assistant feature props.
- * @param { SiteAIAssistantFeatureEndpointResponseProps } response - The response from the endpoint.
- * @return { AiFeatureProps }                                       The AI Assistant feature props.
- */
-export function mapAiFeatureResponseToAiFeatureProps(
-	response: SiteAIAssistantFeatureEndpointResponseProps
-): AiFeatureProps {
-	return {
-		hasFeature: !! response[ 'has-feature' ],
-		isOverLimit: !! response[ 'is-over-limit' ],
-		requestsCount: response[ 'requests-count' ],
-		requestsLimit: response[ 'requests-limit' ],
-		requireUpgrade: !! response[ 'site-require-upgrade' ],
-		errorMessage: response[ 'error-message' ],
-		errorCode: response[ 'error-code' ],
-		upgradeType: response[ 'upgrade-type' ],
-		usagePeriod: {
-			currentStart: response[ 'usage-period' ]?.[ 'current-start' ],
-			nextStart: response[ 'usage-period' ]?.[ 'next-start' ],
-			requestsCount: response[ 'usage-period' ]?.[ 'requests-count' ] || 0,
-		},
-		currentTier: response[ 'current-tier' ],
-		nextTier: response[ 'next-tier' ],
-		tierPlansEnabled: !! response[ 'tier-plans-enabled' ],
-		costs: response.costs,
-		featuresControl: response[ 'features-control' ],
-	};
-}
-
-const actions = {
-	setPlans( plans: Array< Plan > ) {
-		return {
-			type: ACTION_SET_PLANS,
-			plans,
-		};
-	},
-
-	fetchFromAPI( url: string ) {
-		return {
-			type: ACTION_FETCH_FROM_API,
-			url,
-		};
-	},
-
-	storeAiAssistantFeature( feature: AiFeatureProps ) {
-		return {
-			type: ACTION_STORE_AI_ASSISTANT_FEATURE,
-			feature,
-		};
-	},
-
-	/**
-	 * Thunk action to fetch the AI Assistant feature from the API.
-	 *
-	 * @return {Function} The thunk action.
-	 */
-	fetchAiAssistantFeature() {
-		return async ( { dispatch } ) => {
-			// Dispatch isFetching action.
-			dispatch( { type: ACTION_REQUEST_AI_ASSISTANT_FEATURE } );
-
-			try {
-				const response: SiteAIAssistantFeatureEndpointResponseProps = await apiFetch( {
-					path: ENDPOINT_AI_ASSISTANT_FEATURE,
-				} );
-
-				// Store the feature in the store.
-				dispatch(
-					actions.storeAiAssistantFeature( mapAiFeatureResponseToAiFeatureProps( response ) )
-				);
-			} catch ( err ) {
-				// @todo: Handle error.
-				console.error( err ); // eslint-disable-line no-console
-			}
-		};
-	},
-
-	/**
-	 * This thunk action is used to increase
-	 * the requests count for the current usage period.
-	 * @param {number} count - The number of requests to increase. Default is 1.
-	 * @return {Function}     The thunk action.
-	 */
-	increaseAiAssistantRequestsCount( count: number = 1 ) {
-		return ( { dispatch } ) => {
-			dispatch( {
-				type: ACTION_INCREASE_AI_ASSISTANT_REQUESTS_COUNT,
-				count,
-			} );
-
-			// Every time the requests count is increased, decrease the countdown
-			dispatch( actions.decreaseAsyncRequestCountdownValue() );
-		};
-	},
-
-	/**
-	 * This thunk action is used to decrease
-	 * the countdown value for the new async request.
-	 * When the countdown reaches 0, enqueue a new async request.
-	 *
-	 * @return {Function} The thunk action.
-	 */
-	decreaseAsyncRequestCountdownValue() {
-		return async ( { dispatch, select } ) => {
-			dispatch( { type: ACTION_DECREASE_NEW_ASYNC_REQUEST_COUNTDOWN } );
-
-			const asyncCoundown = select.getAsyncRequestCountdownValue();
-			if ( asyncCoundown <= 0 ) {
-				dispatch( actions.enqueueAiAssistantFeatureAsyncRequest() );
-			}
-		};
-	},
-
-	/**
-	 * This thunk action is used to enqueue a new async request.
-	 * If already exist an enqueue request, clear it and enqueue a new one.
-	 *
-	 * @return {Function} The thunk action.
-	 */
-	enqueueAiAssistantFeatureAsyncRequest() {
-		return ( { dispatch } ) => {
-			// Check if there is already a timer running
-			dispatch.dequeueAiAssistantFeatureAsyncRequest();
-
-			const contdownTimerId = setTimeout( () => {
-				dispatch( actions.fetchAiAssistantFeature() );
-			}, NEW_ASYNC_REQUEST_TIMER_INTERVAL ); // backend process requires a delay to be able to see the new value
-
-			dispatch( { type: ACTION_ENQUEUE_ASYNC_REQUEST, timerId: contdownTimerId } );
-		};
-	},
-
-	/**
-	 * This thunk action is used to dequeue a new async request.
-	 * It will clear the timer if there is one,
-	 * canceling the enqueue async request.
-	 *
-	 * @return {Function} The thunk action.
-	 */
-	dequeueAiAssistantFeatureAsyncRequest() {
-		return ( { dispatch, select } ) => {
-			dispatch( { type: ACTION_DEQUEUE_ASYNC_REQUEST, timerId: 0 } );
-
-			const timerId = select.getAsyncRequestCountdownTimerId();
-			// If there is no timer, there is nothing to clear
-			if ( ! timerId ) {
-				return;
-			}
-
-			window?.clearTimeout( timerId );
-		};
-	},
-
-	setAiAssistantFeatureRequireUpgrade( requireUpgrade: boolean = true ) {
-		return {
-			type: ACTION_SET_AI_ASSISTANT_FEATURE_REQUIRE_UPGRADE,
-			requireUpgrade,
-		};
-	},
-
-	setTierPlansEnabled( tierPlansEnabled: boolean = true ) {
-		return {
-			type: ACTION_SET_TIER_PLANS_ENABLED,
-			tierPlansEnabled,
-		};
-	},
-};
-
-export default actions;

package/src/store/wordpress-com/constants.ts

@@ -1,31 +0,0 @@
-/**
- * Plans actions
- */
-export const ACTION_SET_PLANS = 'SET_PLANS';
-export const ACTION_FETCH_FROM_API = 'FETCH_FROM_API';
-
-/**
- * AI Assistant feature Actions
- */
-export const ACTION_STORE_AI_ASSISTANT_FEATURE = 'STORE_AI_ASSISTANT_FEATURE';
-export const ACTION_REQUEST_AI_ASSISTANT_FEATURE = 'REQUEST_AI_ASSISTANT_FEATURE';
-export const ACTION_INCREASE_AI_ASSISTANT_REQUESTS_COUNT = 'INCREASE_AI_ASSISTANT_REQUESTS_COUNT';
-export const ACTION_SET_AI_ASSISTANT_FEATURE_REQUIRE_UPGRADE =
-	'SET_AI_ASSISTANT_FEATURE_REQUIRE_UPGRADE';
-export const ACTION_SET_TIER_PLANS_ENABLED = 'SET_TIER_PLANS_ENABLED';
-
-/**
- * Endpoints
- */
-export const ENDPOINT_AI_ASSISTANT_FEATURE = '/wpcom/v2/jetpack-ai/ai-assistant-feature';
-
-/**
- * New AI Assistant feature async request
- */
-export const FREE_PLAN_REQUESTS_LIMIT = 20;
-export const UNLIMITED_PLAN_REQUESTS_LIMIT = 3000;
-export const ASYNC_REQUEST_COUNTDOWN_INIT_VALUE = 3;
-export const NEW_ASYNC_REQUEST_TIMER_INTERVAL = 5000;
-export const ACTION_DECREASE_NEW_ASYNC_REQUEST_COUNTDOWN = 'DECREASE_NEW_ASYNC_REQUEST_COUNTDOWN';
-export const ACTION_ENQUEUE_ASYNC_REQUEST = 'ENQUEUE_ASYNC_COUNTDOWN_REQUEST';
-export const ACTION_DEQUEUE_ASYNC_REQUEST = 'DEQUEUE_ASYNC_COUNTDOWN_REQUEST';

package/src/store/wordpress-com/reducer.ts

@@ -1,224 +0,0 @@
-/**
- * Types & Constants
- */
-import {
-	ACTION_DECREASE_NEW_ASYNC_REQUEST_COUNTDOWN,
-	ACTION_ENQUEUE_ASYNC_REQUEST,
-	ACTION_INCREASE_AI_ASSISTANT_REQUESTS_COUNT,
-	ACTION_REQUEST_AI_ASSISTANT_FEATURE,
-	ACTION_SET_PLANS,
-	ACTION_SET_AI_ASSISTANT_FEATURE_REQUIRE_UPGRADE,
-	ACTION_STORE_AI_ASSISTANT_FEATURE,
-	ASYNC_REQUEST_COUNTDOWN_INIT_VALUE,
-	FREE_PLAN_REQUESTS_LIMIT,
-	UNLIMITED_PLAN_REQUESTS_LIMIT,
-	ACTION_SET_TIER_PLANS_ENABLED,
-} from './constants.ts';
-import type { PlanStateProps, TierLimitProp } from './types.ts';
-
-const INITIAL_STATE: PlanStateProps = {
-	plans: [],
-	features: {
-		aiAssistant: {
-			hasFeature: true,
-			isOverLimit: false,
-			requestsCount: 0,
-			requestsLimit: FREE_PLAN_REQUESTS_LIMIT,
-			requireUpgrade: false,
-			errorMessage: '',
-			errorCode: '',
-			upgradeType: 'default',
-			currentTier: {
-				slug: 'ai-assistant-tier-free',
-				value: 0,
-				limit: 20,
-			},
-			usagePeriod: {
-				currentStart: '',
-				nextStart: '',
-				requestsCount: 0,
-			},
-			nextTier: null,
-			tierPlansEnabled: false,
-			_meta: {
-				isRequesting: false,
-				asyncRequestCountdown: ASYNC_REQUEST_COUNTDOWN_INIT_VALUE,
-				asyncRequestTimerId: 0,
-			},
-		},
-	},
-};
-
-/**
- * The reducer of the plan state
- * @param {PlanStateProps} state  - The plan state.
- * @param {object}         action - The action.
- * @return {PlanStateProps} - The plan state.
- */
-export default function reducer( state = INITIAL_STATE, action ) {
-	switch ( action.type ) {
-		case ACTION_SET_PLANS:
-			return {
-				...state,
-				plans: action.plans,
-			};
-
-		case ACTION_REQUEST_AI_ASSISTANT_FEATURE:
-			return {
-				...state,
-				features: {
-					...state.features,
-					aiAssistant: {
-						...state.features.aiAssistant,
-						_meta: {
-							...state.features.aiAssistant._meta,
-							isRequesting: true,
-							asyncRequestCountdown: ASYNC_REQUEST_COUNTDOWN_INIT_VALUE, // restore the countdown
-							asyncRequestTimerId: 0, // reset the timer id
-						},
-					},
-				},
-			};
-
-		case ACTION_STORE_AI_ASSISTANT_FEATURE: {
-			return {
-				...state,
-				features: {
-					...state.features,
-					aiAssistant: {
-						...action.feature,
-						_meta: {
-							...state.features.aiAssistant._meta,
-							isRequesting: false,
-						},
-					},
-				},
-			};
-		}
-
-		case ACTION_INCREASE_AI_ASSISTANT_REQUESTS_COUNT: {
-			// Usage Period data
-			const usagePeriod = state.features.aiAssistant.usagePeriod || { requestsCount: 0 };
-
-			// Increase requests counters
-			const requestsCount = state.features.aiAssistant.requestsCount + action.count;
-			usagePeriod.requestsCount += action.count;
-
-			// Current tier value
-			const currentTierValue = state.features.aiAssistant.currentTier?.value;
-
-			const isFreeTierPlan =
-				( typeof currentTierValue === 'undefined' && ! state.features.aiAssistant.hasFeature ) ||
-				currentTierValue === 0;
-
-			const isUnlimitedTierPlan =
-				( typeof currentTierValue === 'undefined' && state.features.aiAssistant.hasFeature ) ||
-				currentTierValue === 1;
-
-			// Request limit defined with the current tier limit by default.
-			let requestsLimit = state.features.aiAssistant.currentTier?.limit;
-
-			if ( isUnlimitedTierPlan ) {
-				requestsLimit = UNLIMITED_PLAN_REQUESTS_LIMIT;
-			} else if ( isFreeTierPlan ) {
-				requestsLimit = state.features.aiAssistant.requestsLimit as TierLimitProp;
-			}
-
-			const currentCount = isFreeTierPlan
-				? requestsCount // Free tier plan counts all time requests
-				: state.features.aiAssistant.usagePeriod?.requestsCount; // Unlimited tier plan counts usage period requests
-
-			/**
-			 * Compute the AI Assistant Feature data optimistically,
-			 * based on the Jetpack_AI_Helper::get_ai_assistance_feature() helper.
-			 *
-			 * @see _inc/lib/class-jetpack-ai-helper.php
-			 */
-			const isOverLimit = currentCount >= requestsLimit;
-
-			// highest tier holds a soft limit so requireUpgrade is false on that case (nextTier null means highest tier)
-			const requireUpgrade = isOverLimit && state.features.aiAssistant.nextTier !== null;
-
-			return {
-				...state,
-				features: {
-					...state.features,
-					aiAssistant: {
-						...state.features.aiAssistant,
-						isOverLimit,
-						requestsCount,
-						requireUpgrade,
-						usagePeriod: { ...usagePeriod },
-					},
-				},
-			};
-		}
-
-		case ACTION_DECREASE_NEW_ASYNC_REQUEST_COUNTDOWN: {
-			return {
-				...state,
-				features: {
-					...state.features,
-					aiAssistant: {
-						...state.features.aiAssistant,
-						_meta: {
-							...state.features.aiAssistant._meta,
-							asyncRequestCountdown: state.features.aiAssistant._meta.asyncRequestCountdown - 1,
-						},
-					},
-				},
-			};
-		}
-
-		case ACTION_ENQUEUE_ASYNC_REQUEST: {
-			return {
-				...state,
-				features: {
-					...state.features,
-					aiAssistant: {
-						...state.features.aiAssistant,
-						_meta: {
-							...state.features.aiAssistant._meta,
-							asyncRequestTimerId: action.timerId,
-						},
-					},
-				},
-			};
-		}
-
-		case ACTION_SET_AI_ASSISTANT_FEATURE_REQUIRE_UPGRADE: {
-			/*
-			 * If we require an upgrade, we are also over the limit;
-			 * The opposite is not true, we can be over the limit without
-			 * requiring an upgrade, for example when we are on the highest tier.
-			 * In this case, we don't want to set isOverLimit to false.
-			 */
-			return {
-				...state,
-				features: {
-					...state.features,
-					aiAssistant: {
-						...state.features.aiAssistant,
-						requireUpgrade: action.requireUpgrade,
-						...( action.requireUpgrade ? { isOverLimit: true } : {} ),
-					},
-				},
-			};
-		}
-
-		case ACTION_SET_TIER_PLANS_ENABLED: {
-			return {
-				...state,
-				features: {
-					...state.features,
-					aiAssistant: {
-						...state.features.aiAssistant,
-						tierPlansEnabled: action.tierPlansEnabled,
-					},
-				},
-			};
-		}
-	}
-
-	return state;
-}