FtsFolder.cs

using System.Threading.Tasks;
using Waher.Persistence.FullTextSearch.Files;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Exceptions;
using Waher.Script.Model;
using Waher.Script.Objects;
 
namespace Waher.Script.FullTextSearch.Functions
{
    public class FtsFolder : FunctionMultiVariate
    {
        public FtsFolder(ScriptNode IndexCollection, ScriptNode Folder,
            int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { IndexCollection, Folder }, argumentTypes2Scalar,
                  Start, Length, Expression)
        {
        }
 
        public FtsFolder(ScriptNode IndexCollection, ScriptNode Folder, ScriptNode Recursive,
            int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { IndexCollection, Folder, Recursive },
                  argumentTypes3Scalar, Start, Length, Expression)
        {
        }
 
        public FtsFolder(ScriptNode IndexCollection, ScriptNode Folder, ScriptNode Recursive, ScriptNode SubFoldersToExclude,
            int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { IndexCollection, Folder, Recursive, SubFoldersToExclude },
                  new ArgumentType[] { ArgumentType.Scalar, ArgumentType.Scalar, ArgumentType.Scalar, ArgumentType.Vector }, 
                  Start, Length, Expression)
        {
        }
 
        public override string FunctionName => nameof(FtsFolder);
 
        public override string[] DefaultArgumentNames => new string[] { "IndexCollection", "Folder", "Recursive" };
 
        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)
        {
            int i = 0;
            int c = Arguments.Length;
            string Index = Arguments[i++].AssociatedObjectValue?.ToString() ?? string.Empty;
            string Folder = Arguments[i++].AssociatedObjectValue?.ToString() ?? string.Empty;
            bool? Recursive = i < c ? ToBoolean(Arguments[i++]) : false;
 
            if (!Recursive.HasValue)
                throw new ScriptRuntimeException("Expected Boolean Value for Recursive argument.", this);
 
            string[] SubFoldersToExclude;
 
            if (i < c)
            {
                if (Arguments[i] is IVector V)
                {
                    int j, d = V.Dimension;
 
                    SubFoldersToExclude = new string[d];
 
                    for (j = 0; j < d; j++)
                        SubFoldersToExclude[j] = V.GetElement(j).AssociatedObjectValue?.ToString();
                }
                else
                    SubFoldersToExclude = new string[] { Arguments[i].AssociatedObjectValue?.ToString() };
            }
            else
                SubFoldersToExclude = new string[0];
 
            FolderIndexationStatistics Result = await Waher.Persistence.FullTextSearch.Search.IndexFolder(Index, Folder, Recursive.Value, SubFoldersToExclude);
 
            return new ObjectValue(Result);
        }
    }
}