aboutsummaryrefslogtreecommitdiffstats
path: root/activesupport/lib/active_support/duration.rb
blob: 5279c4435a60696ca6637fdc4605a013a42f1df8 (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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
module ActiveSupport
  # Provides accurate date and time measurements using Date#advance and 
  # Time#advance, respectively. It mainly supports the methods on Numeric,
  # such as in this example:
  #
  #   1.month.ago       # equivalent to Time.now.advance(:months => -1)
  class Duration < Builder::BlankSlate
    attr_accessor :value, :parts
    
    def initialize(value, parts) #:nodoc:
      @value, @parts = value, parts
    end
    
    # Adds another Duration or a Numeric to this Duration. Numeric values
    # are treated as seconds.
    def +(other)
      if Duration === other
        Duration.new(value + other.value, @parts + other.parts)
      else
        Duration.new(value + other, @parts + [[:seconds, other]])
      end
    end
    
    # Subtracts another Duration or a Numeric from this Duration. Numeric
    # values are treated as seconds.
    def -(other)
      self + (-other)
    end
    
    def -@ #:nodoc:
      Duration.new(-value, parts.map { |type,number| [type, -number] })
    end
    
    def is_a?(klass) #:nodoc:
      klass == Duration || super
    end
    
    def self.===(other) #:nodoc:
      other.is_a?(Duration) rescue super
    end
    
    # Calculates a new Time or Date that is as far in the future
    # as this Duration represents.
    def since(time = ::Time.now)
      sum(1, time)
    end
    alias :from_now :since
    
    # Calculates a new Time or Date that is as far in the past
    # as this Duration represents.
    def ago(time = ::Time.now)
      sum(-1, time)
    end
    alias :until :ago
    
    def inspect #:nodoc:
      consolidated = parts.inject(Hash.new(0)) { |h,part| h[part.first] += part.last; h }
      [:years, :months, :days, :minutes, :seconds].map do |length|
        n = consolidated[length]
        "#{n} #{n == 1 ? length.to_s.singularize : length.to_s}" if n.nonzero?
      end.compact.to_sentence
    end
    
    protected
    
    def sum(sign, time = ::Time.now) #:nodoc:
      parts.inject(time) do |t,(type,number)|
        if t.acts_like?(:time)
          if type == :seconds
            t + (sign * number)
          else
            t.advance(type => sign * number)
          end
        elsif t.acts_like?(:date)
          raise ArgumentError, "Adding seconds to a Date does not make sense" if type == :seconds
          t.advance(type => sign * number)
        else
          raise ArgumentError, "expected a time or date, got #{time.inspect}"
        end
      end
    end
    
    private
    
    def method_missing(method, *args, &block) #:nodoc:
      value.send(method, *args)
    end
  end
end