From 91e8d1302030e60879f4773aaf0b357994f14b23 Mon Sep 17 00:00:00 2001
From: drewgillies <drewgil@microsoft.com>
Date: Wed, 16 Dec 2020 13:15:00 +1000
Subject: [PATCH] Add vulnerabilities service

---
 src/NuGetGallery.Core/Entities/EntitiesContext.cs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/NuGetGallery.Core/Entities/EntitiesContext.cs b/src/NuGetGallery.Core/Entities/EntitiesContext.cs
index d45efb3ad5..54a28ed221 100644
--- a/src/NuGetGallery.Core/Entities/EntitiesContext.cs
+++ b/src/NuGetGallery.Core/Entities/EntitiesContext.cs
@@ -491,6 +491,13 @@ protected override void OnModelCreating(DbModelBuilder modelBuilder)
                 .HasIndex(pv => new { pv.VulnerabilityKey, pv.PackageId, pv.PackageVersionRange })
                 .IsUnique();
 
+            modelBuilder.Entity<VulnerablePackageVersionRange>()
+                .HasMany<Package>(vpvr => vpvr.Packages)
+                .WithMany(p => p.VulnerablePackageRanges)
+                .Map(c => c.ToTable("VulnerablePackageVersionRangePackages")
+                    .MapLeftKey("VulnerablePackageVersionRange_Key")
+                    .MapRightKey("Package_Key"));
+
             modelBuilder.Entity<PackageRename>()
                 .HasKey(r => r.Key)
                 .HasIndex(r => r.TransferPopularity);