actions.ts
2 years ago
constants.ts
2 years ago
index.ts
2 years ago
reducer.ts
2 years ago
types.ts
2 years ago
index.ts
117 lines
| 1 | /** |
| 2 | * External dependencies |
| 3 | */ |
| 4 | import { createReduxStore, register, select } from '@wordpress/data'; |
| 5 | /** |
| 6 | * Internal dependencies |
| 7 | */ |
| 8 | import actions from './actions'; |
| 9 | import reducer from './reducer'; |
| 10 | /** |
| 11 | * Types |
| 12 | */ |
| 13 | import type { AiFeatureProps, PlanStateProps } from './types'; |
| 14 | |
| 15 | const store = 'wordpress-com/plans'; |
| 16 | |
| 17 | export const selectors = { |
| 18 | /* |
| 19 | * Return the plan with the given slug. |
| 20 | * |
| 21 | * @param {Object} state - The Plans state tree. |
| 22 | * @param {string} planSlug - The plan slug to find. |
| 23 | * @return {Object} The plan. |
| 24 | */ |
| 25 | getPlan( state: PlanStateProps, planSlug: string ) { |
| 26 | return state.plans.find( plan => plan.product_slug === planSlug ); |
| 27 | }, |
| 28 | |
| 29 | /** |
| 30 | * Return the AI Assistant feature. |
| 31 | * |
| 32 | * @param {PlanStateProps} state - The Plans state tree. |
| 33 | * @returns {AiFeatureProps} The AI Assistant feature data. |
| 34 | */ |
| 35 | getAiAssistantFeature( state: PlanStateProps ): AiFeatureProps { |
| 36 | // Clean up the _meta property. |
| 37 | const data = { ...state.features.aiAssistant }; |
| 38 | delete data._meta; |
| 39 | |
| 40 | return data; |
| 41 | }, |
| 42 | |
| 43 | /** |
| 44 | * Get the isRequesting flag for the AI Assistant feature. |
| 45 | * |
| 46 | * @param {PlanStateProps} state - The Plans state tree. |
| 47 | * @returns {boolean} The isRequesting flag. |
| 48 | */ |
| 49 | getIsRequestingAiAssistantFeature( state: PlanStateProps ): boolean { |
| 50 | return state.features.aiAssistant?._meta?.isRequesting; |
| 51 | }, |
| 52 | |
| 53 | getAsyncRequestCountdownValue( state: PlanStateProps ): number { |
| 54 | return state.features.aiAssistant?._meta?.asyncRequestCountdown; |
| 55 | }, |
| 56 | |
| 57 | getAsyncRequestCountdownTimerId( state: PlanStateProps ): number { |
| 58 | return state.features.aiAssistant?._meta?.asyncRequestTimerId; |
| 59 | }, |
| 60 | }; |
| 61 | |
| 62 | export const wordpressPlansStore = createReduxStore( store, { |
| 63 | actions, |
| 64 | |
| 65 | reducer, |
| 66 | |
| 67 | selectors, |
| 68 | |
| 69 | controls: { |
| 70 | FETCH_FROM_API( { url } ) { |
| 71 | // We cannot use `@wordpress/api-fetch` here since it unconditionally sends |
| 72 | // the `X-WP-Nonce` header, which is disallowed by WordPress.com. |
| 73 | // (To reproduce, note that you need to call `apiFetch` with ` |
| 74 | // `{ credentials: 'same-origin', mode: 'cors' }`, since its defaults are |
| 75 | // different from `fetch`'s.) |
| 76 | return fetch( url ).then( response => response.json() ); |
| 77 | }, |
| 78 | }, |
| 79 | |
| 80 | resolvers: { |
| 81 | *getPlan() { |
| 82 | const url = 'https://public-api.wordpress.com/rest/v1.5/plans'; |
| 83 | const plans = yield actions.fetchFromAPI( url ); |
| 84 | return actions.setPlans( plans ); |
| 85 | }, |
| 86 | |
| 87 | getAiAssistantFeature: ( state: PlanStateProps ) => { |
| 88 | if ( state?.features?.aiAssistant ) { |
| 89 | return; |
| 90 | } |
| 91 | |
| 92 | return actions.fetchAiAssistantFeature(); |
| 93 | }, |
| 94 | }, |
| 95 | } ); |
| 96 | |
| 97 | register( wordpressPlansStore ); |
| 98 | |
| 99 | /* |
| 100 | * Ensure to request the AI Assistant feature data |
| 101 | * by calling the selector. Resolver will take care. |
| 102 | */ |
| 103 | if ( window.Jetpack_Editor_Initial_State?.[ 'ai-assistant' ]?.[ 'is-enabled' ] ) { |
| 104 | select( store ).getAiAssistantFeature(); |
| 105 | } |
| 106 | |
| 107 | // Types |
| 108 | |
| 109 | // eslint-disable-next-line @typescript-eslint/no-explicit-any |
| 110 | type OmitFirstArg< F > = F extends ( _: any, ...args: infer P ) => infer R |
| 111 | ? ( ...args: P ) => R |
| 112 | : never; |
| 113 | |
| 114 | export type WordPressPlansSelectors = { |
| 115 | [ key in keyof typeof selectors ]: OmitFirstArg< ( typeof selectors )[ key ] >; |
| 116 | }; |
| 117 |