by_star 3.0.0 → 4.0.1

data/lib/by_star/directional.rb

@@ -1,37 +1,35 @@
-module ByStar
-
-  module Directional
-
-    def before(*args)
-      with_by_star_options(*args) do |time, options|
-        scope = by_star_scope(options)
-        field = by_star_start_field(options)
-        time = ByStar::Normalization.time(time)
-        by_star_before_query(scope, field, time)
-      end
-    end
-    alias_method :before_now, :before
-
-    def after(*args)
-      with_by_star_options(*args) do |time, options|
-        scope = by_star_scope(options)
-        field = by_star_start_field(options)
-        time = ByStar::Normalization.time(time)
-        by_star_after_query(scope, field, time)
-      end
-    end
-    alias_method :after_now, :after
-
-    def oldest(*args)
-      with_by_star_options(*args) do |time, options|
-        oldest_query(options)
-      end
-    end
-
-    def newest(*args)
-      with_by_star_options(*args) do |time, options|
-        newest_query(options)
-      end
-    end
-  end
-end
+module ByStar
+
+  module Directional
+
+    def before(*args)
+      with_by_star_options(*args) do |time, options|
+        field = by_star_start_field(options)
+        time = ByStar::Normalization.time(time)
+        by_star_before_query(self, field, time)
+      end
+    end
+    alias_method :before_now, :before
+
+    def after(*args)
+      with_by_star_options(*args) do |time, options|
+        field = by_star_start_field(options)
+        time = ByStar::Normalization.time(time)
+        by_star_after_query(self, field, time)
+      end
+    end
+    alias_method :after_now, :after
+
+    def oldest(*args)
+      with_by_star_options(*args) do |time, options|
+        oldest_query(options)
+      end
+    end
+
+    def newest(*args)
+      with_by_star_options(*args) do |time, options|
+        newest_query(options)
+      end
+    end
+  end
+end

data/lib/by_star/kernel/date.rb

@@ -1,50 +1,41 @@
-module ByStar
-
-  module Kernel
-
-    module Date
-      extend ActiveSupport::Concern
-
-      included do
-
-        # Shim to alias Rails 3 #to_time_in_current_zone to Rails 4 name #in_time_zone
-        if method_defined?(:to_time_in_current_zone) && !method_defined?(:in_time_zone)
-          alias_method :in_time_zone, :to_time_in_current_zone
-        end
-      end
-
-      # A "Weekend" is defined as beginning of Saturday to end of Sunday.
-      # The weekend for a given date will be the the next weekend if the day Mon-Thurs,
-      # otherwise the current weekend if the day is Fri-Sun.
-      def beginning_of_weekend
-        beginning_of_week(:monday).advance(days: 5)
-      end
-
-      def end_of_weekend
-        beginning_of_weekend + 1
-      end
-
-      # A "Fortnight" is defined as a two week period, with the first fortnight of the
-      # year beginning on 1st January.
-      def beginning_of_fortnight
-        beginning_of_year + 14 * ((self - beginning_of_year) / 14).to_i
-      end
-
-      def end_of_fortnight
-        beginning_of_fortnight + 13
-      end
-
-      # A "Calendar Month" is defined as a month as it appears on a calendar, including days form
-      # previous/following months which are part of the first/last weeks of the given month.
-      def beginning_of_calendar_month(*args)
-        beginning_of_month.beginning_of_week(*args)
-      end
-
-      def end_of_calendar_month(*args)
-        end_of_month.end_of_week(*args)
-      end
-    end
-  end
-end
-
-::Date.__send__(:include, ByStar::Kernel::Date)
+module ByStar
+
+  module Kernel
+
+    module Date
+
+      # A "Weekend" is defined as beginning of Saturday to end of Sunday.
+      # The weekend for a given date will be the the next weekend if the day Mon-Thurs,
+      # otherwise the current weekend if the day is Fri-Sun.
+      def beginning_of_weekend
+        beginning_of_week(:monday).advance(days: 5)
+      end
+
+      def end_of_weekend
+        beginning_of_weekend + 1
+      end
+
+      # A "Fortnight" is defined as a two week period, with the first fortnight of the
+      # year beginning on 1st January.
+      def beginning_of_fortnight
+        beginning_of_year + 14 * ((self - beginning_of_year) / 14).to_i
+      end
+
+      def end_of_fortnight
+        beginning_of_fortnight + 13
+      end
+
+      # A "Calendar Month" is defined as a month as it appears on a calendar, including days form
+      # previous/following months which are part of the first/last weeks of the given month.
+      def beginning_of_calendar_month(*args)
+        beginning_of_month.beginning_of_week(*args)
+      end
+
+      def end_of_calendar_month(*args)
+        end_of_month.end_of_week(*args)
+      end
+    end
+  end
+end
+
+::Date.__send__(:include, ByStar::Kernel::Date)

