WoWInterface/Interface/AddOns/Narcissus/Modules/GemManager/DataProvider_Pandaria.lua

local _, addon = ...
local L = Narci.L;
local DoesItemExistByID = addon.DoesItemExistByID;
local Gemma = addon.Gemma;
local AtlasUtil = Gemma.AtlasUtil;
local BagScan = Gemma.BagScan;
local DataProvider = {};
local GemManagerMixin = {};
DataProvider.GemManagerMixin = GemManagerMixin;
Gemma:AddDataProvider("Pandaria", DataProvider);

local BagUtil = {};

local pairs = pairs;
local ipairs = ipairs;
local tsort = table.sort;

local PATH = "Interface/AddOns/Narcissus/Art/Modules/GemManager/";

local GetItemGemID = C_Item.GetItemGemID;
local GetItemNumSockets = C_Item.GetItemNumSockets;     --10.2.7
local IsEquippableItem = C_Item.IsEquippableItem;
local GetItemInfoInstant = C_Item.GetItemInfoInstant;
local GetItemCount = C_Item.GetItemCount;
local GetInventoryItemLink = GetInventoryItemLink;
local GetCVarBool = C_CVar.GetCVarBool;


local MainFrame;

local GEM_TYPES = {
    [1] = "META",
    [2] = "COGWHEEL",
    [3] = "TINKER",
    [4] = "PRISMATIC",
};

local SLOT_ID = {
    HEAD = 1,
    FEET = 8,

    SHOULDER = 3,
    WRIST = 9,
    HANDS = 10,
    WAIST = 6,

    NECK = 2,
    CHEST = 5,
    LEGS = 7,
    RING1 = 11,
    RING2 = 12,
    TRINKET1 = 13,
    TRINKET2 = 14,
};

local GEM_DATA = {
    --[itemID] = {type, spellID, role, uiOrder, scaleWithIlvl}
    --role: bits 000 (Tank/Healer/DPS): Tank 100(4), DPS 001(1), H/D 011(3), H 010(2)

    --Total: 11
    [221982] = {1, 447598, 4, 00},  --Bulwark of the Black Ox: Charge, Taunt, Ward
    [221977] = {1, 447566, 1, 20},  --Funeral Pyre: Stat, Self Harm
    [220211] = {1, 444954, 2, 60},  --Precipice of Madness: Ward
    [220120] = {1, 444677, 4, 70},  --Soul Tether: Redirect Damage
    [220117] = {1, 444622, 2, 90},  --Ward of Salvation: Restore HP, Overhealing to Ward, AoE
    [219878] = {1, 444128, 7, 85},  --Tireless Spirit: Reduce Resouce Cost
    [219386] = {1, 443389, 7, 35},  --Locus of Power: Stats
    --[216974] = {1, 437495, 1, 40},  --Morphing Elements: Summon Portal, AoE
    [216711] = {1, 426268, 3, 10},  --Chi-ji, the Red Crane --426268, 437018
    [216695] = {1, 437011, 3, 30},  --Lifestorm: Damage then Restore HP and Haste
    [216671] = {1, 426748, 7, 80},  --Thundering Orb: Transform, DR, Movement
    [216663] = {1, 435313, 1, 50},  --Oblivion Sphere: Crit Damage Taken, AoE, Control

    --Total：17
    [218110] = {2, 441759, 7, 45},  --Soulshape
    [218109] = {2, 441749, 7, 10},  --Death's Advance
    [218108] = {2, 441741, 7, 05},  --Dark Pack
    [218082] = {2, 441617, 7, 55},  --Spiritwalker's Grace (Cast while Moving)
    [218046] = {2, 441576, 7, 50},  --Spirit Walk
    [218045] = {2, 441569, 7, 20},  --Door of Shadows
    [218044] = {2, 441564, 7, 35},  --Pursuit of Justice (Passive)
    [218043] = {2, 441559, 7, 80},  --Wild Charge
    [218005] = {2, 441493, 7, 65},  --Stampeding Roar
    [218004] = {2, 441479, 7, 75},  --Vanish
    [218003] = {2, 441467, 7, 30},  --Leap of Faith
    [217989] = {2, 441348, 7, 70},  --Trailblazer
    [217983] = {2, 441299, 7, 15},  --Disengage
    [216632] = {2, 427030, 7, 60},  --Sprint
    [216631] = {2, 427026, 7, 40},  --Roll
    [216630] = {2, 427033, 7, 25},  --Heroic Leap
    [216629] = {2, 427053, 7, 05},  --Blink

    --Total: 36
    [219801] = {3, 427064, 7, 00},  --Ankh of Reincarnation: Self-rez
    [212366] = {3, 429270, 3, 28},  --Arcanist's Edge: Consume shield to deal damage *
    [219944] = {3, 444455, 2, 03},  --Bloodthirsty Coral: Damage taken to Healing
    [219818] = {3, 429007, 7, 06},  --Brilliance: Party Resouce Regen
    [216649] = {3, 436578, 1, 09},  --Brittle: Store Damage Done, Death Trigger AoE
    [216648] = {3, 436577, 7, 12},  --Cold Front: Allies Ward, Enemies Debuff
    [217957] = {3, 441165, 2, 15},  --Deliverance: Store Healing. Healing when Low HP
    [212694] = {3, 433362, 3, 29},  --Enkindle: Grant shield, Damage attackers, Increase haste *
    [212749] = {3, 433361, 1, 102},  --Explosive Barrage: AoE
    [212365] = {3, 429389, 1, 103},  --Fervor: Consume HP to deal holy damage
    [219817] = {3, 429026, 7, 18},  --Freedom: Ckear Loss of Control
    [212916] = {3, 436528, 5, 104},  --Frost Armor: Damage and slow attackers
    [219777] = {3, 428854, 4, 21},  --Grounding: Redirect Harmful Spell
    [217964] = {3, 441209, 2, 27},  --Holy Martyr: Damage Taken to Party Healing
    [216647] = {3, 436571, 1, 24},  --Hailstorm: AoE and Debuff
    [212758] = {3, 433360, 1, 105},  --Incendiary Terror: Damage and Horrify
    [219389] = {3, 443498, 3, 30},  --Lightning Rod: Crit on Ally or Dot on Enemy
    [216624] = {3, 436461, 5, 33},  --Mark of Arrogance: Dot on Attackers
    [216650] = {3, 436583, 7, 36},  --Memory of Vegeance: For every 10s, gain primary stat for every 5% missing HP
    [212759] = {3, 433358, 1, 106},  --Metero Storm: AoE and stun
    [212361] = {3, 427054, 1, 107},  --Opportunist: Grant crit when damaging stunned enemey
    [216625] = {3, 429373, 1, 39},  --Quick Strike: Melee Ability Triggers Additional Autoattacks
    [217961] = {3, 441198, 2, 42},  --Righteous Frenzy: Healing Proc Haste on Ally
    [217927] = {3, 441150, 2, 45},  --Savior: Healing Low HP Ally Grants Ward
    [216651] = {3, 436586, 3, 48},  --Searing Light: Healing to Heal and AoE Damage
    [216626] = {3, 429378, 1, 51},  --Slay: Extra Damage to Low Health Enemy
    [219452] = {3, 443670, 3, 54},  --Static Charge: Heal or Damage
    [219523] = {3, 443834, 1, 57},  --Storm Overload: AoE, Control
    [212362] = {3, 436465, 1, 108},  --Sunstrider's Flourish: Crit cause AoE
    [216627] = {3, 429230, 2, 60},  --Tinkmaster's Shield: Ward after not being damaged for 5s
    [219527] = {3, 443855, 7, 63},  --Vampiric Aura: +Leech, Party Leech
    [216628] = {3, 436467, 3, 66},  --Victory Fire: Enemy Death trigger AoE Damage and Healing
    [217903] = {3, 441092, 3, 69},  --Vindication: Damage Done Heals Allies
    [217907] = {3, 441115, 4, 72},  --Warmth: +Healing Taken, Redistribute Overhealing
    [212760] = {3, 433356, 1, 109},  --Wildfire: Dot, Spreading
    [219516] = {3, 443787, 7, 110},  --Windweaver: +Movement Speed, Falling damage immunity, chance to increase party haste


    [210714] = {4, nil, 1, 30},  --Crit +
    [216644] = {4, nil, 1, 20},  --Crit ++
    [211123] = {4, nil, 1, 10},  --Crit +++
    [211102] = {4, nil, 1, 00},  --Crit +++, STAM
    [210681] = {4, nil, 2, 31},  --Haste +
    [216643] = {4, nil, 2, 21},  --Haste ++
    [211107] = {4, nil, 2, 11},  --Haste +++
    [211110] = {4, nil, 2, 01},  --Haste +++, STAM
    [210715] = {4, nil, 3, 32},  --Mastery +
    [216640] = {4, nil, 3, 22},  --Mastery ++
    [211106] = {4, nil, 3, 12},  --Mastery +++
    [211108] = {4, nil, 3, 02},  --Mastery +++, STAM
    [220371] = {4, nil, 4, 33},  --Vers +
    [220372] = {4, nil, 4, 23},  --Vers ++
    [220374] = {4, nil, 4, 13},  --Vers +++
    [220373] = {4, nil, 4, 03},  --Vers +++, STAM
    [220367] = {4, nil, 5, 35},  --Armor +
    [220368] = {4, nil, 5, 25},  --Armor ++
    [220370] = {4, nil, 5, 15},  --Armor +++
    [220369] = {4, nil, 5, 05},  --Armor +++, STAM
    --[211109] = {4, nil, 6, 36},  --Regen +
    --[216642] = {4, nil, 6, 26},  --Regen ++
    --[211125] = {4, nil, 6, 16},  --Regen +++
    --[211105] = {4, nil, 6, 06},  --Regen +++, STAM
    [210717] = {4, nil, 6, 37},  --Leech +
    [216641] = {4, nil, 6, 27},  --Leech ++
    [210718] = {4, nil, 6, 17},  --Leech +++
    [211103] = {4, nil, 6, 07},  --Leech +++, STAM
    [210716] = {4, nil, 7, 38},  --Speed +
    [216639] = {4, nil, 7, 28},  --Speed ++
    [211124] = {4, nil, 7, 18},  --Speed +++
    [211101] = {4, nil, 7, 08},  --Speed +++, STAM
};

