feat: add staked and unstaked action to activity details

iotaledger · Dec 24, 2024 · 0a7587f · 0a7587f
1 parent 434807e
commit 0a7587f
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/apps/core/src/utils/transaction/getTransactionAction.ts b/apps/core/src/utils/transaction/getTransactionAction.ts
@@ -4,11 +4,24 @@
 
 import { IotaTransactionBlockResponse } from '@iota/iota-sdk/client';
 import { TransactionAction } from '../../interfaces';
+import { STAKING_REQUEST_EVENT, UNSTAKING_REQUEST_EVENT } from '../../constants';
 
 export const getTransactionAction = (
     transaction: IotaTransactionBlockResponse,
     currentAddress?: string,
 ) => {
-    const isSender = transaction.transaction?.data.sender === currentAddress;
-    return isSender ? TransactionAction.Send : TransactionAction.Receive;
+    const stakeTypeTransaction = transaction?.events?.find(
+        ({ type }) => type === STAKING_REQUEST_EVENT,
+    );
+    const unstakeTypeTransaction = transaction?.events?.find(
+        ({ type }) => type === UNSTAKING_REQUEST_EVENT,
+    );
+    if (stakeTypeTransaction) {
+        return TransactionAction.Staked;
+    } else if (unstakeTypeTransaction) {
+        return TransactionAction.Unstaked;
+    } else {
+        const isSender = transaction.transaction?.data.sender === currentAddress;
+        return isSender ? TransactionAction.Send : TransactionAction.Receive;
+    }
 };