1
0
mirror of https://github.com/bitwarden/server.git synced 2024-12-05 14:23:30 +01:00
bitwarden-server/util/SqliteMigrations/Factories.cs

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

50 lines
1.7 KiB
C#
Raw Normal View History

using Bit.Core.Settings;
using Bit.Infrastructure.EntityFramework.Repositories;
using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Design;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
namespace Bit.SqliteMigrations;
public class GlobalSettingsFactory
{
public GlobalSettings GlobalSettings { get; }
public GlobalSettingsFactory(string[] args)
{
GlobalSettings = new GlobalSettings();
// UserSecretsId here should match what is in Api.csproj
var config = new ConfigurationBuilder()
.AddUserSecrets("bitwarden-Api")
.AddCommandLine(args)
.Build();
config.GetSection("GlobalSettings").Bind(GlobalSettings);
}
}
public class DatabaseContextFactory : IDesignTimeDbContextFactory<DatabaseContext>
{
public DatabaseContext CreateDbContext(string[] args)
{
var services = new ServiceCollection();
services.AddDataProtection();
var serviceProvider = services.BuildServiceProvider();
var globalSettings = new GlobalSettingsFactory(args)
.GlobalSettings;
var optionsBuilder = new DbContextOptionsBuilder<DatabaseContext>();
var connectionString = globalSettings.Sqlite?.ConnectionString ?? "Data Source=:memory:";
if (string.IsNullOrWhiteSpace(connectionString))
{
throw new Exception("No Sqlite connection string found.");
}
optionsBuilder.UseSqlite(
connectionString,
b => b.MigrationsAssembly("SqliteMigrations"))
.UseApplicationServiceProvider(serviceProvider);
return new DatabaseContext(optionsBuilder.Options);
}
}