fix: retry button on transfaction failure should check old firmware e…

…rror for ledger (#961) * fix: retry button on transfaction failure should check old firmware error for ledger * fix: enable single pool delegation on tx failure re-try button click * fix: remove eslint warning --------- Co-authored-by: Tomislav Horaček <[email protected]> Co-authored-by: Rhys Bartels-Waller <[email protected]> Co-authored-by: Piotr Czeglik <[email protected]>
input-output-hk · May 1, 2024 · 00f0c52 · 00f0c52
1 parent d18829e
commit 00f0c52
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 5 deletions.
diff --git a/packages/staking/src/features/Drawer/TransactionFail.tsx b/packages/staking/src/features/Drawer/TransactionFail.tsx
@@ -1,4 +1,5 @@
 /* eslint-disable react/no-multi-comp */
+import { WalletType } from '@cardano-sdk/web-extension';
 import { Button, WarningBanner } from '@lace/common';
 import { Box } from '@lace/ui';
 import cn from 'classnames';
@@ -45,17 +46,20 @@ export const TransactionFailFooter = ({ popupView }: TransactionFailProps): Reac
   const [isLoading, setIsLoading] = useState<boolean>(false);
   const {
     delegationStoreSetDelegationTxBuilder: setDelegationTxBuilder,
+    walletStoreWalletType: walletType,
     delegationStoreDelegationTxBuilder: delegationTxBuilder,
     password: { password, removePassword },
     walletStoreInMemoryWallet: inMemoryWallet,
     walletManagerExecuteWithPassword: executeWithPassword,
+    isMultidelegationSupportedByDevice,
   } = useOutsideHandles();
   // TODO implement analytics for the new flow
   const analytics = {
     // eslint-disable-next-line @typescript-eslint/no-empty-function
     sendEvent: () => {},
   };
-  const { portfolioMutators } = useDelegationPortfolioStore((store) => ({
+  const { portfolioMutators, draftPortfolio } = useDelegationPortfolioStore((store) => ({
+    draftPortfolio: store.draftPortfolio,
     portfolioMutators: store.mutators,
   }));
 
@@ -72,23 +76,38 @@ export const TransactionFailFooter = ({ popupView }: TransactionFailProps): Reac
     portfolioMutators.executeCommand({ type: 'CancelDrawer' });
   };
 
+  const isInMemory = walletType === WalletType.InMemory;
+
   // TODO unify
   const signAndSubmitTransaction = useCallback(async () => {
     if (!delegationTxBuilder) throw new Error('Unable to submit transaction. The delegationTxBuilder not available');
+
+    const isMultidelegation = draftPortfolio && draftPortfolio.length > 1;
+    if (!isInMemory && isMultidelegation) {
+      const isSupported = await isMultidelegationSupportedByDevice(walletType);
+      if (!isSupported) {
+        throw new Error('MULTIDELEGATION_NOT_SUPPORTED');
+      }
+    }
     const signedTx = await delegationTxBuilder.build().sign();
     await inMemoryWallet.submitTx(signedTx);
-  }, [delegationTxBuilder, inMemoryWallet]);
+  }, [delegationTxBuilder, draftPortfolio, inMemoryWallet, isInMemory, isMultidelegationSupportedByDevice, walletType]);
 
   const onSubmit = async () => {
     setIsLoading(true);
+
     try {
       await signAndSubmitTransaction();
       setIsLoading(false);
       portfolioMutators.executeCommand({ type: 'DrawerContinue' });
       removePassword();
-    } catch (error) {
+    } catch (error: unknown) {
       console.error('failed to sign or submit tx due to:', error);
       setIsLoading(false);
+
+      if (error instanceof Error && error.message === 'MULTIDELEGATION_NOT_SUPPORTED') {
+        portfolioMutators.executeCommand({ type: 'HwSkipToDeviceFailure' });
+      }
     }
   };
 

diff --git a/packages/staking/src/features/store/delegationPortfolioStore/stateMachine/commands.ts b/packages/staking/src/features/store/delegationPortfolioStore/stateMachine/commands.ts
@@ -181,7 +181,7 @@ export type PortfolioManagementConfirmationCommand =
 
 export type PortfolioManagementSignCommand = CancelDrawer | DrawerContinue | DrawerFailure | DrawerBack;
 
-export type PortfolioManagementFailureCommand = CancelDrawer | DrawerContinue;
+export type PortfolioManagementFailureCommand = CancelDrawer | DrawerContinue | HwSkipToDeviceFailure;
 
 export type PortfolioManagementHwFailureCommand = CancelDrawer | DrawerBack;
 
@@ -206,7 +206,7 @@ export type NewPortfolioConfirmationCommand =
 
 export type NewPortfolioSignCommand = CancelDrawer | DrawerContinue | DrawerFailure | DrawerBack;
 
-export type NewPortfolioFailureCommand = CancelDrawer | DrawerContinue;
+export type NewPortfolioFailureCommand = CancelDrawer | DrawerContinue | HwSkipToDeviceFailure;
 
 export type NewPortfolioHwFailureCommand = CancelDrawer | DrawerBack;
 

diff --git a/...king/src/features/store/delegationPortfolioStore/stateMachine/processExpandedViewCases.ts b/...king/src/features/store/delegationPortfolioStore/stateMachine/processExpandedViewCases.ts
@@ -425,6 +425,12 @@ export const processExpandedViewCases: Handler = (params) =>
                   activeDrawerStep: DrawerManagementStep.Success,
                 })
               ),
+              HwSkipToDeviceFailure: handler<HwSkipToDeviceFailure, StatePortfolioManagement, StatePortfolioManagement>(
+                ({ state }) => ({
+                  ...state,
+                  activeDrawerStep: DrawerManagementStep.HwDeviceFailure,
+                })
+              ),
             },
             params.command.type,
             DrawerManagementStep.Failure
@@ -593,6 +599,12 @@ export const processExpandedViewCases: Handler = (params) =>
                 ...state,
                 activeDrawerStep: DrawerManagementStep.Success,
               })),
+              HwSkipToDeviceFailure: handler<HwSkipToDeviceFailure, StatePortfolioManagement, StatePortfolioManagement>(
+                ({ state }) => ({
+                  ...state,
+                  activeDrawerStep: DrawerManagementStep.HwDeviceFailure,
+                })
+              ),
             },
             params.command.type,
             DrawerManagementStep.Failure