add file header support

Author: pwelter34

src/EntityFrameworkCore.Generator.Core/Metadata/Generation/Model.cs

@@ -19,6 +19,8 @@ public Model()
 
     public string? ModelAttributes { get; set; }
 
+    public string? ModelHeader { get; internal set; }
+
 
     public string? ValidatorNamespace { get; set; }
 

src/EntityFrameworkCore.Generator.Core/ModelGenerator.cs

@@ -538,6 +538,10 @@ private void CreateModel<TOption>(Entity entity, TOption options, ModelType mode
             ? options.Namespace
             : _options.Model.Shared.Namespace;
 
+        var modelHeader = options.Header.HasValue()
+            ? options.Header
+            : _options.Model.Shared.Header;
+
         modelNamespace ??= "Data.Models";
 
         var modelClass = ToLegalName(options.Name);
@@ -551,6 +555,7 @@ private void CreateModel<TOption>(Entity entity, TOption options, ModelType mode
             ModelNamespace = modelNamespace,
             ModelClass = modelClass,
             ModelAttributes = options.Attributes,
+            ModelHeader = modelHeader
         };
 
         foreach (var property in entity.Properties)

src/EntityFrameworkCore.Generator.Core/OptionMapper.cs

@@ -63,6 +63,7 @@ private static void MapClassBase(ClassOptionsBase option, ClassBase classBase)
         option.Name = classBase.Name;
         option.BaseClass = classBase.BaseClass;
         option.Attributes = classBase.Attributes;
+        option.Header = classBase.Header;
     }
 
     private static void MapModelBase(ModelOptionsBase option, ModelBase modelBase)
@@ -109,6 +110,7 @@ private static void MapShared(SharedModelOptions option, SharedModel shared)
     {
         option.Namespace = shared.Namespace;
         option.Directory = shared.Directory;
+        option.Header = shared.Header;
 
         MapSelection(option.Include, shared.Include);
         MapSelection(option.Exclude, shared.Exclude);

src/EntityFrameworkCore.Generator.Core/Options/ClassOptionsBase.cs

@@ -87,4 +87,13 @@ public string? Attributes
         get => GetProperty();
         set => SetProperty(value);
     }
+
+    /// <summary>
+    /// Gets or sets the file header.
+    /// </summary>
+    public string? Header
+    {
+        get => GetProperty();
+        set => SetProperty(value);
+    }
 }

src/EntityFrameworkCore.Generator.Core/Options/SharedModelOptions.cs

@@ -45,6 +45,15 @@ public string? Directory
         set => SetProperty(value);
     }
 
+    /// <summary>
+    /// Gets or sets the file header
+    /// </summary>
+    public string? Header
+    {
+        get => GetProperty();
+        set => SetProperty(value);
+    }
+
     /// <summary>
     /// Gets or sets the include selection options.
     /// </summary>
@@ -60,5 +69,4 @@ public string? Directory
     /// The exclude selection options.
     /// </value>
     public SelectionOptions Exclude { get; }
-
 }

src/EntityFrameworkCore.Generator.Core/Serialization/ClassBase.cs

@@ -55,4 +55,9 @@ public abstract class ClassBase
     /// The attributes to add to the class
     /// </value>
     public string? Attributes { get; set; }
+
+    /// <summary>
+    /// Gets or sets the file header.
+    /// </summary>
+    public string? Header { get; set; }
 }

src/EntityFrameworkCore.Generator.Core/Serialization/SharedModel.cs

@@ -32,6 +32,11 @@ public SharedModel()
     /// </value>
     public string Directory { get; set; }
 
+    /// <summary>
+    /// Gets or sets the file header.
+    /// </summary>
+    public string? Header { get; set; }
+
     /// <summary>
     /// Gets or sets the include selection options.
     /// </summary>

src/EntityFrameworkCore.Generator.Core/Templates/DataContextTemplate.cs

@@ -19,7 +19,11 @@ public override string WriteCode()
     {
         CodeBuilder.Clear();
 
+        if (Options.Data.Context.Header.HasValue())
+            CodeBuilder.AppendLine(Options.Data.Context.Header).AppendLine();
+
         CodeBuilder.AppendLine("using System;");
+        CodeBuilder.AppendLine();
         CodeBuilder.AppendLine("using Microsoft.EntityFrameworkCore;");
         CodeBuilder.AppendLine("using Microsoft.EntityFrameworkCore.Metadata;");
         CodeBuilder.AppendLine();

src/EntityFrameworkCore.Generator.Core/Templates/EntityClassTemplate.cs

@@ -21,6 +21,9 @@ public override string WriteCode()
     {
         CodeBuilder.Clear();
 
+        if (Options.Data.Entity.Header.HasValue())
+            CodeBuilder.AppendLine(Options.Data.Entity.Header).AppendLine();
+
         CodeBuilder.AppendLine("using System;");
         CodeBuilder.AppendLine("using System.Collections.Generic;");
         CodeBuilder.AppendLine();

src/EntityFrameworkCore.Generator.Core/Templates/MapperClassTemplate.cs

@@ -19,8 +19,13 @@ public override string WriteCode()
     {
         CodeBuilder.Clear();
 
+        if (Options.Model.Mapper.Header.HasValue())
+            CodeBuilder.AppendLine(Options.Model.Mapper.Header).AppendLine();
+
         CodeBuilder.AppendLine("using System;");
+        CodeBuilder.AppendLine();
         CodeBuilder.AppendLine("using AutoMapper;");
+        CodeBuilder.AppendLine();
 
         var imports = new SortedSet<string>
         {