GroupNavigator.lua

local ADDON, addon = ...
local config = addon.Config

local GamePadButtonList = addon.GamePadButtonList
local GamePadModifierList = addon.GamePadModifierList

local ActionList = {
   ["UNITNAVUP"] = true,
   ["UNITNAVDOWN"] = true,
   ["UNITNAVLEFT"] = true,
   ["UNITNAVRIGHT"] = true,
   ["CLEARTARGETING"] = true
}
config:ConfigListAdd("GamePadActions", ActionList, "NONE")
config:ConfigListAdd("GamePadModifierActions", ActionList, "NONE")

local GroupNavigatorMixin = {
   SoftTargetFrame = CrossHotbarAddon_GroupNavigator_SoftTarget,
   ActiveBindings = {}
}

function GroupNavigatorMixin:OnLoad()
   self:RegisterEvent("ADDON_LOADED")
   self:RegisterEvent("GROUP_JOINED")
   self:RegisterEvent("GROUP_LEFT")
   self:RegisterEvent("GROUP_ROSTER_UPDATE")
   self:RegisterEvent("ZONE_CHANGED_NEW_AREA")
   self:RegisterEvent("UNIT_NAME_UPDATE")
   self:RegisterEvent("PLAYER_ROLES_ASSIGNED")
   self:RegisterEvent("PLAYER_ENTERING_WORLD")
   self:WrapOnClickDiscrete()
   
   addon:AddApplyCallback(GenerateClosure(self.ApplyConfig, self))
end

function GroupNavigatorMixin:OnEvent(event, ...)
   if event == "ADDON_LOADED" then
      self:updateRoster()
      local value = GetCVarBool("ActionButtonUseKeyDown");
      self:RegisterForClicks("AnyUp", "AnyDown")
   elseif event == "GROUP_JOINED" then
      self:updateRoster()
   elseif event == "GROUP_LEFT" then
      self:updateRoster()
      self.SoftTargetFrame:SetAlpha(0)      
      if not InCombatLockdown() then
         self.SoftTargetFrame:Hide()
      end
   elseif event == "GROUP_ROSTER_UPDATE" then
      self:updateRoster()
   elseif event == "ZONE_CHANGED_NEW_AREA" then
      self:updateRoster()
   elseif event == "UNIT_NAME_UPDATE" then
      self:updateRoster()
   elseif event == "PLAYER_ROLES_ASSIGNED" then
      self:updateRoster()                        
   elseif event == "PLAYER_ENTERING_WORLD" then
      self:updateRoster()
      if ChatFrame1EditBox then
         ChatFrame1EditBox:SetAltArrowKeyMode(false)
      end
      local texCoords = { ["Raid-TargetFrame"] = { 0.00781250, 0.55468750, 0.28906250, 0.55468750 }}
      
      self.SoftTargetFrame.selectionHighlight:SetTexture("Interface\\RaidFrame\\Raid-FrameHighlights");
      self.SoftTargetFrame.selectionHighlight:SetTexCoord(unpack(texCoords["Raid-TargetFrame"]));
      self.SoftTargetFrame.selectionHighlight:SetVertexColor(0.2, 1.0, 0.6);
      self.SoftTargetFrame.selectionHighlight:SetAllPoints(self.SoftTargetFrame);
      
      local frameWidth = EditModeManagerFrame:GetRaidFrameWidth(true);
      local frameHeight = EditModeManagerFrame:GetRaidFrameHeight(true);
      
      self.SoftTargetFrame:SetSize(frameWidth+6, frameHeight+6)
      self.SoftTargetFrame:SetAlpha(0)
      self.SoftTargetFrame:Hide()
      
      SecureHandlerSetFrameRef(self, "softtarget", self.SoftTargetFrame)
      if UnitInRaid("player") or UnitInParty("player") then
         self:updateRoster()
      end
   end
end

function GroupNavigatorMixin:AddStateHandlers()
   self:SetAttribute("modstate", 0)
   self:SetAttribute("modname", "")
   
   self:SetAttribute("SetActionBindings", [[
      local prefix = ...
      if prefix == "" or prefix == "TRIG" then
         self:ClearBindings()
      end
      local binding = self:GetAttribute(prefix .. "UNITNAVUP")
      if binding and binding ~= "" then self:SetBindingClick(true, binding, self:GetName(), "Button4") end
      binding = self:GetAttribute(prefix .. "UNITNAVDOWN")                  
      if binding and binding ~= "" then self:SetBindingClick(true, binding, self:GetName(), "Button5") end
      binding = self:GetAttribute(prefix .. "UNITNAVLEFT")                  
      if binding and binding ~= "" then self:SetBindingClick(true, binding, self:GetName(), "LeftButton") end
      binding = self:GetAttribute(prefix .. "UNITNAVRIGHT")
      if binding and binding ~= "" then self:SetBindingClick(true, binding, self:GetName(), "RightButton") end
      binding = self:GetAttribute(prefix .. "CLEARTARGETING")                  
      if binding and binding ~= "" then self:SetBindingClick(true, binding, self:GetName(), "MiddleButton") end
    ]])
