diff --git a/projects/ui/src/components/Silo/Overview.tsx b/projects/ui/src/components/Silo/Overview.tsx index 40baccf506..9e6f0fd005 100644 --- a/projects/ui/src/components/Silo/Overview.tsx +++ b/projects/ui/src/components/Silo/Overview.tsx @@ -7,11 +7,7 @@ import { AppState } from '~/state'; import useTabs from '~/hooks/display/useTabs'; import TokenIcon from '~/components/Common/TokenIcon'; import { SEEDS, STALK } from '~/constants/tokens'; -import { - displayPercentage, - displayStalk, - displayUSD, -} from '~/util'; +import { displayPercentage, displayStalk, displayUSD } from '~/util'; import { ChipLabel, StyledTab } from '~/components/Common/Tabs'; import { ZERO_BN } from '~/constants'; import Row from '~/components/Common/Row'; @@ -78,7 +74,10 @@ const Overview: FC<{ const migrationNeeded = useMigrationNeeded(); const siloBalance = useFarmerSiloBalances(); // - const [tab, handleChange] = useTabs(migrationNeeded ? SLUGS : altSLUGS, 'view'); + const [tab, handleChange] = useTabs( + migrationNeeded ? SLUGS : altSLUGS, + 'view' + ); // const ownership = @@ -86,13 +85,15 @@ const Overview: FC<{ ? farmerSilo.stalk.active.div(beanstalkSilo.stalk.total) : ZERO_BN; - const deposits = Object.values(siloBalance).map(token => token.deposited.crates).flat(Infinity) + const deposits = Object.values(siloBalance) + .map((token) => token.deposited.crates) + .flat(Infinity); - let totalStalkGrown = farmerSilo.stalk.grown; + const totalStalkGrown = farmerSilo.stalk.grown; - deposits.forEach((deposit: any) => { - totalStalkGrown = totalStalkGrown.plus(deposit.stalk.grown) - }) + // deposits.forEach((deposit: any) => { + // totalStalkGrown = totalStalkGrown.plus(deposit.stalk.grown) + // }) const stalkStats = useCallback( (s: BigNumber, v: BigNumber[], d: string) => ( @@ -118,9 +119,7 @@ const Overview: FC<{ )} - + - + - + axios .get( @@ -145,8 +153,6 @@ export const useFetchFarmerSilo = () => { sdk.silo.getSeeds(account), ]); - console.log('Fetched migration data', balances); - // Pre-migration, # of seeds is calc'd from the contract getter activeSeedBalance = _activeSeedBalance; @@ -264,9 +270,59 @@ export const useFetchFarmerSilo = () => { dispatch(updateFarmerSiloBalanceSdk(balances)); } - /// earnedStalk (this is already included in activeStalk) - /// earnedSeed (aka plantable seeds) - /// these work because 1 BEAN = 1 BDV. + /** + * We need to calculate the stalk for un-migrated accounts differently than migrated ones + */ + + // First aggregate all crates per token + const stalkPerTokenForUnMigrated = Object.entries(payload).reduce< + TokenMap + >((prev, [tokenAddress, tokenBalances]) => { + if (!season) return prev; + prev[tokenAddress] = + tokenBalances.deposited!.crates.reduce( + (acc, crate) => { + acc.base = acc.base.plus(crate.stalk.base); + acc.grown = acc.grown.plus(crate.stalk.grown); + acc.seeds = acc.seeds.plus(crate.seeds); + acc.unclaimed = ZERO_BN; + return acc; + }, + { + base: ZERO_BN, + grown: ZERO_BN, + unclaimed: ZERO_BN, + seeds: ZERO_BN, + } + ); + return prev; + }, {}); + + // Then aggregate all tokens + const stalkForUnMigrated = Object.entries( + stalkPerTokenForUnMigrated + ).reduce( + (prev, [_, data]) => { + prev.base = prev.base.plus(data.base); + prev.grown = prev.grown.plus(data.grown); + + return prev; + }, + { + base: ZERO_BN, + grown: ZERO_BN, + earned: transform( + sdk.tokens.BEAN.getStalk(earnedBeanBalance), + 'bnjs' + ), + total: ZERO_BN, + } + ); + stalkForUnMigrated.total = stalkForUnMigrated.base + .plus(stalkForUnMigrated.grown) + .plus(stalkForUnMigrated.earned); + // End of un-migrated stalk calculation + const earnedStalkBalance = sdk.tokens.BEAN.getStalk(earnedBeanBalance); const earnedSeedBalance = sdk.tokens.BEAN.getSeeds(earnedBeanBalance); const totalStalkBalance = activeStalkBalance.add(grownStalkBalance); @@ -281,10 +337,18 @@ export const useFetchFarmerSilo = () => { earned: transform(earnedBeanBalance, 'bnjs', sdk.tokens.BEAN), }, stalk: { - active: transform(activeStalkBalance, 'bnjs', sdk.tokens.STALK), - earned: transform(earnedStalkBalance, 'bnjs', sdk.tokens.STALK), - grown: transform(grownStalkBalance, 'bnjs', sdk.tokens.STALK), - total: transform(totalStalkBalance, 'bnjs', sdk.tokens.STALK), + active: migrationNeeded + ? stalkForUnMigrated.base + : transform(activeStalkBalance, 'bnjs', sdk.tokens.STALK), + earned: migrationNeeded + ? stalkForUnMigrated.earned + : transform(earnedStalkBalance, 'bnjs', sdk.tokens.STALK), + grown: migrationNeeded + ? stalkForUnMigrated.grown + : transform(grownStalkBalance, 'bnjs', sdk.tokens.STALK), + total: migrationNeeded + ? stalkForUnMigrated.total + : transform(totalStalkBalance, 'bnjs', sdk.tokens.STALK), grownByToken: grownStalkByToken, }, seeds: { @@ -297,12 +361,6 @@ export const useFetchFarmerSilo = () => { }, }; - // console.log("Silo Rewards", rewards, { - // totalStalkBalance: totalStalkBalance.toHuman(), - // grownStalkBalance: grownStalkBalance.toHuman(), - // earnedBeanBalance: earnedBeanBalance.toHuman(), - // }) - dispatch(updateLegacyFarmerSiloRewards(rewards)); // HEADS UP: this has to be called after updateLegacyFarmerSiloRewards