Class: DevLXC::Cluster

Inherits:
Object
  • Object
show all
Defined in:
lib/dev-lxc/cluster.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(cluster_config) ⇒ Cluster

Returns a new instance of Cluster.



9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
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
83
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
173
174
175
176
177
178
179
180
181
182
183
184
185
186
# File 'lib/dev-lxc/cluster.rb', line 9

def initialize(cluster_config)
  FileUtils.mkdir_p('/var/dev-lxc') unless Dir.exist?('/var/dev-lxc')

  @config = Hash.new { |hash, key| hash[key] = {} }
  @server_configs = Hash.new

  %w(adhoc analytics automate build-nodes chef-backend chef-server compliance nodes runners supermarket).each do |server_type|
    if cluster_config[server_type]
      if cluster_config[server_type]["servers"]
        cluster_config[server_type]["servers"].each do |server_name, server_config|
          server_config ||= Hash.new

          products = server_config['products']
          products ||= Hash.new

          mounts = ["/var/dev-lxc var/dev-lxc"]
          if cluster_config[server_type]["mounts"]
            mounts.concat(cluster_config[server_type]["mounts"])
          elsif cluster_config["mounts"]
            mounts.concat(cluster_config["mounts"])
          end
          mounts = ["/var/dev-lxc var/dev-lxc"] + server_config["mounts"] if server_config["mounts"]

          ssh_keys = cluster_config[server_type]["ssh-keys"]
          ssh_keys ||= cluster_config["ssh-keys"]
          ssh_keys = server_config["ssh-keys"] if server_config["ssh-keys"]

          base_container_name = cluster_config[server_type]["base_container"]
          base_container_name ||= cluster_config["base_container"]
          base_container_name = server_config["base_container"] if server_config["base_container"]

          @server_configs[server_name] = {
            server_type: server_type,
            products: products,
            ipaddress: server_config['ipaddress'],
            additional_fqdn: nil,
            mounts: mounts,
            ssh_keys: ssh_keys,
            base_container_name: base_container_name
          }
          # gather configuration from only the first "automate", "compliance" or "supermarket" server
          break if %w(automate compliance supermarket).include?(server_type)
        end
      end

      case server_type
      when "analytics"
        @config[server_type][:topology] = cluster_config[server_type]["topology"]
        @config[server_type][:topology] ||= 'standalone'
        @config[server_type][:fqdn] = cluster_config[server_type]["analytics_fqdn"]
        @config[server_type][:frontends] = Array.new

        if cluster_config[server_type]["servers"]
          cluster_config[server_type]["servers"].each do |server_name, server_config|
            server_config ||= Hash.new
            additional_fqdn = nil
            case @config[server_type][:topology]
            when 'standalone'
              @config[server_type][:bootstrap_backend] = server_name if server_config["role"].nil?
              @config[server_type][:fqdn] ||= @config[server_type][:bootstrap_backend]
            when 'tier'
              @config[server_type][:bootstrap_backend] = server_name if server_config["role"] == "backend" && server_config["bootstrap"] == true
              if server_config["role"] == "frontend"
                additional_fqdn = @config[server_type][:fqdn]
                @config[server_type][:frontends] << server_name
              end
            end
            @server_configs[server_name].merge!({
              additional_fqdn: additional_fqdn
            })
          end
        end
      when "chef-backend"
        @config[server_type][:fqdn] = cluster_config[server_type]["api_fqdn"]
        @config[server_type][:users] = cluster_config[server_type]["users"]
        @config[server_type][:users] ||= Array.new
        @config[server_type][:orgs] = cluster_config[server_type]["orgs"]
        @config[server_type][:orgs] ||= Hash.new
        @config[server_type][:backends] = Array.new
        @config[server_type][:frontends] = Array.new

        servers = cluster_config[server_type]["servers"]
        if servers
          @config[server_type][:leader_backend] = servers.select { |s,sc| sc['role'] == 'backend' && sc['leader'] == true }.keys.first
          @config[server_type][:bootstrap_frontend] = servers.select { |s,sc| sc['role'] == 'frontend' && sc['bootstrap'] == true }.keys.first
          @config[server_type][:backends] << @config[server_type][:leader_backend]
          @config[server_type][:frontends] << @config[server_type][:bootstrap_frontend]
          servers.each do |server_name, server_config|
            server_config ||= Hash.new
            additional_fqdn = nil
            case server_config["role"]
            when "backend"
              @config[server_type][:backends] << server_name unless server_name == @config[server_type][:leader_backend]
            when "frontend"
              additional_fqdn = @config[server_type][:fqdn]
              @config[server_type][:frontends] << server_name unless server_name == @config[server_type][:bootstrap_frontend]
            end
            @server_configs[server_name].merge!({
              additional_fqdn: additional_fqdn,
              chef_server_type: 'chef-server'
            })
          end
        end
      when "chef-server"
        @config[server_type][:topology] = cluster_config[server_type]["topology"]
        @config[server_type][:topology] ||= 'standalone'
        @config[server_type][:fqdn] = cluster_config[server_type]["api_fqdn"]
        @config[server_type][:users] = cluster_config[server_type]["users"]
        @config[server_type][:users] ||= Array.new
        @config[server_type][:orgs] = cluster_config[server_type]["orgs"]
        @config[server_type][:orgs] ||= Hash.new
        @config[server_type][:frontends] = Array.new

        if cluster_config[server_type]["servers"]
          cluster_config[server_type]["servers"].each do |server_name, server_config|
            server_config ||= Hash.new
            additional_fqdn = nil
            chef_server_type = 'private-chef' if @server_configs[server_name][:products].has_key?('private-chef')
            chef_server_type = 'chef-server' if @server_configs[server_name][:products].has_key?('chef-server')
            case @config[server_type][:topology]
            when 'standalone'
              @config[server_type][:bootstrap_backend] = server_name if server_config["role"].nil?
              @config[server_type][:fqdn] ||= @config[server_type][:bootstrap_backend]
            when 'tier'
              @config[server_type][:bootstrap_backend] = server_name if server_config["role"] == "backend" && server_config["bootstrap"] == true
              if server_config["role"] == "frontend"
                additional_fqdn = @config[server_type][:fqdn]
                @config[server_type][:frontends] << server_name
              end
            end
            @server_configs[server_name].merge!({
              additional_fqdn: additional_fqdn,
              chef_server_type: chef_server_type
            })
          end
        end
      when "supermarket"
        unless cluster_config[server_type]["servers"].first.nil?
          (server_name, server_config) = cluster_config[server_type]["servers"].first
          @config[server_type][:fqdn] = server_name
        end
      when "compliance"
        unless cluster_config[server_type]["servers"].first.nil?
          (server_name, server_config) = cluster_config[server_type]["servers"].first
          @server_configs[server_name][:admin_user] = cluster_config[server_type]["admin_user"]
        end
      when "automate"
        unless cluster_config[server_type]["servers"].first.nil?
          (server_name, server_config) = cluster_config[server_type]["servers"].first
          server_config ||= Hash.new
          @server_configs[server_name].merge!({
            license_path: server_config['license_path'],
            chef_org: server_config['chef_org'],
            enterprise_name: server_config['enterprise_name']
          })
        end
      when "nodes"
        chef_server_url = cluster_config[server_type]['chef_server_url']
        validation_client_name = cluster_config[server_type]['validation_client_name']
        validation_key = cluster_config[server_type]['validation_key']
        if cluster_config[server_type]["servers"]
          cluster_config[server_type]["servers"].each do |server_name, server_config|
            server_config ||= Hash.new
            chef_server_url = server_config['chef_server_url'] if server_config['chef_server_url']
            validation_client_name = server_config['validation_client_name'] if server_config['validation_client_name']
            validation_key = server_config['validation_key'] if server_config['validation_key']
            @server_configs[server_name].merge!({
              chef_server_url: chef_server_url,
              validation_client_name: validation_client_name,
              validation_key: validation_key
            })
          end
        end
      end
    end
  end
  validate_cluster_config
