qreport_time_parser 0.0.2

data/lib/qreport/time_parser/examples.rb

@@ -0,0 +1,64 @@
+require 'qreport/time_parser'
+
+module Qreport
+  class TimeParser
+    def self.examples
+      now = ::Time.parse("2011-03-10T15:10:37.981304-06:00")
+      examples = {
+        "now" => "nil 2011-03-10T15:10:37.981304-06:00",
+        now.to_s => "nil 2011-03-10T15:10:37.000000-06:00",
+        "2011-03-10 15:10:37.981304 -0600" => "nil 2011-03-10T15:10:37.981304-06:00",
+        "today" => ":day 2011-03-10T00:00:00.000000-06:00",
+        "tomorrow" => ":day 2011-03-11T00:00:00.000000-06:00",
+        "yesterday" => ":day 2011-03-09T00:00:00.000000-06:00",
+        "9:15am yesterday" => ":min 2011-03-09T09:15:00.000000-06:00",
+        "yesterday 9:15am" => ":day 2011-03-09T00:00:00.000000-06:00", # FIXME
+        "10 days ago" => ":day 2011-02-28T00:00:00.000000-06:00",
+        "10 s ago" => ":sec 2011-03-10T15:10:27.000000-06:00",
+        "day before yesterday" => ":day 2011-03-08T00:00:00.000000-06:00",
+        "hr before tomorrow" => ":hour 2011-03-10T23:00:00.000000-06:00",
+        "3 days before today" => ":day 2011-03-07T00:00:00.000000-06:00",
+        "5 days after today" => ":day 2011-03-15T00:00:00.000000-05:00",
+        "5 days before now" => "nil 2011-03-05T15:10:37.981304-06:00",
+        "3 days before this minute" => ":min 2011-03-07T15:10:00.000000-06:00",
+        "5 days before yesterday" => ":day 2011-03-04T00:00:00.000000-06:00",
+        "2 days before 50 hours after tomorrow" => ":hour 2011-03-11T02:00:00.000000-06:00",
+        "2 centuries after today" => ":day 2211-01-21T00:00:00.000000-06:00",
+        "1pm" => ":hour 2011-03-10T13:00:00.000000-06:00",
+        "12:30pm" => ":min 2011-03-10T12:30:00.000000-06:00",
+        "9:20am tomorrow" => ":min 2011-03-11T09:20:00.000000-06:00",
+        "6am 3 days from yesterday" => ":hour 2011-03-12T06:00:00.000000-06:00",
+        "2001/01" => ":mon 2001-01-01T00:00:00.000000-06:00",
+        "2001-01" => ":mon 2001-01-01T00:00:00.000000-06:00",
+        # "01/2001" => ":mon 2001-01-01T00:00:00.000000-06:00",
+        "01/2001" => "#<Qreport::TimeParser::Error::Syntax: syntax error at position 2: \"01 |^| /2001\">",
+        "2001/02/03 12:23pm" => ":min 2001-02-03T12:23:00.000000-06:00",
+        "12/31 12:59pm" => ":min 2011-12-31T12:59:00.000000-06:00",
+        "12/31 last year" => ":day 2010-12-31T00:00:00.000000-06:00",
+        "12:59:59pm 12/31 next year" => ":sec 2012-12-31T12:59:59.000000-06:00",
+        "1:23:45pm 1/2 in 2 years" => ":sec 2013-01-01T13:23:45.000000-06:00",
+        "2011-03-10T15:10:37-06:00" => "nil 2011-03-10T15:10:37.000000-06:00",
+        "2011-03-10T15:10:37.981304-06:00" => "nil 2011-03-10T15:10:37.981304-06:00",
+        "2011-03-10T15:10:37-06:00 plus 10 sec" => ":sec 2011-03-10T15:10:47.000000-06:00",
+        "2011-03-10T15:10:37.981304-06:00 - 2 weeks" => "nil 2011-02-24T15:10:37.981304-06:00",
+        "now minus 2.5 weeks" => "nil 2011-03-10T15:10:35.481304-06:00",
+        "t - 10 sec" => "nil 2011-03-10T15:10:27.981304-06:00",
+        "123.45 sec ago" => "nil 2011-03-10T15:08:34.531303-06:00",
+        "year 2010" => ":year 2010-01-01T00:00:00.000000-06:00",
+        "between 12:45pm and 1:15pm" => ":min 2011-03-10T12:45:00.000000-06:00 ... :min 2011-03-10T13:15:00.000000-06:00",
+        "before 1:23pm tomorrow" => ":min 2011-03-11T13:22:00.000000-06:00",
+        "this minute" => ":min 2011-03-10T15:10:00.000000-06:00",
+        "last hour"     => ":hour 2011-03-10T14:00:00.000000-06:00",
+        "previous hour" => ":hour 2011-03-10T14:00:00.000000-06:00",
+        "last day"   => [ ":day 2011-03-09T00:00:00.000000-06:00", ":day 2011-03-09T00:00:00.000000-06:00 ... :day 2011-03-10T00:00:00.000000-06:00" ],
+        "previous day" => ":day 2011-03-09T00:00:00.000000-06:00",
+        "  2001-01 + 1234 ajsdkfsd hours" => "#<Qreport::TimeParser::Error::Syntax: syntax error at position 17: \"  2001-01 + 1234  |^| ajsdkfsd hours\">",
+        "15 sec" => "#<Qreport::TimeParser::Error: Qreport::TimeParser::Error>",
+        "12 minutes" => "#<Qreport::TimeParser::Error: Qreport::TimeParser::Error>",
+      }
+      examples[:now] = now
+      examples
+    end
+  end
+end
+