end

function GroupNavigatorMixin:ClearConfig()
   for action in pairs(ActionList) do
      self:SetAttribute(action, "")
      for i,modifier in ipairs(GamePadModifierList) do
         self:SetAttribute(modifier .. action, "")
      end
   end
end

function GroupNavigatorMixin:ApplyConfig()
   self:ClearConfig()
   self.ActiveBindings = {}
   for button, attributes in pairs(config.PadActions) do
      if ActionList[ attributes.ACTION ] then
         self:SetAttribute(attributes.ACTION, attributes.BIND)
      end
      
      for i,modifier in ipairs(GamePadModifierList) do
         if ActionList[ attributes[modifier .. "ACTION"] ] then
            self:SetAttribute(modifier .. attributes[modifier .. "ACTION"], attributes.BIND)
            self:SetAttribute(modifier .. "TRIG" .. attributes[modifier .. "ACTION"], attributes.BIND)
         end
      end
   end
   self:Execute([[ self:ClearBindings(); self:RunAttribute("SetActionBindings", "") ]])
end

function GroupNavigatorMixin:updateRoster() 
   if not InCombatLockdown() then 
      local raid_id = UnitInRaid("player")
      if raid_id then
         
         local raid_units = {}
         for i=1, 40 do
            table.insert(raid_units, "raid"..i)
         end
         
         if CompactRaidFrameContainer.groupMode == "flush" then
            table.sort(raid_units, CompactRaidFrameContainer.flowSortFunc);
         end
         
         local raid_size = GetNumGroupMembers()
         
         local player_unit = "1"
         local player_group = "1"
         
         group_units = {}
         for i = 1, raid_size do  
            local id = tonumber(string.sub(raid_units[i], 5));
            name, rank, subgroup = GetRaidRosterInfo(id);
            if group_units[subgroup] == nil then group_units[subgroup] = {} end
            table.insert(group_units[subgroup], raid_units[i])
            if id == raid_id then 
               player_unit = tostring(table.getn(group_units[subgroup]))
               player_group = tostring(subgroup)
            end
         end
         
         local r_str = ""
         for i, group in ipairs(group_units) do
            r_str = r_str .. "["
            for j, unit in ipairs(group) do
               r_str = r_str .. " " .. unit
            end
            r_str = r_str .. "]"
         end
         
         self:SetAttribute("player_unit", player_unit)
         self:SetAttribute("player_group", player_group)
         self:SetAttribute("group_change", "true")
         self:SetAttribute("group_units", r_str) 
         self:AddUnitFrameRefs()
      elseif UnitInParty("player") then
         local party_size = GetNumGroupMembers()
         
         local party_units = {}
         table.insert(party_units, "player")
         for i=2, 5 do
            table.insert(party_units, "party"..(i-1))
         end
         
         if CompactRaidFrameContainer.groupMode == "flush" then
            table.sort(party_units, CompactRaidFrameContainer.flowSortFunc);
         end
         
         local player_unit = "1"
         local p_str = ""
         for i = 1, party_size do
            if party_units[i] == "player" then 
               player_unit = tostring(i)
            end
            if (i-1) % party_size == 0 then
               p_str = p_str .. "["
            end
            p_str = p_str .. " " .. party_units[i]
            if i % party_size == 0 then
               p_str = p_str .. "]"
            end
         end
         
         self:SetAttribute("player_unit", player_unit)
         self:SetAttribute("player_group", "1")
         self:SetAttribute("group_change", "true")
         self:SetAttribute("group_units", p_str)
         self:AddUnitFrameRefs()
      end
   end
end

