Properly tag Tron stake/unstake/claim txs

EdgeApp · Jan 10, 2024 · 4ba48e5 · 4ba48e5
1 parent 1f03775
commit 4ba48e5
Show file tree

Hide file tree

Showing 5 changed files with 45 additions and 0 deletions.
diff --git a/src/actions/CategoriesActions.ts b/src/actions/CategoriesActions.ts
@@ -410,6 +410,7 @@ export const getTxActionDisplayInfo = (tx: EdgeTransaction, account: EdgeAccount
             direction = 'send'
             break
           }
+          case 'claim':
           case 'unstake': {
             let subcategory
             if (action.stakeAssets.length === 1) subcategory = sprintf(lstrings.transaction_details_unstake_subcat_1s, ...getCurrencyCodes(action.stakeAssets))
@@ -423,6 +424,7 @@ export const getTxActionDisplayInfo = (tx: EdgeTransaction, account: EdgeAccount
             direction = 'receive'
             break
           }
+          case 'claimOrder':
           case 'unstakeOrder': {
             if (action.stakeAssets.length === 1) notes = sprintf(lstrings.transaction_details_unstake_order_notes_1s, ...getCurrencyCodes(action.stakeAssets))
             else if (action.stakeAssets.length === 2)
@@ -518,6 +520,7 @@ export const pluginIdIcons: Record<string, string> = {
   swapuz: EDGE_CONTENT_SERVER_URI + '/swapuz.png',
   thorchain: EDGE_CONTENT_SERVER_URI + '/thorchain.png',
   thorchainda: EDGE_CONTENT_SERVER_URI + '/thorchain.png',
+  tronResources: EDGE_CONTENT_SERVER_URI + '/TRON/TRON.png',
   velodrome: EDGE_CONTENT_SERVER_URI + '/velodrome.png',
   xrpdex: EDGE_CONTENT_SERVER_URI + '/xrpdex.png'
 }
diff --git a/src/constants/txActionConstants.ts b/src/constants/txActionConstants.ts
@@ -4,6 +4,8 @@ import { lstrings } from '../locales/strings'
 
 export const TX_ACTION_LABEL_MAP: Record<EdgeAssetActionType, string> = {
   buy: lstrings.fiat_plugin_buy_currencycode,
+  claim: lstrings.transaction_details_claim,
+  claimOrder: lstrings.transaction_details_claim_order,
   sell: lstrings.fiat_plugin_sell_currencycode_s,
   sellNetworkFee: lstrings.fiat_plugin_sell_network_fee,
   swap: lstrings.transaction_details_swap,

diff --git a/src/locales/en_US.ts b/src/locales/en_US.ts
@@ -696,6 +696,8 @@ const strings = {
   transaction_details_swap_order_cancel: 'Swap Order Canceled',
   transaction_details_swap_order_post: 'Swap Order Opened',
   transaction_details_swap_order_fill: 'Swap Order Filled',
+  transaction_details_claim: 'Claim Staked Funds',
+  transaction_details_claim_order: 'Claim Order',
   transaction_details_stake: 'Stake Funds',
   transaction_details_stake_order: 'Stake Order',
   transaction_details_stake_network_fee: 'Stake Network Fee',

diff --git a/src/locales/strings/enUS.json b/src/locales/strings/enUS.json
@@ -627,6 +627,8 @@
   "transaction_details_swap_order_cancel": "Swap Order Canceled",
   "transaction_details_swap_order_post": "Swap Order Opened",
   "transaction_details_swap_order_fill": "Swap Order Filled",
+  "transaction_details_claim": "Claim Staked Funds",
+  "transaction_details_claim_order": "Claim Order",
   "transaction_details_stake": "Stake Funds",
   "transaction_details_stake_order": "Stake Order",
   "transaction_details_stake_network_fee": "Stake Network Fee",

diff --git a/src/plugins/stake-plugins/currency/tronStakePlugin.ts b/src/plugins/stake-plugins/currency/tronStakePlugin.ts
@@ -179,6 +179,18 @@ export const makeTronStakePlugin = async (): Promise<StakePlugin> => {
             otherParams: {
               type: 'addV2',
               params: { nativeAmount, resource }
+            },
+            assetAction: { assetActionType: 'stake' },
+            savedAction: {
+              actionType: 'stake',
+              pluginId: stakeProviderInfo.pluginId,
+              stakeAssets: [
+                {
+                  pluginId,
+                  tokenId: null,
+                  nativeAmount
+                }
+              ]
             }
           }
           edgeTransaction = await wallet.makeSpend(spendInfo)
@@ -191,6 +203,18 @@ export const makeTronStakePlugin = async (): Promise<StakePlugin> => {
             otherParams: {
               type: 'removeV2',
               params: { nativeAmount, resource }
+            },
+            assetAction: { assetActionType: 'unstake' },
+            savedAction: {
+              actionType: 'stake',
+              pluginId: stakeProviderInfo.pluginId,
+              stakeAssets: [
+                {
+                  pluginId,
+                  tokenId: null,
+                  nativeAmount
+                }
+              ]
             }
           }
           edgeTransaction = await wallet.makeSpend(spendInfo)
@@ -202,6 +226,18 @@ export const makeTronStakePlugin = async (): Promise<StakePlugin> => {
             spendTargets,
             otherParams: {
               type: 'withdrawExpireUnfreeze'
+            },
+            assetAction: { assetActionType: 'claim' },
+            savedAction: {
+              actionType: 'stake',
+              pluginId: stakeProviderInfo.pluginId,
+              stakeAssets: [
+                {
+                  pluginId,
+                  tokenId: null,
+                  nativeAmount
+                }
+              ]
             }
           }
           edgeTransaction = await wallet.makeSpend(spendInfo)