mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-29 09:45:18 +00:00
110 lines
3.1 KiB
Go
110 lines
3.1 KiB
Go
package keeper
|
|
|
|
import (
|
|
errorsmod "cosmossdk.io/errors"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
"github.com/0glabs/0g-chain/x/hard/types"
|
|
)
|
|
|
|
// Withdraw returns some or all of a deposit back to original depositor
|
|
func (k Keeper) Withdraw(ctx sdk.Context, depositor sdk.AccAddress, coins sdk.Coins) error {
|
|
// Call incentive hooks
|
|
existingDeposit, found := k.GetDeposit(ctx, depositor)
|
|
if !found {
|
|
return errorsmod.Wrapf(types.ErrDepositNotFound, "no deposit found for %s", depositor)
|
|
}
|
|
k.BeforeDepositModified(ctx, existingDeposit)
|
|
|
|
existingBorrow, hasExistingBorrow := k.GetBorrow(ctx, depositor)
|
|
if hasExistingBorrow {
|
|
k.BeforeBorrowModified(ctx, existingBorrow)
|
|
}
|
|
|
|
// Sync interest
|
|
k.SyncBorrowInterest(ctx, depositor)
|
|
k.SyncSupplyInterest(ctx, depositor)
|
|
|
|
// Refresh Deposit after syncing interest
|
|
deposit, _ := k.GetDeposit(ctx, depositor)
|
|
|
|
amount, err := k.CalculateWithdrawAmount(deposit.Amount, coins)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
borrow, found := k.GetBorrow(ctx, depositor)
|
|
if !found {
|
|
borrow = types.Borrow{}
|
|
}
|
|
|
|
proposedDeposit := types.NewDeposit(deposit.Depositor, deposit.Amount.Sub(amount...), types.SupplyInterestFactors{})
|
|
valid, err := k.IsWithinValidLtvRange(ctx, proposedDeposit, borrow)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !valid {
|
|
return errorsmod.Wrapf(types.ErrInvalidWithdrawAmount, "proposed withdraw outside loan-to-value range")
|
|
}
|
|
|
|
err = k.bankKeeper.SendCoinsFromModuleToAccount(ctx, types.ModuleAccountName, depositor, amount)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// If any coin denoms have been completely withdrawn reset the denom's supply index factor
|
|
for _, coin := range deposit.Amount {
|
|
if !sdk.NewCoins(coin).DenomsSubsetOf(proposedDeposit.Amount) {
|
|
depositIndex, removed := deposit.Index.RemoveInterestFactor(coin.Denom)
|
|
if !removed {
|
|
return errorsmod.Wrapf(types.ErrInvalidIndexFactorDenom, "%s", coin.Denom)
|
|
}
|
|
deposit.Index = depositIndex
|
|
}
|
|
}
|
|
|
|
deposit.Amount = deposit.Amount.Sub(amount...)
|
|
if deposit.Amount.Empty() {
|
|
k.DeleteDeposit(ctx, deposit)
|
|
} else {
|
|
k.SetDeposit(ctx, deposit)
|
|
}
|
|
|
|
// Update total supplied amount
|
|
err = k.DecrementSuppliedCoins(ctx, amount)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Call incentive hook
|
|
k.AfterDepositModified(ctx, deposit)
|
|
|
|
ctx.EventManager().EmitEvent(
|
|
sdk.NewEvent(
|
|
types.EventTypeHardWithdrawal,
|
|
sdk.NewAttribute(sdk.AttributeKeyAmount, amount.String()),
|
|
sdk.NewAttribute(types.AttributeKeyDepositor, depositor.String()),
|
|
),
|
|
)
|
|
return nil
|
|
}
|
|
|
|
// CalculateWithdrawAmount enables full withdraw of deposited coins by adjusting withdraw amount
|
|
// to equal total deposit amount if the requested withdraw amount > current deposit amount
|
|
func (k Keeper) CalculateWithdrawAmount(available sdk.Coins, request sdk.Coins) (sdk.Coins, error) {
|
|
result := sdk.Coins{}
|
|
|
|
if !request.DenomsSubsetOf(available) {
|
|
return result, types.ErrInvalidWithdrawDenom
|
|
}
|
|
|
|
for _, coin := range request {
|
|
if coin.Amount.GT(available.AmountOf(coin.Denom)) {
|
|
result = append(result, sdk.NewCoin(coin.Denom, available.AmountOf(coin.Denom)))
|
|
} else {
|
|
result = append(result, coin)
|
|
}
|
|
}
|
|
return result, nil
|
|
}
|