diff options
Diffstat (limited to 'src/main/scala/OutputHelpers.scala')
-rw-r--r-- | src/main/scala/OutputHelpers.scala | 15 |
1 files changed, 3 insertions, 12 deletions
diff --git a/src/main/scala/OutputHelpers.scala b/src/main/scala/OutputHelpers.scala index 975bed9..6b0ab2e 100644 --- a/src/main/scala/OutputHelpers.scala +++ b/src/main/scala/OutputHelpers.scala @@ -6,22 +6,13 @@ trait TableOutputHelpers extends TableFileGenerator with OutputHelpers { self: S def schemaName: String def imports: String - def packageTableCode(headerComment: String, - pkg: String, - schemaName: String, - imports: String, - profile: String): String = + def packageTableCode(headerComment: String, pkg: String, schemaName: String, imports: String): String = s"""|${headerComment.trim().lines.map("// " + _).mkString("\n")} |package $pkg |package $schemaName | |$imports | - |/** Stand-alone Slick data model for immediate use */ - |object tables extends { - | val profile = $profile - |} with Tables - | |/** Slick data model trait for extension, choice of backend or usage in the cake pattern. (Make sure to initialize this late.) */ |trait Tables${parentType.fold("")(" extends " + _)} { | import profile.api._ @@ -29,8 +20,8 @@ trait TableOutputHelpers extends TableFileGenerator with OutputHelpers { self: S |} |""".stripMargin.trim() - def writeTablesToFile(profile: String, folder: String, pkg: String, fileName: String): Unit = { - writeStringToFile(content = packageTableCode(headerComment, pkg, schemaName, imports, profile), + def writeTablesToFile(folder: String, pkg: String, fileName: String): Unit = { + writeStringToFile(content = packageTableCode(headerComment, pkg, schemaName, imports), folder = folder, pkg = s"$pkg.$schemaName", fileName = fileName) |