Cysharp · mayuki · Jan 22, 2025 · Jan 22, 2025
diff --git a/src/MagicOnion.Client.SourceGenerator/CodeAnalysis/SerializationFormatterNameMapper.cs b/src/MagicOnion.Client.SourceGenerator/CodeAnalysis/SerializationFormatterNameMapper.cs
@@ -22,15 +22,17 @@ public interface IWellKnownSerializationTypes
 public class MessagePackFormatterNameMapper : ISerializationFormatterNameMapper
 {
     readonly string userDefinedFormatterNamespace;
+    readonly bool allowToMapUserDefinedFormatter;
 
     public IWellKnownSerializationTypes WellKnownTypes => MessagePackWellKnownSerializationTypes.Instance;
 
-    public MessagePackFormatterNameMapper(string userDefinedFormatterNamespace)
+    public MessagePackFormatterNameMapper(string userDefinedFormatterNamespace, bool allowToMapUserDefinedFormatter)
     {
         userDefinedFormatterNamespace = string.IsNullOrWhiteSpace(userDefinedFormatterNamespace) ? "MessagePack.Formatters" : userDefinedFormatterNamespace;
         if (!userDefinedFormatterNamespace.StartsWith("global::")) userDefinedFormatterNamespace = "global::" + userDefinedFormatterNamespace;
 
         this.userDefinedFormatterNamespace = userDefinedFormatterNamespace;
+        this.allowToMapUserDefinedFormatter = allowToMapUserDefinedFormatter;
     }
 
     public bool TryMapGeneric(MagicOnionTypeInfo type, [NotNullWhen(true)] out string? formatterName, [NotNullWhen(true)] out string? formatterConstructorArgs)
@@ -52,7 +54,7 @@ public bool TryMapGeneric(MagicOnionTypeInfo type, [NotNullWhen(true)] out strin
             formatterName = $"{mappedFormatterName}<{genericTypeArgs}>";
             formatterConstructorArgs = "()";
         }
-        else
+        else if (allowToMapUserDefinedFormatter)
         {
             // User-defined generic types
             formatterName = $"{userDefinedFormatterNamespace}{(string.IsNullOrWhiteSpace(userDefinedFormatterNamespace) ? "" : ".")}{type.ToDisplayName(MagicOnionTypeInfo.DisplayNameFormat.Namespace | MagicOnionTypeInfo.DisplayNameFormat.WithoutGenericArguments)}Formatter<{genericTypeArgs}>";

diff --git a/src/MagicOnion.Client.SourceGenerator/CodeGen/MagicOnionInitializerGenerator.cs b/src/MagicOnion.Client.SourceGenerator/CodeGen/MagicOnionInitializerGenerator.cs
@@ -11,9 +11,8 @@ public static string Build(GenerationContext generationContext, MagicOnionServic
 
         writer.AppendLine($$"""
             // <auto-generated />
-            #pragma warning disable CS0618 // 'member' is obsolete: 'text'
-            #pragma warning disable CS0612 // 'member' is obsolete
-            #pragma warning disable CS8019 // Unnecessary using directive.
+            #pragma warning disable
+
             """);
         if (!string.IsNullOrEmpty(generationContext.Namespace))
         {

diff --git a/...Onion.Client.SourceGenerator/CodeGen/MessagePack/MessagePackFormatterResolverGenerator.cs b/...Onion.Client.SourceGenerator/CodeGen/MessagePack/MessagePackFormatterResolverGenerator.cs
@@ -3,7 +3,7 @@
 
 namespace MagicOnion.Client.SourceGenerator.CodeGen.MessagePack;
 
-internal class MessagePackFormatterResolverGenerator(bool emitGenericFormatterInstantiationAndTypeHints) : ISerializerFormatterGenerator
+internal class MessagePackFormatterResolverGenerator() : ISerializerFormatterGenerator
 {
     public (string HintNameSuffix, string Source) Build(GenerationContext generationContext, SerializationFormatterCodeGenContext ctx)
     {
@@ -49,10 +49,7 @@ partial class {{generationContext.InitializerPartialTypeName}}
 
         EmitResolver(ctx, writer);
         EmitGetFormatterHelper(ctx, writer);
-        if (emitGenericFormatterInstantiationAndTypeHints)
-        {
-            EmitTypeHints(ctx, writer);
-        }
+        EmitTypeHints(ctx, writer);
 
         writer.AppendLineWithFormat($$"""
                 }
@@ -97,9 +94,7 @@ static FormatterCache()
 
     void EmitGetFormatterHelper(SerializationFormatterCodeGenContext ctx, StringBuilder writer)
     {
-        var formatterRegistrations = emitGenericFormatterInstantiationAndTypeHints
-            ? ctx.FormatterRegistrations
-            : ctx.FormatterRegistrations.Where(x => x is EnumSerializationInfo).ToArray();
+        var formatterRegistrations = ctx.FormatterRegistrations;
 
         writer.AppendLineWithFormat($$"""
                     static class MessagePackGeneratedGetFormatterHelper

diff --git a/src/MagicOnion.Client.SourceGenerator/CodeGen/StaticMagicOnionClientGenerator.cs b/src/MagicOnion.Client.SourceGenerator/CodeGen/StaticMagicOnionClientGenerator.cs
@@ -38,9 +38,7 @@ static void EmitHeader(GenerationContext generationContext, StringBuilder writer
     {
         writer.AppendLine("""
             // <auto-generated />
-            #pragma warning disable CS0618 // 'member' is obsolete: 'text'
-            #pragma warning disable CS0612 // 'member' is obsolete
-            #pragma warning disable CS8019 // Unnecessary using directive.
+            #pragma warning disable
 
             """);
     }

diff --git a/src/MagicOnion.Client.SourceGenerator/CodeGen/StaticStreamingHubClientGenerator.cs b/src/MagicOnion.Client.SourceGenerator/CodeGen/StaticStreamingHubClientGenerator.cs
@@ -41,12 +41,7 @@ static void EmitHeader(GenerationContext generationContext, StringBuilder writer
     {
         writer.AppendLine("""
             // <auto-generated />
-            #pragma warning disable CS0618 // 'member' is obsolete: 'text'
-            #pragma warning disable CS0612 // 'member' is obsolete
-            #pragma warning disable CS0414 // The private field 'field' is assigned but its value is never used
-            #pragma warning disable CS8019 // Unnecessary using directive.
-            #pragma warning disable CS1522 // Empty switch block
-            #pragma warning disable CS1998 // This async method lacks 'await' operators and will run synchronously.
+            #pragma warning disable
 
             """);
     }

diff --git a/src/MagicOnion.Client.SourceGenerator/GenerationContext.cs b/src/MagicOnion.Client.SourceGenerator/GenerationContext.cs
@@ -32,16 +32,15 @@ public enum SerializerType
 public record GenerationOptions(
     SerializerType Serializer,
     bool DisableAutoRegistration,
-    string MessagePackFormatterNamespace,
     bool EnableStreamingHubDiagnosticHandler,
-    string GenerateFileHintNamePrefix
-)
+    string GenerateFileHintNamePrefix,
+    IReadOnlyDictionary<string, object> AdditionalOptions)
 {
     public static GenerationOptions Default { get; } = new (
         SerializerType.MessagePack,
         DisableAutoRegistration: false,
-        MessagePackFormatterNamespace: "MessagePack.Formatters",
         EnableStreamingHubDiagnosticHandler: false,
-        GenerateFileHintNamePrefix: string.Empty
+        GenerateFileHintNamePrefix: string.Empty,
+        AdditionalOptions: new Dictionary<string, object>()
     );
 }
diff --git a/src/MagicOnion.Client.SourceGenerator/MagicOnionClientSourceGenerator.Emitter.cs b/src/MagicOnion.Client.SourceGenerator/MagicOnionClientSourceGenerator.Emitter.cs
@@ -13,6 +13,10 @@ public partial class MagicOnionClientSourceGenerator
     public const string MagicOnionClientGenerationAttributeName = $"{MagicOnionClientGenerationAttributeShortName}Attribute";
     public const string MagicOnionClientGenerationAttributeFullName = $"MagicOnion.Client.{MagicOnionClientGenerationAttributeName}";
 
+    public const string MagicOnionClientGenerationOptionAttributeShortName = "MagicOnionClientGenerationOption";
+    public const string MagicOnionClientGenerationOptionAttributeName = $"{MagicOnionClientGenerationOptionAttributeShortName}Attribute";
+    public const string MagicOnionClientGenerationOptionAttributeFullName = $"MagicOnion.Client.{MagicOnionClientGenerationOptionAttributeName}";
+
     static class Emitter
     {
         public static void Emit(GenerationContext context, ImmutableArray<INamedTypeSymbol> interfaceSymbols, ReferenceSymbols referenceSymbols)
@@ -45,8 +49,10 @@ public static void Emit(GenerationContext context, ImmutableArray<INamedTypeSymb
                     EnumFormatterGenerator: _ => string.Empty
                 ),
                 SerializerType.MessagePack => (
-                    Mapper: new MessagePackFormatterNameMapper(context.Options.MessagePackFormatterNamespace),
-                    Generator: new MessagePackFormatterResolverGenerator(emitGenericFormatterInstantiationAndTypeHints: false),
+                    Mapper: new MessagePackFormatterNameMapper(
+                        (context.Options.AdditionalOptions.TryGetValue("MessagePack.FormatterNamespace", out var formatterNamespace) ? formatterNamespace as string : null) ?? "MessagePack.Formatters",
+                        context.Options.AdditionalOptions.TryGetValue("MessagePack.GenerateResolverForCustomFormatter", out var generateTypeHintsForCustomFormatter) && generateTypeHintsForCustomFormatter.Equals(true)),
+                    Generator: new MessagePackFormatterResolverGenerator(),
                     EnumFormatterGenerator: x => MessagePackEnumFormatterGenerator.Build(context, x)
                 ),
                 _ => throw new NotImplementedException(),

diff --git a/src/MagicOnion.Client.SourceGenerator/MagicOnionClientSourceGenerator.Parser.cs b/src/MagicOnion.Client.SourceGenerator/MagicOnionClientSourceGenerator.Parser.cs
@@ -74,9 +74,13 @@ static void Traverse(INamespaceOrTypeSymbol rootNamespaceOrTypeSymbol, List<INam
         }
     }
 
-    static GenerationOptions ParseClientGenerationOptions(AttributeData attr)
+    static GenerationOptions ParseClientGenerationOptions(AttributeData attr, ClassDeclarationSyntax initializerClassDecl, SemanticModel semanticModel)
     {
-        var options = GenerationOptions.Default;
+        var initializerClassSymbol = semanticModel.GetDeclaredSymbol(initializerClassDecl) ?? throw new InvalidOperationException();
+        var additionalOptionAttrs = initializerClassSymbol.GetAttributes().Where(x => x.AttributeClass?.Name == MagicOnionClientGenerationOptionAttributeName).ToArray();
+
+        var additionalOptions = new Dictionary<string, object>();
+        var options = GenerationOptions.Default with { AdditionalOptions = additionalOptions };
 
         foreach (var namedArg in attr.NamedArguments)
         {
@@ -90,8 +94,8 @@ static GenerationOptions ParseClientGenerationOptions(AttributeData attr)
                 case nameof(GenerationOptions.Serializer):
                     options = options with { Serializer = (SerializerType)(int)namedArg.Value.Value! };
                     break;
-                case nameof(GenerationOptions.MessagePackFormatterNamespace):
-                    options = options with { MessagePackFormatterNamespace = (string)namedArg.Value.Value! };
+                case "MessagePackFormatterNamespace": // Backward compatibility
+                    additionalOptions["MessagePack.FormatterNamespace"] = (string)namedArg.Value.Value!;
                     break;
                 case nameof(GenerationOptions.EnableStreamingHubDiagnosticHandler):
                     options = options with { EnableStreamingHubDiagnosticHandler = (bool)namedArg.Value.Value! };
@@ -102,6 +106,14 @@ static GenerationOptions ParseClientGenerationOptions(AttributeData attr)
             }
         }
 
+        if (additionalOptionAttrs.Length > 0)
+        {
+            foreach (var additionalOptionAttr in additionalOptionAttrs)
+            {
+                additionalOptions[additionalOptionAttr.ConstructorArguments[0].Value!.ToString()!] = additionalOptionAttr.ConstructorArguments[1].Value!;
+            }
+        }
+
         return options;
     }
 

diff --git a/src/MagicOnion.Client.SourceGenerator/MagicOnionClientSourceGenerator.cs b/src/MagicOnion.Client.SourceGenerator/MagicOnionClientSourceGenerator.cs
@@ -25,7 +25,7 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
             var attr = attrs.FirstOrDefault(x => x.AttributeClass?.Name == MagicOnionClientGenerationAttributeName);
             if (attr is null) return; // TODO: ReportDiagnostic
 
-            var options = ParseClientGenerationOptions(attr);
+            var options = ParseClientGenerationOptions(attr, initializerClassDecl, semanticModel);
             if (!TryParseClientGenerationSpec(sourceProductionContext, semanticModel, initializerClassDecl, attr, out var spec))
             {
                 return;

diff --git a/src/MagicOnion.Client/MagicOnionClientGenerationAttribute.cs b/src/MagicOnion.Client/MagicOnionClientGenerationAttribute.cs
@@ -4,7 +4,7 @@ namespace MagicOnion.Client;
 /// Marker attribute for generating clients of MagicOnion.
 /// The source generator collects the classes specified by this attribute and uses them to generate source.
 /// </summary>
-[AttributeUsage(global::System.AttributeTargets.Class, AllowMultiple = false)]
+[AttributeUsage(AttributeTargets.Class, AllowMultiple = false)]
 public class MagicOnionClientGenerationAttribute : Attribute
 {
     /// <summary>
@@ -20,6 +20,7 @@ public class MagicOnionClientGenerationAttribute : Attribute
     /// <summary>
     /// Gets or set the namespace of pre-generated MessagePackFormatters. The default value is <c>MessagePack.Formatters</c>.
     /// </summary>
+    [Obsolete("This property is obsolete. Use [MagicOnionClientGenerationOption(\"MessagePack.FormatterNamespace\", \"MessagePack.Formatters\")] instead.")]
     public string MessagePackFormatterNamespace { get; set; } = "MessagePack.Formatters";
 
     /// <summary>
@@ -44,3 +45,25 @@ public enum GenerateSerializerType
         MemoryPack = 1,
     }
 }
+
+/// <summary>
+/// Specifies the options for generating clients of MagicOnion.
+/// </summary>
+[AttributeUsage(AttributeTargets.Class, AllowMultiple = true)]
+public class MagicOnionClientGenerationOptionAttribute : Attribute
+{
+    /// <summary>
+    /// Gets or sets the key of the option.
+    /// </summary>
+    public string Key { get; set; }
+    /// <summary>
+    /// Gets or sets the value of the option.
+    /// </summary>
+    public object? Value { get; set; }
+
+    public MagicOnionClientGenerationOptionAttribute(string key, object value)
+    {
+        Key = key;
+        Value = value;
+    }
+}
diff --git a/src/MagicOnion.Client/PublicAPI.Shipped.txt b/src/MagicOnion.Client/PublicAPI.Shipped.txt
@@ -87,11 +87,11 @@ MagicOnion.Client.MagicOnionClientGenerationAttribute.GenerateSerializerType
 MagicOnion.Client.MagicOnionClientGenerationAttribute.GenerateSerializerType.MemoryPack = 1 -> MagicOnion.Client.MagicOnionClientGenerationAttribute.GenerateSerializerType
 MagicOnion.Client.MagicOnionClientGenerationAttribute.GenerateSerializerType.MessagePack = 0 -> MagicOnion.Client.MagicOnionClientGenerationAttribute.GenerateSerializerType
 MagicOnion.Client.MagicOnionClientGenerationAttribute.MagicOnionClientGenerationAttribute(params System.Type![]! typesContainedInTargetAssembly) -> void
-MagicOnion.Client.MagicOnionClientGenerationAttribute.MessagePackFormatterNamespace.get -> string!
-MagicOnion.Client.MagicOnionClientGenerationAttribute.MessagePackFormatterNamespace.set -> void
 MagicOnion.Client.MagicOnionClientGenerationAttribute.Serializer.get -> MagicOnion.Client.MagicOnionClientGenerationAttribute.GenerateSerializerType
 MagicOnion.Client.MagicOnionClientGenerationAttribute.Serializer.set -> void
 MagicOnion.Client.MagicOnionClientGenerationAttribute.TypesContainedInTargetAssembly.get -> System.Type![]!
+MagicOnion.Client.MagicOnionClientGenerationOptionAttribute
+MagicOnion.Client.MagicOnionClientGenerationOptionAttribute(string! key, object? value) -> void
 MagicOnion.Client.MagicOnionClientOptions
 MagicOnion.Client.MagicOnionClientOptions.CallInvoker.get -> Grpc.Core.CallInvoker!
 MagicOnion.Client.MagicOnionClientOptions.CallOptions.get -> Grpc.Core.CallOptions

diff --git a/src/MagicOnion.Client/PublicAPI.Unshipped.txt b/src/MagicOnion.Client/PublicAPI.Unshipped.txt
@@ -5,5 +5,7 @@ MagicOnion.Client.IStreamingHubDiagnosticHandler.OnBroadcastEventRaw<THub>(THub
 MagicOnion.Client.IStreamingHubDiagnosticHandler.OnBroadcastEvent<THub, T>(THub hubInstance, string! methodName, T value) -> void
 MagicOnion.Client.IStreamingHubDiagnosticHandler.OnMethodInvoke<THub, TRequest, TResponse>(THub hubInstance, int methodId, string! methodName, TRequest request, bool isFireAndForget, MagicOnion.Client.IStreamingHubDiagnosticHandler.InvokeMethodDelegate<TRequest, TResponse>! invokeMethod) -> System.Threading.Tasks.ValueTask<TResponse>
 MagicOnion.Client.IStreamingHubDiagnosticHandler.OnResponseEvent<THub, T>(THub hubInstance, string! methodName, System.ReadOnlyMemory<byte> data) -> void
+MagicOnion.Client.MagicOnionClientGenerationAttribute.MessagePackFormatterNamespace.get -> string!
+MagicOnion.Client.MagicOnionClientGenerationAttribute.MessagePackFormatterNamespace.set -> void
 static MagicOnion.Client.StreamingHubClient.Connect<TStreamingHub, TReceiver>(Grpc.Core.CallInvoker! callInvoker, TReceiver receiver, string? host = null, Grpc.Core.CallOptions option = default(Grpc.Core.CallOptions), MagicOnion.Serialization.IMagicOnionSerializerProvider? serializerProvider = null, MagicOnion.Client.IStreamingHubClientFactoryProvider? factoryProvider = null, MagicOnion.Client.IMagicOnionClientLogger? logger = null) -> TStreamingHub
 static MagicOnion.Client.StreamingHubClient.Connect<TStreamingHub, TReceiver>(Grpc.Core.ChannelBase! channel, TReceiver receiver, string? host = null, Grpc.Core.CallOptions option = default(Grpc.Core.CallOptions), MagicOnion.Serialization.IMagicOnionSerializerProvider? serializerProvider = null, MagicOnion.Client.IStreamingHubClientFactoryProvider? factoryProvider = null, MagicOnion.Client.IMagicOnionClientLogger? logger = null) -> TStreamingHub