package keeper import ( "time" "github.com/cosmos/cosmos-sdk/codec" "github.com/cosmos/cosmos-sdk/store/prefix" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/cosmos/cosmos-sdk/x/params/subspace" "github.com/kava-labs/kava/x/incentive/types" ) // Keeper keeper for the incentive module type Keeper struct { accountKeeper types.AccountKeeper cdc *codec.Codec cdpKeeper types.CdpKeeper hardKeeper types.HardKeeper key sdk.StoreKey paramSubspace subspace.Subspace supplyKeeper types.SupplyKeeper stakingKeeper types.StakingKeeper } // NewKeeper creates a new keeper func NewKeeper( cdc *codec.Codec, key sdk.StoreKey, paramstore subspace.Subspace, sk types.SupplyKeeper, cdpk types.CdpKeeper, hk types.HardKeeper, ak types.AccountKeeper, stk types.StakingKeeper, ) Keeper { return Keeper{ accountKeeper: ak, cdc: cdc, cdpKeeper: cdpk, hardKeeper: hk, key: key, paramSubspace: paramstore.WithKeyTable(types.ParamKeyTable()), supplyKeeper: sk, stakingKeeper: stk, } } // GetUSDXMintingClaim returns the claim in the store corresponding the the input address collateral type and id and a boolean for if the claim was found func (k Keeper) GetUSDXMintingClaim(ctx sdk.Context, addr sdk.AccAddress) (types.USDXMintingClaim, bool) { store := prefix.NewStore(ctx.KVStore(k.key), types.USDXMintingClaimKeyPrefix) bz := store.Get(addr) if bz == nil { return types.USDXMintingClaim{}, false } var c types.USDXMintingClaim k.cdc.MustUnmarshalBinaryBare(bz, &c) return c, true } // SetUSDXMintingClaim sets the claim in the store corresponding to the input address, collateral type, and id func (k Keeper) SetUSDXMintingClaim(ctx sdk.Context, c types.USDXMintingClaim) { store := prefix.NewStore(ctx.KVStore(k.key), types.USDXMintingClaimKeyPrefix) bz := k.cdc.MustMarshalBinaryBare(c) store.Set(c.Owner, bz) } // DeleteUSDXMintingClaim deletes the claim in the store corresponding to the input address, collateral type, and id func (k Keeper) DeleteUSDXMintingClaim(ctx sdk.Context, owner sdk.AccAddress) { store := prefix.NewStore(ctx.KVStore(k.key), types.USDXMintingClaimKeyPrefix) store.Delete(owner) } // IterateUSDXMintingClaims iterates over all claim objects in the store and preforms a callback function func (k Keeper) IterateUSDXMintingClaims(ctx sdk.Context, cb func(c types.USDXMintingClaim) (stop bool)) { store := prefix.NewStore(ctx.KVStore(k.key), types.USDXMintingClaimKeyPrefix) iterator := sdk.KVStorePrefixIterator(store, []byte{}) defer iterator.Close() for ; iterator.Valid(); iterator.Next() { var c types.USDXMintingClaim k.cdc.MustUnmarshalBinaryBare(iterator.Value(), &c) if cb(c) { break } } } // GetAllUSDXMintingClaims returns all Claim objects in the store func (k Keeper) GetAllUSDXMintingClaims(ctx sdk.Context) types.USDXMintingClaims { cs := types.USDXMintingClaims{} k.IterateUSDXMintingClaims(ctx, func(c types.USDXMintingClaim) (stop bool) { cs = append(cs, c) return false }) return cs } // GetPreviousUSDXMintingAccrualTime returns the last time a collateral type accrued USDX minting rewards func (k Keeper) GetPreviousUSDXMintingAccrualTime(ctx sdk.Context, ctype string) (blockTime time.Time, found bool) { store := prefix.NewStore(ctx.KVStore(k.key), types.PreviousUSDXMintingRewardAccrualTimeKeyPrefix) bz := store.Get([]byte(ctype)) if bz == nil { return time.Time{}, false } k.cdc.MustUnmarshalBinaryBare(bz, &blockTime) return blockTime, true } // SetPreviousUSDXMintingAccrualTime sets the last time a collateral type accrued USDX minting rewards func (k Keeper) SetPreviousUSDXMintingAccrualTime(ctx sdk.Context, ctype string, blockTime time.Time) { store := prefix.NewStore(ctx.KVStore(k.key), types.PreviousUSDXMintingRewardAccrualTimeKeyPrefix) store.Set([]byte(ctype), k.cdc.MustMarshalBinaryBare(blockTime)) } // IterateUSDXMintingAccrualTimes iterates over all previous USDX minting accrual times and preforms a callback function func (k Keeper) IterateUSDXMintingAccrualTimes(ctx sdk.Context, cb func(string, time.Time) (stop bool)) { store := prefix.NewStore(ctx.KVStore(k.key), types.PreviousUSDXMintingRewardAccrualTimeKeyPrefix) iterator := sdk.KVStorePrefixIterator(store, []byte{}) defer iterator.Close() for ; iterator.Valid(); iterator.Next() { var accrualTime time.Time var collateralType string k.cdc.MustUnmarshalBinaryBare(iterator.Value(), &collateralType) k.cdc.MustUnmarshalBinaryBare(iterator.Value(), &accrualTime) if cb(collateralType, accrualTime) { break } } } // GetUSDXMintingRewardFactor returns the current reward factor for an individual collateral type func (k Keeper) GetUSDXMintingRewardFactor(ctx sdk.Context, ctype string) (factor sdk.Dec, found bool) { store := prefix.NewStore(ctx.KVStore(k.key), types.USDXMintingRewardFactorKeyPrefix) bz := store.Get([]byte(ctype)) if bz == nil { return sdk.ZeroDec(), false } k.cdc.MustUnmarshalBinaryBare(bz, &factor) return factor, true } // SetUSDXMintingRewardFactor sets the current reward factor for an individual collateral type func (k Keeper) SetUSDXMintingRewardFactor(ctx sdk.Context, ctype string, factor sdk.Dec) { store := prefix.NewStore(ctx.KVStore(k.key), types.USDXMintingRewardFactorKeyPrefix) store.Set([]byte(ctype), k.cdc.MustMarshalBinaryBare(factor)) } // GetHardLiquidityProviderClaim returns the claim in the store corresponding the the input address collateral type and id and a boolean for if the claim was found func (k Keeper) GetHardLiquidityProviderClaim(ctx sdk.Context, addr sdk.AccAddress) (types.HardLiquidityProviderClaim, bool) { store := prefix.NewStore(ctx.KVStore(k.key), types.HardLiquidityClaimKeyPrefix) bz := store.Get(addr) if bz == nil { return types.HardLiquidityProviderClaim{}, false } var c types.HardLiquidityProviderClaim k.cdc.MustUnmarshalBinaryBare(bz, &c) return c, true } // SetHardLiquidityProviderClaim sets the claim in the store corresponding to the input address, collateral type, and id func (k Keeper) SetHardLiquidityProviderClaim(ctx sdk.Context, c types.HardLiquidityProviderClaim) { store := prefix.NewStore(ctx.KVStore(k.key), types.HardLiquidityClaimKeyPrefix) bz := k.cdc.MustMarshalBinaryBare(c) store.Set(c.Owner, bz) } // DeleteHardLiquidityProviderClaim deletes the claim in the store corresponding to the input address, collateral type, and id func (k Keeper) DeleteHardLiquidityProviderClaim(ctx sdk.Context, owner sdk.AccAddress) { store := prefix.NewStore(ctx.KVStore(k.key), types.HardLiquidityClaimKeyPrefix) store.Delete(owner) } // IterateHardLiquidityProviderClaims iterates over all claim objects in the store and preforms a callback function func (k Keeper) IterateHardLiquidityProviderClaims(ctx sdk.Context, cb func(c types.HardLiquidityProviderClaim) (stop bool)) { store := prefix.NewStore(ctx.KVStore(k.key), types.HardLiquidityClaimKeyPrefix) iterator := sdk.KVStorePrefixIterator(store, []byte{}) defer iterator.Close() for ; iterator.Valid(); iterator.Next() { var c types.HardLiquidityProviderClaim k.cdc.MustUnmarshalBinaryBare(iterator.Value(), &c) if cb(c) { break } } } // GetAllHardLiquidityProviderClaims returns all Claim objects in the store func (k Keeper) GetAllHardLiquidityProviderClaims(ctx sdk.Context) types.HardLiquidityProviderClaims { cs := types.HardLiquidityProviderClaims{} k.IterateHardLiquidityProviderClaims(ctx, func(c types.HardLiquidityProviderClaim) (stop bool) { cs = append(cs, c) return false }) return cs } // SetHardSupplyRewardFactor sets the current interest factor for an individual market func (k Keeper) SetHardSupplyRewardFactor(ctx sdk.Context, denom string, borrowIndex sdk.Dec) { store := prefix.NewStore(ctx.KVStore(k.key), types.HardSupplyRewardFactorKeyPrefix) bz := k.cdc.MustMarshalBinaryBare(borrowIndex) store.Set([]byte(denom), bz) } // GetHardSupplyRewardFactor returns the current interest factor for an individual market func (k Keeper) GetHardSupplyRewardFactor(ctx sdk.Context, denom string) (sdk.Dec, bool) { store := prefix.NewStore(ctx.KVStore(k.key), types.HardSupplyRewardFactorKeyPrefix) bz := store.Get([]byte(denom)) if bz == nil { return sdk.ZeroDec(), false } var interestFactor sdk.Dec k.cdc.MustUnmarshalBinaryBare(bz, &interestFactor) return interestFactor, true } // SetHardBorrowRewardFactor sets the current interest factor for an individual market func (k Keeper) SetHardBorrowRewardFactor(ctx sdk.Context, denom string, borrowIndex sdk.Dec) { store := prefix.NewStore(ctx.KVStore(k.key), types.HardBorrowRewardFactorKeyPrefix) bz := k.cdc.MustMarshalBinaryBare(borrowIndex) store.Set([]byte(denom), bz) } // GetHardBorrowRewardFactor returns the current interest factor for an individual market func (k Keeper) GetHardBorrowRewardFactor(ctx sdk.Context, denom string) (sdk.Dec, bool) { store := prefix.NewStore(ctx.KVStore(k.key), types.HardBorrowRewardFactorKeyPrefix) bz := store.Get([]byte(denom)) if bz == nil { return sdk.ZeroDec(), false } var interestFactor sdk.Dec k.cdc.MustUnmarshalBinaryBare(bz, &interestFactor) return interestFactor, true } // GetHardDelegatorRewardFactor returns the current reward factor for an individual collateral type func (k Keeper) GetHardDelegatorRewardFactor(ctx sdk.Context, ctype string) (factor sdk.Dec, found bool) { store := prefix.NewStore(ctx.KVStore(k.key), types.HardDelegatorRewardFactorKeyPrefix) bz := store.Get([]byte(ctype)) if bz == nil { return sdk.ZeroDec(), false } k.cdc.MustUnmarshalBinaryBare(bz, &factor) return factor, true } // SetHardDelegatorRewardFactor sets the current reward factor for an individual collateral type func (k Keeper) SetHardDelegatorRewardFactor(ctx sdk.Context, ctype string, factor sdk.Dec) { store := prefix.NewStore(ctx.KVStore(k.key), types.HardDelegatorRewardFactorKeyPrefix) store.Set([]byte(ctype), k.cdc.MustMarshalBinaryBare(factor)) } // GetPreviousHardSupplyRewardAccrualTime returns the last time a denom accrued Hard protocol supply-side rewards func (k Keeper) GetPreviousHardSupplyRewardAccrualTime(ctx sdk.Context, denom string) (blockTime time.Time, found bool) { store := prefix.NewStore(ctx.KVStore(k.key), types.PreviousHardSupplyRewardAccrualTimeKeyPrefix) bz := store.Get([]byte(denom)) if bz == nil { return time.Time{}, false } k.cdc.MustUnmarshalBinaryBare(bz, &blockTime) return blockTime, true } // SetPreviousHardSupplyRewardAccrualTime sets the last time a denom accrued Hard protocol supply-side rewards func (k Keeper) SetPreviousHardSupplyRewardAccrualTime(ctx sdk.Context, denom string, blockTime time.Time) { store := prefix.NewStore(ctx.KVStore(k.key), types.PreviousHardSupplyRewardAccrualTimeKeyPrefix) store.Set([]byte(denom), k.cdc.MustMarshalBinaryBare(blockTime)) } // GetPreviousHardBorrowRewardAccrualTime returns the last time a denom accrued Hard protocol borrow-side rewards func (k Keeper) GetPreviousHardBorrowRewardAccrualTime(ctx sdk.Context, denom string) (blockTime time.Time, found bool) { store := prefix.NewStore(ctx.KVStore(k.key), types.PreviousHardBorrowRewardAccrualTimeKeyPrefix) bz := store.Get([]byte(denom)) if bz == nil { return time.Time{}, false } k.cdc.MustUnmarshalBinaryBare(bz, &blockTime) return blockTime, true } // SetPreviousHardBorrowRewardAccrualTime sets the last time a denom accrued Hard protocol borrow-side rewards func (k Keeper) SetPreviousHardBorrowRewardAccrualTime(ctx sdk.Context, denom string, blockTime time.Time) { store := prefix.NewStore(ctx.KVStore(k.key), types.PreviousHardBorrowRewardAccrualTimeKeyPrefix) store.Set([]byte(denom), k.cdc.MustMarshalBinaryBare(blockTime)) } // GetPreviousHardDelegatorRewardAccrualTime returns the last time a denom accrued Hard protocol delegator rewards func (k Keeper) GetPreviousHardDelegatorRewardAccrualTime(ctx sdk.Context, denom string) (blockTime time.Time, found bool) { store := prefix.NewStore(ctx.KVStore(k.key), types.PreviousHardDelegatorRewardAccrualTimeKeyPrefix) bz := store.Get([]byte(denom)) if bz == nil { return time.Time{}, false } k.cdc.MustUnmarshalBinaryBare(bz, &blockTime) return blockTime, true } // SetPreviousHardDelegatorRewardAccrualTime sets the last time a denom accrued Hard protocol delegator rewards func (k Keeper) SetPreviousHardDelegatorRewardAccrualTime(ctx sdk.Context, denom string, blockTime time.Time) { store := prefix.NewStore(ctx.KVStore(k.key), types.PreviousHardDelegatorRewardAccrualTimeKeyPrefix) store.Set([]byte(denom), k.cdc.MustMarshalBinaryBare(blockTime)) }