Merge pull request #56 from serverscom/implement-rbs-commands

Implement rbs commands

Author: kzaitsev

cmd/entities/rbs_volumes/add.go

@@ -0,0 +1,105 @@
+package rbsvolumes
+
+import (
+	serverscom "github.com/serverscom/serverscom-go-client/pkg"
+	"github.com/serverscom/srvctl/cmd/base"
+	"github.com/spf13/cobra"
+)
+
+type AddFlags struct {
+	Skeleton   bool
+	InputPath  string
+	Name       string
+	Size       int64
+	LocationID int
+	FlavorID   int
+	Labels     []string
+}
+
+func newAddCmd(cmdContext *base.CmdContext) *cobra.Command {
+	flags := &AddFlags{}
+
+	cmd := &cobra.Command{
+		Use:   "add",
+		Short: "Add a RBS volume",
+		Long:  "Add a new Remote Block Storage volume",
+		Args:  cobra.ExactArgs(0),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			formatter := cmdContext.GetOrCreateFormatter(cmd)
+
+			if flags.Skeleton {
+				return formatter.FormatSkeleton("rbs-volumes/add.json")
+			}
+
+			manager := cmdContext.GetManager()
+
+			ctx, cancel := base.SetupContext(cmd, manager)
+			defer cancel()
+
+			base.SetupProxy(cmd, manager)
+
+			input := &serverscom.RemoteBlockStorageVolumeCreateInput{}
+
+			if flags.InputPath != "" {
+				if err := base.ReadInputJSON(flags.InputPath, cmd.InOrStdin(), input); err != nil {
+					return err
+				}
+			} else {
+				required := []string{"name", "size", "flavor-id"}
+				if err := base.ValidateFlags(cmd, required); err != nil {
+					return err
+				}
+			}
+
+			if err := flags.FillInput(cmd, input); err != nil {
+				return err
+			}
+
+			scClient := cmdContext.GetClient().SetVerbose(manager.GetVerbose(cmd)).GetScClient()
+			volume, err := scClient.RemoteBlockStorageVolumes.Create(ctx, *input)
+			if err != nil {
+				return err
+			}
+
+			if volume != nil {
+				return formatter.Format(volume)
+			}
+			return nil
+		},
+	}
+
+	cmd.Flags().StringVarP(&flags.InputPath, "input", "i", "", "path to input file or '-' to read from stdin")
+	cmd.Flags().BoolVarP(&flags.Skeleton, "skeleton", "s", false, "JSON object with structure that is required to be passed")
+
+	cmd.Flags().StringVarP(&flags.Name, "name", "n", "", "name of the RBS volume")
+	cmd.Flags().Int64Var(&flags.Size, "size", 0, "size of the volume in GB")
+	cmd.Flags().IntVar(&flags.LocationID, "location-id", 0, "ID of the location")
+	cmd.Flags().IntVar(&flags.FlavorID, "flavor-id", 0, "ID of the flavor")
+	cmd.Flags().StringArrayVarP(&flags.Labels, "label", "l", []string{}, "string in key=value format")
+
+	return cmd
+}
+
+func (f *AddFlags) FillInput(cmd *cobra.Command, input *serverscom.RemoteBlockStorageVolumeCreateInput) error {
+	if cmd.Flags().Changed("name") {
+		input.Name = f.Name
+	}
+	if cmd.Flags().Changed("size") {
+		input.Size = f.Size
+	}
+	if cmd.Flags().Changed("location-id") {
+		input.LocationID = f.LocationID
+	}
+	if cmd.Flags().Changed("flavor-id") {
+		input.FlavorID = f.FlavorID
+	}
+	if cmd.Flags().Changed("label") {
+		labelsMap, err := base.ParseLabels(f.Labels)
+		if err != nil {
+			return err
+		}
+		input.Labels = labelsMap
+	}
+
+	return nil
+}

cmd/entities/rbs_volumes/delete.go

@@ -0,0 +1,30 @@
+package rbsvolumes
+
+import (
+	"github.com/serverscom/srvctl/cmd/base"
+	"github.com/spf13/cobra"
+)
+
+func newDeleteCmd(cmdContext *base.CmdContext) *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "delete <volume-id>",
+		Short: "Delete a RBS volume",
+		Long:  "Delete a Remote Block Storage volume by ID",
+		Args:  cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			manager := cmdContext.GetManager()
+
+			ctx, cancel := base.SetupContext(cmd, manager)
+			defer cancel()
+
+			base.SetupProxy(cmd, manager)
+
+			scClient := cmdContext.GetClient().SetVerbose(manager.GetVerbose(cmd)).GetScClient()
+
+			volumeID := args[0]
+			return scClient.RemoteBlockStorageVolumes.Delete(ctx, volumeID)
+		},
+	}
+
+	return cmd
+}

cmd/entities/rbs_volumes/get.go

