TownsfolkUtil.lua

local L = LibStub("AceLocale-3.0"):GetLocale("TownsfolkTracker")

-- Lua APIs
local pairs, table = pairs, table

-- WoW APIs
local UnitLevel, UnitGUID = UnitLevel, UnitGUID
local GetPlayerInfoByGUID = GetPlayerInfoByGUID

function TownsfolkUtil_GetPlayerLevel()
    return UnitLevel("player");
end

function TownsfolkUtil_GetPlayerClass()
    local playerGUID = UnitGUID("player");
    local _, englishClass = GetPlayerInfoByGUID(playerGUID);
    return englishClass;
end

function TownsfolkUtil_GetPlayerFaction()
    local englishFaction = UnitFactionGroup("player");
    return englishFaction;
end

function TownsfolkUtil_GetPlayerDungeonRecommendation(playerLevel, lowLevelBound, highLevelBound)
    if (playerLevel < lowLevelBound) then
        return 4
    end

    if (playerLevel > highLevelBound) then
        return 0
    end

    local levelRange = highLevelBound - lowLevelBound
    local modifiedLevel = playerLevel - lowLevelBound
    local levelGroup = modifiedLevel / levelRange

    if (levelGroup > 0.66) then
        return 1
    end
    if (levelGroup > 0.33) then
        return 2
    end
    return 3
end

function TownsfolkUtil_DungeonAttunement(attunement, alwaysShow)
    local r, g, b = 1, 0, 0
    if (attunement.item) then
        local count = GetItemCount(attunement.item, false, false)
        if (count == 0 or alwaysShow) then
            if (count > 0) then
                g, b = 1, 1
            end
            GameTooltip:AddLine(format("  " .. L["Requires %s"], L[attunement.name]), r, g, b)
        end
    end
    if (attunement.quest) then
        local complete = false
        if (type(attunement.quest) == "number") then
            complete = IsQuestComplete(attunement.quest)
        else
            for questId in pairs(attunement.quest) do
                if (not complete) then
                    complete = IsQuestComplete(questId)
                end
            end
        end

        if (not complete or alwaysShow) then
            if (complete) then
                g, b = 1, 1
            end
            GameTooltip:AddLine(format("  " .. L["Complete %s"], L[attunement.name]), r, g, b)
        end
    end
end

function TownsfolkUtil_IsInstanceType(folktype)
    return folktype == TF_DUNGEON or folktype == TF_RAID
end

function TownsfolkUtil_IsTrainerType(folktype)
    return folktype == TF_PROFESSION_TRAINER or folktype == TF_CLASS_TRAINER
end

function TownsfolkUtil_PairsByKeys(t, f)
    local a = {}
    for n in pairs(t) do table.insert(a, n) end
    table.sort(a, f)
    local i = 0 -- iterator variable
    local iter = function() -- iterator function
        i = i + 1
        if a[i] == nil then return nil
        else return a[i], t[a[i]]
        end
    end
    return iter
end

function TownsfolkUtil_GetTrainerTitle(trainerType)
    if (trainerType == TF_CLASS.DRUID) then
        return "Druid Trainer"
    elseif (trainerType == TF_CLASS.DRUID) then
        return "Druid Trainer"
    elseif (trainerType == TF_CLASS.HUNTER) then
        return "Hunter Trainer"
    elseif (trainerType == TF_CLASS.MAGE) then
        return "Mage Trainer"
    elseif (trainerType == TF_CLASS.PALADIN) then
        return "Paladin Trainer"
    elseif (trainerType == TF_CLASS.PRIEST) then
        return "Priest Trainer"
    elseif (trainerType == TF_CLASS.ROGUE) then
        return "Rogue Trainer"
    elseif (trainerType == TF_CLASS.SHAMAN) then
        return "Shaman Trainer"
    elseif (trainerType == TF_CLASS.WARLOCK) then
        return "Warlock Trainer"
    elseif (trainerType == TF_CLASS.WARRIOR) then
        return "Warrior Trainer"
    elseif (trainerType == TF_PROFESSION.WEAPON) then
        return "Weapon Master"
    elseif (trainerType == TF_PROFESSION.RIDING) then
        return "Riding Instructor"
    else
        return trainerType .. " Trainer"
    end
end

function TownsfolkUtil_GetTrainerTag(trainerType, expertise, branch)
    local profession

    -- profession
    if (trainerType == TF_PROFESSION.TAILORING) then
        profession = "Tailor"
    elseif (trainerType == TF_PROFESSION.LEATHERWORKING) then
        profession = "Leatherworker"
    elseif (trainerType == TF_PROFESSION.BLACKSMITHING) then
        profession = "Blacksmith"
    elseif (trainerType == TF_PROFESSION.ALCHEMY) then
        profession = "Alchemist"
    elseif (trainerType == TF_PROFESSION.ENGINEERING) then
        profession = "Engineer"
    elseif (trainerType == TF_PROFESSION.ENCHANTING) then
        profession = "Enchanter"
    else
        profession = trainerType
    end

    if (branch == TF_PROFESSION_BRANCH.ARMORSMITH or branch == TF_PROFESSION_BRANCH.WEAPONSMITH) then
        return branch
    end

    if (branch) then
        return expertise .. " " .. branch .. " " .. profession
    else
        return expertise .. " " .. profession
    end
end

function TownsfolkUtil_MenuLabel(label)
    local info = Lib_UIDropDownMenu_CreateInfo()
    info.text, info.notCheckable, info.isTitle = label, true, true
    return info
end

function TownsfolkUtil_ContainsValue(table, value)
    for _, val in pairs(table) do
        if (val == value) then
            return true
        end
    end
    return false
end

function TownsfolkUtil_CopyTable(table)
    local newTable = {}
    for key, val in pairs(table) do
        newTable[key] = val
    end
    return newTable
end