local CUSTOM_SORT_ORDER = {
    [219801] = 3,   --Ankh of Reincarnation
    [212366] = 26,  --Arcanist's Edge
    [219944] = 10,  --Bloodthirsty Coral
    [219818] = 13,  --Brilliance
    [216649] = 11,  --Brittle
    [216648] = 16,  --Cold Front
    [217957] = 36,  --Deliverance
    [212694] = 8,   --Enkindle
    [212749] = 30,  --Explosive Barrage
    [212365] = 32,  --Fervor
    [219817] = 2,   --Freedom
    [212916] = 4,   --Frost Armor
    [219777] = 1,   --Grounding
    [216647] = 22,  --Hailstorm
    [217964] = 5,   --Holy Martyr
    [212758] = 31,  --Incendiary Terror
    [219389] = 33,  --Lightning Rod
    [216624] = 7,   --Mark of Arrogance
    [216650] = 9,   --Memory of Vegeance
    [212759] = 23,  --Metero Storm
    [212361] = 12,  --Opportunist
    [216625] = 29,  --Quick Strike
    [217961] = 27,  --Righteous Frenzy
    [217927] = 28,  --Savior
    [216651] = 35,  --Searing Light
    [216626] = 19,  --Slay
    [219452] = 14,  --Static Charge
    [219523] = 18,  --Storm Overload
    [212362] = 24,  --Sunstrider's Flourish
    [216627] = 21,  --Tinkmaster's Shield
    [219527] = 20,  --Vampiric Aura
    [216628] = 34,  --Victory Fire
    [217903] = 25,  --Vindication
    [217907] = 6,   --Warmth
    [212760] = 17,  --Wildfire
    [219516] = 15,  --Windweaver
};