end

Instance Attribute Details

#configObject (readonly)

Returns the value of attribute config.



7
8
9
# File 'lib/dev-lxc/cluster.rb', line 7

def config
  @config
end

Instance Method Details

#abort_up(servers) ⇒ Object



378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
# File 'lib/dev-lxc/cluster.rb', line 378

def abort_up(servers)
  abort_up = false
  servers.each do |server|
    next if server.container.defined?
    if (@config['chef-server'][:frontends] && @config['chef-server'][:frontends].include?(server.name)) || server.name == @config['analytics'][:bootstrap_backend]
      if @config['chef-server'][:bootstrap_backend].nil?
        puts "ERROR: '#{server.name}' requires a Chef Server bootstrap backend to be configured first."
        abort_up = true
      elsif !get_server(@config['chef-server'][:bootstrap_backend]).container.defined? && servers.select { |s| s.name == @config['chef-server'][:bootstrap_backend] }.empty?
        puts "ERROR: '#{server.name}' requires '#{@config['chef-server'][:bootstrap_backend]}' to be configured first."
        abort_up = true
      end
    end
    if @config['chef-server'][:bootstrap_backend] && @server_configs[server.name][:server_type] == 'supermarket'
      if !get_server(@config['chef-server'][:bootstrap_backend]).container.defined? && servers.select { |s| s.name == @config['chef-server'][:bootstrap_backend] }.empty?
        puts "ERROR: '#{server.name}' requires '#{@config['chef-server'][:bootstrap_backend]}' to be configured first."
        abort_up = true
      end
    end
    if @config['analytics'][:frontends] && @config['analytics'][:frontends].include?(server.name)
      if @config['analytics'][:bootstrap_backend].nil?
        puts "ERROR: '#{server.name}' requires an Analytics Server bootstrap backend to be configured first."
        abort_up = true
      elsif !get_server(@config['analytics'][:bootstrap_backend]).container.defined? && servers.select { |s| s.name == @config['analytics'][:bootstrap_backend] }.empty?
        puts "ERROR: '#{server.name}' requires '#{@config['analytics'][:bootstrap_backend]}' to be configured first."
        abort_up = true
      end
    end
    if @config['chef-backend'][:frontends] && @config['chef-backend'][:frontends].include?(server.name) && server.name != @config['chef-backend'][:bootstrap_frontend]
      if @config['chef-backend'][:bootstrap_frontend].nil?
        puts "ERROR: '#{server.name}' requires a Chef Server bootstrap frontend to be configured first."
        abort_up = true
      elsif !get_server(@config['chef-backend'][:bootstrap_frontend]).container.defined? && servers.select { |s| s.name == @config['chef-backend'][:bootstrap_frontend] }.empty?
        puts "ERROR: '#{server.name}' requires '#{@config['chef-backend'][:bootstrap_frontend]}' to be configured first."
        abort_up = true
      end
    end
    if server.name == @config['chef-backend'][:bootstrap_frontend]
      if (@config['chef-backend'][:backends].select { |s| get_server(s).container.running? }.length + servers.select { |s| @config['chef-backend'][:backends].include?(s.name) }.length) < 2
        puts "ERROR: '#{server.name}' requires at least two nodes in the backend cluster to be running first."
        abort_up = true
      end
    end
    if @config['chef-backend'][:backends] && @config['chef-backend'][:backends].include?(server.name) && server.name != @config['chef-backend'][:leader_backend]
      if !get_server(@config['chef-backend'][:leader_backend]).container.running? && servers.select { |s| s.name == @config['chef-backend'][:leader_backend] }.empty?
        puts "ERROR: '#{server.name}' requires '#{@config['chef-backend'][:leader_backend]}' to be running first."
        abort_up = true
      end
    end
    if @server_configs[server.name][:server_type] == 'nodes'
      if @server_configs[server.name][:chef_server_url] && @server_configs[server.name][:validation_client_name] && @server_configs[server.name][:validation_key].nil?
        if @config['chef-server'][:bootstrap_backend] && !get_server(@config['chef-server'][:bootstrap_backend]).container.defined? && servers.select { |s| s.name == @config['chef-server'][:bootstrap_backend] }.empty?
          puts "ERROR: '#{server.name}' requires '#{@config['chef-server'][:bootstrap_backend]}' to be configured first."
          abort_up = true
        elsif @config['chef-backend'][:bootstrap_frontend] && !get_server(@config['chef-backend'][:bootstrap_frontend]).container.defined? && servers.select { |s| s.name == @config['chef-backend'][:bootstrap_frontend] }.empty?
          puts "ERROR: '#{server.name}' requires '#{@config['chef-backend'][:bootstrap_frontend]}' to be configured first."
          abort_up = true
        end
      end
    end
    if @server_configs[server.name][:server_type] == 'automate'
      if @config['chef-server'][:bootstrap_backend].nil?
        puts "ERROR: '#{server.name}' requires a Chef Server bootstrap backend to be configured first."
        abort_up = true
      elsif !get_server(@config['chef-server'][:bootstrap_backend]).container.defined? && servers.select { |s| s.name == @config['chef-server'][:bootstrap_backend] }.empty?
        puts "ERROR: '#{server.name}' requires '#{@config['chef-server'][:bootstrap_backend]}' to be configured first."
        abort_up = true
      end
    end
    if %w(build-nodes runners).include?(@server_configs[server.name][:server_type])
      if @config['chef-server'][:bootstrap_backend].nil?
        puts "ERROR: '#{server.name}' requires a Chef Server bootstrap backend to be configured first."
        abort_up = true
      elsif !get_server(@config['chef-server'][:bootstrap_backend]).container.running? && servers.select { |s| s.name == @config['chef-server'][:bootstrap_backend] }.empty?
        puts "ERROR: '#{server.name}' requires '#{@config['chef-server'][:bootstrap_backend]}' to be running first."
        abort_up = true
      end
      if @config['chef-server'][:topology] == 'tier'
        if @config[server_type][:frontends].empty?
          puts "ERROR: '#{server.name}' requires at least one Chef Server frontend to be configured first."
          abort_up = true
        elsif (@config['chef-server'][:frontends].select { |s| get_server(s).container.running? }.length + servers.select { |s| @config['chef-server'][:frontends].include?(s.name) }.length) < 1
          puts "ERROR: '#{server.name}' requires at least one Chef Server frontend to be running first."
          abort_up = true
        end
      end
      automate_server_name = @server_configs.select {|name, config| config[:server_type] == 'automate'}.keys.first
      if automate_server_name
        if !get_server(automate_server_name).container.running? && servers.select { |s| s.name == automate_server_name }.empty?
          puts "ERROR: '#{server.name}' requires '#{automate_server_name}' to be running first."
          abort_up = true
        end
      else
        puts "ERROR: '#{server.name}' requires an Automate Server to be configured first."
        abort_up = true
      end
    end
  end
  return abort_up
end

#analytics_configObject



1117
1118
1119
1120
1121
1122
1123
1124
1125
1126
1127
1128
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
1139
1140
# File 'lib/dev-lxc/cluster.rb', line 1117