data/lib/by_star/kernel/in_time_zone.rb

@@ -0,0 +1,20 @@
+module ByStar
+
+  module Kernel
+
+    module InTimeZone
+      extend ActiveSupport::Concern
+
+      included do
+        if method_defined?(:to_time_in_current_zone) && !method_defined?(:in_time_zone)
+          alias_method :in_time_zone, :to_time_in_current_zone
+        end
+      end
+    end
+  end
+end
+
+::Date.__send__(:include, ByStar::Kernel::InTimeZone)
+::Time.__send__(:include, ByStar::Kernel::InTimeZone)
+::DateTime.__send__(:include, ByStar::Kernel::InTimeZone)
+::ActiveSupport::TimeWithZone.__send__(:include, ByStar::Kernel::InTimeZone)

data/lib/by_star/kernel/time.rb

@@ -1,41 +1,41 @@
-module ByStar
-
-  module Kernel
-
-    module Time
-
-      # A "Weekend" is defined as beginning of Saturday to end of Sunday.
-      # The weekend for a given date will be the the next weekend if the day Mon-Thurs,
-      # otherwise the current weekend if the day is Fri-Sun.
-      def beginning_of_weekend
-        beginning_of_week(:monday).advance(days: 5)
-      end
-
-      def end_of_weekend
-        (beginning_of_weekend + 47.hours).end_of_hour
-      end
-
-      # A "Fortnight" is defined as a two week period, with the first fortnight of the
-      # year beginning on 1st January.
-      def beginning_of_fortnight
-        (beginning_of_year + 1.fortnight * ((self - beginning_of_year) / 1.fortnight).to_i).beginning_of_day
-      end
-
-      def end_of_fortnight
-        (beginning_of_fortnight + 13.days).end_of_day
-      end
-
-      # A "Calendar Month" is defined as a month as it appears on a calendar, including days form
-      # previous/following months which are part of the first/last weeks of the given month.
-      def beginning_of_calendar_month(*args)
-        beginning_of_month.beginning_of_week(*args)
-      end
-
-      def end_of_calendar_month(*args)
-        end_of_month.end_of_week(*args)
-      end
-    end
-  end
-end
-
-::Time.__send__(:include, ByStar::Kernel::Time)
+module ByStar
+
+  module Kernel
+
+    module Time
+
+      # A "Weekend" is defined as beginning of Saturday to end of Sunday.
+      # The weekend for a given date will be the the next weekend if the day Mon-Thurs,
+      # otherwise the current weekend if the day is Fri-Sun.
+      def beginning_of_weekend
+        beginning_of_week(:monday).advance(days: 5)
+      end
+
+      def end_of_weekend
+        (beginning_of_weekend + 47.hours).end_of_hour
+      end
+
+      # A "Fortnight" is defined as a two week period, with the first fortnight of the
+      # year beginning on 1st January.
+      def beginning_of_fortnight
+        (beginning_of_year + 1.fortnight * ((self - beginning_of_year) / 1.fortnight).to_i).beginning_of_day
+      end
+
+      def end_of_fortnight
+        (beginning_of_fortnight + 13.days).end_of_day
+      end
+
+      # A "Calendar Month" is defined as a month as it appears on a calendar, including days form
+      # previous/following months which are part of the first/last weeks of the given month.
+      def beginning_of_calendar_month(*args)
+        beginning_of_month.beginning_of_week(*args)
+      end
+
+      def end_of_calendar_month(*args)
+        end_of_month.end_of_week(*args)
+      end
+    end
+  end
+end
+
+::Time.__send__(:include, ByStar::Kernel::Time)

