feat: runtime comment provider

Author: ExerciseBook

Tomlet.Tests/CommentProvider/TestInlineCommentProvider.cs

@@ -0,0 +1,20 @@
+using System.Collections.Generic;
+
+namespace Tomlet.Tests.CommentProvider;
+
+public class TestInlineCommentProvider : ICommentProvider
+{
+    public static Dictionary<string, string> Comments = new Dictionary<string, string>();
+
+    private readonly string _name;
+
+    public TestInlineCommentProvider(string name)
+    {
+        _name = name;
+    }
+
+    public string GetComment()
+    {
+        return Comments[_name];
+    }
+}

Tomlet.Tests/CommentProvider/TestPrecedingCommentProvider.cs

@@ -0,0 +1,20 @@
+using System.Collections.Generic;
+
+namespace Tomlet.Tests.CommentProvider;
+
+public class TestPrecedingCommentProvider : ICommentProvider
+{
+    public static Dictionary<string, string> Comments = new Dictionary<string, string>();
+
+    private readonly string _name;
+
+    public TestPrecedingCommentProvider(string name)
+    {
+        _name = name;
+    }
+
+    public string GetComment()
+    {
+        return Comments[_name];
+    }
+}

Tomlet.Tests/CommentSerializationTests.cs

@@ -1,6 +1,9 @@
+using System;
 using Tomlet.Models;
+using Tomlet.Tests.CommentProvider;
 using Tomlet.Tests.TestModelClasses;
 using Xunit;
+using Xunit.Abstractions;
 
 namespace Tomlet.Tests;
 
@@ -72,7 +75,7 @@ public void CommentsOnTableArraysWork()
         tomlString.Comments.InlineComment = "Inline comment on value";
         table.PutValue("key", tomlString);
 
-        var tableArray = new TomlArray {table};
+        var tableArray = new TomlArray { table };
         tableArray.Comments.PrecedingComment = "This is a preceding comment on the table-array itself";
 
         doc.PutValue("table-array", tableArray);
@@ -91,7 +94,7 @@ public void CommentsOnTableArraysWork()
     public void CommentsOnPrimitiveArraysWork()
     {
         var doc = TomlDocument.CreateEmpty();
-        var tomlNumbers = new TomlArray {1, 2, 3};
+        var tomlNumbers = new TomlArray { 1, 2, 3 };
         doc.PutValue("numbers", tomlNumbers);
 
         tomlNumbers[0].Comments.PrecedingComment = "This is a preceding comment on the first value of the array";
@@ -103,7 +106,7 @@ public void CommentsOnPrimitiveArraysWork()
     2, # This is an inline comment on the second value of the array
     3,
 ]".ReplaceLineEndings();
-        
+
         //Replace tabs with spaces because this source file uses spaces
         var actual = doc.SerializedValue.Trim().Replace("\t", "    ").ReplaceLineEndings();
         Assert.Equal(expected, actual);
@@ -115,10 +118,33 @@ public void CommentAttributesWork()
         var config = TomletMain.To<ExampleMailboxConfigClass>(TestResources.ExampleMailboxConfigurationTestInput);
 
         var doc = TomletMain.DocumentFrom(config);
-        
+
         Assert.Equal("The name of the mailbox", doc.GetValue("mailbox").Comments.InlineComment);
         Assert.Equal("Your username for the mailbox", doc.GetValue("username").Comments.InlineComment);
         Assert.Equal("The password you use to access the mailbox", doc.GetValue("password").Comments.InlineComment);
         Assert.Equal("The rules for the mailbox follow", doc.GetArray("rules").Comments.PrecedingComment);
     }
+
+    [Fact]
+    public void CommentProviderTest()
+    {
+        TestPrecedingCommentProvider.Comments["PrecedingComment"] = Guid.NewGuid().ToString();
+        TestInlineCommentProvider.Comments["InlineComment"] = Guid.NewGuid().ToString();
+
+        var data = new CommentProviderTestModel()
+        {
+            PrecedingComment = "Dynamic Preceding Comment",
+            InlineComment = "Inline Comment",
+        };
+
+        var doc = TomletMain.DocumentFrom(data);
+
+        Assert.Equal(TestPrecedingCommentProvider.Comments["PrecedingComment"],
+            doc.GetValue("PrecedingComment").Comments.PrecedingComment);
+        Assert.Equal("PlainInlineComment", doc.GetValue("PrecedingComment").Comments.InlineComment);
+
+        Assert.Equal(TestInlineCommentProvider.Comments["InlineComment"],
+            doc.GetValue("InlineComment").Comments.InlineComment);
+        Assert.Equal("PlainPrecedingComment", doc.GetValue("InlineComment").Comments.PrecedingComment);
+    }
 }

Tomlet.Tests/TestModelClasses/CommentProviderTestModel.cs

