@@ -41,6 +41,7 @@ type Driver struct {
4141 volumes []string
4242 networks []string
4343 UsePrivateNetwork bool
44+ firewalls []string
4445 cachedServer * hcloud.Server
4546 serverLabels map [string ]string
4647
@@ -64,6 +65,7 @@ const (
6465 flagVolumes = "hetzner-volumes"
6566 flagNetworks = "hetzner-networks"
6667 flagUsePrivateNetwork = "hetzner-use-private-network"
68+ flagFirewalls = "hetzner-firewalls"
6769 flagAdditionalKeys = "hetzner-additional-key"
6870 flagServerLabel = "hetzner-server-label"
6971)
@@ -151,6 +153,12 @@ func (d *Driver) GetCreateFlags() []mcnflag.Flag {
151153 Name : flagUsePrivateNetwork ,
152154 Usage : "Use private network" ,
153155 },
156+ mcnflag.StringSliceFlag {
157+ EnvVar : "HETZNER_FIREWALLS" ,
158+ Name : flagFirewalls ,
159+ Usage : "Firewall IDs or names which should be applied on the server" ,
160+ Value : []string {},
161+ },
154162 mcnflag.StringSliceFlag {
155163 EnvVar : "HETZNER_ADDITIONAL_KEYS" ,
156164 Name : flagAdditionalKeys ,
@@ -179,6 +187,7 @@ func (d *Driver) SetConfigFromFlags(opts drivers.DriverOptions) error {
179187 d .volumes = opts .StringSlice (flagVolumes )
180188 d .networks = opts .StringSlice (flagNetworks )
181189 d .UsePrivateNetwork = opts .Bool (flagUsePrivateNetwork )
190+ d .firewalls = opts .StringSlice (flagFirewalls )
182191 d .additionalKeys = opts .StringSlice (flagAdditionalKeys )
183192
184193 err := d .setLabelsFromFlags (opts )
@@ -327,6 +336,7 @@ func (d *Driver) Create() error {
327336 UserData : d .userData ,
328337 Labels : d .serverLabels ,
329338 }
339+
330340 networks := []* hcloud.Network {}
331341 for _ , networkIDorName := range d .networks {
332342 network , _ , err := d .getClient ().Network .Get (context .Background (), networkIDorName )
@@ -340,6 +350,19 @@ func (d *Driver) Create() error {
340350 }
341351 srvopts .Networks = networks
342352
353+ firewalls := []* hcloud.ServerCreateFirewall {}
354+ for _ , firewallIDorName := range d .firewalls {
355+ firewall , _ , err := d .getClient ().Firewall .Get (context .Background (), firewallIDorName )
356+ if err != nil {
357+ return errors .Wrap (err , "could not get firewall by ID or name" )
358+ }
359+ if firewall == nil {
360+ return errors .Errorf ("firewall '%s' not found" , firewallIDorName )
361+ }
362+ firewalls = append (firewalls , & hcloud.ServerCreateFirewall {Firewall : * firewall })
363+ }
364+ srvopts .Firewalls = firewalls
365+
343366 volumes := []* hcloud.Volume {}
344367 for _ , volumeIDorName := range d .volumes {
345368 volume , _ , err := d .getClient ().Volume .Get (context .Background (), volumeIDorName )
0 commit comments