ScriptExtensions.cs

using System.Reflection;
using Waher.Script;
using Waher.Script.Content;
using Waher.Script.Content.Functions.Encoding;
using Waher.Script.Data.Functions;
using Waher.Script.Fractals;
using Waher.Script.Functions.Runtime;
using Waher.Script.Model;
using Waher.Script.Networking.Functions;
using Waher.Script.Operators;
using Waher.Script.Operators.Assignments;
using Waher.Script.Operators.Membership;
using Waher.Script.Persistence.SQL;
 
namespace LegalLab.Extensions
{
    public static class ScriptExtensions
    {
        private readonly static Assembly scriptContent = typeof(GraphEncoder).Assembly;
        private readonly static Assembly[] prohibitedAssemblies = new Assembly[]
        {
            scriptContent,                      // Waher.Script.Content
            typeof(FractalGraph).Assembly,      // Waher.Script.Fractals
            typeof(WhoIs).Assembly,             // Waher.Script.Networking
            typeof(Select).Assembly,            // Waher.Script.Persistence
            typeof(ConnectMsSql).Assembly       // Waher.Script.Data
        };
 
        public static bool CheckExpressionSafe(this Expression Expression, out ScriptNode Prohibited)
        {
            return CheckExpressionSafe(Expression, false, false, false, out Prohibited);
        }
 
        public static bool CheckExpressionSafe(this Expression Expression, bool AllowNamedMembers, bool AllowError,
            bool AllowCustomFunctions, out ScriptNode Prohibited)
        {
            ScriptNode Prohibited2 = null;
            bool Safe = Expression.ForAll((ScriptNode Node, out ScriptNode NewNode, object State) =>
            {
                NewNode = null;
 
                Assembly Assembly = Node.GetType().Assembly;
 
                foreach (Assembly A in prohibitedAssemblies)
                {
                    if (A.FullName == Assembly.FullName)
                    {
                        if (A == scriptContent &&
                            (Node is Waher.Script.Content.Functions.Duration ||
                            Node.GetType().Namespace == typeof(Utf8Encode).Namespace))
                        {
                            return true;
                        }
 
                        Prohibited2 = Node;
                        return false;
                    }
                }
 
                if ((Node is NamedMember && !AllowNamedMembers) ||
                    (Node is NamedMemberAssignment && !AllowNamedMembers) ||
                    (Node is LambdaDefinition && !AllowCustomFunctions) ||
                    Node is NamedMethodCall ||
                    Node is DynamicFunctionCall ||
                    Node is DynamicMember ||
                    Node is Create ||
                    Node is Destroy ||
                    (Node is Error && !AllowError))
                {
                    Prohibited2 = Node;
                    return false;
                }
 
                return true;
 
            }, null, SearchMethod.TreeOrder);
 
            Prohibited = Prohibited2;
            return Safe;
        }
    }
}