feat: add timelock stake and unstake and use transaciton instead of send

iotaledger · Dec 27, 2024 · d97e7ca · d97e7ca
1 parent aa6ff76
commit d97e7ca
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 3 deletions.
diff --git a/apps/core/src/interfaces/transactions.interfaces.ts b/apps/core/src/interfaces/transactions.interfaces.ts
@@ -16,6 +16,8 @@ export enum TransactionAction {
     Transaction = 'Transaction',
     Staked = 'Staked',
     Unstaked = 'Unstaked',
+    TimelockedStaked = 'TimelockedStaked',
+    TimelockedUnstaked = 'TimelockedUnstaked',
     Rewards = 'Rewards',
     PersonalMessage = 'PersonalMessage',
 }

diff --git a/apps/core/src/utils/transaction/getTransactionAction.ts b/apps/core/src/utils/transaction/getTransactionAction.ts
@@ -16,12 +16,19 @@ export const getTransactionAction = (
     const unstakeTypeTransaction = transaction?.events?.find(
         ({ type }) => type === UNSTAKING_REQUEST_EVENT,
     );
+
+    const isTimelockedStaking = stakeTypeTransaction?.transactionModule === 'timelocked_staking';
+    const isTimelockedUnstaking =
+        unstakeTypeTransaction?.transactionModule === 'timelocked_staking';
+
     if (stakeTypeTransaction) {
-        return TransactionAction.Staked;
+        return isTimelockedStaking ? TransactionAction.TimelockedStaked : TransactionAction.Staked;
     } else if (unstakeTypeTransaction) {
-        return TransactionAction.Unstaked;
+        return isTimelockedUnstaking
+            ? TransactionAction.TimelockedUnstaked
+            : TransactionAction.Unstaked;
     } else {
         const isSender = transaction.transaction?.data.sender === currentAddress;
-        return isSender ? TransactionAction.Send : TransactionAction.Receive;
+        return isSender ? TransactionAction.Transaction : TransactionAction.Receive;
     }
 };
diff --git a/apps/wallet-dashboard/components/transactions/TransactionIcon.tsx b/apps/wallet-dashboard/components/transactions/TransactionIcon.tsx
@@ -19,6 +19,8 @@ const icons = {
     Failed: <Info className={ICON_COLORS.error} />,
     Loading: <LoadingIndicator />,
     PersonalMessage: <Person className={ICON_COLORS.primary} />,
+    TimelockedStaked: <Stake className={ICON_COLORS.primary} />,
+    TimelockedUnstaked: <Stake className={ICON_COLORS.primary} />,
 };
 
 interface TransactionIconProps {