add codegen header

1 files changed, 6 insertions, 0 deletions

diff --git a/src/main/scala/CodegenPlugin.scala b/src/main/scala/CodegenPlugin.scala
index 0c8020f..c90fe66 100644
--- a/src/main/scala/CodegenPlugin.scala
+++ b/src/main/scala/CodegenPlugin.scala
@@ -46,6 +46,10 @@ object CodegenPlugin extends AutoPlugin {
       "codegen-type-replacements",
       "A map of types to find and replace"
     )
+    lazy val codegenHeader = SettingKey[String](
+      "codegen-header",
+      "Comments that go at the top of generated files; notices and tooling directives."
+    )
 
     lazy val slickCodeGenTask =
       TaskKey[Unit]("gen-tables", "generate the table definitions")
@@ -58,6 +62,7 @@ object CodegenPlugin extends AutoPlugin {
     codegenIdType := Option.empty,
     codegenSchemaImports := List.empty,
     codegenTypeReplacements := Map.empty,
+    codegenHeader := "AUTO-GENERATED Slick data model",
     slickCodeGenTask := Def.taskDyn {
       Def.task {
         codegenDatabaseConfigs.value.foreach {
@@ -77,6 +82,7 @@ object CodegenPlugin extends AutoPlugin {
                 Some(parts).filter(_.nonEmpty).map(_.mkString(" with "))
               },
               codegenIdType.value,
+              codegenHeader.value,
               codegenSchemaImports.value,
               codegenTypeReplacements.value
             )