tzinfo 1.2.6

data/lib/tzinfo/ruby_data_source.rb

@@ -0,0 +1,138 @@
+module TZInfo
+  using RubyCoreSupport::UntaintExt if RubyCoreSupport.const_defined?(:UntaintExt)
+
+  # A DataSource that loads data from the set of Ruby modules included in the
+  # TZInfo::Data library (tzinfo-data gem).
+  #
+  # To have TZInfo use this DataSource, call TZInfo::DataSource.set as follows:
+  #
+  #   TZInfo::DataSource.set(:ruby)
+  class RubyDataSource < DataSource
+    # Whether the timezone index has been loaded yet.
+    @@timezone_index_loaded = false
+    
+    # Whether the country index has been loaded yet.
+    @@country_index_loaded = false
+  
+    # Initializes a new RubyDataSource instance.
+    def initialize
+      tzinfo_data = File.join('tzinfo', 'data')
+      begin
+        require(tzinfo_data)
+
+        data_file = File.join('', 'tzinfo', 'data.rb')
+        path = $".reverse_each.detect {|p| p.end_with?(data_file) }
+        if path
+          @base_path = File.join(File.dirname(path), 'data').untaint
+        else
+          @base_path = tzinfo_data
+        end
+      rescue LoadError
+        @base_path = tzinfo_data
+      end
+    end
+
+    # Returns a TimezoneInfo instance for a given identifier. 
+    # Raises InvalidTimezoneIdentifier if the timezone is not found or the 
+    # identifier is invalid.
+    def load_timezone_info(identifier)
+      raise InvalidTimezoneIdentifier, 'Invalid identifier' if identifier !~ /^[A-Za-z0-9\+\-_]+(\/[A-Za-z0-9\+\-_]+)*$/
+      
+      identifier = identifier.gsub(/-/, '__m__').gsub(/\+/, '__p__')
+      
+      # Untaint identifier after it has been reassigned to a new string. We
+      # don't want to modify the original identifier. identifier may also be 
+      # frozen and therefore cannot be untainted.
+      identifier.untaint
+      
+      identifier = identifier.split('/')
+      begin
+        require_definition(identifier)
+        
+        m = Data::Definitions
+        identifier.each {|part|
+          m = m.const_get(part)
+        }
+        
+        m.get
+      rescue LoadError, NameError => e
+        raise InvalidTimezoneIdentifier, e.message
+      end
+    end
+    
+    # Returns an array of all the available timezone identifiers.
+    def timezone_identifiers
+      load_timezone_index
+      Data::Indexes::Timezones.timezones
+    end
+    
+    # Returns an array of all the available timezone identifiers for
+    # data timezones (i.e. those that actually contain definitions).
+    def data_timezone_identifiers
+      load_timezone_index
+      Data::Indexes::Timezones.data_timezones
+    end
+    
+    # Returns an array of all the available timezone identifiers that
+    # are links to other timezones.
+    def linked_timezone_identifiers
+      load_timezone_index
+      Data::Indexes::Timezones.linked_timezones
+    end
+    
+    # Returns a CountryInfo instance for the given ISO 3166-1 alpha-2
+    # country code. Raises InvalidCountryCode if the country could not be found
+    # or the code is invalid.
+    def load_country_info(code)
+      load_country_index
+      info = Data::Indexes::Countries.countries[code]
+      raise InvalidCountryCode, 'Invalid country code' unless info
+      info
+    end
+    
+    # Returns an array of all the available ISO 3166-1 alpha-2
+    # country codes.
+    def country_codes
+      load_country_index
+      Data::Indexes::Countries.countries.keys.freeze
+    end
+    
+    # Returns the name of this DataSource.
+    def to_s
+      "Ruby DataSource"
+    end
+    
+    private
+    
+    # Requires a zone definition by its identifier (split on /).
+    def require_definition(identifier)
+      require_data(*(['definitions'] + identifier))
+    end
+    
+    # Requires an index by its name.
+    def require_index(name)
+      require_data(*['indexes', name])
+    end
+    
+    # Requires a file from tzinfo/data.
+    def require_data(*file)
+      require(File.join(@base_path, *file))
+    end
+    
+    # Loads in the index of timezones if it hasn't already been loaded.
+    def load_timezone_index
+      unless @@timezone_index_loaded
+        require_index('timezones')
+        @@timezone_index_loaded = true
+      end        
+    end
+    
+    # Loads in the index of countries if it hasn't already been loaded.
+    def load_country_index
+      unless @@country_index_loaded
+        require_index('countries')
+        @@country_index_loaded = true
+      end
+    end
+  end
+end