local FALLBACK_TINKERS = {
    --When player doesn't own all saved tinkers and have empty sockets. Prioritize DPS gem
    216626, 219389, 212694, 219516, 212749, 212365,
    212758, 216647, 212362, 212760, 216649, 212759,
    219818, 216648, 219523, 219452, 216651, 216628,
    212916, 212366, 219801, 219944, 217957, 219817,
    219777, 217964, 216624, 216650, 212361, 216625,
    217961, 217927, 216627, 219527, 217903, 217907,
};

local STAT_GEMS = {
    [1] = {210714, 216644, 211123, 211102},     --Crit
    [2] = {210681, 216643, 211107, 211110},     --Haste
    [3] = {210715, 216640, 211106, 211108},     --Mastery
    [4] = {220371, 220372, 220374, 220373},     --Vers
    [5] = {220367, 220368, 220370, 220369},     --Armor
    --[6] = {211109, 216642, 211125, 211105},     --Regen
    [6] = {210717, 216641, 210718, 211103},     --Leech
    [7] = {210716, 216639, 211124, 211101},     --Speed
};

local GEM_REMOVAL_TOOL = {"spell", 433397};


local STATS_DATA = {
    --Data change dynamically
    --{name, }
    {STAT_CRITICAL_STRIKE, },
    {STAT_HASTE, },
    {STAT_MASTERY, },
    {STAT_VERSATILITY, },
    {STAT_ARMOR, },
    --{L["Stat Health Regen"], },
    {STAT_LIFESTEAL, },
    {STAT_SPEED},
};

local function SortFunc_UIOrder(a, b)
    if CUSTOM_SORT_ORDER[a] and CUSTOM_SORT_ORDER[b] then
        return CUSTOM_SORT_ORDER[a] < CUSTOM_SORT_ORDER[b]
    end

    if GEM_DATA[a][4] and GEM_DATA[b][4] then
        return GEM_DATA[a][4] < GEM_DATA[b][4]
    end

    return a < b
end

function DataProvider:GetSortedItemList()
    --Loaded once when used

    if self.gemList then return self.gemList end;

    local tinsert = table.insert;
    local tsort = tsort;

    local tbl = {};
    local numTypes = #GEM_TYPES;

    for i = 1, numTypes do
        tbl[i] = {};
    end

    for itemID, data in pairs(GEM_DATA) do
        if DoesItemExistByID(itemID) then
            local gemType = data[1];
            tinsert(tbl[gemType], itemID);
            Gemma:SetGemRemovalTool(itemID, GEM_REMOVAL_TOOL);
        else

        end
    end

    for gemType, gems in pairs(tbl) do
        tsort(gems, SortFunc_UIOrder);
    end

    self.gemList = tbl;

    return tbl
end

function DataProvider:GetGemTypeName(gemType)
    local lookup = "EMPTY_SOCKET_"..GEM_TYPES[gemType];
    if _G[lookup] then
        return _G[lookup]
    else
        return GEM_TYPES[gemType]
    end
end

function DataProvider:GetItemListByType(gemType)
    local itemList = self:GetSortedItemList();
    return itemList[gemType];
end

local function GetNumSocketsForSlot(slotID)
    local itemLink = GetInventoryItemLink("player", slotID);
    if itemLink then
        return GetItemNumSockets(itemLink) or 0
    else
        return 0
    end
end

local function GetItemGemFromSlot(slotID, index)
    local itemLink = GetInventoryItemLink("player", slotID);
    if itemLink then
        return GetItemGemID(itemLink, index);
    end
end

function DataProvider:DoesPlayerHaveHead()
    return GetNumSocketsForSlot(SLOT_ID.HEAD) > 0
end

function DataProvider:DoesPlayerHaveFeet()
    return GetNumSocketsForSlot(SLOT_ID.FEET) > 0
end

function DataProvider:GetHeadGem()
    return GetItemGemFromSlot(SLOT_ID.HEAD, 1)
end

function DataProvider:GetFeetGem()
    return GetItemGemFromSlot(SLOT_ID.FEET, 1)
end

function DataProvider:GetGemType(itemID)
    if GEM_DATA[itemID] then
        return GEM_DATA[itemID][1]
    end
end

function DataProvider:GetGemSpell(itemID)
    if GEM_DATA[itemID] then
        return GEM_DATA[itemID][2]
    end
end

function DataProvider:GetStatType(itemID)
    if self:GetGemType(itemID) == 4 then
        return GEM_DATA[itemID][3]
    end
end

function DataProvider:GetConflictGemItemID(itemID)
    local gemType = self:GetGemType(itemID);
    if gemType == 1 then
        local existingGem = self:GetHeadGem();
        if itemID ~= existingGem then
            return existingGem
        end
    elseif gemType == 2 then
        local existingGem = self:GetFeetGem();
        if itemID ~= existingGem then
            return existingGem
        end
    end
end

function DataProvider:GetActiveGems()
    return BagUtil.activeGemList
end




