montrose 0.1.0

data/lib/montrose/rule/after.rb

@@ -0,0 +1,27 @@
+module Montrose
+  module Rule
+    class After
+      include Montrose::Rule
+
+      def self.apply_options(opts)
+        opts[:starts]
+      end
+
+      # Initializes rule
+      #
+      # @param [Time] start_time - lower bound timestamp
+      #
+      def initialize(start_time)
+        @start_time = start_time
+      end
+
+      def include?(time)
+        time >= @start_time
+      end
+
+      def continue?
+        false
+      end
+    end
+  end
+end

data/lib/montrose/rule/before.rb

@@ -0,0 +1,23 @@
+module Montrose
+  module Rule
+    class Before
+      include Montrose::Rule
+
+      def self.apply_options(opts)
+        opts[:until]
+      end
+
+      def initialize(end_time)
+        @end_time = end_time
+      end
+
+      def include?(time)
+        time < @end_time
+      end
+
+      def continue?
+        false
+      end
+    end
+  end
+end

data/lib/montrose/rule/day_of_month.rb

@@ -0,0 +1,31 @@
+module Montrose
+  module Rule
+    class DayOfMonth
+      include Montrose::Rule
+
+      def self.apply_options(opts)
+        opts[:mday]
+      end
+
+      # Initializes rule
+      #
+      # @param [Array<Fixnum>] days - valid days of month, i.e. [1, 2, -1]
+      #
+      def initialize(days)
+        @days = days
+      end
+
+      def include?(time)
+        @days.include?(time.mday) || included_from_end_of_month?(time)
+      end
+
+      private
+
+      # matches days specified at negative numbers
+      def included_from_end_of_month?(time)
+        month_days = Time.days_in_month(time.month, time.year) # given by activesupport
+        @days.any? { |d| month_days + d + 1 == time.mday }
+      end
+    end
+  end
+end

data/lib/montrose/rule/day_of_week.rb

@@ -0,0 +1,23 @@
+module Montrose
+  module Rule
+    class DayOfWeek
+      include Montrose::Rule
+
+      def self.apply_options(opts)
+        opts[:day]
+      end
+
+      # Initializes rule
+      #
+      # @param [Array<Fixnum>] days - valid days of week, e.g. [1, 2, 7]
+      #
+      def initialize(days)
+        @days = days
+      end
+
+      def include?(time)
+        @days.include?(time.wday)
+      end
+    end
+  end
+end

data/lib/montrose/rule/day_of_year.rb

@@ -0,0 +1,37 @@
+module Montrose
+  module Rule
+    class DayOfYear
+      include Montrose::Rule
+
+      def self.apply_options(opts)
+        opts[:yday]
+      end
+
+      # Initializes rule
+      #
+      # @param [Array<Fixnum>] days - valid days of year, e.g. [1, 2, -1]
+      #
+      def initialize(days)
+        @days = days
+      end
+
+      def include?(time)
+        @days.include?(time.yday) || included_from_end_of_month?(time)
+      end
+
+      private
+
+      def included_from_end_of_month?(time)
+        year_days = days_in_year(time.year) # given by activesupport
+        @days.any? { |d| year_days + d + 1 == time.yday }
+      end
+
+      # Returns the number of days in the given year.
+      # If no year is specified, it will use the current year.
+      # https://github.com/rails/rails/pull/22244
+      def days_in_year(year)
+        Time.days_in_month(2, year) + 337
+      end
+    end
+  end
+end

data/lib/montrose/rule/hour_of_day.rb

@@ -0,0 +1,23 @@
+module Montrose
+  module Rule
+    class HourOfDay
+      include Montrose::Rule
+
+      def self.apply_options(opts)
+        opts[:hour]
+      end
+
+      # Initializes rule
+      #
+      # @param [Array<Fixnum>] hour - valid hours of days, e.g. [1, 2, 24]
+      #
+      def initialize(hours)
+        @hours = hours
+      end
+
+      def include?(time)
+        @hours.include?(time.hour)
+      end
+    end
+  end
+end

data/lib/montrose/rule/month_of_year.rb

@@ -0,0 +1,23 @@
+module Montrose
+  module Rule
+    class MonthOfYear
+      include Montrose::Rule
+
+      def self.apply_options(opts)
+        opts[:month]
+      end
+
+      # Initializes rule
+      #
+      # @param [Array] months - valid month numbers
+      #
+      def initialize(months)
+        @months = months
+      end
+
+      def include?(time)
+        @months.include?(time.month)
+      end
+    end
+  end
+end

data/lib/montrose/rule/nth_day_matcher.rb

@@ -0,0 +1,32 @@
+module Montrose
+  module Rule
+    class NthDayMatcher
+      extend Forwardable
+
+      def_delegators :@period_day, :nth_day, :first_wday, :total_days
+
+      def initialize(wday, period_day)
+        @wday = wday
+        @period_day = period_day
+      end
+
+      def matches?(nth_occ)
+        nth_occ == current_occ || (nth_occ < 0 && (total_occ + nth_occ + 1) == current_occ)
+      end
+
+      private
+
+      def current_occ
+        @current_occ ||= (nth_day - first_occ) / 7 + 1
+      end
+
+      def total_occ
+        @total_occ ||= ((total_days - first_occ + 1) / 7.0).ceil
+      end
+
+      def first_occ
+        @first_occ ||= ((7 - first_wday) + @wday) % 7 + 1
+      end
+    end
+  end
+end