function GroupNavigatorMixin:AddUnitFrameRefs()

   if not InCombatLockdown() then 
      local inparty = true
      local groupType = CompactRaidGroupTypeEnum.Party;
      if UnitInRaid("player") then 
         inparty = false
         groupType = CompactRaidGroupTypeEnum.Raid;
      end 

      local hasUnits = false
      
      if CompactRaidFrameContainer.groupMode == "flush" then
         for _,frame in pairs(CompactRaidFrameContainer.flowFrames) do
            if frame.unit then
               hasUnits = true
               SecureHandlerSetFrameRef(self, frame.unit, frame)
            end
         end
         self:WrapOnClickFlush()
      else
         if inparty == false then
            for i = 1,8 do
               for j = 1,5 do
                  local frame = _G["CompactRaidGroup"..i.."Member"..j]
                  if frame and frame:IsVisible() then 
                     local frame_unit = frame:GetAttribute("unit")
                     if frame_unit then
                        hasUnits = true
                        SecureHandlerSetFrameRef(self, i .. "_" .. j, frame)
                     end
                  end               
                  local frame = _G["CellRaidFrameHeader"..i.."UnitButton"..j]
                  if frame and frame:IsVisible() then
                     local frame_unit = frame:GetAttribute("unit")
                     if frame_unit then
                        hasUnits = true
                        SecureHandlerSetFrameRef(self, i .. "_" .. j, frame)
                     end
                  end
                  local frame = _G["Grid2LayoutHeader"..i.."UnitButton"..j]
                  if frame and frame:IsVisible() then
                     local frame_unit = frame:GetAttribute("unit")
                     if frame_unit then
                        hasUnits = true
                        SecureHandlerSetFrameRef(self, i .. "_" .. j, frame)
                     end
                  end
                  local frame = _G["ElvUF_Raid1Group"..i.."UnitButton"..j]
                  if frame and frame:IsVisible() then
                     local frame_unit = frame:GetAttribute("unit")
                     if frame_unit then
                        hasUnits = true
                        SecureHandlerSetFrameRef(self, i .. "_" .. j, frame)
                     end
                  end
                  local frame = _G["ElvUF_Raid2Group"..i.."UnitButton"..j]
                  if frame and frame:IsVisible() then
                     local frame_unit = frame:GetAttribute("unit")
                     if frame_unit then
                        hasUnits = true
                        SecureHandlerSetFrameRef(self, i .. "_" .. j, frame)
                     end
                  end
                  local frame = _G["ElvUF_Raid3Group"..i.."UnitButton"..j]
                  if frame and frame:IsVisible() then
                     local frame_unit = frame:GetAttribute("unit")
                     if frame_unit then
                        hasUnits = true
                        SecureHandlerSetFrameRef(self, i .. "_" .. j, frame)
                     end
                  end
               end
            end
         else
            for j = 1,5 do
               local frame = _G["PartyFrame"]
               if frame and frame:IsVisible() then
                  local memberFrame = frame["MemberFrame" .. j]
                  if memberFrame and memberFrame:IsVisible() then
                     local frame_unit = memberFrame:GetAttribute("unit")
                     if frame_unit then
                        hasUnits = true
                        SecureHandlerSetFrameRef(self, "1_" .. j, memberFrame)
                     end
                  end
               end
               local frame = _G["CompactPartyFrameMember"..j]
               if frame and frame:IsVisible() then
                  local frame_unit = frame:GetAttribute("unit")
                  if frame_unit then
                     hasUnits = true
                     SecureHandlerSetFrameRef(self, "1_" .. j, frame)
                  end
               end
               local frame = _G["CellPartyFrameHeaderUnitButton"..j]
               if frame and frame:IsVisible() then
                  local frame_unit = frame:GetAttribute("unit")
                  if frame_unit then
                     hasUnits = true
                     SecureHandlerSetFrameRef(self, "1_" .. j, frame)
                  end
               end
               local frame = _G["Grid2LayoutHeader1UnitButton"..j]
               if frame and frame:IsVisible() then
                  local frame_unit = frame:GetAttribute("unit")
                  if frame_unit then
                     hasUnits = true
                     SecureHandlerSetFrameRef(self, "1_" .. j, frame)
                  end
               end
               local frame = _G["ElvUF_PartyGroup1UnitButton"..j]
               if frame and frame:IsVisible() then
                  local frame_unit = frame:GetAttribute("unit")
                  if frame_unit then
                     hasUnits = true
                     SecureHandlerSetFrameRef(self, "1_" .. j, frame)
                  end
               end
            end
         end
         self:WrapOnClickDiscrete()
      end
      
      if hasUnits then
         self.SoftTargetFrame:Show()
         SecureHandlerSetFrameRef(self, "softtarget", self.SoftTargetFrame)
      else
         self.SoftTargetFrame:Hide()
      end
   end
end

