feat: add take profit price Msg

proto/elys/perpetual/tx.proto

@@ -21,7 +21,8 @@ service Msg {
   rpc AddCollateral       (MsgAddCollateral      ) returns (MsgAddCollateralResponse      );
   rpc BrokerAddCollateral (MsgBrokerAddCollateral) returns (MsgAddCollateralResponse      );
   rpc UpdateStopLoss      (MsgUpdateStopLoss     ) returns (MsgUpdateStopLossResponse     );
-  rpc ClosePositions      (MsgClosePositions     ) returns (MsgClosePositionsResponse);
+  rpc ClosePositions      (MsgClosePositions     ) returns (MsgClosePositionsResponse     );
+  rpc UpdateTakeProfitPrice(MsgUpdateTakeProfitPrice) returns (MsgUpdateTakeProfitPriceResponse);
 }
 message MsgOpen {
   string                   creator           = 1;
@@ -126,4 +127,12 @@ message MsgUpdateStopLoss {
   string price = 3 [(gogoproto.customtype) = "github.com/cosmos/cosmos-sdk/types.Dec", (gogoproto.nullable) = false];
 }
 
-message MsgUpdateStopLossResponse {}
+message MsgUpdateStopLossResponse {}
+
+message MsgUpdateTakeProfitPrice {
+  string creator = 1;
+  uint64 id = 2;
+  string price = 3 [(gogoproto.customtype) = "github.com/cosmos/cosmos-sdk/types.Dec", (gogoproto.nullable) = false];
+}
+
+message MsgUpdateTakeProfitPriceResponse {}

x/perpetual/client/cli/tx.go

@@ -32,6 +32,7 @@
 	cmd.AddCommand(CmdAddCollateral())
 	cmd.AddCommand(CmdClosePositions())
 	cmd.AddCommand(CmdUpdateStopLoss())
+	cmd.AddCommand(CmdUpdateTakeProfitPrice())
 	// this line is used by starport scaffolding # 1
 
 	return cmd

x/perpetual/client/cli/tx_update_take_profit_price.go

@@ -0,0 +1,51 @@
+package cli
+
+import (
+	"errors"
+	"strconv"
+
+	"cosmossdk.io/math"
+	"github.com/cosmos/cosmos-sdk/client"
+	"github.com/cosmos/cosmos-sdk/client/flags"
+	"github.com/cosmos/cosmos-sdk/client/tx"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/elys-network/elys/x/perpetual/types"
+	"github.com/spf13/cobra"
+)
+
+func CmdUpdateTakeProfitPrice() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "update-take-profit-price [id] [amount]",
+		Short: "Broadcast message update-take-profit-price",
+		Args:  cobra.ExactArgs(2),
+		RunE: func(cmd *cobra.Command, args []string) (err error) {
+			argPrice, err := sdk.NewDecFromStr(args[0])
+			if err != nil {
+				return errors.New("invalid take profit amount")
+			}
+			positionId, err := strconv.Atoi(args[1])
+			if err != nil {
+				return err
+			}
+
+			clientCtx, err := client.GetClientTxContext(cmd)
+			if err != nil {
+				return err
+			}
+
+			msg := types.NewMsgUpdateTakeProfitPrice(
+				clientCtx.GetFromAddress().String(),
+				uint64(positionId),
+				math.LegacyDec(argPrice),
+			)
+			if err := msg.ValidateBasic(); err != nil {
+				return err
+			}
+			return tx.GenerateOrBroadcastTxCLI(clientCtx, cmd.Flags(), msg)
+		},
+	}
+
+	flags.AddTxFlagsToCmd(cmd)
+
+	return cmd
+}

x/perpetual/keeper/msg_server_update_take_profit_price.go

