Baggins-Search.lua

--luacheck: no max line length

local NumberofAddons = C_AddOns and C_AddOns.GetNumAddOns and C_AddOns.GetNumAddOns() or GetNumAddOns()
local DisableAddOn = C_AddOns and C_AddOns.DisableAddOn and C_AddOns.DisableAddOn or DisableAddOn
local GetAddOnInfo = C_AddOns and C_AddOns.GetAddOnInfo and C_AddOns.GetAddOnInfo or GetAddOnInfo
StaticPopupDialogs["BAGGINS_SEARCH"] = { --luacheck: ignore 112
    text = "Addon Baggins Search was found, however Baggins has search built-in, Would you like to disable Baggins Search?(requires reloadui, may need to disable manually)",
    button1 = "Yes",
    button2 = "No",
    OnAccept = function()
        for i=1,NumberofAddons do
            local _, title, _, _, _, _, _ = GetAddOnInfo(i)
            if title == "Baggins Search" or title == "Baggins_Search" then
                DisableAddOn(i)
                DisableAddOn(i)
            end
        end
    end,
    timeout = 0,
    whileDead = true,
    hideOnEscape = true,
    preferredIndex = 3,  -- avoid some UI taint, see http://www.wowace.com/announcements/how-to-avoid-some-ui-taint/
}


if NumberofAddons >= 1 then
    for i=1,NumberofAddons do
        local _, title, _, _, reason, _, _ = GetAddOnInfo(i)
        if title == "Baggins Search" or title == "Baggins_Search" then
            if reason == nil or reason ~= "DISABLED" then
                StaticPopup_Show("BAGGINS_SEARCH")
                return
            end
        end
    end
end

local Baggins = Baggins
local GetItemInfo = C_Item and C_Item.GetItemInfo or GetItemInfo
local GetDetailedItemLevelInfo = C_Item.GetDetailedItemLevelInfo
local strlen = strlen
local strfind = strfind
local GetContainerItemLink = C_Container and C_Container.GetContainerItemLink or GetContainerItemLink
local CreateFrame = CreateFrame
local ChatFontNormal = ChatFontNormal
local GameTooltip = GameTooltip
local GameTooltip_SetDefaultAnchor = GameTooltip_SetDefaultAnchor
local getglobal = getglobal
local IsControlKeyDown = IsControlKeyDown
local CreateColor = CreateColor

-- Simple search inspired by vBagnon for Baggins

local BagginsSearch = {}

local itemBindTypes = {
    [0] = "None",
    [1] = "Bind on Pickup",
    [2] = "Bind on Equip",
    [3] = "Bind on Use",
    [4] = "Quest"
}

local itemBindTypesAB = {
    [0] = "None",
    [1] = "BoP",
    [2] = "BoE",
    [3] = "BoU",
    [4] = "Quest"
}

local itemExpansion = {
    [0] = "Classic",
    [1] = "The Burning Crusade",
    [2] = "Wrath of the Lich King",
    [3] = "Cataclysm",
    [4] = "Mists of Pandaria",
    [5] = "Warlords of Draenor",
    [6] = "Legion",
    [7] = "Battle for Azeroth",
    [8] = "Shadowlands",
    [9] = "Dragonflight",
}

local itemExpansionAB = {
    [0] = "Classic",
    [1] = "BC",
    [2] = "Wrath",
    [3] = "Cata",
    [4] = "MoP",
    [5] = "WoD",
    [6] = "Legion",
    [7] = "BFA",
    [8] = "Shadowlands",
    [9] = "DF",
}

local itemQualityT = {
    [0]	= "Poor",
    [1]	= "Common",
    [2]	= "Uncommon",
    [3]	= "Rare",
    [4]	= "Epic",
    [5]	= "Legendary",
    [6]	= "Artifact",
    [7]	= "Heirloom",
    [8]	= "WoWToken",
}

