Support named argument

Author: andrewlock

NetEscapades.EnumGenerators4/src/NetEscapades.EnumGenerators/EnumGenerator.cs

@@ -111,6 +111,26 @@ static List<EnumToGenerate> GetTypesToGenerate(Compilation compilation, IEnumera
             }
 
             string enumName = enumSymbol.ToString();
+            string extensionName = "EnumExtensions";
+
+            foreach (AttributeData attributeData in enumSymbol.GetAttributes())
+            {
+                if (!enumAttribute.Equals(attributeData.AttributeClass, SymbolEqualityComparer.Default))
+                {
+                    continue;
+                }
+
+                foreach (KeyValuePair<string, TypedConstant> namedArgument in attributeData.NamedArguments)
+                {
+                    if (namedArgument.Key == "ExtensionClassName"
+                        && namedArgument.Value.Value?.ToString() is { } n)
+                    {
+                        extensionName = n;
+                    }
+                }
+                break;
+            }
+
             ImmutableArray<ISymbol> enumMembers = enumSymbol.GetMembers();
             var members = new List<string>(enumMembers.Length);
 
@@ -122,7 +142,7 @@ static List<EnumToGenerate> GetTypesToGenerate(Compilation compilation, IEnumera
                 }
             }
 
-            enumsToGenerate.Add(new EnumToGenerate(enumName, members));
+            enumsToGenerate.Add(new EnumToGenerate(extensionName, enumName, members));
         }
 
         return enumsToGenerate;

NetEscapades.EnumGenerators4/src/NetEscapades.EnumGenerators/EnumToGenerate.cs

@@ -2,12 +2,14 @@
 
 public readonly struct EnumToGenerate
 {
+    public readonly string ExtensionName;
     public readonly string Name;
     public readonly List<string> Values;
 
-    public EnumToGenerate(string name, List<string> values)
+    public EnumToGenerate(string extensionName, string name, List<string> values)
     {
         Name = name;
         Values = values;
+        ExtensionName = extensionName;
     }
 }

NetEscapades.EnumGenerators4/src/NetEscapades.EnumGenerators/SourceGenerationHelper.cs