do  --Scan bag and equipment slot
    function BagUtil:ResetBagInfo()
        self.gemCount = {};             --Including all locations
        self.bagGemCount = {};          --Including those in equipment
        self.bagUsedGemCount = {};      --Gems in your bag equipment
        self.slotData = {};
        self.gemTypeAvailable = {};
        self.activeGemCount = {};
        self.activeGemList = nil;
    end

    local function SortFunc_ActiveTraits(itemID1, itemID2)
        local type1 = GEM_DATA[itemID1][1];
        local type2 = GEM_DATA[itemID2][1];

        if type1 ~= type2 then
            return type1 < type2
        end

        return itemID1 < itemID2
    end

    function BagUtil:OnScanComplete()
        if not self.activeGemCount then return end;

        local gemType, statType;
        local traitList = {};
        local statList = {};
        local n = 0;
        for itemID, count in pairs(self.activeGemCount) do
            gemType = DataProvider:GetGemType(itemID);
            if gemType then
                if gemType == 4 then
                    statType = GEM_DATA[itemID][3];
                    if not statList[statType] then
                        local name = STATS_DATA[statType][1];
                        statList[statType] = {0, name};
                    end

                    statList[statType][1] = statList[statType][1] + count;
                else
                    n = n + 1;
                    traitList[n] = itemID;
                end
            end
        end

        tsort(traitList, SortFunc_ActiveTraits);

        self.activeGemList = {
            traits = traitList,
            stats = statList,
        };

        if (not Gemma.MainFrame:IsShown()) and Gemma.PaperdollWidget:IsShown() and DataProvider:ShouldShowGreenDot() then
            Gemma.MainFrame:Show();
        end
    end

    local function BagUtil_ProcessItem(itemLink, id1, id2)
        if IsEquippableItem(itemLink) then
            local numSockets = GetItemNumSockets(itemLink);
            local slotData;

            if (not id2) and (numSockets > 0) then
                --id1: slotID
                slotData = {
                    gems = {},
                    numSockets = numSockets,
                    numMissing = 0,
                };
                BagUtil.slotData[id1] = slotData;
            end

            for index = 1, numSockets do
                local itemID = GetItemGemID(itemLink, index);
                if itemID and GEM_DATA[itemID] then
                    if not BagUtil.gemCount[itemID] then
                        BagUtil.gemCount[itemID] = 1;
                    else
                        BagUtil.gemCount[itemID] = BagUtil.gemCount[itemID] + 1;
                    end

                    if id2 then
                        --when there are supported gems in you bag
                        local gemType = DataProvider:GetGemType(itemID);
                        if not BagUtil.gemTypeAvailable[gemType] then
                            BagUtil.gemTypeAvailable[gemType] = true;
                        end

                        if not BagUtil.bagGemCount[itemID] then
                            BagUtil.bagGemCount[itemID] = 0;
                        end
                        BagUtil.bagGemCount[itemID] = BagUtil.bagGemCount[itemID] + 1;

                        if not BagUtil.bagUsedGemCount[itemID] then
                            BagUtil.bagUsedGemCount[itemID] = 0;
                        end
                        BagUtil.bagUsedGemCount[itemID] = BagUtil.bagUsedGemCount[itemID] + 1;
                    else
                        if not BagUtil.activeGemCount[itemID] then
                            BagUtil.activeGemCount[itemID] = 0;
                        end
                        BagUtil.activeGemCount[itemID] = BagUtil.activeGemCount[itemID] + 1;
                    end
                end

                if not id2 then
                    slotData.gems[index] = itemID;
                    if not itemID then
                        slotData.numMissing = slotData.numMissing + 1;
                    end
                end
            end
        else
            local itemID = GetItemInfoInstant(itemLink);
            if GEM_DATA[itemID] then
                if not BagUtil.gemCount[itemID] then
                    BagUtil.gemCount[itemID] = 1;
                else
                    BagUtil.gemCount[itemID] = BagUtil.gemCount[itemID] + 1;    --stackCount of stat gem isn't relevant here
                end

                if id2 then
                    local gemType = DataProvider:GetGemType(itemID);
                    if not BagUtil.gemTypeAvailable[gemType] then
                        BagUtil.gemTypeAvailable[gemType] = true;
                    end

                    if not BagUtil.bagGemCount[itemID] then
                        BagUtil.bagGemCount[itemID] = 0;
                    end
                    BagUtil.bagGemCount[itemID] = BagUtil.bagGemCount[itemID] + 1;
                end
            end
        end
    end

    local AdditionalScanSlots;

    function BagUtil:InitiateScan()
        if not AdditionalScanSlots then
            AdditionalScanSlots = {};
            local n = 0;
            for slotName, slotID in pairs(SLOT_ID) do
                n = n + 1;
                AdditionalScanSlots[n] = slotID;
            end
        end

        BagScan:SetProcessor(BagUtil_ProcessItem, AdditionalScanSlots);
        BagScan:FullUpdate();
    end

    function BagUtil:StopScan()
        BagScan:StopIfProcessorSame(BagUtil_ProcessItem);
    end


    local CallbackRegistry = addon.CallbackRegistry;

    local function PaperdollWidget_OnShow()
        BagUtil:InitiateScan();
    end

    local function PaperdollWidget_OnHide()
        BagUtil:StopScan();
    end

    CallbackRegistry:Register("PaperdollWidget.Gem.OnShow", PaperdollWidget_OnShow);
    CallbackRegistry:Register("PaperdollWidget.Gem.OnHide", PaperdollWidget_OnHide);

    CallbackRegistry:Register("GemManager.BagScan.OnStart", BagUtil.ResetBagInfo, BagUtil);
    CallbackRegistry:Register("GemManager.BagScan.OnStop", BagUtil.OnScanComplete, BagUtil);
end


local TINKER_SLOT = {
    SLOT_ID.SHOULDER, SLOT_ID.WRIST, SLOT_ID.HANDS, SLOT_ID.WAIST,
};

