using System; using System.Linq; using Microsoft.AspNetCore.Identity.EntityFrameworkCore; using Microsoft.EntityFrameworkCore; using Microsoft.EntityFrameworkCore.Design; using Microsoft.EntityFrameworkCore.Infrastructure; namespace BTCPayServer.Data { public class DesignTimeDbContextFactory : IDesignTimeDbContextFactory { public ApplicationDbContext CreateDbContext(string[] args) { var builder = new DbContextOptionsBuilder(); builder.UseSqlite("Data Source=temp.db"); return new ApplicationDbContext(builder.Options, true); } } public class ApplicationDbContext : IdentityDbContext { private readonly bool _designTime; public ApplicationDbContext(DbContextOptions options, bool designTime = false) : base(options) { _designTime = designTime; } public DbSet Invoices { get; set; } public DbSet Refunds { get; set; } public DbSet PlannedTransactions { get; set; } public DbSet PayjoinLocks { get; set; } public DbSet Apps { get; set; } public DbSet InvoiceEvents { get; set; } public DbSet OffchainTransactions { get; set; } public DbSet HistoricalAddressInvoices { get; set; } public DbSet PendingInvoices { get; set; } public DbSet Payments { get; set; } public DbSet PaymentRequests { get; set; } public DbSet PullPayments { get; set; } public DbSet Payouts { get; set; } public DbSet Wallets { get; set; } public DbSet WalletTransactions { get; set; } public DbSet Stores { get; set; } public DbSet UserStore { get; set; } public DbSet AddressInvoices { get; set; } public DbSet Settings { get; set; } public DbSet PairingCodes { get; set; } public DbSet PairedSINData { get; set; } public DbSet ApiKeys { get; set; } public DbSet Files { get; set; } public DbSet U2FDevices { get; set; } public DbSet Notifications { get; set; } protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder) { var isConfigured = optionsBuilder.Options.Extensions.OfType().Any(); if (!isConfigured) optionsBuilder.UseSqlite("Data Source=temp.db"); } protected override void OnModelCreating(ModelBuilder builder) { base.OnModelCreating(builder); NotificationData.OnModelCreating(builder); InvoiceData.OnModelCreating(builder); PaymentData.OnModelCreating(builder); Data.UserStore.OnModelCreating(builder); APIKeyData.OnModelCreating(builder); AppData.OnModelCreating(builder); AddressInvoiceData.OnModelCreating(builder); PairingCodeData.OnModelCreating(builder); PendingInvoiceData.OnModelCreating(builder); Data.PairedSINData.OnModelCreating(builder); HistoricalAddressInvoiceData.OnModelCreating(builder); InvoiceEventData.OnModelCreating(builder); PaymentRequestData.OnModelCreating(builder); WalletTransactionData.OnModelCreating(builder); PullPaymentData.OnModelCreating(builder); PayoutData.OnModelCreating(builder); RefundData.OnModelCreating(builder); U2FDevice.OnModelCreating(builder); if (Database.IsSqlite() && !_designTime) { // SQLite does not have proper support for DateTimeOffset via Entity Framework Core, see the limitations // here: https://docs.microsoft.com/en-us/ef/core/providers/sqlite/limitations#query-limitations // To work around this, when the Sqlite database provider is used, all model properties of type DateTimeOffset // use the DateTimeOffsetToBinaryConverter // Based on: https://github.com/aspnet/EntityFrameworkCore/issues/10784#issuecomment-415769754 // This only supports millisecond precision, but should be sufficient for most use cases. foreach (var entityType in builder.Model.GetEntityTypes()) { var properties = entityType.ClrType.GetProperties().Where(p => p.PropertyType == typeof(DateTimeOffset)); foreach (var property in properties) { builder .Entity(entityType.Name) .Property(property.Name) .HasConversion(new Microsoft.EntityFrameworkCore.Storage.ValueConversion.DateTimeOffsetToBinaryConverter()); } } } } } }