# encoding: utf-8

require "date"

module DateAndTimeFormatting
  def self.included(base)
    base.class_eval do
      include DateAndTimeFormatting::InstanceMethods
      include OrdinalizedFormatting
      extend DateAndTimeFormatting::ClassMethods
    end
  end

  module InstanceMethods
    # Formats a date/time instance using a defined format
    #
    # ==== Parameters
    # format<Symbol>:: of the format key from Date.date_formats
    #
    # ==== Returns
    # String:: formatted string
    #
    # ==== Example
    #   Time.now.formatted(:rfc822) # => "Sun, 16 Nov 2007 00:21:16 -0800"
    #   Time.now.formatted(:db) # => "2008-11-16 00:22:09"
    #
    # You can also add your own formats using +Date.add_format+ when your app loads.
    #
    # # Add the following to your init.rb
    #   Rango::BootLoader.before_app_loads do
    #     Date.add_format(:matt, "%H:%M:%S %Y-%m-%d")
    #   end
    #
    # # Format a Time instance with the format you just specified
    #   Time.now.formatted(:matt) # => "00:00:00 2007-11-02"
    #
    #--
    # @public
    def formatted(format = :default)
      self.strftime(Date.formats[format])
    end
  end

  module ClassMethods
    @@formats = {
      :db             => "%Y-%m-%d %H:%M:%S",
      :time           => "%H:%M", # 21:12
      :date           => "%Y-%m-%d", # 2008-12-04
      :short          => "%d %b %H:%M", # 01 Sep 21:12
      :long           => "%B %d, %Y %H:%M",
      :rfc822         => "%a, %d %b %Y %H:%M:%S %z"
    }

    # Lists the date and time formats
    #
    # ==== Returns
    # Hash:: a hash with all formats available
    # --
    # @public
    def formats
      @@formats
    end

    # Adds a date and time format
    #
    # Because this operation is not thread safe, you should define
    # custom formats when you load you application. The recommended way
    # to do that, is to use the before_app_loads bootloader.
    #
    # If you want to add a format at runtime, you will need to use a mutex
    # and synchronize it yourself.
    #
    # ==== Parameters
    # key<Symbol>:: name of the format
    # format<Hash>:: time format to use
    #
    # ==== Returns
    # Hash:: a hash with all formats available
    #
    # ==== Example
    #
    #   Rango::BootLoader.before_app_loads do
    #     Date.add_format(:matt, "%H:%M:%S %Y-%m-%d")
    #   end
    #
    #
    # --
    # @public
    def add_format(key, format)
      formats.merge!({key => format})
    end
  end
end

module Ordinalize
  # Ordinalize turns a number into an ordinal string used to denote the
  # position in an ordered sequence such as 1st, 2nd, 3rd, 4th.
  #
  # ==== Examples
  #   1.ordinalize     # => "1st"
  #   2.ordinalize     # => "2nd"
  #   1002.ordinalize  # => "1002nd"
  #   1003.ordinalize  # => "1003rd"
  def ordinalize
    if (11..13).include?(self % 100)
      "#{self}th"
    else
      case self % 10
        when 1; "#{self}st"
        when 2; "#{self}nd"
        when 3; "#{self}rd"
        else    "#{self}th"
      end
    end
  end
end

Integer.send :include, Ordinalize

# Time.now.to_ordinalized_s :long
# => "February 28th, 2006 21:10"
module OrdinalizedFormatting
  def to_ordinalized_s(format = :default)
    format = Date.formats[format]
    return self.to_s if format.nil?
    strftime_ordinalized(format)
  end

  # Gives you a relative date in an attractive format
  #
  # ==== Parameters
  # format<String>:: strftime string used to format a time/date object
  # locale<String, Symbol>:: An optional value which can be used by localization plugins
  #
  # ==== Returns
  # String:: Ordinalized time/date object
  #
  # ==== Examples
  #    5.days.ago.strftime_ordinalized('%b %d, %Y')     # =>
  def strftime_ordinalized(fmt, format = nil)
    strftime(fmt.gsub(/(^|[^-])%d/, '\1_%d_')).gsub(/_(\d+)_/) { |s| s[1..-2].to_i.ordinalize }
  end
end

Time.send(:include, DateAndTimeFormatting)
Date.send(:include, DateAndTimeFormatting)