feat(x/bank): autocli tx support (#17868)

Co-authored-by: Julien Robert <julien@rbrt.fr>
2023-10-05 14:39:45 +00:00
parent c0d2d7038e
commit 6f53f7c4a3
16 changed files with 676 additions and 583 deletions
@@ -1,16 +1,21 @@
 package cli

 import (
+	"bytes"
 	"context"
-	"fmt"

+	"github.com/cosmos/gogoproto/proto"
 	"github.com/spf13/cobra"

-	"cosmossdk.io/core/address"
+	sdkmath "cosmossdk.io/math"

 	"github.com/cosmos/cosmos-sdk/client"
+	"github.com/cosmos/cosmos-sdk/client/flags"
+	"github.com/cosmos/cosmos-sdk/client/tx"
 	"github.com/cosmos/cosmos-sdk/testutil"
-	"github.com/cosmos/cosmos-sdk/x/bank/client/cli"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	txtypes "github.com/cosmos/cosmos-sdk/types/tx"
+	"github.com/cosmos/cosmos-sdk/types/tx/signing"
 )

 // ExecTestCLICmd builds the client context, mocks the output and executes the command.
@@ -30,9 +35,123 @@ func ExecTestCLICmd(clientCtx client.Context, cmd *cobra.Command, extraArgs []st
 	return out, nil
 }

-func MsgSendExec(clientCtx client.Context, from, to, amount fmt.Stringer, ac address.Codec, extraArgs ...string) (testutil.BufferWriter, error) {
-	args := []string{from.String(), to.String(), amount.String()}
-	args = append(args, extraArgs...)
+type TestTxConfig struct {
+	Simulate             bool
+	GenOnly              bool
+	Offline              bool
+	Memo                 string
+	Gas                  uint64
+	AccNum               uint64
+	Seq                  uint64
+	Fee                  sdk.Coins
+	IsAsyncBroadcastMode bool
+}

-	return ExecTestCLICmd(clientCtx, cli.NewSendTxCmd(ac), args)
+func SubmitTestTx(clientCtx client.Context, msg proto.Message, from sdk.AccAddress, config TestTxConfig) (testutil.BufferWriter, error) {
+	txBuilder := clientCtx.TxConfig.NewTxBuilder()
+
+	err := txBuilder.SetMsgs(msg)
+	if err != nil {
+		return nil, err
+	}
+
+	if config.Fee != nil {
+		txBuilder.SetFeeAmount(config.Fee)
+	} else {
+		txBuilder.SetFeeAmount(sdk.NewCoins(sdk.NewCoin("stake", sdkmath.NewInt(10)))) // Arbitrary fee
+	}
+
+	if config.Gas != 0 {
+		txBuilder.SetGasLimit(config.Gas)
+	} else {
+		txBuilder.SetGasLimit(flags.DefaultGasLimit) // Need at least 100386
+	}
+
+	if config.Memo != "" {
+		txBuilder.SetMemo(config.Memo)
+	}
+
+	if config.GenOnly {
+		txBz, err := clientCtx.TxConfig.TxJSONEncoder()(txBuilder.GetTx())
+		if err != nil {
+			return nil, err
+		}
+
+		out := bytes.NewBuffer(txBz)
+		return out, nil
+	}
+
+	txFactory := tx.Factory{}
+	txFactory = txFactory.
+		WithChainID(clientCtx.ChainID).
+		WithKeybase(clientCtx.Keyring).
+		WithTxConfig(clientCtx.TxConfig).
+		WithSignMode(signing.SignMode_SIGN_MODE_DIRECT)
+
+	if config.Offline {
+		txFactory = txFactory.
+			WithAccountNumber(config.AccNum).
+			WithSequence(config.Seq)
+	} else {
+		accNum, accSeq, err := clientCtx.AccountRetriever.GetAccountNumberSequence(clientCtx, from)
+		if err != nil {
+			return nil, err
+		}
+
+		txFactory = txFactory.
+			WithAccountNumber(accNum).
+			WithSequence(accSeq)
+	}
+
+	accBytes, err := clientCtx.AddressCodec.StringToBytes(from.String())
+	if err != nil {
+		return nil, err
+	}
+
+	keyRecord, err := clientCtx.Keyring.KeyByAddress(accBytes)
+	if err != nil {
+		return nil, err
+	}
+
+	err = tx.Sign(context.Background(), txFactory, keyRecord.Name, txBuilder, true)
+	if err != nil {
+		return nil, err
+	}
+
+	txBz, err := clientCtx.TxConfig.TxEncoder()(txBuilder.GetTx())
+	if err != nil {
+		return nil, err
+	}
+
+	clientCtx.BroadcastMode = flags.BroadcastSync
+	if config.IsAsyncBroadcastMode {
+		clientCtx.BroadcastMode = flags.BroadcastAsync
+	}
+
+	var res proto.Message
+
+	if config.Simulate {
+		txSvcClient := txtypes.NewServiceClient(clientCtx)
+		res, err = txSvcClient.Simulate(context.Background(), &txtypes.SimulateRequest{
+			TxBytes: txBz,
+		})
+		if err != nil {
+			return nil, err
+		}
+
+	} else {
+		res, err = clientCtx.BroadcastTxSync(txBz)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	bz, err := clientCtx.Codec.MarshalJSON(res)
+	if err != nil {
+		return nil, err
+	}
+
+	out := bytes.NewBuffer(bz)
+
+	return out, err
 }