Class: Gogetit::GogetLXD

Inherits:
Object
  • Object
show all
Includes:
Util
Defined in:
lib/providers/lxd.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from Util

#check_ip_available, #get_gateway, #knife_bootstrap, #knife_remove, #ping_available?, #recognize_env, #run_command, #run_through_ssh, #ssh_available?, #symbolize_keys, #update_databags, #wait_until_available

Constructor Details

#initialize(conf, maas, logger) ⇒ GogetLXD 



11
12
13
14
15
16
17
18
19
 
# File 'lib/providers/lxd.rb', line 11

def initialize(conf, maas, logger)
  @config = conf
  @conn = Hyperkit::Client.new(
      api_endpoint: config[:lxd][:nodes][0][:url],
      verify_ssl: false
    )
  @maas = maas
  @logger = logger
end

Instance Attribute Details

#config ⇒ Object (readonly)

Returns the value of attribute config.



9
10
11
 
# File 'lib/providers/lxd.rb', line 9

def config
  @config
end

#conn ⇒ Object (readonly)

Returns the value of attribute conn.



9
10
11
 
# File 'lib/providers/lxd.rb', line 9

def conn
  @conn
end

#logger ⇒ Object (readonly)

Returns the value of attribute logger.



9
10
11
 
# File 'lib/providers/lxd.rb', line 9

def logger
  @logger
end

#maas ⇒ Object (readonly)

Returns the value of attribute maas.



9
10
11
 
# File 'lib/providers/lxd.rb', line 9

def maas
  @maas
end

Instance Method Details

#container_exists?(name) ⇒ Boolean 



26
27
28
29
30
31
32
 
# File 'lib/providers/lxd.rb', line 26

def container_exists?(name)
  logger.info("Calling <#{__method__.to_s}>")
  list.each do |c|
    return true if c == name
  end
  false
end

#create(name, options = {}) ⇒ Object 



280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
 
# File 'lib/providers/lxd.rb', line 280

def create(name, options = {})
  logger.info("Calling <#{__method__.to_s}>")

  abort("Container #{name} already exists!") \
    if container_exists?(name)

  abort("Domain #{name}.#{maas.get_domain} already exists!") \
    if maas.domain_name_exists?(name) unless options['no-maas']

  args = generate_user_data(args, options)
  args = generate_network_config(args, options)
  args = generate_devices(args, options)

  if options['alias'].nil? or options['alias'].empty?
    args[:alias] = config[:lxd][:default_alias]
  else
    args[:alias] = options['alias']
  end

  args[:sync] ||= true

  conn.create_container(name, args)
  container = conn.container(name)

  container.devices = args[:devices].merge!(container.devices.to_hash)
  conn.update_container(name, container)
  # Fetch container object again
  container = conn.container(name)

  reserve_ips(name, options, container) \
    if options['vlans'] or options['ipaddresses'] \
      unless options['no-maas']

  conn.start_container(name, :sync=>"true")

  if options['no-maas']
    ip_or_fqdn = options['ip_to_access']
  else
    ip_or_fqdn = name + '.' + maas.get_domain
  end

  if conn.execute_command(name, "ls /etc/lsb-release")[:metadata][:return] == 0
    default_user = 'ubuntu'
  elsif conn.execute_command(name, "ls /etc/redhat-release")[:metadata][:return] == 0
    default_user = 'centos'
  else
    default_user = config[:default][:user]
  end

  wait_until_available(ip_or_fqdn, default_user, logger)
  logger.info("#{name} has been created.")

  if options['no-maas']
    puts "ssh #{default_user}@#{options['ip_to_access']}"
  else
    puts "ssh #{default_user}@#{name}"
  end

  true
end

#destroy(name, args = {}) ⇒ Object 



341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
 
# File 'lib/providers/lxd.rb', line 341

def destroy(name, args = {})
  logger.info("Calling <#{__method__.to_s}>")

  container = conn.container(name)
  args[:sync] ||= true

  if get_state(name) == 'Running'
    conn.stop_container(name, args)
  end

  wait_until_state(name, 'Stopped')

  if YAML.load(container[:config][:"user.user-data"])['maas']
    logger.info("This is a MAAS enabled container.")
    if container[:config][:"user.network-config"]
      net_conf = YAML.load(
        container[:config][:"user.network-config"]
      )['config']
      # To remove DNS configuration
      net_conf.shift

      net_conf.each do |nic|
        if nic['subnets'][0]['type'] == 'static'
          # It assumes we only assign a single subnet on a VLAN.
          # Subnets in a VLAN, VLANs in a Fabric
          ip = nic['subnets'][0]['address'].split('/')[0]

          if maas.ipaddresses_reserved?(ip)
            maas.ipaddresses('release', { 'ip' => ip })
          end
        end
      end
    end

    maas.delete_dns_record(name)
  end

  conn.delete_container(name, args)

  # When multiple static IPs were reserved, it will not delete anything
  # since they are deleted when releasing the IPs above.
  logger.info("#{name} has been destroyed.")
  true
