Add security groups support

Author: GennadySpb

README.md

@@ -39,12 +39,13 @@ $ docker-machine create \
 | ``--yandex-platform-id`` or ``$YC_PLATFORM_ID``         | ID of the hardware platform configuration         | standard-v1              |
 | ``--yandex-preemptible`` or ``$YC_PREEMPTIBLE``         | Yandex.Cloud Instance preemptibility flag         | false                    |
 | ``--yandex-sa-key-file`` or ``$YC_SA_KEY_FILE``         | Yandex.Cloud Service Account key file             |                          |
+| ``--yandex-security-groups`` or ``$YC_SECURITY_GROUPS`` | Set security groups                               |                          |
 | ``--yandex-ssh-port`` or ``$YC_SSH_PORT``               | SSH port                                          | 22                       |
 | ``--yandex-ssh-user`` or ``$YC_SSH_USER``               | SSH username                                      | yc-user                  |
+| ``--yandex-static-address`` or ``$YC_STATIC_ADDRESS``   | Set public static IPv4 address                    |                          |
 | ``--yandex-subnet-id`` or ``$YC_SUBNET_ID``             | Subnet ID                                         |                          |
 | ``--yandex-token`` or ``$YC_TOKEN``                     | Yandex.Cloud OAuth token or IAM token             |                          |
 | ``--yandex-use-internal-ip`` or ``$YC_USE_INTERNAL_IP`` | Use the internal Instance IP to communicate       | false                    |
 | ``--yandex-userdata`` or ``$YC_USERDATA``               | Path to file with cloud-init user-data            |                          |
 | ``--yandex-zone`` or ``$YC_ZONE``                       | Yandex.Cloud zone                                 | ru-central1-a            |
-| ``--yandex-static-address`` or ``$YC_STATIC_ADDRESS``   | Set public static IPv4 address                                |                          |
 ---

driver/client.go

@@ -112,6 +112,7 @@ func prepareInstanceCreateRequest(d *Driver, imageID string) *compute.CreateInst
 			{
 				SubnetId:             d.SubnetID,
 				PrimaryV4AddressSpec: &compute.PrimaryAddressSpec{},
+				SecurityGroupIds:     d.SecurityGroups,
 			},
 		},
 		SchedulingPolicy: &compute.SchedulingPolicy{

driver/driver.go

@@ -28,29 +28,30 @@ type Driver struct {
 	ServiceAccountKeyFile string
 	Token                 string
 
-	CloudID       string
-	Cores         int
-	CoreFraction  int
-	DiskSize      int
-	DiskType      string
-	FolderID      string
-	ImageFamily   string
-	ImageFolderID string
-	ImageID       string
-	InstanceID    string
-	Labels        []string
-	Memory        int
-	Metadata      map[string]string
-	Nat           bool
-	PlatformID    string
-	Preemptible   bool
-	SSHUser       string
-	SubnetID      string
-	UseIPv6       bool
-	UseInternalIP bool
-	UserDataFile  string
-	Zone          string
-	StaticAddress string
+	CloudID        string
+	Cores          int
+	CoreFraction   int
+	DiskSize       int
+	DiskType       string
+	FolderID       string
+	ImageFamily    string
+	ImageFolderID  string
+	ImageID        string
+	InstanceID     string
+	Labels         []string
+	Memory         int
+	Metadata       map[string]string
+	Nat            bool
+	PlatformID     string
+	Preemptible    bool
+	SSHUser        string
+	SubnetID       string
+	UseIPv6        bool
+	UseInternalIP  bool
+	UserDataFile   string
+	Zone           string
+	StaticAddress  string
+	SecurityGroups []string
 }
 
 const (
@@ -223,6 +224,11 @@ func (d *Driver) GetCreateFlags() []mcnflag.Flag {
 			Usage:  "Set static address",
 			Value:  "",
 		},
+		mcnflag.StringSliceFlag{
+			EnvVar: "YC_SECURITY_GROUPS",
+			Name:   "yandex-security-groups",
+			Usage:  "Set security groups",
+		},
 	}
 }
 
@@ -260,6 +266,7 @@ func (d *Driver) SetConfigFromFlags(flags drivers.DriverOptions) error {
 	d.UserDataFile = flags.String("yandex-userdata")
 	d.Zone = flags.String("yandex-zone")
 	d.StaticAddress = flags.String("yandex-static-address")
+	d.SecurityGroups = flags.StringSlice("yandex-security-groups")
 
 	return nil
 }