From 400867f01a32af311c40abd2fa6f1ddabe13deb5 Mon Sep 17 00:00:00 2001
From: Stewart Stewart <stewinsalot@gmail.com>
Date: Tue, 21 Feb 2017 17:58:05 -0500
Subject: override code to remove ForeignKeyAction import

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

(limited to 'src')

diff --git a/src/main/scala/NamespacedCodegen.scala b/src/main/scala/NamespacedCodegen.scala
index 095eefb..4114f69 100644
--- a/src/main/scala/NamespacedCodegen.scala
+++ b/src/main/scala/NamespacedCodegen.scala
@@ -102,7 +102,7 @@ class Generator(pkg: String,
        |}
        |""".stripMargin
 
-  val filteredCode = code.lines.drop(1).mkString
+  override def code = super.code.lines.drop(1).mkString("\n")
   // Drops needless import: `"import slick.model.ForeignKeyAction\n"`.
   // Alias to ForeignKeyAction is in profile.api
   // TODO: fix upstream
@@ -124,9 +124,8 @@ class Generator(pkg: String,
         |/** Slick data model trait for extension, choice of backend or usage in the cake pattern. (Make sure to initialize this late.) */
         |trait $traitName${parentType.fold("")(" extends " + _)} {
         |  import profile.api._
-        |  ${indent(filteredCode)}
+        |  ${indent(code)}
         |}""".stripMargin.trim()
-    // TODO: default to upstream packageCode after filteredCode is fixed
   }
 
   override def Table = new Table(_) { table =>
-- 
cgit v1.2.3