add partial unlocking flag for cli

osmosis-labs · Mar 7, 2022 · 151990d · 151990d
1 parent bff6f64
commit 151990d
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/x/lockup/client/cli/flags.go b/x/lockup/client/cli/flags.go
@@ -8,6 +8,7 @@ import (
 const (
 	FlagDuration    = "duration"
 	FlagMinDuration = "min-duration"
+	FlagAmount      = "amount"
 )
 
 // FlagSetLockTokens returns flags for LockTokens msg builder
@@ -18,6 +19,13 @@ func FlagSetLockTokens() *flag.FlagSet {
 	return fs
 }
 
+func FlagSetUnlockTokens() *flag.FlagSet {
+	fs := flag.NewFlagSet("", flag.ContinueOnError)
+
+	fs.String(FlagAmount, "", "The amount to be unlocked. e.g. 1osmo")
+	return fs
+}
+
 func FlagSetMinDuration() *flag.FlagSet {
 	fs := flag.NewFlagSet("", flag.ContinueOnError)
 

diff --git a/x/lockup/client/cli/tx.go b/x/lockup/client/cli/tx.go
@@ -125,16 +125,31 @@ func NewBeginUnlockByIDCmd() *cobra.Command {
 				return err
 			}
 
+			coins := sdk.Coins(nil)
+			amountStr, err := cmd.Flags().GetString(FlagAmount)
+			if err != nil {
+				return err
+			}
+
+			if amountStr != "" {
+				coins, err = sdk.ParseCoinsNormalized(amountStr)
+				if err != nil {
+					return err
+				}
+			}
+
 			msg := types.NewMsgBeginUnlocking(
 				clientCtx.GetFromAddress(),
 				uint64(id),
-				nil,
+				coins,
 			)
 
 			return tx.GenerateOrBroadcastTxWithFactory(clientCtx, txf, msg)
 		},
 	}
 
+	cmd.Flags().AddFlagSet(FlagSetUnlockTokens())
+
 	flags.AddTxFlagsToCmd(cmd)
 	return cmd
 }