X-Git-Url: https://git.madduck.net/etc/awesome.git/blobdiff_plain/12a8febabf1d5fe90d7c623377507d28a0f0fc80..8cde629906fd8d8e571fd519b3ca5ab3496ebbc1:/luatz/timetable.lua

diff --git a/luatz/timetable.lua b/luatz/timetable.lua
index e1649a0..f096372 100644
--- a/luatz/timetable.lua
+++ b/luatz/timetable.lua
@@ -1,3 +1,4 @@
+local strftime = require "luatz.strftime".strftime
 local strformat = string.format
 local floor = math.floor
 local function idiv ( n , d )
@@ -143,6 +144,10 @@ function timetable_methods:rfc_3339 ( )
 	return strformat ( "%04u-%02u-%02uT%02u:%02u:%06.3f" , self:unpack ( ) )
 end
 
+function timetable_methods:strftime ( format_string )
+	return strftime ( format_string , self )
+end
+
 local timetable_mt
 
 local function coerce_arg ( t )
@@ -156,10 +161,13 @@ timetable_mt = {
 	__index    = timetable_methods ;
 	__tostring = timetable_methods.rfc_3339 ;
 	__eq = function ( a , b )
-		return coerce_arg ( a ) == coerce_arg ( b )
+		return a:timestamp ( ) == b:timestamp ( )
 	end ;
 	__lt = function ( a , b )
-		return coerce_arg ( a ) < coerce_arg ( b )
+		return a:timestamp ( ) < b:timestamp ( )
+	end ;
+	__sub = function ( a , b )
+		return coerce_arg ( a ) - coerce_arg ( b )
 	end ;
 }
 
@@ -185,6 +193,9 @@ function timetable_methods:clone ( )
 end
 
 local function new_from_timestamp ( ts )
+	if type ( ts ) ~= "number" then
+		error ( "bad argument #1 to 'new_from_timestamp' (number expected, got " .. type ( ts ) .. ")" , 2 )
+	end
 	return new_timetable ( 1970 , 1 , 1 , 0 , 0 , ts )
 end