cmd/rofl: Add refresh-trust-root subcommand

Fetch the latest consensus trust root and update the rofl.yaml app
manifest automatically instead of editing it by hand.

Author: anzoman

cmd/rofl/refresh_trust_root.go

@@ -0,0 +1,54 @@
+package rofl
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/spf13/cobra"
+
+	buildRofl "github.com/oasisprotocol/cli/build/rofl"
+	"github.com/oasisprotocol/cli/cmd/common"
+	roflCommon "github.com/oasisprotocol/cli/cmd/rofl/common"
+
+	"github.com/oasisprotocol/oasis-sdk/client-sdk/go/connection"
+)
+
+var refreshTrustRootCmd = &cobra.Command{
+	Use:   "refresh-trust-root",
+	Short: "Fetch latest trust root and update the rofl.yaml manifest",
+	Args:  cobra.NoArgs,
+	Run: func(_ *cobra.Command, _ []string) {
+		// Load manifest and deployment, and configure network/paratime/account selection.
+		manifest, deployment, npa := roflCommon.LoadManifestAndSetNPA(nil)
+
+		// Establish connection with the target network.
+		ctx := context.Background()
+		conn, err := connection.Connect(ctx, npa.Network)
+		cobra.CheckErr(err)
+
+		// Determine actual height (respects --height flag if set).
+		height, err := common.GetActualHeight(ctx, conn.Consensus().Core())
+		cobra.CheckErr(err)
+
+		blk, err := conn.Consensus().Core().GetBlock(ctx, height)
+		cobra.CheckErr(err)
+
+		if deployment.TrustRoot == nil {
+			deployment.TrustRoot = &buildRofl.TrustRootConfig{}
+		}
+		deployment.TrustRoot.Height = uint64(height)
+		deployment.TrustRoot.Hash = blk.Hash.Hex()
+
+		if err = manifest.Save(); err != nil {
+			cobra.CheckErr(fmt.Errorf("failed to update manifest: %w", err))
+		}
+
+		fmt.Printf("Updated '%s' with trust root: height=%d, hash=%s\n", manifest.SourceFileName(), height, blk.Hash.Hex())
+	},
+}
+
+func init() {
+	refreshTrustRootCmd.Flags().AddFlagSet(common.SelectorNPFlags)
+	refreshTrustRootCmd.Flags().AddFlagSet(common.HeightFlag)
+	refreshTrustRootCmd.Flags().AddFlagSet(roflCommon.DeploymentFlags)
+}

cmd/rofl/refresh_trust_root_test.go

@@ -0,0 +1,32 @@
+package rofl
+
+import "testing"
+
+func TestRefreshTrustRootUse(t *testing.T) {
+	if refreshTrustRootCmd.Use != "refresh-trust-root" {
+		t.Fatalf("unexpected Use for command: %s", refreshTrustRootCmd.Use)
+	}
+}
+
+func TestRefreshTrustRootCommandRegistered(t *testing.T) {
+	found := false
+	for _, c := range Cmd.Commands() {
+		if c == refreshTrustRootCmd || c.Name() == "refresh-trust-root" {
+			found = true
+			break
+		}
+	}
+	if !found {
+		t.Fatalf("refresh-trust-root command not registered under rofl.Cmd")
+	}
+}
+
+func TestRefreshTrustRootCommandFlags(t *testing.T) {
+	// The command should expose selector flags (network/paratime), height and deployment flags.
+	names := []string{"network", "paratime", "no-paratime", "height", "deployment"}
+	for _, n := range names {
+		if refreshTrustRootCmd.Flags().Lookup(n) == nil {
+			t.Errorf("expected flag %q to be present", n)
+		}
+	}
+}

cmd/rofl/rofl.go

@@ -25,6 +25,7 @@ func init() {
 	Cmd.AddCommand(listCmd)
 	Cmd.AddCommand(setDefaultCmd)
 	Cmd.AddCommand(trustRootCmd)
+	Cmd.AddCommand(refreshTrustRootCmd)
 	Cmd.AddCommand(build.Cmd)
 	Cmd.AddCommand(identityCmd)
 	Cmd.AddCommand(secretCmd)