forked from cosmos/cosmos-sdk
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(cosmovisor): Add prepare-upgrade cmd (cosmos#21972)
- Loading branch information
1 parent
52d8b2e
commit 3f9c9a0
Showing
6 changed files
with
176 additions
and
5 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,126 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "context" | ||
| "crypto/tls" | ||
| "fmt" | ||
| "path/filepath" | ||
| "strings" | ||
| "time" | ||
|
|
||
| "github.com/spf13/cobra" | ||
| "google.golang.org/grpc" | ||
| "google.golang.org/grpc/credentials" | ||
| "google.golang.org/grpc/credentials/insecure" | ||
|
|
||
| "cosmossdk.io/tools/cosmovisor" | ||
| "cosmossdk.io/x/upgrade/plan" | ||
| upgradetypes "cosmossdk.io/x/upgrade/types" | ||
| ) | ||
|
|
||
| func NewPrepareUpgradeCmd() *cobra.Command { | ||
| cmd := &cobra.Command{ | ||
| Use: "prepare-upgrade", | ||
| Short: "Prepare for the next upgrade", | ||
| Long: `Prepare for the next upgrade by downloading and verifying the upgrade binary. | ||
| This command will query the chain for the current upgrade plan and download the specified binary. | ||
| gRPC must be enabled on the node for this command to work.`, | ||
| RunE: prepareUpgradeHandler, | ||
| SilenceUsage: false, | ||
| Args: cobra.NoArgs, | ||
| } | ||
|
|
||
| return cmd | ||
| } | ||
|
|
||
| func prepareUpgradeHandler(cmd *cobra.Command, _ []string) error { | ||
| configPath, err := cmd.Flags().GetString(cosmovisor.FlagCosmovisorConfig) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to get config flag: %w", err) | ||
| } | ||
|
|
||
| cfg, err := cosmovisor.GetConfigFromFile(configPath) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to get config: %w", err) | ||
| } | ||
|
|
||
| logger := cfg.Logger(cmd.OutOrStdout()) | ||
|
|
||
| grpcAddress := cfg.GRPCAddress | ||
| logger.Info("Using gRPC address", "address", grpcAddress) | ||
|
|
||
| upgradeInfo, err := queryUpgradeInfoFromChain(grpcAddress) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to query upgrade info: %w", err) | ||
| } | ||
|
|
||
| if upgradeInfo == nil { | ||
| logger.Info("No active upgrade plan found") | ||
| return nil | ||
| } | ||
|
|
||
| logger.Info("Preparing for upgrade", "name", upgradeInfo.Name, "height", upgradeInfo.Height) | ||
|
|
||
| upgradeInfoParsed, err := plan.ParseInfo(upgradeInfo.Info, plan.ParseOptionEnforceChecksum(cfg.DownloadMustHaveChecksum)) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to parse upgrade info: %w", err) | ||
| } | ||
|
|
||
| binaryURL, err := cosmovisor.GetBinaryURL(upgradeInfoParsed.Binaries) | ||
| if err != nil { | ||
| return fmt.Errorf("binary URL not found in upgrade plan. Cannot prepare for upgrade: %w", err) | ||
| } | ||
|
|
||
| logger.Info("Downloading upgrade binary", "url", binaryURL) | ||
|
|
||
| upgradeBin := filepath.Join(cfg.UpgradeBin(upgradeInfo.Name), cfg.Name) | ||
| if err := plan.DownloadUpgrade(filepath.Dir(upgradeBin), binaryURL, cfg.Name); err != nil { | ||
| return fmt.Errorf("failed to download and verify binary: %w", err) | ||
| } | ||
|
|
||
| logger.Info("Upgrade preparation complete", "name", upgradeInfo.Name, "height", upgradeInfo.Height) | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func queryUpgradeInfoFromChain(grpcAddress string) (*upgradetypes.Plan, error) { | ||
| if grpcAddress == "" { | ||
| return nil, fmt.Errorf("gRPC address is empty") | ||
| } | ||
|
|
||
| grpcConn, err := getClient(grpcAddress) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to open gRPC client: %w", err) | ||
| } | ||
| defer grpcConn.Close() | ||
|
|
||
| queryClient := upgradetypes.NewQueryClient(grpcConn) | ||
|
|
||
| ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) | ||
| defer cancel() | ||
|
|
||
| res, err := queryClient.CurrentPlan(ctx, &upgradetypes.QueryCurrentPlanRequest{}) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to query current upgrade plan: %w", err) | ||
| } | ||
|
|
||
| return res.Plan, nil | ||
| } | ||
|
|
||
| func getClient(endpoint string) (*grpc.ClientConn, error) { | ||
| var creds credentials.TransportCredentials | ||
| if strings.HasPrefix(endpoint, "https://") { | ||
| tlsConfig := &tls.Config{ | ||
| MinVersion: tls.VersionTLS12, | ||
| } | ||
| creds = credentials.NewTLS(tlsConfig) | ||
| } else { | ||
| creds = insecure.NewCredentials() | ||
| } | ||
|
|
||
| opts := []grpc.DialOption{ | ||
| grpc.WithTransportCredentials(creds), | ||
| } | ||
|
|
||
| return grpc.NewClient(endpoint, opts...) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters