Thievery.lua

local T = Thievery_Translate

local addonName, tv = ...
local gameVersion = tv.gameVersion

local SAP_SPELLID = 6770
local PICKPOCKET_SPELLID = 921

local function clearTable(teeburu)
    for i, v in pairs(teeburu) do
        teeburu[i] = nil
    end
end

local isAlliance
local faction = UnitFactionGroup("player")
if faction == "Alliance" then
    isAlliance = true
elseif faction == "Horde" then
    isAlliance = false
end

local target = Thievery_Target


local UIActive = false
local PPMode = false
local sapMode = false




local Session_EnemyIDTable = {

}



local function getIDFromGUID(guid)
    if not guid then return end
    local starto = 0
    local endo = 0
    for i=1,5,1 do
        starto, endo = string.find(guid, "(%d+)", endo + 1)
        if not endo then
            print("Thievery: Faulty GUID, please contact addon author.")
            return false
        end
    end
    local npcID = string.sub(guid, starto, endo)
    return npcID
end

local function printTargetInfo()
    Thievery_BetaPrint("Creature Family: ", UnitCreatureFamily("target"))
    Thievery_BetaPrint("Creature Type: ", UnitCreatureType("target"))
    Thievery_BetaPrint("Classification: ", UnitClassification("target"))
    Thievery_BetaPrint(UnitGUID("target"))
    Thievery_BetaPrint(getIDFromGUID(UnitGUID("target")))
end
SLASH_THIEVERYTARGETINFO1 = "/teeftarget"
SlashCmdList["THIEVERYTARGETINFO"] = printTargetInfo

local lastPrint
local ppName = C_Spell.GetSpellName(PICKPOCKET_SPELLID)
local sapName = C_Spell.GetSpellName(SAP_SPELLID)
local handleKeybind
local checkSapAura
-- /dump UnitExists("target")
-- /dump UnitExists("softenemy")
--____________________________________________________________________________
--     Pickpocketing through SecureActionButton + OverrideBindingClick
--                          bugs out in Classic
--              use SetOverrideBindingSpell directly instead
--____________________________________________________________________________
if gameVersion == 1 then
    handleKeybind = function(mode, button, assignKey) 
        if mode == "pp" then
            button:SetAttribute("macrotext", "/cast " .. ppName)
            SetOverrideBindingClick(button, true, assignKey, "Thievery_PickpocketButton")
        elseif mode == "sap" then
            button:SetAttribute("macrotext", "/cast " .. sapName)
            SetOverrideBindingClick(button, true, assignKey, "Thievery_PickpocketButton")
        end
    end
    checkSapAura = function()
        local sapped = false
        local targetString 
        if UnitExists("target") then
            targetString = "target"
        elseif UnitExists("softenemy") then
            targetString = "softenemy"
        end
        if not targetString then return false end
        AuraUtil.ForEachAura(targetString, "HARMFUL", nil, function(name, icon, _, _, _, _, _, _, _, spellID, ...)
            if spellID == SAP_SPELLID then
                sapped = true
            end
        end)
        return sapped
    end
elseif gameVersion == 2 or gameVersion == 3 or gameVersion == 4 then
    handleKeybind = function(mode, button, assignKey)
        if mode == "pp" then
            SetOverrideBindingSpell(button, true, assignKey, ppName)
        elseif mode == "sap" then
            SetOverrideBindingSpell(button, true, assignKey, sapName)
        end
    end
    checkSapAura = function()
        local sapped = false
        AuraUtil.FindAura(function(criteria, _,_,_,_,_,_,_,_,_,_,_,spellID) 
            if criteria == spellID then
                sapped = true
            end
        end, "target", "HARMFUL", SAP_SPELLID)
        return sapped
    end
end


local function setPPMode(self)
    local remaining = Thievery_UpdatePPTimers(target.guid)
    if remaining then
        remaining = string.format("%.0f", remaining)
        if remaining ~= lastPrint then
            lastPrint = remaining
            Thievery_BetaPrint("Can pickpocket again in: " .. remaining)
        end
    end
    if not PPMode then
        local parent = self:GetParent()
        local assignKey = "E"
        if Thievery_Config.ppKey then
            assignKey = Thievery_Config.ppKey
        end

        handleKeybind("pp", self, assignKey)

        PPMode = true
        sapMode = false
        parent.visual:Show()
        parent.visual.npcName:SetText(target.name)
        parent.visual.promptText:SetText(T["Pickpocket"])
        if remaining then
            parent.visual.promptText:SetTextColor(0.8, 0.8, 0.8)
            parent.visual.throughLine:Show()
            Thievery_PPCooldownFrame:SetCooldown(GetTime(), remaining)
        else
            parent.visual.promptText:SetText(T["Pickpocket"])
            parent.visual.promptText:SetTextColor(1, 0, 0)
            parent.visual.throughLine:Hide()
        end
    end