@@ -0,0 +1,15 @@
+using Tomlet.Attributes;
+using Tomlet.Tests.CommentProvider;
+
+namespace Tomlet.Tests.TestModelClasses;
+
+public class CommentProviderTestModel
+{
+    [TomlPrecedingCommentProvider(typeof(TestPrecedingCommentProvider), new object[] { "PrecedingComment" })]
+    [TomlInlineComment("PlainInlineComment")]
+    public string PrecedingComment { get; set; }
+    
+    [TomlInlineCommentProvider(typeof(TestInlineCommentProvider), new object[] { "InlineComment" })]
+    [TomlPrecedingComment("PlainPrecedingComment")]
+    public string InlineComment { get; set; }
+}

Tomlet/Attributes/TomlCommentProviderAttribute.cs

@@ -0,0 +1,33 @@
+using System;
+using System.Linq;
+
+namespace Tomlet.Attributes;
+
+[AttributeUsage(AttributeTargets.Field | AttributeTargets.Property)]
+public class TomlCommentProviderAttribute : Attribute
+{
+    private readonly Type _provider;
+    private readonly object[] _args;
+    private readonly Type[] _constructorParamsType;
+
+    public string GetComment()
+    {
+        var constructor = _provider.GetConstructor(_constructorParamsType) ??
+                          throw new ArgumentException("Fail to get a constructor matching the parameters");
+        var instance = constructor.Invoke(_args) as ICommentProvider ??
+                       throw new Exception("Fail to create an instance of the provider");
+        return instance.GetComment();
+    }
+
+    public TomlCommentProviderAttribute(Type provider, object[] args)
+    {
+        if (!typeof(ICommentProvider).IsAssignableFrom(provider))
+        {
+            throw new ArgumentException("Provider must implement ICommentProvider");
+        }
+
+        _provider = provider;
+        _args = args ?? new object[] { };
+        _constructorParamsType = args?.Select(a => a.GetType()).ToArray() ?? new Type[] { };
+    }
+}

Tomlet/Attributes/TomlInlineCommentAttribute.cs

@@ -3,12 +3,10 @@
 namespace Tomlet.Attributes;
 
 [AttributeUsage(AttributeTargets.Field | AttributeTargets.Property)]
-public class TomlInlineCommentAttribute : Attribute
+public class TomlInlineCommentAttribute : TomlInlineCommentProviderAttribute
 {
-    internal string Comment { get; }
-
-    public TomlInlineCommentAttribute(string comment)
+    public TomlInlineCommentAttribute(string comment) : base(typeof(TomlSimpleCommentProvider),
+        new object[] { comment })
     {
-        Comment = comment;
     }
 }

Tomlet/Attributes/TomlInlineCommentProviderAttribute.cs

@@ -0,0 +1,15 @@
+using System;
+
+namespace Tomlet.Attributes;
+
+[AttributeUsage(AttributeTargets.Field | AttributeTargets.Property)]
+public class TomlInlineCommentProviderAttribute : TomlCommentProviderAttribute
+{
+    public TomlInlineCommentProviderAttribute(Type provider) : base(provider, new object[] { })
+    {
+    }
+
+    public TomlInlineCommentProviderAttribute(Type provider, object[] args) : base(provider, args)
+    {
+    }
+}

Tomlet/Attributes/TomlPrecedingCommentAttribute.cs

@@ -3,12 +3,10 @@
 namespace Tomlet.Attributes;
 
 [AttributeUsage(AttributeTargets.Field | AttributeTargets.Property)]
-public class TomlPrecedingCommentAttribute : Attribute
+public class TomlPrecedingCommentAttribute : TomlPrecedingCommentProviderAttribute
 {
-    internal string Comment { get; }
-
-    public TomlPrecedingCommentAttribute(string comment)
+    public TomlPrecedingCommentAttribute(string comment) : base(typeof(TomlSimpleCommentProvider),
+        new object[] { comment })
     {
-        Comment = comment;
     }
 }

Tomlet/Attributes/TomlPrecedingCommentProviderAttribute.cs

@@ -0,0 +1,15 @@
+using System;
+
+namespace Tomlet.Attributes;
+
+[AttributeUsage(AttributeTargets.Field | AttributeTargets.Property)]
+public class TomlPrecedingCommentProviderAttribute : TomlCommentProviderAttribute
+{
+    public TomlPrecedingCommentProviderAttribute(Type provider) : base(provider, new object[] { })
+    {
+    }
+
+    public TomlPrecedingCommentProviderAttribute(Type provider, object[] args) : base(provider, args)
+    {
+    }
+}

Tomlet/CommentProviderUtil.cs

@@ -0,0 +1,18 @@
+using System;
+
+namespace Tomlet;
+
+internal static class CommentProviderUtil
+{
+    public static string GetComment(Type provider)
+    {
+        var constructor = provider.GetConstructor(Type.EmptyTypes);
+        if (constructor == null)
+        {
+            throw new ArgumentException("Provider must have a parameterless constructor");
+        }
+
+        var instance = (ICommentProvider)constructor.Invoke(null);
+        return instance.GetComment();
+    }
+}