--## Author: Choonster ## Version: 1.30 ## SavedVariables: KillingBlow_Enhanced_DB
-- IMPORTANT: If you make any changes to this file, make sure you back it up before installing a new version.
-- This will allow you to restore your custom configuration with ease.
-- Also back up any custom textures or sounds you add.

-------
-- The first four variables control the appearance of the texture.
-------

-- The path of the texture file you want to use for characters of each faction relative to the main WoW directory (without the texture's file extension).
-- The default texture is "Interface\\AddOns\\KillingBlow_Enhanced\\KillingBlow_Enhanced\\KillingBlow_Alliance" for Alliance
-- and "Interface\\AddOns\\KillingBlow_Enhanced\\KillingBlow_Enhanced\\KillingBlow_Horde" for Horde; both by OligoFriends.
-- The AddOn also includes seven other textures:
--	"Interface\\AddOns\\KillingBlow_Enhanced\\KillingBlow_Enhanced\\Textures\\KillingBlow_HordeAlliance" by OligoFriends
--	"Interface\\AddOns\\KillingBlow_Enhanced\\KillingBlow_Enhanced\\Textures\\KillingBlow_Skull" by OligoFriends
--	"Interface\\AddOns\\KillingBlow_Enhanced\\KillingBlow_Enhanced\\Textures\\KillingBlow_HordeSword" by whitefreli
--	"Interface\\AddOns\\KillingBlow_Enhanced\\KillingBlow_Enhanced\\Textures\\KillingBlow_Death" by OligoFriends
--	"Interface\\AddOns\\KillingBlow_Enhanced\\KillingBlow_Enhanced\\Textures\\KillingBlow_SkullShield" by OligoFriends
--	"Interface\\AddOns\\KillingBlow_Enhanced\\KillingBlow_Enhanced\\Textures\\KillingBlow_Alliance2" by OligoFriends
--	"Interface\\AddOns\\KillingBlow_Enhanced\\KillingBlow_Enhanced\\Textures\\KillingBlow_Horde2" by OligoFriends
local ALLIANCE_TEXTURE_PATH = "Interface\\AddOns\\OrzUI\\Media\\Hutu\\DPS.tga"
local HORDE_TEXTURE_PATH    = "Interface\\AddOns\\OrzUI\\Media\\Hutu\\star.tga"

-- You can add your own texture by placing a TGA image in the WoW\Interface\AddOns\KillingBlowImage directory and changing the string after
-- ALLIANCE_TEXTURE_PATH or HORDE_TEXTURE_PATH to match its name.
-- See the "filename" argument on the following page for details on the required texture file format:
-- http://www.wowpedia.org/API_Texture_SetTexture
--
-- GIMP (www.gimp.org) is a free image editing program that can easily convert almost any image format to TGA as well as let you create your own TGA images.
-- If you want your texture to be packaged with the AddOn, just leave a comment on Curse or WoWI with the image embedded or a direct link to download the image.
-- I can convert PNG and other formats to TGA if needed.
-- Make sure that you have ownership rights of any image that you contribute.

-- The height/width of the texture. Using a height:width ratio different to that of the texture file may result in distortion.
local TEXTURE_WIDTH         = 21
local TEXTURE_HEIGHT        = 21

-------
-- These four variables control how the image is anchored to the screen.
-------

-- Used in image:SetPoint(TEXTURE_POINT, UIParent, ANCHOR_POINT, OFFSET_X, OFFSET_Y)
-- See http://www.wowpedia.org/API_Region_SetPoint for explanation.
local TEXTURE_POINT         = "CENTER" -- The point of the texture that should be anchored to the screen.
local ANCHOR_POINT          = "CENTER" -- The point of the screen the texture should be anchored to.
local OFFSET_X              = 0        -- The x/y offset of the texture relative to the anchor point.
local OFFSET_Y              = 5

-------
-- These four variables control the animation that plays when the image is shown
-------

local SCALE_X               = 1.5  -- The X scalar that the image should scale by
local SCALE_Y               = 1.5  -- The Y scalar that the image should scale by
local SCALE_DURATION        = 0.75 -- The duration of the scaling animation in seconds

local DELAY_DURATION        = 0.75 -- The amount of time between the end of the scaling animation and the image hiding

-------
-- Other options
-------

-- The sound to play when you get a killing blow
local SOUND_PATH            = "Interface\\AddOns\\OrzUI\\Media\\Sounds\\Sonar.ogg"

-- The channel to play the sound through. This can be "Master", "SFX", "Music" or "Ambience"
local SOUND_CHANNEL         = "Master"

-- If true, the AddOn will only record killing blows on players. If false, it will record all killing blows.
local PLAYER_KILLS_ONLY     = true

-- If true, the AddOn will only activate in battlegrounds and arenas. If false, it will work everywhere.
local PVP_ZONES_ONLY        = false

-- If true, the AddOn will print a message in your chat frame when you get a killing blow showing your current total.
-- This is reset any time you go through a loading screen (e.g. when entering or leaving a battleground or instance)
local DO_CHAT               = true

-------------------
-- END OF CONFIG --
-------------------
-- Do not change anything below here!

--[=[@alpha@
local DEBUG                 = true

local function debugprint(...)
	if DEBUG then
		print("KillingBlow_Enhanced:", ...)
	end
end
--@end-alpha@]=]


------
-- Animations
------
local frame = CreateFrame("Frame", "KillingBlow_EnhancedFrame", UIParent)
frame:SetPoint(TEXTURE_POINT, UIParent, ANCHOR_POINT, OFFSET_X, OFFSET_Y)
frame:SetFrameStrata("HIGH")
frame:Hide()

local texture = frame:CreateTexture()
texture:SetAllPoints()

local group = texture:CreateAnimationGroup()

group:SetScript("OnPlay", function(self)
	frame:SetSize(TEXTURE_WIDTH, TEXTURE_HEIGHT) -- Set the frame to the configured size before scaling animation starts
end)

local scale = group:CreateAnimation("Scale")
scale:SetScale(SCALE_X, SCALE_Y)
scale:SetDuration(SCALE_DURATION)

local delay = group:CreateAnimation("Animation")
delay:SetDuration(DELAY_DURATION)

delay:SetScript("OnPlay", function(self)
	frame:SetSize(TEXTURE_WIDTH * SCALE_X, TEXTURE_HEIGHT * SCALE_Y) -- Set the frame to the scaled size after the scaling animation ends
end)

group:SetScript("OnFinished", function(self)
	frame:Hide()
end)

frame:SetScript("OnShow", function(self)
	group:Play()
	PlaySoundFile(SOUND_PATH, SOUND_CHANNEL)
end)


------
-- Events
------

local band = bit.band

-- true if we have the AddOn security restrictions added in 12.0.0
-- TODO: Confirm if this works in the next Classic version to include the secrets API
local isCombatLogSecret = C_Secrets and C_Secrets.HasSecretRestrictions and C_Secrets.HasSecretRestrictions()

-- "YYYY-MM-DDThh:mm:ssZ" (ISO 8601 Complete date plus hours, minutes and seconds [UTC])
-- We use date strings instead of Unix times (seconds since epoch) because Lua offers no easy way to get the current Unix time in the UTC timezone (`time` only supports local time).
local DATE_FORMAT = "!%Y-%m-%dT%H:%M:%SZ"

local function GetTimestamp()
	return date(DATE_FORMAT)
end

local FILTER_MINE = bit.bor( -- Matches any "unit" under the player's control
	COMBATLOG_OBJECT_AFFILIATION_MINE,
	COMBATLOG_OBJECT_REACTION_FRIENDLY,
	COMBATLOG_OBJECT_CONTROL_PLAYER
)

local PLAYER_GUID = UnitGUID("player")
local PLAYER_NAME = GetUnitName("player", true)

-- Thanks to ErnestasBaltinas' HK Sounds for this method of tracking killing blows
local TOTAL_KILLING_BLOWS_ACHIEVEMENT_ID = 1487

local PlayerDB, CurrentSession

local FirstLoad = true
local KillCount = 0
local RecentKills = setmetatable({}, { __mode = "kv" }) -- [GUID] = killTime (from GetTime())
local PreviousKillingBlows = 0

local function GetKillingBlows()
	local _, _, _, killingBlows = GetAchievementCriteriaInfoByID(TOTAL_KILLING_BLOWS_ACHIEVEMENT_ID, 0)
	return killingBlows
end

local function CheckKillingBlowsIncreased()
	local killingBlows = GetKillingBlows()
	if killingBlows <= PreviousKillingBlows then
		return false, killingBlows
	end

	PreviousKillingBlows = killingBlows
	return true, killingBlows
end

local function KillingBlow(destGUID, destName, now)
	frame:Show()

	if not isCombatLogSecret then
		RecentKills[destGUID] = now
	end

	if CurrentSession then
		CurrentSession[GetTimestamp()] = destName
	end

	if DO_CHAT then
		KillCount = KillCount + 1
		UIErrorsFrame:AddMessage(("|cffCC0033Killing Blows Counter: %d|r"):format(KillCount))
		--RaidNotice_AddMessage(RaidWarningFrame, "----------   "..("|cffCC0033Killing Blows Counter: %d|r"):format(KillCount).."   ----------", ChatTypeInfo["RAID_WARNING"])
	end
end

local function StartSession(sessionType)
	CurrentSession = { SessionType = sessionType, StartTime = GetTimestamp() }
	PlayerDB[#PlayerDB + 1] = CurrentSession
end

local function EndSession()
	CurrentSession.EndTime = GetTimestamp()
	CurrentSession = nil
end

local GetPVPStatus, SetPVPStatus
do
	local InPVP = nil -- Initialise to nil so the the initial detection of a non-PvP zone and change to false unregisters COMBAT_LOG_EVENT_UNFILTERED
	local PVPStatus = { instance = false, world = false, ffa = false }

	function GetPVPStatus(pvpType)
		assert(PVPStatus[pvpType] ~= nil, ("Invalid pvpType: %s"):format(pvpType))

		return PVPStatus[pvpType]
	end

	function SetPVPStatus(pvpType, status, sessionType)
		assert(PVPStatus[pvpType] ~= nil, ("Invalid pvpType: %s"):format(pvpType))
		assert(type(status) == "boolean", "status must be boolean")

		PVPStatus[pvpType] = status

		local oldPVPStatus = InPVP
		InPVP = PVPStatus.instance or PVPStatus.world or PVPStatus.ffa
		if oldPVPStatus ~= InPVP then
			if CurrentSession then
				EndSession()
			end

			KillCount = 0

			if InPVP then
				StartSession(sessionType or pvpType)
			end

			if PVP_ZONES_ONLY then
				if InPVP then
					frame:RegisterCombatLogEvents()
				else
					frame:UnregisterCombatLogEvents()
				end
			end
		end
	end
end

function frame:RegisterCombatLogEvents()
	if isCombatLogSecret then
		frame:RegisterEvent("PARTY_KILL")
		frame:RegisterEvent("PLAYER_PVP_KILLS_CHANGED")
	else
		frame:RegisterEvent("COMBAT_LOG_EVENT_UNFILTERED")
	end
end

function frame:UnregisterCombatLogEvents()
	if isCombatLogSecret then
		frame:UnregisterEvent("PARTY_KILL")
		frame:UnregisterEvent("PLAYER_PVP_KILLS_CHANGED")
	else
		frame:UnregisterEvent("COMBAT_LOG_EVENT_UNFILTERED")
	end
end

frame:RegisterEvent("ADDON_LOADED")
frame:RegisterEvent("PLAYER_LOGIN")
frame:RegisterCombatLogEvents()

-- Instance
frame:RegisterEvent("PLAYER_ENTERING_WORLD")

-- FFA PvP
frame:RegisterUnitEvent("UNIT_FACTION", "player")

frame:SetScript("OnEvent", function(self, event, ...)
	self[event](self, ...)
end)

function frame:ADDON_LOADED(name)
	if name == "OrzUI" then
		ShiGuangDB[PLAYER_NAME] = ShiGuangDB[PLAYER_NAME] or {}  --KillingBlow_Enhanced_DB = KillingBlow_Enhanced_DB or {}
		self:UnregisterEvent("ADDON_LOADED")
	end
end

function frame:PLAYER_LOGIN()
	PLAYER_GUID = UnitGUID("player")
	PLAYER_NAME = GetUnitName("player", true)
end

function frame:PLAYER_ENTERING_WORLD()
	if FirstLoad then
		FirstLoad = false
		texture:SetTexture(UnitFactionGroup("player") == "Alliance" and ALLIANCE_TEXTURE_PATH or HORDE_TEXTURE_PATH)

		ShiGuangDB[PLAYER_NAME] = ShiGuangDB[PLAYER_NAME] or {}
		PlayerDB = ShiGuangDB[PLAYER_NAME]

		PreviousKillingBlows = GetKillingBlows()

	end

	local inInstance, instanceType = IsInInstance()
	if inInstance and (instanceType == "pvp" or instanceType == "arena") then
		SetPVPStatus("instance", true, instanceType)
	else
		SetPVPStatus("instance", false)
	end
end

function frame:UNIT_FACTION(unit)
	SetPVPStatus("ffa", UnitIsPVPFreeForAll("player"))
end

if isCombatLogSecret then
	function frame:PARTY_KILL(attackerGUID, targetGUID)
		-- If we're in instanced PvP, the kill should be handled by PLAYER_PVP_KILLS_CHANGED
		if GetPVPStatus("instance") then
			return
		end

		-- If the player's total killing blows hasn't increased, return now
		local killingBlowsIncreased, killingBlows = CheckKillingBlowsIncreased()
		if not killingBlowsIncreased then
			return
		end


		-- If attacker is secret or not the player or their pet, return now
		if not canaccessvalue(attackerGUID) or (attackerGUID ~= PLAYER_GUID and UnitTokenFromGUID(attackerGUID) ~= "pet")
		then
			return
		end

		-- If we're only recording player kills and the target is secret or not a player, return now
		if PLAYER_KILLS_ONLY and (not canaccessvalue(targetGUID) or not targetGUID:find("^Player%-")) then
			return
		end

		local now = GetTime()

		-- May be secret
		local targetName = UnitNameFromGUID(targetGUID)

		KillingBlow(targetGUID, targetName, now)
	end

	function frame:PLAYER_PVP_KILLS_CHANGED(unitTarget)
		-- If we're not in instanced PvP, the kill should be handled by PARTY_KILL
		if not GetPVPStatus("instance") then
			return
		end

		-- If the player's total killing blows hasn't increased, return now
		local killingBlowsIncreased, killingBlows = CheckKillingBlowsIncreased()
		if not killingBlowsIncreased then
			return
		end



		local now = GetTime()
		local targetGUID = UnitGUID(unitTarget)
		local targetName = UnitName(unitTarget)

		KillingBlow(targetGUID, targetName, now)
	end
else
	local function HandleCLEU(timestamp, event, hideCaster, sourceGUID, sourceName, sourceFlags, sourceRaidFlags,
							  destGUID,
							  destName, destFlags, destRaidFlags, ...)
		-- If there isn't a valid destination GUID
		if not destGUID or destGUID == "" or
			-- Or the source unit isn't the player or something controlled by the player (the latter check was suggested by Caellian)
			(sourceGUID ~= PLAYER_GUID and band(sourceFlags, FILTER_MINE) ~= FILTER_MINE) or
			-- Or we're only recording player kills and the destination unit isn't a player
			(PLAYER_KILLS_ONLY and not destGUID:find("^Player%-"))
		then
			return
		end -- Return now

		local _, overkill
		if event == "SWING_DAMAGE" then
			_, overkill = ...
		elseif event:find("_DAMAGE", 1, true) and not event:find("_DURABILITY_DAMAGE", 1, true) then
			_, _, _, _, overkill = ...
		end

		local now, previousKill = GetTime(), RecentKills[destGUID]

		-- Caellian has noted that PARTY_KILL doesn't always fire correctly and suggested checking the overkill argument
		-- (which will be 0 [or maybe -1] for non-killing blows) to mitigate against this.
		--
		-- Because most kills will trigger PARTY_KILL and an overkill _DAMAGE, we need to keep a record of recent kill times
		-- and only record kills of the same unit when they're at least 1 second apart.
		if (event == "PARTY_KILL" or (overkill and overkill > 0)) and (not previousKill or now - previousKill > 1.0) then
			KillingBlow(destGUID, destName, now)
		end
	end

	function frame:COMBAT_LOG_EVENT_UNFILTERED()
		HandleCLEU(CombatLogGetCurrentEventInfo())
	end
end
