Files
Umbraco-CMS/src/Umbraco.Web/Install/InstallSteps/DatabaseConfigureStep.cs

113 lines
3.8 KiB
C#
Raw Normal View History

using System;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using System.Text;
using System.Threading.Tasks;
using Umbraco.Core;
using Umbraco.Core.Persistence;
using Umbraco.Web.Install.Models;
namespace Umbraco.Web.Install.InstallSteps
{
//internal class DatabaseInstallStep : IInstallStep<object>
//{
// private readonly ApplicationContext _applicationContext;
// public DatabaseInstallStep(ApplicationContext applicationContext)
// {
// _applicationContext = applicationContext;
// }
// public IDictionary<string, object> Setup(object model)
// {
// if (CheckConnection(database) == false)
// {
// throw new InvalidOperationException("Could not connect to the database");
// }
// ConfigureConnection(database);
// return null;
// }
//}
internal class DatabaseConfigureStep : InstallSetupStep<DatabaseModel>
{
private readonly ApplicationContext _applicationContext;
public DatabaseConfigureStep(ApplicationContext applicationContext)
{
_applicationContext = applicationContext;
}
public override IDictionary<string, object> Execute(DatabaseModel database)
{
if (CheckConnection(database) == false)
{
throw new InvalidOperationException("Could not connect to the database");
}
ConfigureConnection(database);
return null;
}
private bool CheckConnection(DatabaseModel database)
{
string connectionString;
var dbContext = _applicationContext.DatabaseContext;
if (database.ConnectionString.IsNullOrWhiteSpace() == false)
{
connectionString = database.ConnectionString;
}
else if (database.DatabaseType == DatabaseType.SqlCe)
{
//we do not test this connection
return true;
//connectionString = dbContext.GetEmbeddedDatabaseConnectionString();
}
else if (database.IntegratedAuth)
{
connectionString = dbContext.GetIntegratedSecurityDatabaseConnectionString(
database.Server, database.DatabaseName);
}
else
{
string providerName;
connectionString = dbContext.GetDatabaseConnectionString(
database.Server, database.DatabaseName, database.Login, database.Password,
database.DatabaseType.ToString(),
out providerName);
}
return SqlExtensions.IsConnectionAvailable(connectionString);
}
private void ConfigureConnection(DatabaseModel database)
{
var dbContext = _applicationContext.DatabaseContext;
if (database.ConnectionString.IsNullOrWhiteSpace() == false)
{
dbContext.ConfigureDatabaseConnection(database.ConnectionString);
}
else if (database.DatabaseType == DatabaseType.SqlCe)
{
dbContext.ConfigureEmbeddedDatabaseConnection();
}
else if (database.IntegratedAuth)
{
dbContext.ConfigureIntegratedSecurityDatabaseConnection(
database.Server, database.DatabaseName);
}
else
{
dbContext.ConfigureDatabaseConnection(
database.Server, database.DatabaseName, database.Login, database.Password,
database.DatabaseType.ToString());
}
}
public override string View
{
get { return "database"; }
}
}
}