FishingHelper.lua

local addonName, addonTable = ...
local FH = {}
addonTable[1] = FH
_G["FishingHelper"] = FH
_G["BINDING_NAME_CLICK FishingHelperMainButton:LeftButton"] = "Cast Fishing"
_G["BINDING_NAME_CLICK FishingHelperStopButton:LeftButton"] = "Stop Fishing"
-- Default database
local defaultDB = {
    framePosition = { point = "CENTER", relativePoint = "CENTER", xOfs = 0, yOfs = 0 },
    outfit = {
        [1] = nil,  -- Head (INVSLOT_HEAD)
        [10] = nil, -- Hands (INVSLOT_HANDS)
        [8] = nil,  -- Feet (INVSLOT_FEET)
        [16] = nil, -- Main Hand (INVSLOT_MAINHAND)
    },
    lureID = nil,
    savedGear = {}, -- Used to restore previous gear
    minimapPos = 45 -- Minimap button angle in degrees
}

local frame = CreateFrame("Frame")
frame:RegisterEvent("ADDON_LOADED")
frame:RegisterEvent("PLAYER_REGEN_ENABLED")
frame:RegisterEvent("BAG_UPDATE")
frame:RegisterEvent("PLAYER_EQUIPMENT_CHANGED")
frame:RegisterEvent("UNIT_SPELLCAST_CHANNEL_START")
frame:RegisterEvent("UNIT_SPELLCAST_CHANNEL_STOP")
frame:RegisterEvent("UNIT_SPELLCAST_START")
frame:RegisterEvent("UNIT_SPELLCAST_STOP")
frame:RegisterEvent("UNIT_SPELLCAST_SUCCEEDED")
frame:RegisterEvent("UNIT_SPELLCAST_FAILED")
frame:RegisterEvent("UNIT_SPELLCAST_INTERRUPTED")

-- Slash command
SLASH_FISHINGHELPER1 = "/fh"
function SlashCmdList.FISHINGHELPER(msg, editbox)
    if FH.UI and FH.UI.frame then
        if FH.UI.frame:IsShown() then
            FH.UI.frame:Hide()
        else
            FH.UI.frame:Show()
        end
    end
end

function FH:Initialize()
    if not FishingHelperDB then
        FishingHelperDB = CopyTable(defaultDB)
    end
    -- Ensure all keys exist
    for k, v in pairs(defaultDB) do
        if FishingHelperDB[k] == nil then
            FishingHelperDB[k] = type(v) == "table" and CopyTable(v) or v
        end
    end
    self.db = FishingHelperDB

    self.updateTimer = 0
    frame:SetScript("OnUpdate", function(self, elapsed)
        FH.updateTimer = FH.updateTimer + elapsed
        if FH.updateTimer > 0.2 then
            FH.updateTimer = 0
            if FH.UpdateMacro then
                FH:UpdateMacro()
            end
        end
    end)

    self.volumeSaved = false
    self.autoLootSaved = false

    print("|cFF00FF00Fishing Helper|r loaded! Type /fh to toggle the frame.")
end

function FH:EnhanceVolume()
    if self.volumeSaved then return end
    self.oldMasterVolume = GetCVar("Sound_MasterVolume")
    self.oldSFXVolume = GetCVar("Sound_SFXVolume")

    SetCVar("Sound_MasterVolume", "1.0")
    SetCVar("Sound_SFXVolume", "1.0")
    self.volumeSaved = true
end

function FH:RestoreVolume()
    if not self.volumeSaved then return end
    SetCVar("Sound_MasterVolume", self.oldMasterVolume or "1.0")
    SetCVar("Sound_SFXVolume", self.oldSFXVolume or "1.0")
    self.volumeSaved = false
end

function FH:EnableAutoLoot()
    if self.autoLootSaved then return end
    self.oldAutoLoot = GetCVar("autoLootDefault")
    if self.oldAutoLoot ~= "1" then
        SetCVar("autoLootDefault", "1")
        self.autoLootSaved = true
    end
end

function FH:RestoreAutoLoot()
    if not self.autoLootSaved then return end
    SetCVar("autoLootDefault", self.oldAutoLoot or "0")
    self.autoLootSaved = false
end

function FH:SaveCurrentGear()
    if InCombatLockdown() then return end
    local needEquip = false
    for slot, itemID in pairs(self.db.outfit) do
        if itemID and GetInventoryItemID("player", slot) ~= itemID then
            needEquip = true
            break
        end
    end

    if needEquip then
        wipe(self.db.savedGear)
        for slot, itemID in pairs(self.db.outfit) do
            if itemID then
                local link = GetInventoryItemLink("player", slot)
                self.db.savedGear[slot] = link
            end
        end
        if self.db.outfit[16] and not self.db.outfit[17] then
            self.db.savedGear[17] = GetInventoryItemLink("player", 17)
        end
        self:UpdateStopMacro()
    end
