Merge pull request #4 from cioxideru/master

ability to skip some enums which should not be in database

Author: pergerch

src/SpatialFocus.EntityFrameworkCore.Extensions/EnumLookupAttribute.cs

@@ -0,0 +1,13 @@
+// <copyright file="EnumLockupAttribute.cs" company="Spatial Focus">
+// Copyright (c) Spatial Focus. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+// </copyright>
+
+namespace SpatialFocus.EntityFrameworkCore.Extensions
+{
+	using System;
+
+	public class EnumLookupAttribute : Attribute
+	{
+	}
+}

src/SpatialFocus.EntityFrameworkCore.Extensions/EnumLookupExtension.cs

@@ -26,6 +26,11 @@ public static void ConfigureEnumLookup(this ModelBuilder modelBuilder, EnumLooku
 					continue;
 				}
 
+				if (enumOptions.UseEnumsWithAttributesOnly && !propertyType.GetCustomAttributes(typeof(EnumLookupAttribute), true).Any())
+				{
+					continue;
+				}
+
 				IMutableEntityType entityType = property.DeclaringEntityType;
 
 				Type concreteType = enumOptions.UseNumberLookup
@@ -37,8 +42,7 @@ public static void ConfigureEnumLookup(this ModelBuilder modelBuilder, EnumLooku
 				string tableName = enumOptions.NamingFunction(typeName);
 				enumLookupBuilder.ToTable(tableName);
 
-				string keyName = enumOptions.UseNumberLookup
-					? nameof(EnumWithNumberLookup<Enum>.Id)
+				string keyName = enumOptions.UseNumberLookup ? nameof(EnumWithNumberLookup<Enum>.Id)
 					: nameof(EnumWithStringLookup<Enum>.Id);
 
 				modelBuilder.Entity(entityType.Name).HasOne(concreteType).WithMany().HasPrincipalKey(keyName).HasForeignKey(property.Name);

src/SpatialFocus.EntityFrameworkCore.Extensions/EnumLookupOptions.cs

@@ -21,13 +21,16 @@ public static EnumLookupOptions Default
 				EnumLookupOptions enumOptions = new EnumLookupOptions();
 				enumOptions.SetNamingScheme(NamingScheme.SnakeCase);
 				enumOptions.UseNumberLookup = true;
+				enumOptions.UseEnumsWithAttributesOnly = false;
 
 				return enumOptions;
 			}
 		}
 
 		internal Func<string, string> NamingFunction => name => this.postProcessingTableNamingFunction(this.namingFunction(name));
 
+		internal bool UseEnumsWithAttributesOnly { get; private set; }
+
 		internal bool UseNumberLookup { get; private set; }
 
 		public EnumLookupOptions Pluralize()
@@ -51,6 +54,13 @@ public EnumLookupOptions Singularize()
 			return this;
 		}
 
+		public EnumLookupOptions UseEnumsWithAttributeOnly()
+		{
+			UseEnumsWithAttributesOnly = true;
+
+			return this;
+		}
+
 		public EnumLookupOptions UseNumberAsIdentifier()
 		{
 			UseNumberLookup = true;

src/SpatialFocus.EntityFrameworkCore.Extensions/SpatialFocus.EntityFrameworkCore.Extensions.csproj

@@ -1,7 +1,7 @@
 <Project Sdk="Microsoft.NET.Sdk">
 
 	<PropertyGroup>
-		<Version>1.0.0</Version>
+		<Version>1.1.0</Version>
 		<PackageId>SpatialFocus.EntityFrameworkCore.Extensions</PackageId>
 		<Title>Spatial Focus EntityFrameworkCore Extensions</Title>
 		<Description>A set of useful extensions for EntityFrameworkCore (Enum Lookup Tables, Naming of tables / properties / keys, Pluralize).</Description>