cmd/rofl: Add providers show command

Author: ptrus

cmd/rofl/common/flags.go

@@ -19,6 +19,9 @@ var (
 	// TermFlags provide the term and count setting.
 	TermFlags *flag.FlagSet
 
+	// ShowOffersFlag is the flag for showing all provider offers.
+	ShowOffersFlag *flag.FlagSet
+
 	// DeploymentName is the name of the ROFL app deployment.
 	DeploymentName string
 
@@ -33,6 +36,9 @@ var (
 
 	// TermCount specific the rental base unit multiplier.
 	TermCount uint64
+
+	// ShowOffers controls whether to display all offers for each provider.
+	ShowOffers bool
 )
 
 func init() {
@@ -48,4 +54,7 @@ func init() {
 	TermFlags = flag.NewFlagSet("", flag.ContinueOnError)
 	TermFlags.StringVar(&Term, "term", "", "term to pay for in advance [hour, month, year]")
 	TermFlags.Uint64Var(&TermCount, "term-count", 1, "number of terms to pay for in advance")
+
+	ShowOffersFlag = flag.NewFlagSet("", flag.ContinueOnError)
+	ShowOffersFlag.BoolVar(&ShowOffers, "show-offers", false, "show all offers for each provider")
 }

cmd/rofl/common/term.go

@@ -31,3 +31,43 @@ func ParseMachineTerm(term string) roflmarket.Term {
 		return 0
 	}
 }
+
+// FormatTerm formats a roflmarket.Term into a human-readable string.
+func FormatTerm(term roflmarket.Term) string {
+	switch term {
+	case roflmarket.TermHour:
+		return TermHour
+	case roflmarket.TermMonth:
+		return TermMonth
+	case roflmarket.TermYear:
+		return TermYear
+	default:
+		return fmt.Sprintf("<unknown: %d>", term)
+	}
+}
+
+// FormatTermAdjectival formats a roflmarket.Term into an adjectival form (e.g., "hourly").
+func FormatTermAdjectival(term roflmarket.Term) string {
+	switch term {
+	case roflmarket.TermHour:
+		return "hourly"
+	case roflmarket.TermMonth:
+		return "monthly"
+	case roflmarket.TermYear:
+		return "yearly"
+	default:
+		return fmt.Sprintf("term_%d", term)
+	}
+}
+
+// FormatTeeType formats a roflmarket.TeeType into a human-readable string.
+func FormatTeeType(tee roflmarket.TeeType) string {
+	switch tee {
+	case roflmarket.TeeTypeSGX:
+		return "sgx"
+	case roflmarket.TeeTypeTDX:
+		return "tdx"
+	default:
+		return fmt.Sprintf("<unknown: %d>", tee)
+	}
+}

cmd/rofl/deploy.go

@@ -40,7 +40,6 @@ var (
 	deployOffer          string
 	deployMachine        string
 	deployForce          bool
-	deployShowOffers     bool
 	deployReplaceMachine bool
 
 	deployCmd = &cobra.Command{
@@ -129,7 +128,7 @@ var (
 
 			fmt.Printf("Using provider: %s (%s)\n", machine.Provider, providerAddr)
 
-			if deployShowOffers {
+			if roflCommon.ShowOffers {
 				// Display all offers supported by the provider.
 				showProviderOffers(ctx, npa, conn, *providerAddr)
 				return
@@ -203,7 +202,7 @@ var (
 				}
 				cobra.CheckErr(totalPrice.Mul(qTermCount))
 				tp := types.NewBaseUnits(totalPrice, offer.Payment.Native.Denomination)
-				fmt.Printf("Selected per-%s pricing term, total price is ", term2str(term))
+				fmt.Printf("Selected per-%s pricing term, total price is ", roflCommon.FormatTerm(term))
 				tp.PrettyPrint(ctx, "", os.Stdout)
 				fmt.Println(".")
 				// Warn the user about the non-refundable rental policy before first renting.
@@ -350,7 +349,7 @@ func pushBundleToOciRepository(orcFilename string, ociRepository string) (string
 func detectTerm(offer *roflmarket.Offer) (term roflmarket.Term) {
 	if offer == nil {
 		cobra.CheckErr(fmt.Errorf("no offers exist to determine payment term"))
-		return // Linter complains otherwise.
+		return term // Linter complains otherwise.
 	}
 	if offer.Payment.Native == nil {
 		cobra.CheckErr(fmt.Errorf("no payment terms available for offer '%s'", offer.ID))
@@ -362,7 +361,7 @@ func detectTerm(offer *roflmarket.Offer) (term roflmarket.Term) {
 		if _, ok := offer.Payment.Native.Terms[term]; !ok {
 			cobra.CheckErr(fmt.Errorf("term '%s' is not available for offer '%s'", roflCommon.Term, offer.ID))
 		}
-		return
+		return term
 	}
 
 	// Take the longest payment period.
@@ -372,20 +371,20 @@ func detectTerm(offer *roflmarket.Offer) (term roflmarket.Term) {
 			term = t
 		}
 	}
-	return
+	return term
 }
 
 func fetchProviderOffers(ctx context.Context, npa *common.NPASelection, conn connection.Connection, provider types.Address) (offers []*roflmarket.Offer, err error) {
 	offers, err = conn.Runtime(npa.ParaTime).ROFLMarket.Offers(ctx, client.RoundLatest, provider)
 	if err != nil {
 		err = fmt.Errorf("failed to query provider: %s", err)
-		return
+		return offers, err
 	}
 	// Order offers, newer first.
 	sort.Slice(offers, func(i, j int) bool {
 		return bytes.Compare(offers[i].ID[:], offers[j].ID[:]) > 0
 	})
-	return
+	return offers, err
 }
 
 func showProviderOffers(ctx context.Context, npa *common.NPASelection, conn connection.Connection, provider types.Address) {
@@ -409,19 +408,9 @@ func showProviderOffer(ctx context.Context, offer *roflmarket.Offer) {
 		name = "<unnamed>"
 	}
 
-	var tee string
-	switch offer.Resources.TEE {
-	case roflmarket.TeeTypeSGX:
-		tee = "sgx"
-	case roflmarket.TeeTypeTDX:
-		tee = "tdx"
-	default:
-		tee = "<unknown>"
-	}
-
 	fmt.Printf("- %s [%s]\n", name, offer.ID)
 	fmt.Printf("  TEE: %s | Memory: %d MiB | vCPUs: %d | Storage: %.2f GiB\n",
-		tee,
+		roflCommon.FormatTeeType(offer.Resources.TEE),
 		offer.Resources.Memory,
 		offer.Resources.CPUCount,
 		float64(offer.Resources.Storage)/1024.,
@@ -455,27 +444,13 @@ func showProviderOffer(ctx context.Context, offer *roflmarket.Offer) {
 
 			bu := types.NewBaseUnits(price, offer.Payment.Native.Denomination)
 			bu.PrettyPrint(ctx, "", os.Stdout)
-			fmt.Printf("/%s", term2str(term))
+			fmt.Printf("/%s", roflCommon.FormatTerm(term))
 			gotPrev = true
 		}
 		fmt.Println()
 	}
 }
 
-// Helper to convert roflmarket term into string.
-func term2str(term roflmarket.Term) string {
-	switch term {
-	case roflmarket.TermHour:
-		return "hour"
-	case roflmarket.TermMonth:
-		return "month"
-	case roflmarket.TermYear:
-		return "year"
-	default:
-		return "<unknown>"
-	}
-}
-
 func resolveAndMarshalPermissions(npa *common.NPASelection, permissions map[string][]string) (string, error) {
 	perms := make(scheduler.Permissions)
 	for action, addresses := range permissions {
@@ -514,12 +489,12 @@ func init() {
 	providerFlags.StringVar(&deployOffer, "offer", "", "set the provider's offer identifier")
 	providerFlags.StringVar(&deployMachine, "machine", buildRofl.DefaultMachineName, "machine to deploy into")
 	providerFlags.BoolVar(&deployForce, "force", false, "force deployment")
-	providerFlags.BoolVar(&deployShowOffers, "show-offers", false, "show all provider offers and quit")
 	providerFlags.BoolVar(&deployReplaceMachine, "replace-machine", false, "rent a new machine if the provided one expired")
 
 	deployCmd.Flags().AddFlagSet(common.AccountFlag)
 	deployCmd.Flags().AddFlagSet(common.RuntimeTxFlags)
 	deployCmd.Flags().AddFlagSet(providerFlags)
+	deployCmd.Flags().AddFlagSet(roflCommon.ShowOffersFlag)
 	deployCmd.Flags().AddFlagSet(roflCommon.DeploymentFlags)
 	deployCmd.Flags().AddFlagSet(roflCommon.WipeFlags)
 	deployCmd.Flags().AddFlagSet(roflCommon.TermFlags)