function BagginsSearch:Search(search) --luacheck: ignore 212
    local itemName, itemQuality, itemLevel, itemType, itemSubType, itemEquipLoc, bindType, expacID, setID
    local effectiveILvl, baseILvl, GearILevel -- effectiveILvl, isPreview, baseILvl, GearILevel
    for _, bag in ipairs(Baggins.bagframes) do --bagid,bag
        for _, section in ipairs(bag.sections) do --sectionid, section
            for _, button in ipairs(section.items) do --buttonid, button
                if button:IsVisible() then
                    if strlen(search) == 0 or search == "" then
                        button:UnlockHighlight()
                        button:SetAlpha(1)
                    else
                        local link = GetContainerItemLink(button:GetParent():GetID(), button:GetID())
                        if link then
                            --itemName, itemLink, itemQuality, itemLevel, itemMinLevel, itemType, itemSubType, itemStackCount,itemEquipLoc, itemTexture, sellPrice, classID, subclassID, bindType, expacID, setID, isCraftingReagent
                            if Baggins:IsRetailWow() then
                                itemName, _, itemQuality, itemLevel, _, itemType, itemSubType, _,itemEquipLoc, _, _, _, _, bindType, expacID, setID = GetItemInfo(link)
                                effectiveILvl = select(1,GetDetailedItemLevelInfo(link))
                                baseILvl = select(3,GetDetailedItemLevelInfo(link))
                                GearILevel = effectiveILvl or baseILvl or itemLevel or 0
                            else
                                itemName, _, itemQuality, itemLevel, _, itemType, itemSubType, _,itemEquipLoc, _, _, _, _, bindType, expacID = GetItemInfo(link)
                            end
                            if not itemName then
                                -- hack hack hack
                                itemName = string.match(link, "|h%[(.*)%]") or ""
                                -- TODO: should figure out what type of thing this is so we can populate these:
                                --itemType = ""
                                --itemSubType = ""
                                --itemEquipLoc = ""
                            end
                            if Baggins:IsRetailWow() then
                                if itemName and strfind(itemName:lower(), search:lower()) or
                                itemType and strfind(itemType:lower(), search:lower()) or
                                itemSubType and strfind(itemSubType:lower(), search:lower()) or
                                itemEquipLoc and strfind(itemEquipLoc:lower(), search:lower()) or
                                bindType and strfind(itemBindTypes[bindType]:lower(), search:lower()) or
                                bindType and strfind(itemBindTypesAB[bindType]:lower(), search:lower()) or
                                expacID and itemExpansion[expacID] and strfind(itemExpansion[expacID]:lower(), search:lower()) or
                                expacID and itemExpansionAB[expacID] and strfind(itemExpansionAB[expacID]:lower(), search:lower()) or
                                itemQuality and strfind(itemQualityT[itemQuality]:lower(), search:lower()) or
                                GearILevel and strfind(GearILevel, search) or
                                setID and strfind(setID, search) then
                                    button:LockHighlight()
                                    button:SetAlpha(1)
                                else
                                    button:UnlockHighlight()
                                    button:SetAlpha(tonumber(Baggins.db.profile.unmatchedAlpha) or 0.2)
                                end
                            else
                                if itemName and strfind(itemName:lower(), search:lower()) or
                                itemType and strfind(itemType:lower(), search:lower()) or
                                itemSubType and strfind(itemSubType:lower(), search:lower()) or
                                itemEquipLoc and strfind(itemEquipLoc:lower(), search:lower()) or
                                bindType and strfind(itemBindTypes[bindType]:lower(), search:lower()) or
                                bindType and strfind(itemBindTypesAB[bindType]:lower(), search:lower()) or
                                itemQuality and strfind(itemQualityT[itemQuality]:lower(), search:lower()) or
                                itemLevel and strfind(itemLevel, search) then
                                    button:LockHighlight()
                                    button:SetAlpha(1)
                                else
                                    button:UnlockHighlight()
                                    button:SetAlpha(tonumber(Baggins.db.profile.unmatchedAlpha) or 0.2)
                                end
                            end
                        end
                    end
                end
            end
        end
    end
end

