123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448 |
- --[[
- ItemSearch
- An item text search engine of some sort
- Grammar:
- <search> := <intersect search>
- <intersect search> := <union search> & <union search> ; <union search>
- <union search> := <negatable search> | <negatable search> ; <negatable search>
- <negatable search> := !<primitive search> ; <primitive search>
- <primitive search> := <tooltip search> ; <quality search> ; <type search> ; <text search>
- <tooltip search> := bop ; boa ; bou ; boe ; quest
- <quality search> := q<op><text> ; q<op><digit>
- <ilvl search> := ilvl<op><number>
- <type search> := t:<text>
- <text search> := <text>
- <op> := : | = | == | != | ~= | < | > | <= | >=
- I kindof half want to make a full parser for this
- --]]
- -- local MAJOR, MINOR = "LibItemSearch-1.0", 3
- -- local ItemSearch = LibStub:NewLibrary(MAJOR, MINOR)
- -- if not ItemSearch then return end
- local myname, ns = ...
- ns.LibItemSearch = {}
- local ItemSearch = ns.LibItemSearch
- --[[ general search ]]--
- function ItemSearch:Find(itemLink, search)
- if not search or search == "" then
- return true
- end
- if not itemLink then
- return false
- end
- local search = search:lower()
- if search:match('\124') then
- return self:FindUnionSearch(itemLink, strsplit('\124', search))
- end
- return self:FindUnionSearch(itemLink, search)
- end
- --[[ union search: <search>&<search> ]]--
- function ItemSearch:FindUnionSearch(itemLink, ...)
- for i = 1, select('#', ...) do
- local search = select(i, ...)
- if search and search ~= '' then
- if search:match('\038') then
- if self:FindIntersectSearch(itemLink, strsplit('\038', search)) then
- return true
- end
- else
- if self:FindIntersectSearch(itemLink, search) then
- return true
- end
- end
- end
- end
- return false
- end
- --[[ intersect search: <search>|<search> ]]--
- function ItemSearch:FindIntersectSearch(itemLink, ...)
- for i = 1, select('#', ...) do
- local search = select(i, ...)
- if search and search ~= '' then
- if not self:FindNegatableSearch(itemLink, search) then
- return false
- end
- end
- end
- return true
- end
- --[[ negated search: !<search> ]]--
- function ItemSearch:FindNegatableSearch(itemLink, search)
- local negatedSearch = search:match('^\033(.+)$')
- if negatedSearch then
- return not self:FindTypedSearch(itemLink, negatedSearch)
- end
- return self:FindTypedSearch(itemLink, search)
- end
- --[[
- typed search:
- user defined search types
- A typed search object should look like the following:
- {
- string id
- unique identifier for the search type,
- string searchCapture = function isSearch(self, search)
- returns a capture if the given search matches this typed search
- returns nil if the search is not a match for this type
- bool isMatch = function findItem(self, itemLink, searchCapture)
- returns true if <itemLink> is in the search defined by <searchCapture>
- }
- --]]
- local typedSearches = {}
- function ItemSearch:RegisterTypedSearch(typedSearchObj)
- typedSearches[typedSearchObj.id] = typedSearchObj
- end
- function ItemSearch:GetTypedSearches()
- return pairs(typedSearches)
- end
- function ItemSearch:GetTypedSearch(id)
- return typedSearches[id]
- end
- function ItemSearch:FindTypedSearch(itemLink, search)
- if not search then
- return false
- end
- for id, searchInfo in self:GetTypedSearches() do
- local capture1, capture2, capture3 = searchInfo:isSearch(search)
- if capture1 then
- return searchInfo:findItem(itemLink, capture1, capture2, capture3)
- end
- end
- return self:GetTypedSearch('itemTypeGeneric'):findItem(itemLink, search) or self:GetTypedSearch('itemName'):findItem(itemLink, search)
- end
- --[[
- Basic typed searches
- --]]
- function ItemSearch:Compare(op, lhs, rhs)
- --ugly, but it works
- if op == ':' or op == '=' or op == '==' then
- return lhs == rhs
- end
- if op == '!=' or op == '~=' then
- return lhs ~= rhs
- end
- if op == '<=' then
- return lhs <= rhs
- end
- if op == '<' then
- return lhs < rhs
- end
- if op == '>' then
- return lhs > rhs
- end
- if op == '>=' then
- return lhs >= rhs
- end
- return false
- end
- --[[ basic text search n:(.+) ]]--
- local function search_IsInText(search, ...)
- for i = 1, select('#', ...) do
- local text = select(i, ...)
- text = text and tostring(text):lower()
- if text and (text == search or text:match(search)) then
- return true
- end
- end
- return false
- end
- ItemSearch:RegisterTypedSearch{
- id = 'itemName',
- isSearch = function(self, search)
- return search and search:match('^n:(.+)$')
- end,
- findItem = function(self, itemLink, search)
- local itemName = (GetItemInfo(itemLink))
- return search_IsInText(search, itemName)
- end
- }
- --[[ item type,subtype,equip loc search t:(.+) ]]--
- ItemSearch:RegisterTypedSearch{
- id = 'itemTypeGeneric',
- isSearch = function(self, search)
- return search and search:match('^t:(.+)$')
- end,
- findItem = function(self, itemLink, search)
- local name, link, quality, iLevel, reqLevel, type, subType, maxStack, equipSlot = GetItemInfo(itemLink)
- if not name then
- return false
- end
- return search_IsInText(search, type, subType, _G[equipSlot])
- end
- }
- --[[ item quality search: q(sign)(%d+) | q:(qualityName) ]]--
- ItemSearch:RegisterTypedSearch{
- id = 'itemQuality',
- isSearch = function(self, search)
- if search then
- return search:match('^q([%~%:%<%>%=%!]+)(%w+)$')
- end
- end,
- descToQuality = function(self, desc)
- local q = 0
- local quality = _G['ITEM_QUALITY' .. q .. '_DESC']
- while quality and quality:lower() ~= desc do
- q = q + 1
- quality = _G['ITEM_QUALITY' .. q .. '_DESC']
- end
- if quality then
- return q
- end
- end,
- findItem = function(self, itemLink, op, search)
- local name, link, quality = GetItemInfo(itemLink)
- if not name then
- return false
- end
- local num = tonumber(search) or self:descToQuality(search)
- return num and ItemSearch:Compare(op, quality, num) or false
- end,
- }
- --[[ item level search: lvl(sign)(%d+) ]]--
- ItemSearch:RegisterTypedSearch{
- id = 'itemLevel',
- isSearch = function(self, search)
- if search then
- return search:match('^ilvl([:<>=!]+)(%d+)$')
- end
- end,
- findItem = function(self, itemLink, op, search)
- local name, link, quality, iLvl = GetItemInfo(itemLink)
- if not iLvl then
- return false
- end
- local num = tonumber(search)
- return num and ItemSearch:Compare(op, iLvl, num) or false
- end,
- }
- --[[ tooltip keyword search ]]--
- local tooltipCache = setmetatable({}, {__index = function(t, k) local v = {} t[k] = v return v end})
- local tooltipScanner = _G['LibItemSearchTooltipScanner'] or CreateFrame('GameTooltip', 'LibItemSearchTooltipScanner', UIParent, 'GameTooltipTemplate')
- local function link_FindSearchInTooltip(itemLink, search)
- --look in the cache for the result
- local itemID = itemLink:match('item:(%d+)')
- local cachedResult = tooltipCache[search][itemID]
- if cachedResult ~= nil then
- return cachedResult
- end
- --no match?, pull in the resut from tooltip parsing
- tooltipScanner:SetOwner(UIParent, 'ANCHOR_NONE')
- tooltipScanner:SetHyperlink(itemLink)
- local result = false
- if tooltipScanner:NumLines() > 1 and _G[tooltipScanner:GetName() .. 'TextLeft2']:GetText() == search then
- result = true
- elseif tooltipScanner:NumLines() > 2 and _G[tooltipScanner:GetName() .. 'TextLeft3']:GetText() == search then
- result = true
- end
- tooltipScanner:Hide()
- tooltipCache[search][itemID] = result
- return result
- end
- ItemSearch:RegisterTypedSearch{
- id = 'tooltip',
- isSearch = function(self, search)
- return self.keywords[search]
- end,
- findItem = function(self, itemLink, search)
- return search and link_FindSearchInTooltip(itemLink, search)
- end,
- keywords = {
- ['boe'] = ITEM_BIND_ON_EQUIP,
- ['bop'] = ITEM_BIND_ON_PICKUP,
- ['bou'] = ITEM_BIND_ON_USE,
- ['quest'] = ITEM_BIND_QUEST,
- ['boa'] = ITEM_BIND_TO_ACCOUNT
- }
- }
- ItemSearch:RegisterTypedSearch{
- id = 'tooltipDesc',
- isSearch = function(self, search)
- return search and search:match('^tt:(.+)$')
- end,
- findItem = function(self, itemLink, search)
- --no match?, pull in the resut from tooltip parsing
- tooltipScanner:SetOwner(UIParent, 'ANCHOR_NONE')
- tooltipScanner:SetHyperlink(itemLink)
- local i = 1
- while i <= tooltipScanner:NumLines() do
- local text = _G[tooltipScanner:GetName() .. 'TextLeft' .. i]:GetText():lower()
- if text and text:match(search) then
- tooltipScanner:Hide()
- return true
- end
- i = i + 1
- end
- tooltipScanner:Hide()
- return false
- end,
- }
- --[[ equipment set search ]]--
- local function IsWardrobeLoaded()
- local name, title, notes, enabled, loadable, reason, security = GetAddOnInfo('Wardrobe')
- return enabled
- end
- local function findEquipmentSetByName(search)
- local startsWithSearch = '^' .. search
- local partialMatch = nil
- for i = 1, GetNumEquipmentSets() do
- local setName = (GetEquipmentSetInfo(i))
- local lSetName = setName:lower()
- if lSetName == search then
- return setName
- end
- if lSetName:match(startsWithSearch) then
- partialMatch = setName
- end
- end
- -- Wardrobe Support
- if Wardrobe then
- for i, outfit in ipairs( Wardrobe.CurrentConfig.Outfit) do
- local setName = outfit.OutfitName
- local lSetName = setName:lower()
- if lSetName == search then
- return setName
- end
- if lSetName:match(startsWithSearch) then
- partialMatch = setName
- end
- end
- end
- return partialMatch
- end
- local function isItemInEquipmentSet(itemLink, setName)
- if not setName then
- return false
- end
- local itemIDs = GetEquipmentSetItemIDs(setName)
- if not itemIDs then
- return false
- end
- local itemID = tonumber(itemLink:match('item:(%d+)'))
- for inventoryID, setItemID in pairs(itemIDs) do
- if itemID == setItemID then
- return true
- end
- end
- return false
- end
- local function isItemInWardrobeSet(itemLink, setName)
- if not Wardrobe then return false end
- local itemName = (GetItemInfo(itemLink))
- for i, outfit in ipairs(Wardrobe.CurrentConfig.Outfit) do
- if outfit.OutfitName == setName then
- for j, item in pairs(outfit.Item) do
- if item and (item.IsSlotUsed == 1) and (item.Name == itemName) then
- return true
- end
- end
- end
- end
- return false
- end
- ItemSearch:RegisterTypedSearch{
- id = 'equipmentSet',
- isSearch = function(self, search)
- return search and search:match('^s:(.+)$')
- end,
- findItem = function(self, itemLink, search)
- local setName = findEquipmentSetByName(search)
- if not setName then
- return false
- end
- return isItemInEquipmentSet(itemLink, setName)
- or isItemInWardrobeSet(itemLink, setName)
- end,
- }
|