jps.lua

--[[
	 JPS - WoW Protected Lua DPS AddOn
	Copyright (C) 2011 Jp Ganis

	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program. If not, see <http://www.gnu.org/licenses/>.
]]--

-- Huge thanks to everyone who's helped out on this, <3
-- Universal
local L = MyLocalizationTable
jps = {}
jps.Version = "1.5"
jps.Revision = "r700"
jps.NextSpell = nil
jps.Rotation = nil
jps.UpdateInterval = 0.05
jps.Enabled = false
jps.Combat = false
jps.Debug = false
jps.DebugLevel = 1
jps.PLuaFlag = false
jps.MoveToTarget = false
jps.FaceTarget = true

jps.MultiTarget = false
jps.Interrupts = false
jps.UseCDs = false
jps.PvP = false
jps.Defensive = false

-- Utility
jps.Class = nil
jps.Spec = nil
jps.Race = nil
jps.Level = 1
jps.IconSpell = nil
jps.Message = ""
jps.LastTarget = nil
jps.LastTargetGUID = nil
jps.Target = nil
jps.Casting = false
jps.LastCast = nil
jps.ThisCast = nil
jps.NextCast = nil
jps.Error = nil
jps.Lag = nil
jps.Moving = nil
jps.startedMoving = 0
jps.Falling = nil
jps.startedFalling = 0
jps.MovingTarget = nil
jps.HarmSpell = nil
jps.IconSpell = nil
jps.CurrentCast = nil
jps.detectSpecDisabled = false

-- Class
jps.isNotBehind = false
jps.isBehind = true
jps.isHealer = false
jps.DPSRacial = nil
jps.DefRacial = nil
jps.Lifeblood = nil
jps.EngiGloves = nil
jps.isTank = false
jps.CrowdControl = false
jps.CrowdControlTarget = nil

-- Raccourcis
cast = CastSpellByName

-- Misc
jps.raid = {}
jps.Opening = true
jps.RakeBuffed = false
jps.RipBuffed = false
jps.BlacklistTimer = 2
jps.RaidStatus = {}
jps.RaidTarget = {}
jps.HealerBlacklist = {}
jps.Timers = {}
Healtable = {}
jps.EnemyTable =  {}
jps.initializedRotation = false
jps.firstInitializingLoop = true
jps.settings = {}
jps.settingsQueue = {}
jps.combatStart = 0
jps.RaidMode = false
jps.functionQueues = {}
jps.timedCasting = {}

-- Config.
jps.Configged = false
jps_variablesLoaded = false
jpsName = select(1,UnitName("player"))
jpsRealm = GetRealmName()
jps.ExtraButtons = true
jps.ResetDB = false

jps.Count = 1
jps.Tooltip = ""
jps.ToggleRotationName = {"No Rotations"}
jps.MultiRotation = false
rotationDropdownHolder = nil
jps.timeToDieAlgorithm= "LeastSquared"  --  WeightedLeastSquares , LeastSquared , InitialMidpoints
jps.maxTDDLifetime = 30 -- resetting time to die if there was no hp change withing 30 seconds
jps.TimeToDieData = {}
jps.RaidTimeToDie = {}
jps.customRotationFunc = ""

-- Latency
jps.CastBar = {}
jps.CastBar.latency = 0
jps.CastBar.latencySpell = nil
jps.CastBar.nextSpell = ""
jps.CastBar.nextTarget = ""
jps.CastBar.currentSpell = ""
jps.CastBar.currentTarget = ""
jps.CastBar.currentMessage = ""

jps.pulltimerDefault = 900
jps.currentTimer = 900

-- Slash Cmd
SLASH_jps1 = '/jps'

function write(...)
	if jps.getConfigVal("quiet mode",0) == 0 then
		DEFAULT_CHAT_FRAME:AddMessage("|cffff8000JPS: " .. strjoin(" ", tostringall(...))); -- color orange
	end
end
function macrowrite(...)
   DEFAULT_CHAT_FRAME:AddMessage("|cffff8000MACRO: " .. strjoin(" ", tostringall(...))); -- color orange
end

------------------------
-- DETECT CLASS SPEC
------------------------

