aboutsummaryrefslogtreecommitdiff
path: root/lib/leap_cli/config_list.rb
blob: c8ff23ba5e1734504de23b818689ada9fc2063e3 (plain)
1
2
3
4
5
6
7
8
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
module LeapCli
  class ConfigList < Hash

    def initialize(config=nil)
      if config
        self << config
      end
    end

    #
    # if the key is a hash, we treat it as a condition and filter all the configs using the condition
    #
    # for example:
    #
    #   nodes[:public_dns => true]
    #
    # will return a ConfigList with node configs that have public_dns set to true
    #
    def [](key)
      if key.is_a? Hash
        results = ConfigList.new
        field, match_value = key.to_a.first
        field = field.is_a?(Symbol) ? field.to_s : field
        match_value = match_value.is_a?(Symbol) ? match_value.to_s : match_value
        each do |name, config|
          value = config[field]
          if !value.nil?
            if value.is_a? Array
              if value.includes?(match_value)
                results[name] = config
              end
            else
              if value == match_value
                results[name] = config
              end
            end
          end
        end
        results
      else
        super
      end
    end

    def <<(config)
      if config.is_a? ConfigList
        self.merge!(config)
      else
        self[config['name']] = config
      end
    end

    #
    # converts the hash of configs into an array of hashes, with ONLY the specified fields
    #
    def fields(*fields)
      result = []
      keys.sort.each do |name|
        result << self[name].to_h(*fields)
      end
      result
    end

    #
    # like fields(), but returns an array of values instead of an array of hashes.
    #
    def field(field)
      field = field.to_s
      result = []
      keys.sort.each do |name|
        result << self[name][field]
      end
      result
    end

  end
end