class Kafo::Configuration
Constants
- DEFAULT
Attributes
answer_file[R]
config_file[R]
Public Class Methods
new(file, persist = true)
click to toggle source
# File lib/kafo/configuration.rb, line 33 def initialize(file, persist = true) @config_file = file @persist = persist configure_application @logger = KafoConfigure.logger @answer_file = app[:answer_file] begin @data = load_yaml_file(@answer_file) rescue Errno::ENOENT => e puts "No answer file at #{@answer_file} found, can not continue" KafoConfigure.exit(:no_answer_file) end @config_dir = File.dirname(@config_file) end
Public Instance Methods
[](key)
click to toggle source
if a value is a true we return empty hash because we have no specific options for a particular puppet module
# File lib/kafo/configuration.rb, line 160 def [](key) value = @data[key] value.is_a?(Hash) ? value : {} end
add_mapping(module_name, mapping)
click to toggle source
# File lib/kafo/configuration.rb, line 119 def add_mapping(module_name, mapping) app[:mapping][module_name] = mapping save_configuration(app) end
add_module(name)
click to toggle source
# File lib/kafo/configuration.rb, line 111 def add_module(name) mod = PuppetModule.new(name, KafoParsers::PuppetModuleParser, self).parse unless modules.map(&:name).include?(mod.name) mod.enable @modules << mod end end
answers()
click to toggle source
# File lib/kafo/configuration.rb, line 240 def answers @data end
app()
click to toggle source
# File lib/kafo/configuration.rb, line 63 def app @app ||= begin begin configuration = load_yaml_file(@config_file) rescue => e configuration = {} end result = DEFAULT.merge(configuration || {}) result[:password] ||= PasswordManager.new.password result[:module_dirs] = result[:modules_dir] || result[:module_dirs] result.delete(:modules_dir) result end end
check_dirs()
click to toggle source
# File lib/kafo/configuration.rb, line 95 def check_dirs [app[:check_dirs] || File.join(root_dir, 'checks')].flatten end
config_header()
click to toggle source
# File lib/kafo/configuration.rb, line 170 def config_header files = [app[:config_header_file], File.join(gem_root, '/config/config_header.txt')].compact file = files.select { |f| File.exists?(f) }.first @config_header ||= file.nil? ? '' : File.read(file) end
configure_application()
click to toggle source
# File lib/kafo/configuration.rb, line 57 def configure_application result = app save_configuration(result) result end
gem_root()
click to toggle source
# File lib/kafo/configuration.rb, line 103 def gem_root File.join(File.dirname(__FILE__), '../../') end
get_custom(key)
click to toggle source
# File lib/kafo/configuration.rb, line 79 def get_custom(key) custom_storage[key.to_sym] end
kafo_modules_dir()
click to toggle source
# File lib/kafo/configuration.rb, line 107 def kafo_modules_dir app[:kafo_modules_dir] || (gem_root + '/modules') end
log_exists?()
click to toggle source
# File lib/kafo/configuration.rb, line 236 def log_exists? File.exists?(log_file) && File.size(log_file) > 0 end
log_file()
click to toggle source
# File lib/kafo/configuration.rb, line 232 def log_file File.join(app[:log_dir], app[:log_name]) end
migrate_configuration(from_config, options={})
click to toggle source
# File lib/kafo/configuration.rb, line 124 def migrate_configuration(from_config, options={}) keys_to_skip = options.fetch(:skip, []) keys = [:log_dir, :log_name, :log_level, :no_prefix, :default_values_dir, :colors, :color_of_background, :custom, :password, :verbose_log_level] keys += options.fetch(:with, []) keys.each do |key| next if keys_to_skip.include?(key) app[key] = from_config.app[key] end save_configuration(app) end
migrations_dir()
click to toggle source
# File lib/kafo/configuration.rb, line 256 def migrations_dir @config_file.gsub(/\.yaml$/, '.migrations') end
module_dirs()
click to toggle source
# File lib/kafo/configuration.rb, line 99 def module_dirs [app[:module_dirs] || (app[:installer_dir] + '/modules')].flatten.map { |dir| File.expand_path(dir) } end
module_enabled?(mod)
click to toggle source
# File lib/kafo/configuration.rb, line 165 def module_enabled?(mod) value = @data[mod.is_a?(String) ? mod : mod.identifier] !!value || value.is_a?(Hash) end
modules()
click to toggle source
# File lib/kafo/configuration.rb, line 87 def modules @modules ||= @data.keys.map { |mod| PuppetModule.new(mod, KafoParsers::PuppetModuleParser, self).parse }.sort end
param(mod, name)
click to toggle source
# File lib/kafo/configuration.rb, line 187 def param(mod, name) params.detect { |p| p.name == name && p.module.name == mod } end
params()
click to toggle source
# File lib/kafo/configuration.rb, line 183 def params @params ||= modules.map(&:params).flatten end
params_changed(old_config)
click to toggle source
# File lib/kafo/configuration.rb, line 211 def params_changed(old_config) # finds params that had different value in the old config params.select do |par| next unless par.module.enabled? old_param = old_config.param(par.module.class_name, par.name) old_param && old_param.value != par.value end end
params_default_values()
click to toggle source
# File lib/kafo/configuration.rb, line 136 def params_default_values @params_default_values ||= begin @logger.debug "Creating tmp dir within #{app[:default_values_dir]}..." temp_dir = Dir.mktmpdir(nil, app[:default_values_dir]) KafoConfigure.exit_handler.register_cleanup_path temp_dir @logger.info 'Loading default values from puppet modules...' command = PuppetCommand.new("$temp_dir=\"#{temp_dir}\" #{includes} dump_values(#{params_to_dump})", ['--noop'], self).append('2>&1').command result = %x`#{command}` @logger.debug result unless $?.exitstatus == 0 log = app[:log_dir] + '/' + app[:log_name] puts "Could not get default values, check log file at #{log} for more information" @logger.error command @logger.error result @logger.error 'Could not get default values, cannot continue' KafoConfigure.exit(:defaults_error) end @logger.info "... finished" load_yaml_file(File.join(temp_dir, 'default_values.yaml')) end end
params_missing(old_config)
click to toggle source
# File lib/kafo/configuration.rb, line 220 def params_missing(old_config) # finds params that are present but will be missing in the new config old_config.params.select do |par| next if !par.module.enabled? || !module_enabled?(par.module.name) param(par.module.class_name, par.name).nil? end end
parser_cache()
click to toggle source
# File lib/kafo/configuration.rb, line 260 def parser_cache if app[:parser_cache_path] @parser_cache ||= Kafo::ParserCacheReader.new_from_file(File.expand_path(app[:parser_cache_path])) end end
preset_defaults_from_other_config(other_config)
click to toggle source
# File lib/kafo/configuration.rb, line 205 def preset_defaults_from_other_config(other_config) params_changed(other_config).each do |par| param(par.module.class_name, par.name).value = other_config.param(par.module.class_name, par.name).value end end
preset_defaults_from_puppet()
click to toggle source
# File lib/kafo/configuration.rb, line 191 def preset_defaults_from_puppet # set values based on default_values params.each do |param| param.set_default(params_default_values) end end
preset_defaults_from_yaml()
click to toggle source
# File lib/kafo/configuration.rb, line 198 def preset_defaults_from_yaml # set values based on YAML params.each do |param| param.set_value_by_config(self) end end
root_dir()
click to toggle source
# File lib/kafo/configuration.rb, line 91 def root_dir File.expand_path(app[:installer_dir]) end
run_migrations()
click to toggle source
# File lib/kafo/configuration.rb, line 244 def run_migrations migrations = Kafo::Migrations.new(migrations_dir) @app, @data = migrations.run(app, answers) if migrations.migrations.count > 0 @modules = nil # force the lazy loaded modules to reload next time they are used save_configuration(app) store(answers) migrations.store_applied @logger.info("#{migrations.migrations.count} migration/s were applied. Updated configuration was saved.") end end
save_configuration(configuration)
click to toggle source
# File lib/kafo/configuration.rb, line 50 def save_configuration(configuration) return true unless @persist FileUtils.touch @config_file File.chmod 0600, @config_file File.open(@config_file, 'w') { |file| file.write(format(YAML.dump(configuration))) } end
set_custom(key, value)
click to toggle source
# File lib/kafo/configuration.rb, line 83 def set_custom(key, value) custom_storage[key.to_sym] = value end
store(data, file = nil)
click to toggle source
# File lib/kafo/configuration.rb, line 176 def store(data, file = nil) filename = file || answer_file FileUtils.touch filename File.chmod 0600, filename File.open(filename, 'w') { |file| file.write(config_header + format(YAML.dump(data))) } end
temp_config_file()
click to toggle source
# File lib/kafo/configuration.rb, line 228 def temp_config_file @temp_config_file ||= "/tmp/kafo_answers_#{rand(1_000_000)}.yaml" end
Private Instance Methods
custom_storage()
click to toggle source
# File lib/kafo/configuration.rb, line 268 def custom_storage app[:custom] end
format(data)
click to toggle source
# File lib/kafo/configuration.rb, line 288 def format(data) data.gsub('!ruby/sym ', ':') end
includes()
click to toggle source
# File lib/kafo/configuration.rb, line 272 def includes modules.map do |mod| module_dir = module_dirs.find do |dir| params_file = File.join(dir, mod.params_path) @logger.debug "checking presence of #{params_file}" File.exist?(params_file) end module_dir ? "include #{mod.dir_name}::#{mod.params_class_name}" : nil end.uniq.compact.join(' ') end
load_yaml_file(filename)
click to toggle source
# File lib/kafo/configuration.rb, line 292 def load_yaml_file(filename) YAML.load_file(filename) end
params_to_dump()
click to toggle source
# File lib/kafo/configuration.rb, line 283 def params_to_dump parameters = params.select { |p| p.default != 'UNSET' } parameters.map { |param| "#{param.dump_default}" }.join(',') end