local BagginsSearch_Label
local BagginsSearch_EditBox
local function BagginsSearch_CreateEditBox()
    -- Create Baggins Search EditBox
    local LSM = LibStub:GetLibrary("LibSharedMedia-3.0", true) --luacheck:ignore 113
    BagginsSearch_EditBox = CreateFrame('EditBox', nil, UIParent, BackdropTemplateMixin and "BackdropTemplate") --luacheck: ignore 113
    BagginsSearch_EditBox:SetWidth(100)
    BagginsSearch_EditBox:SetHeight(24)
    --editBox:SetScale(Baggins.db.profile.scale)
    BagginsSearch_EditBox:SetFrameStrata("HIGH")

    BagginsSearch_EditBox:SetFontObject(ChatFontNormal)
    BagginsSearch_EditBox:SetFont(LSM and LSM:Fetch("font", Baggins.db.profile.Font) or STANDARD_TEXT_FONT,Baggins.db.profile.FontSize or 10, "")
    BagginsSearch_EditBox:SetTextInsets(8, 8, 0, 0)
    BagginsSearch_EditBox:SetAutoFocus(false)

    BagginsSearch_EditBox:SetBackdrop({edgeFile = "Interface/Tooltips/UI-Tooltip-Border", edgeSize = 16, insets = {left = 2, right = 2, top = 2, bottom = 2}})
    BagginsSearch_EditBox:SetBackdropBorderColor(0.6, 0.6, 0.6)

    local background = BagginsSearch_EditBox:CreateTexture(nil, "BACKGROUND")
    background:SetTexture("Interface/ChatFrame/ChatFrameBackground")
    background:SetPoint("TOPLEFT", 4, -4)
    background:SetPoint("BOTTOMRIGHT", -4, 4)
    background:SetGradient("VERTICAL", CreateColor(0, 0, 0, 0.9), CreateColor(0.2, 0.2, 0.2, 0.9))

    BagginsSearch_EditBox:SetScript("OnHide", function(self)
            self:SetText("")
            BagginsSearch_Label:Show()
        end)
    BagginsSearch_EditBox:SetScript("OnEnterPressed", function(self) self:ClearFocus() end)
    BagginsSearch_EditBox:SetScript("OnEscapePressed", function(self)
            self:SetText("")
            self:ClearFocus()
            BagginsSearch_Label:Show()
        end)
    BagginsSearch_EditBox:SetScript("OnEditFocusGained", function(self)
            if IsControlKeyDown() then
                self:SetText("")
                self:ClearFocus()
                BagginsSearch_Label:Show()
            else
                BagginsSearch_Label:Hide()
                self:HighlightText()
            end
        end)
    BagginsSearch_EditBox:SetScript("OnTextChanged", function(self)
            BagginsSearch:Search(self:GetText())
        end)
    BagginsSearch_EditBox:SetScript("OnEnter", function(self)
            GameTooltip_SetDefaultAnchor(GameTooltip, self)
            GameTooltip:SetText("Baggins Search")
            --GameTooltip:AddLine("|c00FFFFFFv" .. BagginsSearch.version .. "|r")
            GameTooltip:Show()
        end)
    BagginsSearch_EditBox:SetScript("OnLeave", function() GameTooltip:Hide() end)

    BagginsSearch_Label = BagginsSearch_EditBox:CreateFontString(nil, "OVERLAY", "GameFontHighlight")
    BagginsSearch_Label:SetAlpha(0.2)
    BagginsSearch_Label:SetFont(LSM and LSM:Fetch("font", Baggins.db.profile.Font) or STANDARD_TEXT_FONT,Baggins.db.profile.FontSize or 10, "")
    BagginsSearch_Label:SetText("Search")
    BagginsSearch_Label:SetPoint("TOPLEFT", 8, 0)
    BagginsSearch_Label:SetPoint("BOTTOMLEFT", -8, 0)
    BagginsSearch_Label:Show()

end

