feat: reload credits or wallet after create module (#1105)

Signed-off-by: David Dal Busco <[email protected]>

src/frontend/src/lib/components/guards/CreditsGuard.svelte

@@ -14,6 +14,7 @@
 	interface Props {
 		detail: JunoModalDetail;
 		priceLabel: string;
+		withCredits: boolean;
 		insufficientFunds?: boolean;
 		children: Snippet;
 		onclose: () => void;
@@ -23,6 +24,7 @@
 		detail,
 		priceLabel,
 		insufficientFunds = $bindable(true),
+		withCredits = $bindable(false),
 		children,
 		onclose
 	}: Props = $props();
@@ -38,6 +40,10 @@
 	$effect(() => {
 		insufficientFunds = $balanceOrZero < fee && notEnoughCredits;
 	});
+
+	$effect(() => {
+		withCredits = !notEnoughCredits;
+	});
 </script>
 
 {#if notEnoughCredits}

src/frontend/src/lib/components/modals/OrbiterCreateModal.svelte

@@ -23,6 +23,7 @@
 
 	let { detail, onclose }: Props = $props();
 
+	let withCredits = $state(false);
 	let insufficientFunds = $state(true);
 
 	let step: 'init' | 'in_progress' | 'ready' | 'error' = $state('init');
@@ -46,6 +47,7 @@
 			missionControlId: $missionControlIdDerived,
 			subnetId,
 			monitoringStrategy,
+			withCredits,
 			onProgress
 		});
 
@@ -85,6 +87,7 @@
 		<CreditsGuard
 			{onclose}
 			bind:insufficientFunds
+			bind:withCredits
 			{detail}
 			priceLabel={$i18n.analytics.create_orbiter_price}
 		>

src/frontend/src/lib/components/modals/SatelliteCreateModal.svelte

@@ -28,6 +28,7 @@
 
 	let { detail, onclose }: Props = $props();
 
+	let withCredits = $state(false);
 	let insufficientFunds = $state(true);
 
 	let step: 'init' | 'in_progress' | 'ready' | 'error' = $state('init');
@@ -53,6 +54,7 @@
 			subnetId,
 			monitoringStrategy,
 			satelliteName,
+			withCredits,
 			onProgress
 		});
 
@@ -101,6 +103,7 @@
 
 		<CreditsGuard
 			{onclose}
+			bind:withCredits
 			bind:insufficientFunds
 			{detail}
 			priceLabel={$i18n.satellites.create_satellite_price}

src/frontend/src/lib/services/wizard.services.ts

@@ -31,6 +31,7 @@ import type { PrincipalText } from '$lib/types/principal';
 import { type WizardCreateProgress, WizardCreateProgressStep } from '$lib/types/progress-wizard';
 import type { Option } from '$lib/types/utils';
 import { emit } from '$lib/utils/events.utils';
+import { waitAndRestartWallet } from '$lib/utils/wallet.utils';
 import type { Identity } from '@dfinity/agent';
 import { Principal } from '@dfinity/principal';
 import { assertNonNullish, isNullish, nonNullish, toNullable } from '@dfinity/utils';
@@ -190,6 +191,7 @@ interface CreateWizardParams {
 	identity: OptionIdentity;
 	subnetId: PrincipalText | undefined;
 	monitoringStrategy: CyclesMonitoringStrategy | undefined;
+	withCredits: boolean;
 	onProgress: (progress: WizardCreateProgress | undefined) => void;
 }
 
@@ -351,7 +353,8 @@ const createWizard = async <T>({
 	createFn,
 	reloadFn,
 	monitoringFn,
-	onProgress
+	onProgress,
+	withCredits
 }: Omit<CreateWizardParams, 'subnetId' | 'monitoringStrategy'> & {
 	errorLabel: keyof I18nErrors;
 	createFn: (params: { identity: Identity }) => Promise<T>;
@@ -382,11 +385,22 @@ const createWizard = async <T>({
 		});
 
 		const reload = async () => {
-			await reloadFn({ missionControlId, reload: true });
+			await Promise.allSettled([
+				...(withCredits
+					? [
+							loadCredits({
+								identity,
+								reload: true
+							})
+						]
+					: [waitAndRestartWallet()]),
+				reloadFn({ missionControlId, reload: true })
+			]);
 		};
 
 		if (nonNullish(monitoringFn)) {
 			const executeMonitoringFn = async () => {
+				await waitAndRestartWallet();
 				await monitoringFn({ identity, segment });
 			};
 
@@ -397,7 +411,7 @@ const createWizard = async <T>({
 			});
 		}
 
-		// Reload list of segments before navigation
+		// Reload list of segments and wallet or credits before navigation
 		await execute({
 			fn: reload,
 			onProgress,