end

function FH:UpdateStopMacro()
    if InCombatLockdown() or not self.UI or not self.UI.stopButton then return end

    local lines = {}
    -- Restore offhand AFTER mainhand, so sort it (16 comes before 17)
    local sortedSlots = {}
    for slot, link in pairs(self.db.savedGear) do
        table.insert(sortedSlots, slot)
    end
    table.sort(sortedSlots)

    for _, slot in ipairs(sortedSlots) do
        local link = self.db.savedGear[slot]
        if link then
            local name = C_Item.GetItemInfo(link)
            if name then
                table.insert(lines, "/equipslot " .. slot .. " " .. name)
            end
        end
    end

    -- When stopping fishing, run a macro to equip old gear, AND a secure snippet or lua to restore AutoLoot.
    -- We can just hook the "OnClick" of stopButton securely, no, insecurely to restore autoloot, since we aren't in combat.

    self.UI.stopButton:SetAttribute("type", "macro")
    if #lines > 0 then
        self.UI.stopButton:SetAttribute("macrotext", table.concat(lines, "\n"))
    else
        self.UI.stopButton:SetAttribute("macrotext", "")
    end
end

function FH:UpdateMacro()
    if InCombatLockdown() then return end
    if not self.UI or not self.UI.mainButton then return end

    local needEquip = false
    local itemsToEquip = {}
    local anyOutfitConfigured = false
    local tooltipItem = nil

    -- Check outfit
    for slot, itemID in pairs(self.db.outfit) do
        if itemID then
            anyOutfitConfigured = true
            local equippedID = GetInventoryItemID("player", slot)
            if equippedID ~= itemID then
                needEquip = true
                local name = C_Item.GetItemInfo(itemID)
                if name then
                    table.insert(itemsToEquip, "/equipslot " .. slot .. " " .. name)
                    if not tooltipItem then tooltipItem = name end
                end
            end
        end
    end

    local function ApplyMacro(macroStr, icon, tooltip)
        self.UI.mainButton:SetAttribute("type", "macro")
        self.UI.mainButton:SetAttribute("macrotext", macroStr)
        if self.UI.mainButton.icon then
            self.UI.mainButton.icon:SetTexture(icon)
        end

        local idx = GetMacroIndexByName("FishHelper")
        if idx > 0 then
            local mText = macroStr
            if tooltip then
                mText = "#showtooltip " .. tooltip .. "\n" .. macroStr
            else
                mText = "#showtooltip\n" .. macroStr
            end
            local name, icon, body = GetMacroInfo(idx)
            if body ~= mText then
                EditMacro(idx, "FishHelper", "INV_Misc_QuestionMark", mText)
            end
        end
        self.currentMacroText = macroStr
        self.currentMacroTooltip = tooltip
    end

    if needEquip then
        -- Add the script call to save gear right before we equip
        table.insert(itemsToEquip, 1, "/run FishingHelper:SaveCurrentGear()")
        local macroStr = table.concat(itemsToEquip, "\n")
        ApplyMacro(macroStr, "Interface\\Icons\\INV_Chest_Cloth_17", tooltipItem)
        return
    end

    -- We are equipped, so we enable auto loot.
    if anyOutfitConfigured then
        self:EnableAutoLoot()
    end

    -- Outfit is equipped. Check Lure.
    local lureID = self.db.lureID
    local recentlySucceeded = self.lastLureSuccess and (GetTime() - self.lastLureSuccess < 2.0)
    if lureID and anyOutfitConfigured and not self.isApplyingLure and not recentlySucceeded then
        local name = C_Item.GetItemInfo(lureID)
        if name and C_Item.GetItemCount(lureID) > 0 then
            local hasMainHandEnchant, mainHandExpiration = GetWeaponEnchantInfo()
            if not hasMainHandEnchant or (mainHandExpiration and mainHandExpiration < 20000) then
                local macroStr = "/use " .. name .. "\n/use 16"
                ApplyMacro(macroStr, C_Item.GetItemIconByID(lureID), name)
                return
            end
        end
    end

    -- Ready to fish
    ---@diagnostic disable-next-line: undefined-global
    local _, _, icon = GetSpellInfo("Fishing")
    if not icon then icon = "Interface\\Icons\\Trade_Fishing" end
    ApplyMacro("/cast Fishing", icon, "Fishing")