def analytics_config
  analytics_config = %Q(analytics_fqdn "#{@config['analytics'][:fqdn]}"
topology "#{@config['analytics'][:topology]}"
)
  if @config['analytics'][:topology] == 'tier'
    analytics_config += %Q(
server "#{@config['analytics'][:bootstrap_backend]}",
  :ipaddress => "#{@server_configs[@config['analytics'][:bootstrap_backend]][:ipaddress]}",
  :role => "backend",
  :bootstrap => true

backend_vip "#{@config['analytics'][:bootstrap_backend]}",
  :ipaddress => "#{@server_configs[@config['analytics'][:bootstrap_backend]][:ipaddress]}"
)
    @config['analytics'][:frontends].each do |frontend_name|
      analytics_config += %Q(
server "#{frontend_name}",
  :ipaddress => "#{@server_configs[frontend_name][:ipaddress]}",
  :role => "frontend"
)
    end
  end
  return analytics_config
end

#calculate_required_products(servers, force = false) ⇒ Object



537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
# File 'lib/dev-lxc/cluster.rb', line 537

def calculate_required_products(servers, force=false)
  all_required_products = Hash.new
  servers.each do |server|
    products = @server_configs[server.name][:products]
    @server_configs[server.name][:required_products] = Hash.new
    if !force && !server.snapshot_list.select { |sn| sn[2].to_s.start_with?("dev-lxc build: products installed") }.empty?
      # Skipping product cache preparation for container because it has a 'products installed' snapshot
      next
    elsif !force && !server.snapshot_list.select { |sn| sn[2].to_s.start_with?("dev-lxc build: completed") }.empty?
      # Skipping product cache preparation for container because it has a 'build: completed' snapshot
      next
    end
    products.each do |product_name, product_options|
      if product_options && product_options['package_source']
        package_source = product_options['package_source']
        all_required_products[package_source] = [product_name, nil]
        @server_configs[server.name][:required_products][product_name] = package_source
      else
        package_source, package_sha256 = get_product_download_info(server, product_name, product_options)
        all_required_products[package_source] = [product_name, package_sha256]
        product_cache_path = "/var/dev-lxc/cache/chef-products/#{product_name}/#{File.basename(package_source)}"
        @server_configs[server.name][:required_products][product_name] = product_cache_path
      end
    end
  end
  all_required_products
end

#chef_repo(force = false, pivotal = false) ⇒ Object



1025
1026
1027
1028
1029
1030
1031
1032
1033
1034
1035
1036
1037
1038
1039
1040
1041
1042
1043
1044
1045
1046
1047
1048
1049
1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
1064
# File 'lib/dev-lxc/cluster.rb', line 1025

def chef_repo(force=false, pivotal=false)
  chef_server_dot_chef_path = "/root/chef-repo/.chef"
  dot_chef_path = "./chef-repo/.chef"

  if @config['chef-server'][:bootstrap_backend]
    chef_server = get_server(@config['chef-server'][:bootstrap_backend])
    chef_server_fqdn = @config['chef-server'][:fqdn]
  elsif @config['chef-backend'][:bootstrap_frontend]
    chef_server = get_server(@config['chef-backend'][:bootstrap_frontend])
    chef_server_fqdn = @config['chef-backend'][:fqdn]
  else
    puts "ERROR: A Chef Server is not defined in the cluster's config. Please define it first."
    exit 1
  end
  unless chef_server.container.defined?
    puts "ERROR: The '#{chef_server.name}' Chef Server does not exist."
    exit 1
  end

  puts "Creating chef-repo with pem files and knife.rb in the current directory"
  FileUtils.mkdir_p(dot_chef_path)

  pem_files = Dir.glob("#{chef_server.container.config_item('lxc.rootfs')}#{chef_server_dot_chef_path}/*.pem")
  pem_files.delete_if { |pem_file| pem_file.end_with?("/pivotal.pem") } unless pivotal
  FileUtils.cp(pem_files, dot_chef_path) unless pem_files.empty?

  if pivotal
    if File.exists?("#{dot_chef_path}/pivotal.rb") && ! force
      puts "Skipping pivotal.rb because it already exists in `#{dot_chef_path}`"
    else
      create_pivotal_knife_config(chef_server_fqdn, dot_chef_path)
    end
  end

  if File.exists?("./chef-repo/.chef/knife.rb") && ! force
    puts "Skipping knife.rb because it already exists in `#{dot_chef_path}`"
  else
    create_knife_config(chef_server_fqdn, dot_chef_path)
  end
end

#chef_server_configObject



1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
1080
1081
1082
1083
1084
1085
1086
1087
1088
1089
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101
1102
1103
1104
1105
1106
1107
1108
1109
1110
1111
1112
1113
1114
1115
# File 'lib/dev-lxc/cluster.rb', line 1066

def chef_server_config
  chef_server_config = %Q(api_fqdn "#{@config['chef-server'][:fqdn]}"\n)
  if @config['chef-server'][:topology] == 'tier'
    chef_server_config += %Q(
topology "#{@config['chef-server'][:topology]}"

server "#{@config['chef-server'][:bootstrap_backend]}",
  :ipaddress => "#{@server_configs[@config['chef-server'][:bootstrap_backend]][:ipaddress]}",
  :role => "backend",
  :bootstrap => true

backend_vip "#{@config['chef-server'][:bootstrap_backend]}",
  :ipaddress => "#{@server_configs[@config['chef-server'][:bootstrap_backend]][:ipaddress]}"
)
    @config['chef-server'][:frontends].each do |frontend_name|
      chef_server_config += %Q(
server "#{frontend_name}",
  :ipaddress => "#{@server_configs[frontend_name][:ipaddress]}",
  :role => "frontend"
)
    end
  end
  if @config['analytics'][:fqdn]
    chef_server_config += %Q(
oc_id['applications'] ||= {}
oc_id['applications']['analytics'] = {
  'redirect_uri' => 'https://#{@config['analytics'][:fqdn]}/'
}
rabbitmq['vip'] = '#{@config['chef-server'][:bootstrap_backend]}'
rabbitmq['node_ip_address'] = '0.0.0.0'
)
  end
  if @config['supermarket'][:fqdn]
    chef_server_config += %Q(
oc_id['applications'] ||= {}
oc_id['applications']['supermarket'] = {
  'redirect_uri' => 'https://#{@config['supermarket'][:fqdn]}/auth/chef_oauth2/callback'
}
)
  end
  automate_server_name = @server_configs.select {|name, config| config[:server_type] == 'automate'}.keys.first
  if automate_server_name
    chef_server_config += %Q(
data_collector['root_url'] = "https://#{automate_server_name}/data-collector/v0/"
data_collector['token'] = "93a49a4f2482c64126f7b6015e6b0f30284287ee4054ff8807fb63d9cbd1c506"
profiles['root_url'] = "https://#{automate_server_name}"
)
  end
  return chef_server_config
end

#clone_from_base_container(server) ⇒ Object



493
494
495
496
497
498
499
500
501
# File 'lib/dev-lxc/cluster.rb', line 493

def clone_from_base_container(server)
  server_type = @server_configs[server.name][:server_type]
  base_container = DevLXC::Container.new(@server_configs[server.name][:base_container_name])
  puts "Cloning base container '#{base_container.name}' into container '#{server.name}'"
  base_container.clone(server.name, {:flags => LXC::LXC_CLONE_SNAPSHOT})
  server.container.load_config
  puts "Deleting SSH Server Host Keys"
  FileUtils.rm_f(Dir.glob("#{server.container.config_item('lxc.rootfs')}/etc/ssh/ssh_host*_key*"))