@@ -0,0 +1,70 @@
+package keeper
+
+import (
+	"context"
+	"fmt"
+	"strconv"
+
+	errorsmod "cosmossdk.io/errors"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/elys-network/elys/x/perpetual/types"
+	assetprofiletypes "github.com/elys-network/elys/x/assetprofile/types"
+	ptypes "github.com/elys-network/elys/x/parameter/types"
+)
+
+func (k msgServer) UpdateTakeProfitPrice(goCtx context.Context, msg *types.MsgUpdateTakeProfitPrice) (*types.MsgUpdateTakeProfitPriceResponse, error) {
+	ctx := sdk.UnwrapSDKContext(goCtx)
+
+	// Load existing mtp
+	creator := sdk.MustAccAddressFromBech32(msg.Creator)
+	mtp, err := k.GetMTP(ctx, creator, msg.Id)
+	if err != nil {
+		return nil, err
+	}
+
+	poolId := mtp.AmmPoolId
+	pool, found := k.GetPool(ctx, poolId)
+	if !found {
+		return nil, errorsmod.Wrap(types.ErrPoolDoesNotExist, fmt.Sprintf("poolId: %d", poolId))
+	}
+
+	if !pool.Enabled {
+		return nil, errorsmod.Wrap(types.ErrPerpetualDisabled, fmt.Sprintf("poolId: %d", poolId))
+	}
+
+	entry, found := k.assetProfileKeeper.GetEntry(ctx, ptypes.BaseCurrency)
+	if !found {
+		return nil, errorsmod.Wrapf(assetprofiletypes.ErrAssetProfileNotFound, "asset %s not found", ptypes.BaseCurrency)
+	}
+	baseCurrency := entry.Denom
+
+	mtp.TakeProfitPrice = msg.Price
+	mtp.TakeProfitCustody = types.CalcMTPTakeProfitCustody(&mtp)
+	mtp.TakeProfitLiabilities, err = k.CalcMTPTakeProfitLiability(ctx, &mtp, baseCurrency)
+	if err != nil {
+		return nil, err
+	}
+
+	// All take profit liability has to be in liabilities asset
+	err = pool.UpdateTakeProfitLiabilities(ctx, mtp.LiabilitiesAsset, mtp.TakeProfitLiabilities, true, mtp.Position)
+	if err != nil {
+		return nil, err
+	}
+
+	// All take profit custody has to be in custody asset
+	err = pool.UpdateTakeProfitCustody(ctx, mtp.CustodyAsset, mtp.TakeProfitCustody, true, mtp.Position)
+	if err != nil {
+		return nil, err
+	}
+
+	k.SetMTP(ctx, &mtp)
+
+	event := sdk.NewEvent(types.EventOpen,
+		sdk.NewAttribute("id", strconv.FormatInt(int64(mtp.Id), 10)),
+		sdk.NewAttribute("address", mtp.Address),
+		sdk.NewAttribute("take_profit_price", mtp.TakeProfitPrice.String()),
+	)
+	ctx.EventManager().EmitEvent(event)
+
+	return &types.MsgUpdateTakeProfitPriceResponse{}, nil
+}

x/perpetual/types/codec.go

@@ -21,6 +21,7 @@
 	legacy.RegisterAminoMsg(cdc, &MsgAddCollateral{}, "perpetual/MsgAddCollateral")
 	legacy.RegisterAminoMsg(cdc, &MsgClosePositions{}, "perpetual/ClosePositions")
 	legacy.RegisterAminoMsg(cdc, &MsgUpdateStopLoss{}, "perpetual/MsgUpdateStopLoss")
+	legacy.RegisterAminoMsg(cdc, &MsgUpdateTakeProfitPrice{}, "perpetual/MsgUpdateTakeProfitPrice")
 	// this line is used by starport scaffolding # 2
 }
 
@@ -33,6 +34,7 @@
 		&MsgDewhitelist{},
 		&MsgAddCollateral{},
 		&MsgUpdateStopLoss{},
