Handle different access levels like private, protected, internal

KafkaWannaFly · May 13, 2024 · b41c762 · b41c762
1 parent 7b74d67
commit b41c762
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/BindableProps/BindableProps.csproj b/BindableProps/BindableProps.csproj
@@ -13,7 +13,7 @@
         <PackageLicenseFile>LICENSE.txt</PackageLicenseFile>
         <PackageTags>source-generator;maui;net-standard;net-6;net-7;net-8;utility;helper</PackageTags>
         <PackageProjectUrl>https://github.com/KafkaWannaFly/BindableProps</PackageProjectUrl>
-        <Version>1.4.2</Version>
+        <Version>1.4.3</Version>
         <RepositoryUrl>https://github.com/KafkaWannaFly/BindableProps</RepositoryUrl>
         <RepositoryType>git</RepositoryType>
         <PackageReadmeFile>README.md</PackageReadmeFile>

diff --git a/BindablePropsSG/Generators/BaseGenerator.cs b/BindablePropsSG/Generators/BaseGenerator.cs
@@ -1,4 +1,5 @@
 using System.Collections.Immutable;
+using System.Diagnostics;
 using System.Diagnostics.CodeAnalysis;
 using System.Text;
 using BindablePropsSG.Utils;
@@ -86,6 +87,17 @@ protected virtual string ProcessClass(TypeDeclarationSyntax? typeDeclarationSynt
         var namespaceSyntax = typeDeclarationSyntax.Parent as BaseNamespaceDeclarationSyntax;
         var namespaceName = namespaceSyntax?.Name.ToString() ?? "global";
 
+        var accessLevelKeyWords = typeDeclarationSyntax?.Modifiers
+            .Where(
+                modifier => modifier.IsKind(SyntaxKind.PrivateKeyword)
+                            || modifier.IsKind(SyntaxKind.PublicKeyword)
+                            || modifier.IsKind(SyntaxKind.InternalKeyword)
+                            || modifier.IsKind(SyntaxKind.ProtectedKeyword)
+            )
+            .Select(modifier => modifier.Text) ?? ImmutableList<string>.Empty;
+        var accessLevel = string.Join(" ", accessLevelKeyWords);
+
+
         var source = new StringBuilder($@"
 #pragma warning disable
 
@@ -94,13 +106,13 @@ protected virtual string ProcessClass(TypeDeclarationSyntax? typeDeclarationSynt
 
 namespace {namespaceName}
 {{
-    public partial {typeDeclarationSyntax.Keyword.Text} {typeDeclarationSyntax.Identifier}
+    {accessLevel} partial {typeDeclarationSyntax.Keyword.Text} {typeDeclarationSyntax.Identifier}
     {{
 ");
 
         foreach (var (syntax, symbol) in syntaxSymbols)
         {
-            ProcessField(source, typeDeclarationSyntax, syntax, symbol);
+            ProcessField(source, typeDeclarationSyntax!, syntax, symbol);
         }
 
         source.Append(@$"