end

#configure_analytics(server) ⇒ Object



871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
# File 'lib/dev-lxc/cluster.rb', line 871

def configure_analytics(server)
  FileUtils.mkdir_p("#{server.container.config_item('lxc.rootfs')}/var/opt/opscode-analytics")
  FileUtils.touch("#{server.container.config_item('lxc.rootfs')}/var/opt/opscode-analytics/.license.accepted")
  if @config['analytics'][:topology] == "standalone" || @config['analytics'][:bootstrap_backend] == server.name
    puts "Copying /etc/opscode-analytics from Chef Server bootstrap backend '#{@config['chef-server'][:bootstrap_backend]}'"
    FileUtils.cp_r("#{get_server(@config['chef-server'][:bootstrap_backend]).container.config_item('lxc.rootfs')}/etc/opscode-analytics",
                   "#{server.container.config_item('lxc.rootfs')}/etc", preserve: true)

    IO.write("#{server.container.config_item('lxc.rootfs')}/etc/opscode-analytics/opscode-analytics.rb", analytics_config)
  elsif @config['analytics'][:frontends].include?(server.name)
    puts "Copying /etc/opscode-analytics from Analytics bootstrap backend '#{@config['analytics'][:bootstrap_backend]}'"
    FileUtils.cp_r("#{get_server(@config['analytics'][:bootstrap_backend]).container.config_item('lxc.rootfs')}/etc/opscode-analytics",
                   "#{server.container.config_item('lxc.rootfs')}/etc", preserve: true)
  end
  server.run_command("opscode-analytics-ctl reconfigure")
end

#configure_automate(server) ⇒ Object



670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
# File 'lib/dev-lxc/cluster.rb', line 670

def configure_automate(server)
  license_path = @server_configs[server.name][:license_path]
  chef_org = @server_configs[server.name][:chef_org]
  enterprise_name = @server_configs[server.name][:enterprise_name]
  chef_server_url = @config['chef-server'][:fqdn]
  supermarket_fqdn = @config['supermarket'][:fqdn]

  FileUtils.cp(license_path, "#{server.container.config_item('lxc.rootfs')}/root/automate.license")

  chef_server = get_server(@config['chef-server'][:bootstrap_backend])
  automate_chef_user_key = "#{chef_server.container.config_item('lxc.rootfs')}/root/chef-repo/.chef/delivery.pem"
  FileUtils.cp(automate_chef_user_key, "#{server.container.config_item('lxc.rootfs')}/root/automate_chef_user_key.pem")

  setup_cmd = "setup"
  setup_cmd += " --license /root/automate.license"
  setup_cmd += " --fqdn #{server.name}"
  setup_cmd += " --key /root/automate_chef_user_key.pem"
  setup_cmd += " --server-url https://#{chef_server_url}/organizations/#{chef_org}"
  setup_cmd += " --supermarket-fqdn #{supermarket_fqdn}" if supermarket_fqdn
  setup_cmd += " --enterprise #{enterprise_name}"
  setup_cmd += " --no-build-node"
  setup_cmd += " --no-configure"
  server.run_command("delivery-ctl #{setup_cmd}")

  # enable Compliance profiles asset store
  DevLXC::append_line_to_file("#{server.container.config_item('lxc.rootfs')}/etc/delivery/delivery.rb", "compliance_profiles['enable'] = true\n")

  server.run_command("delivery-ctl reconfigure")

  # give time for all services to come up completely
  sleep 10

  server.run_command("delivery-ctl create-enterprise #{enterprise_name} --ssh-pub-key-file /etc/delivery/builder_key.pub", "#{server.container.config_item('lxc.rootfs')}/etc/delivery/#{enterprise_name}-admin-credentials")
end

#configure_build_node(server) ⇒ Object



720
721
722
723
724
725
726
727
728
729
730
731
732
# File 'lib/dev-lxc/cluster.rb', line 720

def configure_build_node(server)
  automate_server_name = @server_configs.select {|name, config| config[:server_type] == 'automate'}.keys.first
  if automate_server_name
    automate_server = get_server(automate_server_name)
    install_build_node_cmd = "install-build-node"
    install_build_node_cmd += " --fqdn #{server.name}"
    install_build_node_cmd += " --username dev-lxc"
    install_build_node_cmd += " --password dev-lxc"
    install_build_node_cmd += " --installer #{@server_configs[server.name][:required_products]["chefdk"]}"
    install_build_node_cmd += " --overwrite-registration"
    automate_server.run_command("delivery-ctl #{install_build_node_cmd}")
  end
end

#configure_chef_backend(server) ⇒ Object



785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
# File 'lib/dev-lxc/cluster.rb', line 785

def configure_chef_backend(server)
  FileUtils.mkdir_p("#{server.container.config_item('lxc.rootfs')}/var/opt/chef-backend")
  FileUtils.touch("#{server.container.config_item('lxc.rootfs')}/var/opt/chef-backend/.license.accepted")
  if server.name == @config['chef-backend'][:leader_backend]
    puts "Creating /etc/chef-backend/chef-backend.rb"
    FileUtils.mkdir_p("#{server.container.config_item('lxc.rootfs')}/etc/chef-backend")
    chef_backend_config = "publish_address '#{@server_configs[server.name][:ipaddress]}'\n"
    IO.write("#{server.container.config_item('lxc.rootfs')}/etc/chef-backend/chef-backend.rb", chef_backend_config)
    server.run_command("chef-backend-ctl bootstrap --yes")
  else
    puts "Joining #{server.name} to the chef-backend cluster"
    leader_backend = get_server(@config['chef-backend'][:leader_backend])
    FileUtils.cp("#{leader_backend.container.config_item('lxc.rootfs')}/etc/chef-backend/chef-backend-secrets.json",
                 "#{server.container.config_item('lxc.rootfs')}/root/")
    server.run_command("chef-backend-ctl join-cluster #{@server_configs[leader_backend.name][:ipaddress]} -p #{@server_configs[server.name][:ipaddress]} -s /root/chef-backend-secrets.json --yes")
  end
end

#configure_chef_client(server, dot_chef_path) ⇒ Object



746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
# File 'lib/dev-lxc/cluster.rb', line 746

def configure_chef_client(server, dot_chef_path)
  puts "Configuring Chef Client in container '#{server.name}'"

  FileUtils.mkdir_p("#{server.container.config_item('lxc.rootfs')}/etc/chef")

  chef_server_url = @server_configs[server.name][:chef_server_url]
  validation_client_name = @server_configs[server.name][:validation_client_name]
  validation_key = @server_configs[server.name][:validation_key]
  if validation_key.nil?
    chef_server_name = @config['chef-server'][:bootstrap_backend]
    chef_server_name ||= @config['chef-backend'][:bootstrap_frontend]
    if chef_server_name
      chef_server = get_server(chef_server_name)
      if chef_server.container.defined?
        validator_pem_files = Dir.glob("#{chef_server.container.config_item('lxc.rootfs')}#{dot_chef_path}/*-validator.pem")
        FileUtils.cp(validator_pem_files, "#{server.container.config_item('lxc.rootfs')}/etc/chef/") unless validator_pem_files.empty?
      end
    end
  else
    FileUtils.cp(validation_key, "#{server.container.config_item('lxc.rootfs')}/etc/chef/") if File.exists?(validation_key)
  end

  client_rb = %Q(chef_server_url '#{chef_server_url}'
validation_client_name '#{validation_client_name}'
validation_key '/etc/chef/#{validation_client_name}.pem'
ssl_verify_mode :verify_none
)

  automate_server_name = @server_configs.select {|name, config| config[:server_type] == 'automate'}.keys.first
  if automate_server_name
    client_rb += %Q(
# data_collector.server_url "https://#{automate_server_name}/data-collector/v0/"
# data_collector.token "93a49a4f2482c64126f7b6015e6b0f30284287ee4054ff8807fb63d9cbd1c506"
)
  end

  IO.write("#{server.container.config_item('lxc.rootfs')}/etc/chef/client.rb", client_rb)
