CleverAlchemistHelper.lua

CAHelper = {}

local ADDON_NAME = "CleverAlchemistHelper"
local ADDON_VERSION	= "1.1"
local ESOUI_URL = "https://www.esoui.com/downloads/info2688-CleverAlchemistHelper.html"

local variableVersion = 2
local savedVarsName = "CAHelperVars"

local defaults = {
	pvpOnly = false
}

local LAM = LibAddonMenu2

local panelData = {
    type = "panel",
	name = "Clever Alchemist Helper",
	author = "@adjutant [EU]",
	version = ADDON_VERSION,
	website = ESOUI_URL,
}

local optionsData = {

	[1] = {
		type = "checkbox",
        name = "PvP only mode",
		tooltip = "Only activate in Cyrodiil and Battlegrounds",
        getFunc = function() return CAHelper.settings.pvpOnly end,
        setFunc = function(value) CAHelper.settings.pvpOnly = value end,
        width = "full",
		requiresReload = true
  	},

    [2] = {
			type    = "header",
			name    = nil,
    },

    [3] = {
        type = "description",
        title = nil,
        text = "This addon is in active development. Feature requests and bug reports are very welcome! Please leave your comments at ESOUI.",
        width = "full",
    },
}

local currentHotbar
local slotIndex
local SET_ID = 225 -- hardcoded Clever Alchemist set ID
local pastState
local ZoneIsPvP
local isPotion

local gear = {}

local STATE -- an attempt to make an ENUM

local NEVER = 'never'
local ALWAYS = 'always'
local MAINBAR = 'mainBar'
local OFFBAR = 'offBar'

local SLOTS = { -- a lot of rudimentary experimental/debug stuff here, will remove later.
    [EQUIP_SLOT_HEAD] = { EQUIP_SLOT_HEAD, 0, 'Head'},
    [EQUIP_SLOT_NECK] = { EQUIP_SLOT_NECK, 1, 'Neck' },
    [EQUIP_SLOT_CHEST] = { EQUIP_SLOT_CHEST, 2, 'Chest' },
    [EQUIP_SLOT_SHOULDERS] =  { EQUIP_SLOT_SHOULDERS, 3, 'Shoulder'},
    [EQUIP_SLOT_MAIN_HAND] = { EQUIP_SLOT_MAIN_HAND, 4, 'MainHand'},
    [EQUIP_SLOT_OFF_HAND] =  { EQUIP_SLOT_OFF_HAND, 5, 'OffHand' },
    [EQUIP_SLOT_WAIST] =  { EQUIP_SLOT_WAIST, 6, 'Belt' },
    -- [7] = { EQUIP_SLOT_WRIST, 7 , 'wut' },
    [EQUIP_SLOT_LEGS] =  { EQUIP_SLOT_LEGS, 8, 'Leg' },
    [EQUIP_SLOT_FEET] =  { EQUIP_SLOT_FEET, 9, 'Foot' },
    -- [10] = { EQUIP_SLOT_COSTUME, 10, 'disguise?' },
    [EQUIP_SLOT_RING1] =  { EQUIP_SLOT_RING1, 11, 'Ring1' },
    [EQUIP_SLOT_RING2] =  { EQUIP_SLOT_RING2, 12, 'Ring2' },
    --[13] = { EQUIP_SLOT_POISON, 13, 'poison1'},
    --[14] = { EQUIP_SLOT_BACKUP_POISON, 14, 'poison2'},
    -- [15] = { EQUIP_SLOT_RANGED, 15, 'wut'},
    [EQUIP_SLOT_HAND] = { EQUIP_SLOT_HAND, 16, 'Glove'},
    -- [17] = { EQUIP_SLOT_CLASS1, 17, 'wut'},
    -- [18] = { EQUIP_SLOT_CLASS2, 18, 'wut'},
    -- [19] = { EQUIP_SLOT_CLASS3, 19, 'wut'},
    [EQUIP_SLOT_BACKUP_MAIN] =  { EQUIP_SLOT_BACKUP_MAIN, 20, 'BackupMain' },
    [EQUIP_SLOT_BACKUP_OFF] = { EQUIP_SLOT_BACKUP_OFF, 21, 'BackupOff' }
}

local function stateToChat()
    if pastState ~= STATE and pastState then
        if (STATE == MAINBAR) then
            d("[Clever Alchemist Helper]: Set detected on Mainbar")
        elseif (STATE == OFFBAR) then
            d("[Clever Alchemist Helper]: Set detected on Offbar")
        elseif (STATE == NEVER) then
            d("[Clever Alchemist Helper]: Not enough pieces equipped. Addon is OFF.")
        elseif (STATE == ALWAYS) then
            d("[Clever Alchemist Helper]: Too many set pieces equipped. Addon is OFF.")
        end
		if (CAHelper.settings.pvpOnly) then d("[Clever Alchemist Helper]: Warning! PvP zone only mode.") end
    end

    pastState = STATE
end

