From 4a4c212938f3c7c689819e9f41c7aee929adb013 Mon Sep 17 00:00:00 2001
From: Jeff Wu <jeff@notional.finance>
Date: Sun, 21 Apr 2024 14:55:40 -0700
Subject: [PATCH] Bugfix/various fee bugs (#769)

* fix: hero stats (#766)

* fix: some initial issues

* fix: withdraw liquidity bugs
---
 .../components/hero/components/hero-stats.tsx | 12 ++++++++----
 .../src/client/analytics-registry-client.ts   |  5 ++++-
 .../hooks/use-liquidity-details.tsx           | 19 -------------------
 .../side-drawers/create-increase-position.tsx |  2 +-
 .../src/side-drawers/convert-asset.tsx        | 15 ++-------------
 .../portfolio/src/side-drawers/roll-debt.tsx  | 15 ++-------------
 .../src/summary/use-trade-summary.tsx         |  8 ++++++--
 packages/transaction/src/calculate.ts         | 11 ++++++++---
 8 files changed, 31 insertions(+), 56 deletions(-)

diff --git a/apps/web/src/containers/LandingPageView/components/hero/components/hero-stats.tsx b/apps/web/src/containers/LandingPageView/components/hero/components/hero-stats.tsx
index 735f659d2..19340095b 100644
--- a/apps/web/src/containers/LandingPageView/components/hero/components/hero-stats.tsx
+++ b/apps/web/src/containers/LandingPageView/components/hero/components/hero-stats.tsx
@@ -10,32 +10,36 @@ import statsImg from '../images/stats_overlay.svg';
 import { FormattedMessage } from 'react-intl';
 import { formatNumber } from '@notional-finance/helpers';
 import { useNotionalContext } from '@notional-finance/notionable-hooks';
+import { SupportedNetworks } from '@notional-finance/util';
 
 const oneMillion = 1_000_000;
 
 export const HeroStats = () => {
   const theme = useTheme();
   const {
-    globalState: { heroStats },
+    globalState: { heroStats, activeAccounts },
   } = useNotionalContext();
+  const isReady = SupportedNetworks.every(
+    (n) => activeAccounts && activeAccounts[n]
+  );
 
   return (
     <StatsContainer>
       <ImgContainer>
         <StatsContent>
-          {heroStats?.totalAccounts ? (
+          {isReady && heroStats ? (
             <div>
               <LargeInputText>
                 {`$${formatNumber(
                   heroStats.totalValueLocked / oneMillion,
-                  0
+                  2
                 )}M`}
                 <BodySecondary>
                   <FormattedMessage defaultMessage={'Total Value Locked'} />
                 </BodySecondary>
               </LargeInputText>
               <LargeInputText sx={{ marginTop: theme.spacing(6) }}>
-                {`$${formatNumber(heroStats.totalOpenDebt / oneMillion, 0)}M`}
+                {`$${formatNumber(heroStats.totalOpenDebt / oneMillion, 1)}M`}
                 <BodySecondary>
                   <FormattedMessage defaultMessage={'Total Open Debt'} />
                 </BodySecondary>
diff --git a/packages/core-entities/src/client/analytics-registry-client.ts b/packages/core-entities/src/client/analytics-registry-client.ts
index c7b718924..0705da670 100644
--- a/packages/core-entities/src/client/analytics-registry-client.ts
+++ b/packages/core-entities/src/client/analytics-registry-client.ts
@@ -331,7 +331,10 @@ export class AnalyticsRegistryClient extends ClientRegistry<unknown> {
         Registry.getTokenRegistry()
           .getAllTokens(n)
           .filter(
-            (t) => t.tokenType === 'PrimeDebt' || t.tokenType === 'PrimeCash'
+            (t) =>
+              t.tokenType === 'PrimeDebt' ||
+              t.tokenType === 'PrimeCash' ||
+              t.tokenType === 'VaultShare'
           )
           .reduce(
             (acc, token) =>
diff --git a/packages/features/liquidity/src/liquidity-leveraged/hooks/use-liquidity-details.tsx b/packages/features/liquidity/src/liquidity-leveraged/hooks/use-liquidity-details.tsx
index 803e1fd89..8ddd9611b 100644
--- a/packages/features/liquidity/src/liquidity-leveraged/hooks/use-liquidity-details.tsx
+++ b/packages/features/liquidity/src/liquidity-leveraged/hooks/use-liquidity-details.tsx
@@ -137,25 +137,6 @@ export const useLiquidityDetails = () => {
         greenOnArrowUp: true,
       },
     },
-    {
-      label: 'Net Worth',
-      current: currentHoldings?.presentValue.toDisplayStringWithSymbol() || '-',
-      updated: {
-        value: newNetWorth?.toDisplayStringWithSymbol() || '-',
-        arrowUp:
-          getChangeType(
-            currentHoldings?.presentValue.toFloat(),
-            newNetWorth?.toFloat()
-          ) === 'increase',
-        checkmark:
-          getChangeType(
-            currentHoldings?.presentValue.toFloat(),
-            newNetWorth?.toFloat()
-          ) === 'cleared',
-        greenOnCheckmark: false,
-        greenOnArrowUp: true,
-      },
-    },
     {
       label: 'Leverage Ratio',
       current: currentHoldings?.leverageRatio
diff --git a/packages/features/liquidity/src/liquidity-leveraged/side-drawers/create-increase-position.tsx b/packages/features/liquidity/src/liquidity-leveraged/side-drawers/create-increase-position.tsx
index 01a4d03ee..a18cf11a5 100644
--- a/packages/features/liquidity/src/liquidity-leveraged/side-drawers/create-increase-position.tsx
+++ b/packages/features/liquidity/src/liquidity-leveraged/side-drawers/create-increase-position.tsx
@@ -56,7 +56,7 @@ export const CreateOrIncreasePosition = () => {
       {currentPosition ? (
         <ManageTerms
           borrowType={
-            currentPosition.debt.tokenType === 'PrimeDebt'
+            currentPosition.debt.balance.tokenType === 'PrimeDebt'
               ? 'Variable'
               : 'Fixed'
           }
diff --git a/packages/features/portfolio/src/side-drawers/convert-asset.tsx b/packages/features/portfolio/src/side-drawers/convert-asset.tsx
index 710b00590..233f9d202 100644
--- a/packages/features/portfolio/src/side-drawers/convert-asset.tsx
+++ b/packages/features/portfolio/src/side-drawers/convert-asset.tsx
@@ -1,17 +1,15 @@
-import { Box, useTheme } from '@mui/material';
+import { Box } from '@mui/material';
 import { TokenBalance } from '@notional-finance/core-entities';
-import { SideBarSubHeader, useCurrencyInputRef } from '@notional-finance/mui';
+import { useCurrencyInputRef } from '@notional-finance/mui';
 import { useTradeContext } from '@notional-finance/notionable-hooks';
 import { AssetInput } from '@notional-finance/trade';
 import { PORTFOLIO_ACTIONS } from '@notional-finance/util';
-import { defineMessage } from 'react-intl';
 import { PortfolioSideDrawer } from './components/portfolio-side-drawer';
 import { SelectConvertAsset } from './components/select-convert-asset';
 import { messages } from './messages';
 import { useState } from 'react';
 
 export const ConvertAsset = () => {
-  const theme = useTheme();
   const context = useTradeContext('ConvertAsset');
   const { currencyInputRef } = useCurrencyInputRef();
   const {
@@ -41,15 +39,6 @@ export const ConvertAsset = () => {
       </Box>
       {collateral && (
         <PortfolioSideDrawer context={context}>
-          <SideBarSubHeader
-            callback={() => updateState({ collateral: undefined })}
-            sx={{
-              position: 'absolute',
-              top: 0,
-              background: theme.palette.background.paper,
-            }}
-            titleText={defineMessage({ defaultMessage: 'Back' })}
-          />
           <AssetInput
             ref={currencyInputRef}
             debtOrCollateral="Debt"
diff --git a/packages/features/portfolio/src/side-drawers/roll-debt.tsx b/packages/features/portfolio/src/side-drawers/roll-debt.tsx
index 4a91bc048..e535c104e 100644
--- a/packages/features/portfolio/src/side-drawers/roll-debt.tsx
+++ b/packages/features/portfolio/src/side-drawers/roll-debt.tsx
@@ -1,17 +1,15 @@
-import { Box, useTheme } from '@mui/material';
+import { Box } from '@mui/material';
 import { TokenBalance } from '@notional-finance/core-entities';
-import { SideBarSubHeader, useCurrencyInputRef } from '@notional-finance/mui';
+import { useCurrencyInputRef } from '@notional-finance/mui';
 import { useTradeContext } from '@notional-finance/notionable-hooks';
 import { AssetInput } from '@notional-finance/trade';
 import { PORTFOLIO_ACTIONS } from '@notional-finance/util';
-import { defineMessage } from 'react-intl';
 import { PortfolioSideDrawer } from './components/portfolio-side-drawer';
 import { SelectConvertAsset } from './components/select-convert-asset';
 import { messages } from './messages';
 import { useState } from 'react';
 
 export const RollDebt = () => {
-  const theme = useTheme();
   const context = useTradeContext('RollDebt');
   const { currencyInputRef } = useCurrencyInputRef();
   const {
@@ -44,15 +42,6 @@ export const RollDebt = () => {
           context={context}
           enablePrimeBorrow={debt?.tokenType === 'PrimeDebt'}
         >
-          <SideBarSubHeader
-            callback={() => updateState({ debt: undefined })}
-            sx={{
-              position: 'absolute',
-              top: 0,
-              background: theme.palette.background.paper,
-            }}
-            titleText={defineMessage({ defaultMessage: 'Back' })}
-          />
           <AssetInput
             ref={currencyInputRef}
             debtOrCollateral="Collateral"
diff --git a/packages/shared/notionable-hooks/src/summary/use-trade-summary.tsx b/packages/shared/notionable-hooks/src/summary/use-trade-summary.tsx
index 41484f644..a8cf8937e 100644
--- a/packages/shared/notionable-hooks/src/summary/use-trade-summary.tsx
+++ b/packages/shared/notionable-hooks/src/summary/use-trade-summary.tsx
@@ -505,7 +505,9 @@ function getRollDebtOrConvertAssetSummary(
   if (netRealizedDebtBalance && debtBalance) {
     summary.push(
       getTradeDetail(
-        netRealizedDebtBalance.abs().neg(),
+        tradeType === 'RollDebt'
+          ? netRealizedDebtBalance.abs()
+          : netRealizedDebtBalance.neg(),
         tradeType === 'RollDebt' ? 'Debt' : 'Asset',
         tradeType === 'RollDebt' ? 'none' : 'withdraw',
         intl,
@@ -519,7 +521,9 @@ function getRollDebtOrConvertAssetSummary(
   if (netRealizedCollateralBalance && collateralBalance)
     summary.push(
       getTradeDetail(
-        netRealizedCollateralBalance,
+        tradeType === 'RollDebt'
+          ? netRealizedCollateralBalance.neg()
+          : netRealizedCollateralBalance.abs(),
         tradeType === 'RollDebt' ? 'Debt' : 'Asset',
         tradeType === 'RollDebt' ? 'repay' : 'none',
         intl,
diff --git a/packages/transaction/src/calculate.ts b/packages/transaction/src/calculate.ts
index 7655040c0..da89f0bc6 100644
--- a/packages/transaction/src/calculate.ts
+++ b/packages/transaction/src/calculate.ts
@@ -60,7 +60,7 @@ export function exchangeToLocalPrime(
     return {
       localPrime: tokensOut[0].toToken(outToken),
       fees: feesPaid[0],
-      netRealized: tokensOut[0].sub(feesPaid[0]).toUnderlying(),
+      netRealized: tokensOut[0].add(feesPaid[0]).toUnderlying(),
     };
   } else if (token.tokenType === 'nToken' && balance.isPositive()) {
     if (!pool) throw Error('Pool is undefined');
@@ -73,7 +73,7 @@ export function exchangeToLocalPrime(
     return {
       localPrime: tokensIn[0].toToken(outToken),
       fees: feesPaid[0],
-      netRealized: balance.sub(feesPaid[0]).toUnderlying(),
+      netRealized: balance.toPrimeCash().sub(feesPaid[0]).toUnderlying(),
     };
   } else if (token.tokenType === 'fCash') {
     if (!pool) throw Error('Pool is undefined');
@@ -275,7 +275,11 @@ export function calculateDebt({
       debtBalance: lpTokens.neg(),
       debtFee: feesPaid[0],
       collateralFee,
-      netRealizedDebtBalance: totalDebtPrime.sub(feesPaid[0]).toUnderlying(),
+      // Ensure netRealizedDebtBalance is positive
+      netRealizedDebtBalance: totalDebtPrime
+        .sub(feesPaid[0])
+        .neg()
+        .toUnderlying(),
       netRealizedCollateralBalance,
     };
   } else if (debt.tokenType === 'fCash') {
@@ -676,6 +680,7 @@ export function calculateDebtCollateralGivenDepositRiskLimit({
           debtBalance: results.debtBalance.add(debtDeposit.debtBalance),
           debtFee: results.debtFee.add(debtDeposit.debtFee),
           netRealizedDebtBalance: results.netRealizedDebtBalance.add(
+            // TODO: this is negative
             debtDeposit.netRealizedDebtBalance
           ),
         }