do  --Use the result from bag scan
    local META_SLOT = { SLOT_ID.HEAD };

    local COGWHEEL_SLOT = { SLOT_ID.FEET };

    local PRISMATIC_SLOT = {
        SLOT_ID.CHEST, SLOT_ID.LEGS, SLOT_ID.TRINKET1, SLOT_ID.TRINKET2, SLOT_ID.NECK, SLOT_ID.RING1, SLOT_ID.RING2
    };


    function DataProvider:GetSlotSocketsCount(slots, gemType)
        local slotData = BagUtil.slotData;
        local totalSockets = 0;
        local totalMissing = 0;

        for _, slotID in ipairs(slots) do
            if slotData[slotID] then
                totalSockets = totalSockets + slotData[slotID].numSockets;
                totalMissing = totalMissing + slotData[slotID].numMissing;
            end
        end

        local anySpareGemInBags = BagUtil.gemTypeAvailable and BagUtil.gemTypeAvailable[gemType]

        return totalSockets, totalMissing, anySpareGemInBags
    end

    function DataProvider:GetMetaSocketCount()
        return self:GetSlotSocketsCount(META_SLOT, 1)
    end

    function DataProvider:GetCogwheelSocketCount()
        return self:GetSlotSocketsCount(COGWHEEL_SLOT, 2)
    end

    function DataProvider:GetTinkerSocketCount()
        return self:GetSlotSocketsCount(TINKER_SLOT, 3)
    end
    
    function DataProvider:GetPrismaticSocketCount()
        return self:GetSlotSocketsCount(PRISMATIC_SLOT, 4)
    end

    function DataProvider:GetGemInventorySlotAndIndex(itemID)
        if not self:IsGemCollected(itemID) then return end;

        for slotID, slotData in pairs(BagUtil.slotData) do
            for socketIndex = 1, slotData.numSockets do
                if slotData.gems[socketIndex] == itemID then
                    return slotID, socketIndex
                end
            end
        end
    end

    function DataProvider:CanSwapGemInOneStep(itemID)
        --You will have to perform two actions if the target socket is occupied and
        --the gem you want to insert is in another equipment
        local existingGem = self:GetConflictGemItemID(itemID);
        local inbagCount = self:GetInBagGemCount(itemID);
        local spareCount = GetItemCount(itemID);

        if existingGem and (inbagCount > 0 and spareCount == 0) then
            return false
        end

        return true
    end

    function DataProvider:GetBestSlotToPlaceGem(itemID)
        if not self:IsGemCollected(itemID) then return end;

        local gemType = self:GetGemType(itemID);
        if not gemType then return end;

        local slots;

        if gemType == 1 then
            slots = META_SLOT;
        elseif gemType == 2 then
            slots = COGWHEEL_SLOT;
        elseif gemType == 3 then
            slots = TINKER_SLOT;
        elseif gemType == 4 then
            slots = PRISMATIC_SLOT;
        end

        local slotData;

        for _, slotID in ipairs(slots) do
            slotData = BagUtil.slotData[slotID];
            if slotData then
                for socketIndex = 1, slotData.numSockets do
                    if slotData.gems[socketIndex] == nil then
                        return slotID, socketIndex
                    end
                end
            end
        end
    end

    function DataProvider:GetInBagGemCount(itemID)
        return BagUtil.bagGemCount and BagUtil.bagGemCount[itemID] or 0;
    end

    function DataProvider:GetInBagUsedGemCount(itemID)
        return BagUtil.bagUsedGemCount and BagUtil.bagUsedGemCount[itemID] or 0;
    end

    function DataProvider:GetOnPlayerGemCount(itemID)
        return BagUtil.activeGemCount[itemID] and BagUtil.activeGemCount[itemID] or 0;
    end

    function DataProvider:IsGemActive(itemID)
        return self:GetOnPlayerGemCount(itemID) > 0
    end

    function DataProvider:IsGemCollected(itemID)
        return BagUtil.gemCount[itemID] and BagUtil.gemCount[itemID] > 0
    end

    function DataProvider:ShouldShowGreenDot()
        local totalSockets, totalMissing, anySpareGemInBags;
        local showGreenDot;

        if not showGreenDot then
            totalSockets, totalMissing, anySpareGemInBags = self:GetMetaSocketCount();
            showGreenDot = totalMissing > 0 and anySpareGemInBags;
        end

        if not showGreenDot then
            totalSockets, totalMissing, anySpareGemInBags = self:GetCogwheelSocketCount();
            showGreenDot = totalMissing > 0 and anySpareGemInBags;
        end

        if not showGreenDot then
            totalSockets, totalMissing, anySpareGemInBags = self:GetTinkerSocketCount();
            showGreenDot = totalMissing > 0 and anySpareGemInBags;
        end

        if not showGreenDot then
            totalSockets, totalMissing, anySpareGemInBags = self:GetPrismaticSocketCount();
            showGreenDot = totalMissing > 0 and anySpareGemInBags;
        end

        return showGreenDot
    end
end

do
    local ACTION_BUTTON_METHODS = {
        [1] = "SetAction_RemovePandariaMetaGem",
        [2] = "SetAction_RemovePandariaGem",
    };

    function DataProvider:GetActionButtonMethod(itemID)
        local gemType = self:GetGemType(itemID);
        if gemType == 1 or gemType == 2 then
            return ACTION_BUTTON_METHODS[1]
        elseif gemType == 3 then
            return ACTION_BUTTON_METHODS[2]
        end
    end
end


DataProvider.schematic = {
    background = "remix-ui-bg",
    topDivider = "remix-ui-divider",

    tabData = {
        {
            name = L["Pandamonium Gem Category 1"],  --Major
            method = "ShowMajors",
            background = nil,
            useCustomTooltip = false,
        },

        {
            name = L["Pandamonium Gem Category 2"],  --Tinker
            method = "ShowTraits",
            background = "remix-ui-tinker-bg",
            useCustomTooltip = true,
        },

        {
            name = L["Pandamonium Gem Category 3"],  --Prismatic
            method = "ShowStats",
            background = nil,
            useCustomTooltip = false,
        },
    },
};

local BorderTextures_Hexagon = {
    active = "remix-hexagon-yellow",
    inactive = "remix-hexagon-grey",
    available = "remix-hexagon-green",
    dimmed = "remix-hexagon-darkyellow",
};

local BorderTextures_BigSquare = {
    active = "remix-bigsquare-yellow",
    inactive = "remix-bigsquare-grey",
    available = "remix-bigsquare-green",
};


local function sin(deg)
    return math.sin(math.rad(deg));
end

local function cos(deg)
    return math.cos(math.rad(deg));
end

local function CreateSlotShadow(self, slotButton)
    local shadow = self:AcquireTexture("Back", "BACKGROUND");
    AtlasUtil:SetAtlas(shadow, "remix-bigsquare-shadow");
    shadow:SetPoint("CENTER", slotButton, "CENTER", 0, 0);
end


function GemManagerMixin:SetupMajorSlotButton(itemID)
    local button = self:AcquireSlotButton("BigSquare");
    button.borderTextures = BorderTextures_BigSquare;
    button:SetButtonSize(64, 53);   --buttonSize, iconSize

    if itemID then
        button:SetItem(itemID);
        button:SetActive();
    else
        button:ClearItem();
    end

    local shadow = self:AcquireTexture("Back", "BACKGROUND");
    AtlasUtil:SetAtlas(shadow, "remix-bigsquare-shadow");
    shadow:SetPoint("CENTER", button, "CENTER", 0, 0);

    return button
end

local function TooltipFunc_EmptyMeta(tooltip)
    tooltip:SetText(EMPTY_SOCKET_META, 1, 1, 1);
    tooltip:AddLine(L["Click To Show Gem List"], 0.098, 1.000, 0.098, true);
    tooltip:Show();
    return true
end