local function toggleVisibility() -- TODO: rewrite this garbage
	if (STATE == MAINBAR) then
        CAHelperReticleIcon:SetHidden(false)
		CAHelperReticleIcon:SetAlpha(0)
		if currentHotbar == 0 then
			CAHelperReticleIcon:SetAlpha(0.5)
		end
    elseif (STATE == OFFBAR) then
        CAHelperReticleIcon:SetHidden(false)
		CAHelperReticleIcon:SetAlpha(0)
		if currentHotbar == 1 then
			CAHelperReticleIcon:SetAlpha(0.5)
		end
    elseif (STATE == NEVER) then
        CAHelperReticleIcon:SetHidden(true)
    elseif (STATE == ALWAYS) then
		CAHelperReticleIcon:SetHidden(true)
    end
end

function eval()
    local equipped = 0
    local totalBonus = 0
    local armorBonus = 0
    local mainBarBonus = 0
    local offBarBonus = 0

    for i, v in pairs(gear) do
        if gear[i]["isAlchemist"] == true then
            equipped = equipped + 1
            if gear[i]["type"] == ITEMTYPE_ARMOR then
                armorBonus = armorBonus + 1
            elseif gear[i]["slot"] == EQUIP_SLOT_MAIN_HAND or gear[i]["slot"] == EQUIP_SLOT_OFF_HAND then
                if gear[i]["type"] == EQUIP_TYPE_TWO_HAND then
                    mainBarBonus = mainBarBonus + 2
                elseif gear[i]["type"] == EQUIP_TYPE_ONE_HAND then
                    mainBarBonus = mainBarBonus + 1
                end
            elseif gear[i]["slot"] == EQUIP_SLOT_BACKUP_MAIN or gear[i]["slot"] == EQUIP_SLOT_BACKUP_OFF then
                if gear[i]["type"] == EQUIP_TYPE_TWO_HAND then
                    offBarBonus = offBarBonus + 2
                elseif gear[i]["type"] == EQUIP_TYPE_ONE_HAND then
                    offBarBonus = offBarBonus + 1
                end
            end
        end
    end

    if equipped < 4 then
        STATE = NEVER
    elseif armorBonus >= 5 then
        STATE = ALWAYS
    elseif armorBonus <= 2 then
        STATE = NEVER
    elseif mainBarBonus == offBarBonus == 2 then
        STATE = ALWAYS
    elseif mainBarBonus + armorBonus >= 5 then
        STATE = MAINBAR
    elseif offBarBonus + armorBonus >= 5 then
        STATE = OFFBAR
    else
        STATE = NEVER
    end

    stateToChat()
	
	toggleVisibility()
end

function parseSlot(eventCode, bagId, slotIndex, isNewItem, itemSoundCategory, updateReason, stackCountChange)
	local itemType
	itemType, _ = GetItemType(BAG_WORN, slotIndex)
	if itemType == ITEMTYPE_WEAPON then
		_, _, _, _, _, itemType, _, _ = GetItemInfo(BAG_WORN, slotIndex)
	end
    local link = GetItemLink(BAG_WORN, slotIndex)
	local hasSet, setName, numBonuses, numEquipped, _, setId = GetItemLinkSetInfo(link)
	if hasSet and setId == SET_ID then
		gear[slotIndex] = { link = itemLink, type = itemType, isAlchemist = true, slot = slotIndex }
	else
		gear[slotIndex] = { link = itemLink, type = itemType, isAlchemist = false, slot = slotIndex }
	end
	if CAHelper.ready then
		eval()
	end
end

function getEquippedGear()
 --   for x = 1, #SLOTS do
    for i, v in pairs(SLOTS) do
        parseSlot(0, 0, v[1], 0, 0, 0, 0)
    end
end

local function createReticleControl()
	CAHelperReticleIcon = WINDOW_MANAGER:CreateControl("CAHelperRecticleControl", ZO_ReticleContainer, CT_TEXTURE)
	CAHelperReticleIcon:ClearAnchors()
	CAHelperReticleIcon:SetAnchor(CENTER, ZO_ReticleContainer, CENTER, 45, 0)
	CAHelperReticleIcon:SetTexture("/esoui/art/treeicons/gamepad/progression_levelup_choiceofpotion.dds")
	CAHelperReticleIcon:SetDimensions(32, 32)
	CAHelperReticleIcon:SetColor(240, 240, 240, 0.5)
	CAHelperReticleIcon:SetHidden(true)
end

local function registerAnimations(control)
    local isValidAnchor, point, relativeTo, relativePoint, offsetX, offsetY = control:GetAnchor()

    local TranslateAnimation, TranslateTimeline = CreateSimpleAnimation(ANIMATION_TRANSLATE, control)
        TranslateAnimation:SetTranslateOffsets(offsetX, offsetY, offsetX - 15, offsetY)
        TranslateAnimation:SetDuration(70)
        TranslateAnimation:SetEasingFunction(ZO_EaseInQuadratic)

    local fadeInAnimation, FadeInTimeline = CreateSimpleAnimation(ANIMATION_ALPHA, control)
        fadeInAnimation:SetAlphaValues(0, 0.5)
        fadeInAnimation:SetDuration(200)
        fadeInAnimation:SetEasingFunction(ZO_EaseOutQuadratic)

    return TranslateTimeline, FadeInTimeline