@@ -0,0 +1,39 @@
+package rbsvolumes
+
+import (
+	"github.com/serverscom/srvctl/cmd/base"
+	"github.com/spf13/cobra"
+)
+
+func newGetCmd(cmdContext *base.CmdContext) *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "get <volume-id>",
+		Short: "Get a RBS volume",
+		Long:  "Get a Remote Block Storage volume by ID",
+		Args:  cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			manager := cmdContext.GetManager()
+
+			ctx, cancel := base.SetupContext(cmd, manager)
+			defer cancel()
+
+			base.SetupProxy(cmd, manager)
+
+			scClient := cmdContext.GetClient().SetVerbose(manager.GetVerbose(cmd)).GetScClient()
+
+			volumeID := args[0]
+			volume, err := scClient.RemoteBlockStorageVolumes.Get(ctx, volumeID)
+			if err != nil {
+				return err
+			}
+
+			if volume != nil {
+				formatter := cmdContext.GetOrCreateFormatter(cmd)
+				return formatter.Format(volume)
+			}
+			return nil
+		},
+	}
+
+	return cmd
+}

cmd/entities/rbs_volumes/get_credentials.go

@@ -0,0 +1,39 @@
+package rbsvolumes
+
+import (
+	"github.com/serverscom/srvctl/cmd/base"
+	"github.com/spf13/cobra"
+)
+
+func newGetCredentialsCmd(cmdContext *base.CmdContext) *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "get-credentials <volume-id>",
+		Short: "Get credentials for a RBS volume",
+		Long:  "Get credentials for a Remote Block Storage volume by ID",
+		Args:  cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			manager := cmdContext.GetManager()
+
+			ctx, cancel := base.SetupContext(cmd, manager)
+			defer cancel()
+
+			base.SetupProxy(cmd, manager)
+
+			scClient := cmdContext.GetClient().SetVerbose(manager.GetVerbose(cmd)).GetScClient()
+
+			volumeID := args[0]
+			credentials, err := scClient.RemoteBlockStorageVolumes.GetCredentials(ctx, volumeID)
+			if err != nil {
+				return err
+			}
+
+			if credentials != nil {
+				formatter := cmdContext.GetOrCreateFormatter(cmd)
+				return formatter.Format(credentials)
+			}
+			return nil
+		},
+	}
+
+	return cmd
+}

cmd/entities/rbs_volumes/list.go

@@ -0,0 +1,21 @@
+package rbsvolumes
+
+import (
+	serverscom "github.com/serverscom/serverscom-go-client/pkg"
+	"github.com/serverscom/srvctl/cmd/base"
+	"github.com/spf13/cobra"
+)
+
+func newListCmd(cmdContext *base.CmdContext) *cobra.Command {
+	factory := func(verbose bool, args ...string) serverscom.Collection[serverscom.RemoteBlockStorageVolume] {
+		scClient := cmdContext.GetClient().SetVerbose(verbose).GetScClient()
+		return scClient.RemoteBlockStorageVolumes.Collection()
+	}
+
+	opts := base.NewListOptions(
+		&base.BaseListOptions[serverscom.RemoteBlockStorageVolume]{},
+		&base.LabelSelectorOption[serverscom.RemoteBlockStorageVolume]{},
+	)
+
+	return base.NewListCmd("list", "RBS servers", factory, cmdContext, opts...)
+}

cmd/entities/rbs_volumes/rbs_volumes.go

@@ -0,0 +1,48 @@
+package rbsvolumes
+
+import (
+	"log"
+
+	serverscom "github.com/serverscom/serverscom-go-client/pkg"
+	"github.com/serverscom/srvctl/cmd/base"
+	"github.com/serverscom/srvctl/internal/output/entities"
+	"github.com/spf13/cobra"
+)
+
+func NewCmd(cmdContext *base.CmdContext) *cobra.Command {
+	rbsVolumeEntity, err := entities.Registry.GetEntityFromValue(serverscom.RemoteBlockStorageVolume{})
+	if err != nil {
+		log.Fatal(err)
+	}
+	rbsCredentialsEntity, err := entities.Registry.GetEntityFromValue(serverscom.RemoteBlockStorageVolumeCredentials{})
+	if err != nil {
+		log.Fatal(err)
+	}
+	entitiesMap := make(map[string]entities.EntityInterface)
+	entitiesMap["rbs"] = rbsVolumeEntity
+	entitiesMap["rbs-credentials"] = rbsCredentialsEntity
+	cmd := &cobra.Command{
+		Use:   "rbs",
+		Short: "Manage RBS servers",
+		PersistentPreRunE: base.CombinePreRunE(
+			base.CheckFormatterFlags(cmdContext, entitiesMap),
+			base.CheckEmptyContexts(cmdContext),
+		),
+		Args: base.NoArgs,
+		Run:  base.UsageRun,
+	}
+
+	cmd.AddCommand(
+		newListCmd(cmdContext),
+		newAddCmd(cmdContext),
+		newGetCmd(cmdContext),
+		newUpdateCmd(cmdContext),
+		newDeleteCmd(cmdContext),
+		newGetCredentialsCmd(cmdContext),
+		newResetCredentialsCmd(cmdContext),
+	)
+
+	base.AddFormatFlags(cmd)
+
+	return cmd
+}