function jps.detectSpec()
	if jps.detectSpecDisabled == true then return false end

	jps.Count = 1
	jps.Tooltip = ""
	jps.ToggleRotationName = {"No Rotations"}
	jps.MultiRotation = false
	jps.initializedRotation = false
	jps.firstInitializingLoop = true
	rotationDropdownHolder:Hide()

	jps.Race = UnitRace("player")
	jps.Class = UnitClass("player")
	jps.Level = Ternary(jps.Level > 1, jps.Level, UnitLevel("player"))
	if jps.Class then
		local id = GetSpecialization() -- remplace GetPrimaryTalentTree() patch 5.0.4
		if not id then 
			if jps.Level < 10 then 
				write("You need to be at least at level 10 and have a specialization choosen to use JPS is using now random spells") 
				jps.getHarmSpellTable()
				jps.detectSpecDisabled = true
			else
				write("jps couldn't find your talent tree... One second please.") 
			end
		else
			jps.customRotationFunc = ""
		-- local id, name, description, icon, background, role = GetSpecializationInfo(specIndex [, isInspect [, isPet]])
			local _,name,_,_,_,_ = GetSpecializationInfo(id) -- patch 5.0.4 remplace GetTalentTabInfo( id )
			if name then
				jps.Spec = name
				if jps.Spec then 
					write("Online for your",jps.Spec,jps.Class)
				end
			end
		end
	end
   if jps.Spec == L["Discipline"] or jps.Spec == L["Holy"] or jps.Spec == L["Restoration"] or jps.Spec == L["Mistweaver"] then jps.isHealer = true end
   if jps.Spec == L["Blood"] or jps.Spec == L["Protection"] or jps.Spec == L["Brewmaster"] or jps.Spec == L["Guardian"] then
	   jps.isTank = true
	   jps.gui_toggleDef(true) 
	end
	jps.HarmSpell = jps.GetHarmfulSpell()
	--write("jps.HarmSpell_","|cff1eff00",jps.HarmSpell)
	jps.setClassCooldowns()
	jps_VARIABLES_LOADED()
	if jps.initializedRotation == false then
		jps_Combat()
	end
end

------------------------
-- SLASHCMDLIST
------------------------

function SlashCmdList.jps(cmd, editbox)
	local msg, rest = cmd:match("^(%S*)%s*(.-)$");
	if msg == "toggle" or msg == "t" then
		if jps.Enabled == false then msg = "e"
		else msg = "d" end
	end
	if msg == "config" then
		InterfaceOptionsFrame_OpenToCategory(jpsConfigFrame)
	elseif msg == "show" then
		jpsIcon:Show()
		write("Icon set to show")
	elseif msg == "hide" then
		jpsIcon:Hide()
		write("Icon set to hide")
	elseif msg== "disable" or msg == "d" then
		jps.Enabled = false
		jps.gui_toggleEnabled(false)
		write("jps Disabled.")
	elseif msg== "enable" or msg == "e" then
		jps.Enabled = true
		jps.gui_toggleEnabled(true)
		write("jps Enabled.")
	elseif msg == "respec" then
		jps.detectSpec()
	elseif msg == "multi" or msg == "aoe" then
		jps.gui_toggleMulti()
	elseif msg == "cds" then
		jps.gui_toggleCDs()
	elseif msg == "int" then
		jps.gui_toggleInt()
	elseif msg == "pvp" then
		jps.togglePvP()
		write("PvP mode is now set to",tostring(jps.PvP))
	elseif msg == "def" then
		jps.gui_toggleDef()
		write("Defensive set to",tostring(jps.Defensive))
	elseif msg == "heal" then
		jps.isHealer = not jps.isHealer
		write("Healing set to", tostring(jps.isHealer))
	elseif msg == "opening" then
		jps.Opening = not jps.Opening
		write("Opening flag set to",tostring(jps.Opening))
	elseif msg == "debug" and rest ~="" then
		if tonumber(rest) then
			jps.DebugLevel = rest
			write("Debug level set to",tostring(rest))		
		else
			jps.DebugLevel = 1
			write("Debug level set to 1")		
		end
	elseif msg == "debug" then
		jps.Debug = not jps.Debug
		write("Debug mode set to",tostring(jps.Debug))
	elseif msg == "face" then
		jps.gui_toggleRot()
		write("jps.FaceTarget set to",tostring(jps.FaceTarget))
	elseif msg == "db" then
		jps.ResetDB = not jps.ResetDB
		jps_VARIABLES_LOADED()
		write("jps.ResetDB set to",tostring(jps.ResetDB))
		jps.Macro("/reload")
	elseif msg == "ver"  or msg == "version" or msg == "revision" or msg == "v" then
		write("You have JPS version: "..jps.Version..", revision: "..jps.Revision)
	elseif msg == "raid"  or msg == "raidmode" then
		jps.RaidMode = not jps.RaidMode
		if jps.RaidMode then
			write("Raid Mode is now enabled")
		else
			write("Raid Mode is now disabled")
		end
	elseif msg == "opening" then
		jps.Opening = not jps.Opening
		write("Opening flag is now set to",tostring(jps.Opening))
	elseif msg == "size" then
		jps.resize( rest )
	elseif msg == "update" then
		jps.loadDefaultSettings()
	elseif msg == "reset" then
		jps.resetView()
	elseif msg == "help" then
		write("Slash Commands:")
		write("/jps - Show enabled status.")
		write("/jps enable/disable - Enable/Disable the addon.")
		write("/jps spam - Toggle spamming of a given macro.")
		write("/jps cds - Toggle use of cooldowns.")
		write("/jps pew - Spammable macro to do your best moves, if for some reason you don't want it fully automated")
		write("/jps interrupts - Toggle interrupting")
		write("/jps reset - reset position of jps icons and UI")
		write("/jps db - cleares your local jps DB")
		write("/jps help - Show this help text.")
	elseif msg == "pew" then
	  	jps_Combat()
	else
		if jps.Enabled then
			print("jps Enabled - Ready and Waiting.")
		else 
			print "jps Disabled - Waiting on Standby."
		end
	end