data/lib/qreport/time_parser/time_interval.rb

@@ -0,0 +1,117 @@
+require 'qreport/time_parser'
+require 'qreport/time_parser/time_unit'
+require 'rational'
+
+module Qreport
+  class TimeParser
+     class TimeInterval
+      include TimeUnit
+      include Comparable
+      attr_reader :amount
+
+      def initialize amount, unit
+        @amount, @unit = amount, unit
+        normalize!
+      end
+
+      def to_unit! unit
+        @amount *= unit_multiplier(unit).to_r / unit_multipler
+        @unit = unit
+        normalize!
+        self
+      end
+
+      def + x
+        case x
+        when Numeric
+          new(@amount + x, @unit)
+        when TimeInterval
+          new(@amount + x.to_sec, :sec)
+        else
+          raise TypeError, x.class.to_s
+        end
+      end
+
+      def - x
+        case x
+        when Numeric
+          new(@amount - x, @unit)
+        when TimeInterval
+          new(@amount - x.to_sec, :sec)
+        else
+          raise TypeError, x.class.to_s
+        end
+      end
+
+      def * x
+        case x
+        when Numeric
+          new(@amount * x, @unit)
+        else
+          raise TypeError, x.class.to_s
+        end
+      end
+
+      def / x
+        case x
+        when Numeric
+          new(@amount / x, @unit)
+        when TimeInterval
+          to_sec / x.to_sec
+        else
+          raise TypeError, x.class.to_s
+        end
+      end
+
+      def normalize!
+        @unit = nil if Float === @amount
+
+        super
+
+        case @amount
+        when String, Symbol
+          @amount = 
+            case @amount.to_s
+            when /\A(?:this)\b/i
+              0
+            when /\A(?:last|previous)\b/i
+              -1
+            when /\A(?:next|after)\b/i
+              1
+            else
+              raise ArgumentError, amount.inspect
+            end
+        end
+
+        self
+      end
+
+      def to_int; to_sec; end # see ruby/time.c
+
+      def to_sec
+        @to_sec ||=
+          @amount * unit_multiplier
+      end
+
+      def <=> x
+        case x
+        when TimeInterval
+          to_sec <=> x.to_sec
+        when Numeric
+          to_sec <=> x
+        else
+          raise TypeError, x.inspect
+        end
+      end
+
+      def to_s
+        "#{@amount.inspect} #{@unit.inspect}"
+      end
+
+      def inspect
+        "#<#{self.class} #{to_s}>"
+      end
+
+    end
+  end
+end

data/lib/qreport/time_parser/time_range.rb

@@ -0,0 +1,46 @@
+require 'qreport/time_parser'
+require 'qreport/time_parser/time_unit'
+
+module Qreport
+  class TimeParser
+    class TimeRange
+      include Comparable
+      attr_reader :a, :b
+
+      def initialize a, b
+        @a = a
+        @b = b
+      end
+
+      def inspect
+        "#<#{self.class} #{to_s}>"
+      end
+
+      def min
+        a <= b ? a : b
+      end
+
+      def max
+        a <= b ? b : a
+      end
+
+      def <=> x
+        min <=> x.min
+      end
+
+      def to_s
+        "#{a} ... #{b}"
+      end
+
+      def to_TimeRange; self; end
+
+      def to_range
+        if a <= b
+          (a.to_time ... b.to_time)
+        else
+          (b.to_time ... a.to_time)
+        end
+      end
+    end
+  end
+end

