cmd/rofl/machine: Pretty print metadata

Author: matevz

cmd/rofl/machine/show.go

@@ -2,6 +2,7 @@ package machine
 
 import (
 	"context"
+	"encoding/base64"
 	"encoding/binary"
 	"encoding/json"
 	"fmt"
@@ -132,9 +133,7 @@ func prettyPrintMachine(mCfg *machineCfg, out *machineShowOutput) {
 
 	if len(out.Machine.Metadata) > 0 {
 		fmt.Printf("Metadata:\n")
-		for key, value := range out.Machine.Metadata {
-			fmt.Printf("  %s: %s\n", key, value)
-		}
+		prettyPrintMachineMetadata(out.Machine.Metadata, "  ", "  ")
 	}
 
 	fmt.Printf("Resources:\n")
@@ -169,9 +168,7 @@ func prettyPrintMachine(mCfg *machineCfg, out *machineShowOutput) {
 
 		if len(out.Machine.Deployment.Metadata) > 0 {
 			fmt.Printf("  Metadata:\n")
-			for key, value := range out.Machine.Deployment.Metadata {
-				fmt.Printf("    %s: %s\n", key, value)
-			}
+			prettyPrintMachineMetadata(out.Machine.Deployment.Metadata, "    ", "  ")
 		}
 	case nil:
 		fmt.Printf("Deployment: <no current deployment>\n")
@@ -239,6 +236,37 @@ func prettyPrintMachinePorts(extraCfg *roflCmdBuild.AppExtraConfig, appID rofl.A
 	}
 }
 
+func prettyPrintMachineMetadata(metadata map[string]string, prefix, indent string) {
+	fallBackPrint := func(key, value, prefix string) {
+		fmt.Printf("%s%s: %s\n", prefix, key, value)
+	}
+
+	for key, value := range metadata {
+		switch key {
+		case scheduler.MetadataKeyPermissions:
+			// Try to decode value as base64 CBOR value.
+			cborValue, err := base64.StdEncoding.DecodeString(value)
+			if err != nil {
+				fallBackPrint(key, value, prefix)
+			}
+
+			var permissions map[string][]types.Address
+			if err = cbor.Unmarshal(cborValue, &permissions); err != nil {
+				fallBackPrint(key, value, prefix)
+			}
+			fmt.Printf("%s%s:\n", prefix, key)
+			for p, addresses := range permissions {
+				fmt.Printf("%s%s:\n", prefix+indent, p)
+				for _, a := range addresses {
+					fmt.Printf("%s- %s\n", prefix+indent+indent, a)
+				}
+			}
+		default:
+			fallBackPrint(key, value, prefix)
+		}
+	}
+}
+
 func init() {
 	common.AddSelectorFlags(showCmd)
 	showCmd.Flags().AddFlagSet(common.FormatFlag)