mirror of
https://github.com/bitwarden/server.git
synced 2025-04-05 05:00:19 -05:00
58 lines
1.8 KiB
C#
58 lines
1.8 KiB
C#
using Bit.Seeder.Factories;
|
|
using Bit.Seeder.Settings;
|
|
using Bit.SharedWeb.Utilities;
|
|
using Microsoft.AspNetCore.DataProtection;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.Logging;
|
|
using DatabaseContext = Bit.Infrastructure.EntityFramework.Repositories.DatabaseContext;
|
|
|
|
namespace Bit.Seeder.Commands;
|
|
|
|
public class GenerateCommand
|
|
{
|
|
public bool Execute(string name, int users, string domain)
|
|
{
|
|
// Create service provider with necessary services
|
|
var services = new ServiceCollection();
|
|
ConfigureServices(services);
|
|
var serviceProvider = services.BuildServiceProvider();
|
|
|
|
var logger = serviceProvider.GetRequiredService<ILogger<GenerateCommand>>();
|
|
|
|
var organization = OrganizationSeeder.CreateEnterprise(name, domain, users);
|
|
var user = UserSeeder.CreateUser($"admin@{domain}");
|
|
var orgUser = organization.CreateOrganizationUser(user);
|
|
|
|
using (var scope = serviceProvider.CreateScope())
|
|
{
|
|
var scopedServices = scope.ServiceProvider;
|
|
var db = scopedServices.GetRequiredService<DatabaseContext>();
|
|
|
|
db.Add(organization);
|
|
db.Add(user);
|
|
db.Add(orgUser);
|
|
|
|
db.SaveChanges();
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
private void ConfigureServices(ServiceCollection services)
|
|
{
|
|
// Load configuration using the GlobalSettingsFactory
|
|
var globalSettings = GlobalSettingsFactory.GlobalSettings;
|
|
|
|
// Register services
|
|
services.AddLogging(builder => builder.AddConsole());
|
|
services.AddSingleton(globalSettings);
|
|
|
|
// Add Data Protection services
|
|
services.AddDataProtection()
|
|
.SetApplicationName("Bitwarden");
|
|
|
|
services.AddDatabaseRepositories(globalSettings);
|
|
}
|
|
}
|