feat(database): add title resolver and argument completion

Author: kangasta

internal/commands/database/show.go

@@ -2,7 +2,9 @@ package database
 
 import (
 	"github.com/UpCloudLtd/upcloud-cli/internal/commands"
+	"github.com/UpCloudLtd/upcloud-cli/internal/completion"
 	"github.com/UpCloudLtd/upcloud-cli/internal/output"
+	"github.com/UpCloudLtd/upcloud-cli/internal/resolver"
 	"github.com/UpCloudLtd/upcloud-cli/internal/ui"
 	"github.com/UpCloudLtd/upcloud-go-api/v4/upcloud"
 	"github.com/UpCloudLtd/upcloud-go-api/v4/upcloud/request"
@@ -23,6 +25,8 @@ func ShowCommand() commands.Command {
 
 type showCommand struct {
 	*commands.BaseCommand
+	resolver.CachingDatabase
+	completion.Database
 }
 
 // Execute implements commands.MultipleArgumentCommand

internal/completion/database.go

@@ -0,0 +1,27 @@
+package completion
+
+import (
+	"github.com/UpCloudLtd/upcloud-cli/internal/service"
+	"github.com/UpCloudLtd/upcloud-go-api/v4/upcloud/request"
+	"github.com/spf13/cobra"
+)
+
+// Database implements argument completion for databases, by uuid or title.
+type Database struct {
+}
+
+// make sure Database implements the interface
+var _ Provider = Database{}
+
+// CompleteArgument implements completion.Provider
+func (s Database) CompleteArgument(svc service.AllServices, toComplete string) ([]string, cobra.ShellCompDirective) {
+	databases, err := svc.GetManagedDatabases(&request.GetManagedDatabasesRequest{})
+	if err != nil {
+		return None(toComplete)
+	}
+	var vals []string
+	for _, db := range databases {
+		vals = append(vals, db.UUID, db.Title)
+	}
+	return MatchStringPrefix(vals, toComplete, true), cobra.ShellCompDirectiveNoFileComp
+}

internal/resolver/database.go

@@ -0,0 +1,40 @@
+package resolver
+
+import (
+	internal "github.com/UpCloudLtd/upcloud-cli/internal/service"
+	"github.com/UpCloudLtd/upcloud-go-api/v4/upcloud/request"
+)
+
+// CachingDatabase implements resolver for servers, caching the results
+type CachingDatabase struct{}
+
+// make sure we implement the ResolutionProvider interface
+var _ ResolutionProvider = CachingDatabase{}
+
+// Get implements ResolutionProvider.Get
+func (s CachingDatabase) Get(svc internal.AllServices) (Resolver, error) {
+	databases, err := svc.GetManagedDatabases(&request.GetManagedDatabasesRequest{})
+	if err != nil {
+		return nil, err
+	}
+	return func(arg string) (uuid string, err error) {
+		rv := ""
+		for _, db := range databases {
+			if db.Title == arg || db.UUID == arg {
+				if rv != "" {
+					return "", AmbiguousResolutionError(arg)
+				}
+				rv = db.UUID
+			}
+		}
+		if rv != "" {
+			return rv, nil
+		}
+		return "", NotFoundError(arg)
+	}, nil
+}
+
+// PositionalArgumentHelp implements resolver.ResolutionProvider
+func (s CachingDatabase) PositionalArgumentHelp() string {
+	return "<UUID/Title...>"
+}