src/timetable: Comparison operators can take raw timestamps

diff --git a/src/timetable.lua b/src/timetable.lua
index 88016238e77aed0fcaf92169e98e9f245f389cc7..d1cd534a624f8c8653929b72dd556e1966601027 100644 (file)
--- a/src/timetable.lua
+++ b/src/timetable.lua
@@ -151,14 +151,23 @@ function timetable_methods:rfc_3339 ( )
        return strformat ( "%04u-%02u-%02uT%02u:%02u:%06.4g" , unpack_tm ( self ) )
 end
 
        return strformat ( "%04u-%02u-%02uT%02u:%02u:%06.4g" , unpack_tm ( self ) )
 end
 

-local timetable_mt = {
+local timetable_mt
+
+local function coerce_arg ( t )
+       if getmetatable ( t ) == timetable_mt then
+               return t:timestamp ( )
+       end
+       return t
+end
+
+timetable_mt = {

        __index    = timetable_methods ;
        __tostring = timetable_methods.rfc_3339 ;
        __eq = function ( a , b )
        __index    = timetable_methods ;
        __tostring = timetable_methods.rfc_3339 ;
        __eq = function ( a , b )

-               return a:timestamp() == b:timestamp()
+               return coerce_arg ( a ) == coerce_arg ( b )

        end ;
        __lt = function ( a , b )
        end ;
        __lt = function ( a , b )

-               return a:timestamp() < b:timestamp()
+               return coerce_arg ( a ) < coerce_arg ( b )

        end ;
 }
 
        end ;
 }