aboutsummaryrefslogtreecommitdiffstats
path: root/activesupport/lib/active_support/ordered_hash.rb
blob: 5de94c67e0a589cfc25727a9ac99b8ac1afe888c (plain) (blame)
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
# OrderedHash is namespaced to prevent conflicts with other implementations
module ActiveSupport
  # Hash is ordered in Ruby 1.9!
  if RUBY_VERSION >= '1.9'
    OrderedHash = ::Hash
  else
    class OrderedHash < Array #:nodoc:
      def []=(key, value)
        if pair = assoc(key)
          pair.pop
          pair << value
        else
          self << [key, value]
        end
        value
      end

      def [](key)
        pair = assoc(key)
        pair ? pair.last : nil
      end

      def delete(key)
        pair = assoc(key)
        pair ? array_index = index(pair) : nil
        array_index ? delete_at(array_index).last : nil
      end

      def keys
        collect { |key, value| key }
      end

      def values
        collect { |key, value| value }
      end

      def to_hash
        returning({}) do |hash|
          each { |array| hash[array[0]] = array[1] }
        end
      end

      def has_key?(k)
        !assoc(k).nil?
      end

      alias_method :key?, :has_key?
      alias_method :include?, :has_key?
      alias_method :member?, :has_key?

      def has_value?(v)
        any? { |key, value| value == v }
      end

      alias_method :value?, :has_value?

      def each_key
        each { |key, value| yield key }
      end

      def each_value
        each { |key, value| yield value }
      end
    end
  end
end