From 8c9bf4808da961da89694fc4543964bea94a7109 Mon Sep 17 00:00:00 2001
From: Aaron Cox <aaron@greymass.com>
Date: Tue, 16 Jul 2024 14:15:52 -0700
Subject: [PATCH] REX calculation fixes

---
 src/pages/dashboard/index.svelte | 23 +++++++-----
 src/pages/earn/index.svelte      | 61 ++++++++++++++++++++++++++------
 2 files changed, 65 insertions(+), 19 deletions(-)

diff --git a/src/pages/dashboard/index.svelte b/src/pages/dashboard/index.svelte
index 15256552..e52aa531 100644
--- a/src/pages/dashboard/index.svelte
+++ b/src/pages/dashboard/index.svelte
@@ -1,5 +1,5 @@
 <script>
-    import {Asset} from 'anchor-link'
+    import {Asset, Int128} from 'anchor-link'
     import {derived} from 'svelte/store'
     import type {Readable} from 'svelte/store'
 
@@ -11,7 +11,7 @@
     import {activeSession, activeBlockchain, currentAccount, activePriceTicker} from '~/store'
     import {balances, fetchBalances} from '~/stores/balances'
     import {isLoading} from '~/stores/balances-provider'
-    import {getToken, systemTokenKey, tokens} from '~/stores/tokens'
+    import {getToken, systemToken, systemTokenKey, tokens} from '~/stores/tokens'
     import {stateREX} from '~/pages/resources/resources'
 
     import Page from '~/components/layout/page.svelte'
@@ -70,14 +70,21 @@
     )
 
     const rexTokens: Readable<number> = derived(
-        [currentAccount, stateREX],
-        ([$currentAccount, $stateREX]) => {
+        [currentAccount, stateREX, systemToken],
+        ([$currentAccount, $stateREX, $systemToken]) => {
             if ($currentAccount && $currentAccount.rex_info && $stateREX && $stateREX.value) {
                 if ($stateREX.value === 0.0001) {
-                    return (
-                        ($stateREX.total_lendable.value / $stateREX.total_rex.value) *
-                        $currentAccount.rex_info.rex_balance.value
-                    )
+                    const pool = $stateREX
+                    if (!$systemToken || !pool) {
+                        return 0
+                    }
+                    const {total_lendable, total_rex} = pool
+                    const R1 = total_rex.units.adding($currentAccount.rex_info.rex_balance.units)
+                    const S1 = Int128.from(R1)
+                        .multiplying(total_lendable.units)
+                        .dividing(total_rex.units)
+                    const result = S1.subtracting(total_lendable.units)
+                    return Asset.fromUnits(result, $systemToken!.symbol).value
                 } else {
                     return $stateREX.value * $currentAccount.rex_info.rex_balance.value
                 }
diff --git a/src/pages/earn/index.svelte b/src/pages/earn/index.svelte
index ab7ab5fc..ccdf0ac1 100644
--- a/src/pages/earn/index.svelte
+++ b/src/pages/earn/index.svelte
@@ -1,7 +1,7 @@
 <script lang="ts">
     import type {Readable, Writable} from 'svelte/store'
     import {derived, writable, get} from 'svelte/store'
-    import {AnyAction, Asset} from 'anchor-link'
+    import {AnyAction, Asset, Int128} from 'anchor-link'
 
     import {currentAccount} from '~/store'
     import {activeBlockchain, activeSession} from '~/store'
@@ -66,20 +66,59 @@
 
     const convertEosToRex = (eos: number) => {
         const pool = $stateREX
-        const S0 = pool!.total_lendable.value
-        const R0 = pool!.total_rex.value
-        const S1 = S0 + eos
-        const R1 = (S1 * R0) / S0
-        return Asset.from(R1 - R0, $currentAccount!.rex_info!.rex_balance.symbol)
+        if (!$systemToken || !pool) {
+            return Asset.from(0, $systemToken!.symbol)
+        }
+
+        const {total_lendable, total_rex} = pool
+        const asset = Asset.from(eos, $systemToken!.symbol)
+        const S1 = total_lendable.units.adding(asset.units)
+        const R1 = Int128.from(S1).multiplying(total_rex.units).dividing(total_lendable.units)
+        const result = R1.subtracting(total_rex.units)
+
+        // Debugging using floats
+        // const oS0 = parseFloat(String(pool.total_lendable).split(' ')[0])
+        // const oS1 = oS0 + eos
+        // const oR0 = parseFloat(String(pool.total_rex).split(' ')[0])
+        // const oR1 = (oS1 * oR0) / oS0
+
+        // console.table({
+        //     S0: [String(S0), String(oS0)],
+        //     R0: [String(R0), String(oR0)],
+        //     R1: [String(R1), String(oR1)],
+        //     S1: [String(S1), String((oS0 * oR1) / oR0)],
+        //     result: [String(result), String(oR1 - oR0)],
+        // })
+
+        return Asset.fromUnits(result, $currentAccount!.rex_info!.rex_balance.symbol)
     }
 
     const convertRexToEos = (rex: number) => {
         const pool = $stateREX
-        const S0 = pool!.total_lendable.value
-        const R0 = pool!.total_rex.value
-        const R1 = R0 + rex
-        const S1 = (S0 * R1) / R0
-        return Asset.from(S1 - S0, $systemToken!.symbol)
+        if (!$systemToken || !pool) {
+            return Asset.from(0, $systemToken!.symbol)
+        }
+
+        const {total_lendable, total_rex} = pool
+        const asset = Asset.from(rex, '4,REX')
+        const R1 = total_rex.units.adding(asset.units)
+        const S1 = Int128.from(R1).multiplying(total_lendable.units).dividing(total_rex.units)
+        const result = S1.subtracting(total_lendable.units)
+
+        // Debugging using floats
+        // const oS0 = parseFloat(String(pool.total_lendable).split(' ')[0])
+        // const oR0 = parseFloat(String(pool.total_rex).split(' ')[0])
+        // const oR1 = oR0 + rex
+        // const oS1 = (oS0 * oR1) / oR0
+        // console.table({
+        //     S0: [String(S0), String(oS0)],
+        //     R0: [String(R0), String(oR0)],
+        //     R1: [String(R1), String(oR1)],
+        //     S1: [String(S1), String((oS0 * oR1) / oR0)],
+        //     result: [String(result), String(oS1 - oS0)],
+        // })
+
+        return Asset.fromUnits(result, $systemToken!.symbol)
     }
 
     const rexInfo: Readable<REXInfo> = derived(