InstanceCounters

hooksecurefunc("ResetInstances", INSTANCECOUNTERSResetInstances)

-- [ VERSION: 1.0.2 ] --
-- [ MODIFIED: 20091101 ] --

-- [ Instance Lock Information ] --
-- { Name, Year, Month, Day, Hour, Minute, Been Reset, Player, Realm } 
INSTANCECOUNTERSinstances = {
	{"---", -1, -1, -1, -1, -1, false, "---", "---"},
	{"---", -1, -1, -1, -1, -1, false, "---", "---"},
	{"---", -1, -1, -1, -1, -1, false, "---", "---"},
	{"---", -1, -1, -1, -1, -1, false, "---", "---"},
	{"---", -1, -1, -1, -1, -1, false, "---", "---"}};

-- [ Text Colors & Other Objects] --
local colorWhite = "|cFFFFFFFF";
local colorRed = "|cFFFF0000";
local colorGreen = "|cFF00FF00";
local debugHeader = "InstanceCounters: ";
local classcolors = {};
for i,v in pairs(RAID_CLASS_COLORS) do classcolors[i] = string.format("|cff%02x%02x%02x", v.r*255, v.g*255, v.b*255) end

-- [ Core Databroker Object & Libraries ] --
local dataobj = LibStub:GetLibrary("LibDataBroker-1.1"):NewDataObject("InstanceCounters", {type = "data source", text = "InstanceCounters"});

-- [ Namespace ] --
local f = CreateFrame("Frame");
f:SetScript("OnEvent", function(self, event, ...) if self[event] then return self[event](self, event, ...) end end);

-- [ Initialization ] --
function f:PLAYER_LOGIN()
	print(colorRed .. debugHeader .. colorWhite .. "Loaded");
	self:Show();
	self:RegisterEvent("PLAYER_ENTERING_WORLD");
	self:UnregisterEvent("PLAYER_LOGIN");
	self.PLAYER_LOGIN = nil;
end

function f:PLAYER_ENTERING_WORLD()
	inInstance, instanceType = IsInInstance();
	if (inInstance and (instanceType == "party" or instanceType == "raid")) then
		local iName, iType, iDifficulty, iDifficultyName = GetInstanceInfo();
		local lDate = date("*t");
		local weekday, month, day, year = lDate.wday, lDate.month, lDate.day, lDate.year;
		local hour, minute = lDate.hour, lDate.min;
		INSTANCECOUNTERSAddInstance(iName, year, month, day, hour, minute);
	end
end

-- [ OnClick Handler ] --
function dataobj.OnClick(self,button)
	if (button == "RightButton") then
		inInstance, instanceType = IsInInstance();
		if (not inInstance) then
			ResetInstances();
		else
			print(colorRed .. debugHeader .. colorWhite .. "Cannot reset instances while inside an instance.");
		end
	end
end

-- [ Tooltip Stuffs ] --
local tip = LibStub("tektip-1.0").new(6, "LEFT", "LEFT", "CENTER", "RIGHT", "RIGHT", "RIGHT")
function dataobj.OnLeave() tip:Hide() end
function dataobj.OnEnter(self)
	tip:AnchorTo(self);

	tip:AddLine(colorRed .. "Current Instance Timers");
	tip:AddLine(colorWhite .. "All timestamps use local time of your computer");
	tip:AddLine(" ");

	for i = 1, 5, 1	do
		dataobj:CheckTimeStamp(i);
			
		-- Instance Values
		local iName = INSTANCECOUNTERSinstances[i][1];
		local iYear = INSTANCECOUNTERSinstances[i][2];
		local iMonth = INSTANCECOUNTERSinstances[i][3];
		local iDay = INSTANCECOUNTERSinstances[i][4];
		local iHour = INSTANCECOUNTERSinstances[i][5];
		local iMinute = INSTANCECOUNTERSinstances[i][6];
			
		-- Format the timestamp
		local iTimestamp = "";
		if (iMonth < 10) then iTimestamp = iTimestamp .. "0" .. iMonth;
		else iTimestamp = iTimestamp .. iMonth;	end
		iTimestamp = iTimestamp .. "/";
			
		if (iDay < 10) then iTimestamp = iTimestamp .. "0" .. iDay;
		else iTimestamp = iTimestamp .. iDay; end
		iTimestamp = iTimestamp .. "/" .. iYear .. " ";
			
		if (iHour < 10) then iTimestamp = iTimestamp .. "0" .. iHour;
		else iTimestamp = iTimestamp .. iHour; end
		iTimestamp = iTimestamp .. ":";

		if (iMinute < 10) then iTimestamp = iTimestamp .. "0" .. iMinute;
		else iTimestamp = iTimestamp .. iMinute; end

		-- Current Values
		local lDate = date("*t");
		local weekday, month, day, year = lDate.wday, lDate.month, lDate.day, lDate.year;
		local hour, minute = lDate.hour, lDate.min;
			
		local minuteDiff = iMinute - minute;
		if (minuteDiff <= 0) then
			minuteDiff = 60 + minuteDiff;
		end
			
		if (iName == "---") then
			tip:AddLine(i .. ": " .. iName);
		else
			local left = i .. ": " .. iTimestamp .. " - " .. iName .. colorGreen .. " (Free in " .. minuteDiff .. " minutes)";
			local right = "     " .. INSTANCECOUNTERSinstances[i][8] .. colorWhite .. " (" .. INSTANCECOUNTERSinstances[i][9] .. ")";
			tip:AddMultiLine(left, right);
		end
	end

	tip:AddLine(" ");
	tip:AddLine(colorWhite .. "Right click to reset instances.");
	tip:Show();
