ConnectMsSql.cs

using Microsoft.Data.SqlClient;
using System.Security;
using System.Threading.Tasks;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Data.Model;
using Waher.Script.Model;
using Waher.Script.Objects;
 
namespace Waher.Script.Data.Functions
{
    public class ConnectMsSql : FunctionMultiVariate
    {
        public ConnectMsSql(ScriptNode ConnectionString, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { ConnectionString }, argumentTypes1Scalar, Start, Length, Expression)
        {
        }
 
        public ConnectMsSql(ScriptNode Host, ScriptNode Database, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Host, Database }, argumentTypes2Scalar, Start, Length, Expression)
        {
        }
 
        public ConnectMsSql(ScriptNode ConnectionString, ScriptNode UserName, ScriptNode Password, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { ConnectionString, UserName, Password }, argumentTypes3Scalar, Start, Length, Expression)
        {
        }
 
        public ConnectMsSql(ScriptNode Host, ScriptNode Database, ScriptNode UserName, ScriptNode Password, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Host, Database, UserName, Password }, argumentTypes4Scalar, Start, Length, Expression)
        {
        }
 
        public override string FunctionName => nameof(ConnectMsSql);
 
        public override string[] DefaultArgumentNames => new string[] { "Host", "Database", "UserName", "Password" };
 
        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 ConnectionString = ToString(Arguments[0]) ?? string.Empty;
            SqlConnection Connection;
 
            switch (Arguments.Length)
            {
                case 1:
                default:
                    Connection = new SqlConnection(ConnectionString);
                    break;
 
                case 2:
                    string Database = ToString(Arguments[1]) ?? string.Empty;
 
                    ConnectionString = "Data Source=" + ConnectionString + ";Initial Catalog=" + Database + ";Integrated Security=true";
                    Connection = new SqlConnection(ConnectionString);
                    break;
                case 3:
                    string UserName = ToString(Arguments[1]) ?? string.Empty;
                    string Password = ToString(Arguments[2]) ?? string.Empty;
                    SecureString Password2 = new SecureString();
 
                    foreach (char ch in Password)
                        Password2.AppendChar(ch);
 
                    Password2.MakeReadOnly();
                    Connection = new SqlConnection(ConnectionString, new SqlCredential(UserName, Password2));
                    break;
 
                case 4:
                    Database = ToString(Arguments[1]) ?? string.Empty;
                    UserName = ToString(Arguments[2]) ?? string.Empty;
                    Password = ToString(Arguments[3]) ?? string.Empty;
                    Password2 = new SecureString();
 
                    foreach (char ch in Password)
                        Password2.AppendChar(ch);
 
                    Password2.MakeReadOnly();
 
                    ConnectionString = "Data Source=" + ConnectionString + ";Initial Catalog=" + Database;
                    Connection = new SqlConnection(ConnectionString, new SqlCredential(UserName, Password2));
                    break;
            }
 
            await Connection.OpenAsync();
 
            return new ObjectValue(new MsSqlDatabase(Connection));
        }
    }
}