end

#configure_chef_frontend(server) ⇒ Object



803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
# File 'lib/dev-lxc/cluster.rb', line 803

def configure_chef_frontend(server)
  puts "Creating /etc/opscode/chef-server.rb"
  FileUtils.mkdir_p("#{server.container.config_item('lxc.rootfs')}/etc/opscode")
  leader_backend = get_server(@config['chef-backend'][:leader_backend])
  leader_backend.run_command("chef-backend-ctl gen-server-config #{server.name} --filename /tmp/#{server.name}.rb")
  FileUtils.cp("#{leader_backend.container.config_item('lxc.rootfs')}/tmp/#{server.name}.rb",
               "#{server.container.config_item('lxc.rootfs')}/etc/opscode/chef-server.rb")
  unless server.name == @config['chef-backend'][:bootstrap_frontend]
    bootstrap_frontend = get_server(@config['chef-backend'][:bootstrap_frontend])
    puts "Copying /etc/opscode/private-chef-secrets.json from bootstrap frontend '#{bootstrap_frontend.name}'"
    FileUtils.cp("#{bootstrap_frontend.container.config_item('lxc.rootfs')}/etc/opscode/private-chef-secrets.json",
                 "#{server.container.config_item('lxc.rootfs')}/etc/opscode/")
    puts "Copying /etc/opscode/pivotal.pem from bootstrap frontend '#{bootstrap_frontend.name}'"
    FileUtils.cp("#{bootstrap_frontend.container.config_item('lxc.rootfs')}/etc/opscode/pivotal.pem",
                 "#{server.container.config_item('lxc.rootfs')}/etc/opscode/")
  end
  server.run_command("chef-server-ctl reconfigure")
end

#configure_chef_server(server) ⇒ Object



822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
# File 'lib/dev-lxc/cluster.rb', line 822

def configure_chef_server(server)
  if @config['chef-server'][:topology] == "standalone" || @config['chef-server'][:bootstrap_backend] == server.name
    case @server_configs[server.name][:chef_server_type]
    when 'private-chef'
      puts "Creating /etc/opscode/private-chef.rb"
      FileUtils.mkdir_p("#{server.container.config_item('lxc.rootfs')}/etc/opscode")
      IO.write("#{server.container.config_item('lxc.rootfs')}/etc/opscode/private-chef.rb", chef_server_config)
    when 'chef-server'
      puts "Creating /etc/opscode/chef-server.rb"
      FileUtils.mkdir_p("#{server.container.config_item('lxc.rootfs')}/etc/opscode")
      IO.write("#{server.container.config_item('lxc.rootfs')}/etc/opscode/chef-server.rb", chef_server_config)
    end
  elsif @config['chef-server'][:frontends].include?(server.name)
    puts "Copying /etc/opscode from bootstrap backend '#{@config['chef-server'][:bootstrap_backend]}'"
    FileUtils.cp_r("#{get_server(@config['chef-server'][:bootstrap_backend]).container.config_item('lxc.rootfs')}/etc/opscode",
                   "#{server.container.config_item('lxc.rootfs')}/etc", preserve: true)
  end
  server.run_command("#{@server_configs[server.name][:chef_server_type]}-ctl reconfigure")
end

#configure_compliance(server) ⇒ Object



888
889
890
891
892
893
894
895
896
897
# File 'lib/dev-lxc/cluster.rb', line 888

def configure_compliance(server)
  FileUtils.mkdir_p("#{server.container.config_item('lxc.rootfs')}/var/opt/chef-compliance")
  FileUtils.touch("#{server.container.config_item('lxc.rootfs')}/var/opt/chef-compliance/.license.accepted")
  server.run_command("chef-compliance-ctl reconfigure")
  admin_user = @server_configs[server.name][:admin_user]
  if admin_user
    server.run_command("chef-compliance-ctl user-create #{admin_user} #{admin_user}")
    server.run_command("chef-compliance-ctl restart core")
  end
end

#configure_manage(server) ⇒ Object



859
860
861
862
863
864
865
866
867
868
869
# File 'lib/dev-lxc/cluster.rb', line 859

def configure_manage(server)
  FileUtils.mkdir_p("#{server.container.config_item('lxc.rootfs')}/var/opt/chef-manage")
  FileUtils.touch("#{server.container.config_item('lxc.rootfs')}/var/opt/chef-manage/.license.accepted")
  if @server_configs[server.name][:chef_server_type] == 'private-chef'
    puts "Disabling old opscode-webui in /etc/opscode/private-chef.rb"
    DevLXC.search_file_delete_line("#{server.container.config_item('lxc.rootfs')}/etc/opscode/private-chef.rb", /opscode_webui[.enable.]/)
    DevLXC.append_line_to_file("#{server.container.config_item('lxc.rootfs')}/etc/opscode/private-chef.rb", "\nopscode_webui['enable'] = false\n")
    server.run_command("#{@server_configs[server.name][:chef_server_type]}-ctl reconfigure")
  end
  server.run_command("opscode-manage-ctl reconfigure")
end

#configure_products(server) ⇒ Object



610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
# File 'lib/dev-lxc/cluster.rb', line 610

def configure_products(server)
  puts "Configuring container '#{server.name}'"
  server.start
  installed_products = @server_configs[server.name][:products].keys if @server_configs[server.name][:products]
  installed_products ||= Array.new
  server_type = @server_configs[server.name][:server_type]
  dot_chef_path = "/root/chef-repo/.chef"
  case server_type
  when 'adhoc'
    # Allow adhoc servers time to generate SSH Server Host Keys
    sleep 5
  when 'analytics'
    configure_analytics(server) if installed_products.include?('analytics')
  when 'build-nodes'
    sleep 5     # give time for DNS resolution to be available
    configure_build_node(server)
  when 'runners'
    sleep 5     # give time for DNS resolution to be available
    configure_runner(server)
  when 'chef-backend'
    configure_chef_backend(server) if installed_products.include?('chef-backend')
    if installed_products.include?('chef-server')
      configure_chef_frontend(server)
      if server.name == @config['chef-backend'][:bootstrap_frontend]
        create_users_orgs_knife_configs(server, dot_chef_path)
      end
    end
    configure_manage(server) if installed_products.include?('manage')
  when 'chef-server'
    if installed_products.include?('chef-server') || installed_products.include?('private-chef')
      configure_chef_server(server)
      if server.name == @config['chef-server'][:bootstrap_backend]
        create_users_orgs_knife_configs(server, dot_chef_path)

        automate_server_name = @server_configs.select {|name, config| config[:server_type] == 'automate'}.keys.first
        if automate_server_name
          automate_user = "delivery"
          automate_chef_org = @server_configs[automate_server_name][:chef_org]
          create_user(server, automate_user, dot_chef_path)
          create_org(server, automate_chef_org, dot_chef_path)
          org_add_user(server, automate_chef_org, automate_user, true, dot_chef_path)
        end
      end
    end
    configure_reporting(server) if installed_products.include?('reporting')
    configure_push_jobs_server(server) if installed_products.include?('push-jobs-server')
    configure_manage(server) if installed_products.include?('manage')
  when 'compliance'
    configure_compliance(server) if installed_products.include?('compliance')
  when 'automate'
    configure_automate(server) if installed_products.include?('delivery')
  when 'nodes'
    # Allow servers time to generate SSH Server Host Keys
    sleep 5
    configure_chef_client(server, dot_chef_path) if installed_products.include?('chef') || installed_products.include?('chefdk')
  when 'supermarket'
    configure_supermarket(server) if installed_products.include?('supermarket')
  end
