ManageIQ/manageiq

class MiqExpression::RelativeDatetime
  def self.relative?(value)
    v = value.downcase
    v.starts_with?("this", "last") || v.ends_with?("ago") || ["today", "yesterday", "now"].include?(v)
  end

  def self.normalize(rel_time, tz, mode = "beginning", is_date = nil)
    # time_spec =
    #   <value> <interval> Ago
    #   "Today"
    #   "Yesterday"
    #   "Now"
    #   "Last Week"
    #   "Last Month"
    #   "Last Quarter"
    #   "This Week"
    #   "This Month"
    #   "This Quarter"

    rt = rel_time.downcase

    if rt.starts_with?("this", "last")
      # Convert these into the time spec form: <value> <interval> Ago
      value, interval = rt.split
      rt = "#{value == "this" ? 0 : 1} #{interval} ago"
    end

    if rt.ends_with?("ago")
      # Time spec <value> <interval> Ago
      value, interval, _ago = rt.split
      interval = interval.pluralize

      if interval == "quarters"
        ts = Time.now.in_time_zone(tz).beginning_of_quarter
        coerce((ts - (value.to_i * 3.months)).send(:"#{mode}_of_quarter"), is_date)
      else
        coerce(value.to_i.send(interval).ago.in_time_zone(tz).send(:"#{mode}_of_#{interval.singularize}"), is_date)
      end
    elsif rt == "today"
      coerce(Time.now.in_time_zone(tz).send(:"#{mode}_of_day"), is_date)
    elsif rt == "yesterday"
      coerce(1.day.ago.in_time_zone(tz).send(:"#{mode}_of_day"), is_date)
    elsif rt == "now"
      t = Time.now.in_time_zone(tz)
      coerce(mode == "beginning" ? t.beginning_of_hour : t.end_of_hour, is_date)
    else
      # Assume it's an absolute date or time
      value_is_date = !rel_time.include?(":")
      ts = Time.use_zone(tz) { Time.zone.parse(rel_time) }
      ts = ts.send(:"#{mode}_of_day") if mode && value_is_date
      coerce(ts, is_date)
    end
  end

  def self.coerce(value, is_date)
    return value if is_date.nil?

    is_date ? value.to_date : value.utc
  end
end