Add BracingStyle and IndentString to options (#293)

Author: sungam3r

src/PublicApiGenerator/ApiGenerator.cs

@@ -26,8 +26,6 @@ public static string GeneratePublicApi(this Assembly assembly, ApiGeneratorOptio
     {
         options ??= new ApiGeneratorOptions();
 
-        var attributeFilter = new AttributeFilter(options.ExcludeAttributes);
-
         using (var assemblyResolver = new DefaultAssemblyResolver())
         {
             var assemblyPath = assembly.Location;
@@ -43,14 +41,11 @@ public static string GeneratePublicApi(this Assembly assembly, ApiGeneratorOptio
             {
                 return CreatePublicApiForAssembly(
                     asm,
+                    options,
                     typeDefinition => !typeDefinition.IsNested &&
                                       ShouldIncludeType(typeDefinition, options.DenyNamespacePrefixes, options.AllowNamespacePrefixes, options.UseDenyNamespacePrefixesForExtensionMethods) &&
-                                      (options.IncludeTypes == null || options.IncludeTypes.Any(type => type.FullName == typeDefinition.FullName && type.Assembly.FullName == typeDefinition.Module.Assembly.FullName)),
-                    options.IncludeAssemblyAttributes,
-                    options.DenyNamespacePrefixes,
-                    options.AllowNamespacePrefixes,
-                    options.UseDenyNamespacePrefixesForExtensionMethods,
-                    attributeFilter);
+                                      (options.IncludeTypes == null || options.IncludeTypes.Any(type => type.FullName == typeDefinition.FullName && type.Assembly.FullName == typeDefinition.Module.Assembly.FullName))
+                );
             }
         }
     }
@@ -86,12 +81,14 @@ public static string GeneratePublicApi(this Type type, ApiGeneratorOptions? opti
 
     // TODO: Assembly references?
     // TODO: Better handle namespaces - using statements? - requires non-qualified type names
-    private static string CreatePublicApiForAssembly(AssemblyDefinition assembly, Func<TypeDefinition, bool> shouldIncludeType, bool shouldIncludeAssemblyAttributes, string[] denyNamespacePrefixes, string[] allowNamespacePrefixes, bool useDenyNamespacePrefixesForExtensionMethods, AttributeFilter attributeFilter)
+    private static string CreatePublicApiForAssembly(AssemblyDefinition assembly, ApiGeneratorOptions options, Func<TypeDefinition, bool> shouldIncludeType)
     {
+        var attributeFilter = new AttributeFilter(options.ExcludeAttributes);
+
         using var provider = new CSharpCodeProvider();
 
         var compileUnit = new CodeCompileUnit();
-        if (shouldIncludeAssemblyAttributes && assembly.HasCustomAttributes)
+        if (options.IncludeAssemblyAttributes && assembly.HasCustomAttributes)
         {
             PopulateCustomAttributes(assembly, compileUnit.AssemblyCustomAttributes, attributeFilter);
         }
@@ -110,7 +107,7 @@ private static string CreatePublicApiForAssembly(AssemblyDefinition assembly, Fu
 
             using (NullableContext.Push(publicType))
             {
-                var typeDeclaration = CreateTypeDeclaration(publicType, denyNamespacePrefixes, allowNamespacePrefixes, useDenyNamespacePrefixesForExtensionMethods, attributeFilter);
+                var typeDeclaration = CreateTypeDeclaration(publicType, options.DenyNamespacePrefixes, options.AllowNamespacePrefixes, options.UseDenyNamespacePrefixesForExtensionMethods, attributeFilter);
                 @namespace.Types.Add(typeDeclaration);
             }
         }
@@ -119,10 +116,10 @@ private static string CreatePublicApiForAssembly(AssemblyDefinition assembly, Fu
         {
             var cgo = new CodeGeneratorOptions
             {
-                BracingStyle = "C",
+                BracingStyle = options.BracingStyle,
                 BlankLinesBetweenMembers = false,
                 VerbatimOrder = false,
-                IndentString = "    "
+                IndentString = options.IndentString
             };
 
             provider.GenerateCodeFromCompileUnit(compileUnit, writer, cgo);

src/PublicApiGenerator/ApiGeneratorOptions.cs

@@ -71,4 +71,14 @@ public class ApiGeneratorOptions
     private static readonly string[] _defaultAllowNamespacePrefixes = [];
 
     private static readonly string[] _defaultDenyNamespacePrefixes = ["System", "Microsoft"];
+
+    /// <summary>
+    /// Indentation string. Defaults to 4 whitespace.
+    /// </summary>
+    public string IndentString { get; set; } = "    ";
+
+    /// <summary>
+    /// Style for braces. Available values: C, Block. Defaults to C.
+    /// </summary>
+    public string BracingStyle { get; set; } = "C";
 }

src/PublicApiGeneratorTests/ApiGeneratorTestsBase.cs

@@ -8,34 +8,47 @@ public abstract class ApiGeneratorTestsBase
 {
     private static readonly Regex StripEmptyLines = new(@"^\s+$[\r\n]*", RegexOptions.Multiline | RegexOptions.Compiled);
 
-    protected void AssertPublicApi<T>(string expectedOutput, ApiGeneratorOptions? options = null)
+    protected void AssertPublicApi<T>(string expectedOutput) => AssertPublicApi<T>(expectedOutput, _ => { });
+
+    protected void AssertPublicApi<T>(string expectedOutput, Action<ApiGeneratorOptions> configure)
     {
-        AssertPublicApi(typeof(T), expectedOutput, options);
+        AssertPublicApi(typeof(T), expectedOutput, configure);
     }
 
-    protected void AssertPublicApi(Type type, string expectedOutput, ApiGeneratorOptions? options = null)
+    protected void AssertPublicApi(Type type, string expectedOutput)
     {
-        AssertPublicApi([type], expectedOutput, options);
+        AssertPublicApi(type, expectedOutput, _ => { });
     }
 
-    protected void AssertPublicApi(Type[] types, string expectedOutput, ApiGeneratorOptions? options = null)
+    protected void AssertPublicApi(Type type, string expectedOutput, Action<ApiGeneratorOptions> configure)
     {
-        options ??= new DefaultApiGeneratorOptions();
-        options.IncludeTypes = types;
+        AssertPublicApi([type], expectedOutput, configure);
+    }
 
-        AssertPublicApi(types[0].Assembly, expectedOutput, options);
+    protected void AssertPublicApi(Type[] types, string expectedOutput)
+    {
+        AssertPublicApi(types, expectedOutput, _ => { });
     }
 
-    private static void AssertPublicApi(Assembly assembly, string expectedOutput, ApiGeneratorOptions? options = null)
+    protected void AssertPublicApi(Type[] types, string expectedOutput, Action<ApiGeneratorOptions> configure)
     {
-        options ??= new DefaultApiGeneratorOptions();
+        var options = new ApiGeneratorOptions
+        {
+            IncludeAssemblyAttributes = false,
+            IncludeTypes = types
+        };
+        configure(options);
 
+        AssertPublicApi(types[0].Assembly, expectedOutput, options);
+    }
+
+    private static void AssertPublicApi(Assembly assembly, string expectedOutput, ApiGeneratorOptions options)
+    {
         string actualOutput = assembly.GeneratePublicApi(options);
         actualOutput = StripEmptyLines.Replace(actualOutput, string.Empty);
         try
         {
-            Assert.Equal(expectedOutput, actualOutput, ignoreCase: false, ignoreLineEndingDifferences: true,
-                ignoreWhiteSpaceDifferences: true);
+            Assert.Equal(expectedOutput, actualOutput, ignoreCase: false, ignoreLineEndingDifferences: true, ignoreWhiteSpaceDifferences: true);
         }
         catch
         {

src/PublicApiGeneratorTests/Assembly_attributes.cs

@@ -1,5 +1,4 @@
 using System.Runtime.InteropServices;
-using PublicApiGenerator;
 using PublicApiGeneratorTests.Examples;
 
 [assembly: Guid("3B8D506A-5247-47FF-B053-D29A51A97C33")]
@@ -54,7 +53,7 @@ public NotImportant() {{ }}
     }}
 }}";
 
-            AssertPublicApi<NotImportant>(api, new ApiGeneratorOptions { IncludeAssemblyAttributes = true });
+            AssertPublicApi<NotImportant>(api, opt => opt.IncludeAssemblyAttributes = true);
         }
     }
 

src/PublicApiGeneratorTests/Class_attributes.cs

@@ -290,11 +290,6 @@ public ClassWithInternalAttribute() { }
         [Fact]
         public void Should_skip_excluded_attribute()
         {
-            var options = new DefaultApiGeneratorOptions
-            {
-                ExcludeAttributes = ["PublicApiGeneratorTests.Examples.Attribute_MM"]
-            };
-
             AssertPublicApi<ClassWithMultipleAttributes>(
                 @"namespace PublicApiGeneratorTests.Examples
 {
@@ -304,7 +299,7 @@ public class ClassWithMultipleAttributes
     {
         public ClassWithMultipleAttributes() { }
     }
-}", options);
+}", opt => opt.ExcludeAttributes = ["PublicApiGeneratorTests.Examples.Attribute_MM"]);
         }
 
         [Fact]

src/PublicApiGeneratorTests/Class_event_attributes.cs

@@ -22,11 +22,6 @@ public ClassWithEventWithAttribute() { }
         [Fact]
         public void Should_skip_excluded_attribute()
         {
-            var options = new DefaultApiGeneratorOptions
-            {
-                ExcludeAttributes = ["PublicApiGeneratorTests.Examples.SimpleAttribute"]
-            };
-
             AssertPublicApi<ClassWithEventWithAttribute>(
                 @"namespace PublicApiGeneratorTests.Examples
 {
@@ -35,15 +30,15 @@ public class ClassWithEventWithAttribute
         public ClassWithEventWithAttribute() { }
         public event System.EventHandler OnClicked;
     }
-}", options);
+}", opt => opt.ExcludeAttributes = ["PublicApiGeneratorTests.Examples.SimpleAttribute"]);
         }
     }
 
     namespace Examples
     {
         public class ClassWithEventWithAttribute
         {
-            [SimpleAttribute]
+            [Simple]
             public event EventHandler OnClicked;
         }
     }