local function TooltipFunc_EmptyCogwheel(tooltip)
    tooltip:SetText(EMPTY_SOCKET_COGWHEEL, 1, 1, 1);
    tooltip:AddLine(L["Click To Show Gem List"], 0.098, 1.000, 0.098, true);
    tooltip:Show();
    return true
end

local function OnClickFunc_ShowMetaGemList(self, mouseButton)
    if DataProvider:GetHeadGem() and mouseButton == "RightButton" then
        MainFrame.autoShowGemList = 1;
        return false
    end

    GemManagerMixin.ShowGemList(MainFrame, 1);

    return true
end

local function OnClickFunc_ShowCogwheelGemList(self, mouseButton)
    if DataProvider:GetFeetGem() and mouseButton == "RightButton" then
        MainFrame.autoShowGemList = 2;
        return false
    end

    GemManagerMixin.ShowGemList(MainFrame, 2);

    return true
end

local function OnClickFunc_CloseGemList(self, mouseButton)
    if mouseButton == "RightButton" then
        --if not DataProvider:IsGemActive(self.itemID) then
            MainFrame:CloseGemList();
            return true
        --end
    elseif mouseButton == "LeftButton" then
        if DataProvider:IsGemActive(self.itemID) then
            MainFrame:CloseGemList();
            return true
        end
    end
end

function GemManagerMixin:ShowMajors()
    MainFrame = self;
    self.useSlotFrame = true;

    local shape = "BigSquare";
    local container = self.SlotFrame;

    local numButtons = 0;
    local button1, button2;

    local newlyEquippedSocket;

    if DataProvider:DoesPlayerHaveHead() then
        local itemID = DataProvider:GetHeadGem();
        local button = self:SetupMajorSlotButton(itemID);

        numButtons = numButtons + 1;
        button1 = button;

        local typeIcon = self:AcquireTexture("Front");
        AtlasUtil:SetAtlas(typeIcon, "gemtypeicon-offensive");
        typeIcon:SetPoint("CENTER", button, "BOTTOM", 0, 0);

        button.onClickFunc = OnClickFunc_ShowMetaGemList;

        if itemID then
            typeIcon:SetDesaturation(0);
        else
            typeIcon:SetDesaturation(1);
            button.tooltipFunc = TooltipFunc_EmptyMeta;
            local _, _, anySpareGemInBags = DataProvider:GetPrismaticSocketCount();  --GetMetaSocketCount
            if anySpareGemInBags then
                button:SetSelectable();
            else
                button:SetUncollected();
            end
        end

        if (not self.headGemID) and itemID then
            newlyEquippedSocket = button;
        end
        self.headGemID = itemID;
    else
        self.headGemID = nil;
    end

    if DataProvider:DoesPlayerHaveFeet() then
        local itemID = DataProvider:GetFeetGem();
        local button = self:SetupMajorSlotButton(itemID);

        numButtons = numButtons + 1;
        if not button1 then
            button1 = button;
        else
            button2 = button;
        end

        local typeIcon = self:AcquireTexture("Front");
        AtlasUtil:SetAtlas(typeIcon, "gemtypeicon-movement");
        typeIcon:SetPoint("CENTER", button, "BOTTOM", 0, 0);

        button.onClickFunc = OnClickFunc_ShowCogwheelGemList;

        if itemID then
            typeIcon:SetDesaturation(0);
        else
            typeIcon:SetDesaturation(1);
            button.tooltipFunc = TooltipFunc_EmptyCogwheel;
            local _, _, anySpareGemInBags = DataProvider:GetCogwheelSocketCount();
            if anySpareGemInBags then
                button:SetSelectable();
            else
                button:SetUncollected();
            end
        end

        if (not self.feetGemID) and itemID then
            if newlyEquippedSocket then
                newlyEquippedSocket = nil;
            else
                newlyEquippedSocket = button;
            end
        end
        self.feetGemID = itemID;
    else
        self.feetGemID = nil;
    end

    if numButtons == 0 then
        self:ShowNoSocketAlert(true);
    else
        if numButtons == 2 then
            local offsetX = 8;
            local offsetY = 16;
            button1:SetPoint("BOTTOMRIGHT", container, "CENTER", -offsetX, offsetY);
            button2:SetPoint("TOPLEFT", container, "CENTER", offsetX, -offsetY);
        elseif numButtons == 1 then
            button1:SetPoint("CENTER", container, "CENTER", 0, 0);
        end
        self:ShowNoSocketAlert(false);
    end

    self.SlotFrame.ButtonHighlight:SetShape(shape);
    self.UpdateCurrentTab = self.UpdateMajors;

    if self.autoShowGemList then
        self:ShowGemList(self.autoShowGemList);
        self.autoShowGemList = nil;
    else
        if newlyEquippedSocket then
            self:CloseGemList();

            local shine = self.SlotFrame.ButtonShine;
            shine.Mask:SetTexture(PATH.."IconMask-"..shape, "CLAMPTOBLACKADDITIVE", "CLAMPTOBLACKADDITIVE");
            shine.Mask:SetSize(53, 53);
            shine.Texture:SetTexture(PATH.."SlotShine");
            shine.Texture:SetSize(80, 80);
            shine.Texture:SetBlendMode("ADD");

            self:ShineSlot(newlyEquippedSocket);
        else
            self:UpdateGemList();
        end
    end
end

function GemManagerMixin:UpdateMajors()
    self:ReleaseContent();
    self:ShowMajors();
end

function GemManagerMixin:UpdateGemList()
    if self.GemList:IsShown() and self.gemType then
        local activeGemID;
        if self.gemType == 1 then
            activeGemID = DataProvider:GetHeadGem();
        elseif self.gemType == 2 then
            activeGemID = DataProvider:GetFeetGem();
        end
        self.GemList.activeGemID = activeGemID;
        self.GemList:UpdatePage();
    end
end

function GemManagerMixin:ShowGemList(gemType)
    self.gemType = gemType;

    if not gemType then return end;

    local activeGemID;
    if gemType == 1 then
        activeGemID = DataProvider:GetHeadGem();
    elseif gemType == 2 then
        activeGemID = DataProvider:GetFeetGem();
    end
    self.GemList.activeGemID = activeGemID;

    local gems = DataProvider:GetItemListByType(gemType);
    self.GemList.onClickFunc = OnClickFunc_CloseGemList;
    self.GemList:SetItemList(gems, DataProvider:GetGemTypeName(gemType), DataProvider);

    self:OpenGemList();
