Refactored multitenant stuff out further

-Remaining setup moved to extension methods.

src/Finbuckle.MultiTenant.EntityFrameworkCore/Extensions/IMultiTenantDbContextExtensions.cs

@@ -0,0 +1,175 @@
+// Copyright 2019 Andrew White
+// 
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// 
+//     http://www.apache.org/licenses/LICENSE-2.0
+// 
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+using System.Linq;
+using Microsoft.EntityFrameworkCore;
+
+namespace Finbuckle.MultiTenant.EntityFrameworkCore
+{
+    static class IMultiTenantDbContextExtensions
+    {
+        /// <summary>
+        /// Checks the TenantId on entities taking into account
+        /// TenantNotSetMode and TenantMismatchMode.
+        /// </summary>
+        public static void EnforceMultiTenant(this IMultiTenantDbContext context)
+        {
+            var changeTracker = context.Context.ChangeTracker;
+            var tenantInfo = context.TenantInfo;
+            var tenantMismatchMode = context.TenantMismatchMode;
+            var tenantNotSetMode = context.TenantNotSetMode;
+
+            var changedMultiTenantEntities = changeTracker.Entries().
+                Where(e => e.State == EntityState.Added || e.State == EntityState.Modified || e.State == EntityState.Deleted).
+                Where(e => Shared.HasMultiTenantAnnotation(e.Metadata));
+
+            // ensure tenant context is valid
+            if (changedMultiTenantEntities.Any())
+            {
+                if (tenantInfo == null)
+                    throw new MultiTenantException("MultiTenant Entity cannot be changed if TenantInfo is null.");
+            }
+
+            // get list of all added entities with MultiTenant annotation
+            var addedMultiTenantEntities = changedMultiTenantEntities.
+                Where(e => e.State == EntityState.Added);
+
+            // handle Tenant Id mismatches for added entities
+            var mismatchedAdded = addedMultiTenantEntities.
+                Where(e => (string)e.Property("TenantId").CurrentValue != null &&
+                (string)e.Property("TenantId").CurrentValue != tenantInfo.Id);
+
+            if (mismatchedAdded.Any())
+            {
+                switch (tenantMismatchMode)
+                {
+                    case TenantMismatchMode.Throw:
+                        throw new MultiTenantException($"{mismatchedAdded.Count()} added entities with Tenant Id mismatch."); ;
+
+                    case TenantMismatchMode.Ignore:
+                        // no action needed
+                        break;
+
+                    case TenantMismatchMode.Overwrite:
+                        foreach (var e in mismatchedAdded)
+                        {
+                            e.Property("TenantId").CurrentValue = tenantInfo.Id;
+                        }
+                        break;
+                }
+            }
+
+            // for added entities TenantNotSetMode is always Overwrite
+            var notSetAdded = addedMultiTenantEntities.
+                Where(e => (string)e.Property("TenantId").CurrentValue == null);
+
+            foreach (var e in notSetAdded)
+            {
+                e.Property("TenantId").CurrentValue = tenantInfo.Id;
+            }
+
+            // get list of all modified entities with MultiTenant annotation
+            var modifiedMultiTenantEntities = changedMultiTenantEntities.
+                Where(e => e.State == EntityState.Modified);
+
+            // handle Tenant Id mismatches for modified entities
+            var mismatchedModified = modifiedMultiTenantEntities.
+                Where(e => (string)e.Property("TenantId").CurrentValue != null &&
+                (string)e.Property("TenantId").CurrentValue != tenantInfo.Id);
+
+            if (mismatchedModified.Any())
+            {
+                switch (tenantMismatchMode)
+                {
+                    case TenantMismatchMode.Throw:
+                        throw new MultiTenantException($"{mismatchedModified.Count()} modified entities with Tenant Id mismatch."); ;
+
+                    case TenantMismatchMode.Ignore:
+                        // no action needed
+                        break;
+
+                    case TenantMismatchMode.Overwrite:
+                        foreach (var e in mismatchedModified)
+                        {
+                            e.Property("TenantId").CurrentValue = tenantInfo.Id;
+                        }
+                        break;
+                }
+            }
+
+            // handle Tenant Id not set for modified entities
+            var notSetModified = modifiedMultiTenantEntities.
+                Where(e => (string)e.Property("TenantId").CurrentValue == null);
+
+            if (notSetModified.Any())
+            {
+                switch (tenantNotSetMode)
+                {
+                    case TenantNotSetMode.Throw:
+                        throw new MultiTenantException($"{notSetModified.Count()} modified entities with Tenant Id not set."); ;
+
+                    case TenantNotSetMode.Overwrite:
+                        foreach (var e in notSetModified)
+                        {
+                            e.Property("TenantId").CurrentValue = tenantInfo.Id;
+                        }
+                        break;
+                }
+            }
+
+            // get list of all deleted  entities with MultiTenant annotation
+            var deletedMultiTenantEntities = changedMultiTenantEntities.
+                Where(e => e.State == EntityState.Deleted);
+
+            // handle Tenant Id mismatches for deleted entities
+            var mismatchedDeleted = deletedMultiTenantEntities.
+                Where(e => (string)e.Property("TenantId").CurrentValue != null &&
+                (string)e.Property("TenantId").CurrentValue != tenantInfo.Id);
+
+            if (mismatchedDeleted.Any())
+            {
+                switch (tenantMismatchMode)
+                {
+                    case TenantMismatchMode.Throw:
+                        throw new MultiTenantException($"{mismatchedDeleted.Count()} deleted entities with Tenant Id mismatch."); ;
+
+                    case TenantMismatchMode.Ignore:
+                        // no action needed
+                        break;
+
+                    case TenantMismatchMode.Overwrite:
+                        // no action needed
+                        break;
+                }
+            }
+
+            // handle Tenant Id not set for deleted entities
+            var notSetDeleted = deletedMultiTenantEntities.
+                Where(e => (string)e.Property("TenantId").CurrentValue == null);
+
+            if (notSetDeleted.Any())
+            {
+                switch (tenantNotSetMode)
+                {
+                    case TenantNotSetMode.Throw:
+                        throw new MultiTenantException($"{notSetDeleted.Count()} deleted entities with Tenant Id not set."); ;
+
+                    case TenantNotSetMode.Overwrite:
+                        // no action needed
+                        break;
+                }
+            }
+        }
+    }
+}

