feat(lb): add loadbalancer list command

Author: kangasta

internal/commands/all/all.go

@@ -5,6 +5,7 @@ import (
 	"github.com/UpCloudLtd/upcloud-cli/internal/commands/account"
 	"github.com/UpCloudLtd/upcloud-cli/internal/commands/database"
 	"github.com/UpCloudLtd/upcloud-cli/internal/commands/ipaddress"
+	"github.com/UpCloudLtd/upcloud-cli/internal/commands/loadbalancer"
 	"github.com/UpCloudLtd/upcloud-cli/internal/commands/network"
 	"github.com/UpCloudLtd/upcloud-cli/internal/commands/networkinterface"
 	"github.com/UpCloudLtd/upcloud-cli/internal/commands/root"
@@ -106,6 +107,10 @@ func BuildCommands(rootCmd *cobra.Command, conf *config.Config) {
 	commands.BuildCommand(database.TypesCommand(), databaseCommand.Cobra(), conf)
 	commands.BuildCommand(database.PlansCommand(), databaseCommand.Cobra(), conf)
 
+	// LoadBalancers
+	loadbalancerCommand := commands.BuildCommand(loadbalancer.BaseLoadBalancerCommand(), rootCmd, conf)
+	commands.BuildCommand(loadbalancer.ListCommand(), loadbalancerCommand.Cobra(), conf)
+
 	// Misc
 	commands.BuildCommand(
 		&root.CompletionCommand{

internal/commands/database/database.go

@@ -4,7 +4,7 @@ import (
 	"github.com/UpCloudLtd/upcloud-cli/internal/commands"
 )
 
-// BaseDatabaseCommand creates the base "zone" command
+// BaseDatabaseCommand creates the base "database" command
 func BaseDatabaseCommand() commands.Command {
 	return &databaseCommand{
 		commands.New("database", "Manage databases"),

internal/commands/loadbalancer/list.go

@@ -0,0 +1,52 @@
+package loadbalancer
+
+import (
+	"github.com/UpCloudLtd/upcloud-cli/internal/commands"
+	"github.com/UpCloudLtd/upcloud-cli/internal/output"
+	"github.com/UpCloudLtd/upcloud-cli/internal/ui"
+	"github.com/UpCloudLtd/upcloud-go-api/v4/upcloud/request"
+)
+
+// ListCommand creates the "loadbalancer list" command
+func ListCommand() commands.Command {
+	return &listCommand{
+		BaseCommand: commands.New("list", "List current load balancers", "upctl loadbalancer list"),
+	}
+}
+
+type listCommand struct {
+	*commands.BaseCommand
+}
+
+// ExecuteWithoutArguments implements commands.NoArgumentCommand
+func (s *listCommand) ExecuteWithoutArguments(exec commands.Executor) (output.Output, error) {
+	svc := exec.All()
+	loadbalancers, err := svc.GetLoadBalancers(&request.GetLoadBalancersRequest{})
+	if err != nil {
+		return nil, err
+	}
+
+	rows := []output.TableRow{}
+	for _, lb := range loadbalancers {
+		coloredState := commands.LoadBalancerOperationalStateColour(lb.OperationalState).Sprint(lb.OperationalState)
+
+		rows = append(rows, output.TableRow{
+			lb.UUID,
+			lb.Name,
+			lb.Plan,
+			lb.Zone,
+			coloredState,
+		})
+	}
+
+	return output.Table{
+		Columns: []output.TableColumn{
+			{Key: "uuid", Header: "UUID", Colour: ui.DefaultUUUIDColours},
+			{Key: "name", Header: "Name"},
+			{Key: "plan", Header: "Plan"},
+			{Key: "zone", Header: "Zone"},
+			{Key: "state", Header: "State"},
+		},
+		Rows: rows,
+	}, nil
+}

internal/commands/loadbalancer/loadbalancer.go

@@ -0,0 +1,16 @@
+package loadbalancer
+
+import (
+	"github.com/UpCloudLtd/upcloud-cli/internal/commands"
+)
+
+// BaseLoadBalancerCommand creates the base "loadbalancer" command
+func BaseLoadBalancerCommand() commands.Command {
+	return &loadbalancerCommand{
+		commands.New("loadbalancer", "Manage load balancers"),
+	}
+}
+
+type loadbalancerCommand struct {
+	*commands.BaseCommand
+}

internal/commands/util.go

@@ -84,6 +84,18 @@ func DatabaseStateColour(state upcloud.ManagedDatabaseState) text.Colors {
 	}
 }
 
+// LoadBalancerOperationalStateColour maps load balancer states to colours
+func LoadBalancerOperationalStateColour(state upcloud.LoadBalancerOperationalState) text.Colors {
+	switch state {
+	case upcloud.LoadBalancerOperationalStateRunning:
+		return text.Colors{text.FgGreen}
+	case upcloud.LoadBalancerOperationalStateCheckup, upcloud.LoadBalancerOperationalStatePending, upcloud.LoadBalancerOperationalStateSetupAgent, upcloud.LoadBalancerOperationalStateSetupDNS, upcloud.LoadBalancerOperationalStateSetupLB, upcloud.LoadBalancerOperationalStateSetupNetwork, upcloud.LoadBalancerOperationalStateSetupServer:
+		return text.Colors{text.FgYellow}
+	default:
+		return text.Colors{text.FgHiBlack}
+	}
+}
+
 // ServerStateColour is a helper mapping server states to colours
 func ServerStateColour(state string) text.Colors {
 	switch state {