function BagginsSearch:UpdateEditBoxPosition() --luacheck: ignore 212
    if Baggins.db.profile.enableSearch then
        if not BagginsSearch_EditBox then
            BagginsSearch_CreateEditBox()
        end
    end
    if not Baggins.db.profile.enableSearch then
        if BagginsSearch_EditBox then
            BagginsSearch_EditBox:Hide()
            return
        end
    end
    local initialcorner = Baggins.db.profile.layoutanchor
    local lastBag
    if (initialcorner == "BOTTOMRIGHT" or initialcorner == "BOTTOMLEFT") then
      if type(Baggins.bagframes) == "table" then
        for bagid, _ in ipairs(Baggins.bagframes) do --bagid, bag
          if Baggins.bagframes[bagid]:IsVisible() then
            lastBag = bagid
          end
        end
      end
    elseif (initialcorner == "TOPRIGHT" or initialcorner == "TOPLEFT") then
     if Baggins.bagframes[1]:IsVisible() then
        lastBag = 1
      end
    end
    if BagginsSearch_EditBox and lastBag then
        BagginsSearch_EditBox:ClearAllPoints()
        BagginsSearch_EditBox:SetPoint("BOTTOMRIGHT", "BagginsBag"..lastBag, "TOPRIGHT", 0, 0)
        BagginsSearch_EditBox:SetWidth(getglobal("BagginsBag"..lastBag):GetWidth())
        BagginsSearch_EditBox:Show()
    else
        if BagginsSearch_EditBox then
            BagginsSearch_EditBox:Hide()
        end
    end
end

Baggins:RegisterSignal("Baggins_UpdateBagScale",function() --luacheck: ignore 212
    if BagginsSearch_EditBox then
        BagginsSearch_EditBox:SetScale(Baggins.db.profile.scale)
    end
end, Baggins)


Baggins:RegisterSignal("Baggins_RefreshBags",function() --luacheck: ignore 212
    BagginsSearch:UpdateEditBoxPosition()
    if BagginsSearch_EditBox then
        BagginsSearch:Search(BagginsSearch_EditBox:GetText())
    end
end, Baggins)

Baggins:RegisterSignal("Baggins_AllBagsClosed",function() --luacheck: ignore 212
    local initialcorner = Baggins.db.profile.layoutanchor
    local lastBag
    if (initialcorner == "BOTTOMRIGHT" or initialcorner == "BOTTOMLEFT") then
      if type(Baggins.bagframes) == "table" then
        for bagid, _ in ipairs(Baggins.bagframes) do --bagid, bag
          if Baggins.bagframes[bagid]:IsVisible() then
            lastBag = bagid
          end
        end
      end
    elseif (initialcorner == "TOPRIGHT" or initialcorner == "TOPLEFT") then
     if Baggins.bagframes[1]:IsVisible() then
        lastBag = 1
      end
    end
    if BagginsSearch_EditBox and lastBag then
        BagginsSearch_EditBox:ClearAllPoints()
        BagginsSearch_EditBox:SetPoint("BOTTOMRIGHT", "BagginsBag"..lastBag, "TOPRIGHT", 0, 0)
        BagginsSearch_EditBox:SetWidth(getglobal("BagginsBag"..lastBag):GetWidth())
        BagginsSearch_EditBox:Show()
    else
        if BagginsSearch_EditBox then
            BagginsSearch_EditBox:Hide()
        end
    end
end, Baggins)

local f = CreateFrame('Frame')
f:RegisterEvent("PLAYER_ENTERING_WORLD")
f:SetScript("OnEvent", function()
    if Baggins.db.profile.enableSearch then
        BagginsSearch_CreateEditBox()
        BagginsSearch:UpdateEditBoxPosition()
        BagginsSearch_EditBox:SetScale(Baggins.db.profile.scale)
    end
    Baggins.OnMenuRequest.args.General.args.BagginsSearchEnable = {
        name = "Enable Search",
        type = "toggle",
        desc = "Enable/Disable Search",
        order = 250,
        get = function() return Baggins.db.profile.enableSearch end,
        set = function(_, value)
            Baggins.db.profile.enableSearch = value
            BagginsSearch:UpdateEditBoxPosition()
        end
    }
    Baggins.OnMenuRequest.args.General.args.BagginsSearchFade = {
        name = "Search Item Fade",
        type = "range",
        desc = "Set the transparency for unmatched items",
        order = 251,
        max = 1,
        min = 0,
        step = 0.05,
        get = function() return Baggins.db.profile.unmatchedAlpha end,
        set = function(_, value)
            Baggins.db.profile.unmatchedAlpha = value
            BagginsSearch:Search(BagginsSearch_EditBox:GetText())
        end
    }
end)