data/lib/qreport/time_parser/time_relative.rb

@@ -0,0 +1,77 @@
+require 'qreport/time_parser'
+require 'qreport/time_parser/time_unit'
+
+module Qreport
+  class TimeParser
+    class TimeRelative
+      include TimeUnit
+      SLOTS = [ :year, :mon, :day, :hour, :min, :sec, :zone ].freeze
+      attr_accessor *SLOTS
+
+      def unit
+        case 
+        when Float === @sec
+          nil
+        when @sec
+          :sec
+        when @min
+          :min
+        when @hour
+          :hour
+        when @day
+          :day
+        when @mon
+          :mon
+        when @year
+          :year
+        else
+          nil
+        end
+      end
+
+      def merge! x
+        unless self.class === x
+          x = x.to_time if x.respond_to?(:to_time)
+        end
+        @year ||= x.year
+        @mon ||= x.mon
+        @day ||= x.day
+        @hour ||= x.hour
+        @min ||= x.min
+        @sec ||= x.sec
+        @zone ||= x.zone
+        self
+      end
+
+      def to_s
+        str = ''
+
+        [ [ '',  @year, 4 ],
+          [ '-', @mon,  2 ],
+          [ '-', @day,  2 ],
+          [ 'T', @hour, 2 ],
+          [ ':', @min,  2 ],
+          [ ':', @sec,  2 ],
+          [ '-', @zone,   ],
+        ].each do | sep, val, size |
+          str << sep << (size ? (val ? "%0#{size}d" % val : '?' * size) : val).to_s
+        end
+
+        str
+      end
+
+      def inspect
+        "#<#{self.class} #{to_s}>"
+      end
+
+      def from_time! t
+        merge! t
+      end
+
+      def to_time
+        Time.send(@zone == 'UTC' ? :utc : :local, 
+                  @year || 0, @mon || 1, @day || 1, @hour || 0, @min || 0, @sec || 0)
+      end
+    end
+  end
+end

data/lib/qreport/time_parser/time_unit.rb

@@ -0,0 +1,110 @@
+require 'qreport/time_parser'
+
+module Qreport
+  class TimeParser
+    module TimeUnit
+      include Comparable
+      attr_reader :unit
+
+      def new *args
+        self.class.new *args
+      end
+
+      def normalize!
+        case @unit
+        when nil
+        when Symbol
+          @unit = @@unit_alias[@unit] || @unit
+        when String
+          @unit = @unit.downcase.to_sym
+          @unit = @@unit_alias[@unit] || @unit
+        when TimeInterval, TimeWithUnit
+          @unit = @unit.unit
+        else
+          raise ArgumentError, @unit.inspect
+        end
+        self
+      end
+
+      @@unit_alias = {
+        :second => :sec,
+        :s => :sec,
+        :minute => :min,
+        :min => :min,
+        :m => :min,
+        :hr => :hour,
+        :h => :hour,
+        :d => :day,
+        :w => :week,
+        :mo => :mon,
+        :month => :mon,
+        :mth => :mon,
+        :yr => :year,
+        :y => :year,
+      }
+
+      # in seconds
+      @@unit_interval = {
+        nil => 1,
+        :sec => 1,
+        :min => 60,
+        :hour => 60 * 60,
+        :day => 60 * 60 * 24, # + 1 leap second
+        :week => 60 * 60 * 24 * 7,
+        :mon => (60 * 60 * 24) * 31,
+        :year => (60 * 60 * 24) * 365, # + 1 leap day
+        :decade => (60 * 60 * 24) * 3650,
+        :century => (60 * 60 * 24) * 36500,
+        :millenium => (60 * 60 * 24) * 365000,
+      }
+
+      def self.pluralize str
+        str = str.to_s
+        case str
+        when "day"
+          "days"
+        when /y\Z/
+          str.sub(/y\Z/, 'ies')
+        else
+          str + "s"
+        end
+      end
+
+      (@@unit_alias.to_a.flatten + @@unit_interval.keys).each do | x |
+        next unless x
+        next if x.size == 1
+        x = x.to_sym
+        @@unit_alias[pluralize(x).to_sym] ||= @@unit_alias[x] || x
+      end
+
+      def unit_multiplier unit = @unit
+        @@unit_interval[unit] || 1
+      end
+
+      UNIT_REGEXP = 
+        (@@unit_interval.keys + 
+         @@unit_alias.keys + 
+         @@unit_alias.values).
+        uniq.
+        map(&:to_s).
+        reject(&:empty?).
+        sort_by(&:size) * '|'
+      # $stderr.puts "UNIT_REGEXP = #{UNIT_REGEXP}"
+
+      def unit_interval
+        @unit_interval ||=
+          TimeInterval.new(1, @unit)
+      end
+
+      def <=> x
+        case x
+        when TimeInterval
+          unit_interval <=> x.unit_interval
+        else
+          unit_interval <=> x
+        end
+      end
+
+    end
+  end
+end