src/PublicApiGeneratorTests/DefaultApiGeneratorOptions.cs

@@ -1,4 +1,4 @@
-using PublicApiGeneratorTests.Examples;
+using PublicApiGeneratorTests.Examples;
 
 namespace PublicApiGeneratorTests
 {
@@ -214,18 +214,13 @@ public void Should_output_attribute_for_parameter()
         [Fact]
         public void Should_skip_excluded_attribute()
         {
-            var options = new DefaultApiGeneratorOptions
-            {
-                ExcludeAttributes = ["PublicApiGeneratorTests.Examples.Attribute_ZZ"]
-            };
-
             AssertPublicApi<DelegateWithMultipleAttributes>(
                 @"namespace PublicApiGeneratorTests.Examples
 {
     [PublicApiGeneratorTests.Examples.Attribute_AA]
     [PublicApiGeneratorTests.Examples.Attribute_MM]
     public delegate void DelegateWithMultipleAttributes();
-}", options);
+}", opt => opt.ExcludeAttributes = ["PublicApiGeneratorTests.Examples.Attribute_ZZ"]);
         }
     }
 

src/PublicApiGeneratorTests/Delegate_attributes.cs

@@ -160,11 +160,6 @@ public FieldWithMultipleAttributes() { }
         [Fact]
         public void Should_skip_excluded_attributes()
         {
-            var options = new DefaultApiGeneratorOptions
-            {
-                ExcludeAttributes = ["PublicApiGeneratorTests.Examples.Attribute_MM", "PublicApiGeneratorTests.Examples.Attribute_ZZ"]
-            };
-
             AssertPublicApi<FieldWithMultipleAttributes>(
                 @"namespace PublicApiGeneratorTests.Examples
 {
@@ -174,7 +169,7 @@ public class FieldWithMultipleAttributes
         public string Value;
         public FieldWithMultipleAttributes() { }
     }
-}", options);
+}", opt => opt.ExcludeAttributes = ["PublicApiGeneratorTests.Examples.Attribute_MM", "PublicApiGeneratorTests.Examples.Attribute_ZZ"]);
         }
     }
 

src/PublicApiGeneratorTests/Field_attributes.cs

@@ -153,16 +153,11 @@ public interface IInterfaceWithAttributeWithStringArrayInitialiser { }
         [Fact]
         public void Should_skip_excluded_attribute()
         {
-            var options = new DefaultApiGeneratorOptions
-            {
-                ExcludeAttributes = ["PublicApiGeneratorTests.Examples.SimpleAttribute"]
-            };
-
             AssertPublicApi<IInterfaceWithSimpleAttribute>(
                 @"namespace PublicApiGeneratorTests.Examples
 {
     public interface IInterfaceWithSimpleAttribute { }
-}", options);
+}", opt => opt.ExcludeAttributes = ["PublicApiGeneratorTests.Examples.SimpleAttribute"]);
         }
     }