WoWInterface/Interface/AddOns/Narcissus/Modules/CharacterFrame/ItemSocketing.lua

local L = Narci.L;
local GetItemQualityColor = NarciAPI.GetItemQualityColor;
local GetGemBonus = NarciAPI.GetGemBonus; --(Gem's itemID or hyperlink)
local GetShardBonus = NarciAPI.GetDominationShardBonus;
local max = math.max;
local min = math.min;
local floor = math.floor;
local format = string.format;
local After = C_Timer.After;

--[[ Blizzard APIs:
PickupItem(item)            --Place the item on the cursor
ClickSocketButton(id)       --Put the picked-up item into SocketButton
CursorHasItem()
GetCursorInfo()
GetItemCount(item)
GetContainerNumSlots(bagID) 0-4
GetContainerItemID(bag, slot)
SocketInventoryItem(slot)

Item:CreateFromItemID(itemID)
local itemLocation = ItemLocation:CreateFromEquipmentSlot(slotId)
ItemMixin:GetItemLocation()
ItemLocationMixin:GetBagAndSlot
PlaySound(SOUNDKIT.JEWEL_CRAFTING_FINALIZE);

Events:

ITEM_UNLOCKED
ITEM_LOCK_CHANGED

local function GetBagPosition(itemLink)
    for bag = 0, NUM_BAG_SLOTS do
        for slot = 1, GetContainerNumSlots(bag) do
            if(GetContainerItemLink(bag, slot) == itemLink) then
                return bag, slot
            end
        end
    end
end
--]]

local ItemBorderTexture = {
    [1] = {0,     0.125},        --Krazken
    [2] = {0.75,  0.875},        --Green
    [3] = {0.875,     1},        --White
    [4] = {0.875,     1},        --Meta
    [5] = {0.125,  0.25},        --Orange
    [6] = {0.5,   0.625},        --Purple
    [7] = {0.25,  0.375},        --Yellow
    [8] = {0.375,   0.5},        --Blue
    [9] = {0.875,     1},        --White
    [10] = {0.625,  0.75},       --Red
}

local GetGemBorderTexture = NarciAPI.GetGemBorderTexture;

local function GetBorderTexCoord(itemSubClassID, itemID)
    local _, index = GetGemBorderTexture(itemSubClassID, itemID);
    return ItemBorderTexture[index][1], ItemBorderTexture[index][2];
end

local function ShowFlyoutBlack(bool)
    --Narci_FlyoutBlack.Black:SetColorTexture(28/255, 32/255, 40/255, 0.5);
    if Narci_EquipmentFlyoutFrame:IsShown() then return; end
    local frame = Narci_FlyoutBlack;
	frame.AnimFrame:Hide();
	frame.AnimFrame.OppoDirection = not bool;
	frame.AnimFrame:Show();
end

local GemIDList, GemTypesByID = {}, {};
local DominationShardIDs = {};
local GET_BONUS_FUNC;
local GetContainerItemID = GetContainerItemID;
local GetItemCount = GetItemCount;
local numBags = NUM_BAG_SLOTS;
local FadeFrame = NarciAPI_FadeFrame;

local function StatAbbreviationToFullForm(abbr)
    if abbr == "crit" then
        abbr = NARCI_CRITICAL_STRIKE;
    elseif abbr == "haste" then
        abbr = STAT_HASTE;
    elseif abbr == "versatility" then
        abbr = STAT_VERSATILITY;
    elseif abbr == "mastery" then
        abbr = STAT_MASTERY;
    elseif abbr == "EXP" then
        abbr = POWER_TYPE_EXPERIENCE;
    elseif abbr == "MSPD" then
        abbr = STAT_MOVEMENT_SPEED;
    elseif abbr == "STR" then
        abbr = NARCI_STAT_STRENGTH;
    elseif abbr == "AGI" then
        abbr = NARCI_STAT_AGILITY;
    elseif abbr == "INT" then
        abbr = SPEC_FRAME_PRIMARY_STAT_INTELLECT;
    end
    return abbr;
end

for gemID, info in pairs(Narci.GemData) do
    tinsert(GemIDList, gemID);
    GemTypesByID[gemID] = StatAbbreviationToFullForm(info[1]);
end

if Narci.DominationShards then
    for gemID, info in pairs(Narci.DominationShards) do
        tinsert(DominationShardIDs, gemID);
        GemTypesByID[gemID] = "Domination";
    end
end

local function SortedByID(a, b) return a > b; end
table.sort(GemIDList, SortedByID);
table.sort(DominationShardIDs, SortedByID);

local GemCountList = {};
local BUTTON_HEIGHT = 48 --buttons[1] and buttons[1]:GetHeight()
local NUM_BUTTONS_PER_PAGE = 4;
local function GetTypeCount(referenceList)
    local count = 0;
    local i = 0;
    for k, itemID in pairs(referenceList) do
        count = GetItemCount(itemID);
        if count ~= 0 then
            i = i + 1;
        end
    end
    return i;
end

local function GetMatchCount(referenceList, outputList)
    wipe(outputList);
    local count = 0;
    local i = 1;
    local types = {};
    for k, itemID in pairs(referenceList) do
        count = GetItemCount(itemID);
        if count ~= 0 then
            outputList[i] = {itemID, count};
            if i < 3 then
                tinsert(types, GemTypesByID[itemID] );
            end
            i = i + 1;
        end
    end
    
    return (i - 1), types[1], types[2]
end

local function RefreshAvailableGems()
    local gemType = GetSocketTypes(1);
    if gemType == "Domination" then
        GET_BONUS_FUNC = GetShardBonus;
        GetMatchCount(DominationShardIDs, GemCountList);
    else
        GET_BONUS_FUNC = GetGemBonus;
        GetMatchCount(GemIDList, GemCountList);
    end
end

local function AjustShadowHeight(frame, numButtons)
    local tex = frame.Shadow; 
    if numButtons == 1 then
        tex:SetTexCoord(0, 0.37109375, 0.75, 1);
    elseif numButtons == 2 then
        tex:SetTexCoord(0.62890625, 1, 0.625, 1);
    elseif numButtons == 3 then
        tex:SetTexCoord(0.62890625, 1, 0, 0.5);
    elseif numButtons == 4 then
        tex:SetTexCoord(0, 0.37109375, 0, 0.671875);
    elseif numButtons == 0 then
        tex:SetTexCoord(0, 0, 0, 0);
    else
        tex:SetTexCoord(0, 0.37109375, 0, 0.671875);
    end
end

local SOCKETED_ITEM_LEVEL = 1208;    --the ilvl of the item that's currently being socketed
local function DisplayButtons(itemCountList, disabledID, rootFrame, buttonTemplate)
    local button, itemID, count;
    rootFrame = rootFrame or Narci_ItemSocketing;
    local scrollchild = rootFrame.ScrollFrame.scrollChild;
    buttonTemplate = buttonTemplate or "Narci_GearEnhancement_Template";
    if not rootFrame.buttons then
        rootFrame.buttons = {};
    end
    local buttons = rootFrame.buttons;
    local numGems = #itemCountList;
    local numMax = max(#buttons, numGems);
    for i = 1, numMax do
        button = buttons[i];
        if not button then
            button = CreateFrame("BUTTON", nil, scrollchild, buttonTemplate);
            button.index = i;
            if i == 1 then
                button:SetPoint("TOPLEFT", scrollchild, "TOPLEFT", 0 , 0);
                button.TopDivider:Hide();
            else
                button:SetPoint("TOPLEFT", buttons[i-1], "BOTTOMLEFT", 0 , 0);
            end
            tinsert(buttons, button);
        end

        if i == numMax then button.BottomDivider:Hide(); end
        
        if i > numGems then
            button:Hide();
        else
            itemID = itemCountList[i][1];
            count = itemCountList[i][2];
            button:SetItem(itemID, count, disabledID);
            button:Show();
        end
    end

    --Scroll Frame
    local ScrollFrame = rootFrame.ScrollFrame;
    local range = max( (numGems- NUM_BUTTONS_PER_PAGE - 0.35)*BUTTON_HEIGHT, 0);
    local scrollBar = ScrollFrame.scrollBar;
    scrollBar:SetMinMaxValues(0, range);
    ScrollFrame.range = range;
    scrollBar:SetShown(range ~= 0);
    local totalButtons = min(numGems, NUM_BUTTONS_PER_PAGE + 0.4);
    rootFrame:SetHeight(totalButtons*BUTTON_HEIGHT);
    AjustShadowHeight(rootFrame, totalButtons);
end

local function GetBagPosition(itemID)
    for bagID = 0, numBags do
        for slotID = 1, GetContainerNumSlots(bagID) do
            if(GetContainerItemID(bagID, slotID) == itemID) then
                return bagID, slotID
            end
        end
    end
end

local function AutoSocket(slotID, gemID)
    --local item = Item:CreateFromItemID(itemID)
    ClearCursor();
    if not slotID or not gemID then return; end
    local bagID, slotIndex = GetBagPosition(gemID);

    if not(bagID and slotIndex) then return; end
    PickupContainerItem(bagID, slotIndex);
    SocketInventoryItem(slotID);
    ClickSocketButton(1);
    AcceptSockets();
    ClearCursor();
    --CloseSocketInfo();
end

local function UpdateInnerShadowStates(self)
	local currValue = self:GetValue();
    local minVal, maxVal = self:GetMinMaxValues();
    
	if ( currValue >= maxVal -20) then
		self.BottomShadow:Hide();
    else
        self.BottomShadow:Show();
    end
    
	if ( currValue <= minVal +20) then
		self.TopShadow:Hide();
    else
        self.TopShadow:Show();
	end
end

function Narci_ItemSocketing_ScrollFrame_OnLoad(self)
    local TabHeight = BUTTON_HEIGHT;
    local TotalTab = 5;
    local TotalHeight = floor(TotalTab * TabHeight + 0.5);
    local MaxScroll = floor((TotalTab-1) * TabHeight + 0.5);
    self.buttonHeight = TotalHeight;
    self.range = MaxScroll;
    self.scrollBar:SetScript("OnValueChanged", function(self, value)
        self:GetParent():SetVerticalScroll(value);
        UpdateInnerShadowStates(self);
    end)
    NarciAPI_SmoothScroll_Initialization(self, nil, nil, 1/(TotalTab), 0.14, 0.4);      --(self, updatedList, updateFunc, deltaRatio, timeRatio, minOffset)
end


function Narci_ItemSocketing_Succeed()
    FadeFrame(Narci_ItemSocketing, 0.5, "OUT");
    ShowFlyoutBlack(false);
    HideUIPanel(ItemSocketingFrame);
    Narci_ItemSocketing_GemFrame.Flare.Rotate:Play();
    PlaySound(84378);
    After(0.25, function()
        local slot = Narci_ItemSocketing.anchorSlot;
        slot:Refresh();
        After(0.5, function()
            slot:Refresh();
        end);
    end);
    RefreshAvailableGems();
end

NarciGemListButtonMixin = {};

function NarciGemListButtonMixin:OnEnter()
    UIFrameFadeIn(self.Icon3, 0.12, self.Icon3:GetAlpha(), 0.66);
end

function NarciGemListButtonMixin:OnLeave()
    UIFrameFadeIn(self.Icon3, 0.2, self.Icon3:GetAlpha(), 0);
end

function NarciGemListButtonMixin:OnClick()

end

function NarciGemListButtonMixin:ConfirmSocketing()
    local slot = Narci_ItemSocketing.anchorSlot;
    local slotID = slot:GetID();
    if not slotID then return; end

    Narci_ItemSocketing:RegisterEvent("UI_ERROR_MESSAGE");
    Narci_AlertFrame_Autohide:SetAnchor(Narci_ItemSocketing_GemFrame, -14, true);
    AutoSocket(slotID, self.gemID);
end

function NarciGemListButtonMixin:OnDoubleClick()
    self:ConfirmSocketing()
    --self.ButtonHighlight.animQuickFill:Play();
    self.ButtonFill.animFill:Stop();
end

function NarciGemListButtonMixin:OnMouseDown()
    if self:IsEnabled() then
        self.ButtonFill.animFill:Play();
    end
end

function NarciGemListButtonMixin:OnMouseUp()
    self.ButtonFill.animFill:Stop();
end

function NarciGemListButtonMixin:InsertGem()
    if not self.gemID then return; end
    ClearCursor();
    local bagID, slotIndex = GetBagPosition(self.gemID);
    if not(bagID and slotIndex) then return; end
    PickupContainerItem(bagID, slotIndex);
    ClickSocketButton(1)
    ClearCursor();
end

function NarciGemListButtonMixin:SetItem(itemID, quantity, disabledID)
    local _, _, _, _, icon, _, itemSubClassID = GetItemInfoInstant(itemID);
    local name = C_Item.GetItemNameByID(itemID);
    self.gemID = itemID;
    self.ItemName:SetText(name);
    self.Icon:SetTexture(icon);
    self.Icon2:SetTexture(icon);
    self.Icon3:SetTexture(icon);
    self.Count:SetText(quantity);
    local bonus, minLevel = GET_BONUS_FUNC(itemID);
    self.Bonus:SetText(bonus);
    local texCoord1, texCoord2 = GetBorderTexCoord(itemSubClassID, itemID);
    self.Border0:SetTexCoord(texCoord1, texCoord2, 0, 0.5);
    self.Border1:SetTexCoord(texCoord1, texCoord2, 0.5, 1);

    if itemID == disabledID or (minLevel and minLevel > SOCKETED_ITEM_LEVEL) then
        --Irrelevent Item Alpha
        self:Disable();
        self.ItemName:SetTextColor(0.44, 0.44, 0.44, 1);
    else
        local quality = C_Item.GetItemQualityByID(itemID);
        local r, g, b = GetItemQualityColor(quality);
        self:Enable();
        self.ItemName:SetTextColor(r, g, b, 1);
    end
end

--------------------------------------
--[[
local Finder = CreateFrame("Frame")
Finder:RegisterEvent("ITEM_LOCKED");
Finder:RegisterEvent("SOCKET_INFO_UPDATE");
Finder:RegisterEvent("SOCKET_INFO_CLOSE");
Finder:RegisterEvent("SOCKET_INFO_BIND_CONFIRM");
Finder:RegisterEvent("SOCKET_INFO_ACCEPT");
Finder:RegisterEvent("SOCKET_INFO_SUCCESS");
Finder:RegisterEvent("SOCKET_INFO_FAILURE");
Finder:RegisterEvent("ADDON_LOADED");
Finder:SetScript("OnEvent", function(self,event,...)
    print(event)
    if event == "ITEM_LOCKED" then
        local bagOrSlotIndex, slotIndex = ...;
        if slotIndex ~= nil then
            --print("bag: "..bagOrSlotIndex.."  slotIndex: "..slotIndex)
        else
            --print("bag: "..bagOrSlotIndex)
        end
    end
end)
--]]


---------------------------------------
NarciGemSlotMixin = {};

function NarciGemSlotMixin:CountGems()
    local numGems, type1, type2;
    local list;
    if self.isDomiationSocket then
        GET_BONUS_FUNC = GetShardBonus;
        list = DominationShardIDs;
    else
        GET_BONUS_FUNC = GetGemBonus;
        list = GemIDList;
    end
    numGems, type1, type2 = GetMatchCount(list, GemCountList);
    self.numGems = numGems;
    return numGems, type1, type2;
end

function NarciGemSlotMixin:OnEnter()
    if Narci_ItemSocketing:IsShown() then
		return;
    end
    local link = self:GetParent().GemLink;
    local tooltip = Narci_GearEnhancement_Tooltip;

    --Show optional gem types in your inventory
    local numGems, type1, type2 = self:CountGems();

    local text;     --Show how many types of gems in bags
    if numGems > 0 then
        if numGems == 1 then
            text = type1;
        else
            if type1 == type2 then
                type1 = type1.."+";    --Greater +40/+30 Versa
            end
            if numGems == 2 then
                text = format( L["Gem Tooltip Format1"], type1, type2);
            elseif numGems > 2 then
                text = format( L["Gem Tooltip Format2"], type1, type2, (numGems - 2) );
            end
        end
        tooltip.OtherGems.Text:SetText(text);
        tooltip.OtherGems:Show();
    else
        tooltip.OtherGems:Hide();
    end

    local ItemName = tooltip.ArtFrame.ItemName;
    ItemName:ClearAllPoints();
    local _, bonus, name, quality, icon;

    if link then
        if self.isDomiationSocket then
            bonus = GetShardBonus(self.sockedGemItemID);
        else
            bonus = GetGemBonus(link);
        end
        name, _, quality, _, _, _, _, _, _, icon = GetItemInfo(link);
        ItemName:SetPoint("TOPLEFT", tooltip.ArtFrame, "LEFT", 25, -3);
    else
        name = EMPTY;
        quality = 0;
        icon = 458977;
        ItemName:SetPoint("LEFT", tooltip.ArtFrame, "LEFT", 25, 0);
    end

	local r, g, b = GetItemQualityColor(quality);

	tooltip.ArtFrame.Icon:SetTexture(icon);
	ItemName:SetText(name);
	ItemName:SetTextColor(r, g, b);
	tooltip.ArtFrame.Bonus:SetText(bonus);

	tooltip:ClearAllPoints();
	tooltip:SetParent(self);
	tooltip:SetFrameStrata("TOOLTIP");

	if self:GetParent().isRight then
		tooltip:SetPoint("TOPRIGHT", self, "TOPLEFT", 1, 10);
	else
		tooltip:SetPoint("TOPLEFT", self, "TOPRIGHT", -1, 10);
	end

    tooltip:SetMouseMotionEnabled(false);
	FadeFrame(tooltip, 0.15, "IN");
end


function NarciGemSlotMixin:OnLeave()
    FadeFrame(Narci_GearEnhancement_Tooltip, 0.25, "OUT");
end

function NarciGemSlotMixin:OnClick()
    local frame = Narci_ItemSocketing;
    local GemFrame = Narci_ItemSocketing_GemFrame;
    frame.anchorSlot = self:GetParent();
    if frame:IsShown() then
        frame:Hide();
        ShowFlyoutBlack(false);
        HideUIPanel(ItemSocketingFrame);
    else
        self:CountGems();
        SOCKETED_ITEM_LEVEL= self.ItemLevel;
        if self.numGems == 0 then return; end;
        DisplayButtons(GemCountList, self.sockedGemItemID);
        frame:ClearAllPoints();
        GemFrame:ClearAllPoints();
        --local scale = self:GetEffectiveScale();
        --frame:SetScale(scale);
        --GemFrame:SetScale(scale); 
        frame:SetParent(Narci_Character);
        GemFrame:SetParent(Narci_Character);
        frame:SetFrameLevel(20);
        GemFrame:SetPoint("CENTER", self, "CENTER");
        GemFrame.GemBorder:SetTexture(self.GemBorder:GetTexture());
        GemFrame.GemIcon:SetTexture(self.GemIcon:GetTexture());
        if self.isRight then
            frame:SetPoint("TOPRIGHT", self, "TOPLEFT", 1 ,10);
            GemFrame.GemBorder:SetTexCoord(1, 0, 0, 1);
            GemFrame.Bling:SetTexCoord(0.5, 0, 0, 1);
        else
            frame:SetPoint("TOPLEFT", self, "TOPRIGHT", -1 ,10);
            GemFrame.GemBorder:SetTexCoord(0, 1, 0, 1);
            GemFrame.Bling:SetTexCoord(0, 0.5, 0, 1);
        end
        frame:SetAlpha(1);
        frame:Show();
        GemFrame:SetFrameStrata("TOOLTIP");
        GemFrame:Show();
        ShowFlyoutBlack(true);
        FadeFrame(Narci_GearEnhancement_Tooltip, 0.15, "OUT");
    end
    Narci:HideButtonTooltip();
end

function Narci_ItemSocketing_Close()
    Narci_ItemSocketing:Hide();
    ShowFlyoutBlack(false);
    HideUIPanel(ItemSocketingFrame);
end
---------------------------------------------
--------For Blizzard ItemSocketing UI--------
---------------------------------------------
local currentSocketingItemLink;
local currentSocketingItemGemID = 1208;

local function GetCurrentSocketingItem(bag, slot)
    local itemlocation;
    if not bag then
        itemlocation = ItemLocation:CreateFromEquipmentSlot(slot);
    else
        itemlocation = ItemLocation:CreateFromBagAndSlot(bag, slot);
    end
    return C_Item.GetItemLink(itemlocation);
end

local function GetCurrentGemID(itemLink, gemIndex)
    if not itemLink then return; end
    local index = gemIndex or 1;
    local _, gemLink = GetItemGem(itemLink, index);
    if not gemLink or type(gemLink) ~= "string" then return; end
    local itemID = GetItemInfoInstant(gemLink)
    return itemID;
end

function Narci_ItemSocketing_OnEvent(self, event, ...)
    local buttons = self.buttons;
    if not buttons then return; end
    local button, count, id;
    After(0.6, function()
        for i = 1, #buttons do
            button = buttons[i]
            id = button.gemID
            if not id then return; end
            count = GetItemCount(id);
            button.Count:SetText(count);
            if count == 0 then
                button:Disable();
            end
        end
    end)
end

hooksecurefunc("SocketInventoryItem", function(slot)
    currentSocketingItemLink = GetCurrentSocketingItem(nil, slot)
end)

hooksecurefunc("SocketContainerItem", function(bag, slot)
    currentSocketingItemLink = GetCurrentSocketingItem(bag, slot)
end)

hooksecurefunc("ItemSocketingFrame_LoadUI", function(name)
    if not Narci_Character or Narci_Character:IsShown() or not NarcissusDB or not NarcissusDB.GemManager then return; end

    local frame = Narci_ItemSocketing_ForBlizzardUI;
    if not frame then
        frame = CreateFrame("Frame", "Narci_ItemSocketing_ForBlizzardUI", ItemSocketingFrame, "Narci_ItemSocketing_ForBlizzardUI_Template")
    end

    RefreshAvailableGems();

    if #GemCountList == 0 then return; end
    DisplayButtons(GemCountList, 1208, frame, "Narci_OptionalGems_Template")
    After(0.5, function()
        currentSocketingItemGemID = GetCurrentGemID(currentSocketingItemLink)
        DisplayButtons(GemCountList, currentSocketingItemGemID, frame, "Narci_OptionalGems_Template")
    end)
    frame:SetAlpha(1);
    frame:ClearAllPoints();
    frame:SetPoint("TOPLEFT", ItemSocketingFrame, "TOPRIGHT", 4 ,0)
    local UIScale = UIParent:GetEffectiveScale()
    frame:SetScale(math.max(UIScale, 0.75));
    frame:Show();
end)