function GroupNavigatorMixin:WrapOnClickDiscrete()
   SecureHandlerUnwrapScript(self, "OnClick")
   SecureHandlerWrapScript(self, "OnClick", self,  [[

   if not down then return end

   if not lastunit then lastunit = 1 end
   if not lastgroup then lastgroup = 1 end
   
   local newunit = "player"
   local player_unit = tonumber(self:GetAttribute("player_unit"))
   local player_group = tonumber(self:GetAttribute("player_group"))
   
   if not group_units then group_units = table.new() end
   if not hidden_units then hidden_units = table.new() end

   local num_units = 0
   local num_groups = #group_units
   local recalc = self:GetAttribute("group_change")

   for _,frame in pairs(hidden_units) do
      if frame and frame:IsVisible() then
         recalc = true;
      end
   end

   if recalc then
      -- print("Re-Calc Units")

      local max_group = 1
      if PlayerInGroup() == "raid" then
         max_group = 8
      end

      group_units = table.new()
      hidden_units = table.new()

      for x = 1,max_group do
         local newgroup = true
         for y = 1,5 do
            local frame = self:GetFrameRef(x .. "_" .. y)
            if frame then
               if frame:IsVisible() then 
                  if newgroup then 
                     group_units[#group_units + 1] = table.new()
                     newgroup = false
                  end
                  table.insert(group_units[#group_units], frame)
               else
                  -- print("hidden")
                  table.insert(hidden_units, frame)
               end
            end
         end
      end

      num_groups = #group_units

      if num_groups > 0 then 
         if lastgroup > num_groups then
            lastgroup = num_groups
         end

         num_units = #(group_units[lastgroup])
         if num_units > 0 then
            if lastunit > num_units then
               lastunit = num_units
            end
         end
      end

      self:SetAttribute("group_change", false)
   end

   if num_groups > 0 then 
      num_units = #(group_units[lastgroup])
      if num_units > 0 then
         if PlayerInGroup() == "party" or PlayerInGroup() == "raid" then
            
            local unitfound = false
            local softtarget = self:GetFrameRef("softtarget")
            local frame = group_units[lastgroup][lastunit]

            if softtarget then
               if frame and frame:IsVisible() then
                  if softtarget:GetParent() == frame then
                     unitfound = true
                  end
               end
            end
            
            if unitfound then
               if UnitPlayerOrPetInRaid("target") or
                  UnitPlayerOrPetInParty("target") then
    
                  if button == "Button4" then
                     lastunit = (lastunit + num_units - 1) % num_units
                  elseif button == "Button5" then
                     lastunit = (lastunit + 1) % num_units
                  elseif button == "LeftButton" then
                     lastgroup = (lastgroup + num_groups - 1) % num_groups
                  elseif button == "RightButton" then
                     lastgroup = (lastgroup + 1) % num_groups
                  end
                  
                  if lastgroup == 0 then
                     lastgroup = num_groups
                  end
                  
                  num_units = #(group_units[lastgroup])
                  
                  if lastunit > num_units then
                     lastunit = num_units
                  elseif lastunit == 0 then
                     lastunit = num_units
                  end

                  -- print(lastgroup, lastunit)
               end
            else 
               lastgroup = player_group
               lastunit = player_unit
            end
            
            frame = nil
            if group_units[lastgroup] then 
               frame = group_units[lastgroup][lastunit]
            end

            if frame and frame:IsVisible() then
               newunit = frame:GetAttribute("unit")
               self:SetAttribute("unit", newunit)
               if softtarget then
                  softtarget:SetParent(frame)
                  softtarget:ClearAllPoints()
                  softtarget:SetPoint("CENTER", frame, "CENTER")
                  local frameWidth = frame:GetWidth();
                  local frameHeight = frame:GetHeight();
                  softtarget:SetWidth(frameWidth+6)
                  softtarget:SetHeight(frameHeight+6)
                  softtarget:SetAlpha(0.75)
               end
            elseif softtarget then
               softtarget:SetAlpha(0)
               self:SetAttribute("group_change", true)
            end
            
         else
            self:SetAttribute("unit", newunit)
         end
      else
         self:SetAttribute("unit", newunit)
      end
   else
      self:SetAttribute("unit", newunit)
   end
   ]])
end

function GroupNavigatorMixin:WrapOnClickFlush()
   SecureHandlerUnwrapScript(self, "OnClick")
   SecureHandlerWrapScript(self, "OnClick", self,  [[

   if not down then return end

   if not lastunit then lastunit = 1 end
   if not lastgroup then lastgroup = 1 end
   if not group_units then group_units = table.new() end
   local newunit = "player"
   local player_unit = tonumber(self:GetAttribute("player_unit"))
   local player_group = tonumber(self:GetAttribute("player_group"))
   if self:GetAttribute("group_change") == "true" then
      local g_str = self:GetAttribute("group_units")
      group_units = table.new()
      local i = 1
      for subgrp in string.gmatch(g_str, "%[([%w+%s]+)%]") do 
         group_units[i] = table.new()
         local j = 1
         for unitid in string.gmatch(subgrp, "%w+") do 
            table.insert(group_units[i], unitid)
            j = j + 1
         end
         i = i + 1
      end
      local num_groups = #group_units
      if num_groups > 0 then 
         if lastgroup > num_groups then
            lastgroup = num_groups
         end
         local num_units = #(group_units[lastgroup])
         if num_units > 0 then
            if lastunit > num_units then
               lastunit = num_units
            end
         end
      end
      self:SetAttribute("group_change", "false")
   end
   
   local num_groups = #group_units
   if num_groups > 0 then 
      local num_units = #(group_units[lastgroup])
      if num_units > 0 then
         if PlayerInGroup() == "party" or PlayerInGroup() == "raid" then
            
            local unitfound = false
            local softtarget = self:GetFrameRef("softtarget")
            if softtarget then
               local frame = self:GetFrameRef(group_units[lastgroup][lastunit])
               if frame and frame:IsVisible() then
                  local frame_unit = frame:GetAttribute("unit")
                  if frame_unit == group_units[lastgroup][lastunit] then
                     unitfound = true
                  end
               end
            end
            
            if unitfound then
               local targetunit = group_units[lastgroup][lastunit]
               if (UnitPlayerOrPetInRaid("target") or UnitPlayerOrPetInParty("target")) or
                  not (UnitPlayerOrPetInRaid(targetunit) or UnitPlayerOrPetInParty(targetunit)) then      
                  if button == "Button4" then
                     lastunit = (lastunit + num_units - 1) % num_units
                  elseif button == "Button5" then
                     lastunit = (lastunit + 1) % num_units
                  elseif button == "LeftButton" then
                     lastgroup = (lastgroup + num_groups - 1) % num_groups
                  elseif button == "RightButton" then
                     lastgroup = (lastgroup + 1) % num_groups
                  end
                  
                  if lastgroup == 0 then
                     lastgroup = num_groups
                  end
                  
                  num_units = #(group_units[lastgroup])
                  
                  if lastunit > num_units then
                     lastunit = num_units
                  elseif lastunit == 0 then
                     lastunit = num_units
                  end
               end
            else 
               lastgroup = player_group
               lastunit = player_unit
            end
            
            newunit = group_units[lastgroup][lastunit]
            self:SetAttribute("unit", newunit)
               
            if softtarget then
               local found = false
               local frame = self:GetFrameRef(newunit)
               if frame and frame:IsVisible() then
                  local frame_unit = frame:GetAttribute("unit")
                  if frame_unit == newunit then
                     softtarget:SetParent(frame)
                     softtarget:ClearAllPoints()
                     softtarget:SetPoint("CENTER", frame, "CENTER")
                     softtarget:SetAlpha(0.75)
                     found = true
                  end
               end
               if found == false then
                  softtarget:SetAlpha(0)
               end
            end
         else
            self:SetAttribute("unit", newunit)
         end
      else
         self:SetAttribute("unit", newunit)
      end
   else
      self:SetAttribute("unit", newunit)
   end
   ]])
end

local CreateGroupNavigator = function(parent)
   local GroupNavigator = CreateFrame("Button", ADDON .. "GroupNavigator", parent,
                                      "SecureActionButtonTemplate, SecureHandlerStateTemplate")

   Mixin(GroupNavigator, GroupNavigatorMixin)

   GroupNavigator:SetFrameStrata("BACKGROUND")
   GroupNavigator:SetPoint("TOP", parent:GetName(), "LEFT", 0, 0)
   GroupNavigator:Hide()

   GroupNavigator:SetAttribute("player_unit", "1")
   GroupNavigator:SetAttribute("player_group", "1")
   GroupNavigator:SetAttribute("*type1", "target")
   GroupNavigator:SetAttribute("*type2", "target")
   GroupNavigator:SetAttribute("*type3", "macro")
   GroupNavigator:SetAttribute("*type4", "target")
   GroupNavigator:SetAttribute("*type5", "target")
   GroupNavigator:SetAttribute("unit", "player")
   GroupNavigator:SetAttribute("group_change", "true")
   GroupNavigator:SetAttribute("group_units", "[player party1 party2 party3 party4]")
   GroupNavigator:SetAttribute("macrotext3", "/cleartarget\n/stopspelltarget\n")

   GroupNavigator:AddStateHandlers()

   GroupNavigator:HookScript("OnEvent", GroupNavigator.OnEvent)
   GroupNavigator:OnLoad()

   addon.GroupNavigator = GroupNavigator
end

addon.CreateGroupNavigator = CreateGroupNavigator