Add square sizes flag to the payForMessage cli command

app/abci.go

@@ -121,4 +121,4 @@ func hasWirePayForMessage(tx sdk.Tx) bool {
 // hardcoded. todo(evan): don't hardcode the square size
 func (app *App) SquareSize() uint64 {
 	return consts.MaxSquareSize
-}
+}

flags/flags.go

@@ -0,0 +1,17 @@
+package flags
+
+import (
+	"github.com/cosmos/cosmos-sdk/client/flags"
+
+	"github.com/spf13/cobra"
+	"github.com/tendermint/tendermint/pkg/consts"
+)
+
+// List of CLI flags
+const FlagSquareSizes = "square-sizes"
+
+// AddTxFlagsToCmd adds common flags to a module tx command.
+func AddTxFlagsToCmd(cmd *cobra.Command) {
+	flags.AddTxFlagsToCmd(cmd)
+	cmd.Flags().UintSlice(FlagSquareSizes, []uint{consts.MaxSquareSize, 128, 64}, "Specify the square size")
+}

x/payment/client/cli/wirepayformessage.go

@@ -6,11 +6,12 @@ import (
 	"fmt"
 
 	"github.com/spf13/cobra"
-	"github.com/tendermint/tendermint/pkg/consts"
+	//"github.com/tendermint/tendermint/pkg/consts"
 
 	"github.com/celestiaorg/celestia-app/x/payment/types"
 	"github.com/cosmos/cosmos-sdk/client"
 	"github.com/cosmos/cosmos-sdk/client/flags"
+	cflags "github.com/celestiaorg/celestia-app/flags"
 	"github.com/cosmos/cosmos-sdk/client/tx"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 )
@@ -52,7 +53,17 @@ func CmdWirePayForMessage() *cobra.Command {
 			}
 
 			// create the MsgPayForMessage
-			pfmMsg, err := types.NewWirePayForMessage(namespace, message, consts.MaxSquareSize)
+			squareSizes, err := cmd.Flags().GetUintSlice(cflags.FlagSquareSizes)
+			if err != nil {
+				return err
+			}
+
+			squareSizes64 := make([]uint64, len(squareSizes))
+			for i := range squareSizes {
+				squareSizes64[i] = uint64(squareSizes[i])
+			}
+			pfmMsg, err := types.NewWirePayForMessage(namespace, message, squareSizes64...)
+
 			if err != nil {
 				return err
 			}
@@ -101,7 +112,7 @@ func CmdWirePayForMessage() *cobra.Command {
 		},
 	}
 
-	flags.AddTxFlagsToCmd(cmd)
+	cflags.AddTxFlagsToCmd(cmd)
 
 	return cmd
 }

x/payment/client/testutil/integration_test.go

@@ -6,6 +6,7 @@ import (
 	"time"
 
 	"github.com/cosmos/cosmos-sdk/client/flags"
+	cflags "github.com/celestiaorg/celestia-app/flags"
 	"github.com/cosmos/cosmos-sdk/crypto/keyring"
 	"github.com/gogo/protobuf/proto"
 	"github.com/stretchr/testify/suite"
@@ -82,6 +83,19 @@ func (s *IntegrationTestSuite) TestSubmitWirePayForMessage() {
 			},
 			false, 0, &sdk.TxResponse{},
 		},
+		{
+			"valid transaction list of square sizes",
+			[]string{
+				hexNS,
+				hexMsg,
+				fmt.Sprintf("--from=%s", username),
+				fmt.Sprintf("--%s=%s", flags.FlagBroadcastMode, flags.BroadcastBlock),
+				fmt.Sprintf("--%s=%s", flags.FlagFees, sdk.NewCoins(sdk.NewCoin(s.cfg.BondDenom, sdk.NewInt(2))).String()),
+				fmt.Sprintf("--%s=true", flags.FlagSkipConfirmation),
+				fmt.Sprintf("--%s=%s", cflags.FlagSquareSizes, "256,128,64"),
+			},
+			false, 0, &sdk.TxResponse{},
+		},
 	}
 
 	for _, tc := range testCases {