@automattic/data-stores 2.0.1 → 3.0.0

package/src/site/reducer.ts

@@ -1,12 +1,4 @@
-/**
- * External dependencies
- */
-import type { Reducer } from 'redux';
 import { combineReducers } from '@wordpress/data';
-
-/**
- * Internal dependencies
- */
 import {
 	NewSiteBlogDetails,
 	NewSiteErrorResponse,
@@ -14,8 +6,19 @@ import {
 	Domain,
 	SiteLaunchState,
 	SiteLaunchStatus,
+	SiteSettings,
+	AtomicTransferStatus,
+	LatestAtomicTransferStatus,
 } from './types';
+import {
+	AtomicTransferState,
+	LatestAtomicTransferState,
+	AtomicSoftwareStatusState,
+	AtomicSoftwareInstallState,
+	AtomicSoftwareInstallStatus,
+} from '.';
 import type { Action } from './actions';
+import type { Reducer } from 'redux';
 
 export const newSiteData: Reducer< NewSiteBlogDetails | undefined, Action > = ( state, action ) => {
 	if ( action.type === 'RECEIVE_NEW_SITE' ) {
@@ -78,12 +81,15 @@ export const isFetchingSiteDetails: Reducer< boolean | undefined, Action > = (
 	return state;
 };
 
-export const sites: Reducer< { [ key: number ]: SiteDetails | undefined }, Action > = (
+export const sites: Reducer< { [ key: number | string ]: SiteDetails | undefined }, Action > = (
 	state = {},
 	action
 ) => {
 	if ( action.type === 'RECEIVE_SITE' ) {
-		return { ...state, [ action.siteId ]: action.response };
+		if ( action.response ) {
+			return { ...state, [ action.response.ID ]: action.response };
+		}
+		return state;
 	} else if ( action.type === 'RECEIVE_SITE_FAILED' ) {
 		const { [ action.siteId ]: idToBeRemoved, ...remainingState } = state;
 		return { ...remainingState };
@@ -92,7 +98,26 @@ export const sites: Reducer< { [ key: number ]: SiteDetails | undefined }, Actio
 	} else if ( action.type === 'RECEIVE_SITE_TITLE' ) {
 		return {
 			...state,
-			[ action.siteId ]: { ...( state[ action.siteId ] as SiteDetails ), name: action.title },
+			[ action.siteId ]: { ...( state[ action.siteId ] as SiteDetails ), name: action.name },
+		};
+	} else if ( action.type === 'RECEIVE_SITE_TAGLINE' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				...( state[ action.siteId ] as SiteDetails ),
+				description: action.tagline ?? '',
+			},
+		};
+	} else if ( action.type === 'RECEIVE_SITE_VERTICAL_ID' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				...( state[ action.siteId ] as SiteDetails ),
+				options: {
+					...state[ action.siteId ]?.options,
+					site_vertical_id: action.verticalId,
+				},
+			},
 		};
 	}
 	return state;
@@ -108,6 +133,25 @@ export const sitesDomains: Reducer< { [ key: number ]: Domain[] }, Action > = (
 	return state;
 };
 
+export const sitesSettings: Reducer< { [ key: number ]: SiteSettings }, Action > = (
+	state = {},
+	action
+) => {
+	if ( action.type === 'RECEIVE_SITE_SETTINGS' ) {
+		return { ...state, [ action.siteId ]: action.settings };
+	}
+	if ( action.type === 'UPDATE_SITE_SETTINGS' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				...state?.[ action.siteId ],
+				...action.settings,
+			},
+		};
+	}
+	return state;
+};
+
 export const launchStatus: Reducer< { [ key: number ]: SiteLaunchState }, Action > = (
 	state = {},
 	action
@@ -136,6 +180,187 @@ export const launchStatus: Reducer< { [ key: number ]: SiteLaunchState }, Action
 	return state;
 };
 
