Enable internal classes

codecentric · Mar 20, 2024 · b12d15b · b12d15b
1 parent 7d87360
commit b12d15b
Show file tree

Hide file tree

Showing 3 changed files with 82 additions and 33 deletions.
diff --git a/AutomaticInterface/AutomaticInterface/Builder.cs b/AutomaticInterface/AutomaticInterface/Builder.cs
@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using System.Globalization;
 using System.Linq;
 using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.CSharp;
@@ -25,7 +26,9 @@ is not ClassDeclarationSyntax classSyntax
 
         var interfaceName = $"I{classSyntax.GetClassName()}";
 
-        var interfaceGenerator = new InterfaceBuilder(namespaceName, interfaceName);
+	var accessSpecifier = typeSymbol.DeclaredAccessibility.ToString().ToLower(new CultureInfo("C"));
+
+        var interfaceGenerator = new InterfaceBuilder(namespaceName, interfaceName, accessSpecifier);
 
         interfaceGenerator.AddUsings(GetUsings(typeSymbol));
         interfaceGenerator.AddClassDocumentation(GetDocumentationForClass(classSyntax));

diff --git a/AutomaticInterface/AutomaticInterface/InterfaceBuilder.cs b/AutomaticInterface/AutomaticInterface/InterfaceBuilder.cs
@@ -23,7 +23,7 @@ internal sealed record MethodInfo(
 
     public record EventInfo(string Name, string Type, string Documentation);
 
-    public class InterfaceBuilder(string nameSpaceName, string interfaceName)
+    public class InterfaceBuilder(string nameSpaceName, string interfaceName, string accessSpecifier="public")
     {
         private const string Autogenerated = """
             //--------------------------------------------------------------------------------------------------
@@ -117,7 +117,7 @@ public string Build()
             cb.AppendAndNormalizeMultipleLines(classDocumentation);
 
             cb.AppendLine($"[GeneratedCode(\"AutomaticInterface\", \"\")]");
-            cb.AppendLine($"public partial interface {interfaceName}{genericType}");
+            cb.AppendLine($"{accessSpecifier} partial interface {interfaceName}{genericType}");
             cb.AppendLine("{");
 
             cb.Indent();

diff --git a/AutomaticInterface/Tests/GeneratorTests.cs b/AutomaticInterface/Tests/GeneratorTests.cs
@@ -201,7 +201,7 @@ public void GeneratesEmptyInterface()
             namespace AutomaticInterfaceExample
             {
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                                  }
             }
@@ -244,7 +244,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                     public string Hello { get; set; }
                 }
@@ -290,7 +290,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                     private string x;
                     public string Hello { set => x = value; }
@@ -337,7 +337,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                     private string x;
                     public string Hello { get; }
@@ -385,7 +385,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                     public DirectoryInfo Hello { get; set; }
                                  }
@@ -433,7 +433,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                               
                     public string Hello(){return "";}
@@ -483,7 +483,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                    public async Task<string> Hello(){return "";}
                 }
@@ -532,7 +532,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                               
                     public string Hello(string x){return x;}
@@ -582,7 +582,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                     public string Hello(Task<string> x){return "";}
                 }
@@ -631,7 +631,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                     public string Hello(string x, int y, double z){return x;}
                 }
@@ -679,7 +679,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                     private string Hello(string x, int y, double z){return x;}
                     internal string Hello2(string x, int y, double z){return x;}
@@ -725,7 +725,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
 
                     /// <summary>
@@ -778,7 +778,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
 
                     /**
@@ -829,7 +829,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                     /// <inheritdoc />
                     public string Hello { get; private set; }
@@ -876,7 +876,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                     /// <summary>
                     /// Bla bla
@@ -927,7 +927,7 @@ namespace AutomaticInterfaceExample
                     /// Bla bla
                     /// </summary>
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                     public string Hello { get; private set; }
                 }
@@ -978,7 +978,7 @@ namespace AutomaticInterfaceExample
                     /// Bla bla
                     /// </summary>
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                     DemoClass(string x)
                     {
@@ -1034,7 +1034,7 @@ namespace AutomaticInterfaceExample
                     /// Bla bla
                     /// </summary>
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                     public static string Hello => "abc"; // property
 
@@ -1087,7 +1087,7 @@ namespace AutomaticInterfaceExample
                     /// Bla bla
                     /// </summary>
                 [GenerateAutomaticInterface]
-                class DemoClass<T,U> where T:class
+                public class DemoClass<T,U> where T:class
                 {
                 }
             }
@@ -1135,7 +1135,7 @@ namespace AutomaticInterfaceExample
                     /// Bla bla
                     /// </summary>
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
 
                     /// <summary>
@@ -1194,7 +1194,7 @@ namespace AutomaticInterfaceExample
                     /// Bla bla
                     /// </summary>
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
 
                     private int[] arr = new int[100];
@@ -1254,7 +1254,7 @@ namespace AutomaticInterfaceExample
                     /// Bla bla
                     /// </summary>
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                     /// <summary>
                     /// Property Documentation will be copied
@@ -1353,7 +1353,7 @@ public void WorksWithFileScopedNamespace()
             namespace AutomaticInterfaceExample;
 
             [GenerateAutomaticInterface]
-            class DemoClass
+            public class DemoClass
             {
                 public string Hello { get; set; }
             }
@@ -1452,7 +1452,7 @@ namespace AutomaticInterfaceExample
             {
 
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
                     public string AMethod(string x, string y)
                     {
@@ -1606,7 +1606,7 @@ namespace AutomaticInterfaceExample
                     /// Bla bla
                     /// </summary>
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
 
                     /// <summary>
@@ -1665,7 +1665,7 @@ namespace AutomaticInterfaceExample
                     /// Bla bla
                     /// </summary>
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
 
                     /// <summary>
@@ -1724,7 +1724,7 @@ namespace AutomaticInterfaceExample
                     /// Bla bla
                     /// </summary>
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
 
                     /// <summary>
@@ -1784,7 +1784,7 @@ namespace AutomaticInterfaceExample
                     /// Bla bla
                     /// </summary>
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
 
                     /// <summary>
@@ -1844,7 +1844,7 @@ namespace AutomaticInterfaceExample
                     /// Bla bla
                     /// </summary>
                 [GenerateAutomaticInterface]
-                class DemoClass
+                public class DemoClass
                 {
 
                    public async Task<Stream?> GetFinalDocumentsByIDFails(
@@ -1916,7 +1916,7 @@ public GenerateAutomaticInterfaceAttribute(string namespaceName = "") { }
                 /// Bla bla
                 /// </summary>
                 [GenerateAutomaticInterface("CustomNameSpace")]
-                class DemoClass
+                public class DemoClass
                 {
 
                    public async Task<Stream?> GetFinalDocumentsByIDFails(
@@ -2211,4 +2211,50 @@ public partial interface ISecondClass
             """;
         GenerateCode(code).Should().Be(expected);
     }
+
+    [Fact]
+    public void WorksWithInternal()
+    {
+        const string code = """
+
+            using AutomaticInterfaceAttribute;
+            using System.Threading.Tasks;
+
+            namespace AutomaticInterfaceExample;
+
+            [GenerateAutomaticInterface]
+            internal class InternalClass
+            {
+                public int AProperty { get; set; }
+            }
+
+            """;
+
+        const string expected = """
+            //--------------------------------------------------------------------------------------------------
+            // <auto-generated>
+            //     This code was generated by a tool.
+            //
+            //     Changes to this file may cause incorrect behavior and will be lost if the code is regenerated.
+            // </auto-generated>
+            //--------------------------------------------------------------------------------------------------
+
+            using System.CodeDom.Compiler;
+            using AutomaticInterfaceAttribute;
+            using System.Threading.Tasks;
+
+            namespace AutomaticInterfaceExample
+            {
+                [GeneratedCode("AutomaticInterface", "")]
+                internal partial interface IInternalClass
+                {
+                    /// <inheritdoc />
+                    int AProperty { get; set; }
+                    
+                }
+            }
+
+            """;
+        GenerateCode(code).Should().Be(expected);
+    }
 }