end

function GemManagerMixin:ShowTraits()
    self.useSlotFrame = true;

    local shape = "Hexagon";
    local diagonal = 40;    --46
    local gap = 1;
    local container = self.SlotFrame;

    local deltaXPerRow = ((diagonal * cos(30)) + gap) * 0.5;
    local deltaYPerRow = (diagonal * cos(30) + gap) * cos(30);
    local offsetX = diagonal * cos(30) + gap;

    local contentHeight = 5 * deltaYPerRow + diagonal;

    local frameWidth = container:GetWidth();
    local frameHeight = container:GetHeight();

    local paddingY = (frameHeight - contentHeight) * 0.5;

    local refX = frameWidth * 0.5;
    refX = 0;
    local refY = (deltaYPerRow - diagonal * 0.5) - paddingY;

    local gemType = 3;  --Tinker
    local gems = DataProvider:GetItemListByType(gemType);

    local button;
    local row = 1;
    local col = 1;
    local maxCol = 1;
    local fromX, fromY = refX, refY;
    local x, y;

    for index, itemID in ipairs(gems) do
        button = self:AcquireSlotButton(shape);
        button:ResetButtonSize();
        button.borderTextures = BorderTextures_Hexagon;

        if col > maxCol then
            maxCol = maxCol + 1;
            col = 1;
            row = row + 1;

            fromX = refX - (row - 1) * deltaXPerRow;
            fromY = refY - (row - 1) * deltaYPerRow;
        end

        x = fromX + (col - 1) * offsetX;
        y = fromY;
        button:SetPoint("CENTER", self.SlotFrame, "TOP", x, y);
        button:SetItem(itemID);

        col = col + 1;
    end

    if self.SlotFrame.ButtonHighlight then
        self.SlotFrame.ButtonHighlight:SetShape(shape);
    end

    local shine = self.SlotFrame.ButtonShine;
    if shine then
        shine.Mask:SetTexture(PATH.."IconMask-"..shape, "CLAMPTOBLACKADDITIVE", "CLAMPTOBLACKADDITIVE");
        shine.Mask:SetSize(32, 32);
        shine.Texture:SetTexture(PATH.."SlotShine");
        shine.Texture:SetSize(48, 48);
        shine.Texture:SetBlendMode("ADD");
    end

    self.UpdateCurrentTab = self.UpdateSlots;

    if not self.isLoadoutPlanner then
        if self.TooltipFrame then
            self.TooltipFrame:ClearAllPoints();
            self.TooltipFrame:SetPoint("TOP", container, "CENTER", 0, -contentHeight * 0.5 - 44);
    
            if GetCVarBool("colorblindMode") then
                self.TooltipFrame:SetDescriptionLine(7);
            else
                self.TooltipFrame:SetDescriptionLine(6);
            end
        end

        self:UpdateSlots();
    end
end

function GemManagerMixin:UpdateSlots()
    if not self.slotButtons then return end;

    local totalSockets, totalMissing, anySpareGemInBags = DataProvider:GetTinkerSocketCount();
    self:SetPointDisplayAmount(totalMissing);

    local chooseItem = totalMissing > 0;

    for index, button in ipairs(self.slotButtons) do
        if button:IsShown() then
            if DataProvider:IsGemActive(button.itemID) then
                if chooseItem then
                    button:SetDimmed();
                else
                    button:SetActive();
                end
            elseif DataProvider:IsGemCollected(button.itemID) then
                if chooseItem then
                    button:SetAvailable();
                else
                    button:SetInactive();
                end
                
            else
                button:SetUncollected();
            end
        else
            break
        end
    end
end


do  --Stats Assignment Tab
    function DataProvider:GetNumAvailableGemForStat(statType)
        local gems = STAT_GEMS[statType];
        local total = 0;

        if gems then
            for _, itemID in ipairs(gems) do
                total = total + self:GetInBagGemCount(itemID);
            end
        end

        return total
    end

    function DataProvider:GetBestStatGemToRemove(statType)
        --From lower tier to higher
        local gems = STAT_GEMS[statType];
        if gems then
            for _, itemID in ipairs(gems) do
                if self:IsGemActive(itemID) then
                    return itemID
                end
            end
        end
    end

    function DataProvider:GetBestStatGemToInsert(statType)
        --From higher tier to lower
        local gems = STAT_GEMS[statType];
        if gems then
            local itemID;
            for i = #gems, 1, -1 do
                itemID = gems[i];
                if self:GetInBagGemCount(itemID) > 0 then
                    return itemID
                end
            end
        end
    end

    function DataProvider:GetBestStatGemForAction(statType, direction)
        if direction > 0 then   --Insert gem
            return self:GetBestStatGemToInsert(statType)
        else    --remove gem
            return self:GetBestStatGemToRemove(statType)
        end
    end

    function DataProvider:GetAvailableGemListForStat(statType)
        --Used in Loadout Action
        local gems = STAT_GEMS[statType];
        local statGemCount = {};

        if gems then
            local itemID, inBagCount, spareCount;
            local n = 0;

            for i = #gems, 1, -1 do
                itemID = gems[i];

                inBagCount = self:GetInBagUsedGemCount(itemID);
                spareCount = GetItemCount(itemID);

                --if inBagCount > 0 or spareCount > 0 then
                    local tbl = {
                        itemID = itemID,
                        inBagCount = inBagCount,
                        spareCount = spareCount,
                    };

                    n = n + 1;
                    statGemCount[n] = tbl;
                --end
            end
        end

        return statGemCount
    end

    function GemManagerMixin:ShowStats()
        self.useSlotFrame = true;

        local buttonHeight = 24;
        local gap = 0.2;    --Compensation for IsMouseOver()
        local numButtons = #STATS_DATA;
        local container = self.SlotFrame;
        local fromY = -0.5 * (container:GetHeight() - (numButtons * (buttonHeight + gap) -gap));

        local button;

        for i, statData in ipairs(STATS_DATA) do
            button = self:AcquireStatButton();
            button:SetPoint("TOP", container, "TOP", 0, fromY + (1 - i) * (buttonHeight + gap));
            button:SetName(statData[1]);
        end

        self.UpdateCurrentTab = self.UpdateStats;

        if not self.isLoadoutPlanner then
            self:UpdateStats();
        end
    end

    function GemManagerMixin:UpdateStats()
        local totalSockets, totalMissing, anySpareGemInBags = DataProvider:GetPrismaticSocketCount();
        local isEditMode;

        if totalMissing > 0 then
            isEditMode = true;
            self.PointsDisplay:SetAmount(totalMissing);
            self.PointsDisplay:Show();
        else
            isEditMode = false;
            self.PointsDisplay:Hide();
        end
        
        local activeGems = DataProvider:GetActiveGems();
        local activeStatsGems = activeGems and activeGems.stats;

        local button, count;

        for i, statData in ipairs(STATS_DATA) do
            button = self.statButtons[i];
            count = activeStatsGems[i] and activeStatsGems[i][1] or 0;
            button:SetCount(count);
            button:SetPlusButtonVisibility( isEditMode and DataProvider:GetNumAvailableGemForStat(i) > 0 );
        end
    end
