enumerable.rb 961 Bytes
module Enumerable
  Groupdate::PERIODS.each do |period|
    define_method :"group_by_#{period}" do |*args, &block|
      if block
        Groupdate::Magic.new(period, args[0] || {}).group_by(self, &block)
      elsif respond_to?(:scoping)
        scoping { @klass.send(:"group_by_#{period}", *args, &block) }
      else
        raise ArgumentError, "no block given"
      end
    end
  end

  def group_by_period(*args, &block)
    if block || !respond_to?(:scoping)
      period = args[0]
      options = args[1] || {}

      # to_sym is unsafe on user input, so convert to strings
      permitted_periods = ((options[:permit] || Groupdate::PERIODS).map(&:to_sym) & Groupdate::PERIODS).map(&:to_s)
      if permitted_periods.include?(period.to_s)
        send("group_by_#{period}", options, &block)
      else
        raise ArgumentError, "Unpermitted period"
      end
    else
      scoping { @klass.send(:group_by_period, *args, &block) }
    end
  end
end