debian-mirror-gitlab/lib/gitlab/import_export/config.rb
2019-12-26 22:10:19 +05:30

58 lines
1.3 KiB
Ruby

# frozen_string_literal: true
module Gitlab
module ImportExport
class Config
def initialize(config: Gitlab::ImportExport.config_file)
@config = config
@hash = parse_yaml
@hash.deep_symbolize_keys!
@ee_hash = @hash.delete(:ee) || {}
@hash[:tree] = normalize_tree(@hash[:tree])
@ee_hash[:tree] = normalize_tree(@ee_hash[:tree] || {})
end
# Returns a Hash of the YAML file, including EE specific data if EE is
# used.
def to_h
if merge_ee?
deep_merge(@hash, @ee_hash)
else
@hash
end
end
private
def deep_merge(hash_a, hash_b)
hash_a.deep_merge(hash_b) do |_, this_val, other_val|
this_val.to_a + other_val.to_a
end
end
def normalize_tree(item)
case item
when Array
item.reduce({}) do |hash, subitem|
hash.merge!(normalize_tree(subitem))
end
when Hash
item.transform_values(&method(:normalize_tree))
when Symbol
{ item => {} }
else
raise ArgumentError, "#{item} needs to be Array, Hash, Symbol or NilClass"
end
end
def merge_ee?
Gitlab.ee?
end
def parse_yaml
YAML.load_file(@config)
end
end
end
end