littleshop/LittleShop/Data/LittleShopContext.cs

using Microsoft.EntityFrameworkCore;
using LittleShop.Models;

namespace LittleShop.Data;

public class LittleShopContext : DbContext
{
    public LittleShopContext(DbContextOptions<LittleShopContext> options) : base(options)
    {
    }

    public DbSet<User> Users { get; set; }
    public DbSet<Category> Categories { get; set; }
    public DbSet<Product> Products { get; set; }
    public DbSet<ProductPhoto> ProductPhotos { get; set; }
    public DbSet<Order> Orders { get; set; }
    public DbSet<OrderItem> OrderItems { get; set; }
    public DbSet<CryptoPayment> CryptoPayments { get; set; }
    public DbSet<ShippingRate> ShippingRates { get; set; }

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        base.OnModelCreating(modelBuilder);

        // User entity
        modelBuilder.Entity<User>(entity =>
        {
            entity.HasIndex(e => e.Username).IsUnique();
        });

        // Category entity
        modelBuilder.Entity<Category>(entity =>
        {
            entity.HasMany(c => c.Products)
                  .WithOne(p => p.Category)
                  .HasForeignKey(p => p.CategoryId)
                  .OnDelete(DeleteBehavior.Restrict);
        });

        // Product entity
        modelBuilder.Entity<Product>(entity =>
        {
            entity.HasMany(p => p.Photos)
                  .WithOne(pp => pp.Product)
                  .HasForeignKey(pp => pp.ProductId)
                  .OnDelete(DeleteBehavior.Cascade);

            entity.HasMany(p => p.OrderItems)
                  .WithOne(oi => oi.Product)
                  .HasForeignKey(oi => oi.ProductId)
                  .OnDelete(DeleteBehavior.Restrict);
        });

        // Order entity
        modelBuilder.Entity<Order>(entity =>
        {
            entity.HasMany(o => o.Items)
                  .WithOne(oi => oi.Order)
                  .HasForeignKey(oi => oi.OrderId)
                  .OnDelete(DeleteBehavior.Cascade);

            entity.HasMany(o => o.Payments)
                  .WithOne(cp => cp.Order)
                  .HasForeignKey(cp => cp.OrderId)
                  .OnDelete(DeleteBehavior.Cascade);

            entity.HasIndex(e => e.IdentityReference);
        });

        // OrderItem entity
        modelBuilder.Entity<OrderItem>(entity =>
        {
            entity.HasKey(oi => oi.Id);
        });

        // CryptoPayment entity
        modelBuilder.Entity<CryptoPayment>(entity =>
        {
            entity.HasIndex(e => e.BTCPayInvoiceId);
            entity.HasIndex(e => e.WalletAddress);
        });
    }
}