Mercurial > wow > lfg-premade-filter

annotate LFGFilter.lua @ 12:64c9517efa88

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
convert local boss name lookups to GetLFGDungeonEncounterInfo
author ovolkov
date Tue, 03 Feb 2015 11:15:14 +0300
parents bd476e399376
children b9af92090f5e
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 -- EJ_GetCurrentInstance()
ovolkov@11 33 -- EJ_SelectInstance(477)
ovolkov@11 34 -- EJ_GetEncounterInfoByIndex(4)
ovolkov@11 35
ovolkov@11 36 local token_to_encounter_id = {
ovolkov@11 37 highmaul = {
ovolkov@12 38 lfg_dungeon_id = 849,
ovolkov@12 39 { "kargath", "bladefist", "kargath_bladefist" },
ovolkov@12 40 { "butcher", "the_butcher" },
ovolkov@12 41 { "tectus" },
ovolkov@12 42 { "brackenspore", "bracken" },
ovolkov@12 43 { "twin_orgon", "twins" },
ovolkov@12 44 { "koragh", "breaker" },
ovolkov@12 45 { "imperator", "margok" },
ovolkov@11 46 }
ovolkov@11 47 }
ovolkov@11 48
ovolkov@3 49 function LFGListSearchPanel_DoSearch(self)
ovolkov@3 50 local searchText = self.SearchBox:GetText();
ovolkov@3 51 local real_search, filter_expression = searchText:match("^([^=]-)=(.+)$")
ovolkov@3 52 if filter_expression then
ovolkov@3 53 filter_expression = filter_expression:lower()
ovolkov@3 54 self.filter_func = filter_expression_functions[filter_expression]
ovolkov@3 55 end
ovolkov@3 56 self.filter_expression = filter_expression
ovolkov@3 57
ovolkov@3 58 -- print("lfgsearch", real_search, filter_expression)
ovolkov@3 59 C_LFGList.Search(self.categoryID, real_search or searchText, self.filters, self.preferredFilters);
ovolkov@3 60 self.searching = true;
ovolkov@3 61 self.searchFailed = false;
ovolkov@3 62 self.selectedResult = nil;
ovolkov@3 63 LFGListSearchPanel_UpdateResultList(self);
ovolkov@3 64 LFGListSearchPanel_UpdateResults(self);
ovolkov@3 65 end
ovolkov@3 66
ovolkov@12 67 local localized_encounter_name_to_idx = {}
ovolkov@11 68
ovolkov@11 69 local function InsertEncounterStateAliases(result_env, raid_token, completed_encounters)
ovolkov@11 70 local encounter_aliases = token_to_encounter_id[raid_token]
ovolkov@11 71 if not encounter_aliases then return end
ovolkov@11 72
ovolkov@12 73 local lfg_dungeon_id = encounter_aliases.lfg_dungeon_id
ovolkov@12 74 local encounter_names = localized_encounter_name_to_idx[lfg_dungeon_id]
ovolkov@11 75 if not encounter_names then
ovolkov@11 76 encounter_names = {}
ovolkov@12 77 for idx = 1, GetLFGDungeonNumEncounters(lfg_dungeon_id) do
ovolkov@12 78 local bossName, texture, isKilled = GetLFGDungeonEncounterInfo(lfg_dungeon_id, idx)
ovolkov@12 79 encounter_names[bossName] = idx
ovolkov@12 80 end
ovolkov@12 81 localized_encounter_name_to_idx[lfg_dungeon_id] = encounter_names
ovolkov@11 82 end
ovolkov@11 83
ovolkov@11 84 for idx = 1, #completed_encounters do
ovolkov@11 85 local encounter_id = encounter_names[completed_encounters[idx]]
ovolkov@11 86 if encounter_id then
ovolkov@11 87 local aliases = encounter_aliases[encounter_id]
ovolkov@11 88 if aliases then
ovolkov@11 89 for alias_idx = 1, #aliases do
ovolkov@11 90 result_env[aliases[alias_idx]] = true
ovolkov@11 91 end
ovolkov@11 92 end
ovolkov@11 93 end
ovolkov@11 94 end
ovolkov@11 95 end
ovolkov@11 96
ovolkov@3 97 local result_env = {}
ovolkov@3 98 -- =highmaul and ((normal and (name:match("imp") or defeated == 6)) or (heroic and defeated == 2))
ovolkov@0 99 function LFGListUtil_SortSearchResults(results)
ovolkov@3 100 local self = LFGListFrame.SearchPanel
ovolkov@3 101 if self.filter_expression then
ovolkov@3 102 local check = self.filter_func
ovolkov@0 103 local shift_down = 0
ovolkov@0 104 local original_size = #results
ovolkov@0 105 for idx = 1, original_size do
ovolkov@3 106 local id = results[idx]
ovolkov@3 107 local _, activityID, name, comment, voiceChat, iLvl, age, numBNetFriends, numCharFriends, numGuildMates, isDelisted, leaderName, numMembers = C_LFGList.GetSearchResultInfo(id)
ovolkov@3 108 local completedEncounters = C_LFGList.GetSearchResultEncounterInfo(id)
ovolkov@3 109 local memberCounts = C_LFGList.GetSearchResultMemberCounts(id)
ovolkov@3 110
ovolkov@3 111 wipe(result_env)
ovolkov@3 112 result_env.name = name:lower()
ovolkov@3 113 result_env.comment = comment:lower()
ovolkov@3 114 result_env.ilvl = iLvl
ovolkov@3 115 -- TODO: should be calculated in meta
ovolkov@3 116 result_env.defeated = completedEncounters and #completedEncounters or 0
ovolkov@6 117 result_env.member = numMembers
ovolkov@6 118 result_env.tank = memberCounts.TANK
ovolkov@6 119 result_env.healer = memberCounts.HEALER
ovolkov@6 120 result_env.damager = memberCounts.DAMAGER + memberCounts.NOROLE
ovolkov@3 121 result_env.my_server = leaderName and not leaderName:find('-')
ovolkov@3 122
ovolkov@7 123 for idx = 1, numMembers do
ovolkov@7 124 local role, class, classLocalized = C_LFGList.GetSearchResultMemberInfo(id, idx)
ovolkov@7 125 local class_lc = class:lower()
ovolkov@7 126 local prev_count = result_env[class_lc]
ovolkov@7 127 result_env[class_lc] = prev_count and (prev_count + 1) or 0
ovolkov@7 128 end
ovolkov@7 129 for idx = 1, MAX_CLASSES do
ovolkov@7 130 local class_lc = CLASS_SORT_ORDER[idx]:lower()
ovolkov@7 131 local count = result_env[class_lc]
ovolkov@10 132 if not count then result_env[class_lc] = 0 end
ovolkov@7 133 end
ovolkov@7 134
ovolkov@11 135 local raid_token
ovolkov@3 136 if activityID == 37 then
ovolkov@11 137 raid_token = "highmaul"
ovolkov@3 138 result_env.normal = true
ovolkov@3 139 elseif activityID == 38 then
ovolkov@11 140 raid_token = "highmaul"
ovolkov@3 141 result_env.heroic = true
ovolkov@11 142
ovolkov@3 143 end
ovolkov@3 144
ovolkov@11 145 result_env[raid_token] = true
ovolkov@11 146 if completedEncounters then InsertEncounterStateAliases(result_env, raid_token, completedEncounters) end
ovolkov@11 147
ovolkov@11 148 result_env.dead = true
ovolkov@11 149 result_env.defeated = true
ovolkov@11 150
ovolkov@11 151 -- dump(result_env)
ovolkov@10 152
ovolkov@10 153 for alias, original in pairs(aliases) do result_env[alias] = result_env[original] end
ovolkov@10 154
ovolkov@10 155 -- dump(result_env)
ovolkov@7 156
ovolkov@3 157 local pass
ovolkov@3 158 if check then
ovolkov@3 159 setfenv(check, result_env)
ovolkov@3 160 pass = check()
ovolkov@3 161 end
ovolkov@3 162
ovolkov@3 163 if pass then
ovolkov@0 164 if shift_down > 0 then
ovolkov@3 165 results[idx - shift_down] = id
ovolkov@0 166 end
ovolkov@0 167 else
ovolkov@0 168 shift_down = shift_down + 1
ovolkov@0 169 end
ovolkov@3 170
ovolkov@0 171 end
ovolkov@0 172 for idx = original_size - shift_down + 1, original_size do
ovolkov@0 173 results[idx] = nil
ovolkov@0 174 end
ovolkov@0 175 end
ovolkov@3 176 table.sort(results, LFGListUtil_SortSearchResultsCB);
ovolkov@0 177 end
ovolkov@0 178
ovolkov@0 179 function LFGListUtil_SortSearchResultsCB(id1, id2)
ovolkov@0 180 local id1, activityID1, name1, comment1, voiceChat1, iLvl1, age1, numBNetFriends1, numCharFriends1, numGuildMates1, isDelisted1 = C_LFGList.GetSearchResultInfo(id1);
ovolkov@0 181 local id2, activityID2, name2, comment2, voiceChat2, iLvl2, age2, numBNetFriends2, numCharFriends2, numGuildMates2, isDelisted2 = C_LFGList.GetSearchResultInfo(id2);
ovolkov@0 182
ovolkov@0 183 --If one has more friends, do that one first
ovolkov@0 184 if ( numBNetFriends1 ~= numBNetFriends2 ) then
ovolkov@0 185 return numBNetFriends1 > numBNetFriends2;
ovolkov@0 186 end
ovolkov@0 187
ovolkov@0 188 if ( numCharFriends1 ~= numCharFriends2 ) then
ovolkov@0 189 return numCharFriends1 > numCharFriends2;
ovolkov@0 190 end
ovolkov@0 191
ovolkov@0 192 if ( numGuildMates1 ~= numGuildMates2 ) then
ovolkov@0 193 return numGuildMates1 > numGuildMates2;
ovolkov@0 194 end
ovolkov@0 195
ovolkov@0 196 if ( activityID1 ~= activityID2 ) then
ovolkov@0 197 return activityID1 > activityID2;
ovolkov@0 198 end
ovolkov@0 199
ovolkov@0 200 --If we aren't sorting by anything else, just go by ID
ovolkov@0 201 return id1 < id2;
ovolkov@0 202 end
ovolkov@0 203
ovolkov@0 204 function LFGPrintRawResults()
ovolkov@0 205 local totalResults, results = C_LFGList.GetSearchResults()
ovolkov@0 206 for idx = 1, totalResults do
ovolkov@0 207 local id1, activityID1, name1, comment1, voiceChat1, iLvl1, age1, numBNetFriends1, numCharFriends1, numGuildMates1, isDelisted1 = C_LFGList.GetSearchResultInfo(results[idx])
ovolkov@0 208 print(id1, activityID1, C_LFGList.GetActivityInfo(activityID1), '*', name1)
ovolkov@0 209 end
ovolkov@11 210 end
ovolkov@11 211
ovolkov@11 212 function LFGPrintInstanceScan(ej_instance)
ovolkov@11 213 EJ_SelectInstance(ej_instance)
ovolkov@11 214 local encounter_idx = 1
ovolkov@11 215 local encounter_name, encounter_id, root_section_id, _
ovolkov@11 216 repeat
ovolkov@11 217 encounter_name, _, encounter_id, root_section_id = EJ_GetEncounterInfoByIndex(encounter_idx)
ovolkov@11 218 if encounter_name then
ovolkov@11 219 print(encounter_id, encounter_name)
ovolkov@11 220 local next_section_id = root_section_id
ovolkov@11 221 repeat
ovolkov@11 222 local title, description, headerType, abilityIcon, displayInfo, siblingID, _, fileredByDifficulty, link, startsOpen, flag1, flag2, flag3, flag4 = EJ_GetSectionInfo(next_section_id)
ovolkov@11 223 if displayInfo ~= 0 then
ovolkov@11 224 print(encounter_id, title)
ovolkov@11 225 end
ovolkov@11 226 next_section_id = siblingID
ovolkov@11 227 until not next_section_id
ovolkov@11 228 encounter_idx = encounter_idx + 1
ovolkov@11 229 end
ovolkov@11 230 until not encounter_name
ovolkov@12 231 localized_encounter_name_to_idx[ej_instance] = encounter_names
ovolkov@11 232 end