end

local function setSapMode(self)
    if not sapMode then 
        local parent = self:GetParent()
        local assignKey = "E"
        if Thievery_Config.ppKey then
            assignKey = Thievery_Config.ppKey
        end

        handleKeybind("sap", self, assignKey)

        PPMode = true
        parent.visual:Show()
        parent.visual.promptText:SetText(T["Sap"])
        parent.visual.promptText:SetTextColor(1, 0, 0)
        parent.visual.throughLine:Hide()
        parent.visual.npcName:SetText(target.name)
        sapMode = true
        PPMode = false
    end
end
function Thievery_Activate(self)
    if InCombatLockdown() then 
        print("Thievery: State change occured during combat. Please contact author.")
        return 
    end
    UIActive = true

    local sapped = checkSapAura()

    if Thievery_Config.Checkboxes[1].enableSap == true then
        if sapped == false then
            setSapMode(self)
        else
            setPPMode(self)
        end
    else
        setPPMode(self)
    end
end
function Thievery_Deactivate(self)
    if InCombatLockdown() then 
        print("Thievery: State change occured during combat. Please contact author.")
        return 
    end
    if UIActive == true then
        local parent = self:GetParent()
        ClearOverrideBindings(self)
        parent.visual:Hide()
        parent.visual.npcName:SetText(nil)
        UIActive = false
        PPMode = false
        sapMode = false
    end
end

function Thievery_CheckTargetLocal(target)
    if not target then return false end
    local npcID = target.npcID
    if not npcID then return false end
    if Session_EnemyIDTable[npcID] then
        if not PPMode and not sapMode then
            Thievery_BetaPrint("session match found!")
        end
        return true
    end
end

local stealthed = false
local validTarget = false
function Thievery_UpdateState(self, resetMode)
    if InCombatLockdown() then return end
    if resetMode then
        PPMode = false
        sapMode = false
        ClearOverrideBindings(self)
    end
    local inRange = C_Spell.IsSpellInRange(PICKPOCKET_SPELLID)
    if stealthed and validTarget and inRange then
        if Thievery_CheckTargetLocal(target) then
            Thievery_Activate(self)
        elseif Thievery_CheckTargetForPP(target, isAlliance) then
            Session_EnemyIDTable[target.npcID] = true
            Thievery_Activate(self)
        else
            Thievery_Deactivate(self)
        end
    else
        Thievery_Deactivate(self)
    end
end

-- --------------------------------------------------------------------------
local function checkTarget(unitString)
    if not UnitExists(unitString) then 
        --Thievery_BetaPrint("No eligible target")
        return false
    end
    if UnitIsDead(unitString) or UnitIsCorpse(unitString) then
        return false
    end
    if UnitIsPlayer(unitString) then
        return false
    end
    -- 1)player 2)target ORDER ON PURPOSE. To avoid checking reputations
    local reaction = UnitReaction("player", unitString)
    if reaction ~= 2 and reaction ~= 4 then
        return false
    end
    return true
