FindObjects.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.Reflection;
using System.Threading.Tasks;
using Waher.Persistence;
using Waher.Persistence.Filters;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Exceptions;
using Waher.Script.Model;
using Waher.Script.Objects.VectorSpaces;
using Waher.Script.Operators.Comparisons;
using Waher.Script.Operators.Logical;
 
namespace Waher.Script.Persistence.Functions
{
    public class FindObjects : FunctionMultiVariate
    {
        private static readonly MethodInfo findMethodGeneric = GetFindMethod();
 
        public FindObjects(ScriptNode Type, ScriptNode Offset, ScriptNode MaxCount, ScriptNode Filter, ScriptNode SortOrder, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Type, Offset, MaxCount, Filter, SortOrder }, 
                  new ArgumentType[] { ArgumentType.Scalar, ArgumentType.Scalar, ArgumentType.Scalar, ArgumentType.Scalar, ArgumentType.Vector }, 
                  Start, Length, Expression)
        {
        }
 
        private static MethodInfo GetFindMethod()
        {
            Type T = typeof(Database);
            foreach (MethodInfo MI in T.GetTypeInfo().GetDeclaredMethods("Find"))
            {
                ParameterInfo[] P = MI.GetParameters();
                if (P.Length != 4 ||
                    P[0].ParameterType != typeof(int) ||
                    P[1].ParameterType != typeof(int) ||
                    P[2].ParameterType != typeof(Filter) ||
                    P[3].ParameterType != typeof(string[]))
                {
                    continue;
                }
 
                return MI;
            }
 
            return null;
        }
 
        public override string[] DefaultArgumentNames
        {
            get
            {
                return new string[]
                {
                    "Type",
                    "Offset",
                    "MaxCount",
                    "Filter",
                    "SortOrder"
                };
            }
        }
 
        public override string FunctionName => nameof(FindObjects);
 
        public override bool IsAsynchronous => true;
 
        public override IElement Evaluate(IElement[] Arguments, Variables Variables)
        {
            return this.EvaluateAsync(Arguments, Variables).Result;
        }
 
        public override async Task<IElement> EvaluateAsync(IElement[] Arguments, Variables Variables)
        {
            if (!(Arguments[0].AssociatedObjectValue is Type T))
                throw new ScriptRuntimeException("First parameter must be a type.", this);
 
            int Offset = (int)Expression.ToDouble(Arguments[1].AssociatedObjectValue);
            int MaxCount = (int)Expression.ToDouble(Arguments[2].AssociatedObjectValue);
            object FilterObj = Arguments[3].AssociatedObjectValue;
            Filter Filter = FilterObj as Filter;
            IVector V = Arguments[4] as IVector;
            int i, c = V.Dimension;
            string[] SortOrder = new string[c];
 
            for (i = 0; i < c; i++)
                SortOrder[i] = V.GetElement(i).AssociatedObjectValue.ToString();
 
            if (Filter is null && !(FilterObj is null))
            {
                Expression Exp = new Expression(FilterObj.ToString(), this.Expression.Source);
                Filter = await this.ConvertAsync(Exp.Root, Variables);
            }
 
            MethodInfo MI = findMethodGeneric.MakeGenericMethod(new Type[] { T });
            object Result = MI.Invoke(null, new object[] { Offset, MaxCount, Filter, SortOrder });
            Result = await WaitPossibleTask(Result);
 
            if (Result is IEnumerable E)
            {
                LinkedList<IElement> Elements = new LinkedList<IElement>();
                IEnumerator e = E.GetEnumerator();
                while (e.MoveNext())
                    Elements.AddLast(Expression.Encapsulate(e.Current));
 
                return new ObjectVector(Elements);
            }
            else
                return Expression.Encapsulate(Result);
        }
 
