RegexKeyword.cs

using System.Collections.Generic;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using Waher.Persistence.Filters;
using Waher.Runtime.Collections;
 
namespace Waher.Persistence.FullTextSearch.Keywords
{
    public class RegexKeyword : Keyword
    {
        public RegexKeyword(string Expression)
        {
            this.Expression = Expression;
            this.Parsed = new Regex(Expression, RegexOptions.IgnoreCase | RegexOptions.Singleline);
 
            if (Expression.EndsWith(".*"))
            {
                string s = Expression.Substring(0, Expression.Length - 2);
 
                this.Ignore = string.IsNullOrEmpty(s) || FullTextSearchModule.IsStopWord(s);
            }
            else
                this.Ignore = false;
 
        }
 
        public string Expression { get; }
 
        public Regex Parsed { get; }
 
        public override int OrderComplexity => this.Expression.Length;
 
        public override bool Ignore { get; }
 
        public override bool Equals(object obj)
        {
            return obj is RegexKeyword k && this.Expression == k.Expression;
        }
 
        public override string ToString()
        {
            return this.Expression;
        }
 
        public override async Task<IEnumerable<KeyValuePair<string, TokenReferences>>> GetTokenReferences(SearchProcess Process)
        {
            string Preamble = FilterFieldLikeRegEx.GetRegExConstantPrefix(this.Expression, this.Parsed);
            ChunkedList<KeyValuePair<string, TokenReferences>> Result = new ChunkedList<KeyValuePair<string, TokenReferences>>();
 
            if (string.IsNullOrEmpty(Preamble))
            {
                string[] Keys = await Process.Index.GetKeysAsync();
 
                foreach (string Key in Keys)
                {
                    Match M = this.Parsed.Match(Key);
                    if (M.Success && M.Index == 0 && M.Length == Key.Length)
                    {
                        foreach (KeyValuePair<string, object> P in await Process.Index.GetEntriesAsync(Key, Key + "!"))
                        {
                            if (P.Value is TokenReferences References)
                                Result.Add(new KeyValuePair<string, TokenReferences>(P.Key, References));
                        }
                    }
                }
            }
            else
            {
                char[] Characters = Preamble.ToCharArray();
                Characters[Characters.Length - 1]++;
                string ToExclusive = new string(Characters);
 
                KeyValuePair<string, object>[] Records = await Process.Index.GetEntriesAsync(Preamble, ToExclusive);
 
                foreach (KeyValuePair<string, object> Rec in Records)
                {
                    Match M = this.Parsed.Match(Rec.Key);
                    if (M.Success && M.Index == 0 && M.Length == Rec.Key.Length && Rec.Value is TokenReferences References)
                        Result.Add(new KeyValuePair<string, TokenReferences>(Rec.Key, References));
                }
            }
 
            return Result;
        }
    }
}