refactor(sdk-coin-hbar): add recovery for hbar tokens

TICKET: WIN-4342
BitGo · Jan 31, 2025 · 1380cc2 · 1380cc2
1 parent 725c495
commit 1380cc2
Show file tree

Hide file tree

Showing 3 changed files with 459 additions and 12 deletions.
diff --git a/modules/sdk-coin-hbar/src/hbar.ts b/modules/sdk-coin-hbar/src/hbar.ts
@@ -87,9 +87,10 @@ export interface RecoveryOptions {
   maxFee?: string;
   nodeId?: string;
   startTime?: string;
+  tokenName?: string;
 }
 
-interface RecoveryInfo {
+export interface RecoveryInfo {
   id: string;
   tx: string;
   coin: string;
@@ -391,12 +392,19 @@ export class Hbar extends BaseCoin {
 
     const spendableAmount = new BigNumber(nativeBalance).minus(fee).toString();
 
-    const txBuilder = this.getBuilderFactory().getTransferBuilder();
+    const txBuilder = params.tokenName
+      ? this.getBuilderFactory().getTokenTransferBuilder()
+      : this.getBuilderFactory().getTransferBuilder();
+
     txBuilder.node({ nodeId });
     txBuilder.fee({ fee });
     txBuilder.source({ address: params.rootAddress });
-    txBuilder.send({ address: destinationAddress, amount: spendableAmount });
     txBuilder.validDuration(180);
+    if (params.tokenName) {
+      txBuilder.send({ address: destinationAddress, amount: spendableAmount, tokenName: params.tokenName });
+    } else {
+      txBuilder.send({ address: destinationAddress, amount: spendableAmount });
+    }
 
     if (memoId) {
       txBuilder.memo(memoId);