mirror of https://github.com/ErsatzTV/ErsatzTV.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
110 lines
3.8 KiB
110 lines
3.8 KiB
using Lucene.Net.Analysis; |
|
using Lucene.Net.Index; |
|
using Lucene.Net.QueryParsers.Classic; |
|
using Lucene.Net.Search; |
|
using Lucene.Net.Util; |
|
using Query = Lucene.Net.Search.Query; |
|
|
|
namespace ErsatzTV.Infrastructure.Search; |
|
|
|
public class CustomMultiFieldQueryParser : MultiFieldQueryParser |
|
{ |
|
public CustomMultiFieldQueryParser( |
|
LuceneVersion matchVersion, |
|
string[] fields, |
|
Analyzer analyzer, |
|
IDictionary<string, float> boosts) : base(matchVersion, fields, analyzer, boosts) |
|
{ |
|
} |
|
|
|
public CustomMultiFieldQueryParser(LuceneVersion matchVersion, string[] fields, Analyzer analyzer) : base( |
|
matchVersion, |
|
fields, |
|
analyzer) |
|
{ |
|
} |
|
|
|
protected override Query GetFieldQuery(string field, string queryText, bool quoted) |
|
{ |
|
if (field == "released_onthisday") |
|
{ |
|
var todayString = DateTime.Today.ToString("*MMdd"); |
|
return base.GetWildcardQuery(SearchIndex.ReleaseDateField, todayString); |
|
} |
|
|
|
if (CustomQueryParser.NumericFields.Contains(field) && int.TryParse(queryText, out int val)) |
|
{ |
|
var bytesRef = new BytesRef(); |
|
NumericUtils.Int32ToPrefixCoded(val, 0, bytesRef); |
|
return NewTermQuery(new Term(field, bytesRef)); |
|
} |
|
|
|
return base.GetFieldQuery(field, queryText, quoted); |
|
} |
|
|
|
protected override Query GetFieldQuery(string field, string queryText, int slop) |
|
{ |
|
if (field == "released_inthelast" && CustomQueryParser.ParseStart(queryText, out DateTime start)) |
|
{ |
|
var todayString = DateTime.UtcNow.ToString("yyyyMMdd"); |
|
var dateString = start.ToString("yyyyMMdd"); |
|
|
|
return base.GetRangeQuery(SearchIndex.ReleaseDateField, dateString, todayString, true, true); |
|
} |
|
|
|
if (field == "released_notinthelast" && CustomQueryParser.ParseStart(queryText, out DateTime finish)) |
|
{ |
|
var dateString = finish.ToString("yyyyMMdd"); |
|
|
|
return base.GetRangeQuery(SearchIndex.ReleaseDateField, "00000000", dateString, false, false); |
|
} |
|
|
|
if (field == "added_inthelast" && CustomQueryParser.ParseStart(queryText, out DateTime addedStart)) |
|
{ |
|
var todayString = DateTime.UtcNow.ToString("yyyyMMdd"); |
|
var dateString = addedStart.ToString("yyyyMMdd"); |
|
|
|
return base.GetRangeQuery(SearchIndex.AddedDateField, dateString, todayString, true, true); |
|
} |
|
|
|
if (field == "added_notinthelast" && CustomQueryParser.ParseStart(queryText, out DateTime addedFinish)) |
|
{ |
|
var dateString = addedFinish.ToString("yyyyMMdd"); |
|
|
|
return base.GetRangeQuery(SearchIndex.AddedDateField, "00000000", dateString, false, false); |
|
} |
|
|
|
return base.GetFieldQuery(field, queryText, slop); |
|
} |
|
|
|
protected override Query GetRangeQuery( |
|
string field, |
|
string part1, |
|
string part2, |
|
bool startInclusive, |
|
bool endInclusive) |
|
{ |
|
if (CustomQueryParser.NumericFields.Contains(field)) |
|
{ |
|
if (part1 is null or "*" && int.TryParse(part2, out int max1)) |
|
{ |
|
return NumericRangeQuery.NewInt32Range(field, null, max1, startInclusive, endInclusive); |
|
} |
|
|
|
if (int.TryParse(part1, out int min)) |
|
{ |
|
if (part2 is null or "*") |
|
{ |
|
return NumericRangeQuery.NewInt32Range(field, min, null, startInclusive, endInclusive); |
|
} |
|
|
|
if (int.TryParse(part2, out int max)) |
|
{ |
|
return NumericRangeQuery.NewInt32Range(field, min, max, startInclusive, endInclusive); |
|
} |
|
} |
|
} |
|
|
|
return base.GetRangeQuery(field, part1, part2, startInclusive, endInclusive); |
|
} |
|
}
|
|
|