Mercurial > wow > lfg-premade-filter

annotate LFGFilter.lua @ 15:28bee190a79f v4

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
v4 docs + toc
author ovolkov
date Wed, 04 Feb 2015 07:43:36 +0300
parents 9431f27780c3
children ba1754998685
rev   line source
ovolkov@7 1 local dump = DevTools_Dump
ovolkov@3 2 LFGListFrame.SearchPanel.SearchBox:SetMaxLetters(2048)
ovolkov@0 3
ovolkov@3 4 local filter_expression_functions = setmetatable({}, {
ovolkov@3 5 __mode = "k",
ovolkov@3 6 __index = function(t, key)
ovolkov@3 7 local func, error = loadstring("return " .. key)
ovolkov@3 8 if error then print("Error in LFG filter expression:\n", error) end
ovolkov@3 9 t[key] = func
ovolkov@3 10 return func
ovolkov@3 11 end
ovolkov@3 12 })
ovolkov@3 13
ovolkov@10 14 local aliases = {
ovolkov@10 15 hm = "highmaul",
ovolkov@10 16 healers = "healer",
ovolkov@10 17 members = "member",
ovolkov@10 18 tanks = "tank",
ovolkov@10 19 damagers = "damager",
ovolkov@10 20 damage = "damager",
ovolkov@10 21 plates = "plate",
ovolkov@10 22 mails = "mail",
ovolkov@10 23 leathers = "leather",
ovolkov@10 24 cloths = "cloth",
ovolkov@10 25 clothies = "cloth",
ovolkov@10 26 }
ovolkov@10 27 for idx = 1, MAX_CLASSES do
ovolkov@10 28 local class_lc = CLASS_SORT_ORDER[idx]:lower()
ovolkov@10 29 aliases[class_lc .. "s"] = class_lc
ovolkov@10 30 end
ovolkov@10 31
ovolkov@11 32 local token_to_encounter_id = {
ovolkov@11 33 highmaul = {
ovolkov@12 34 lfg_dungeon_id = 849,
ovolkov@12 35 { "kargath", "bladefist", "kargath_bladefist" },
ovolkov@12 36 { "butcher", "the_butcher" },
ovolkov@12 37 { "tectus" },
ovolkov@12 38 { "brackenspore", "bracken" },
ovolkov@12 39 { "twin_orgon", "twins" },
ovolkov@12 40 { "koragh", "breaker" },
ovolkov@12 41 { "imperator", "margok" },
ovolkov@11 42 }
ovolkov@11 43 }
ovolkov@11 44
ovolkov@3 45 function LFGListSearchPanel_DoSearch(self)
ovolkov@3 46 local searchText = self.SearchBox:GetText();
ovolkov@3 47 local real_search, filter_expression = searchText:match("^([^=]-)=(.+)$")
ovolkov@3 48 if filter_expression then
ovolkov@3 49 filter_expression = filter_expression:lower()
ovolkov@3 50 self.filter_func = filter_expression_functions[filter_expression]
ovolkov@3 51 end
ovolkov@3 52 self.filter_expression = filter_expression
ovolkov@3 53
ovolkov@3 54 -- print("lfgsearch", real_search, filter_expression)
ovolkov@3 55 C_LFGList.Search(self.categoryID, real_search or searchText, self.filters, self.preferredFilters);
ovolkov@3 56 self.searching = true;
ovolkov@3 57 self.searchFailed = false;
ovolkov@3 58 self.selectedResult = nil;
ovolkov@3 59 LFGListSearchPanel_UpdateResultList(self);
ovolkov@3 60 LFGListSearchPanel_UpdateResults(self);
ovolkov@3 61 end
ovolkov@3 62
ovolkov@12 63 local localized_encounter_name_to_idx = {}
ovolkov@11 64
ovolkov@11 65 local function InsertEncounterStateAliases(result_env, raid_token, completed_encounters)
ovolkov@11 66 local encounter_aliases = token_to_encounter_id[raid_token]
ovolkov@11 67 if not encounter_aliases then return end
ovolkov@11 68
ovolkov@12 69 local lfg_dungeon_id = encounter_aliases.lfg_dungeon_id
ovolkov@12 70 local encounter_names = localized_encounter_name_to_idx[lfg_dungeon_id]
ovolkov@11 71 if not encounter_names then
ovolkov@11 72 encounter_names = {}
ovolkov@12 73 for idx = 1, GetLFGDungeonNumEncounters(lfg_dungeon_id) do
ovolkov@12 74 local bossName, texture, isKilled = GetLFGDungeonEncounterInfo(lfg_dungeon_id, idx)
ovolkov@12 75 encounter_names[bossName] = idx
ovolkov@12 76 end
ovolkov@12 77 localized_encounter_name_to_idx[lfg_dungeon_id] = encounter_names
ovolkov@11 78 end
ovolkov@11 79
ovolkov@11 80 for idx = 1, #completed_encounters do
ovolkov@11 81 local encounter_id = encounter_names[completed_encounters[idx]]
ovolkov@11 82 if encounter_id then
ovolkov@11 83 local aliases = encounter_aliases[encounter_id]
ovolkov@11 84 if aliases then
ovolkov@11 85 for alias_idx = 1, #aliases do
ovolkov@11 86 result_env[aliases[alias_idx]] = true
ovolkov@11 87 end
ovolkov@11 88 end
ovolkov@11 89 end
ovolkov@11 90 end
ovolkov@11 91 end
ovolkov@11 92
ovolkov@3 93 local result_env = {}
ovolkov@3 94 -- =highmaul and ((normal and (name:match("imp") or defeated == 6)) or (heroic and defeated == 2))
ovolkov@0 95 function LFGListUtil_SortSearchResults(results)
ovolkov@3 96 local self = LFGListFrame.SearchPanel
ovolkov@3 97 if self.filter_expression then
ovolkov@3 98 local check = self.filter_func
ovolkov@0 99 local shift_down = 0
ovolkov@0 100 local original_size = #results
ovolkov@0 101 for idx = 1, original_size do
ovolkov@3 102 local id = results[idx]
ovolkov@3 103 local _, activityID, name, comment, voiceChat, iLvl, age, numBNetFriends, numCharFriends, numGuildMates, isDelisted, leaderName, numMembers = C_LFGList.GetSearchResultInfo(id)
ovolkov@3 104 local completedEncounters = C_LFGList.GetSearchResultEncounterInfo(id)
ovolkov@3 105 local memberCounts = C_LFGList.GetSearchResultMemberCounts(id)
ovolkov@3 106
ovolkov@3 107 wipe(result_env)
ovolkov@3 108 result_env.name = name:lower()
ovolkov@3 109 result_env.comment = comment:lower()
ovolkov@3 110 result_env.ilvl = iLvl
ovolkov@3 111 -- TODO: should be calculated in meta
ovolkov@3 112 result_env.defeated = completedEncounters and #completedEncounters or 0
ovolkov@6 113 result_env.member = numMembers
ovolkov@6 114 result_env.tank = memberCounts.TANK
ovolkov@6 115 result_env.healer = memberCounts.HEALER
ovolkov@6 116 result_env.damager = memberCounts.DAMAGER + memberCounts.NOROLE
ovolkov@3 117 result_env.my_server = leaderName and not leaderName:find('-')
ovolkov@3 118
ovolkov@7 119 for idx = 1, numMembers do
ovolkov@7 120 local role, class, classLocalized = C_LFGList.GetSearchResultMemberInfo(id, idx)
ovolkov@7 121 local class_lc = class:lower()
ovolkov@7 122 local prev_count = result_env[class_lc]
ovolkov@7 123 result_env[class_lc] = prev_count and (prev_count + 1) or 0
ovolkov@7 124 end
ovolkov@7 125 for idx = 1, MAX_CLASSES do
ovolkov@7 126 local class_lc = CLASS_SORT_ORDER[idx]:lower()
ovolkov@7 127 local count = result_env[class_lc]
ovolkov@10 128 if not count then result_env[class_lc] = 0 end
ovolkov@7 129 end
ovolkov@7 130
ovolkov@11 131 local raid_token
ovolkov@3 132 if activityID == 37 then
ovolkov@11 133 raid_token = "highmaul"
ovolkov@3 134 result_env.normal = true
ovolkov@3 135 elseif activityID == 38 then
ovolkov@11 136 raid_token = "highmaul"
ovolkov@3 137 result_env.heroic = true
ovolkov@11 138
ovolkov@3 139 end
ovolkov@3 140
ovolkov@11 141 result_env[raid_token] = true
ovolkov@11 142 if completedEncounters then InsertEncounterStateAliases(result_env, raid_token, completedEncounters) end
ovolkov@11 143
ovolkov@11 144 result_env.dead = true
ovolkov@11 145 result_env.defeated = true
ovolkov@11 146
ovolkov@11 147 -- dump(result_env)
ovolkov@10 148
ovolkov@10 149 for alias, original in pairs(aliases) do result_env[alias] = result_env[original] end
ovolkov@10 150
ovolkov@10 151 -- dump(result_env)
ovolkov@7 152
ovolkov@3 153 local pass
ovolkov@3 154 if check then
ovolkov@3 155 setfenv(check, result_env)
ovolkov@3 156 pass = check()
ovolkov@3 157 end
ovolkov@3 158
ovolkov@3 159 if pass then
ovolkov@0 160 if shift_down > 0 then
ovolkov@3 161 results[idx - shift_down] = id
ovolkov@0 162 end
ovolkov@0 163 else
ovolkov@0 164 shift_down = shift_down + 1
ovolkov@0 165 end
ovolkov@3 166
ovolkov@0 167 end
ovolkov@0 168 for idx = original_size - shift_down + 1, original_size do
ovolkov@0 169 results[idx] = nil
ovolkov@0 170 end
ovolkov@0 171 end
ovolkov@3 172 table.sort(results, LFGListUtil_SortSearchResultsCB);
ovolkov@0 173 end
ovolkov@0 174
ovolkov@0 175 function LFGListUtil_SortSearchResultsCB(id1, id2)
ovolkov@0 176 local id1, activityID1, name1, comment1, voiceChat1, iLvl1, age1, numBNetFriends1, numCharFriends1, numGuildMates1, isDelisted1 = C_LFGList.GetSearchResultInfo(id1);
ovolkov@0 177 local id2, activityID2, name2, comment2, voiceChat2, iLvl2, age2, numBNetFriends2, numCharFriends2, numGuildMates2, isDelisted2 = C_LFGList.GetSearchResultInfo(id2);
ovolkov@0 178
ovolkov@0 179 --If one has more friends, do that one first
ovolkov@0 180 if ( numBNetFriends1 ~= numBNetFriends2 ) then
ovolkov@0 181 return numBNetFriends1 > numBNetFriends2;
ovolkov@0 182 end
ovolkov@0 183
ovolkov@0 184 if ( numCharFriends1 ~= numCharFriends2 ) then
ovolkov@0 185 return numCharFriends1 > numCharFriends2;
ovolkov@0 186 end
ovolkov@0 187
ovolkov@0 188 if ( numGuildMates1 ~= numGuildMates2 ) then
ovolkov@0 189 return numGuildMates1 > numGuildMates2;
ovolkov@0 190 end
ovolkov@0 191
ovolkov@0 192 if ( activityID1 ~= activityID2 ) then
ovolkov@0 193 return activityID1 > activityID2;
ovolkov@0 194 end
ovolkov@0 195
ovolkov@0 196 --If we aren't sorting by anything else, just go by ID
ovolkov@0 197 return id1 < id2;
ovolkov@0 198 end
ovolkov@0 199
ovolkov@0 200 function LFGPrintRawResults()
ovolkov@0 201 local totalResults, results = C_LFGList.GetSearchResults()
ovolkov@0 202 for idx = 1, totalResults do
ovolkov@0 203 local id1, activityID1, name1, comment1, voiceChat1, iLvl1, age1, numBNetFriends1, numCharFriends1, numGuildMates1, isDelisted1 = C_LFGList.GetSearchResultInfo(results[idx])
ovolkov@0 204 print(id1, activityID1, C_LFGList.GetActivityInfo(activityID1), '*', name1)
ovolkov@0 205 end
ovolkov@13 206 end