@@ -10,19 +10,20 @@ namespace NetEscapades.EnumGenerators
     [System.AttributeUsage(System.AttributeTargets.Enum)]
     public class EnumExtensionsAttribute : System.Attribute
     {
+        public string ExtensionClassName { get; set; }
     }
 }";
     public static string GenerateExtensionClass(List<EnumToGenerate> enumsToGenerate)
     {
         var sb = new StringBuilder();
         sb.Append(@"
 namespace NetEscapades.EnumGenerators
-{
-    public static partial class EnumExtensions
-    {");
+{");
         foreach(var enumToGenerate in enumsToGenerate)
         {
             sb.Append(@"
+    public static partial class ").Append(enumToGenerate.ExtensionName).Append(@"
+    {
         public static string ToStringFast(this ").Append(enumToGenerate.Name).Append(@" value)
             => value switch
             {");
@@ -38,11 +39,10 @@ public static string ToStringFast(this ").Append(enumToGenerate.Name).Append(@"
             sb.Append(@"
                 _ => value.ToString(),
             };
+    }
 ");
         }
-        sb.Append(@"
-    }
-}");
+        sb.Append('}');
 
         return sb.ToString();
     }

NetEscapades.EnumGenerators4/tests/NetEscapades.EnumGenerators.IntegrationTests/Direction.cs

@@ -0,0 +1,10 @@
+namespace NetEscapades.EnumGenerators.IntegrationTests;
+
+[EnumExtensions(ExtensionClassName = "DirectionExtensions")]
+public enum Direction
+{
+    Left,
+    Right,
+    Up,
+    Down,
+}

NetEscapades.EnumGenerators4/tests/NetEscapades.EnumGenerators.IntegrationTests/EnumExtensionsTests.cs

@@ -17,4 +17,16 @@ public void FastToStringIsSameAsToString(Colour value)
 
         Assert.Equal(expected, actual);
     }
+
+    [Theory]
+    [InlineData(Direction.Up)]
+    [InlineData((Direction)15)]
+    [InlineData((Direction)0)]
+    public void CustomExtensionNameToStringFast(Direction value)
+    {
+        var expected = value.ToString();
+        var actual = value.ToStringFast();
+
+        Assert.Equal(expected, actual);
+    }
 }

NetEscapades.EnumGenerators4/tests/NetEscapades.EnumGenerators.Tests/EnumGeneratorSnapshotTests.cs

@@ -92,6 +92,48 @@ public enum Colour
 
 [EnumExtensions]
 public enum Direction
+{
+    Left,
+    Right,
+    Up,
+    Down,
+}";
+
+        return TestHelper.Verify(source);
+    }
+
+    [Fact]
+    public Task GeneratesCustomEnumNameCorrectly()
+    {
+        var source = @"
+using NetEscapades.EnumGenerators;
+
+[EnumExtensions(ExtensionClassName = ""ColourExtensions"")]
+public enum Colour // Yes, I'm British
+{
+    Red = 0,
+    Blue = 1,
+}";
+
+        return TestHelper.Verify(source);
+    }
+
+    [Fact]
+    public Task GeneratesCustomEnumNameCorrectly_TwoAttributes()
+    {
+        var source = @"
+using NetEscapades.EnumGenerators;
+
+[EnumExtensions(ExtensionClassName = ""ColourExtensions"")]
+public enum Colour
+{
+    Red = 0,
+    Blue = 1,
+}
+
+
+[EnumExtensions(ExtensionClassName = ""DirectionExtensions"")]
+public enum Direction
 {
     Left,
     Right,

NetEscapades.EnumGenerators4/tests/NetEscapades.EnumGenerators.Tests/Snapshots/EnumGeneratorSnapshotTests.GeneratesCustomEnumNameCorrectly.00.verified.cs

@@ -0,0 +1,11 @@
+//HintName: EnumExtensionsAttribute.g.cs
+
+
+namespace NetEscapades.EnumGenerators
+{
+    [System.AttributeUsage(System.AttributeTargets.Enum)]
+    public class EnumExtensionsAttribute : System.Attribute
+    {
+        public string ExtensionClassName { get; set; }
+    }
+}

NetEscapades.EnumGenerators4/tests/NetEscapades.EnumGenerators.Tests/Snapshots/EnumGeneratorSnapshotTests.GeneratesCustomEnumNameCorrectly.01.verified.cs

@@ -0,0 +1,15 @@
+//HintName: EnumExtensions.g.cs
+
+namespace NetEscapades.EnumGenerators
+{
+    public static partial class ColourExtensions
+    {
+        public static string ToStringFast(this Colour value)
+            => value switch
+            {
+                Colour.Red => nameof(Colour.Red),
+                Colour.Blue => nameof(Colour.Blue),
+                _ => value.ToString(),
+            };
+    }
+}

NetEscapades.EnumGenerators4/tests/NetEscapades.EnumGenerators.Tests/Snapshots/EnumGeneratorSnapshotTests.GeneratesCustomEnumNameCorrectly_TwoAttributes.00.verified.cs

@@ -0,0 +1,11 @@
+//HintName: EnumExtensionsAttribute.g.cs
+
+
+namespace NetEscapades.EnumGenerators
+{
+    [System.AttributeUsage(System.AttributeTargets.Enum)]
+    public class EnumExtensionsAttribute : System.Attribute
+    {
+        public string ExtensionClassName { get; set; }
+    }
+}

NetEscapades.EnumGenerators4/tests/NetEscapades.EnumGenerators.Tests/Snapshots/EnumGeneratorSnapshotTests.GeneratesCustomEnumNameCorrectly_TwoAttributes.01.verified.cs

@@ -0,0 +1,28 @@
+//HintName: EnumExtensions.g.cs
+
+namespace NetEscapades.EnumGenerators
+{
+    public static partial class ColourExtensions
+    {
+        public static string ToStringFast(this Colour value)
+            => value switch
+            {
+                Colour.Red => nameof(Colour.Red),
+                Colour.Blue => nameof(Colour.Blue),
+                _ => value.ToString(),
+            };
+    }
+
+    public static partial class DirectionExtensions
+    {
+        public static string ToStringFast(this Direction value)
+            => value switch
+            {
+                Direction.Left => nameof(Direction.Left),
+                Direction.Right => nameof(Direction.Right),
+                Direction.Up => nameof(Direction.Up),
+                Direction.Down => nameof(Direction.Down),
+                _ => value.ToString(),
+            };
+    }
+}