0g-chain/x/earn/keeper/params.go
2024-08-03 15:03:23 +08:00

63 lines
1.6 KiB
Go

package keeper
import (
"strings"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/0glabs/0g-chain/x/earn/types"
)
const (
bkavaDenom = "bkava"
bkavaPrefix = bkavaDenom + "-"
)
// GetParams returns the params from the store
func (k Keeper) GetParams(ctx sdk.Context) types.Params {
var p types.Params
k.paramSubspace.GetParamSet(ctx, &p)
return p
}
// SetParams sets params on the store
func (k Keeper) SetParams(ctx sdk.Context, params types.Params) {
k.paramSubspace.SetParamSet(ctx, &params)
}
// GetAllowedVaults returns the list of allowed vaults from the module params.
func (k Keeper) GetAllowedVaults(ctx sdk.Context) types.AllowedVaults {
return k.GetParams(ctx).AllowedVaults
}
// getAllowedVaultRaw returns a single vault from the module params specified
// by the denom.
func (k Keeper) getAllowedVaultRaw(
ctx sdk.Context,
vaultDenom string,
) (types.AllowedVault, bool) {
for _, allowedVault := range k.GetAllowedVaults(ctx) {
if allowedVault.Denom == vaultDenom {
return allowedVault, true
}
}
return types.AllowedVault{}, false
}
// GetAllowedVault returns the AllowedVault that corresponds to the
// given denom. If the denom starts with "bkava-" where it will return the
// "bkava" AllowedVault. Otherwise, it will return the exact match for the
// corresponding AllowedVault denom.
func (k *Keeper) GetAllowedVault(
ctx sdk.Context,
vaultDenom string,
) (types.AllowedVault, bool) {
if strings.HasPrefix(vaultDenom, bkavaPrefix) {
return k.getAllowedVaultRaw(ctx, bkavaDenom)
}
return k.getAllowedVaultRaw(ctx, vaultDenom)
}