data/lib/qreport/time_parser/time_with_unit.rb

@@ -0,0 +1,155 @@
+require 'qreport/time_parser'
+require 'qreport/time_parser/time_unit'
+
+module Qreport
+  class TimeParser
+    class TimeWithUnit
+      include TimeUnit
+      include Comparable
+      attr_reader :time
+
+      def initialize time, unit
+        @time, @unit = time, unit
+        normalize!
+      end
+
+      def interval(amount, unit = nil)
+        TimeInterval.new(amount, unit || @unit)
+      end
+
+      def + amount
+        # debugger
+        amount = interval(amount) unless TimeInterval === amount
+        raise TypeError, amount.to_s unless TimeInterval === amount
+        new(@time + amount.to_sec, 
+            [ unit_interval, amount.unit_interval ] )
+      end
+
+      def - x
+        x = interval(x) if Numeric === x
+        case x
+        when TimeInterval
+          new(@time - x.to_sec, unit_interval)
+        when TimeWithUnit, ::Time
+          interval(@time.to_f - x.to_f, unit_interval)
+        else
+          raise TypeError, x.inspect
+        end
+      end
+
+      def to_time
+        @time
+      end
+
+      def to_f
+        @time.to_f
+      end
+
+      def <=> x
+        case x
+        when TimeWithUnit
+          @time <=> x.to_time
+        else
+          @time <=> x
+        end
+      end
+
+      def normalize!
+        case @unit
+        when Array
+          @unit = @unit.reduce{|a, b| a < b ? a : b}
+        end
+
+        super
+
+        case @time
+        when nil
+          return
+        when ::Time
+        when String
+          @time = ::Time.parse(@time)
+        when Numeric
+          @time = ::Time.utc(@time)
+        when ::Date
+          @time = @time.to_time
+        when TimeWithUnit
+          @time = @time.to_time
+        else
+          raise TypeError, @time.inspect
+        end
+
+        # debugger
+        args = 
+          case @unit
+          when nil
+            return self
+          when :decade
+            [ @time.year % 10 * 10, 1, 1, 0, 0, 0 ]
+          when :century
+            [ @time.year % 100 / 100, 1, 1, 0, 0, 0 ]
+          when :millenium
+            [ @time.year % 1000 / 1000, 1, 1, 0, 0, 0 ]
+          else
+            @@unit_args[@unit] or
+              raise ArgumentError, @unit.inspect
+          end
+        if args.any? { | x | Symbol === x }
+          args = args.map do | x | 
+            x = @time.send(x) if Symbol === x
+            x
+          end
+        end
+        # $stderr.puts "  @time #{inspect} => "
+        # $stderr.puts "    args = #{args.inspect}"
+        @time = @time.class.send(zone_method, *args)
+        # $stderr.puts "    #{inspect}"
+        self
+      end
+
+      @@unit_args = {
+        :sec =>
+          [ :year, :mon, :day, :hour, :min, :sec ],
+        :min =>
+          [ :year, :mon, :day, :hour, :min, 0 ],
+        :hour =>
+          [ :year, :mon, :day, :hour, 0, 0 ],
+        :day =>
+          [ :year, :mon, :day, 0,     0, 0 ],
+        :week =>
+          [ :year, :mon, :day, 0,     0, 0 ],
+        :mon =>
+          [ :year, :mon, 1,    0,     0, 0 ],
+        :year =>
+          [ :year, 1,      1,    0,     0, 0 ],
+      }
+
+      UTC = 'UTC'.freeze
+
+      def zone_method
+        case @time.zone
+        when UTC
+          :utc
+        else
+          :local
+        end
+      end
+
+      def inspect
+        "#<#{self.class} #{to_s}>"
+      end
+
+      def to_s
+        "#{@unit.inspect} #{@time && @time.iso8601(6)}"
+      end
+
+      def to_TimeRange
+        TimeRange.new(self, self + unit_interval)
+      end
+
+      def to_range
+        to_TimeRange.to_range
+      end
+    end
+  end
+end
+