Add project machine-reservations command.

cmd/project-reservations.go

@@ -0,0 +1,198 @@
+package cmd
+
+import (
+	"errors"
+	"fmt"
+	"strings"
+
+	"github.com/fi-ts/cloud-go/api/client/project"
+	"github.com/fi-ts/cloud-go/api/models"
+	"github.com/metal-stack/metal-lib/pkg/genericcli"
+	"github.com/metal-stack/metal-lib/pkg/genericcli/printers"
+	"github.com/metal-stack/metal-lib/pkg/pointer"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+type machineReservationsCmd struct {
+	*config
+}
+
+func newMachineReservationsCmd(c *config) *cobra.Command {
+	w := machineReservationsCmd{
+		config: c,
+	}
+
+	cmdsConfig := &genericcli.CmdsConfig[*models.V1MachineReservationCreateRequest, *models.V1MachineReservationUpdateRequest, *models.V1MachineReservationResponse]{
+		BinaryName:  binaryName,
+		GenericCLI:  genericcli.NewGenericCLI(w).WithFS(c.fs),
+		Singular:    "machine-reservation",
+		Plural:      "machine-reservations",
+		Description: "manage machine reservations, ids must be provided in the form <project>@<size>",
+		// Sorter:          sorters.MachineReservationsSorter(),
+		// ValidArgsFn:     c.comp.MachineReservationsListCompletion,
+		DescribePrinter: func() printers.Printer { return c.describePrinter },
+		ListPrinter:     func() printers.Printer { return c.listPrinter },
+		ListCmdMutateFn: func(cmd *cobra.Command) {
+			cmd.Flags().String("project", "", "show projects of given id")
+			cmd.Flags().String("size", "", "show projects of given name")
+			cmd.Flags().String("tenant", "", "show projects of given name")
+			genericcli.Must(cmd.RegisterFlagCompletionFunc("tenant", c.comp.TenantListCompletion))
+			genericcli.Must(cmd.RegisterFlagCompletionFunc("project", c.comp.ProjectListCompletion))
+			genericcli.Must(cmd.RegisterFlagCompletionFunc("size", c.comp.SizeListCompletion))
+		},
+		UpdateCmdMutateFn: func(cmd *cobra.Command) {
+			cmd.Flags().Int32("amount", 0, "the amount of machines to reserve")
+			cmd.Flags().String("description", "", "the description of the reservation")
+			cmd.Flags().StringSlice("partitions", nil, "the partitions in which this reservation is being made")
+			genericcli.Must(cmd.RegisterFlagCompletionFunc("partitions", c.comp.PartitionListCompletion))
+		},
+		CreateCmdMutateFn: func(cmd *cobra.Command) {
+			cmd.Flags().String("project", "", "the project of the reservation")
+			cmd.Flags().String("size", "", "the size of the reservation")
+			cmd.Flags().Int32("amount", 0, "the amount of machines to reserve")
+			cmd.Flags().String("description", "", "the description of the reservation")
+			cmd.Flags().StringSlice("partitions", nil, "the partitions in which this reservation is being made")
+			genericcli.Must(cmd.RegisterFlagCompletionFunc("project", c.comp.ProjectListCompletion))
+			genericcli.Must(cmd.RegisterFlagCompletionFunc("size", c.comp.SizeListCompletion))
+			genericcli.Must(cmd.RegisterFlagCompletionFunc("partitions", c.comp.PartitionListCompletion))
+		},
+		CreateRequestFromCLI: func() (*models.V1MachineReservationCreateRequest, error) {
+			return &models.V1MachineReservationCreateRequest{
+				Amount:       pointer.PointerOrNil(viper.GetInt32("amount")),
+				Description:  pointer.PointerOrNil(viper.GetString("description")),
+				Partitionids: viper.GetStringSlice("partitions"),
+				Projectid:    pointer.PointerOrNil(viper.GetString("project")),
+				Sizeid:       pointer.PointerOrNil(viper.GetString("size")),
+			}, nil
+		},
+		UpdateRequestFromCLI: func(args []string) (*models.V1MachineReservationUpdateRequest, error) {
+			id, err := genericcli.GetExactlyOneArg(args)
+			if err != nil {
+				return nil, err
+			}
+
+			p, size, err := w.fromCompoundID(id)
+			if err != nil {
+				return nil, err
+			}
+
+			return &models.V1MachineReservationUpdateRequest{
+				Amount:       pointer.PointerOrNil(viper.GetInt32("amount")),
+				Description:  pointer.PointerOrNil(viper.GetString("description")),
+				Partitionids: viper.GetStringSlice("partitions"),
+				Projectid:    &p,
+				Sizeid:       &size,
+			}, nil
+		},
+	}
+
+	return genericcli.NewCmds(cmdsConfig)
+}
+
+func (m machineReservationsCmd) toCompoundID(project, size string) string {
+	return fmt.Sprintf("%s@%s", project, size)
+}
+
+func (m machineReservationsCmd) fromCompoundID(id string) (project, size string, err error) {
+	project, size, ok := strings.Cut(id, "@")
+	if !ok {
+		return "", "", errors.New("reservation id must be in form <project>@<size>")
+	}
+	return project, size, nil
+}
+
+func (m machineReservationsCmd) Convert(r *models.V1MachineReservationResponse) (string, *models.V1MachineReservationCreateRequest, *models.V1MachineReservationUpdateRequest, error) {
+	id := m.toCompoundID(*r.Projectid, *r.Sizeid)
+	return id, toMachineReservationCreateRequest(r), toMachineReservationUpdateRequest(r), nil
+}
+
+func toMachineReservationCreateRequest(r *models.V1MachineReservationResponse) *models.V1MachineReservationCreateRequest {
+	return &models.V1MachineReservationCreateRequest{
+		Amount:       r.Amount,
+		Description:  &r.Description,
+		Partitionids: r.Partitionids,
+		Projectid:    r.Projectid,
+		Sizeid:       r.Sizeid,
+	}
+}
+
+func toMachineReservationUpdateRequest(r *models.V1MachineReservationResponse) *models.V1MachineReservationUpdateRequest {
+	return &models.V1MachineReservationUpdateRequest{
+		Amount:       r.Amount,
+		Description:  &r.Description,
+		Partitionids: r.Partitionids,
+		Projectid:    r.Projectid,
+		Sizeid:       r.Sizeid,
+	}
+}
+
+func (m machineReservationsCmd) Create(rq *models.V1MachineReservationCreateRequest) (*models.V1MachineReservationResponse, error) {
+	resp, err := m.cloud.Project.CreateMachineReservation(project.NewCreateMachineReservationParams().WithBody(rq), nil)
+	if err != nil {
+		return nil, err
+	}
+
+	return resp.Payload, nil
+}
+
+func (m machineReservationsCmd) Delete(id string) (*models.V1MachineReservationResponse, error) {
+	p, size, err := m.fromCompoundID(id)
+	if err != nil {
+		return nil, err
+	}
+
+	resp, err := m.cloud.Project.DeleteMachineReservation(project.NewDeleteMachineReservationParams().WithProject(p).WithSize(size), nil)
+	if err != nil {
+		return nil, err
+	}
+
+	return resp.Payload, nil
+}
+
+func (m machineReservationsCmd) Get(id string) (*models.V1MachineReservationResponse, error) {
+	p, size, err := m.fromCompoundID(id)
+	if err != nil {
+		return nil, err
+	}
+
+	all, err := m.List()
+	if err != nil {
+		return nil, fmt.Errorf("unable to fetch machine reservations: %w", err)
+	}
+
+	for _, rv := range all {
+		if p != pointer.SafeDeref(rv.Projectid) {
+			continue
+		}
+		if size != pointer.SafeDeref(rv.Sizeid) {
+			continue
+		}
+
+		return rv, nil
+	}
+
+	return nil, fmt.Errorf("no reservation found with size %q for project %q", size, p)
+}
+
+func (m machineReservationsCmd) List() ([]*models.V1MachineReservationResponse, error) {
+	resp, err := m.cloud.Project.ListMachineReservations(project.NewListMachineReservationsParams().WithBody(&models.V1MachineReservationFindRequest{
+		Projectid: pointer.PointerOrNil(viper.GetString("project")),
+		Sizeid:    pointer.PointerOrNil(viper.GetString("size")),
+		Tenant:    pointer.PointerOrNil(viper.GetString("tenant")),
+	}), nil)
+	if err != nil {
+		return nil, err
+	}
+
+	return resp.Payload, nil
+}
+
+func (m machineReservationsCmd) Update(rq *models.V1MachineReservationUpdateRequest) (*models.V1MachineReservationResponse, error) {
+	resp, err := m.cloud.Project.UpdateMachineReservation(project.NewUpdateMachineReservationParams().WithBody(rq), nil)
+	if err != nil {
+		return nil, err
+	}
+
+	return resp.Payload, nil
+}

cmd/project.go

@@ -104,6 +104,7 @@ func newProjectCmd(c *config) *cobra.Command {
 	projectCmd.AddCommand(projectListCmd)
 	projectCmd.AddCommand(projectApplyCmd)
 	projectCmd.AddCommand(projectEditCmd)
+	projectCmd.AddCommand(newMachineReservationsCmd(c))
 
 	return projectCmd
 }

go.mod

@@ -8,7 +8,7 @@ require (
 	github.com/dustin/go-humanize v1.0.1
 	github.com/fatih/color v1.17.0
 	github.com/fi-ts/accounting-go v0.10.0
-	github.com/fi-ts/cloud-go v0.28.0
+	github.com/fi-ts/cloud-go v0.28.1-0.20240917130058-a9bec42d1326
 	github.com/gardener/gardener v1.91.0
 	github.com/gardener/machine-controller-manager v0.53.1
 	github.com/go-openapi/runtime v0.28.0

go.sum

@@ -90,8 +90,8 @@ github.com/fatih/color v1.17.0 h1:GlRw1BRJxkpqUCBKzKOw098ed57fEsKeNjpTe3cSjK4=
 github.com/fatih/color v1.17.0/go.mod h1:YZ7TlrGPkiz6ku9fK3TLD/pl3CpsiFyu8N92HLgmosI=
 github.com/fi-ts/accounting-go v0.10.0 h1:vbPgTWq1iicyBWFRajX0bawZ1ADbhKGuJyNEtXjpr08=
 github.com/fi-ts/accounting-go v0.10.0/go.mod h1:ARKouuFYUV44xUKytAlczpzoti/S+o+PnXCN5BQA6nQ=
-github.com/fi-ts/cloud-go v0.28.0 h1:MAg5Vsac9XYxCCL73USnceLtry1pvJovcNjwvqrv0lU=
-github.com/fi-ts/cloud-go v0.28.0/go.mod h1:R7JMkC92eGvxkkMO1oP6lEevBH86DFiO9H9mo7YD5Sw=
+github.com/fi-ts/cloud-go v0.28.1-0.20240917130058-a9bec42d1326 h1:eujakn8xwpEM8pONokO/foDXmS2Bno2AQyxlSh+PgIg=
+github.com/fi-ts/cloud-go v0.28.1-0.20240917130058-a9bec42d1326/go.mod h1:R7JMkC92eGvxkkMO1oP6lEevBH86DFiO9H9mo7YD5Sw=
 github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=
 github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=
 github.com/fsnotify/fsnotify v1.7.0 h1:8JEhPFa5W2WU7YfeZzPNqzMP6Lwt7L2715Ggo0nosvA=