init only & record type

Author: chaowlert

ExpressionDebugger.Tests/DebugInfoInjectorTest.cs

@@ -736,6 +736,12 @@ public void TestProperties()
                 Type = typeof(string),
                 IsReadOnly = true
             });
+            translator.Properties.Add(new PropertyDefinitions
+            {
+                Name = "Prop3",
+                Type = typeof(string),
+                IsInitOnly = true
+            });
             var str = translator.ToString();
             Assert.AreEqual(@"
 namespace ExpressionDebugger.Tests
@@ -744,6 +750,7 @@ public partial class MockClass
     {
         public string Prop1 { get; set; }
         public string Prop2 { get; }
+        public string Prop3 { get; init; }
         
         public MockClass(string prop2)
         {
@@ -753,6 +760,46 @@ public MockClass(string prop2)
 }".Trim(), str);
         }
 
+        
+        [TestMethod]
+        public void TestRecordType()
+        {
+            var translator = new ExpressionTranslator(new TypeDefinitions
+            {
+                IsStatic = false,
+                Namespace = "ExpressionDebugger.Tests",
+                TypeName = "MockClass",
+                IsRecordType = true,
+            });
+            translator.Properties.Add(new PropertyDefinitions
+            {
+                Name = "Prop1",
+                Type = typeof(string)
+            });
+            translator.Properties.Add(new PropertyDefinitions
+            {
+                Name = "Prop2",
+                Type = typeof(string),
+                IsReadOnly = true
+            });
+            translator.Properties.Add(new PropertyDefinitions
+            {
+                Name = "Prop3",
+                Type = typeof(string),
+                IsInitOnly = true
+            });
+            var str = translator.ToString();
+            Assert.AreEqual(@"
+namespace ExpressionDebugger.Tests
+{
+    public partial record MockClass(string prop2)
+    {
+        public string Prop1 { get; set; }
+        public string Prop3 { get; init; }
+    }
+}".Trim(), str);
+        }
+
         public class Data
         {
             public string Id { get; set; }

ExpressionTranslator/ExpressionTranslator.cs

@@ -1841,7 +1841,8 @@ public override string ToString()
                     .ToList();
                 var properties = _properties?
                     .ToDictionary(it => it.Name,
-                        it => $"{Translate(it.Type)} {it.Name} {{ get; {(it.IsReadOnly ? "" : "set; ")}}}");
+                        it => $"{Translate(it.Type)} {it.Name} {{ get; {(it.IsReadOnly ? "" : it.IsInitOnly ? "init; " : "set; ")}}}");
+                var ctorParams = _properties?.Where(it => it.IsReadOnly).ToList();
 
                 if (Definitions?.TypeName != null)
                 {
@@ -1881,7 +1882,11 @@ public override string ToString()
                         isInternal = Definitions.Implements?.Any(it =>
                                 !it.GetTypeInfo().IsInterface && !it.GetTypeInfo().IsPublic) ?? false;
                     WriteModifierNextLine(isInternal ? "internal" : "public");
-                    Write("partial class ", Definitions.TypeName);
+                    Write("partial ", Definitions.IsRecordType ? "record " : "class ", Definitions.TypeName);
+                    if (Definitions.IsRecordType && ctorParams?.Count > 0)
+                    {
+                        WriteCtorParams(ctorParams);
+                    }
                     if (implements?.Any() == true)
                     {
                         Write(" : ", string.Join(", ", implements));
@@ -1901,28 +1906,22 @@ public override string ToString()
                 {
                     foreach (var property in _properties)
                     {
+                        if (Definitions.IsRecordType && property.IsReadOnly)
+                            continue;
                         var isInternal = property.Type.GetTypeInfo().IsNotPublic;
                         WriteModifierNextLine(isInternal ? "internal" : "public");
                         Write(properties![property.Name]);
                     }
                     WriteLine();
 
-                    var parameters = _properties.Where(it => it.IsReadOnly).ToList();
-                    if (parameters.Count > 0 && !Definitions.IsStatic)
+                    if (ctorParams?.Count > 0 && !Definitions.IsRecordType)
                     {
-                        var isInternal = parameters.Any(it => it.Type.GetTypeInfo().IsNotPublic);
+                        var isInternal = ctorParams.Any(it => it.Type.GetTypeInfo().IsNotPublic);
                         WriteModifierNextLine(isInternal ? "internal" : "public");
-                        Write(Definitions.TypeName, "(");
-                        for (var i = 0; i < parameters.Count; i++)
-                        {
-                            var parameter = parameters[i];
-                            if (i > 0)
-                                Write(", ");
-                            Write($"{Translate(parameter.Type)} {char.ToLower(parameter.Name[0]) + parameter.Name.Substring(1)}");
-                        }
-                        Write(")");
+                        Write(Definitions.TypeName);
+                        WriteCtorParams(ctorParams);
                         Indent();
-                        foreach (var parameter in parameters)
+                        foreach (var parameter in ctorParams)
                         {
                             WriteNextLine("this.", parameter.Name, " = ", char.ToLower(parameter.Name[0]).ToString(), parameter.Name.Substring(1), ";");
                         }
@@ -1965,6 +1964,19 @@ public override string ToString()
             }
         }
 
+        private void WriteCtorParams(List<PropertyDefinitions> ctorParams)
+        {
+            Write("(");
+            for (var i = 0; i < ctorParams.Count; i++)
+            {
+                var parameter = ctorParams[i];
+                if (i > 0)
+                    Write(", ");
+                Write($"{Translate(parameter.Type)} {char.ToLower(parameter.Name[0]) + parameter.Name.Substring(1)}");
+            }
+            Write(")");
+        }
+
         public enum LambdaType
         {
             PublicMethod,

ExpressionTranslator/PropertyDefinitions.cs

@@ -7,5 +7,6 @@ public class PropertyDefinitions
         public Type Type { get; set; }
         public string Name { get; set; }
         public bool IsReadOnly { get; set; }
+        public bool IsInitOnly { get; set; }
     }
 }

ExpressionTranslator/TypeDefinitions.cs

@@ -11,5 +11,6 @@ public class TypeDefinitions
         public bool IsInternal { get; set; }
         public IEnumerable<Type>? Implements { get; set; }
         public bool PrintFullTypeName { get; set; }
+        public bool IsRecordType { get; set; }
     }
 }