end

frame:SetScript("OnEvent", function(self, event, ...)
    if event == "ADDON_LOADED" then
        local loadedAddon = ...
        if loadedAddon == addonName then
            FH:Initialize()
            if FH.UI and FH.UI.Initialize then
                FH.UI:Initialize()
            end
            self:UnregisterEvent("ADDON_LOADED")
            FH:UpdateMacro()
            FH:UpdateStopMacro()

            -- Hook Stop Button Click for AutoLoot Restore
            if FH.UI and FH.UI.stopButton then
                FH.UI.stopButton:HookScript("OnClick", function()
                    FH:RestoreAutoLoot()
                end)
            end
        end
    elseif event == "PLAYER_REGEN_ENABLED" or event == "PLAYER_EQUIPMENT_CHANGED" then
        if event == "PLAYER_EQUIPMENT_CHANGED" then
            local slotID = ...
            if slotID and FH.db.savedGear[slotID] then
                local currentItemID = GetInventoryItemID("player", slotID)
                local fishingItemID = FH.db.outfit[slotID]

                -- Check if the current item matches our configured fishing outfit
                if fishingItemID then
                    if currentItemID ~= fishingItemID then
                        -- User manually equipped something else in an outfit slot
                        FH.db.savedGear[slotID] = nil
                    end
                else
                    -- No fishing item configured for this slot, but we have something saved (e.g. slot 17)
                    -- If the slot is no longer empty, it's a manual change.
                    if currentItemID and currentItemID ~= 0 then
                        FH.db.savedGear[slotID] = nil
                    end
                end
            end
        end

        if FH.UpdateMacro then
            FH:UpdateMacro()
            FH:UpdateStopMacro()
        end
    elseif event == "BAG_UPDATE" then
        if FH.UI and FH.UI.UpdateLureCount then
            FH.UI:UpdateLureCount()
        end
    elseif event == "UNIT_SPELLCAST_CHANNEL_START" then
        local unitTarget, _, spellID = ...
        if unitTarget == "player" then
            ---@diagnostic disable-next-line: undefined-global
            local name = GetSpellInfo(spellID)
            if name == "Fishing" then
                FH.isChannelingFishing = true
                FH:EnhanceVolume()
            end
        end
    elseif event == "UNIT_SPELLCAST_CHANNEL_STOP" then
        local unitTarget, _, spellID = ...
        if unitTarget == "player" then
            ---@diagnostic disable-next-line: undefined-global
            local name = GetSpellInfo(spellID)
            if name == "Fishing" then
                FH.isChannelingFishing = false
                C_Timer.After(1, function()
                    if not FH.isChannelingFishing then
                        FH:RestoreVolume()
                    end
                end)
            end
        end
    elseif event == "UNIT_SPELLCAST_START" then
        local unitTarget, _, spellID = ...
        if unitTarget == "player" then
            ---@diagnostic disable-next-line: undefined-global
            local spellName = GetSpellInfo(spellID)
            local lureID = FH.db.lureID
            if lureID then
                local lureName = C_Item.GetItemInfo(lureID)
                if spellName == lureName then
                    FH.isApplyingLure = true
                    FH.lastLureSuccess = nil
                    FH:UpdateMacro()
                end
            end
        end
    elseif event == "UNIT_SPELLCAST_SUCCEEDED" then
        local unitTarget, _, spellID = ...
        if unitTarget == "player" then
            ---@diagnostic disable-next-line: undefined-global
            local spellName = GetSpellInfo(spellID)
            local lureID = FH.db.lureID
            if lureID then
                local lureName = C_Item.GetItemInfo(lureID)
                if spellName == lureName then
                    FH.lastLureSuccess = GetTime()
                end
            end
        end
    elseif event == "UNIT_SPELLCAST_FAILED" or event == "UNIT_SPELLCAST_INTERRUPTED" then
        local unitTarget, _, spellID = ...
        if unitTarget == "player" then
            ---@diagnostic disable-next-line: undefined-global
            local spellName = GetSpellInfo(spellID)
            local lureID = FH.db.lureID
            if lureID then
                local lureName = C_Item.GetItemInfo(lureID)
                if spellName == lureName then
                    FH.lastLureSuccess = nil
                end
            end
        end
    elseif event == "UNIT_SPELLCAST_STOP" then
        local unitTarget = ...
        if unitTarget == "player" then
            FH.isApplyingLure = false
            FH:UpdateMacro()
        end
    end
end)