end

-- cache for WoW API functions that return always the same results for the given params
local spellcache = setmetatable({}, {__index=function(t,v) local a = {GetSpellInfo(v)} if GetSpellInfo(v) then t[v] = a end return a end})
local function GetSpellInfo(a)
	return unpack(spellcache[a])
end

-- set's jps.NextSpell if user manually uses a spell/item
hooksecurefunc("UseAction", function(...)
	if jps.Enabled and (select(3, ...) ~= nil) and InCombatLockdown() == true  then
		local stype,id,_ = GetActionInfo(select(1, ...))
		if stype == "spell" then
			local name = select(1,GetSpellInfo(id))
			if jps.NextSpell ~= name  and not jps.shouldSpellBeIgnored(name) then
				jps.NextSpell = name
				if jps.getConfigVal("print next spells in chat") == 1 and jps.Combat then
					write("Set",name,"for next cast.")
				end
			end
		end
		if stype == "item" then
			local res = jps.useBagItem(id)
			
			if res == nil and IsEquippedItem(id) then
				slot = select(9, GetItemInfo(id))
				if string.find(slot, "TRINKET") ~= nil then
					s1 = select(1,GetInventorySlotInfo("Trinket0Slot"))
					s2 = select(1,GetInventorySlotInfo("Trinket1Slot"))
					t1 = GetInventoryItemID("player", s1)
					t2 = GetInventoryItemID("player", s2)

					if t1 == id then
						jps.NextSpell = "/use "..s1
					end
					if t2 == id then
						jps.NextSpell = "/use "..s2
					end
					
				end
				-- this does not work for engineer glider / nitro boots... coming soon!
			end
			if res ~= nil then
				jps.NextSpell = res[2]
			end
		end
		if stype == "macro" then
			macroText = select(3, GetMacroInfo(id))
			if string.find(macroText,"jps") == nil then
				jps.NextSpell = macroText
			end
		end
	end
end)
------------------------
-- COMBAT
------------------------