end

#generate_devices(args, options) ⇒ Object

To configure devices



175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
 
# File 'lib/providers/lxd.rb', line 175

def generate_devices(args, options)
  args[:devices] = {}

  if options['no-maas']
    args[:devices] = \
      YAML.load_file(options['file'])[:devices]

  elsif options['ipaddresses']
    options[:ifaces].each_with_index do |iface,index|
      if index == 0
        if iface['vlan']['name'] == 'untagged' # or vid == 0
          args[:devices][:"eth#{index}"] = {
            mtu: iface['vlan']['mtu'].to_s,   #This must be string
            name: "eth#{index}",
            nictype: 'bridged',
            parent: config[:default][:root_bridge],
            type: 'nic'
          }
        elsif iface['vlan']['name'] != 'untagged' # or vid != 0
          args[:devices][:"eth#{index}"] = {
            mtu: iface['vlan']['mtu'].to_s,   #This must be string
            name: "eth#{index}",
            nictype: 'bridged',
            parent: config[:default][:root_bridge] + "-" + iface['vlan']['vid'].to_s,
            type: 'nic'
          }
        end
      # When options[:ifaces][0]['vlan']['name'] == 'untagged' and index > 0,
      # it does not need to generate more devices 
      # since it will configure the IPs with tagged VLANs.
      elsif options[:ifaces][0]['vlan']['name'] != 'untagged'
        args[:devices][:"eth#{index}"] = {
          mtu: iface['vlan']['mtu'].to_s,   #This must be string
          name: "eth#{index}",
          nictype: 'bridged',
          parent: config[:default][:root_bridge] + "-" + iface['vlan']['vid'].to_s,
          type: 'nic'
        }
      end
    end
  else
    abort("root_bridge #{config[:default][:root_bridge]} does not exist.") \
       unless conn.networks.include? config[:default][:root_bridge]

    root_bridge_mtu = nil
    # It assumes you only use one fabric as of now,
    # since there might be more fabrics with each untagged vlans on them,
    # which might make finding exact mtu fail as following process.
    default_fabric = 'fabric-0'

    maas.get_subnets.each do |subnet|
      if subnet['vlan']['name'] == 'untagged' and subnet['vlan']['fabric'] == default_fabric
        root_bridge_mtu = subnet['vlan']['mtu']
        break
      end
    end

    args[:devices] = {}
    args[:devices][:"eth0"] = {
      mtu: root_bridge_mtu.to_s,   #This must be string
      name: 'eth0',
      nictype: 'bridged',
      parent: config[:default][:root_bridge],
      type: 'nic'
    }
  end

  return args
end

#generate_network_config(args, options) ⇒ Object 



84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
 
# File 'lib/providers/lxd.rb', line 84

def generate_network_config(args, options)
  logger.info("Calling <#{__method__.to_s}>")
  if options['no-maas']
    args[:config][:'user.network-config'] = \
      YAML.load_file(options['file'])[:config][:'user.network-config']

    options['ip_to_access'] = \
      args[:config][:"user.network-config"]['config'][1]['subnets'][0]['address']
      .split('/')[0]
    args[:config][:"user.network-config"] = \
      YAML.dump(args[:config][:"user.network-config"])[4..-1]

  elsif options['ipaddresses']
    options[:ifaces] = check_ip_available(options['ipaddresses'], maas, logger)
    abort("There is no dns server specified for the gateway network.") \
      unless options[:ifaces][0]['dns_servers'][0]
    abort("There is no gateway specified for the gateway network.") \
      unless options[:ifaces][0]['gateway_ip']

    args[:config][:'user.network-config'] = {
      'version' => 1,
      'config' => [
        {
          'type' => 'nameserver',
          'address' => options[:ifaces][0]['dns_servers'][0]
        }
      ]
    }

    # to generate configuration for [:config][:'user.network-config']['config']
    options[:ifaces].each_with_index do |iface,index|
      if index == 0
        iface_conf = {
          'type' => 'physical',
          'name' => "eth#{index}",
          'subnets' => [
            {
              'type' => 'static',
              'ipv4' => true,
              'address' => iface['ip'] + '/' + iface['cidr'].split('/')[1],
              'gateway' => iface['gateway_ip'],
              'mtu' => iface['vlan']['mtu'],
              'control' => 'auto'
            }
          ]
        }
      elsif index > 0
        if options[:ifaces][0]['vlan']['name'] != 'untagged'
          iface_conf = {
            'type' => 'physical',
            'name' => "eth#{index}",
            'subnets' => [
              {
                'type' => 'static',
                'ipv4' => true,
                'address' => iface['ip'] + '/' + iface['cidr'].split('/')[1],
                'mtu' => iface['vlan']['mtu'],
                'control' => 'auto'
              }
            ]
          }
        elsif options[:ifaces][0]['vlan']['name'] == 'untagged'
          iface_conf = {
            'type' => 'vlan',
            'name' => "eth0.#{iface['vlan']['vid'].to_s}",
            'vlan_id' => iface['vlan']['vid'].to_s,
            'vlan_link' => 'eth0',
            'subnets' => [
              {
                'type' => 'static',
                'ipv4' => true,
                'address' => iface['ip'] + '/' + iface['cidr'].split('/')[1],
                'mtu' => iface['vlan']['mtu'],
                'control' => 'auto'
              }
            ]
          }
        end
      end

      args[:config][:'user.network-config']['config'].push(iface_conf)
    end

    args[:config][:"user.network-config"] = \
      YAML.dump(args[:config][:"user.network-config"])[4..-1]
  end

  return args