end

function dataobj.CheckTimeStamp(self,index)
	if (INSTANCECOUNTERSinstances[index][1] == "---") then return end

	-- Instance Values
	local iYear = INSTANCECOUNTERSinstances[index][2];
	local iMonth = INSTANCECOUNTERSinstances[index][3];
	local iDay = INSTANCECOUNTERSinstances[index][4];
	local iHour = INSTANCECOUNTERSinstances[index][5];
	local iMinute = INSTANCECOUNTERSinstances[index][6];
	
	-- Current Values
	local lDate = date("*t");
	local weekday, month, day, year = lDate.wday, lDate.month, lDate.day, lDate.year;
	local hour, minute = lDate.hour, lDate.min;
	
	local hourDiff = hour - iHour;
	local minuteDiff = minute - iMinute;
	
	-- Big difference in time
	if (year > iYear or month > iMonth) then
		INSTANCECOUNTERSRemoveInstance(index);
	end
	
	if ((hourDiff == -23) or (hourDiff == 1)) then
		if (minuteDiff >= 0) then
			INSTANCECOUNTERSRemoveInstance(index);
		end
	elseif (((hourDiff > -23) and (hourDiff < 0)) or (hourDiff >= 2)) then
		INSTANCECOUNTERSRemoveInstance(index);
	end
end

-- [ Helper Methods, Made Global To Test Without Using Instances ] --
function INSTANCECOUNTERSClearInstances()
	for i = 1, 5, 1	do
		INSTANCECOUNTERSinstances[i][1] = "---";
		INSTANCECOUNTERSinstances[i][2] = -1;
		INSTANCECOUNTERSinstances[i][3] = -1;
		INSTANCECOUNTERSinstances[i][4] = -1;
		INSTANCECOUNTERSinstances[i][5] = -1;
		INSTANCECOUNTERSinstances[i][6] = -1;
		INSTANCECOUNTERSinstances[i][7] = false;
		INSTANCECOUNTERSinstances[i][8] = "---";
		INSTANCECOUNTERSinstances[i][9] = "---";
	end
end

function INSTANCECOUNTERSAddInstance(iName, year, month, day, hour, minute)
	local needToAdd = true;
	for i = 1, 5, 1	do
		dataobj.CheckTimeStamp(self,i);
		if (INSTANCECOUNTERSinstances[i][1] == iName and not INSTANCECOUNTERSinstances[i][7]) then
			needToAdd = false;
		end
		if (INSTANCECOUNTERSinstances[i][1] == "---" and needToAdd) then
			local _, class = UnitClass("player");
			INSTANCECOUNTERSinstances[i][1] = iName;
			INSTANCECOUNTERSinstances[i][2] = year;
			INSTANCECOUNTERSinstances[i][3] = month;
			INSTANCECOUNTERSinstances[i][4] = day;
			INSTANCECOUNTERSinstances[i][5] = hour;
			INSTANCECOUNTERSinstances[i][6] = minute;
			INSTANCECOUNTERSinstances[i][7] = false;
			INSTANCECOUNTERSinstances[i][8] = classcolors[class] .. UnitName("player");
			INSTANCECOUNTERSinstances[i][9] = GetRealmName();
			dataobj.CheckTimeStamp(self,i);
			break
		end
	end
end

function INSTANCECOUNTERSRemoveInstance(index)
	for i = index, 4, 1 do
		INSTANCECOUNTERSinstances[i][1] = INSTANCECOUNTERSinstances[i + 1][1];
		INSTANCECOUNTERSinstances[i][2] = INSTANCECOUNTERSinstances[i + 1][2];
		INSTANCECOUNTERSinstances[i][3] = INSTANCECOUNTERSinstances[i + 1][3];
		INSTANCECOUNTERSinstances[i][4] = INSTANCECOUNTERSinstances[i + 1][4];
		INSTANCECOUNTERSinstances[i][5] = INSTANCECOUNTERSinstances[i + 1][5];
		INSTANCECOUNTERSinstances[i][6] = INSTANCECOUNTERSinstances[i + 1][6];
		INSTANCECOUNTERSinstances[i][7] = INSTANCECOUNTERSinstances[i + 1][7];
		INSTANCECOUNTERSinstances[i][8] = INSTANCECOUNTERSinstances[i + 1][8];
		INSTANCECOUNTERSinstances[i][9] = INSTANCECOUNTERSinstances[i + 1][9];
	end
	INSTANCECOUNTERSinstances[5][1] = "---";
	INSTANCECOUNTERSinstances[5][2] = -1;
	INSTANCECOUNTERSinstances[5][3] = -1;
	INSTANCECOUNTERSinstances[5][4] = -1;
	INSTANCECOUNTERSinstances[5][5] = -1;
	INSTANCECOUNTERSinstances[5][6] = -1;
	INSTANCECOUNTERSinstances[5][7] = false;
	INSTANCECOUNTERSinstances[5][8] = "---";
	INSTANCECOUNTERSinstances[5][9] = "---";
end

function INSTANCECOUNTERSResetInstances()
	for i = 1, 5, 1 do
		INSTANCECOUNTERSinstances[i][7] = true;
	end
end

hooksecurefunc("ResetInstances", INSTANCECOUNTERSResetInstances);

if IsLoggedIn() then f:PLAYER_LOGIN() else f:RegisterEvent("PLAYER_LOGIN") end