Add configurable access for generated clients (#6374)

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Builders/EnumBuilder.cs

@@ -5,13 +5,20 @@ namespace StrawberryShake.CodeGeneration.CSharp.Builders;
 
 public class EnumBuilder : ITypeBuilder
 {
+    private AccessModifier _accessModifier;
     private readonly List<(string, long?, XmlCommentBuilder?)> _elements = new();
     private string? _name;
     private string? _underlyingType;
     private XmlCommentBuilder? _xmlComment;
 
     public static EnumBuilder New() => new();
 
+    public EnumBuilder SetAccessModifier(AccessModifier value)
+    {
+        _accessModifier = value;
+        return this;
+    }
+
     public EnumBuilder SetName(string value)
     {
         _name = value;
@@ -61,14 +68,16 @@ public void Build(CodeWriter writer)
         _xmlComment?.Build(writer);
 
         writer.WriteGeneratedAttribute();
+
+        var modifier = _accessModifier.ToString().ToLowerInvariant();
 
         if (_underlyingType is null)
         {
-            writer.WriteIndentedLine($"public enum {_name}");
+            writer.WriteIndentedLine($"{modifier} enum {_name}");
         }
         else
         {
-            writer.WriteIndentedLine($"public enum {_name} : {_underlyingType}");
+            writer.WriteIndentedLine($"{modifier} enum {_name} : {_underlyingType}");
         }
 
         writer.WriteIndentedLine("{");

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Builders/InterfaceBuilder.cs

@@ -6,12 +6,19 @@ namespace StrawberryShake.CodeGeneration.CSharp.Builders;
 
 public class InterfaceBuilder : AbstractTypeBuilder
 {
+    private AccessModifier _accessModifier;
     private readonly List<MethodBuilder> _methods = new();
 
     private XmlCommentBuilder? _xmlComment;
 
     public static InterfaceBuilder New() => new();
 
+    public InterfaceBuilder SetAccessModifier(AccessModifier value)
+    {
+        _accessModifier = value;
+        return this;
+    }
+
     public new InterfaceBuilder SetName(string name)
     {
         base.SetName(name);
@@ -74,7 +81,9 @@ public override void Build(CodeWriter writer)
 
         writer.WriteIndent();
 
-        writer.Write("public partial interface ");
+        var modifier = _accessModifier.ToString().ToLowerInvariant();
+
+        writer.Write($"{modifier} partial interface ");
         writer.WriteLine(Name);
 
         if (Implements.Count > 0)

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/CSharpGenerator.cs

@@ -312,6 +312,7 @@ private static IReadOnlyList<GeneratorResult> GenerateCSharpDocuments(
         CSharpGeneratorSettings settings)
     {
         var generatorSettings = new CSharpSyntaxGeneratorSettings(
+            settings.AccessModifier,
             settings.NoStore,
             settings.InputRecords,
             settings.EntityRecords,

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/CSharpGeneratorSettings.cs

@@ -18,6 +18,11 @@ public class CSharpGeneratorSettings
     /// The root namespace of the client.
     /// </summary>
     public string Namespace { get; set; } = "StrawberryShake.GraphQL";
+
+    /// <summary>
+    /// The access modifier of the client.
+    /// </summary>
+    public AccessModifier AccessModifier { get; set; } = AccessModifier.Internal;
 
     /// <summary>
     /// Defines if a schema needs to be fully valid.

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/ClientGenerator.cs

@@ -1,3 +1,4 @@
+using System;
 using StrawberryShake.CodeGeneration.CSharp.Builders;
 using StrawberryShake.CodeGeneration.Descriptors;
 using StrawberryShake.CodeGeneration.Descriptors.Operations;
@@ -21,6 +22,7 @@ protected override void Generate(
 
         var classBuilder = ClassBuilder
             .New()
+            .SetAccessModifier(settings.AccessModifier)
             .SetName(fileName)
             .SetComment(descriptor.Documentation)
             .AddImplements(descriptor.InterfaceType.ToString());

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/ClientInterfaceGenerator.cs

@@ -20,6 +20,7 @@ protected override void Generate(ClientDescriptor descriptor,
 
         var interfaceBuilder = InterfaceBuilder
             .New()
+            .SetAccessModifier(settings.AccessModifier)
             .SetName(fileName)
             .SetComment(descriptor.Documentation);
 

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/DataTypeGenerator.cs

@@ -16,18 +16,23 @@ protected override CSharpSyntaxGeneratorResult Generate(
         DataTypeDescriptor descriptor,
         CSharpSyntaxGeneratorSettings settings)
     {
+        var modifier = settings.AccessModifier == AccessModifier.Public
+            ? SyntaxKind.PublicKeyword
+            : SyntaxKind.InternalKeyword;
+
         return descriptor.IsInterface
-            ? GenerateDataInterface(descriptor)
-            : GenerateDataClass(descriptor, settings);
+            ? GenerateDataInterface(descriptor, modifier)
+            : GenerateDataClass(descriptor, modifier, settings.EntityRecords);
     }
 
     private CSharpSyntaxGeneratorResult GenerateDataInterface(
-        DataTypeDescriptor descriptor)
+        DataTypeDescriptor descriptor,
+        SyntaxKind accessModifier)
     {
         var interfaceDeclaration =
             InterfaceDeclaration(descriptor.RuntimeType.Name)
                 .AddModifiers(
-                    Token(SyntaxKind.PublicKeyword),
+                    Token(accessModifier),
                     Token(SyntaxKind.PartialKeyword))
                 .AddGeneratedAttribute()
                 .AddSummary(descriptor.Documentation)
@@ -48,14 +53,15 @@ private CSharpSyntaxGeneratorResult GenerateDataInterface(
 
     private CSharpSyntaxGeneratorResult GenerateDataClass(
         DataTypeDescriptor descriptor,
-        CSharpSyntaxGeneratorSettings settings)
+        SyntaxKind accessModifier,
+        bool hasEntityRecords)
     {
-        if (settings.EntityRecords)
+        if (hasEntityRecords)
         {
             var recordDeclarationSyntax =
                 RecordDeclaration(Token(SyntaxKind.RecordKeyword), descriptor.RuntimeType.Name)
                     .AddModifiers(
-                        Token(SyntaxKind.PublicKeyword),
+                        Token(accessModifier),
                         Token(SyntaxKind.PartialKeyword))
                     .AddGeneratedAttribute()
                     .AddSummary(descriptor.Documentation)
@@ -96,7 +102,7 @@ private CSharpSyntaxGeneratorResult GenerateDataClass(
             var classDeclaration =
                 ClassDeclaration(descriptor.RuntimeType.Name)
                     .AddModifiers(
-                        Token(SyntaxKind.PublicKeyword),
+                        Token(accessModifier),
                         Token(SyntaxKind.PartialKeyword))
                     .AddGeneratedAttribute()
                     .AddSummary(descriptor.Documentation)

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/DependencyInjectionGenerator.cs

@@ -69,7 +69,7 @@ protected override void Generate(
         var factory = ClassBuilder
             .New(fileName)
             .SetStatic()
-            .SetAccessModifier(AccessModifier.Public);
+            .SetAccessModifier(settings.AccessModifier);
 
         var addClientMethod = factory
             .AddMethod($"Add{descriptor.Name}")

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/EntityIdFactoryGenerator.cs

@@ -33,7 +33,7 @@ protected override void Generate(EntityIdFactoryDescriptor descriptor,
 
         var classBuilder = ClassBuilder
             .New()
-            .SetAccessModifier(AccessModifier.Public)
+            .SetAccessModifier(settings.AccessModifier)
             .AddImplements(TypeNames.IEntityIdSerializer)
             .SetName(fileName);
 

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/EntityTypeGenerator.cs

@@ -60,10 +60,14 @@ protected override CSharpSyntaxGeneratorResult Generate(
                 recordDeclarationSyntax);
         }
 
+        var modifier = settings.AccessModifier == AccessModifier.Public
+            ? SyntaxKind.PublicKeyword
+            : SyntaxKind.InternalKeyword;
+
         var classDeclaration =
             ClassDeclaration(descriptor.RuntimeType.Name)
                 .AddModifiers(
-                    Token(SyntaxKind.PublicKeyword),
+                    Token(modifier),
                     Token(SyntaxKind.PartialKeyword))
                 .AddGeneratedAttribute()
                 .AddSummary(descriptor.Documentation);

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/EnumGenerator.cs

@@ -19,6 +19,7 @@ protected override void Generate(EnumTypeDescriptor descriptor,
 
         var enumBuilder = EnumBuilder
             .New()
+            .SetAccessModifier(settings.AccessModifier)
             .SetComment(descriptor.Documentation)
             .SetName(descriptor.RuntimeType.Name)
             .SetUnderlyingType(descriptor.UnderlyingType);

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/EnumParserGenerator.cs

@@ -23,6 +23,7 @@ protected override void Generate(EnumTypeDescriptor descriptor,
 
         var classBuilder = ClassBuilder
             .New(fileName)
+            .SetAccessModifier(settings.AccessModifier)
             .AddImplements(IInputValueFormatter)
             .AddImplements(ILeafValueParser.WithGeneric(String, descriptor.Name));
 

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/InputTypeGenerator.cs

@@ -16,25 +16,30 @@ protected override CSharpSyntaxGeneratorResult Generate(
     {
         var stateNamespace = $"{descriptor.RuntimeType.Namespace}.{State}";
         var infoInterfaceType = $"{stateNamespace}.{CreateInputValueInfo(descriptor.Name)}";
+
+        var modifier = settings.AccessModifier == AccessModifier.Public
+            ? SyntaxKind.PublicKeyword
+            : SyntaxKind.InternalKeyword;
 
         return new(
             descriptor.Name,
             null,
             descriptor.RuntimeType.NamespaceWithoutGlobal,
             settings.InputRecords
-                ? GenerateRecord(descriptor, infoInterfaceType)
-                : GenerateClass(descriptor, infoInterfaceType));
+                ? GenerateRecord(descriptor, modifier, infoInterfaceType)
+                : GenerateClass(descriptor, modifier, infoInterfaceType));
     }
 
     private BaseTypeDeclarationSyntax GenerateRecord(
         InputObjectTypeDescriptor descriptor,
+        SyntaxKind accessModifier,
         string infoInterfaceType)
     {
         var recordDeclaration =
             RecordDeclaration(Token(SyntaxKind.RecordKeyword), descriptor.Name.ToEscapedName())
                 .AddImplements(infoInterfaceType)
                 .AddModifiers(
-                    Token(SyntaxKind.PublicKeyword),
+                    Token(accessModifier),
                     Token(SyntaxKind.PartialKeyword))
                 .AddGeneratedAttribute()
                 .AddEquality(descriptor.Name.ToEscapedName(), descriptor.Properties, true)
@@ -55,13 +60,14 @@ private BaseTypeDeclarationSyntax GenerateRecord(
 
     private BaseTypeDeclarationSyntax GenerateClass(
         InputObjectTypeDescriptor descriptor,
+        SyntaxKind accessModifier,
         string infoInterfaceType)
     {
         var classDeclaration =
             ClassDeclaration(descriptor.Name.ToEscapedName())
                 .AddImplements(infoInterfaceType)
                 .AddModifiers(
-                    Token(SyntaxKind.PublicKeyword),
+                    Token(accessModifier),
                     Token(SyntaxKind.PartialKeyword))
                 .AddGeneratedAttribute()
                 .AddEquality(descriptor.Name.ToEscapedName(), descriptor.Properties)

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/InputValueFormatterGenerator.cs

@@ -41,6 +41,7 @@ protected override void Generate(
 
         var classBuilder = ClassBuilder
             .New()
+            .SetAccessModifier(settings.AccessModifier)
             .SetName(fileName)
             .AddImplements(TypeNames.IInputObjectFormatter);
 

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/JsonResultBuilderGenerator.cs

@@ -39,6 +39,7 @@ resultBuilderDescriptor.ResultNamedType as InterfaceTypeDescriptor ??
 
         var classBuilder = ClassBuilder
             .New()
+            .SetAccessModifier(settings.AccessModifier)
             .SetName(fileName);
 
         var constructorBuilder = classBuilder

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/NoStoreAccessorGenerator.cs

@@ -32,7 +32,7 @@ protected override void Generate(
 
         var factory = ClassBuilder
             .New(fileName)
-            .SetAccessModifier(AccessModifier.Public)
+            .SetAccessModifier(settings.AccessModifier)
             .AddImplements(TypeNames.IStoreAccessor);
 
         AddThrowNotValidWithoutStore(factory, "OperationStore", TypeNames.IOperationStore);

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/OperationDocumentGenerator.cs

@@ -33,6 +33,7 @@ protected override void Generate(OperationDescriptor descriptor,
 
         var classBuilder = ClassBuilder
             .New()
+            .SetAccessModifier(settings.AccessModifier)
             .SetName(fileName)
             .AddImplements(TypeNames.IDocument)
             .SetComment(

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/OperationServiceGenerator.cs

@@ -47,6 +47,7 @@ protected override void Generate(
 
         var classBuilder = ClassBuilder
             .New()
+            .SetAccessModifier(settings.AccessModifier)
             .SetComment(
                 XmlCommentBuilder
                     .New()

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/OperationServiceInterfaceGenerator.cs

@@ -25,6 +25,7 @@ protected override void Generate(OperationDescriptor descriptor,
 
         var interfaceBuilder = InterfaceBuilder
             .New()
+            .SetAccessModifier(settings.AccessModifier)
             .SetComment(
                 XmlCommentBuilder
                     .New()

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/RazorQueryGenerator.cs

@@ -23,12 +23,16 @@ protected override CSharpSyntaxGeneratorResult Generate(
     {
         var componentName = $"Use{descriptor.Name}";
         var resultType = descriptor.ResultTypeReference.GetRuntimeType().ToString();
+
+        var modifier = settings.AccessModifier == AccessModifier.Public
+            ? SyntaxKind.PublicKeyword
+            : SyntaxKind.InternalKeyword;
 
         var classDeclaration =
             ClassDeclaration(componentName)
                 .AddImplements(TypeNames.UseQuery.WithGeneric(resultType))
                 .AddModifiers(
-                    Token(SyntaxKind.PublicKeyword),
+                    Token(modifier),
                     Token(SyntaxKind.PartialKeyword))
                 .AddGeneratedAttribute()
                 .AddMembers(CreateOperationProperty(descriptor.RuntimeType.ToString()));

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/RazorSubscriptionGenerator.cs

@@ -24,12 +24,16 @@ protected override CSharpSyntaxGeneratorResult Generate(
     {
         var componentName = $"Use{descriptor.Name}";
         var resultType = descriptor.ResultTypeReference.GetRuntimeType().ToString();
+
+        var modifier = settings.AccessModifier == AccessModifier.Public
+            ? SyntaxKind.PublicKeyword
+            : SyntaxKind.InternalKeyword;
 
         var classDeclaration =
             ClassDeclaration(componentName)
                 .AddImplements(TypeNames.UseSubscription.WithGeneric(resultType))
                 .AddModifiers(
-                    Token(SyntaxKind.PublicKeyword),
+                    Token(modifier),
                     Token(SyntaxKind.PartialKeyword))
                 .AddGeneratedAttribute()
                 .AddMembers(CreateOperationProperty(descriptor.RuntimeType.ToString()));

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/ResultDataFactoryGenerator.cs

@@ -40,6 +40,7 @@ typeDescriptor as ComplexTypeDescriptor ??
         var classBuilder =
             ClassBuilder
                 .New()
+                .SetAccessModifier(settings.AccessModifier)
                 .SetName(fileName)
                 .AddImplements(
                     TypeNames.IOperationResultDataFactory.WithGeneric(descriptor.RuntimeType));

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/ResultFromEntityTypeMapperGenerator.cs

@@ -38,6 +38,7 @@ protected override void Generate(
 
         var classBuilder = ClassBuilder
             .New()
+            .SetAccessModifier(settings.AccessModifier)
             .AddImplements(
                 IEntityMapper.WithGeneric(
                     descriptor.ExtractType().ToString(),

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/ResultInfoGenerator.cs

@@ -41,6 +41,7 @@ typeDescriptor as ComplexTypeDescriptor ??
 
         var classBuilder = ClassBuilder
             .New()
+            .SetAccessModifier(settings.AccessModifier)
             .AddImplements(TypeNames.IOperationResultDataInfo)
             .SetName(fileName);
 

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/ResultInterfaceGenerator.cs

@@ -21,6 +21,7 @@ protected override void Generate(InterfaceTypeDescriptor descriptor,
 
         var interfaceBuilder = InterfaceBuilder
             .New()
+            .SetAccessModifier(settings.AccessModifier)
             .SetComment(descriptor.Description)
             .SetName(fileName);
 

src/StrawberryShake/CodeGeneration/src/CodeGeneration.CSharp/Generators/ResultTypeGenerator.cs

@@ -39,6 +39,7 @@ protected override void Generate(
 
         var classBuilder = ClassBuilder
             .New()
+            .SetAccessModifier(settings.AccessModifier)
             .SetComment(descriptor.Description)
             .SetName(fileName)
             .AddEquality(fileName, equalityProperties);