From 8795a2deae1d24050a1b5b3d294dcc0c4e35d87b Mon Sep 17 00:00:00 2001
From: Rootul Patel <rootulp@gmail.com>
Date: Mon, 11 Dec 2023 16:34:37 -0500
Subject: [PATCH] submit upgrade

---
 tools/upgrademonitor/cmd/root.go      |  3 ++-
 tools/upgrademonitor/internal/grpc.go | 16 +++++++++++-----
 2 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/tools/upgrademonitor/cmd/root.go b/tools/upgrademonitor/cmd/root.go
index 3f74302600..6ae1d94e02 100644
--- a/tools/upgrademonitor/cmd/root.go
+++ b/tools/upgrademonitor/cmd/root.go
@@ -39,9 +39,10 @@ var rootCmd = &cobra.Command{
 					if err != nil {
 						return err
 					}
-					if err := internal.SubmitTryUpgrade(grpcEndpoint, addr); err != nil {
+					if _, err = internal.SubmitTryUpgrade(grpcEndpoint, addr); err != nil {
 						return err
 					}
+					fmt.Printf("submitted try upgrade for version: %v. Exiting.", version)
 					return nil
 				}
 			}
diff --git a/tools/upgrademonitor/internal/grpc.go b/tools/upgrademonitor/internal/grpc.go
index 42b5336977..ccde12a80b 100644
--- a/tools/upgrademonitor/internal/grpc.go
+++ b/tools/upgrademonitor/internal/grpc.go
@@ -30,16 +30,22 @@ func QueryVersionTally(grpcEndpoint string, version uint64) (*upgradetypes.Query
 	return resp, nil
 }
 
-func SubmitTryUpgrade(grpcEndpoint string, addr sdk.AccAddress) error {
+func SubmitTryUpgrade(grpcEndpoint string, addr sdk.AccAddress) (*upgradetypes.MsgTryUpgradeResponse, error) {
 	conn, err := grpc.Dial(grpcEndpoint, grpc.WithTransportCredentials(insecure.NewCredentials()))
 	if err != nil {
-		return fmt.Errorf("did not connect: %v", err)
+		return nil, fmt.Errorf("did not connect: %v", err)
 	}
 	defer conn.Close()
 
-	// msg := upgradetypes.NewMsgTryUpgrade(signer)
-	// TODO (@rootulp): submit tx with msg
-	return nil
+	client := upgradetypes.NewMsgClient(conn)
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second)
+	defer cancel()
+
+	resp, err := client.TryUpgrade(ctx, upgradetypes.NewMsgTryUpgrade(addr))
+	if err != nil {
+		return nil, fmt.Errorf("could not submit try upgrade: %v", err)
+	}
+	return resp, nil
 }
 
 func IsUpgradeable(response *upgradetypes.QueryVersionTallyResponse) bool {