From 4c4a9cbb739673a21ca4ece5e918e5f59f168cf8 Mon Sep 17 00:00:00 2001
From: microwavedcola1 <microwavedcola@gmail.com>
Date: Tue, 23 Aug 2022 16:21:05 +0200
Subject: [PATCH] Small changes for consistency

Signed-off-by: microwavedcola1 <microwavedcola@gmail.com>
---
 ts/client/src/accounts/mangoAccount.ts | 38 ++++++++++++++++++++++----
 1 file changed, 32 insertions(+), 6 deletions(-)

diff --git a/ts/client/src/accounts/mangoAccount.ts b/ts/client/src/accounts/mangoAccount.ts
index 02dab7c8c..f9a4d9dee 100644
--- a/ts/client/src/accounts/mangoAccount.ts
+++ b/ts/client/src/accounts/mangoAccount.ts
@@ -321,6 +321,30 @@ export class MangoAccount {
       .mul(I80F48.fromNumber(slippageAndFeesFactor));
   }
 
+  /**
+   * The max amount of given source ui token you can swap to a target token.
+   * note: slippageAndFeesFactor is a normalized number, <1,
+   *  e.g. a slippage of 5% and some fees which are 1%, then slippageAndFeesFactor = 0.94
+   *  the factor is used to compute how much target can be obtained by swapping source
+   * @returns max amount of given source ui token you can swap to a target token, in ui token
+   */
+  getMaxSourceUiForTokenSwap(
+    group: Group,
+    sourceMintPk: PublicKey,
+    targetMintPk: PublicKey,
+    slippageAndFeesFactor: number,
+  ): number {
+    return toUiDecimals(
+      this.getMaxSourceForTokenSwap(
+        group,
+        sourceMintPk,
+        targetMintPk,
+        slippageAndFeesFactor,
+      ),
+      group.getMintDecimals(sourceMintPk),
+    );
+  }
+
   /**
    * Simulates new health ratio after applying tokenChanges to the token positions.
    * Note: token changes are expected in native amounts
@@ -359,7 +383,7 @@ export class MangoAccount {
       mintPk: PublicKey;
     }[],
     healthType: HealthType = HealthType.init,
-  ): I80F48 {
+  ): number {
     const nativeTokenChanges = uiTokenChanges.map((tokenChange) => {
       return {
         nativeTokenAmount: toNative(
@@ -369,11 +393,13 @@ export class MangoAccount {
         mintPk: tokenChange.mintPk,
       };
     });
-    return this.accountData.healthCache.simHealthRatioWithTokenPositionChanges(
-      group,
-      nativeTokenChanges,
-      healthType,
-    );
+    return this.accountData.healthCache
+      .simHealthRatioWithTokenPositionChanges(
+        group,
+        nativeTokenChanges,
+        healthType,
+      )
+      .toNumber();
   }
 
   /**