mirror of
https://github.com/rapid7/metasploit-framework
synced 2024-11-12 11:52:01 +01:00
756e6d2ad8
* This will significantly decrease the size of the msf install (~5 MB) * ActiveRecord & ActiveSupport are still used, and have been moved to lib/ git-svn-id: file:///home/svn/framework3/trunk@10682 4d416f70-5f16-0410-b530-b9f4589650da
101 lines
2.8 KiB
Ruby
101 lines
2.8 KiB
Ruby
require 'active_support/basic_object'
|
|
|
|
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 < BasicObject
|
|
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
|
|
|
|
# Returns true if <tt>other</tt> is also a Duration instance with the
|
|
# same <tt>value</tt>, or if <tt>other == value</tt>.
|
|
def ==(other)
|
|
if Duration === other
|
|
other.value == value
|
|
else
|
|
other == value
|
|
end
|
|
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.current)
|
|
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.current)
|
|
sum(-1, time)
|
|
end
|
|
alias :until :ago
|
|
|
|
def inspect #:nodoc:
|
|
consolidated = parts.inject(::Hash.new(0)) { |h,part| h[part.first] += part.last; h }
|
|
parts = [: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
|
|
parts = ["0 seconds"] if parts.empty?
|
|
parts.to_sentence(:locale => :en)
|
|
end
|
|
|
|
protected
|
|
|
|
def sum(sign, time = ::Time.current) #:nodoc:
|
|
parts.inject(time) do |t,(type,number)|
|
|
if t.acts_like?(:time) || t.acts_like?(:date)
|
|
if type == :seconds
|
|
t.since(sign * number)
|
|
else
|
|
t.advance(type => sign * number)
|
|
end
|
|
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
|