From ee5142f4e3c889429c00fddf3d045506f10acf94 Mon Sep 17 00:00:00 2001
From: Stewart Stewart <stewinsalot@gmail.com>
Date: Tue, 21 Feb 2017 22:41:54 -0500
Subject: add codegen header

---
 src/main/scala/NamespacedCodegen.scala | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'src/main/scala/NamespacedCodegen.scala')

diff --git a/src/main/scala/NamespacedCodegen.scala b/src/main/scala/NamespacedCodegen.scala
index affd36c..c545b80 100644
--- a/src/main/scala/NamespacedCodegen.scala
+++ b/src/main/scala/NamespacedCodegen.scala
@@ -24,6 +24,7 @@ object Generator {
           manualForeignKeys: Map[(String, String), (String, String)],
           parentType: Option[String],
           idType: Option[String],
+          header: String,
           schemaImports: List[String],
           typeReplacements: Map[String, String]) = {
     val dc: DatabaseConfig[JdbcProfile] =
@@ -54,6 +55,7 @@ object Generator {
                                         manualForeignKeys,
                                         parentType,
                                         idType,
+                                        header,
                                         schemaImports,
                                         typeReplacements)
           generator.writeToFile(profile = profile,
@@ -75,6 +77,7 @@ class Generator(pkg: String,
                 manualForeignKeys: Map[(String, String), (String, String)],
                 override val parentType: Option[String],
                 idType: Option[String],
+                override val headerComment: String,
                 schemaImports: List[String],
                 typeReplacements: Map[String, String])
     extends SourceCodeGenerator(schemaOnlyModel)
-- 
cgit v1.2.3