+		&MsgUpdateTakeProfitPrice{},
 	)
 	registry.RegisterImplementations((*sdk.Msg)(nil),
 		&MsgClosePositions{},

x/perpetual/types/message_update_take_profit_price.go

@@ -0,0 +1,53 @@
+package types
+
+import (
+	"fmt"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
+)
+
+const TypeMsgUpdateTakeProfitPrice= "update_take_profit_price"
+
+var _ sdk.Msg = &MsgUpdateTakeProfitPrice{}
+
+func NewMsgUpdateTakeProfitPrice(creator string, id uint64, price sdk.Dec) *MsgUpdateTakeProfitPrice {
+	return &MsgUpdateTakeProfitPrice{
+		Creator:  creator,
+		Id: id,
+		Price:    price,
+	}
+}
+
+func (msg *MsgUpdateTakeProfitPrice) Route() string {
+	return RouterKey
+}
+
+func (msg *MsgUpdateTakeProfitPrice) Type() string {
+	return TypeMsgUpdateTakeProfitPrice
+}
+
+func (msg *MsgUpdateTakeProfitPrice) GetSigners() []sdk.AccAddress {
+	creator, err := sdk.AccAddressFromBech32(msg.Creator)
+	if err != nil {
+		panic(err)
+	}
+	return []sdk.AccAddress{creator}
+}
+
+func (msg *MsgUpdateTakeProfitPrice) GetSignBytes() []byte {
+	bz := ModuleCdc.MustMarshalJSON(msg)
+	return sdk.MustSortJSON(bz)
+}
+
+func (msg *MsgUpdateTakeProfitPrice) ValidateBasic() error {
+	_, err := sdk.AccAddressFromBech32(msg.Creator)
+	if err != nil {
+		return sdkerrors.Wrapf(sdkerrors.ErrInvalidAddress, "invalid creator address (%s)", err)
+	}
+
+	if msg.Price.IsNegative() {
+		return fmt.Errorf("take profit price cannot be negative")
+	}
+	return nil
+}

x/perpetual/types/message_update_take_profit_price_test.go

@@ -0,0 +1,67 @@
+package types_test
+
+import (
+	"testing"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/elys-network/elys/testutil/sample"
+	"github.com/elys-network/elys/x/perpetual/types"
+	"github.com/stretchr/testify/require"
+)
+
+func TestMsgUpdatetakeProfitPrice(t *testing.T) {
+	msg := types.NewMsgUpdateTakeProfitPrice(sample.AccAddress(), 1, sdk.OneDec())
+	require.Equal(t, msg.Route(), types.RouterKey)
+	require.Equal(t, msg.Type(), types.TypeMsgUpdateTakeProfitPrice)
+	require.Equal(t, msg.GetSigners(), []sdk.AccAddress{sdk.MustAccAddressFromBech32(msg.Creator)})
+	require.Equal(t, msg.GetSignBytes(), sdk.MustSortJSON(types.ModuleCdc.MustMarshalJSON(msg)))
+	msg.Creator = ""
+	require.PanicsWithError(t, "empty address string is not allowed", func() { msg.GetSigners() })
+	tests := []struct {
+		name   string
+		setter func()
+		errMsg string
+	}{
+		{
+			name: "success",
+			setter: func() {
+				msg.Creator = sample.AccAddress()
+			},
+			errMsg: "",
+		},
+		{
+			name: "invalid address",
+			setter: func() {
+				msg.Creator = "invalid_address"
+			},
+			errMsg: "invalid creator address",
+		},
+		{
+			name: "take profit price is 0",
+			setter: func() {
+				msg.Creator = sample.AccAddress()
+				msg.Price = sdk.ZeroDec()
+			},
+			errMsg: "",
+		},
+		{
+			name: "take profit price is < 0",
+			setter: func() {
+				msg.Creator = sample.AccAddress()
+				msg.Price = sdk.OneDec().MulInt64(-1)
+			},
+			errMsg: "take profit price cannot be negative",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			tt.setter()
+			err := msg.ValidateBasic()
+			if tt.errMsg != "" {
+				require.ErrorContains(t, err, tt.errMsg)
+			} else {
+				require.NoError(t, err)
+			}
+		})
+	}
+}