0g-chain/x/earn/keeper/strategy_hard.go

package keeper

import (
	sdk "github.com/cosmos/cosmos-sdk/types"
	"github.com/kava-labs/kava/x/earn/types"
)

// HardStrategy defines the strategy that deposits assets to Hard
type HardStrategy Keeper

var _ Strategy = (*HardStrategy)(nil)

// GetStrategyType returns the strategy type
func (s *HardStrategy) GetStrategyType() types.StrategyType {
	return types.STRATEGY_TYPE_HARD
}

// GetEstimatedTotalAssets returns the current value of all assets deposited
// in hard.
func (s *HardStrategy) GetEstimatedTotalAssets(ctx sdk.Context, denom string) (sdk.Coin, error) {
	macc := s.accountKeeper.GetModuleAccount(ctx, types.ModuleName)
	deposit, found := s.hardKeeper.GetSyncedDeposit(ctx, macc.GetAddress())
	if !found {
		// Return 0 if no deposit exists for module account
		return sdk.NewCoin(denom, sdk.ZeroInt()), nil
	}

	// Only return the deposit for the vault denom.
	for _, coin := range deposit.Amount {
		if coin.Denom == denom {
			return coin, nil
		}
	}

	// Return 0 if no deposit exists for the vault denom
	return sdk.NewCoin(denom, sdk.ZeroInt()), nil
}

// Deposit deposits the specified amount of coins into hard.
func (s *HardStrategy) Deposit(ctx sdk.Context, amount sdk.Coin) error {
	macc := s.accountKeeper.GetModuleAccount(ctx, types.ModuleName)
	return s.hardKeeper.Deposit(ctx, macc.GetAddress(), sdk.NewCoins(amount))
}

// Withdraw withdraws the specified amount of coins from hard.
func (s *HardStrategy) Withdraw(ctx sdk.Context, amount sdk.Coin) error {
	macc := s.accountKeeper.GetModuleAccount(ctx, types.ModuleName)
	return s.hardKeeper.Withdraw(ctx, macc.GetAddress(), sdk.NewCoins(amount))
}