Cast.lua

-------------------------------------------------------------------------------
-- SmartBuff Cast
-------------------------------------------------------------------------------

local SG = SMARTBUFF_GLOBALS;

-- SMARTBUFF_doCast
-- Validates whether a spell can be cast on the given unit before buffing.
-- Parameters:
--   unit      (string)  Unit token to buff (e.g. "player", "party1")
--   id        (number)  Spell ID; nil returns code 9
--   spellName (string)  Localized spell name used for cooldown/range/mana checks
--   levels    (table)   Rank/level data (passed through from caller; unused here)
--   type      (number)  Buff type constant (SMARTBUFF_CONST_*)
-- Returns:
--   (number)  0 = castable
--             1 = spell on cooldown
--             3 = target out of range
--             4 = cooldown exceeds SG.maxSkipCoolDown
--             6 = not enough mana/rage/energy
--             7 = another tracking ability is already active
--             9 = spell ID is nil
function SMARTBUFF_doCast(unit, id, spellName, levels, type)
  if (id == nil) then return 9; end
  if (type == SMARTBUFF_CONST_TRACK and (GetTrackingTexture() ~= "Interface\\Minimap\\Tracking\\None")) then
    return 7;
  end
  -- check if spell has cooldown (remaining time, not raw duration)
  local cdStart, cdDuration = GetSpellCooldown(spellName);
  local cd = 0;
  if (cdStart and cdStart > 0 and cdDuration) then
    cd = (cdStart + cdDuration) - GetTime();
    if (cd < 0) then
      cd = 0;
    end
  end
  if (cd > SG.maxSkipCoolDown) then
    return 4;
  elseif (cd > 0) then
    return 1;
  end
  -- Rangecheck
  if (type == SMARTBUFF_CONST_GROUP or type == SMARTBUFF_CONST_ITEMGROUP) then
    if (C_Spell.SpellHasRange(spellName)) then
      if (not C_Spell.IsSpellInRange(spellName, unit)) then
        return 3;
      end
    else
      if (UnitInRange(unit) ~= 1) then
        return 3;
      end
    end
  end
  -- check if you have enough mana/energy/rage to cast
  local isUsable, notEnoughMana = IsUsableSpell(spellName);
  if (notEnoughMana) then
    return 6;
  end
  return 0;
end


-- SMARTBUFF_InitActionButtonPos
-- Restores the SmartBuff action button position from saved options or defaults.
-- Parameters: none
-- Returns: nothing (no-op while in combat lockdown)
function SMARTBUFF_InitActionButtonPos()
  if (InCombatLockdown()) then return end

  SG.isInitBtn = true;
  if (SG.O.ActionBtnX == nil) then
    SMARTBUFF_SetButtonPos(SmartBuff_KeyButton);
  else
    SmartBuff_KeyButton:ClearAllPoints();
    SmartBuff_KeyButton:SetPoint("TOPLEFT", UIParent, "TOPLEFT", SG.O.ActionBtnX, SG.O.ActionBtnY);
  end
end

-- SMARTBUFF_ResetAll
-- Wipes all saved buff templates and options, then reloads the UI.
-- Parameters: none
-- Returns: nothing
function SMARTBUFF_ResetAll()
  wipe(SMARTBUFF_Buffs);
  wipe(SMARTBUFF_Options);
  ReloadUI();
end

-- SMARTBUFF_SetButtonPos
-- Saves the current screen position of a frame to SG.O.ActionBtnX / ActionBtnY.
-- Parameters:
--   self (Frame)  Frame whose top-left corner position is stored
-- Returns: nothing
function SMARTBUFF_SetButtonPos(self)
  local x, y = self:GetLeft(), self:GetTop() - UIParent:GetHeight();
  SG.O.ActionBtnX = x;
  SG.O.ActionBtnY = y;
end

