ClientFactory.cs

/*
 *    The contents of this file are subject to the Initial
 *    Developer's Public License Version 1.0 (the "License");
 *    you may not use this file except in compliance with the
 *    License. You may obtain a copy of the License at
 *    https://github.com/FirebirdSQL/NETProvider/raw/master/license.txt.
 *
 *    Software distributed under the License is distributed on
 *    an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, either
 *    express or implied. See the License for the specific
 *    language governing rights and limitations under the License.
 *
 *    All Rights Reserved.
 */

//$Authors = Carlos Guzman Alvarez, Jiri Cincura (jiri@cincura.net)

using System;
using System.Threading;
using System.Threading.Tasks;
using FirebirdSql.Data.Client.Managed;
using FirebirdSql.Data.Common;
using FirebirdSql.Data.FirebirdClient;
using WireCryptOption = FirebirdSql.Data.Client.Managed.Version13.WireCryptOption;

namespace FirebirdSql.Data.Client;

internal static class ClientFactory
{
	public static DatabaseBase CreateDatabase(ConnectionString options)
	{
		return options.ServerType switch
		{
			FbServerType.Default => CreateManagedDatabase(options),
			FbServerType.Embedded => CreateNativeDatabase(options),
			_ => throw IncorrectServerTypeException(),
		};
	}
	public static ValueTask<DatabaseBase> CreateDatabaseAsync(ConnectionString options, CancellationToken cancellationToken = default)
	{
		return options.ServerType switch
		{
			FbServerType.Default => CreateManagedDatabaseAsync(options, cancellationToken),
			FbServerType.Embedded => CreateNativeDatabaseAsync(options),
			_ => throw IncorrectServerTypeException(),
		};
	}

	public static ServiceManagerBase CreateServiceManager(ConnectionString options)
	{
		return options.ServerType switch
		{
			FbServerType.Default => CreateManagedServiceManager(options),
			FbServerType.Embedded => CreateNativeServiceManager(options),
			_ => throw IncorrectServerTypeException(),
		};
	}
	public static ValueTask<ServiceManagerBase> CreateServiceManagerAsync(ConnectionString options, CancellationToken cancellationToken = default)
	{
		return options.ServerType switch
		{
			FbServerType.Default => CreateManagedServiceManagerAsync(options, cancellationToken),
			FbServerType.Embedded => CreateNativeServiceManagerAsync(options),
			_ => throw IncorrectServerTypeException(),
		};
	}

	private static DatabaseBase CreateManagedDatabase(ConnectionString options)
	{
		var charset = GetCharset(options);

		var connection = new GdsConnection(options.UserID, options.Password, options.DataSource, options.Port, options.ConnectionTimeout, options.PacketSize, options.BlobSegmentSize, charset, options.Dialect, options.Compression, FbWireCryptToWireCryptOption(options.WireCrypt), options.CryptKey);
		connection.Connect();
		try
		{
			connection.Identify(options.Database);
		}
		catch
		{
			connection.Disconnect();
			throw;
		}
		return connection.ProtocolVersion switch
		{
			IscCodes.PROTOCOL_VERSION16 => new Managed.Version16.GdsDatabase(connection),
			IscCodes.PROTOCOL_VERSION15 => new Managed.Version15.GdsDatabase(connection),
			IscCodes.PROTOCOL_VERSION13 => new Managed.Version13.GdsDatabase(connection),
			IscCodes.PROTOCOL_VERSION12 => new Managed.Version12.GdsDatabase(connection),
			IscCodes.PROTOCOL_VERSION11 => new Managed.Version11.GdsDatabase(connection),
			IscCodes.PROTOCOL_VERSION10 => new Managed.Version10.GdsDatabase(connection),
			_ => throw UnsupportedProtocolException(),
		};
	}
	private static async ValueTask<DatabaseBase> CreateManagedDatabaseAsync(ConnectionString options, CancellationToken cancellationToken = default)
	{
		var charset = GetCharset(options);

		var connection = new GdsConnection(options.UserID, options.Password, options.DataSource, options.Port, options.ConnectionTimeout, options.PacketSize, options.BlobSegmentSize, charset, options.Dialect, options.Compression, FbWireCryptToWireCryptOption(options.WireCrypt), options.CryptKey);
		await connection.ConnectAsync(cancellationToken).ConfigureAwait(false);
		try
		{
			await connection.IdentifyAsync(options.Database, cancellationToken).ConfigureAwait(false);
		}
		catch
		{
			await connection.DisconnectAsync(cancellationToken).ConfigureAwait(false);
			throw;
		}
		return connection.ProtocolVersion switch
		{
			IscCodes.PROTOCOL_VERSION16 => new Managed.Version16.GdsDatabase(connection),
			IscCodes.PROTOCOL_VERSION15 => new Managed.Version15.GdsDatabase(connection),
			IscCodes.PROTOCOL_VERSION13 => new Managed.Version13.GdsDatabase(connection),
			IscCodes.PROTOCOL_VERSION12 => new Managed.Version12.GdsDatabase(connection),
			IscCodes.PROTOCOL_VERSION11 => new Managed.Version11.GdsDatabase(connection),
			IscCodes.PROTOCOL_VERSION10 => new Managed.Version10.GdsDatabase(connection),
			_ => throw UnsupportedProtocolException(),
		};
	}