end
-- ⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⣠⣤⣤⣤⣤⣤⣤⣤⣤⣄⡀⠀⠀⠀⠀⠀⠀⠀⠀ 
-- ⠀⠀⠀⠀⠀⠀⠀⠀⢀⣴⣿⡿⠛⠉⠙⠛⠛⠛⠛⠻⢿⣿⣷⣤⠀⠀⠀⠀⠀ 
-- ⠀⠀⠀⠀⠀⠀⠀⠀⣼⣿⠋⠀⠀⠀⠀⠀⠀⠀⢀⣀⣀⠈⢻⣿⣿⡄⠀⠀⠀⠀ 
-- ⠀⠀⠀⠀⠀⠀⠀⣸⣿⡏⠀⠀⠀⣠⣶⣾⣿⣿⣿⠿⠿⠿⢿⣿⣿⣿⣄⠀⠀⠀ 
-- ⠀⠀⠀⠀⠀⠀⠀⣿⣿⠁⠀⠀⢰⣿⣿⣯⠁⠀⠀⠀⠀⠀⠀⠀⠈⠙⢿⣷⠀ 
-- ⠀⠀⣀⣤⣴⣶⣶⣿⡟⠀⠀⢸⣿⣿⣿ ⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⣿⣷⠀ 
-- ⠀⢰⣿⡟⠋⠉⣹⣿⡇⠀ ⣿⣿⣿⣿⣷⣦⣤⣤⣤⣶⣶⣶⣶⣿⣿⣿⠀ 
-- ⠀⢸⣿⡇⠀⠀⣿⣿⡇⠀⠀⠀⠹⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⡿ 
-- ⠀⣸⣿⡇⠀⠀⣿⣿⡇⠀⠀⠀⠀⠉⠻⠿⣿⣿⣿⣿⡿⠿⠿⠛⢻⣿⡇
-- ⠀⣿⣿⠁⠀⠀⣿⣿⡇⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⢸⣿⣧⠀⠀ 
-- ⠀⣿⣿⠀⠀⠀⣿⣿⡇⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⢸⣿⣿⠀⠀ 
-- ⠀⣿⣿⠀⠀⠀⣿⣿⡇⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⢸⣿⣿⠀⠀ 
-- ⠀⢿⣿⡆⠀⠀⣿⣿⡇⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⢸⣿⡇⠀⠀ 
-- ⠀⠸⣿⣧⡀⠀⣿⣿⡇⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⣿⣿⠃⠀⠀ 
-- ⠀⠀⠛⢿⣿⣿⣿⣿⣇⠀⠀⠀⠀⠀⣰⣿⣿⣷⣶⣶⣶⣶⠶⢠⣿⣿⠀⠀⠀ 
-- ⠀⠀⠀⠀⠀⠀⠀⣿⣿⠀⠀⠀⠀⠀⣿⣿⡇⠀⣽⣿⡏⠁⠀⠀⢸⣿⡇⠀⠀⠀ 
-- ⠀⠀⠀⠀⠀⠀⠀⣿⣿⠀⠀⠀⠀⠀⣿⣿⡇⠀⢹⣿⡆⠀⠀⠀⣸⣿⠇⠀⠀⠀ 
-- ⠀⠀⠀⠀⠀⠀ ⢿⣿⣦⣄⣀⣠⣴⣿⣿⠁⠀⠈⠻⣿⣿⣿⣿⡿⠏⠀⠀⠀⠀ 
-- ⠀⠀⠀⠀⠀⠀⠀⠈⠛⠻⠿⠿⠿⠿⠋⠁⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀
--                   sus
local function Thievery_CheckHardOrSoft()
    if checkTarget("target") == true then return "target" end
    if checkTarget("softenemy") == true then return "softenemy" end
    return nil
end
-- --------------------------------------------------------------------------
local function checkAndHandleStealth(self)
    if IsStealthed() then
        stealthed = true
        local erapusuThreshold = 0.3
        local erapusuCounter = 0
        self:SetScript("OnUpdate", function(self, elapsed)
            erapusuCounter = erapusuCounter + elapsed
            if erapusuCounter < erapusuThreshold then
                return
            end
            erapusuCounter = 0
            if InCombatLockdown() then
                self:SetScript("OnUpdate", nil)
            end
            if stealthed == false then
                self:SetScript("OnUpdate", nil)
            end
            Thievery_UpdateState(self)
        end)
    else
        stealthed = false
        Thievery_UpdateState(self)
        self:SetScript("OnUpdate", nil)
    end
end
function Thievery_Events(self, event, unit)
    if event == "PLAYER_SOFT_INTERACT_CHANGED" or event == "PLAYER_SOFT_ENEMY_CHANGED" or event == "PLAYER_TARGET_CHANGED" then
        PPMode = false
        sapMode = false
        clearTable(target)
        Thievery_PPCooldownFrame:Clear()
        if InCombatLockdown() then return end
        local targetType = Thievery_CheckHardOrSoft()
        if targetType then
            validTarget = true
            local guid = Thievery_ScrubSecret(UnitGUID(targetType))
            if guid then
                target.guid = guid
                local npcID = getIDFromGUID(target.guid)
                target.npcID = tonumber(npcID)
                target.name = UnitName(targetType)
                local _
                _, target.creatureType = UnitCreatureType(targetType)
                target.classification = UnitClassification(targetType)
                -- 1)player 2)target ORDER ON PURPOSE, to avoid checking reputations
                target.reaction = UnitReaction("player", targetType)
            end
        else
            validTarget = false
        end
        if not IsTargetLoose() then
            --printTargetInfo()
        end
        checkAndHandleStealth(self)
        Thievery_UpdateState(self)
    elseif event == "UPDATE_STEALTH" then
        checkAndHandleStealth(self)
        Thievery_UpdateState(self)
    elseif event == "PLAYER_REGEN_DISABLED" then
        Thievery_Deactivate(self)
        Thievery.configPanel.checkboxes2.lockpicking:Disable()
    elseif event == "PLAYER_REGEN_ENABLED" then
        Thievery_UpdateState(self)
        Thievery.configPanel.checkboxes2.lockpicking:Enable()
    end
end

-- /dump C_Spell.IsSpellUsable(PICKPOCKET_SPELLID)
-- /dump SpellCanTargetUnit()

-- /dump SpellCanTargetUnit("target")


local nameToID = {
    ["ContainerFrame1"] = 1,
    ["ContainerFrame2"] = 2,
    ["ContainerFrame3"] = 3,
    ["ContainerFrame4"] = 4,
    ["ContainerFrame5"] = 5,
    ["ContainerFrame6"] = 6,
}