end


function GemManagerMixin:UpdateTabGreenDot()
    BagScan:FullUpdate();

    local totalSockets, totalMissing, anySpareGemInBags;
    local showGreenDot;

    --Meta and cogwheel are on the same tab
    totalSockets, totalMissing, anySpareGemInBags = DataProvider:GetMetaSocketCount();
    showGreenDot = totalMissing > 0 and anySpareGemInBags;

    if not showGreenDot then
        totalSockets, totalMissing, anySpareGemInBags = DataProvider:GetCogwheelSocketCount();
        showGreenDot = totalMissing > 0 and anySpareGemInBags;
    end

    self.tabButtons[1].GreenDot:SetShown(showGreenDot);

    totalSockets, totalMissing, anySpareGemInBags = DataProvider:GetTinkerSocketCount();
    showGreenDot = totalMissing > 0 and anySpareGemInBags;
    self.tabButtons[2].GreenDot:SetShown(showGreenDot);

    totalSockets, totalMissing, anySpareGemInBags = DataProvider:GetPrismaticSocketCount();
    showGreenDot = totalMissing > 0 and anySpareGemInBags;
    self.tabButtons[3].GreenDot:SetShown(showGreenDot);
end




do --Loadout
    local function AddSlotGemCountToTable(tbl, slotID, gemType)
        local itemLink = GetInventoryItemLink("player", slotID);
        if itemLink then
            local gemItemID, statType;
            for socketIndex = 1, GetItemNumSockets(itemLink) do
                gemItemID = GetItemGemID(itemLink, socketIndex);
                if gemItemID then
                    --local gemType = DataProvider:GetGemType(gemItemID);
                    if gemType == 3 then
                        table.insert(tbl, gemItemID);
                    elseif gemType == 4 then
                        statType = DataProvider:GetStatType(gemItemID);
                        if not tbl[statType] then
                            tbl[statType] = 0;
                        end
                        tbl[statType] = tbl[statType] + 1;
                    end
                end
            end
        end
    end

    function DataProvider:GetEquippedLoadoutGemInfo()
        --[[--Structure:
            {
                head = 221977,
                feet = 218110,
                tinker = {219801, 212366, 219944, 219818},
                stats1 = {  --Chest Legs
                    crit = 1,
                    haste = 2,
                },

                stats2 = {}, --Trinkets
                stats3 = {}, --Neck/Rings
            }
        --]]

        local gemInfo = {};

        gemInfo.head = self:GetHeadGem();
        gemInfo.feet = self:GetFeetGem();


        gemInfo.tinker = {};
        for _, slotID in ipairs(TINKER_SLOT) do
            AddSlotGemCountToTable(gemInfo.tinker, slotID, 3);
        end

        tsort(gemInfo.tinker);

        gemInfo.stats1 = {};
        AddSlotGemCountToTable(gemInfo.stats1, SLOT_ID.CHEST, 4);
        AddSlotGemCountToTable(gemInfo.stats1, SLOT_ID.LEGS, 4);

        gemInfo.stats2 = {};
        AddSlotGemCountToTable(gemInfo.stats2, SLOT_ID.TRINKET1, 4);
        AddSlotGemCountToTable(gemInfo.stats2, SLOT_ID.TRINKET2, 4);

        gemInfo.stats3 = {};
        AddSlotGemCountToTable(gemInfo.stats3, SLOT_ID.NECK, 4);
        AddSlotGemCountToTable(gemInfo.stats3, SLOT_ID.RING1, 4);
        AddSlotGemCountToTable(gemInfo.stats3, SLOT_ID.RING2, 4);

        return gemInfo
    end

    function DataProvider:IsLeftStatGemBetter(gem1, gem2)
        if GEM_DATA[gem1][3] == GEM_DATA[gem2][3] then
            return GEM_DATA[gem1][4] < GEM_DATA[gem2][4]
        end
    end

    function DataProvider:GetFallbackTinkers(numEmptySockets, requiredTinker)
        local tbl = {};
        local n = 0;

        requiredTinker = requiredTinker or {};

        for _, itemID in ipairs(FALLBACK_TINKERS) do
            if (not requiredTinker[itemID]) and (self:IsGemCollected(itemID) and not self:IsGemActive(itemID)) then
                n = n + 1;
                if n <= numEmptySockets then
                    tbl[n] = itemID;
                else
                    break
                end
            end
        end

        return tbl
    end

    function DataProvider:GetFallbackMajorGem(gemType, requiredGem)
        local gems = self:GetItemListByType(gemType);
        for _, itemID in ipairs(gems) do
            if (itemID ~= requiredGem) and (self:IsGemCollected(itemID) and not self:IsGemActive(itemID)) then
                return itemID;
            end
        end
    end

    function DataProvider:GetFallbackMeta(requiredGem)
        return self:GetFallbackMajorGem(1, requiredGem)
    end

    function DataProvider:GetFallbackCogwheel(requiredGem)
        return self:GetFallbackMajorGem(2, requiredGem)
    end
end