local NPC_IDS = {400104}
local ITEM_ID = 60124

local mainHandItems = {6905, 10756, 11087}

local destinations = {
    {x = -482, y = -2710, z = 94.303, o = 4.2},
    {x = -353, y = -2681, z = 95.88, o = 0.0929},
    {x = -579, y = -2650, z = 95.633, o = 3.12},
    {x = -348, y = -2507, z = 95.563, o = 1.49},
}

local messages = {
    "For the Horde!",
    "No mercy for our enemies!",
    "Let's show these Scourge dogs what it means to mess with the Horde!",
}


local function OnGossipHello(event, player, creature)
    if player:HasItem(ITEM_ID) then
        player:GossipClearMenu()
        player:GossipMenuAddItem(8, "|TInterface\\icons\\inv_sword_39:40:40:-35|t|cff610B0BGive Weapon|r", 1, 0)
        player:GossipMenuAddItem(9, "2-Handed Axe", 1, 1001)
        player:GossipMenuAddItem(9, "2-Handed Mace", 1, 1002)
        player:GossipMenuAddItem(9, "2-Handed Sword", 1, 1003)

        if creature:GetData("equipmentSet") then
            player:GossipMenuAddItem(8, "|TInterface\\icons\\ability_hunter_markedfordeath:40:40:-35|t|cffC41F3BSend to Location|r", 2, 0)
            player:GossipMenuAddItem(0, "Reinforce the East Entrance!", 2, 2001)
            player:GossipMenuAddItem(0, "Reinforce the North Entrance!", 2, 2002)
            player:GossipMenuAddItem(0, "Reinforce the South Entrance!", 2, 2003)
            player:GossipMenuAddItem(0, "Reinforce the West Entrance!", 2, 2004)
            player:GossipMenuAddItem(2, "|t|cff0101DFFollow me into battle!|r", 2, 2005) 
        end

        player:GossipSendMenu(1, creature)
    else
        player:SendBroadcastMessage("You need Horde Armaments!")
    end
end

--Define your abilities here
local function CastCleave(eventId, delay, calls, creature)
    if creature:IsInCombat() then
        creature:CastSpell(creature:GetVictim(), 845, true)
    end
end

local function CastHeroicStrike(eventId, delay, calls, creature)
    if creature:IsInCombat() then
        creature:CastSpell(creature:GetVictim(), 78, true)
    end
end

local function CastThunderClap(eventId, delay, calls, creature)
    if creature:IsInCombat() then
        creature:CastSpell(creature:GetVictim(), 8078, true)
    end
end

local function CastEnrage(eventId, delay, calls, creature)
    if creature:IsInCombat() then
        creature:CastSpell(creature:GetVictim(), 3547, true)
    end
end

local function CastRend(eventId, delay, calls, creature)
    if creature:IsInCombat() then
        creature:CastSpell(creature:GetVictim(), 8548, true)
    end
end

local function CastCharge(eventId, delay, calls, creature)
    if creature:IsInCombat() then
        creature:CastSpell(creature:GetVictim(), 24193, true)
    end
end


local function RegisterAbilities(creature, weaponType)
    if creature:GetData("abilitiesRegistered") then
        return
    end
    creature:SetData("abilitiesRegistered", true)

    if weaponType == 1 then -- 2-Handed Axe Abilities
        creature:RegisterEvent(CastCleave, math.random(8000, 12000), 0)
        creature:RegisterEvent(CastHeroicStrike, math.random(5000, 9000), 0)
    elseif weaponType == 2 then -- 2-Handed Mace Abilities
        creature:RegisterEvent(CastThunderClap, math.random(9000, 14000), 0)
        creature:RegisterEvent(CastEnrage, math.random(15000, 19000), 0)
    elseif weaponType == 3 then -- 2-Handed Sword Abilities
        creature:RegisterEvent(CastRend, math.random(11000, 15000), 0)
        creature:RegisterEvent(CastCharge, 100, 1)
    end
end

local function CheckIfReachedDestination(eventId, delay, calls, creature)
    local x, y, z = creature:GetHomePosition()
    local currentPositionX, currentPositionY, currentPositionZ = creature:GetPosition()
    if math.floor(currentPositionX) == math.floor(x) and math.floor(currentPositionY) == math.floor(y) and math.floor(currentPositionZ) == math.floor(z) then
        creature:SetData("hasReachedDestination", true)
        creature:EmoteState(375)
        creature:RemoveEventById(eventId)
    end
end

local function OnEnterCombat(event, creature, target)
    local weaponType = creature:GetData("weaponType")
    if weaponType and not creature:GetData("abilitiesRegistered") then
        RegisterAbilities(creature, weaponType)
    end
end


local function OnLeaveCombat(event, creature)
    creature:RemoveEvents()
    -- creature:EmoteState(375)
    creature:SetData("abilitiesRegistered", false)
end


local function DespawnCreature(eventId, delay, calls, creature)
    creature:DespawnOrUnsummon()
end

local function OnGossipSelect(event, player, creature, sender, intid, code)
    if intid == 0 then
        OnGossipHello(event, player, creature)
    elseif intid >= 1001 and intid <= 1003 then
        local index = intid - 1000
        local selectedMainHand = mainHandItems[index]
        creature:SetEquipmentSlots(selectedMainHand, 0, 0)
		creature:SetData("weaponType", index) 
        creature:SetData("equipmentSet", true)
        creature:PerformEmote(66)
        RegisterAbilities(creature, index) -- Add this line to register abilities
        OnGossipHello(event, player, creature)
    end

    if intid >= 2001 and intid <= 2004 then
        local index = intid - 2000
        local destination = destinations[index]
        local finalX = destination.x + math.random(-10, 10)
        local finalY = destination.y + math.random(-10, 10)
        local finalZ = destination.z
        local finalO = destination.o
        creature:MoveTo(0, finalX, finalY, finalZ, true)
        local battlecryIndex = math.random(1, #messages)
        creature:SendUnitYell(messages[battlecryIndex], 0)
        creature:EmoteState(375)
        creature:SetHomePosition(finalX, finalY, finalZ, finalO)
        creature:SetNPCFlags(0)
        player:RemoveItem(ITEM_ID, 1)
        player:KilledMonsterCredit(creature:GetEntry())
        player:GossipComplete()
		 creature:RegisterEvent(DespawnCreature, 180000)
		creature:SetData("hasReachedDestination", false)
		creature:RegisterEvent(CheckIfReachedDestination, 1000, 0) 
end

    if intid == 2005 then
        local randomAngle = math.random() * 2 * math.pi
        creature:MoveFollow(player, 5, randomAngle) 
        local battlecryIndex = math.random(1, #messages)
        creature:SendUnitYell(messages[battlecryIndex], 0)
		player:KilledMonsterCredit(400104)
        creature:SetNPCFlags(0)
        player:RemoveItem(ITEM_ID, 1)
        player:GossipComplete()
		creature:RegisterEvent(DespawnCreature, 180000)
    end
end

local function OnCreatureDied(event, creature, killer)
    creature:SetEquipmentSlots(0, 0, 0)
	creature:SetData("equipmentSet", false)
    creature:RemoveEvents()
end

for _, npcId in ipairs(NPC_IDS) do
    RegisterCreatureGossipEvent(npcId, 1, OnGossipHello)
    RegisterCreatureGossipEvent(npcId, 2, OnGossipSelect)
    RegisterCreatureEvent(npcId, 1, OnEnterCombat)
    RegisterCreatureEvent(npcId, 2, OnLeaveCombat)
    RegisterCreatureEvent(npcId, 4, OnCreatureDied)
end