From 7f891c3f976d3e9151b885043d323ed69d688520 Mon Sep 17 00:00:00 2001
From: Lucas Nussbaum <lucas.nussbaum@loria.fr>
Date: Sat, 17 Nov 2018 00:19:08 +0100
Subject: [PATCH] [dev] also port lanpowerg5k
---
lib/refrepo/gen/puppet/lanpowerg5k.rb | 186 ++++++++++----------------
1 file changed, 69 insertions(+), 117 deletions(-)
diff --git a/lib/refrepo/gen/puppet/lanpowerg5k.rb b/lib/refrepo/gen/puppet/lanpowerg5k.rb
index 457233f5b66..c4d20194807 100644
--- a/lib/refrepo/gen/puppet/lanpowerg5k.rb
+++ b/lib/refrepo/gen/puppet/lanpowerg5k.rb
@@ -1,134 +1,86 @@
-#!/usr/bin/ruby
-
-if RUBY_VERSION < "2.1"
- puts "This script requires ruby >= 2.1"
- exit
-end
-
# This script generates lanpowerg5k/files/<site_uid>/lanpower.conf from conf/console.yaml and conf/console-password.conf
require 'pp'
require 'yaml'
require 'pathname'
require 'optparse'
-require_relative '../lib/input_loader'
-require_relative '../lib/hash/hash.rb'
+require 'refrepo/input_loader'
+require 'refrepo/hash/hash'
-options = {}
-options[:sites] = %w{grenoble lille luxembourg lyon nancy nantes rennes sophia}
-options[:output_dir] = "/tmp/puppet-repo"
-options[:conf_dir] = File.expand_path("conf-examples/", File.dirname(__FILE__))
-options[:puppet4] = false
+def generate_puppet_lanpowerg5k(options)
+ if not options[:conf_dir]
+ options[:conf_dir] = "#{options[:output_dir]}/platforms/production/generators/ipmitools"
+ end
-OptionParser.new do |opts|
- opts.banner = "Usage: lanpowerg5k.rb [options]"
+ raise("Error: #{options[:conf_dir]} does not exist. The given configuration path is incorrect") unless Pathname(options[:conf_dir].to_s).exist?
- opts.separator ""
- opts.separator "Example: ruby lanpowerg5k.rb -s nancy -o /tmp/puppet-repo"
+ puts "Puppet #{options[:puppet4] ? '4' : '2'} context"
+ puts "Writing lanpower configuration files to: #{options[:output_dir]}"
+ puts "Using configuration directory: #{options[:conf_dir]}"
+ puts "For site(s): #{options[:sites].join(', ')}"
- opts.on('-o', '--output-dir dir', String, 'Select the puppet repo path', "Default: " + options[:output_dir]) do |d|
- options[:output_dir] = d
+ config = YAML::load_file("#{options[:conf_dir]}/console.yaml")
+ credentials = YAML::load_file("#{options[:conf_dir]}/console-password.yaml")
- options[:conf_dir] = begin
- if options[:puppet4]
- "#{options[:output_dir]}/platforms/production/generators/ipmitools"
- else
- "#{options[:output_dir]}/modules/lanpowerg5k/generators/"
- end
- end
- end
+ refapi = load_yaml_file_hierarchy
- opts.on('-c', '--conf-dir dir', String, 'Select the lanpower module configuration path', "Default: #{options[:conf_dir]}") do |d|
- options[:conf_dir] = d
- end
+ refapi['sites'].each { |site_uid, site_refapi|
- opts.on('--puppet4', 'Puppet 4 version', "Default: " + options[:puppet4].to_s) do |d|
- options[:puppet4] = d
- end
+ next unless options[:sites].include?(site_uid)
- opts.separator ""
- opts.separator "Filters:"
+ h = {'clusters' => {} } # output hash
- opts.on('-s', '--sites a,b,c', Array, 'Select site(s)', "Default: " + options[:sites].join(", ")) do |s|
- raise "Wrong argument for -s option." unless (s - options[:sites]).empty?
- options[:sites] = s
- end
+ # Generate config for both cluster and server entries of the refapi
+ site_refapi['servers'] ||= {}
+ cluster_list = site_refapi['clusters'].keys | site_refapi['servers'].keys | config[site_uid].keys | credentials[site_uid].keys
- opts.on_tail("-h", "--help", "Show this message") do
- puts opts
- exit
- end
-end.parse!
-
-raise("Error: #{options[:conf_dir]} does not exist. The given configuration path is incorrect") unless Pathname(options[:conf_dir].to_s).exist?
-
-puts "Puppet #{options[:puppet4] ? '4' : '2'} context"
-puts "Writing lanpower configuration files to: #{options[:output_dir]}"
-puts "Using configuration directory: #{options[:conf_dir]}"
-puts "For site(s): #{options[:sites].join(', ')}"
-
-config = YAML::load_file("#{options[:conf_dir]}/console.yaml")
-credentials = YAML::load_file("#{options[:conf_dir]}/console-password.yaml")
-input_data_dir = "../../input/grid5000/"
-
-refapi = load_yaml_file_hierarchy(File.expand_path(input_data_dir, File.dirname(__FILE__)))
-
-refapi['sites'].each { |site_uid, site_refapi|
-
- next unless options[:sites].include?(site_uid)
-
- h = {'clusters' => {} } # output hash
-
- # Generate config for both cluster and server entries of the refapi
- site_refapi['servers'] ||= {}
- cluster_list = site_refapi['clusters'].keys | site_refapi['servers'].keys | config[site_uid].keys | credentials[site_uid].keys
-
- cluster_list.sort.each { |cluster_uid|
- cluster_refapi = site_refapi['clusters'][cluster_uid].fetch('nodes') rescue site_refapi['servers'][cluster_uid].fetch('nodes') rescue nil
- cluster_config = config[site_uid][cluster_uid]['lanpower'] rescue nil
- cluster_credentials = credentials[site_uid].fetch(cluster_uid) rescue nil
-
- # error handling:
- # - refapi is optional for this generator but every cluster should still be on the ref api => display a warning message
- # - credentials are mandatory and the cluster is skipped if info is missing
- # - config is optional as the cluster might use the default configuration
- puts "Warning: #{site_uid} - #{cluster_uid} not found in the reference api" if cluster_refapi.nil?
- if cluster_credentials.nil?
- puts "Warning: #{site_uid} - #{cluster_uid} not found in console-password.yaml... skipped"
- next
- end
-
- # clusters:
- # griffon:
- # bmc: "ipmi"
- # user: ""
- # password: ""
- # suffix: "-bmc"
- # sleep: "6"
-
- cluster_hash = cluster_config || {}
- cluster_hash['bmc'] ||= 'ipmi'
- cluster_hash['sleep'] ||= '6'
- cluster_hash['suffix'] ||= (cluster_refapi["#{cluster_uid}-1"]['network_adapters']['bmc'].fetch('network_address').split('.')[0].gsub("#{cluster_uid}-1",'') rescue '-bmc')
- cluster_hash['user'] ||= cluster_credentials.split(' ')[0]
- cluster_hash['password'] ||= cluster_credentials.split(' ')[1]
-
- cluster_hash.reject!{ |k,v| v == nil }
-
- h['clusters'][cluster_uid] = cluster_hash
-
- } # clusters.each
-
- # Write output file
- output_file = begin
- if options[:puppet4]
- Pathname("#{options[:output_dir]}//platforms/production/modules/generated/files/grid5000/lanpower/#{site_uid}/lanpower.yaml")
- else
- Pathname("#{options[:output_dir]}/modules/lanpowerg5k/files/#{site_uid}/lanpower.yaml")
- end
- end
- output_file.dirname.mkpath()
- write_yaml(output_file, h)
- add_header(output_file)
+ cluster_list.sort.each { |cluster_uid|
+ cluster_refapi = site_refapi['clusters'][cluster_uid].fetch('nodes') rescue site_refapi['servers'][cluster_uid].fetch('nodes') rescue nil
+ cluster_config = config[site_uid][cluster_uid]['lanpower'] rescue nil
+ cluster_credentials = credentials[site_uid].fetch(cluster_uid) rescue nil
-}
+ # error handling:
+ # - refapi is optional for this generator but every cluster should still be on the ref api => display a warning message
+ # - credentials are mandatory and the cluster is skipped if info is missing
+ # - config is optional as the cluster might use the default configuration
+ puts "Warning: #{site_uid} - #{cluster_uid} not found in the reference api" if cluster_refapi.nil?
+ if cluster_credentials.nil?
+ puts "Warning: #{site_uid} - #{cluster_uid} not found in console-password.yaml... skipped"
+ next
+ end
+
+ # clusters:
+ # griffon:
+ # bmc: "ipmi"
+ # user: ""
+ # password: ""
+ # suffix: "-bmc"
+ # sleep: "6"
+
+ cluster_hash = cluster_config || {}
+ cluster_hash['bmc'] ||= 'ipmi'
+ cluster_hash['sleep'] ||= '6'
+ cluster_hash['suffix'] ||= (cluster_refapi["#{cluster_uid}-1"]['network_adapters']['bmc'].fetch('network_address').split('.')[0].gsub("#{cluster_uid}-1",'') rescue '-bmc')
+ cluster_hash['user'] ||= cluster_credentials.split(' ')[0]
+ cluster_hash['password'] ||= cluster_credentials.split(' ')[1]
+
+ cluster_hash.reject!{ |k,v| v == nil }
+
+ h['clusters'][cluster_uid] = cluster_hash
+
+ } # clusters.each
+
+ # Write output file
+ output_file = begin
+ if options[:puppet4]
+ Pathname("#{options[:output_dir]}//platforms/production/modules/generated/files/grid5000/lanpower/#{site_uid}/lanpower.yaml")
+ else
+ Pathname("#{options[:output_dir]}/modules/lanpowerg5k/files/#{site_uid}/lanpower.yaml")
+ end
+ end
+ output_file.dirname.mkpath()
+ write_yaml(output_file, h)
+ add_header(output_file)
+
+ }
+end
--
GitLab