data/lib/by_star/normalization.rb

@@ -1,127 +1,156 @@
-module ByStar
-
-  class ParseError < StandardError; end
-
-  module Normalization
-
-    class << self
-
-      def time(value)
-        case value
-          when String   then time_string(value)
-          when DateTime then value.to_time
-          when Date     then value.in_time_zone
-          else value
-        end
-      end
-
-      def time_string(value)
-        defined?(Chronic) ? time_string_chronic(value) : time_string_fallback(value)
-      end
-
-      def time_string_chronic(value)
-        Chronic.time_class = Time.zone
-        Chronic.parse(value) || raise(ByStar::ParseError, "Chronic could not parse String #{value.inspect}")
-      end
-
-      def time_string_fallback(value)
-        Time.zone.parse(value) || raise(ByStar::ParseError, "Cannot parse String #{value.inspect}")
-      end
-
-      def week(value, options={})
-        value = try_string_to_int(value)
-        case value
-          when Integer then week_integer(value, options)
-          else time(value)
-        end
-      end
-
-      def week_integer(value, options={})
-        raise ParseError, 'Week number must be between 0 and 52' unless value.in?(0..52)
-        time = Time.zone.local(options[:year] || Time.zone.now.year)
-        time.beginning_of_year + value.to_i.weeks
-      end
-
-      def cweek(value, options={})
-        _value = value
-        if _value.is_a?(Integer)
-          raise ParseError, 'cweek number must be between 1 and 53' unless value.in?(1..53)
-          _value -= 1
-        end
-        week(_value, options)
-      end
-
-      def fortnight(value, options={})
-        value = try_string_to_int(value)
-        case value
-          when Integer then fortnight_integer(value, options)
-          else time(value)
-        end
-      end
-
-      def fortnight_integer(value, options={})
-        raise ParseError, 'Fortnight number must be between 0 and 26' unless value.in?(0..26)
-        time = Time.zone.local(options[:year] || Time.zone.now.year)
-        time + (value * 2).weeks
-      end
-
-      def quarter(value, options={})
-        value = try_string_to_int(value)
-        case value
-          when Integer then quarter_integer(value, options)
-          else time(value)
-        end
-      end
-
-      def quarter_integer(value, options={})
-        raise ParseError, 'Quarter number must be between 1 and 4' unless value.in?(1..4)
-        time = Time.zone.local(options[:year] || Time.zone.now.year)
-        time.beginning_of_year + ((value - 1) * 3).months
-      end
-
-      def month(value, options={})
-        value = try_string_to_int(value)
-        case value
-          when Integer, String then month_integer(value, options)
-          else time(value)
-        end
-      end
-
-      def month_integer(value, options={})
-        year = options[:year] || Time.zone.now.year
-        Time.zone.parse "#{year}-#{value}-01"
-      rescue
-        raise ParseError, 'Month must be a number between 1 and 12 or a month name'
-      end
-
-      def year(value, options={})
-        value = try_string_to_int(value)
-        case value
-          when Integer then year_integer(value)
-          else time(value)
-        end
-      end
-
-      def year_integer(value)
-        Time.zone.local(extrapolate_year(value))
-      end
-
-      def extrapolate_year(value)
-        case value.to_i
-          when 0..69
-            2000 + value
-          when 70..99
-            1900 + value
-          else
-            value.to_i
-        end
-      end
-
-      def try_string_to_int(value)
-        value.is_a?(String) ? Integer(value) : value
-      rescue
-        value
-      end
-    end
-  end
-end
+module ByStar
+
+  class ParseError < StandardError; end
+
+  module Normalization
+
+    class << self
+
+      def date(value)
+        value = parse_time(value) if value.is_a?(String)
+        value = value.try(:in_time_zone) unless value.is_a?(Date)
+        value.try(:to_date)
+      end
+
+      def time(value)
+        value = parse_time(value) if value.is_a?(String)
+        value.try(:in_time_zone)
+      end
+
+      def week(value, options={})
+        value = try_string_to_int(value)
+        case value
+          when Integer then week_integer(value, options)
+          else date(value)
+        end
+      end
+
+      def week_integer(value, options={})
+        raise ParseError, 'Week number must be between 0 and 52' unless value.in?(0..52)
+        time = Time.zone.local(options[:year] || Time.zone.now.year)
+        time.beginning_of_year + value.to_i.weeks
+      end
+
+      def cweek(value, options={})
+        _value = value
+        if _value.is_a?(Integer)
+          raise ParseError, 'cweek number must be between 1 and 53' unless value.in?(1..53)
+          _value -= 1
+        end
+        week(_value, options)
+      end
+
+      def fortnight(value, options={})
+        value = try_string_to_int(value)
+        case value
+          when Integer then fortnight_integer(value, options)
+          else date(value)
+        end
+      end
+
+      def fortnight_integer(value, options={})
+        raise ParseError, 'Fortnight number must be between 0 and 26' unless value.in?(0..26)
+        time = Time.zone.local(options[:year] || Time.zone.now.year)
+        time + (value * 2).weeks
+      end
+
+      def quarter(value, options={})
+        value = try_string_to_int(value)
+        case value
+          when Integer then quarter_integer(value, options)
+          else date(value)
+        end
+      end
+
+      def quarter_integer(value, options={})
+        raise ParseError, 'Quarter number must be between 1 and 4' unless value.in?(1..4)
+        time = Time.zone.local(options[:year] || Time.zone.now.year)
+        time.beginning_of_year + ((value - 1) * 3).months
+      end
+
+      def month(value, options={})
+        value = try_string_to_int(value)
+        case value
+          when Integer, String then month_integer(value, options)
+          else date(value)
+        end
+      end
+
+      def month_integer(value, options={})
+        year = options[:year] || Time.zone.now.year
+        Time.zone.parse "#{year}-#{value}-01"
+      rescue
+        raise ParseError, 'Month must be a number between 1 and 12 or a month name'
+      end
+
+      def year(value, options={})
+        value = try_string_to_int(value)
+        case value
+          when Integer then year_integer(value)
+          else date(value)
+        end
+      end
+
+      def year_integer(value)
+        Time.zone.local(extrapolate_year(value))
+      end
+
+      def extrapolate_year(value)
+        case value.to_i
+          when 0..69
+            2000 + value
+          when 70..99
+            1900 + value
+          else
+            value.to_i
+        end
+      end
+
+      def try_string_to_int(value)
+        value.is_a?(String) ? Integer(value) : value
+      rescue
+        value
+      end
+
+      def time_in_units(seconds)
+        days = seconds / 1.day
+        time = Time.at(seconds).utc
+        { days: days, hour: time.hour, min: time.min, sec: time.sec }
+      end
+
+      def apply_offset_start(time, offset)
+        units = time_in_units(offset)
+        time += units.delete(:days).days
+        time.change(units)
+      end
+
+      def apply_offset_end(time, offset)
+        units = time_in_units(offset)
+        time += units.delete(:days).days
+        (time + 1.day).change(units) - 1.second
+      end
+
+      def extract_range(args)
+        case args[0]
+        when Array, Range then [args[0].first, args[0].last]
+        else args[0..1]
+        end
+      end
+
+      private
+
+      def parse_time(value)
+        defined?(Chronic) ? parse_time_chronic(value) : parse_time_fallback(value)
+      end
+
+      def parse_time_chronic(value)
+        Chronic.time_class = Time.zone
+        Chronic.parse(value) || raise(ByStar::ParseError, "Chronic could not parse String #{value.inspect}")
+      end
+
+      def parse_time_fallback(value)
+        Time.zone.parse(value) || raise(ByStar::ParseError, "Cannot parse String #{value.inspect}")
+      end
+    end
+  end
+end