Merge pull request #554 from civo/support-volume-type

support specifying volume-type when creating volumes

Author: dippydocus

cmd/volume/volume.go

@@ -32,6 +32,7 @@ func init() {
 
 	volumeCreateCmd.Flags().IntVarP(&createSizeGB, "size-gb", "s", 0, "The new size in GB (required)")
 	volumeCreateCmd.Flags().StringVarP(&networkVolumeID, "network", "t", "default", "The network name/ID where the volume will be created")
+	volumeCreateCmd.Flags().StringVarP(&volumeType, "volume-type", "v", "", "The type of the volume that will be created (as provided by 'civo volumetypes ls')")
 	volumeCreateCmd.MarkFlagRequired("size-gb")
 
 	volumeResizeCmd.Flags().IntVarP(&newSizeGB, "size-gb", "s", 0, "The new size in GB (required)")

cmd/volume/volume_create.go

@@ -11,8 +11,11 @@ import (
 	"github.com/spf13/cobra"
 )
 
-var createSizeGB int
-var networkVolumeID string
+var (
+	createSizeGB    int
+	networkVolumeID string
+	volumeType      string
+)
 
 var volumeCreateCmd = &cobra.Command{
 	Use:     "create",
@@ -68,6 +71,10 @@ var volumeCreateCmd = &cobra.Command{
 			volumeConfig.NetworkID = network.ID
 		}
 
+		if volumeType != "" {
+			volumeConfig.VolumeType = volumeType
+		}
+
 		volume, err := client.NewVolume(volumeConfig)
 		if err != nil {
 			if err == civogo.QuotaLimitReachedError {