	private static DatabaseBase CreateNativeDatabase(ConnectionString options)
	{
		var charset = GetCharset(options);

		return new Native.FesDatabase(options.ClientLibrary, charset, options.PacketSize, options.BlobSegmentSize, options.Dialect);
	}
	private static ValueTask<DatabaseBase> CreateNativeDatabaseAsync(ConnectionString options)
	{
		var charset = GetCharset(options);

		return ValueTask.FromResult<DatabaseBase>(new Native.FesDatabase(options.ClientLibrary, charset, options.PacketSize, options.BlobSegmentSize, options.Dialect));
	}

	private static ServiceManagerBase CreateManagedServiceManager(ConnectionString options)
	{
		var charset = GetCharset(options);

		var connection = new GdsConnection(options.UserID, options.Password, options.DataSource, options.Port, options.ConnectionTimeout, options.PacketSize, options.BlobSegmentSize, charset, options.Dialect, options.Compression, FbWireCryptToWireCryptOption(options.WireCrypt), options.CryptKey);
		connection.Connect();
		try
		{
			connection.Identify(!string.IsNullOrEmpty(options.Database) ? options.Database : string.Empty);
		}
		catch
		{
			connection.Disconnect();
			throw;
		}
		return connection.ProtocolVersion switch
		{
			IscCodes.PROTOCOL_VERSION16 => new Managed.Version16.GdsServiceManager(connection),
			IscCodes.PROTOCOL_VERSION15 => new Managed.Version15.GdsServiceManager(connection),
			IscCodes.PROTOCOL_VERSION13 => new Managed.Version13.GdsServiceManager(connection),
			IscCodes.PROTOCOL_VERSION12 => new Managed.Version12.GdsServiceManager(connection),
			IscCodes.PROTOCOL_VERSION11 => new Managed.Version11.GdsServiceManager(connection),
			IscCodes.PROTOCOL_VERSION10 => new Managed.Version10.GdsServiceManager(connection),
			_ => throw UnsupportedProtocolException(),
		};
	}
	private static async ValueTask<ServiceManagerBase> CreateManagedServiceManagerAsync(ConnectionString options, CancellationToken cancellationToken = default)
	{
		var charset = GetCharset(options);

		var connection = new GdsConnection(options.UserID, options.Password, options.DataSource, options.Port, options.ConnectionTimeout, options.PacketSize, options.BlobSegmentSize, charset, options.Dialect, options.Compression, FbWireCryptToWireCryptOption(options.WireCrypt), options.CryptKey);
		await connection.ConnectAsync(cancellationToken).ConfigureAwait(false);
		try
		{
			await connection.IdentifyAsync(!string.IsNullOrEmpty(options.Database) ? options.Database : string.Empty, cancellationToken).ConfigureAwait(false);
		}
		catch
		{
			await connection.DisconnectAsync(cancellationToken).ConfigureAwait(false);
			throw;
		}
		return connection.ProtocolVersion switch
		{
			IscCodes.PROTOCOL_VERSION16 => new Managed.Version16.GdsServiceManager(connection),
			IscCodes.PROTOCOL_VERSION15 => new Managed.Version15.GdsServiceManager(connection),
			IscCodes.PROTOCOL_VERSION13 => new Managed.Version13.GdsServiceManager(connection),
			IscCodes.PROTOCOL_VERSION12 => new Managed.Version12.GdsServiceManager(connection),
			IscCodes.PROTOCOL_VERSION11 => new Managed.Version11.GdsServiceManager(connection),
			IscCodes.PROTOCOL_VERSION10 => new Managed.Version10.GdsServiceManager(connection),
			_ => throw UnsupportedProtocolException(),
		};
	}

	private static ServiceManagerBase CreateNativeServiceManager(ConnectionString options)
	{
		var charset = GetCharset(options);

		return new Native.FesServiceManager(options.ClientLibrary, charset);
	}
	private static ValueTask<ServiceManagerBase> CreateNativeServiceManagerAsync(ConnectionString options)
	{
		var charset = GetCharset(options);

		return ValueTask.FromResult<ServiceManagerBase>(new Native.FesServiceManager(options.ClientLibrary, charset));
	}

	private static Exception UnsupportedProtocolException() => new NotSupportedException("Protocol not supported.");
	private static Exception IncorrectServerTypeException() => new NotSupportedException("Specified server type is not correct.");
	private static Exception InvalidCharsetException() => new ArgumentException("Invalid character set specified.");

	private static Charset GetCharset(ConnectionString options)
	{
		if (!Charset.TryGetByName(options.Charset, out var charset))
			throw InvalidCharsetException();
		return charset;
	}

	private static WireCryptOption FbWireCryptToWireCryptOption(FbWireCrypt wireCrypt)
	{
		return wireCrypt switch
		{
			FbWireCrypt.Disabled => WireCryptOption.Disabled,
			FbWireCrypt.Enabled => WireCryptOption.Enabled,
			FbWireCrypt.Required => WireCryptOption.Required,
			_ => throw new ArgumentOutOfRangeException(nameof(wireCrypt), $"{nameof(wireCrypt)}={wireCrypt}"),
		};
	}
}