-- SMARTBUFF_RebindKeys
-- Applies override key bindings for the SmartBuff secure action button.
-- Binds SMARTBUFF_BIND_TRIGGER and optional mouse-wheel scroll buffing.
-- Parameters: none
-- Returns: nothing (no-op while in combat lockdown)
function SMARTBUFF_RebindKeys()
  if (InCombatLockdown()) then return; end
  local i;
  SG.isRebinding = true;
  local key1, key2;
  for i = 1, GetNumBindings(), 1 do
    local command, k1, k2 = GetBinding(i);
    if (k1 and k1 == "MOUSEWHEELUP" and command ~= "SmartBuff_KeyButton") then
      SG.O.OldWheelUp = command;
    elseif (k1 and k1 == "MOUSEWHEELDOWN" and command ~= "SmartBuff_KeyButton") then
      SG.O.OldWheelDown = command;
    end
    if (command and command == "SMARTBUFF_BIND_TRIGGER") then
      key1, key2 = k1, k2;
      break;
    end
  end

  -- Use override bindings so the key triggers the secure button (works in combat, doesn't alter saved bindings)
  ClearOverrideBindings(SmartBuffFrame);
  if (key1) then
    SetOverrideBindingClick(SmartBuffFrame, false, key1, "SmartBuff_KeyButton", "LeftButton");
  end
  if (key2) then
    SetOverrideBindingClick(SmartBuffFrame, false, key2, "SmartBuff_KeyButton", "LeftButton");
  end

  if (SG.O.ScrollWheelUp) then
    SG.isKeyUpChanged = true;
    SetOverrideBindingClick(SmartBuffFrame, false, "MOUSEWHEELUP", "SmartBuff_KeyButton", "MOUSEWHEELUP");
  else
    if (SG.isKeyUpChanged) then
      SG.isKeyUpChanged = false;
      SetOverrideBinding(SmartBuffFrame, false, "MOUSEWHEELUP");
    end
  end

  if (SG.O.ScrollWheelDown) then
    SG.isKeyDownChanged = true;
    SetOverrideBindingClick(SmartBuffFrame, false, "MOUSEWHEELDOWN", "SmartBuff_KeyButton", "MOUSEWHEELDOWN");
  else
    if (SG.isKeyDownChanged) then
      SG.isKeyDownChanged = false;
      SetOverrideBinding(SmartBuffFrame, false, "MOUSEWHEELDOWN");
    end
  end

  SG.isRebinding = false;
end

-- SMARTBUFF_ResetBindings
-- Restores saved mouse-wheel bindings and reapplies SmartBuff override bindings.
-- Parameters: none
-- Returns: nothing
function SMARTBUFF_ResetBindings()
  if (not SG.isRebinding) then
    SG.isRebinding = true;
    if (SG.O.OldWheelUp == "SmartBuff_KeyButton") then
      SetBinding("MOUSEWHEELUP", "CAMERAZOOMIN");
    else
      SetBinding("MOUSEWHEELUP", SG.O.OldWheelUp);
    end
    if (SG.O.OldWheelDown == "SmartBuff_KeyButton") then
      SetBinding("MOUSEWHEELDOWN", "CAMERAZOOMOUT");
    else
      SetBinding("MOUSEWHEELDOWN", SG.O.OldWheelDown);
    end
    SaveBindings(GetCurrentBindingSet());
    SMARTBUFF_RebindKeys();
  end
end


-- SmartBuff commandline menu ---------------------------------------------------------------------------------------

-- SMARTBUFF_ShowSAButton
-- Shows or hides the SmartBuff action button based on the HideSAButton option.
-- Parameters: none
-- Returns: nothing (no-op while in combat lockdown)
function SMARTBUFF_ShowSAButton()
  if (not InCombatLockdown()) then
    if (SG.O.HideSAButton) then
      SmartBuff_KeyButton:Hide();
    else
      SmartBuff_KeyButton:Show();
    end
  end
end

local sScript;

-- SMARTBUFF_OnClick
-- Debug handler for direct clicks on the SmartBuff action button.
-- Parameters:
--   obj (Frame)  Button frame that received the click
-- Returns: nothing
function SMARTBUFF_OnClick(obj)
  SMARTBUFF_AddMsgD("OnClick");
end


local lastBuffType = "";
local cvarChanged = false;

-- SMARTBUFF_OnPreClick
-- Secure pre-click handler: runs SMARTBUFF_Check and sets button attributes for the cast.
-- Supports left-click (mode 0) and scroll-wheel buffing (mode 5).
-- Parameters:
--   self   (Button)  SmartBuff_KeyButton secure action button
--   button (string)  Mouse button name (e.g. "LeftButton", "MOUSEWHEELUP")
--   down   (boolean) True if the button was pressed, false if released
-- Returns: nothing
function SMARTBUFF_OnPreClick(self, button, down)
  if (not SG.isInit) then return end
  local mode = 0;
  local unitsLevel = nil
  if (button) then
    if (button == "MOUSEWHEELUP" or button == "MOUSEWHEELDOWN") then
      mode = 5;
    end
  end

  if (not InCombatLockdown()) then
    self:SetAttribute("type", nil);
    self:SetAttribute("unit", nil);
    self:SetAttribute("spell", nil);
    self:SetAttribute("item", nil);
    self:SetAttribute("macrotext", nil);
    self:SetAttribute("target-slot", nil);
    self:SetAttribute("target-item", nil);
    self:SetAttribute("action", nil);
  end

  -- we need to adjust the cvar to enable scroll buffing, we 
  -- only need to do this if the cvar is set to 0 and the user has enabled the fix.
  if (SG.O.FixCasting and C_CVar.GetCVar("ActionButtonUseKeyDown") == "0") then
    C_CVar.SetCVar("ActionButtonUseKeyDown", 1);
    cvarChanged = true;
  end


  local td;
  if (lastBuffType == "") then
    td = 0.8;
  else
    td = SG.GlobalCd;
  end

  if (UnitCastingInfo("player")) then
    SG.tAutoBuff = GetTime() + 0.7;
    return;
  end

  if (GetTime() < (SG.tAutoBuff + td)) then return end

  SG.tAutoBuff = GetTime();
  lastBuffType = "";
  SG.currentUnit = nil;
  SG.currentSpell = nil;

  if (not InCombatLockdown()) then
    local ret, actionType, spellName, slot, unit, buffType = SMARTBUFF_Check(mode);
    if (ret and ret == 0 and actionType and spellName and unit) then
      lastBuffType = buffType;
      self:SetAttribute("type", actionType);
      self:SetAttribute("unit", unit);
      if (actionType == SMARTBUFF_ACTION_SPELL) then
        if (slot and slot > 0 and unit == "player") then
          self:SetAttribute("type", "macro");
          self:SetAttribute("macrotext", string.format("/use %s\n/use %i\n/click StaticPopup1Button1", spellName, slot));
          SMARTBUFF_AddMsgD("Weapon buff "..spellName..", "..slot);
        else
           self:SetAttribute("spell", spellName);
        end

        if (SG.cBuffIndex[spellName]) then
          SG.currentUnit = unit;
          SG.currentSpell = spellName;
        end

      elseif (actionType == SMARTBUFF_ACTION_ITEM and slot) then
        self:SetAttribute("item", spellName);
        if (slot > 0) then
          self:SetAttribute("type", "macro");
          self:SetAttribute("macrotext", string.format("/use %s\n/use %i\n/click StaticPopup1Button1", spellName, slot));
        end
      elseif (actionType == "action" and slot) then
        self:SetAttribute("action", slot);
      else
        SMARTBUFF_AddMsgD("Preclick: not supported actiontype -> " .. actionType);
      end
      SG.tLastCheck = GetTime() - SG.O.AutoTimer + SG.GlobalCd;
    end
  end
end

-- SMARTBUFF_OnPostClick
-- Secure post-click handler: clears button attributes, restores camera zoom and CVars.
-- Parameters:
--   self   (Button)  SmartBuff_KeyButton secure action button
--   button (string)  Mouse button name
--   down   (boolean) True if the button was pressed, false if released
-- Returns: nothing
function SMARTBUFF_OnPostClick(self, button, down)
  if (not SG.isInit) then return end
  if (button and not SG.isPrompting) then
    if (button == "MOUSEWHEELUP") then
      CameraZoomIn(1);
    elseif (button == "MOUSEWHEELDOWN") then
      CameraZoomOut(1);
    end
  end

  if (InCombatLockdown()) then return end

  self:SetAttribute("type", nil);
  self:SetAttribute("unit", nil);
  self:SetAttribute("spell", nil);
  self:SetAttribute("item", nil);
  self:SetAttribute("target-slot", nil);
  self:SetAttribute("target-item", nil);
  self:SetAttribute("macrotext", nil);
  self:SetAttribute("action", nil);

  if not SG.O.HideSAButtonNoAction and not SG.O.HideSAButton then
    SMARTBUFF_SetButtonTexture(SmartBuff_KeyButton, SG.imgSB);
  end

  -- if cvarChanged is true then it was originally set to 0, we need to put it back.
  if (cvarChanged) then
    C_CVar.SetCVar("ActionButtonUseKeyDown", 0);
    cvarChanged = false;
  end
  
end

-- SMARTBUFF_SetButtonTexture
-- Updates the SmartBuff action button icon when the texture changes.
-- Parameters:
--   button  (Button)  Button frame to update
--   texture (string)  Texture path or file ID for the normal texture
--   text    (string)  Optional label text (currently unused)
-- Returns: nothing
function SMARTBUFF_SetButtonTexture(button, texture, text)
  if (button and texture and texture ~= SG.sLastTexture) then
    SG.sLastTexture = texture;
    button:SetNormalTexture(texture);
    if (text) then
      --button.title:SetText(spell);
    end
  end
end