function jps_Combat()  
	-- Check for the Rotation
	if not jps.Class then return end
	if not jps.activeRotation() and jps.Combat == false then
		if string.len(jps.customRotationFunc) < 10 then
			write("JPS does not have a rotation for your",jps.Spec,jps.Class)
			jps.Enabled = false
		end
		return 
	end

	-- STOP Combat
	if (IsMounted() == true and jps.getConfigVal("dismount in combat") == 0) or UnitIsDeadOrGhost("player")==1 or jps.buff(L["Drink"],"player") then return end

	-- LagWorld
	jps.Lag = select(4,GetNetStats())/1000 -- amount of lag in milliseconds local down, up, lagHome, lagWorld = GetNetStats()
	
	-- Casting UnitCastingInfo("player")~= nil or UnitChannelInfo("player")~= nil
	local latency = math.max(jps.CastBar.latency,jps.Lag)
	if jps.ChannelTimeLeft() > 0 then
		jps.Casting = true
	elseif (jps.CastTimeLeft() - latency) > 0 then 
		jps.Casting = true
	else
		jps.Casting = false
	end
	
	-- RAID UPDATE
	jps.UpdateHealerBlacklist()
	-- in case you want to play only with /jps pew the RaidStatus table will be updated
	if not jps.Enabled then jps.SortRaidStatus() end
	
	-- Check spell usability 
	if string.len(jps.customRotationFunc) > 10 then
		jps.ThisCast,jps.Target = jps.customRotation() 
	else
		jps.ThisCast,jps.Target = jps.activeRotation().getSpell() -- ALLOW SPELLSTOPCASTING() IN JPS.ROTATION() TABLE
	end
	if jps.firstInitializingLoop == true then
		jps.firstInitializingLoop = false
		return nil
	end
	
	-- Movement
	isMoving = GetUnitSpeed("player") > 0
	if jps.Moving == false and isMoving == true then
		jps.startedMoving = GetTime()
	end
	jps.Moving = isMoving
	
	isFalling = IsFalling()
	if jps.falling == false and isFalling == true then
		jps.startedFalling = GetTime()
	end

	jps.falling = isFalling

	jps.MovingTarget = GetUnitSpeed("target") > 0
	
	if not jps.Casting and jps.ThisCast ~= nil then
		if jps.NextSpell ~= nil then
			if string.find(jps.NextSpell,"/") ~= nil then
				jps.Macro(jps.NextSpell)

				jps.NextSpell = nil
			else
				if jps.canCast(jps.NextSpell, jps.Target) then
					jps.Cast(jps.NextSpell)
					if jps.getConfigVal("print manually casted spells") == 1 then
						write("Next Spell "..jps.NextSpell.. " was casted")
					end
					jps.NextSpell = nil
				else
					if jps.cooldown(jps.NextSpell) > 3 then jps.NextSpell = nil end
					jps.Cast(jps.ThisCast)
				end
			end
		else
			jps.Cast(jps.ThisCast)
		end
	end
	
	-- Return spellcast.
	return jps.ThisCast,jps.Target
end

function jps.addTofunctionQueue(fn,queueName) 
	if not jps.functionQueues[queueName] then
		jps.functionQueues[queueName] = {}
	end
	if not jps.functionQueues[queueName][fn] then
		jps.functionQueues[queueName][fn] = fn
	end
end

function jps.deleteFunctionFromQueue(fn, queueName)
	if jps.functionQueues[queueName] ~= nil then
		if jps.functionQueues[queueName][fn] ~= nil then
			jps.functionQueues[queueName][fn] = nil
		end
	end
end

function jps.runFunctionQueue(queueName)
	local noErrors = true
	if jps.functionQueues[queueName] then
		for _,fn in pairs(jps.functionQueues[queueName]) do
			local status, error = pcall(fn)
			if not status then
				noError = false
				LOG.error("Error %s on function: %s in Queue %s", error, fn, queueName)
			end
			jps.functionQueues[queueName][fn] = nil
		end
		if noErrors then
			jps.functionQueues[queueName] = nil
			return true
		end
	end	
	return false
end

-- TODO wod guid 
function UnitGUIDnorm(guid)
	if guid then
		guid = string.gsub(guid, "Creature","")
		guid = string.gsub(guid, "Player","")
		guid = string.gsub(guid, "Vignette","")
		guid = string.gsub(guid, "GameObject","")
		guid = string.gsub(guid, "-","")
		guid = string.gsub(guid, "%.","")
	end
	return guid or 0
end