data/lib/tzinfo/time_or_datetime.rb

@@ -0,0 +1,340 @@
+require 'date'
+require 'rational' unless defined?(Rational)
+require 'time'
+
+module TZInfo
+  # Used by TZInfo internally to represent either a Time, DateTime or
+  # an Integer timestamp (seconds since 1970-01-01 00:00:00).
+  class TimeOrDateTime
+    include Comparable
+    
+    # Constructs a new TimeOrDateTime. timeOrDateTime can be a Time, DateTime
+    # or Integer. If using a Time or DateTime, any time zone information 
+    # is ignored.
+    #
+    # Integer timestamps must be within the range supported by Time on the
+    # platform being used.    
+    def initialize(timeOrDateTime)
+      @time = nil
+      @datetime = nil
+      @timestamp = nil
+      
+      if timeOrDateTime.is_a?(Time)
+        @time = timeOrDateTime
+        
+        # Avoid using the slower Rational class unless necessary.
+        nsec = RubyCoreSupport.time_nsec(@time)
+        usec = nsec % 1000 == 0 ? nsec / 1000 : Rational(nsec, 1000)
+        
+        @time = Time.utc(@time.year, @time.mon, @time.mday, @time.hour, @time.min, @time.sec, usec) unless @time.utc?        
+        @orig = @time
+      elsif timeOrDateTime.is_a?(DateTime)
+        @datetime = timeOrDateTime
+        @datetime = @datetime.new_offset(0) unless @datetime.offset == 0
+        @orig = @datetime
+      else
+        @timestamp = timeOrDateTime.to_i
+        
+        if !RubyCoreSupport.time_supports_64bit && (@timestamp > 2147483647 || @timestamp < -2147483648 || (@timestamp < 0 && !RubyCoreSupport.time_supports_negative))
+          raise RangeError, 'Timestamp is outside the supported range of Time on this platform'
+        end
+        
+        @orig = @timestamp
+      end
+    end
+    
+    # Returns the time as a Time.
+    #
+    # When converting from a DateTime, the result is truncated to microsecond
+    # precision.
+    def to_time
+      # Thread-safety: It is possible that the value of @time may be 
+      # calculated multiple times in concurrently executing threads. It is not 
+      # worth the overhead of locking to ensure that @time is only 
+      # calculated once.
+    
+      unless @time
+        result = if @timestamp
+          Time.at(@timestamp).utc
+        else
+          Time.utc(year, mon, mday, hour, min, sec, usec)
+        end
+
+        return result if frozen?
+        @time = result
+      end
+      
+      @time      
+    end
+    
+    # Returns the time as a DateTime.
+    #
+    # When converting from a Time, the result is truncated to microsecond
+    # precision.
+    def to_datetime
+      # Thread-safety: It is possible that the value of @datetime may be 
+      # calculated multiple times in concurrently executing threads. It is not 
+      # worth the overhead of locking to ensure that @datetime is only 
+      # calculated once.
+    
+      unless @datetime
+        # Avoid using Rational unless necessary.
+        u = usec
+        s = u == 0 ? sec : Rational(sec * 1000000 + u, 1000000)
+        result = RubyCoreSupport.datetime_new(year, mon, mday, hour, min, s)
+        return result if frozen?
+        @datetime = result
+      end
+      
+      @datetime
+    end
+    
+    # Returns the time as an integer timestamp.
+    def to_i
+      # Thread-safety: It is possible that the value of @timestamp may be 
+      # calculated multiple times in concurrently executing threads. It is not 
+      # worth the overhead of locking to ensure that @timestamp is only 
+      # calculated once.
+    
+      unless @timestamp
+        result = to_time.to_i
+        return result if frozen?
+        @timestamp = result
+      end
+      
+      @timestamp
+    end
+    
+    # Returns the time as the original time passed to new.
+    def to_orig
+      @orig
+    end
+    
+    # Returns a string representation of the TimeOrDateTime.
+    def to_s
+      if @orig.is_a?(Time)
+        "Time: #{@orig.to_s}"
+      elsif @orig.is_a?(DateTime)
+        "DateTime: #{@orig.to_s}"
+      else
+        "Timestamp: #{@orig.to_s}"
+      end
+    end
+    
+    # Returns internal object state as a programmer-readable string.
+    def inspect
+      "#<#{self.class}: #{@orig.inspect}>"
+    end
+    
+    # Returns the year.
+    def year
+      if @time
+        @time.year
+      elsif @datetime
+        @datetime.year
+      else
+        to_time.year
+      end
+    end
+    
+    # Returns the month of the year (1..12).
+    def mon
+      if @time
+        @time.mon
+      elsif @datetime
+        @datetime.mon
+      else
+        to_time.mon
+      end
+    end
+    alias :month :mon
+    
+    # Returns the day of the month (1..n).
+    def mday
+      if @time
+        @time.mday
+      elsif @datetime
+        @datetime.mday
+      else
+        to_time.mday
+      end
+    end
+    alias :day :mday
+    
+    # Returns the hour of the day (0..23).
+    def hour
+      if @time
+        @time.hour
+      elsif @datetime
+        @datetime.hour
+      else
+        to_time.hour
+      end
+    end
+    
+    # Returns the minute of the hour (0..59).
+    def min
+      if @time
+        @time.min
+      elsif @datetime
+        @datetime.min
+      else
+        to_time.min
+      end
+    end
+    
+    # Returns the second of the minute (0..60). (60 for a leap second).
+    def sec
+      if @time
+        @time.sec
+      elsif @datetime
+        @datetime.sec
+      else
+        to_time.sec
+      end
+    end
+    
+    # Returns the number of microseconds for the time.
+    def usec      
+      if @time
+        @time.usec
+      elsif @datetime
+        # Ruby 1.8 has sec_fraction (of which the documentation says
+        # 'I do NOT recommend you to use this method'). sec_fraction no longer
+        # exists in Ruby 1.9.
+        
+        # Calculate the sec_fraction from the day_fraction.
+        ((@datetime.day_fraction - OffsetRationals.rational_for_offset(@datetime.hour * 3600 + @datetime.min * 60 + @datetime.sec)) * 86400000000).to_i
+      else 
+        0
+      end
+    end
+    
+    # Compares this TimeOrDateTime with another Time, DateTime, timestamp 
+    # (Integer) or TimeOrDateTime. Returns -1, 0 or +1 depending 
+    # whether the receiver is less than, equal to, or greater than 
+    # timeOrDateTime.
+    #
+    # Returns nil if the passed in timeOrDateTime is not comparable with 
+    # TimeOrDateTime instances.
+    #
+    # Comparisons involving a DateTime will be performed using DateTime#<=>.
+    # Comparisons that don't involve a DateTime, but include a Time will be
+    # performed with Time#<=>. Otherwise comparisons will be performed with
+    # Integer#<=>.    
+    def <=>(timeOrDateTime)
+      return nil unless timeOrDateTime.is_a?(TimeOrDateTime) || 
+                        timeOrDateTime.is_a?(Time) ||
+                        timeOrDateTime.is_a?(DateTime) ||
+                        timeOrDateTime.respond_to?(:to_i)
+    
+      unless timeOrDateTime.is_a?(TimeOrDateTime)
+        timeOrDateTime = TimeOrDateTime.wrap(timeOrDateTime)
+      end
+          
+      orig = timeOrDateTime.to_orig
+      
+      if @orig.is_a?(DateTime) || orig.is_a?(DateTime)
+        # If either is a DateTime, assume it is there for a reason 
+        # (i.e. for its larger range of acceptable values on 32-bit systems).
+        to_datetime <=> timeOrDateTime.to_datetime
+      elsif @orig.is_a?(Time) || orig.is_a?(Time)
+        to_time <=> timeOrDateTime.to_time
+      else
+        to_i <=> timeOrDateTime.to_i
+      end
+    end
+    
+    # Adds a number of seconds to the TimeOrDateTime. Returns a new 
+    # TimeOrDateTime, preserving what the original constructed type was.
+    # If the original type is a Time and the resulting calculation goes out of
+    # range for Times, then an exception will be raised by the Time class.
+    def +(seconds)
+      if seconds == 0
+        self
+      else
+        if @orig.is_a?(DateTime)
+          TimeOrDateTime.new(@orig + OffsetRationals.rational_for_offset(seconds))
+        else
+          # + defined for Time and Integer
+          TimeOrDateTime.new(@orig + seconds)
+        end
+      end
+    end
+    
+    # Subtracts a number of seconds from the TimeOrDateTime. Returns a new 
+    # TimeOrDateTime, preserving what the original constructed type was.
+    # If the original type is a Time and the resulting calculation goes out of
+    # range for Times, then an exception will be raised by the Time class.
+    def -(seconds)
+      self + (-seconds)
+    end
+   
+    # Similar to the + operator, but converts to a DateTime based TimeOrDateTime
+    # where the  Time or Integer timestamp to go out of the allowed range for a 
+    # Time, converts to a DateTime based TimeOrDateTime.
+    #
+    # Note that the range of Time varies based on the platform.
+    def add_with_convert(seconds)
+      if seconds == 0
+        self
+      else
+        if @orig.is_a?(DateTime)
+          TimeOrDateTime.new(@orig + OffsetRationals.rational_for_offset(seconds))
+        else
+          # A Time or timestamp.
+          result = to_i + seconds
+          
+          if ((result > 2147483647 || result < -2147483648) && !RubyCoreSupport.time_supports_64bit) || (result < 0 && !RubyCoreSupport.time_supports_negative)
+            result = TimeOrDateTime.new(to_datetime + OffsetRationals.rational_for_offset(seconds))
+          else
+            result = TimeOrDateTime.new(@orig + seconds)
+          end
+        end
+      end
+    end
+    
+    # Returns true if todt represents the same time and was originally 
+    # constructed with the same type (DateTime, Time or timestamp) as this 
+    # TimeOrDateTime.
+    def eql?(todt)
+      todt.kind_of?(TimeOrDateTime) && to_orig.eql?(todt.to_orig)      
+    end
+    
+    # Returns a hash of this TimeOrDateTime.
+    def hash
+      @orig.hash
+    end
+    
+    # If no block is given, returns a TimeOrDateTime wrapping the given 
+    # timeOrDateTime. If a block is specified, a TimeOrDateTime is constructed
+    # and passed to the block. The result of the block must be a TimeOrDateTime.
+    #
+    # The result of the block will be converted to the type of the originally 
+    # passed in timeOrDateTime and then returned as the result of wrap.
+    #
+    # timeOrDateTime can be a Time, DateTime, timestamp (Integer) or 
+    # TimeOrDateTime. If a TimeOrDateTime is passed in, no new TimeOrDateTime 
+    # will be constructed and the value passed to wrap will be used when 
+    # calling the block.
+    def self.wrap(timeOrDateTime)      
+      t = timeOrDateTime.is_a?(TimeOrDateTime) ? timeOrDateTime : TimeOrDateTime.new(timeOrDateTime)        
+      
+      if block_given?
+        t = yield t
+        
+        if timeOrDateTime.is_a?(TimeOrDateTime)
+          t
+        elsif timeOrDateTime.is_a?(Time)
+          t.to_time
+        elsif timeOrDateTime.is_a?(DateTime)
+          t.to_datetime
+        else
+          t.to_i
+        end        
+      else
+        t
+      end
+    end
+  end
+end