Merge pull request #32 from dansiegel/autogenerated-message

add ability to customize Autogenerated Message

src/CodeGenHelpers/CodeBuilder.cs

@@ -8,6 +8,11 @@ namespace CodeGenHelpers
 {
     public sealed class CodeBuilder
     {
+        private const string DefaultAutoGeneratedMessage = @"This code was generated.
+
+Changes to this file may cause incorrect behavior and will be lost if
+the code is regenerated.";
+        private string _autoGeneratedMessage;
         private readonly List<string> _namespaceImports = new List<string>();
         private readonly List<string> _assemblyAttributes = new List<string>();
         private readonly Queue<IBuilder> _classes = new Queue<IBuilder>();
@@ -84,6 +89,12 @@ public EnumBuilder AddEnum(string name)
             return builder;
         }
 
+        public CodeBuilder WithAutoGeneratedMessage(string message)
+        {
+            _autoGeneratedMessage = message;
+            return this;
+        }
+
         public void Clear()
         {
             _namespaceImports.Clear();
@@ -96,10 +107,12 @@ private CodeWriter BuildInternal()
             var writer = new CodeWriter(IndentStyle);
             writer.AppendLine("//------------------------------------------------------------------------------");
             writer.AppendLine("// <auto-generated>");
-            writer.AppendLine("//     This code was generated.");
-            writer.AppendLine("//");
-            writer.AppendLine("//     Changes to this file may cause incorrect behavior and will be lost if");
-            writer.AppendLine("//     the code is regenerated.");
+            var autoGeneratedMessage = string.IsNullOrEmpty(_autoGeneratedMessage) ? DefaultAutoGeneratedMessage : _autoGeneratedMessage;
+            var lines = Regex.Replace(autoGeneratedMessage, @"\r\n", "\n").Split('\n');
+            foreach(var line in lines)
+            {
+                writer.AppendLine($"//     {line}".Trim());
+            }
             writer.AppendLine("// </auto-generated>");
             writer.AppendLine("//------------------------------------------------------------------------------");
             writer.NewLine();

tests/CodeGenHelpers.Tests/SampleCode/CustomAutoGeneratedMessage.cs

@@ -0,0 +1,12 @@
+//------------------------------------------------------------------------------
+// <auto-generated>
+//     This is a Custom Defined Message
+//     See my Custom Copyright Message
+//
+//     Don't Make Changes to this File!!!
+// </auto-generated>
+//------------------------------------------------------------------------------
+
+namespace CodeGenHelpers.SampleCode
+{
+}

tests/CodeGenHelpers.Tests/Tests/CodeBuilderTests.cs

@@ -0,0 +1,30 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using Xunit;
+using Xunit.Abstractions;
+
+namespace CodeGenHelpers.Tests
+{
+    public class CodeBuilderTests : TestBase
+    {
+        public CodeBuilderTests(ITestOutputHelper testOutput)
+            : base(testOutput)
+        {
+        }
+
+        [Fact]
+        public void AddsCustomAutoGeneratedMessage()
+        {
+            var builder = CodeBuilder.Create(Namespace)
+                .WithAutoGeneratedMessage(@"This is a Custom Defined Message
+See my Custom Copyright Message
+
+Don't Make Changes to this File!!!");
+
+            MakeAssertion(builder, "CustomAutoGeneratedMessage");
+        }
+    }
+}