end

#configure_push_jobs_server(server) ⇒ Object



854
855
856
857
# File 'lib/dev-lxc/cluster.rb', line 854

def configure_push_jobs_server(server)
  server.run_command("opscode-push-jobs-server-ctl reconfigure")
  server.run_command("#{@server_configs[server.name][:chef_server_type]}-ctl reconfigure")
end

#configure_reporting(server) ⇒ Object



842
843
844
845
846
847
848
849
850
851
852
# File 'lib/dev-lxc/cluster.rb', line 842

def configure_reporting(server)
  FileUtils.mkdir_p("#{server.container.config_item('lxc.rootfs')}/var/opt/opscode-reporting")
  FileUtils.touch("#{server.container.config_item('lxc.rootfs')}/var/opt/opscode-reporting/.license.accepted")
  if @config['chef-server'][:frontends].include?(server.name)
    puts "Copying /etc/opscode-reporting from bootstrap backend '#{@config['chef-server'][:bootstrap_backend]}'"
    FileUtils.cp_r("#{get_server(@config['chef-server'][:bootstrap_backend]).container.config_item('lxc.rootfs')}/etc/opscode-reporting",
                   "#{server.container.config_item('lxc.rootfs')}/etc", preserve: true)
  end
  server.run_command("#{@server_configs[server.name][:chef_server_type]}-ctl reconfigure")
  server.run_command("opscode-reporting-ctl reconfigure")
end

#configure_runner(server) ⇒ Object



734
735
736
737
738
739
740
741
742
743
744
# File 'lib/dev-lxc/cluster.rb', line 734

def configure_runner(server)
  automate_server_name = @server_configs.select {|name, config| config[:server_type] == 'automate'}.keys.first
  if automate_server_name
    automate_server = get_server(automate_server_name)
    install_runner_cmd = "install-runner #{server.name} dev-lxc"
    install_runner_cmd += " --password dev-lxc"
    install_runner_cmd += " --installer #{@server_configs[server.name][:required_products]["chefdk"]}"
    install_runner_cmd += " --yes"
    automate_server.run_command("delivery-ctl #{install_runner_cmd}")
  end
end

#configure_supermarket(server) ⇒ Object



899
900
901
902
903
904
905
906
907
908
909
910
911
912
# File 'lib/dev-lxc/cluster.rb', line 899

def configure_supermarket(server)
  if @config['chef-server'][:bootstrap_backend] && get_server(@config['chef-server'][:bootstrap_backend]).container.defined?
    chef_server_supermarket_config = JSON.parse(IO.read("#{get_server(@config['chef-server'][:bootstrap_backend]).container.config_item('lxc.rootfs')}/etc/opscode/oc-id-applications/supermarket.json"))
    supermarket_config = {
      'chef_server_url' => "https://#{@config['chef-server'][:fqdn]}/",
      'chef_oauth2_app_id' => chef_server_supermarket_config['uid'],
      'chef_oauth2_secret' => chef_server_supermarket_config['secret'],
      'chef_oauth2_verify_ssl' => false
    }
    FileUtils.mkdir_p("#{server.container.config_item('lxc.rootfs')}/etc/supermarket")
    IO.write("#{server.container.config_item('lxc.rootfs')}/etc/supermarket/supermarket.json", JSON.pretty_generate(supermarket_config))
  end
  server.run_command("supermarket-ctl reconfigure")
end

#create_dns_recordsObject



479
480
481
482
483
484
485
486
487
488
489
490
491
# File 'lib/dev-lxc/cluster.rb', line 479

