twitter/twitter-cldr-rb

View on GitHub
lib/twitter_cldr/localized/localized_datetime.rb

Summary

Maintainability
A
1 hr
Test Coverage
# encoding: UTF-8

# Copyright 2012 Twitter, Inc
# http://www.apache.org/licenses/LICENSE-2.0

require 'tzinfo'

module TwitterCldr
  module Localized
    class LocalizedDateTime < LocalizedObject

      attr_reader :calendar_type, :timezone

      class << self
        def types
          TwitterCldr::DataReaders::CalendarDataReader.types
        end
      end

      def initialize(obj, locale, options = {})
        super
        @calendar_type = options[:calendar_type] || TwitterCldr::DEFAULT_CALENDAR_TYPE
        @timezone = options[:timezone] || "UTC"
      end

      types.each do |type|
        define_method "to_#{type}_s" do
          # @ TODO: these need to be cheap to create
          data_reader = data_reader_for(type)
          tokens = data_reader.tokenizer.tokenize(data_reader.pattern)
          data_reader.formatter.format(tokens, base_in_timezone, chain_params)
        end
      end

      def to_additional_s(additional_format)
        data_reader = data_reader_for(:additional, {
          additional_format: additional_format
        })

        tokens = if data_reader.tokenizer.respond_to?(:full_tokenize)
          data_reader.tokenizer.full_tokenize(data_reader.pattern)
        else
          data_reader.tokenizer.tokenize(data_reader.pattern)
        end

        data_reader.formatter.format(tokens, base_in_timezone, chain_params)
      end

      alias :to_default_s :to_medium_s

      def to_timespan(options = {})
        base_time = options[:base_time] || Time.now
        seconds = (self.to_time.base_obj.to_i - base_time.to_i).abs
        TwitterCldr::Localized::LocalizedTimespan.new(seconds, options.merge(locale: @locale, direction: :none))
      end

      def ago(options = {})
        base_time = (options[:base_time] || Time.now).gmtime
        seconds = self.to_time(base_time).base_obj.gmtime.to_i - base_time.to_i
        raise ArgumentError.new('Start date is after end date. Consider using "until" function.') if seconds > 0
        TwitterCldr::Localized::LocalizedTimespan.new(seconds, options.merge(locale: @locale))
      end

      def until(options = {})
        base_time = (options[:base_time] || Time.now).gmtime
        seconds = self.to_time(base_time).base_obj.gmtime.to_i - base_time.to_i
        raise ArgumentError.new('End date is before start date. Consider using "ago" function.') if seconds < 0
        TwitterCldr::Localized::LocalizedTimespan.new(seconds, options.merge(locale: @locale))
      end

      def additional_formats
        data_reader_for(nil).additional_format_selector.patterns
      end

      def to_s
        to_default_s
      end

      def to_date
        LocalizedDate.new(@base_obj, @locale, chain_params)
      end

      def to_time(base = Time.now)
        utc_dt = @base_obj.new_offset(0)

        time = Time.gm(
          utc_dt.year,
          utc_dt.month,
          utc_dt.day,
          utc_dt.hour,
          utc_dt.min,
          utc_dt.sec,
          utc_dt.sec_fraction * 1_000_000
        )

        LocalizedTime.new(time, @locale, chain_params)
      end

      def with_timezone(timezone)
        self.class.new(@base_obj, @locale, chain_params.merge(timezone: timezone))
      end

      protected

      def data_reader_for(type, options = {})
        TwitterCldr::DataReaders::DateTimeDataReader.new(
          locale, options.merge({
            calendar_type: calendar_type,
            type: type
          })
        )
      end

      def chain_params
        { calendar_type: @calendar_type, timezone: @timezone }
      end

      def base_in_timezone
        timezone_info.utc_to_local(@base_obj.new_offset(0))
      end

      def timezone_info
        (@@timezone_info ||= {})[@timezone] ||= TZInfo::Timezone.get(@timezone)
      end

    end
  end
end