+export const siteSetupErrors: Reducer<
+	{
+		error?: string;
+		message?: string;
+	},
+	{
+		type: string;
+		error?: string;
+		message?: string;
+	}
+> = ( state = {}, action ) => {
+	if ( action.type === 'SET_SITE_SETUP_ERROR' ) {
+		const { error, message } = action;
+
+		return {
+			error,
+			message,
+		};
+	}
+
+	if ( action.type === 'CLEAR_SITE_SETUP_ERROR' ) {
+		return {};
+	}
+
+	return state;
+};
+
+export const atomicTransferStatus: Reducer< { [ key: number ]: AtomicTransferState }, Action > = (
+	state = {},
+	action
+) => {
+	if ( action.type === 'ATOMIC_TRANSFER_START' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				status: AtomicTransferStatus.IN_PROGRESS,
+				softwareSet: action.softwareSet,
+				errorCode: undefined,
+			},
+		};
+	}
+	if ( action.type === 'ATOMIC_TRANSFER_SUCCESS' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				status: AtomicTransferStatus.SUCCESS,
+				softwareSet: action.softwareSet,
+				errorCode: undefined,
+			},
+		};
+	}
+	if ( action.type === 'ATOMIC_TRANSFER_FAILURE' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				status: AtomicTransferStatus.FAILURE,
+				softwareSet: action.softwareSet,
+				errorCode: action.error,
+			},
+		};
+	}
+	return state;
+};
+
+export const latestAtomicTransferStatus: Reducer<
+	{ [ key: number ]: LatestAtomicTransferState },
+	Action
+> = ( state = {}, action ) => {
+	if ( action.type === 'LATEST_ATOMIC_TRANSFER_START' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				status: LatestAtomicTransferStatus.IN_PROGRESS,
+				transfer: undefined,
+				errorCode: undefined,
+			},
+		};
+	}
+	if ( action.type === 'LATEST_ATOMIC_TRANSFER_SUCCESS' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				status: LatestAtomicTransferStatus.SUCCESS,
+				transfer: action.transfer,
+				errorCode: undefined,
+			},
+		};
+	}
+	if ( action.type === 'LATEST_ATOMIC_TRANSFER_FAILURE' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				status: LatestAtomicTransferStatus.FAILURE,
+				transfer: undefined,
+				errorCode: action.error,
+			},
+		};
+	}
+	return state;
+};
+
+export const atomicSoftwareStatus: Reducer<
+	{ [ key: number ]: AtomicSoftwareStatusState },
+	Action
+> = ( state = {}, action ) => {
+	if ( action.type === 'ATOMIC_SOFTWARE_STATUS_START' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				[ action.softwareSet ]: {
+					status: undefined,
+					error: undefined,
+				},
+			},
+		};
+	}
+	if ( action.type === 'ATOMIC_SOFTWARE_STATUS_SUCCESS' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				[ action.softwareSet ]: {
+					status: action.status,
+					error: undefined,
+				},
+			},
+		};
+	}
+	if ( action.type === 'ATOMIC_SOFTWARE_STATUS_FAILURE' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				[ action.softwareSet ]: {
+					status: undefined,
+					error: action.error,
+				},
+			},
+		};
+	}
+	return state;
+};
+
+export const atomicSoftwareInstallStatus: Reducer<
+	{ [ key: number ]: AtomicSoftwareInstallState },
+	Action
+> = ( state = {}, action ) => {
+	if ( action.type === 'ATOMIC_SOFTWARE_INSTALL_START' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				[ action.softwareSet ]: {
+					status: AtomicSoftwareInstallStatus.IN_PROGRESS,
+					error: undefined,
+				},
+			},
+		};
+	}
+	if ( action.type === 'ATOMIC_SOFTWARE_INSTALL_SUCCESS' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				[ action.softwareSet ]: {
+					status: AtomicSoftwareInstallStatus.SUCCESS,
+					error: undefined,
+				},
+			},
+		};
+	}
+	if ( action.type === 'ATOMIC_SOFTWARE_INSTALL_FAILURE' ) {
+		return {
+			...state,
+			[ action.siteId ]: {
+				[ action.softwareSet ]: {
+					status: AtomicSoftwareInstallStatus.FAILURE,
+					error: action.error,
+				},
+			},
+		};
+	}
+	return state;
+};
+
 const newSite = combineReducers( {
 	data: newSiteData,
 	error: newSiteError,
@@ -148,6 +373,12 @@ const reducer = combineReducers( {
 	sites,
 	launchStatus,
 	sitesDomains,
+	sitesSettings,
+	siteSetupErrors,
+	atomicTransferStatus,
+	latestAtomicTransferStatus,
+	atomicSoftwareStatus,
+	atomicSoftwareInstallStatus,
 } );
 
 export type State = ReturnType< typeof reducer >;

package/src/site/resolvers.ts

@@ -1,13 +1,6 @@
-/**
- * External dependencies
- */
-import { dispatch } from '@wordpress/data';
-
-/**
- * Internal dependencies
- */
-import { wpcomRequest } from '../wpcom-request-controls';
-import { STORE_KEY } from './constants';
+// wpcomRequest is a temporary rename while we're working on migrating generators to thunks
+import wpcomRequest from 'wpcom-proxy-request';
+import type { SiteDetails, Domain, SiteSettings, Dispatch } from './types';
 
 /**
  * Attempt to find a site based on its id, and if not return undefined.
@@ -17,30 +10,48 @@ import { STORE_KEY } from './constants';
  *
  * @param siteId {number}	The site to look up
  */
-export function* getSite( siteId: number ) {
-	yield dispatch( STORE_KEY ).fetchSite();
-	try {
-		const existingSite = yield wpcomRequest( {
-			path: '/sites/' + encodeURIComponent( siteId ),
-			apiVersion: '1.1',
-		} );
-		yield dispatch( STORE_KEY ).receiveSite( siteId, existingSite );
-	} catch ( err ) {
-		yield dispatch( STORE_KEY ).receiveSiteFailed( siteId, undefined );
-	}
-}
+export const getSite =
+	( siteId: number ) =>
+	async ( { dispatch }: Dispatch ) => {
+		dispatch.fetchSite();
+		try {
+			const existingSite: SiteDetails | undefined = await wpcomRequest( {
+				path: '/sites/' + encodeURIComponent( siteId ),
+				apiVersion: '1.1',
+			} );
+			dispatch.receiveSite( siteId, existingSite );
+		} catch ( err ) {
+			dispatch.receiveSiteFailed( siteId, undefined );
+		}
+	};
 
 /**
  * Get all site domains
  *
  * @param siteId {number} The site id
  */
-export function* getSiteDomains( siteId: number ) {
-	try {
-		const result = yield wpcomRequest( {
+export const getSiteDomains =
+	( siteId: number ) =>
+	async ( { dispatch }: Dispatch ) => {
+		const result: { domains: Domain[] } = await wpcomRequest( {
 			path: '/sites/' + encodeURIComponent( siteId ) + '/domains',
 			apiVersion: '1.2',
 		} );
-		yield dispatch( STORE_KEY ).receiveSiteDomains( siteId, result?.domains );
-	} catch ( e ) {}
-}
+		dispatch.receiveSiteDomains( siteId, result?.domains );
+	};
+
+/**
+ * Get all site settings
+ *
+ * @param siteId {number} The site id
+ */
+export const getSiteSettings =
+	( siteId: number ) =>
+	async ( { dispatch }: Dispatch ) => {
+		const result: { settings: SiteSettings } = await wpcomRequest( {
+			path: '/sites/' + encodeURIComponent( siteId ) + '/settings',
+			apiVersion: '1.4',
+		} );
+
+		dispatch.receiveSiteSettings( siteId, result?.settings );
+	};

package/src/site/selectors.ts

@@ -1,17 +1,9 @@
-/**
- * External dependencies
- */
 import { select } from '@wordpress/data';
-
-/**
- * Internal dependencies
- */
-import type { State } from './reducer';
 import { STORE_KEY } from './constants';
-import { SiteLaunchStatus } from './types';
+import { SiteLaunchStatus, SiteOption } from './types';
+import type { State } from './reducer';
 
 export const getState = ( state: State ) => state;
-
 export const getNewSite = ( state: State ) => state.newSite.data;
 export const getNewSiteError = ( state: State ) => state.newSite.error;
 export const isFetchingSite = ( state: State ) => state.newSite.isFetching;
@@ -26,13 +18,23 @@ export const isNewSite = ( state: State ) => !! state.newSite.data;
  * @param state {State}		state object
  * @param siteId {number}	id of the site to look up
  */
-export const getSite = ( state: State, siteId: number ) => {
-	return state.sites[ siteId ];
+export const getSite = ( state: State, siteId: number | string ) => {
+	return (
+		state.sites[ siteId ] ||
+		Object.values( state.sites ).find( ( site ) => site && new URL( site.URL ).host === siteId )
+	);
+};
+
+export const getSiteIdBySlug = ( _: State, slug: string ) => {
+	return select( STORE_KEY ).getSite( slug )?.ID;
 };
 
 export const getSiteTitle = ( _: State, siteId: number ) =>
 	select( STORE_KEY ).getSite( siteId )?.name;
 
+export const getSiteVerticalId = ( _: State, siteId: number ) =>
+	select( STORE_KEY ).getSite( siteId )?.options?.site_vertical_id;
+
 // @TODO: Return LaunchStatus instead of a boolean
 export const isSiteLaunched = ( state: State, siteId: number ) => {
 	return state.launchStatus[ siteId ]?.status === SiteLaunchStatus.SUCCESS;
@@ -43,10 +45,34 @@ export const isSiteLaunching = ( state: State, siteId: number ) => {
 	return state.launchStatus[ siteId ]?.status === SiteLaunchStatus.IN_PROGRESS;
 };
 
+export const isSiteAtomic = ( state: State, siteId: number | string ) => {
+	return select( STORE_KEY ).getSite( siteId )?.options.is_wpcom_atomic === true;
+};
+
+export const isSiteWPForTeams = ( state: State, siteId: number | string ) => {
+	return select( STORE_KEY ).getSite( siteId )?.options.is_wpforteams_site === true;
+};
+
 export const getSiteDomains = ( state: State, siteId: number ) => {
 	return state.sitesDomains[ siteId ];
 };
 
+export const getSiteSettings = ( state: State, siteId: number ) => {
+	return state.sitesSettings[ siteId ];
+};
+
+export const getSiteSetupError = ( state: State ) => {
+	return state.siteSetupErrors;
+};
+
+export const getSiteOptions = ( state: State, siteId: number ) => {
+	return state.sites[ siteId ]?.options;
+};
+
+export const getSiteOption = ( state: State, siteId: number, optionName: SiteOption ) => {
+	return state.sites[ siteId ]?.options?.[ optionName ];
+};
+
 export const getPrimarySiteDomain = ( _: State, siteId: number ) =>
 	select( STORE_KEY )
 		.getSiteDomains( siteId )
@@ -56,3 +82,60 @@ export const getSiteSubdomain = ( _: State, siteId: number ) =>
 	select( STORE_KEY )
 		.getSiteDomains( siteId )
 		?.find( ( domain ) => domain.is_subdomain );
+
+export const getSiteLatestAtomicTransfer = ( state: State, siteId: number ) => {
+	return state.latestAtomicTransferStatus[ siteId ]?.transfer;
+};
+
+export const getSiteLatestAtomicTransferError = ( state: State, siteId: number ) => {
+	return state.latestAtomicTransferStatus[ siteId ]?.errorCode;
+};
+
+export const getAtomicSoftwareStatus = ( state: State, siteId: number, softwareSet: string ) => {
+	return state.atomicSoftwareStatus[ siteId ]?.[ softwareSet ]?.status;
+};
+
+export const getAtomicSoftwareError = ( state: State, siteId: number, softwareSet: string ) => {
+	return state.atomicSoftwareStatus[ siteId ]?.[ softwareSet ]?.error;
+};
+
+export const getAtomicSoftwareInstallError = (
+	state: State,
+	siteId: number,
+	softwareSet: string
+) => {
+	return state.atomicSoftwareInstallStatus[ siteId ]?.[ softwareSet ]?.error;
+};
+
+export const siteHasFeature = (
+	_: State,
+	siteId: number | undefined,
+	featureKey: string
+): boolean => {
+	return Boolean(
+		siteId && select( STORE_KEY ).getSite( siteId )?.plan?.features.active.includes( featureKey )
+	);
+};
+
+export const requiresUpgrade = ( state: State, siteId: number | null ) => {
+	return siteId && ! select( STORE_KEY ).siteHasFeature( siteId, 'woop' );
+};
+
+export function isJetpackSite( state: State, siteId?: number ): boolean {
+	return Boolean( siteId && select( STORE_KEY ).getSite( siteId )?.jetpack );
+}
+
+export function isEligibleForProPlan( state: State, siteId?: number ): boolean {
+	if ( ! siteId ) {
+		return false;
+	}
+
+	if (
+		( isJetpackSite( state, siteId ) && ! isSiteAtomic( state, siteId ) ) ||
+		isSiteWPForTeams( state, siteId )
+	) {
+		return false;
+	}
+
+	return true;
+}