def create_dns_records
  get_sorted_servers.each do |server|
    ipaddress = server.container.ip_addresses.first
    ipaddress ||= @server_configs[server.name][:ipaddress]
    next unless ipaddress
    additional_fqdn = @server_configs[server.name][:additional_fqdn]
    dns_record = "#{ipaddress} #{server.name}"
    dns_record += " #{additional_fqdn}\n" if additional_fqdn
    DevLXC::search_file_delete_line("/etc/lxc/addn-hosts.conf", /\s#{server.name}/)
    DevLXC::append_line_to_file("/etc/lxc/addn-hosts.conf", dns_record)
  end
  DevLXC::reload_dnsmasq
end

#create_knife_config(fqdn, dot_chef_path) ⇒ Object



966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
# File 'lib/dev-lxc/cluster.rb', line 966

def create_knife_config(fqdn, dot_chef_path)
  knife_rb = %Q(
username = "CHANGEME"
orgname = "CHANGEME"

if [username, orgname].include?("CHANGEME")
  puts "ERROR: Please set 'username' and 'orgname' to proper values in knife.rb"
  exit!
end

current_dir = File.dirname(__FILE__)

chef_server_url "https://#{fqdn}/organizations/\#{orgname}"

node_name username
client_key "\#{current_dir}/\#{username}.pem"

validation_client_name "\#{orgname}-validator"
validation_key "\#{current_dir}/\#{orgname}-validator.pem"

cookbook_path Dir.pwd + "/cookbooks"
knife[:chef_repo_path] = Dir.pwd

ssl_verify_mode :verify_none
)
  IO.write("#{dot_chef_path}/knife.rb", knife_rb)
end

#create_org(server, orgname, dot_chef_path) ⇒ Object



1004
1005
1006
1007
1008
1009
1010
1011
1012
# File 'lib/dev-lxc/cluster.rb', line 1004

def create_org(server, orgname, dot_chef_path)
  create_org_string = "#{orgname} #{orgname} --filename #{dot_chef_path}/#{orgname}-validator.pem"
  case @server_configs[server.name][:chef_server_type]
  when 'private-chef'
    server.run_command("/opt/opscode/embedded/bin/knife opc org create #{create_org_string} -c #{dot_chef_path}/pivotal.rb")
  when 'chef-server'
    server.run_command("chef-server-ctl org-create #{create_org_string}")
  end
end

#create_pivotal_knife_config(fqdn, dot_chef_path) ⇒ Object



948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
# File 'lib/dev-lxc/cluster.rb', line 948

def create_pivotal_knife_config(fqdn, dot_chef_path)
  pivotal_rb = %Q(
current_dir = File.dirname(__FILE__)

chef_server_root "https://#{fqdn}"
chef_server_url "https://#{fqdn}"

node_name "pivotal"
client_key "\#{current_dir}/pivotal.pem"

cookbook_path Dir.pwd + "/cookbooks"
knife[:chef_repo_path] = Dir.pwd

ssl_verify_mode :verify_none
)
  IO.write("#{dot_chef_path}/pivotal.rb", pivotal_rb)
end

#create_user(server, username, dot_chef_path) ⇒ Object



994
995
996
997
998
999
1000
1001
1002
# File 'lib/dev-lxc/cluster.rb', line 994

def create_user(server, username, dot_chef_path)
  create_user_string = "#{username} #{username} #{username} #{username}@noreply.com #{username} --filename #{dot_chef_path}/#{username}.pem"
  case @server_configs[server.name][:chef_server_type]
  when 'private-chef'
    server.run_command("/opt/opscode/embedded/bin/knife opc user create #{create_user_string} -c #{dot_chef_path}/pivotal.rb")
  when 'chef-server'
    server.run_command("chef-server-ctl user-create #{create_user_string}")
  end
end

#create_users_orgs_knife_configs(server, dot_chef_path) ⇒ Object



914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
# File 'lib/dev-lxc/cluster.rb', line 914

def create_users_orgs_knife_configs(server, dot_chef_path)
  server_type = @server_configs[server.name][:server_type]
  # give time for all services to come up completely
  sleep 10
  if @server_configs[server.name][:chef_server_type] == 'private-chef'
    # give more time for all services to come up completely
    sleep 50
    server.run_command("/opt/opscode/embedded/bin/gem install knife-opc --no-ri --no-rdoc -v 0.3.1")
  end
  chef_server_dot_chef_path = "#{server.container.config_item('lxc.rootfs')}#{dot_chef_path}"
  FileUtils.mkdir_p(chef_server_dot_chef_path)
  FileUtils.cp( "#{server.container.config_item('lxc.rootfs')}/etc/opscode/pivotal.pem", chef_server_dot_chef_path )
  create_pivotal_knife_config('127.0.0.1', chef_server_dot_chef_path)
  create_knife_config('127.0.0.1', chef_server_dot_chef_path)
  @config[server_type][:users].each do |username|
    create_user(server, username, dot_chef_path)
  end
  @config[server_type][:orgs].each do |orgname, org_users|
    create_org(server, orgname, dot_chef_path)
    if org_users
      if org_users['admins']
        org_users['admins'].each do |username|
          org_add_user(server, orgname, username, true, dot_chef_path)
        end
      end
      if org_users['non-admins']
        org_users['non-admins'].each do |username|
          org_add_user(server, orgname, username, false, dot_chef_path)
        end
      end
    end
  end
end

#delete_dns_recordsObject



312
313
314
315
# File 'lib/dev-lxc/cluster.rb', line 312

def delete_dns_records
  @server_configs.keys.each { |server_name| DevLXC::search_file_delete_line("/etc/lxc/addn-hosts.conf", /\s#{server_name}/) }
  DevLXC::reload_dnsmasq
end

#destroy(server_name_regex = nil) ⇒ Object



300
301
302
303
304
# File 'lib/dev-lxc/cluster.rb', line 300

def destroy(server_name_regex=nil)
  servers = get_sorted_servers(server_name_regex)
  servers.reverse_each { |s| s.destroy; puts }
  delete_dns_records unless get_sorted_servers.any? { |s| s.container.state != :stopped }
end

#get_product_download_info(server, product_name, product_options) ⇒ Object



503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
# File 'lib/dev-lxc/cluster.rb', line 503

def get_product_download_info(server, product_name, product_options)
  server_type = @server_configs[server.name][:server_type]
  base_container = DevLXC::Container.new(@server_configs[server.name][:base_container_name])
  mixlib_install_platform_detection_path = "#{base_container.config_item('lxc.rootfs')}/mixlib-install-platform-detection"
  IO.write(mixlib_install_platform_detection_path, Mixlib::Install::Generator::Bourne.detect_platform_sh)
  platform_results = `chroot #{base_container.config_item('lxc.rootfs')} bash mixlib-install-platform-detection`
  File.unlink(mixlib_install_platform_detection_path)
  if platform_results.empty?
    puts "ERROR: Unable to detect the platform of container '#{base_container.name}'"
    exit 1
  end
  (platform, platform_version, architecture) = platform_results.split
  product_version = product_options['version'] if product_options
  product_version ||= 'latest'
  channel = product_options['channel'] if product_options
  channel ||= 'stable'
  channel = channel.to_sym
  options = {
    product_name: product_name,
    product_version: product_version,
    channel: channel,
    platform: platform,
    platform_version: platform_version,
    architecture: architecture
  }
  artifact = Mixlib::Install.new(options).artifact_info
  if artifact.class != Mixlib::Install::ArtifactInfo
    puts "ERROR: Unable to find download URL for the following product"
    puts JSON.pretty_generate(options)
    exit 1
  end
  [artifact.url, artifact.sha256]
end

#get_server(server_name) ⇒ Object



259
260
261
262
263
264
265
# File 'lib/dev-lxc/cluster.rb', line 259

def get_server(server_name)
  ipaddress = @server_configs[server_name][:ipaddress]
  additional_fqdn = @server_configs[server_name][:additional_fqdn]
  mounts = @server_configs[server_name][:mounts]
  ssh_keys = @server_configs[server_name][:ssh_keys]
  Server.new(server_name, ipaddress, additional_fqdn, mounts, ssh_keys)
end

#get_sorted_servers(server_name_regex = nil) ⇒ Object



267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
# File 'lib/dev-lxc/cluster.rb', line 267

def get_sorted_servers(server_name_regex=nil)
  servers = Array.new

  # the order of this list of server_types matters
  # it determines the order in which actions are applied to each server_type
  %w(chef-backend chef-server analytics compliance supermarket automate build-nodes runners nodes adhoc).each do |server_type|
    unless @config[server_type].empty?
      case server_type
      when "chef-backend"
        @config[server_type][:backends].each do |backend_name|
          servers << get_server(backend_name)
        end
        @config[server_type][:frontends].each do |frontend_name|
          servers << get_server(frontend_name)
        end
      when "analytics", "chef-server"
        if @config[server_type][:bootstrap_backend]
          server_name = @config[server_type][:bootstrap_backend]
          servers << get_server(server_name)
        end
        @config[server_type][:frontends].each do |frontend_name|
          servers << get_server(frontend_name)
        end
      end
    end
    if %w(adhoc automate build-nodes compliance nodes runners supermarket).include?(server_type)
      server_configs = @server_configs.select { |server_name, server_config| server_config[:server_type] == server_type }
      server_configs.each_key { |server_name| servers << get_server(server_name) }
    end
  end
  servers.select { |s| s.name =~ /#{server_name_regex}/ }
end

#halt(server_name_regex = nil) ⇒ Object



306
307
308
309
310
# File 'lib/dev-lxc/cluster.rb', line 306

def halt(server_name_regex=nil)
  servers = get_sorted_servers(server_name_regex)
  servers.reverse_each { |s| s.shutdown; puts }
  delete_dns_records unless get_sorted_servers.any? { |s| s.container.state != :stopped }
end

#install_products(server) ⇒ Object



587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
# File 'lib/dev-lxc/cluster.rb', line 587

def install_products(server)
  if !server.snapshot_list.select { |sn| sn[2].to_s.start_with?("dev-lxc build: products installed") }.empty?
    puts "Skipping product installation for container '#{server.name}' because it already has a 'products installed' snapshot"
    return
  elsif !server.snapshot_list.select { |sn| sn[2].to_s.start_with?("dev-lxc build: completed") }.empty?
    puts "Skipping product installation for container '#{server.name}' because it already has a 'build: completed' snapshot"
    return
  end
  if server.container.running?
    server_was_running = true
  else
    server_was_running = false
    server.start
  end
  @server_configs[server.name][:required_products].each do |product_name, package_source|
    next if %w(build-nodes runners).include?(@server_configs[server.name][:server_type]) && product_name == "chefdk"
    server.install_package(package_source)
  end
  server.shutdown
  server.snapshot("dev-lxc build: products installed")
  server.start if server_was_running
end

#org_add_user(server, orgname, username, admin, dot_chef_path) ⇒ Object



1014
1015
1016
1017
1018
1019
1020
1021
1022
1023
# File 'lib/dev-lxc/cluster.rb', line 1014

def org_add_user(server, orgname, username, admin, dot_chef_path)
  org_add_user_string = "#{orgname} #{username}"
  org_add_user_string += " --admin" if admin
  case @server_configs[server.name][:chef_server_type]
  when 'private-chef'
    server.run_command("/opt/opscode/embedded/bin/knife opc org user add #{org_add_user_string} -c #{dot_chef_path}/pivotal.rb")
  when 'chef-server'
    server.run_command("chef-server-ctl org-user-add #{org_add_user_string}")
  end
end

#prep_product_cache(servers, force = false) ⇒ Object



565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
# File 'lib/dev-lxc/cluster.rb', line 565

def prep_product_cache(servers, force=false)
  all_required_products = calculate_required_products(servers, force)
  all_required_products.each do |package_source, v|
    product_name, package_sha256 = v
    if package_source.start_with?('http')
      product_cache_path = "/var/dev-lxc/cache/chef-products/#{product_name}/#{File.basename(package_source)}"
      if !File.exist?(product_cache_path)
        FileUtils.mkdir_p(File.dirname(product_cache_path)) unless Dir.exist?(File.dirname(product_cache_path))
        puts "Downloading #{package_source} to #{product_cache_path}"
        open(package_source) { |url| File.open(product_cache_path, 'wb') { |f| f.write(url.read) } }
      end
      if package_sha256 != Digest::SHA256.file(product_cache_path).hexdigest
        puts "ERROR: Incorrect SHA256 for #{product_cache_path}."
        exit 1
      end
    elsif !File.exist?(package_source)
      puts "ERROR: Package source #{package_source} does not exist."
      exit 1
    end
  end
end


705
706
707
708
709
710
711
712
713
714
715
716
717
718
# File 'lib/dev-lxc/cluster.rb', line 705

def print_automate_credentials
  automate_server_name = @server_configs.select {|name, config| config[:server_type] == 'automate'}.keys.first
  if automate_server_name
    automate_server = get_server(automate_server_name)
    automate_credentials_files = Dir.glob("#{automate_server.container.config_item('lxc.rootfs')}/etc/delivery/*-credentials")
    automate_credentials_files.each_with_index do |automate_credentials_file, index|
      puts IO.read(automate_credentials_file)
      puts if index + 1 < automate_credentials_files.length
    end
  else
    puts "WARNING: An Automate server is not defined."
    exit 1
  end
end

#show_config(include_products = false) ⇒ Object



249
250
251
252
253
254
255
256
257
# File 'lib/dev-lxc/cluster.rb', line 249

def show_config(include_products=false)
  require 'pp'
  calculate_required_products(get_sorted_servers, true) if include_products
  puts "Cluster config:"
  pp @config
  puts
  puts "Server configs:"
  pp @server_configs
end

#up(server_name_regex = nil) ⇒ Object



317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
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
# File 'lib/dev-lxc/cluster.rb', line 317

def up(server_name_regex=nil)
  configured_servers = Array.new
  servers = get_sorted_servers(server_name_regex)
  exit 1 if abort_up(servers)
  prep_product_cache(servers)
  servers.each do |server|
    clone_from_base_container(server) unless server.container.defined?
  end
  # get_sorted_servers is called again in order to ensure the container objects are initialized properly in case they were just cloned from the base container
  servers = get_sorted_servers(server_name_regex)
  create_dns_records unless servers.empty?
  servers.each do |server|
    if %w(build-nodes runners).include?(@server_configs[server.name][:server_type])
      next if @server_configs[server.name][:required_products]["chefdk"] && @server_configs[server.name][:required_products].length == 1
    end
    install_products(server) unless @server_configs[server.name][:required_products].empty?
  end
  servers.each do |server|
    if server.snapshot_list.select { |sn| sn[2].to_s.start_with?("dev-lxc build: completed") }.empty?
      if server.name == @config["chef-backend"][:bootstrap_frontend]
        running_backends = Array.new
        @config["chef-backend"][:backends].reverse_each do |server_name|
          backend = get_server(server_name)
          if backend.container.defined? && backend.snapshot_list.select { |sn| sn[2].to_s.start_with?("dev-lxc build: backend cluster configured but frontend not bootstrapped") }.empty?
            if backend.container.running?
              running_backends << backend.name
              backend.shutdown
            end
            backend.snapshot("dev-lxc build: backend cluster configured but frontend not bootstrapped")
            snapshot = backend.snapshot_list.select { |sn| sn[2].to_s.start_with?("dev-lxc build: completed") }.first
            backend.snapshot_destroy(snapshot.first) if snapshot
          end
        end
        @config["chef-backend"][:backends].each do |server_name|
          if running_backends.include?(server_name)
            get_server(server_name).start
            configured_servers << server_name unless configured_servers.include?(server_name)
          end
        end
      end
      configure_products(server)
      configured_servers << server.name
    end
    if server.container.running?
      puts "Container '#{server.name}' is already running"
      puts
    else
      server.start
    end
  end
  configured_servers.reverse_each do |server_name|
    server = get_server(server_name)
    server.shutdown if server.container.running?
    server.snapshot("dev-lxc build: completed")
  end
  configured_servers.each do |server_name|
    server = get_server(server_name)
    server.start if server.container.defined?
  end
end

#validate_cluster_configObject



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
244
245
246
247
# File 'lib/dev-lxc/cluster.rb', line 188

def validate_cluster_config
  base_container_names = Array.new
  mounts = Array.new
  ssh_keys = Array.new
  hostnames = Array.new

  @config.map { |server_type, config| hostnames.push(config[:fqdn]) }

  @server_configs.each do |server_name, server_config|
    base_container_names.push(server_config[:base_container_name]).uniq! if server_config[:base_container_name]
    mounts.concat(server_config[:mounts]).uniq! if server_config[:mounts]
    ssh_keys.concat(server_config[:ssh_keys]).uniq! if server_config[:ssh_keys]
    hostnames.push(server_name)
    case server_config[:server_type]
    when "automate"
      if server_config[:license_path] && !File.file?(server_config[:license_path])
        puts "ERROR: Automate license #{server_config[:license_path]} does not exist or is not a file."
        exit 1
      end
    when "nodes"
      if server_config[:validation_key] && !File.file?(server_config[:validation_key])
        puts "ERROR: Validation key #{server_config[:validation_key]} does not exist or is not a file."
        exit 1
      end
    end
  end

  unless base_container_names.empty?
    base_container_names.each do |base_container_name|
      unless ::DevLXC::Container.new(base_container_name).defined?
        puts "ERROR: Base container #{base_container_name} does not exist."
        exit 1
      end
    end
  end
  unless hostnames.empty?
    hostnames.each do |hostname|
      unless hostname.end_with?(".lxc")
        puts "ERROR: Hostname #{hostname} does not end with '.lxc'."
        exit 1
      end
    end
  end
  unless mounts.empty?
    mounts.each do |mount|
      unless File.exists?(mount.split.first)
        puts "ERROR: Mount source #{mount.split.first} does not exist."
        exit 1
      end
    end
  end
  unless ssh_keys.empty?
    ssh_keys.each do |ssh_key|
      unless File.exists?(ssh_key)
        puts "ERROR: SSH key #{ssh_key} does not exist."
        exit 1
      end
    end
  end
end