src/Finbuckle.MultiTenant.EntityFrameworkCore/Extensions/ModelBuilderExtensions.cs

@@ -0,0 +1,36 @@
+// Copyright 2019 Andrew White
+// 
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// 
+//     http://www.apache.org/licenses/LICENSE-2.0
+// 
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+using System.Linq;
+using Microsoft.EntityFrameworkCore;
+
+namespace Finbuckle.MultiTenant.EntityFrameworkCore
+{
+    static class FinbuckleModelBuilderExtensions
+    {
+        /// <summary>
+        /// Configures any entity's with the [MultiTenant] attribute.
+        /// </summary>
+        public static ModelBuilder SetupMultiTenant(this ModelBuilder modelBuilder)
+        {
+            // Annotate the types marked with the MultiTenant Data Attribute
+            foreach (var t in modelBuilder.Model.GetEntityTypes().Where(t => Shared.HasMultiTenantAttribute(t.ClrType)))
+            {
+                modelBuilder.Entity(t.ClrType).IsMultiTenant();
+            }
+
+            return modelBuilder;
+        }
+    }
+}

src/Finbuckle.MultiTenant.EntityFrameworkCore/IMultiTenantDbContext.cs

@@ -12,10 +12,18 @@
 //    See the License for the specific language governing permissions and
 //    limitations under the License.
 
-namespace Finbuckle.MultiTenant
+using System.Collections.Immutable;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Metadata;
+
+namespace Finbuckle.MultiTenant.EntityFrameworkCore
 {
-    internal interface IMultiTenantDbContext
+    public interface IMultiTenantDbContext
     {
         TenantInfo TenantInfo { get; }
+        TenantMismatchMode TenantMismatchMode { get; set; }
+        TenantNotSetMode TenantNotSetMode { get; set; }
+        IImmutableList<IEntityType> MultiTenantEntityTypes { get; }
+        DbContext Context { get; }
     }
 }