end

#generate_user_data(args, options) ⇒ Object

to generate ‘user.user-data’



46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
 
# File 'lib/providers/lxd.rb', line 46

def generate_user_data(args, options)
  logger.info("Calling <#{__method__.to_s}>")

  args = {}
  args[:config] = {}

  if options['no-maas']
    args[:config][:'user.user-data'] = \
      YAML.load_file(options['file'])[:config]['user.user-data']
  else
    sshkeys = maas.get_sshkeys
    pkg_repos = maas.get_package_repos

    args[:config][:'user.user-data'] = { 'ssh_authorized_keys' => [] }

    sshkeys.each do |key|
      args[:config][:'user.user-data']['ssh_authorized_keys'].push(key['key'])
    end

    pkg_repos.each do |repo|
      if repo['name'] == 'main_archive'
        args[:config][:'user.user-data']['apt_mirror'] = repo['url']
      end
    end

    args[:config][:"user.user-data"]['maas'] = true
  end

  # To disable to update apt database on first boot
  # so chef client can keep doing its job.
  args[:config][:'user.user-data']['package_update'] = false

  args[:config][:"user.user-data"] = \
    "#cloud-config\n" + YAML.dump(args[:config][:"user.user-data"])[4..-1]

  return args
end

#get_state(name) ⇒ Object 



34
35
36
 
# File 'lib/providers/lxd.rb', line 34

def get_state(name)
  conn.container(name)[:status]
end

#list ⇒ Object 



21
22
23
24
 
# File 'lib/providers/lxd.rb', line 21

def list
  logger.info("Calling <#{__method__.to_s}>")
  conn.containers
end

#reserve_ips(name, options, container) ⇒ Object 



245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
 
# File 'lib/providers/lxd.rb', line 245

def reserve_ips(name, options, container)
  # Generate params to reserve IPs
  options[:ifaces].each_with_index do |iface,index|
    if index == 0
      params = {
        'subnet' => iface['cidr'],
        'ip' => iface['ip'],
        'hostname' => name,
        'mac' => container[:expanded_config][:"volatile.eth#{index}.hwaddr"]
      }
    elsif index > 0
      # if dot, '.', is used as a conjunction instead of '-',
      # it fails ocuring '404 not found'.
      # if under score, '_', is used as a conjunction instead of '-',
      # it breaks MAAS DNS somehow..
      if options[:ifaces][0]['vlan']['name'] == 'untagged'
        params = {
          'subnet' => iface['cidr'],
          'ip' => iface['ip'],
          'hostname' => 'eth0' + '-' + iface['vlan']['vid'].to_s  + '-' + name,
          'mac' => container[:expanded_config][:"volatile.eth0.hwaddr"]
        }
      elsif options[:ifaces][0]['vlan']['name'] != 'untagged'
        params = {
          'subnet' => iface['cidr'],
          'ip' => iface['ip'],
          'hostname' => "eth#{index}" + '-' + name,
          'mac' => container[:expanded_config][:"volatile.eth#{index}.hwaddr"]
        }
      end
    end
    maas.ipaddresses('reserve', params)
  end
end

#wait_until_state(name, state) ⇒ Object 



38
39
40
41
42
43
 
# File 'lib/providers/lxd.rb', line 38

def wait_until_state(name, state)
  logger.info("Calling <#{__method__.to_s}> for being #{state}..")
  until get_state(name) == state
    sleep 3
  end
end