X-Git-Url: https://git.madduck.net/etc/awesome.git/blobdiff_plain/a74ef4ee5f93ac14d3ee03bf70a78d6a192e351f..6db06b57e851ec08d5355f015d6df1ebe58b4982:/src/timetable.lua

diff --git a/src/timetable.lua b/src/timetable.lua
index aef8ca4..d1cd534 100644
--- a/src/timetable.lua
+++ b/src/timetable.lua
@@ -75,7 +75,9 @@ local function unpack_tm ( tm )
 		assert ( tm.day   , "day required" ) ,
 		tm.hour or 12 ,
 		tm.min  or 0 ,
-		tm.sec  or 0
+		tm.sec  or 0 ,
+		tm.yday ,
+		tm.wday
 end
 
 -- Modify parameters so they all fit within the "normal" range
@@ -149,14 +151,23 @@ function timetable_methods:rfc_3339 ( )
 	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 )
-		return a:timestamp() == b:timestamp()
+		return coerce_arg ( a ) == coerce_arg ( b )
 	end ;
 	__lt = function ( a , b )
-		return a:timestamp() < b:timestamp()
+		return coerce_arg ( a ) < coerce_arg ( b )
 	end ;
 }
 
@@ -164,7 +175,7 @@ local function cast_timetable ( tm )
 	return setmetatable ( tm , timetable_mt )
 end
 
-local function new_timetable ( year , month , day , hour , min , sec )
+local function new_timetable ( year , month , day , hour , min , sec , yday , wday )
 	return cast_timetable {
 		year  = year ;
 		month = month ;
@@ -172,9 +183,15 @@ local function new_timetable ( year , month , day , hour , min , sec )
 		hour  = hour ;
 		min   = min ;
 		sec   = sec ;
+		yday  = yday ;
+		wday  = wday ;
 	}
 end
 
+function timetable_methods:clone ( )
+	return new_timetable ( unpack_tm ( self ) )
+end
+
 local function new_from_timestamp ( ts )
 	return new_timetable ( 1970 , 1 , 1 , 0 , 0 , ts )
 end