Merge pull request #530 from kescherCode/master

Add options to Mapster.Tool to generate files with #nullable directives

Author: andrerav

src/ExpressionTranslator/ExpressionTranslator.cs

@@ -1896,7 +1896,7 @@ public override string ToString()
                     }
 
                     // NOTE: type alias cannot solve all name conflicted case, user should use PrintFullTypeName
-                    // keep logic here for compatability
+                    // keep logic here for compatibility
                     if (_typeNames != null)
                     {
                         var names = _typeNames

src/Mapster.Tool/ExtensionOptions.cs

@@ -25,6 +25,9 @@ public class ExtensionOptions
         [Option('s', "skipExisting", Required = false, HelpText = "Set true to skip generating already existing files")]
         public bool SkipExistingFiles { get; set; }
 
+        [Option('N', "nullableDirective", Required = false, HelpText = "Set true to add \"#nullable enable\" to the top of generated extension files")]
+        public bool GenerateNullableDirective { get; set; }
+
         [Usage(ApplicationAlias = "dotnet mapster extension")]
         public static IEnumerable<Example> Examples =>
             new List<Example>

src/Mapster.Tool/MapperOptions.cs

@@ -25,6 +25,9 @@ public class MapperOptions
         [Option('s', "skipExisting", Required = false, HelpText = "Set true to skip generating already existing files")]
         public bool SkipExistingFiles { get; set; }
 
+        [Option('N', "nullableDirective", Required = false, HelpText = "Set true to add \"#nullable enable\" to the top of generated mapper files")]
+        public bool GenerateNullableDirective { get; set; }
+
         [Usage(ApplicationAlias = "dotnet mapster mapper")]
         public static IEnumerable<Example> Examples =>
             new List<Example>

src/Mapster.Tool/ModelOptions.cs

@@ -28,6 +28,9 @@ public class ModelOptions
         [Option('s', "skipExisting", Required = false, HelpText = "Set true to skip generating already existing files")]
         public bool SkipExistingFiles { get; set; }
 
+        [Option('N', "nullableDirective", Required = false, HelpText = "Set true to add \"#nullable enable\" to the top of generated model files")]
+        public bool GenerateNullableDirective { get; set; }
+
         [Usage(ApplicationAlias = "dotnet mapster model")]
         public static IEnumerable<Example> Examples =>
             new List<Example>

src/Mapster.Tool/Program.cs

@@ -133,7 +133,9 @@ private static void GenerateMappers(MapperOptions opt)
                     }
                 }
 
-                var code = translator.ToString();
+                var code = opt.GenerateNullableDirective ?
+                    $"#nullable enable{Environment.NewLine}{translator}" :
+                    translator.ToString();
                 WriteFile(code, path);
             }
         }
@@ -266,7 +268,9 @@ private static void CreateModel(ModelOptions opt, Type type, AdaptAttributeBuild
                 });
             }
 
-            var code = translator.ToString();
+            var code = opt.GenerateNullableDirective ?
+                $"#nullable enable{Environment.NewLine}{translator}" :
+                translator.ToString();
             WriteFile(code, path);
 
             static Type getPropType(MemberInfo mem)
@@ -481,7 +485,9 @@ private static void GenerateExtensions(ExtensionOptions opt)
                     GenerateExtensionMethods(mapType, config, tuple, translator, type, mapperAttr.IsHelperClass);
                 }
 
-                var code = translator.ToString();
+                var code = opt.GenerateNullableDirective ?
+                    $"#nullable enable{Environment.NewLine}{translator}" :
+                    translator.ToString();
                 WriteFile(code, path);
             }
         }