AddFtsProperties.cs

using System.Threading.Tasks;
using Waher.Persistence.FullTextSearch;
using Waher.Persistence.FullTextSearch.PropertyEvaluators;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Exceptions;
using Waher.Script.Model;
using Waher.Script.Objects;
using Waher.Script.Operators;
 
namespace Waher.Script.FullTextSearch.Functions
{
    public class AddFtsProperties : FunctionMultiVariate
    {
        public AddFtsProperties(ScriptNode Collection, ScriptNode Properties,
            int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Collection, Properties },
                  new ArgumentType[] { ArgumentType.Scalar, ArgumentType.Vector },
                  Start, Length, Expression)
        {
        }
 
        public override string FunctionName => nameof(AddFtsProperties);
 
        public override string[] DefaultArgumentNames => new string[]
        {
            "Collection", "Properties"
        };
 
        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)
        {
            string Collection = Arguments[0].AssociatedObjectValue?.ToString() ?? string.Empty;
            PropertyDefinition[] Properties = GetPropertyDefinitions(Arguments[1], this);
 
            bool Result = await Waher.Persistence.FullTextSearch.Search.AddFullTextSearch(Collection, Properties);
 
            return Result ? BooleanValue.True : BooleanValue.False;
        }
 
        public static PropertyDefinition[] GetPropertyDefinitions(IElement Argument, ScriptNode Node)
        {
            object Obj = Argument.AssociatedObjectValue;
 
            if (Obj is PropertyDefinition[] Result)
                return Result;
 
            if (Obj is string[] Labels)
                return PropertyDefinition.ToArray(Labels);
 
            if (Obj is PropertyDefinition Def)
                return new PropertyDefinition[] { Def };
 
            if (Obj is string s)
                return new PropertyDefinition[] { new PropertyDefinition(s) };
 
            if (Obj is Expression Exp)
                return new PropertyDefinition[] { new PropertyDefinition(typeof(ExpressionEvaluator).FullName, Exp.Script) };
 
            if (Obj is LambdaDefinition Lambda0 && Lambda0.NrArguments == 1)
                return new PropertyDefinition[] { new PropertyDefinition(typeof(LambdaEvaluator).FullName, Lambda0.SubExpression) };
 
            if (!(Argument is IVector V))
                throw new ScriptRuntimeException("Expected a vector of property definitions.", Node);
 
            int i, c = V.Dimension;
 
            Result = new PropertyDefinition[c];
 
            for (i = 0; i < c; i++)
            {
                Argument = V.GetElement(i);
                Obj = Argument.AssociatedObjectValue;
 
                if (Obj is PropertyDefinition Def2)
                    Result[i] = Def2;
                else if (Obj is string s2)
                    Result[i] = new PropertyDefinition(s2);
                else if (Obj is Expression Expression)
                    Result[i] = new PropertyDefinition(typeof(ExpressionEvaluator).FullName, Expression.Script);
                else if (Obj is LambdaDefinition Lambda && Lambda.NrArguments == 1)
                    Result[i] = new PropertyDefinition(typeof(LambdaEvaluator).FullName, Lambda.SubExpression);
                else
                    throw new ScriptRuntimeException("Unrecognized property definition: " + Obj.GetType().FullName, Node);
            }
 
            return Result;
        }
 
    }
}