Extract pending migration status hook

Automattic · Oct 21, 2024 · 7023a65 · 7023a65
1 parent 5b7c3ec
commit 7023a65
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 42 deletions.
diff --git a/...epper/declarative-flow/internals/steps-repository/site-migration-how-to-migrate/index.tsx b/...epper/declarative-flow/internals/steps-repository/site-migration-how-to-migrate/index.tsx
@@ -1,60 +1,21 @@
 import { StepContainer } from '@automattic/onboarding';
 import { useTranslate } from 'i18n-calypso';
-import { FC, useMemo, useEffect } from 'react';
+import { FC, useMemo } from 'react';
 import DocumentHead from 'calypso/components/data/document-head';
 import FormattedHeader from 'calypso/components/formatted-header';
-import { useUpdateMigrationStatus } from 'calypso/data/site-migration/use-update-migration-status';
 import { useAnalyzeUrlQuery } from 'calypso/data/site-profiler/use-analyze-url-query';
 import { useHostingProviderQuery } from 'calypso/data/site-profiler/use-hosting-provider-query';
 import { HOW_TO_MIGRATE_OPTIONS } from 'calypso/landing/stepper/constants';
 import { useQuery } from 'calypso/landing/stepper/hooks/use-query';
-import { useSite } from 'calypso/landing/stepper/hooks/use-site';
 import { recordTracksEvent } from 'calypso/lib/analytics/tracks';
 import { usePresalesChat } from 'calypso/lib/presales-chat';
 import useHostingProviderName from 'calypso/site-profiler/hooks/use-hosting-provider-name';
 import FlowCard from '../components/flow-card';
-import type { NavigationControls, StepProps } from '../../types';
+import usePendingMigrationStatus from './use-pending-migration-status';
+import type { StepProps } from '../../types';
 
 import './style.scss';
 
-interface PendingMigrationStatusProps {
-	onSubmit?: Pick< NavigationControls, 'submit' >[ 'submit' ];
-}
-
-const usePendingMigrationStatus = ( { onSubmit }: PendingMigrationStatusProps ) => {
-	const site = useSite();
-	const siteId = site?.ID;
-
-	const canInstallPlugins = site?.plan?.features?.active.find(
-		( feature ) => feature === 'install-plugins'
-	)
-		? true
-		: false;
-
-	const { updateMigrationStatus } = useUpdateMigrationStatus();
-
-	// Register pending migration status when loading the step.
-	useEffect( () => {
-		if ( siteId ) {
-			updateMigrationStatus( siteId, 'migration-pending' );
-		}
-	}, [ siteId, updateMigrationStatus ] );
-
-	const onOptionClick = ( how: string ) => {
-		const destination = canInstallPlugins ? 'migrate' : 'upgrade';
-		if ( siteId ) {
-			const parsedHow = how === HOW_TO_MIGRATE_OPTIONS.DO_IT_MYSELF ? 'diy' : how;
-			updateMigrationStatus( siteId, `migration-pending-${ parsedHow }` );
-		}
-
-		if ( onSubmit ) {
-			return onSubmit( { how, destination } );
-		}
-	};
-
-	return { onOptionClick };
-};
-
 interface Props extends StepProps {
 	headerText?: string;
 	subHeaderText?: string;

diff --git a/.../internals/steps-repository/site-migration-how-to-migrate/use-pending-migration-status.ts b/.../internals/steps-repository/site-migration-how-to-migrate/use-pending-migration-status.ts
@@ -0,0 +1,45 @@
+import { useEffect } from 'react';
+import { useUpdateMigrationStatus } from 'calypso/data/site-migration/use-update-migration-status';
+import { HOW_TO_MIGRATE_OPTIONS } from 'calypso/landing/stepper/constants';
+import { useSite } from 'calypso/landing/stepper/hooks/use-site';
+import type { NavigationControls } from '../../types';
+
+interface PendingMigrationStatusProps {
+	onSubmit?: Pick< NavigationControls, 'submit' >[ 'submit' ];
+}
+
+const usePendingMigrationStatus = ( { onSubmit }: PendingMigrationStatusProps ) => {
+	const site = useSite();
+	const siteId = site?.ID;
+
+	const canInstallPlugins = site?.plan?.features?.active.find(
+		( feature ) => feature === 'install-plugins'
+	)
+		? true
+		: false;
+
+	const { updateMigrationStatus } = useUpdateMigrationStatus();
+
+	// Register pending migration status when loading the step.
+	useEffect( () => {
+		if ( siteId ) {
+			updateMigrationStatus( siteId, 'migration-pending' );
+		}
+	}, [ siteId, updateMigrationStatus ] );
+
+	const onOptionClick = ( how: string ) => {
+		const destination = canInstallPlugins ? 'migrate' : 'upgrade';
+		if ( siteId ) {
+			const parsedHow = how === HOW_TO_MIGRATE_OPTIONS.DO_IT_MYSELF ? 'diy' : how;
+			updateMigrationStatus( siteId, `migration-pending-${ parsedHow }` );
+		}
+
+		if ( onSubmit ) {
+			return onSubmit( { how, destination } );
+		}
+	};
+
+	return { onOptionClick };
+};
+
+export default usePendingMigrationStatus;