Module: IncludeDateScopes::DefineTimestampScopes

Included in:
IncludeDateScopes::DateScopes::ClassMethods
Defined in:
lib/include_date_scopes/define_timestamp_scopes.rb

Instance Method Summary collapse

Instance Method Details

#define_timestamp_scopes_for(column_name, prepend_name = false) ⇒ Object 



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
 
# File 'lib/include_date_scopes/define_timestamp_scopes.rb', line 4

def define_timestamp_scopes_for(column_name, prepend_name = false)
  prefix = prepend_name ? "#{column_name}_" : ""
  t = self.arel_table

  define_singleton_method :"#{prefix}between" do |start_date_or_time, stop_date_or_time|
    # If start_date_or_time is a Date, the interval INCLUDES the start of the day; otherwise, the interval INCLUDES the argument time.
    # If stop_date_or_time is a Date, the interval INCLUDES the end date; otherwise, the interval EXCLUDES the argument time.
    # For example between(Date.today, Date.today) is equivalent to between(Date.today.midnight, Date.tomorrow.midnight).
    start_time = (start_date_or_time.is_a?(Date) ? start_date_or_time.to_time : start_date_or_time)
    stop_time = (stop_date_or_time.is_a?(Date) ? stop_date_or_time.to_time + 1.day : stop_date_or_time )
    where(t[column_name].gteq(start_time).and(t[column_name].lt stop_time))
  end

  define_singleton_method :"#{prefix}on_or_before_date" do |date|
    where(t[column_name].lt date.to_date.tomorrow.midnight)
  end

  define_singleton_method :"#{prefix}on_or_before" do |time|
    if time.is_a?(Date)
      where(t[column_name].lt time.to_time + 1.day)
    else
      where(t[column_name].lteq time)
    end
  end

  define_singleton_method :"#{prefix}before" do |time|
    time = time.to_time if time.kind_of? Date
    where(t[column_name].lt time)
  end

  define_singleton_method :"#{prefix}on_or_after_date" do |time|
    where(t[column_name].gteq time.to_date.midnight)
  end

  define_singleton_method :"#{prefix}on_or_after" do |time|
    where(t[column_name].gteq time.to_time)
  end

  define_singleton_method :"#{prefix}after" do |time|
    if time.is_a?(Date)
      where(t[column_name].gteq time.to_time + 1.day)
    else
      where(t[column_name].gt time)
    end
  end

  define_singleton_method :"#{prefix}on" do |day| 
    __send__(:"#{prefix}between", day.to_date, day.to_date)
  end

  define_singleton_method :"#{prefix}last_24_hours" do
    __send__(:"#{prefix}last_day")
  end

  [:minute, :hour].each do |time_unit|
    define_singleton_method :"#{prefix}next_#{time_unit}" do
      start_time = Time.now
      __send__(:"#{prefix}between", start_time, start_time + 1.send(time_unit))
    end

    define_singleton_method :"#{prefix}last_#{time_unit}" do
      now = Time.now
      __send__(:"#{prefix}between", now - 1.send(time_unit), now + 1.second)
    end
  end

  [:seconds, :minutes, :hours].each do |time_unit|
    define_singleton_method :"#{prefix}last_n_#{time_unit}" do |count|
      now = Time.now
      __send__(:"#{prefix}between", now - count.send(time_unit), now + 1.second)
    end

    define_singleton_method :"#{prefix}next_n_#{time_unit}" do |count|
      start_time = Time.now
      __send__(:"#{prefix}between", start_time, start_time + count.send(time_unit))
    end
  end

  define_singleton_method :"#{prefix}this_minute" do
    start_time = Time.now.change(sec: 0)
    __send__(:"#{prefix}between", start_time, start_time + 1.minute)
  end

  [:hour, :day, :week, :month, :year].each do |time_unit|
    define_singleton_method :"#{prefix}this_#{time_unit}" do
      start_time = Time.now.send(:"beginning_of_#{time_unit}")
      __send__(:"#{prefix}between", start_time, start_time + 1.send(:"#{time_unit}"))
    end
  end

  define_common_scopes prefix, column_name
end