add released_inthelast queries (#357)

2021-09-10 21:11:14 -05:00
parent 96bc2c28f2
commit ad2685fb2e
5 changed files with 130 additions and 3 deletions
@@ -0,0 +1,40 @@
+using System;
+using System.Collections.Generic;
+using Lucene.Net.Analysis;
+using Lucene.Net.QueryParsers.Classic;
+using Lucene.Net.Search;
+using Lucene.Net.Util;
+
+namespace ErsatzTV.Infrastructure.Search
+{
+    public class RelativeDateMultiFieldQueryParser : MultiFieldQueryParser
+    {
+        public RelativeDateMultiFieldQueryParser(
+            LuceneVersion matchVersion,
+            string[] fields,
+            Analyzer analyzer,
+            IDictionary<string, float> boosts) : base(matchVersion, fields, analyzer, boosts)
+        {
+        }
+
+        public RelativeDateMultiFieldQueryParser(LuceneVersion matchVersion, string[] fields, Analyzer analyzer) : base(
+            matchVersion,
+            fields,
+            analyzer)
+        {
+        }
+
+        protected override Query GetFieldQuery(string field, string queryText, int slop)
+        {
+            if (field == "released_inthelast" && RelativeDateQueryParser.ParseStart(queryText, out DateTime start))
+            {
+                var todayString = DateTime.Today.ToString("yyyyMMdd");
+                var dateString = start.ToString("yyyyMMdd");
+
+                return base.GetRangeQuery("release_date", dateString, todayString, true, true);
+            }
+
+            return base.GetFieldQuery(field, queryText, slop);
+        }
+    }
+}
@@ -0,0 +1,77 @@
+using System;
+using Lucene.Net.Analysis;
+using Lucene.Net.QueryParsers.Classic;
+using Lucene.Net.Search;
+using Lucene.Net.Util;
+
+namespace ErsatzTV.Infrastructure.Search
+{
+    public class RelativeDateQueryParser : QueryParser
+    {
+        public RelativeDateQueryParser(LuceneVersion matchVersion, string f, Analyzer a) : base(matchVersion, f, a)
+        {
+        }
+
+        protected internal RelativeDateQueryParser(ICharStream stream) : base(stream)
+        {
+        }
+
+        protected RelativeDateQueryParser(QueryParserTokenManager tm) : base(tm)
+        {
+        }
+
+        protected override Query GetFieldQuery(string field, string queryText, int slop)
+        {
+            if (field == "released_inthelast" && ParseStart(queryText, out DateTime start))
+            {
+                var todayString = DateTime.Today.ToString("yyyyMMdd");
+                var dateString = start.ToString("yyyyMMdd");
+
+                return base.GetRangeQuery("release_date", dateString, todayString, true, true);
+            }
+
+            return base.GetFieldQuery(field, queryText, slop);
+        }
+        
+        internal static bool ParseStart(string text, out DateTime start)
+        {
+            start = DateTime.MinValue;
+
+            try
+            {
+                if (int.TryParse(text.Split(" ")[0], out int number))
+                {
+                    if (text.Contains("day"))
+                    {
+                        start = DateTime.Today.AddDays(number * -1);
+                        return true;
+                    }
+
+                    if (text.Contains("week"))
+                    {
+                        start = DateTime.Today.AddDays(number * -7);
+                        return true;
+                    }
+
+                    if (text.Contains("month"))
+                    {
+                        start = DateTime.Today.AddMonths(number * -1);
+                        return true;
+                    }
+
+                    if (text.Contains("year"))
+                    {
+                        start = DateTime.Today.AddYears(number * -1);
+                        return true;
+                    }
+                }
+            }
+            catch
+            {
+                // do nothing
+            }
+
+            return false;
+        }
+    }
+}
@@ -148,8 +148,8 @@ namespace ErsatzTV.Infrastructure.Search
            };
            using var analyzerWrapper = new PerFieldAnalyzerWrapper(analyzer, customAnalyzers);
            QueryParser parser = !string.IsNullOrWhiteSpace(searchField)
-                ? new QueryParser(AppLuceneVersion, searchField, analyzerWrapper)
-                : new MultiFieldQueryParser(AppLuceneVersion, new[] { TitleField }, analyzerWrapper);
+                ? new RelativeDateQueryParser(AppLuceneVersion, searchField, analyzerWrapper)
+                : new RelativeDateMultiFieldQueryParser(AppLuceneVersion, new[] { TitleField }, analyzerWrapper);
            parser.AllowLeadingWildcard = true;
            Query query = ParseQuery(searchQuery, parser);
            var filter = new DuplicateFilter(TitleAndYearField);