Merge pull request #4 from cioxideru/master

ability to skip some enums which should not be in database

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>