end

local function checkIfZoneIsPvP()
    if IsPlayerInAvAWorld() or IsActiveWorldBattleground() then
        ZoneIsPvP = true
    end
end

local function getQuickSlotItemInfo(eventCode, actionSlotIndex)
	local itemLink = GetSlotItemLink(actionSlotIndex)
	local itemType = GetItemLinkItemType(itemLink)
	if itemType == ITEMTYPE_POTION then	
		isPotion = true
	else
		isPotion = false
	end
--	d(actionSlotIndex .. " : " .. itemType .. " " .. GetItemLinkName(itemLink) .. " " .. tostring(isPotion))
end

function Initialize()
    EVENT_MANAGER:UnregisterForEvent(ADDON_NAME, EVENT_ADD_ON_LOADED)

    createReticleControl()
    local TranslateTimeline, FadeInTimeline = registerAnimations(CAHelperReticleIcon)

    for i, v in pairs(SLOTS) do
        gear[i] = { id = 0, link = 0, type = 0, isAlchemist = false, slot = 0 }
    end
	
	CAHelper.settings = ZO_SavedVars:NewAccountWide(savedVarsName, variableVersion, "globals", defaults, GetWorldName(), "AllAccountsTheSame")
	
	LAM:RegisterAddonPanel("Clever Alchemist Helper", panelData)
	LAM:RegisterOptionControls("Clever Alchemist Helper", optionsData)
	
	currentHotbar = ACTION_BAR_ASSIGNMENT_MANAGER:GetCurrentHotbarCategory() -- get initial bar

    getEquippedGear()
    eval()
    CAHelper.ready = true
	
	getQuickSlotItemInfo() -- if there's a potion slotted on login
	EVENT_MANAGER:RegisterForEvent(ADDON_NAME, EVENT_ACTIVE_QUICKSLOT_CHANGED, getQuickSlotItemInfo)

    EVENT_MANAGER:RegisterForEvent(ADDON_NAME, EVENT_INVENTORY_SINGLE_SLOT_UPDATE, parseSlot)
    EVENT_MANAGER:AddFilterForEvent(ADDON_NAME, EVENT_INVENTORY_SINGLE_SLOT_UPDATE, REGISTER_FILTER_BAG_ID, BAG_WORN)
    EVENT_MANAGER:AddFilterForEvent(ADDON_NAME, EVENT_INVENTORY_SINGLE_SLOT_UPDATE, REGISTER_FILTER_INVENTORY_UPDATE_REASON, INVENTORY_UPDATE_REASON_DEFAULT)
	
--	SLASH_COMMANDS["/cahelper"] = stateToChat

    if CAHelper.settings.pvpOnly then
        EVENT_MANAGER:RegisterForEvent(ADDON_NAME, EVENT_ZONE_CHANGED, checkIfZoneIsPvP)
        checkIfZoneIsPvP() -- if we are in AvA/BG on login
    end

    if (CAHelper.settings.pvpOnly and ZoneIsPvP) or not CAHelper.settings.pvpOnly then
	
        EVENT_MANAGER:RegisterForEvent(ADDON_NAME, EVENT_ACTION_SLOTS_FULL_UPDATE, function(eventCode, isHotbarSwap)
			if isHotbarSwap then
				currentHotbar = 1 - currentHotbar -- switch between 0 and 1
			end
			if (currentHotbar == 0 and STATE == MAINBAR) or 
				(currentHotbar == 1 and STATE == OFFBAR) then
				FadeInTimeline:PlayForward()
			else
				FadeInTimeline:PlayBackward() -- fadeOut :P
			end
		end)

		SecurePostHook(ZO_Reticle, "OnUpdate", function(self)
			local interactionPossible = not self.interact:IsHidden()
			if (interactionPossible or PLAYER_TO_PLAYER:HasTarget() or IsGameCameraUnitHighlightedAttackable()) and (GetUnitStealthState("player") == 0) then
				TranslateTimeline:PlayForward()
			else
				TranslateTimeline:PlayBackward()
			end
		end)

		ZO_PreHook("ZO_ActionBar_CanUseActionSlots", function()
			-- flag = not flag -- Since ZO_ActionBar_CanUseActionSlots is called twice for each ability cast
			-- if flag then
			local slotNum = tonumber(debug.traceback():match('keybind = "ACTION_BUTTON_(%d)')) -- get pressed button
				if slotNum == 9 then -- break if consumable or empty slot
					if ((currentHotbar == 1 and STATE == MAINBAR) or 
				(currentHotbar == 0 and STATE == OFFBAR)) and isPotion then
						return true -- ESO won't run ability press if PreHook returns true
					end
				else return false end
		end)
    end
end

local function OnAddOnLoaded(event, addonName)
	if addonName == ADDON_NAME then
		Initialize()
	end
end

EVENT_MANAGER:RegisterForEvent(ADDON_NAME, EVENT_ADD_ON_LOADED, OnAddOnLoaded)