        private async Task<Filter> ConvertAsync(ScriptNode Node, Variables Variables)
        {
            if (Node is null)
                return null;
            else if (Node is And And)
            {
                return new FilterAnd(await this.ConvertAsync(And.LeftOperand, Variables),
                    await this.ConvertAsync(And.RightOperand, Variables));
            }
            else if (Node is Or Or)
            {
                return new FilterOr(await this.ConvertAsync(Or.LeftOperand, Variables),
                    await this.ConvertAsync(Or.RightOperand, Variables));
            }
            else if (Node is Operators.Dual.And And2)
            {
                return new FilterAnd(await this.ConvertAsync(And2.LeftOperand, Variables),
                    await this.ConvertAsync(And2.RightOperand, Variables));
            }
            else if (Node is Operators.Dual.Or Or2)
            {
                return new FilterOr(await this.ConvertAsync(Or2.LeftOperand, Variables),
                    await this.ConvertAsync(Or2.RightOperand, Variables));
            }
            else if (Node is Not Not)
                return new FilterNot(await this.ConvertAsync(Not.Operand, Variables));
            else if (Node is EqualTo EQ)
            {
                KeyValuePair<string, object> P = await this.CheckBinaryOperator(EQ, Variables);
                return new FilterFieldEqualTo(P.Key, P.Value);
            }
            else if (Node is NotEqualTo NEQ)
            {
                KeyValuePair<string, object> P = await this.CheckBinaryOperator(NEQ, Variables);
                return new FilterFieldNotEqualTo(P.Key, P.Value);
            }
            else if (Node is LesserThan LT)
            {
                KeyValuePair<string, object> P = await this.CheckBinaryOperator(LT, Variables);
                return new FilterFieldLesserThan(P.Key, P.Value);
            }
            else if (Node is GreaterThan GT)
            {
                KeyValuePair<string, object> P = await this.CheckBinaryOperator(GT, Variables);
                return new FilterFieldGreaterThan(P.Key, P.Value);
            }
            else if (Node is LesserThanOrEqualTo LTE)
            {
                KeyValuePair<string, object> P = await this.CheckBinaryOperator(LTE, Variables);
                return new FilterFieldLesserOrEqualTo(P.Key, P.Value);
            }
            else if (Node is GreaterThanOrEqualTo GTE)
            {
                KeyValuePair<string, object> P = await this.CheckBinaryOperator(GTE, Variables);
                return new FilterFieldGreaterOrEqualTo(P.Key, P.Value);
            }
            else if (Node is Range Range)
            {
                if (!(Range.MiddleOperand is VariableReference v))
                    throw new ScriptRuntimeException("Middle operands in ternary filter operators need to be a variable references, as they refer to field names.", this);
 
                string FieldName = v.VariableName;
                object Min = (await Range.LeftOperand.EvaluateAsync(Variables)).AssociatedObjectValue;
                object Max = (await Range.RightOperand.EvaluateAsync(Variables)).AssociatedObjectValue;
 
                Filter[] Filters = new Filter[2];
 
                if (Range.LeftInclusive)
                    Filters[0] = new FilterFieldGreaterOrEqualTo(FieldName, Min);
                else
                    Filters[0] = new FilterFieldGreaterThan(FieldName, Min);
 
                if (Range.RightInclusive)
                    Filters[1] = new FilterFieldLesserOrEqualTo(FieldName, Max);
                else
                    Filters[1] = new FilterFieldLesserThan(FieldName, Max);
 
                return new FilterAnd(Filters);
            }
            else if (Node is Like)
            {
                KeyValuePair<string, object> P = await this.CheckBinaryOperator((BinaryOperator)Node, Variables);
                string RegEx = Database.WildcardToRegex(P.Value is string s ? s : Expression.ToString(P.Value), "*");
                return new FilterFieldLikeRegEx(P.Key, RegEx);
            }
            else if (Node is NotLike)
            {
                KeyValuePair<string, object> P = await this.CheckBinaryOperator((BinaryOperator)Node, Variables);
                string RegEx = Database.WildcardToRegex(P.Value is string s ? s : Expression.ToString(P.Value), "*");
                return new FilterNot(new FilterFieldLikeRegEx(P.Key, RegEx));
            }
            else
                throw new ScriptRuntimeException("Invalid operation for filters: " + Node.GetType().FullName, this);
        }
 
        private async Task<KeyValuePair<string, object>> CheckBinaryOperator(BinaryOperator Operator, Variables Variables)
        {
            if (!(Operator.LeftOperand is VariableReference v))
                throw new ScriptRuntimeException("Left operands in binary filter operators need to be a variable references, as they refer to field names.", this);
 
            return new KeyValuePair<string, object>(v.VariableName, (await Operator.RightOperand.EvaluateAsync(Variables)).AssociatedObjectValue);
        }
    }
}