data/lib/montrose/rule/nth_day_of_month.rb

@@ -0,0 +1,63 @@
+require "montrose/rule/nth_day_matcher"
+
+module Montrose
+  module Rule
+    class NthDayOfMonth
+      include Montrose::Rule
+
+      def self.apply_options?(opts)
+        opts[:every] == :month && opts[:day].is_a?(Hash)
+      end
+
+      def self.apply_options(opts)
+        opts[:day]
+      end
+
+      # Initializes rule
+      #
+      # @param [Hash] days - valid days of week to month occurrence pairs
+      #
+      def initialize(days)
+        @days = days
+      end
+
+      def include?(time)
+        @days.key?(time.wday) && nth_day?(time)
+      end
+
+      private
+
+      def nth_day?(time)
+        expected_occurrences = @days[time.wday]
+        nth_day = NthDayMatcher.new(time.wday, MonthDay.new(time))
+        expected_occurrences.any? { |n| nth_day.matches?(n) }
+      end
+
+      class MonthDay
+        def initialize(time)
+          @time = time
+        end
+
+        def nth_day
+          @time.mday
+        end
+
+        def first_wday
+          @time.beginning_of_month.wday
+        end
+
+        def total_days
+          days_in_month(@time)
+        end
+
+        private
+
+        # Get the days in the month for +time
+        def days_in_month(time)
+          date = Date.new(time.year, time.month, 1)
+          ((date >> 1) - date).to_i
+        end
+      end
+    end
+  end
+end

data/lib/montrose/rule/nth_day_of_year.rb

@@ -0,0 +1,63 @@
+require "montrose/rule/nth_day_matcher"
+
+module Montrose
+  module Rule
+    class NthDayOfYear
+      include Montrose::Rule
+
+      def self.apply_options?(opts)
+        opts[:every] == :year && opts[:day].is_a?(Hash)
+      end
+
+      def self.apply_options(opts)
+        opts[:day]
+      end
+
+      # Initializes rule
+      #
+      # @param [Hash] days - valid days of week to year occurrence pairs
+      #
+      def initialize(days)
+        @days = days
+      end
+
+      def include?(time)
+        @days.key?(time.wday) && nth_day?(time)
+      end
+
+      private
+
+      def nth_day?(time)
+        expected_occurrences = @days[time.wday]
+        nth_day = NthDayMatcher.new(time.wday, YearDay.new(time))
+        expected_occurrences.any? { |n| nth_day.matches?(n) }
+      end
+
+      class YearDay
+        def initialize(time)
+          @time = time
+        end
+
+        def nth_day
+          @time.yday
+        end
+
+        def first_wday
+          @time.beginning_of_year.wday
+        end
+
+        def total_days
+          days_in_year(@time)
+        end
+
+        private
+
+        # Get the days in the month for +time
+        def days_in_year(time)
+          date = time.to_date
+          ((date + 1.year) - date).to_i
+        end
+      end
+    end
+  end
+end

data/lib/montrose/rule/time_of_day.rb

@@ -0,0 +1,33 @@
+module Montrose
+  module Rule
+    class TimeOfDay
+      include Montrose::Rule
+
+      def self.apply_options(opts)
+        opts[:at]
+      end
+
+      # Initializes rule
+      #
+      # @param [Array<Time>] times - valid hours of days
+      #
+      def initialize(times)
+        @times = times
+      end
+
+      def include?(time)
+        times_of_day.include?(parts(time))
+      end
+
+      private
+
+      def parts(time)
+        [time.hour, time.min]
+      end
+
+      def times_of_day
+        @times_of_day ||= @times.map { |t| parts(t) }
+      end
+    end
+  end
+end

data/lib/montrose/rule/total.rb

@@ -0,0 +1,29 @@
+module Montrose
+  module Rule
+    class Total
+      include Montrose::Rule
+
+      def self.apply_options(opts)
+        opts[:total]
+      end
+
+      def initialize(max)
+        @max = max
+        @count = 0
+      end
+
+      def include?(_time)
+        continue?
+      end
+
+      def advance!(_time)
+        @count += 1
+        continue?
+      end
+
+      def continue?
+        @count <= @max
+      end
+    end
+  end
+end

data/lib/montrose/rule/week_of_year.rb

@@ -0,0 +1,23 @@
+module Montrose
+  module Rule
+    class WeekOfYear
+      include Montrose::Rule
+
+      def self.apply_options(opts)
+        opts[:week]
+      end
+
+      # Initializes rule
+      #
+      # @param [Array[Fixnum]] weeks - valid weeks of year
+      #
+      def initialize(weeks)
+        @weeks = weeks
+      end
+
+      def include?(time)
+        @weeks.include?(time.to_date.cweek)
+      end
+    end
+  end
+end