refactor: move FilterInfo into its own file
This commit is contained in:
parent
d7c10595c7
commit
466c5c9304
|
@ -0,0 +1,240 @@
|
|||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Reflection;
|
||||
using Dalamud;
|
||||
using Dalamud.Data;
|
||||
using Lumina.Excel.GeneratedSheets;
|
||||
|
||||
namespace Glamaholic.Ui {
|
||||
internal class FilterInfo {
|
||||
private DataManager Data { get; }
|
||||
|
||||
private uint MaxLevel { get; }
|
||||
private string Query { get; }
|
||||
private HashSet<ClassJob> WantedJobs { get; } = new();
|
||||
private HashSet<string> Tags { get; } = new();
|
||||
private HashSet<string> ExcludeTags { get; } = new();
|
||||
private HashSet<uint> ItemIds { get; } = new();
|
||||
private HashSet<string> ItemNames { get; } = new();
|
||||
|
||||
internal FilterInfo(DataManager data, string filter) {
|
||||
this.Data = data;
|
||||
|
||||
var queryWords = new List<string>();
|
||||
|
||||
var quoteType = -1;
|
||||
string? quoted = null;
|
||||
foreach (var immutableWord in filter.Split(' ')) {
|
||||
var word = immutableWord;
|
||||
|
||||
if (quoted != null) {
|
||||
quoted += " ";
|
||||
|
||||
var quoteIndex = word.IndexOf('"');
|
||||
if (quoteIndex > -1) {
|
||||
quoted += word[..quoteIndex];
|
||||
|
||||
switch (quoteType) {
|
||||
case 1:
|
||||
this.Tags.Add(quoted);
|
||||
break;
|
||||
case 2:
|
||||
this.ItemNames.Add(quoted);
|
||||
break;
|
||||
case 3:
|
||||
this.ExcludeTags.Add(quoted);
|
||||
break;
|
||||
}
|
||||
|
||||
quoted = null;
|
||||
quoteType = -1;
|
||||
|
||||
var rest = word[(quoteIndex + 1)..];
|
||||
if (rest.Length > 0) {
|
||||
word = rest;
|
||||
} else {
|
||||
continue;
|
||||
}
|
||||
} else {
|
||||
quoted += word;
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
if (word.StartsWith("j:")) {
|
||||
var abbr = word[2..].ToLowerInvariant();
|
||||
var job = this.Data.GetExcelSheet<ClassJob>()!.FirstOrDefault(row => row.Abbreviation.RawString.ToLowerInvariant() == abbr);
|
||||
if (job != null) {
|
||||
this.WantedJobs.Add(job);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (word.StartsWith("lvl:")) {
|
||||
if (uint.TryParse(word[4..], out var level)) {
|
||||
this.MaxLevel = level;
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (word.StartsWith("t:")) {
|
||||
if (word.StartsWith("t:\"")) {
|
||||
if (word.EndsWith('"') && word.Length >= 5) {
|
||||
this.Tags.Add(word[3..^1]);
|
||||
} else {
|
||||
quoteType = 1;
|
||||
quoted = word[3..];
|
||||
}
|
||||
} else {
|
||||
this.Tags.Add(word[2..]);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (word.StartsWith("!t:")) {
|
||||
if (word.StartsWith("!t:\"")) {
|
||||
if (word.EndsWith('"') && word.Length >= 6) {
|
||||
this.ExcludeTags.Add(word[4..^1]);
|
||||
} else {
|
||||
quoteType = 3;
|
||||
quoted = word[4..];
|
||||
}
|
||||
} else {
|
||||
this.ExcludeTags.Add(word[3..]);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (word.StartsWith("id:")) {
|
||||
if (uint.TryParse(word[3..], out var id)) {
|
||||
this.ItemIds.Add(id);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (word.StartsWith("i:")) {
|
||||
if (word.StartsWith("i:\"")) {
|
||||
if (word.EndsWith('"') && word.Length >= 5) {
|
||||
this.ItemNames.Add(word[3..^1]);
|
||||
} else {
|
||||
quoteType = 2;
|
||||
quoted = word[3..];
|
||||
}
|
||||
} else {
|
||||
this.ItemNames.Add(word[2..]);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
queryWords.Add(word);
|
||||
}
|
||||
|
||||
this.Query = string.Join(' ', queryWords).ToLowerInvariant();
|
||||
}
|
||||
|
||||
internal bool Matches(SavedPlate plate) {
|
||||
// if the name doesn't match the query, it's not a match, obviously
|
||||
if (this.Query.Length != 0 && !plate.Name.ToLowerInvariant().Contains(this.Query)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// if there's nothing custom about this filter, this is a match
|
||||
var notCustom = this.MaxLevel == 0
|
||||
&& this.WantedJobs.Count == 0
|
||||
&& this.Tags.Count == 0
|
||||
&& this.ExcludeTags.Count == 0
|
||||
&& this.ItemIds.Count == 0
|
||||
&& this.ItemNames.Count == 0;
|
||||
if (notCustom) {
|
||||
return true;
|
||||
}
|
||||
|
||||
foreach (var tag in this.Tags) {
|
||||
if (!plate.Tags.Contains(tag)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
foreach (var tag in this.ExcludeTags) {
|
||||
if (plate.Tags.Contains(tag)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
if (this.ItemIds.Count > 0) {
|
||||
var matching = plate.Items.Values
|
||||
.Select(mirage => mirage.ItemId)
|
||||
.Intersect(this.ItemIds)
|
||||
.Count();
|
||||
|
||||
if (matching != this.ItemIds.Count) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
if (this.ItemNames.Count > 0) {
|
||||
var sheet = this.Data.GetExcelSheet<Item>()!;
|
||||
|
||||
var names = plate.Items.Values
|
||||
.Select(mirage => sheet.GetRow(mirage.ItemId % Util.HqItemOffset))
|
||||
.Where(item => item != null)
|
||||
.Cast<Item>()
|
||||
.Select(item => item.Name.RawString.ToLowerInvariant())
|
||||
.ToArray();
|
||||
|
||||
foreach (var needle in this.ItemNames) {
|
||||
var lower = needle.ToLowerInvariant();
|
||||
if (!names.Any(name => name.Contains(lower))) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
foreach (var mirage in plate.Items.Values) {
|
||||
var item = this.Data.GetExcelSheet<Item>()!.GetRow(mirage.ItemId % Util.HqItemOffset);
|
||||
if (item == null) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (this.MaxLevel != 0 && item.LevelEquip > this.MaxLevel) {
|
||||
return false;
|
||||
}
|
||||
|
||||
foreach (var job in this.WantedJobs) {
|
||||
var category = item.ClassJobCategory.Value;
|
||||
if (category == null) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (!this.CanWear(category, job)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
private bool CanWear(ClassJobCategory category, ClassJob classJob) {
|
||||
// get english version
|
||||
var job = this.Data.GetExcelSheet<ClassJob>(ClientLanguage.English)!.GetRow(classJob.RowId)!;
|
||||
var getter = category.GetType().GetProperty(job.Abbreviation.RawString, BindingFlags.Public | BindingFlags.Instance);
|
||||
if (getter == null) {
|
||||
return false;
|
||||
}
|
||||
|
||||
var value = getter.GetValue(category);
|
||||
if (value is bool res) {
|
||||
return res;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
|
@ -5,10 +5,8 @@ using System.Linq;
|
|||
using System.Net;
|
||||
using System.Net.Http;
|
||||
using System.Numerics;
|
||||
using System.Reflection;
|
||||
using System.Threading.Tasks;
|
||||
using Dalamud;
|
||||
using Dalamud.Data;
|
||||
using Dalamud.Interface;
|
||||
using Dalamud.Interface.Colors;
|
||||
using Dalamud.Logging;
|
||||
|
@ -903,237 +901,5 @@ namespace Glamaholic.Ui {
|
|||
.Where(item => this._itemFilter.Length == 0 || item.Name.RawString.ToLowerInvariant().Contains(filter))
|
||||
.ToList();
|
||||
}
|
||||
|
||||
private class FilterInfo {
|
||||
private DataManager Data { get; }
|
||||
|
||||
private uint MaxLevel { get; }
|
||||
private string Query { get; }
|
||||
private HashSet<ClassJob> WantedJobs { get; } = new();
|
||||
private HashSet<string> Tags { get; } = new();
|
||||
private HashSet<string> ExcludeTags { get; } = new();
|
||||
private HashSet<uint> ItemIds { get; } = new();
|
||||
private HashSet<string> ItemNames { get; } = new();
|
||||
|
||||
internal FilterInfo(DataManager data, string filter) {
|
||||
this.Data = data;
|
||||
|
||||
var queryWords = new List<string>();
|
||||
|
||||
var quoteType = -1;
|
||||
string? quoted = null;
|
||||
foreach (var immutableWord in filter.Split(' ')) {
|
||||
var word = immutableWord;
|
||||
|
||||
if (quoted != null) {
|
||||
quoted += " ";
|
||||
|
||||
var quoteIndex = word.IndexOf('"');
|
||||
if (quoteIndex > -1) {
|
||||
quoted += word[..quoteIndex];
|
||||
|
||||
switch (quoteType) {
|
||||
case 1:
|
||||
this.Tags.Add(quoted);
|
||||
break;
|
||||
case 2:
|
||||
this.ItemNames.Add(quoted);
|
||||
break;
|
||||
case 3:
|
||||
this.ExcludeTags.Add(quoted);
|
||||
break;
|
||||
}
|
||||
|
||||
quoted = null;
|
||||
quoteType = -1;
|
||||
|
||||
var rest = word[(quoteIndex + 1)..];
|
||||
if (rest.Length > 0) {
|
||||
word = rest;
|
||||
} else {
|
||||
continue;
|
||||
}
|
||||
} else {
|
||||
quoted += word;
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
if (word.StartsWith("j:")) {
|
||||
var abbr = word[2..].ToLowerInvariant();
|
||||
var job = this.Data.GetExcelSheet<ClassJob>()!.FirstOrDefault(row => row.Abbreviation.RawString.ToLowerInvariant() == abbr);
|
||||
if (job != null) {
|
||||
this.WantedJobs.Add(job);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (word.StartsWith("lvl:")) {
|
||||
if (uint.TryParse(word[4..], out var level)) {
|
||||
this.MaxLevel = level;
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (word.StartsWith("t:")) {
|
||||
if (word.StartsWith("t:\"")) {
|
||||
if (word.EndsWith('"') && word.Length >= 5) {
|
||||
this.Tags.Add(word[3..^1]);
|
||||
} else {
|
||||
quoteType = 1;
|
||||
quoted = word[3..];
|
||||
}
|
||||
} else {
|
||||
this.Tags.Add(word[2..]);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (word.StartsWith("!t:")) {
|
||||
if (word.StartsWith("!t:\"")) {
|
||||
if (word.EndsWith('"') && word.Length >= 6) {
|
||||
this.ExcludeTags.Add(word[4..^1]);
|
||||
} else {
|
||||
quoteType = 3;
|
||||
quoted = word[4..];
|
||||
}
|
||||
} else {
|
||||
this.ExcludeTags.Add(word[3..]);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (word.StartsWith("id:")) {
|
||||
if (uint.TryParse(word[3..], out var id)) {
|
||||
this.ItemIds.Add(id);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (word.StartsWith("i:")) {
|
||||
if (word.StartsWith("i:\"")) {
|
||||
if (word.EndsWith('"') && word.Length >= 5) {
|
||||
this.ItemNames.Add(word[3..^1]);
|
||||
} else {
|
||||
quoteType = 2;
|
||||
quoted = word[3..];
|
||||
}
|
||||
} else {
|
||||
this.ItemNames.Add(word[2..]);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
queryWords.Add(word);
|
||||
}
|
||||
|
||||
this.Query = string.Join(' ', queryWords).ToLowerInvariant();
|
||||
}
|
||||
|
||||
internal bool Matches(SavedPlate plate) {
|
||||
// if the name doesn't match the query, it's not a match, obviously
|
||||
if (this.Query.Length != 0 && !plate.Name.ToLowerInvariant().Contains(this.Query)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// if there's nothing custom about this filter, this is a match
|
||||
var notCustom = this.MaxLevel == 0
|
||||
&& this.WantedJobs.Count == 0
|
||||
&& this.Tags.Count == 0
|
||||
&& this.ExcludeTags.Count == 0
|
||||
&& this.ItemIds.Count == 0
|
||||
&& this.ItemNames.Count == 0;
|
||||
if (notCustom) {
|
||||
return true;
|
||||
}
|
||||
|
||||
foreach (var tag in this.Tags) {
|
||||
if (!plate.Tags.Contains(tag)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
foreach (var tag in this.ExcludeTags) {
|
||||
if (plate.Tags.Contains(tag)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
if (this.ItemIds.Count > 0) {
|
||||
var matching = plate.Items.Values
|
||||
.Select(mirage => mirage.ItemId)
|
||||
.Intersect(this.ItemIds)
|
||||
.Count();
|
||||
|
||||
if (matching != this.ItemIds.Count) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
if (this.ItemNames.Count > 0) {
|
||||
var sheet = this.Data.GetExcelSheet<Item>()!;
|
||||
|
||||
var names = plate.Items.Values
|
||||
.Select(mirage => sheet.GetRow(mirage.ItemId % Util.HqItemOffset))
|
||||
.Where(item => item != null)
|
||||
.Cast<Item>()
|
||||
.Select(item => item.Name.RawString.ToLowerInvariant())
|
||||
.ToArray();
|
||||
|
||||
foreach (var needle in this.ItemNames) {
|
||||
var lower = needle.ToLowerInvariant();
|
||||
if (!names.Any(name => name.Contains(lower))) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
foreach (var mirage in plate.Items.Values) {
|
||||
var item = this.Data.GetExcelSheet<Item>()!.GetRow(mirage.ItemId % Util.HqItemOffset);
|
||||
if (item == null) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (this.MaxLevel != 0 && item.LevelEquip > this.MaxLevel) {
|
||||
return false;
|
||||
}
|
||||
|
||||
foreach (var job in this.WantedJobs) {
|
||||
var category = item.ClassJobCategory.Value;
|
||||
if (category == null) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (!this.CanWear(category, job)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
private bool CanWear(ClassJobCategory category, ClassJob classJob) {
|
||||
// get english version
|
||||
var job = this.Data.GetExcelSheet<ClassJob>(ClientLanguage.English)!.GetRow(classJob.RowId)!;
|
||||
var getter = category.GetType().GetProperty(job.Abbreviation.RawString, BindingFlags.Public | BindingFlags.Instance);
|
||||
if (getter == null) {
|
||||
return false;
|
||||
}
|
||||
|
||||
var value = getter.GetValue(category);
|
||||
if (value is bool res) {
|
||||
return res;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue