X-Git-Url: https://git.madduck.net/etc/awesome.git/blobdiff_plain/c7456258113a30fcb5140d83d138e45fdc2c9c9d..d535197d4d94122f86a485aac740bc7329664129:/src/tzinfo.lua

diff --git a/src/tzinfo.lua b/src/tzinfo.lua
index 66589d9..34b7048 100644
--- a/src/tzinfo.lua
+++ b/src/tzinfo.lua
@@ -1,3 +1,5 @@
+local gettime = require "luatz.gettime".gettime
+
 local tz_info_methods = { }
 local tz_info_mt = {
 	__index = tz_info_methods ;
@@ -9,12 +11,46 @@ local function _find_current ( tzinfo , target , i , j )
 
 	local half = math.ceil ( (j+i) / 2 )
 
-	if target > tzinfo [ half ].transition_time then
+	if target >= tzinfo [ half ].transition_time then
 		return _find_current ( tzinfo , target , half , j )
 	else
 		return _find_current ( tzinfo , target , i , half-1 )
 	end
 end
+
+local function find_current_local ( tzinfo , ts_local )
+	-- Find two best possibilities by searching back and forward a day (assumes transition is never by more than 24 hours)
+	local tz_first = _find_current ( tzinfo , ts_local-86400 , 0 , #tzinfo )
+	local tz_last  = _find_current ( tzinfo , ts_local+86400 , 0 , #tzinfo )
+
+	local n_candidates = tz_last - tz_first + 1
+
+	if n_candidates == 1 then
+		return tz_first
+	elseif n_candidates == 2 then
+		local tz_first_ob = tzinfo [ tz_first ]
+		local tz_last_ob  = tzinfo [ tz_last ]
+
+		local first_gmtoffset = tz_first_ob.info.gmtoff
+		local last_gmtoffset  = tz_last_ob .info.gmtoff
+
+		local t_start = tz_last_ob.transition_time + first_gmtoffset
+		local t_end   = tz_last_ob.transition_time + last_gmtoffset
+
+		-- If timestamp is before start or after end
+		if ts_local < t_start then
+			return tz_first
+		elseif ts_local > t_end then
+			return tz_last
+		end
+
+		-- If we get this far, the local time is ambiguous
+		return tz_first , tz_last
+	else
+		error ( "Too many transitions in a 2 day period" )
+	end
+end
+
 function tz_info_methods:find_current ( current )
 	return self [ _find_current ( self , current , 0 , #self ) ].info
 end
@@ -24,6 +60,18 @@ function tz_info_methods:localize ( utc_ts )
 	return utc_ts + self:find_current ( utc_ts ).gmtoff
 end
 
+function tz_info_methods:utctime ( ts_local , is_dst )
+	local tz1 , tz2 = find_current_local ( self , ts_local )
+	tz1 = self [ tz1 ].info
+	if tz2 == nil then
+		return ts_local - tz1.gmtoff
+	else -- Local time is ambiguous
+		tz2 = self [ tz2 ].info
+
+		return ts_local - tz2.gmtoff , ts_local - tz2.gmtoff
+	end
+end
+
 return {
 	tz_info_mt = tz_info_mt ;
 }