chore: update with RegisterAsDMM message

Author: gorgos

chain/exchange/types/codec.go

@@ -32,7 +32,7 @@ func RegisterLegacyAminoCodec(cdc *codec.LegacyAmino) {
 	cdc.RegisterConcrete(&MsgIncreasePositionMargin{}, "exchange/MsgIncreasePositionMargin", nil)
 	cdc.RegisterConcrete(&MsgLiquidatePosition{}, "exchange/MsgLiquidatePosition", nil)
 	cdc.RegisterConcrete(&MsgBatchUpdateOrders{}, "exchange/MsgBatchUpdateOrders", nil)
-
+	cdc.RegisterConcrete(&MsgRegisterAsDMM{}, "exchange/MsgRegisterAsDMM", nil)
 	cdc.RegisterConcrete(&ExchangeEnableProposal{}, "exchange/ExchangeEnableProposal", nil)
 	cdc.RegisterConcrete(&BatchExchangeModificationProposal{}, "exchange/BatchExchangeModificationProposal", nil)
 	cdc.RegisterConcrete(&SpotMarketParamUpdateProposal{}, "exchange/SpotMarketParamUpdateProposal", nil)

chain/exchange/types/errors.go

@@ -65,4 +65,6 @@ var (
 	ErrInvalidBatchMsgUpdate                    = sdkerrors.Register(ModuleName, 58, "Invalid batch msg update")
 	ErrExceedsTopOfBookPrice                    = sdkerrors.Register(ModuleName, 59, "Post-only order exceeds top of book price")
 	ErrInvalidOrderTypeForMessage               = sdkerrors.Register(ModuleName, 60, "Order type not supported for given message")
+	ErrInvalidDMMSender                         = sdkerrors.Register(ModuleName, 61, "Sender must match dmm account")
+	ErrDMMAlreadyRegistered                     = sdkerrors.Register(ModuleName, 62, "DMM is already registered")
 )

chain/exchange/types/exchange.go

@@ -61,9 +61,28 @@ type TradingRewardAccountPoints struct {
 	Points  sdk.Dec
 }
 
-func (p *PointsMultiplier) GetMultiplier(e ExecutionType) sdk.Dec {
+func (p *PointsMultiplier) GetEffectiveMultiplier(e ExecutionType, isRegisteredDMMCache map[string]bool, account sdk.AccAddress) sdk.Dec {
+	if isRegisteredDMMCache[account.String()] {
+		return sdk.ZeroDec()
+	}
+
 	if e.IsMaker() {
 		return p.MakerPointsMultiplier
 	}
+
 	return p.TakerPointsMultiplier
 }
+
+func (p *PointsMultiplier) GetEffectiveMultipliers(isRegisteredDMMCache map[string]bool, account sdk.AccAddress) PointsMultiplier {
+	if isRegisteredDMMCache[account.String()] {
+		return PointsMultiplier{
+			MakerPointsMultiplier: sdk.ZeroDec(),
+			TakerPointsMultiplier: sdk.ZeroDec(),
+		}
+	}
+
+	return PointsMultiplier{
+		MakerPointsMultiplier: p.MakerPointsMultiplier,
+		TakerPointsMultiplier: p.TakerPointsMultiplier,
+	}
+}

chain/exchange/types/key.go

@@ -78,7 +78,9 @@ var (
 	FeeDiscountAccountTierPrefix                          = []byte{0x56} // prefix to each account's fee discount tier and TTL timestamp
 	FeeDiscountBucketAccountFeesPaidPrefix                = []byte{0x57} // prefix to each account's fee paid amount for a given bucket
 	FeeDiscountAccountPastBucketTotalFeesPaidAmountPrefix = []byte{0x58} // prefix to each account's total past bucket fees paid amount FeeDiscountAccountIndicatorPrefix
-	FeeDiscountAccountOrderIndicatorPrefix                = []byte{0x59} // prefix to each account's transient indicator if the account has placed an order that block
+	FeeDiscountAccountOrderIndicatorPrefix                = []byte{0x59} // prefix to each account's transient indicator if the account has placed an order that block that is relevant for fee discounts
+
+	IsRegisteredDMMPrefix = []byte{0x60} // prefix to each account's is registered DMM address key
 )
 
 // GetFeeDiscountAccountFeesPaidInBucketKey provides the key for the account's fees paid in the given bucket
@@ -132,6 +134,16 @@ func GetFeeDiscountAccountTierKey(account sdk.AccAddress) []byte {
 	return buf
 }
 
+// GetIsRegisteredDMMKey provides the key for the registered DMM address
+func GetIsRegisteredDMMKey(account sdk.AccAddress) []byte {
+	accountBz := account.Bytes()
+
+	buf := make([]byte, 0, len(IsRegisteredDMMPrefix)+len(accountBz))
+	buf = append(buf, IsRegisteredDMMPrefix...)
+	buf = append(buf, accountBz...)
+	return buf
+}
+
 // GetFeeDiscountMarketQualificationKey provides the key for the market fee discount qualification status
 func GetFeeDiscountMarketQualificationKey(marketID common.Hash) []byte {
 	return append(FeeDiscountMarketQualificationPrefix, marketID.Bytes()...)

chain/exchange/types/msgs.go

@@ -2,6 +2,7 @@ package types
 
 import (
 	"bytes"
+	"fmt"
 
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
@@ -31,6 +32,7 @@ var (
 	_ sdk.Msg = &MsgInstantPerpetualMarketLaunch{}
 	_ sdk.Msg = &MsgInstantExpiryFuturesMarketLaunch{}
 	_ sdk.Msg = &MsgBatchUpdateOrders{}
+	_ sdk.Msg = &MsgRegisterAsDMM{}
 )
 
 func (o *SpotOrder) ValidateBasic(senderAddr sdk.AccAddress) error {
@@ -924,6 +926,34 @@ func (msg *MsgIncreasePositionMargin) GetSigners() []sdk.AccAddress {
 	return []sdk.AccAddress{sender}
 }
 
+func (msg *MsgRegisterAsDMM) Route() string {
+	return RouterKey
+}
+
+func (msg *MsgRegisterAsDMM) Type() string {
+	return "redeemFromMarketMakingPool"
+}
+
+func (msg *MsgRegisterAsDMM) ValidateBasic() error {
+	if msg.Sender != msg.DmmAccount {
+		return sdkerrors.Wrap(ErrInvalidDMMSender, fmt.Sprintf("Sender: %s doesn't match dmm account: %s", msg.Sender, msg.DmmAccount))
+	}
+
+	return nil
+}
+
+func (msg *MsgRegisterAsDMM) GetSignBytes() []byte {
+	return sdk.MustSortJSON(ModuleCdc.MustMarshalJSON(msg))
+}
+
+func (msg *MsgRegisterAsDMM) GetSigners() []sdk.AccAddress {
+	sender, err := sdk.AccAddressFromBech32(msg.Sender)
+	if err != nil {
+		panic(err)
+	}
+	return []sdk.AccAddress{sender}
+}
+
 func (msg *MsgLiquidatePosition) Route() string {
 	return RouterKey
 }