Adding unfreeze and withdraw for tron unstaking

modules/sdk-coin-trx/src/lib/enum.ts

@@ -22,6 +22,14 @@ export enum ContractType {
    * This is the contract for voting for witnesses
    */
   VoteWitness,
+  /**
+   * This is the contract for unfreezing balances
+   */
+  UnfreezeBalanceV2,
+  /**
+   * This is the contract for withdrawing expired unfrozen balances
+   */
+  WithdrawExpireUnfreeze,
 }
 
 export enum PermissionType {

modules/sdk-coin-trx/src/lib/iface.ts

@@ -47,7 +47,9 @@ export interface RawData {
     | AccountPermissionUpdateContract[]
     | TriggerSmartContract[]
     | FreezeBalanceV2Contract[]
-    | VoteWitnessContract[];
+    | VoteWitnessContract[]
+    | UnfreezeBalanceV2Contract[]
+    | WithdrawExpireUnfreezeContract[];
 }
 
 export interface Value {
@@ -132,6 +134,15 @@ export interface FreezeBalanceValueFields {
   owner_address: string;
 }
 
+/**
+ * Unfreeze transaction value fields
+ */
+export interface UnfreezeBalanceValueFields {
+  resource: string;
+  unfreeze_balance: number;
+  owner_address: string;
+}
+
 /**
  * Freeze balance contract value interface
  */
@@ -148,6 +159,22 @@ export interface FreezeBalanceV2Contract {
   type?: string;
 }
 
+/**
+ * Unfreeze balance contract value interface
+ */
+export interface UnfreezeBalanceValue {
+  type_url?: string;
+  value: UnfreezeBalanceValueFields;
+}
+
+/**
+ * Unfreeze balance v2 contract interface
+ */
+export interface UnfreezeBalanceV2Contract {
+  parameter: UnfreezeBalanceValue;
+  type?: string;
+}
+
 /**
  * Freeze balance contract parameter interface
  */
@@ -161,6 +188,39 @@ export interface FreezeBalanceContractParameter {
   };
 }
 
+/**
+ * Withdraw transaction value fields
+ */
+export interface WithdrawExpireUnfreezeValueFields {
+  owner_address: string;
+}
+
+/**
+ * Withdraw balance contract value interface
+ */
+export interface WithdrawExpireUnfreezeValue {
+  type_url?: string;
+  value: WithdrawExpireUnfreezeValueFields;
+}
+
+/**
+ * Withdraw expire unfreeze contract interface
+ */
+export interface WithdrawExpireUnfreezeContract {
+  parameter: WithdrawExpireUnfreezeValue;
+  type?: string;
+}
+
+export interface UnfreezeBalanceContractParameter {
+  parameter: {
+    value: {
+      resource: TronResource;
+      unfreeze_balance: number;
+      owner_address: string;
+    };
+  };
+}
+
 /**
  * Freeze balance contract decoded interface
  */
@@ -170,6 +230,22 @@ export interface FreezeContractDecoded {
   frozenBalance?: string | number;
 }
 
+/**
+ * Unfreeze balance contract decoded interface
+ */
+export interface UnfreezeContractDecoded {
+  ownerAddress?: string;
+  resource?: number;
+  unfreezeBalance?: string | number;
+}
+
+/**
+ * Withdraw expire unfreeze contract decoded interface
+ */
+export interface WithdrawContractDecoded {
+  ownerAddress?: string;
+}
+
 /**
  * Vote data in a vote transaction
  */
@@ -227,3 +303,14 @@ export interface VoteWitnessContractParameter {
     };
   };
 }
+
+/**
+ * Withdraw expire unfreeze contract parameter interface
+ */
+export interface WithdrawExpireUnfreezeContractParameter {
+  parameter: {
+    value: {
+      owner_address: string;
+    };
+  };
+}

modules/sdk-coin-trx/src/lib/transaction.ts

@@ -25,6 +25,8 @@ import {
   TransferContract,
   TriggerSmartContract,
   VoteWitnessContract,
+  UnfreezeBalanceV2Contract,
+  WithdrawExpireUnfreezeContract,
 } from './iface';
 
 /**
@@ -162,6 +164,30 @@ export class Transaction extends BaseTransaction {
           value: totalVoteCount.toString(),
         };
         break;
+      case ContractType.UnfreezeBalanceV2:
+        this._type = TransactionType.StakingDeactivate;
+        const unfreezeValues = (rawData.contract[0] as UnfreezeBalanceV2Contract).parameter.value;
+        output = {
+          address: unfreezeValues.owner_address,
+          value: unfreezeValues.unfreeze_balance.toString(),
+        };
+        input = {
+          address: unfreezeValues.owner_address,
+          value: unfreezeValues.unfreeze_balance.toString(),
+        };
+        break;
+      case ContractType.WithdrawExpireUnfreeze:
+        this._type = TransactionType.StakingWithdraw;
+        const withdrawValues = (rawData.contract[0] as WithdrawExpireUnfreezeContract).parameter.value;
+        output = {
+          address: withdrawValues.owner_address,
+          value: '0', // no value field
+        };
+        input = {
+          address: withdrawValues.owner_address,
+          value: '0',
+        };
+        break;
       default:
         throw new ParseTransactionError('Unsupported contract type');
     }