diff options
Diffstat (limited to 'src/main/scala/OutputHelpers.scala')
-rw-r--r-- | src/main/scala/OutputHelpers.scala | 35 |
1 files changed, 12 insertions, 23 deletions
diff --git a/src/main/scala/OutputHelpers.scala b/src/main/scala/OutputHelpers.scala index 97fb4e9..17c2019 100644 --- a/src/main/scala/OutputHelpers.scala +++ b/src/main/scala/OutputHelpers.scala @@ -1,7 +1,6 @@ import slick.codegen.{SourceCodeGenerator, OutputHelpers} -trait TableOutputHelpers extends TableFileGenerator with OutputHelpers { - self: SourceCodeGenerator => +trait TableOutputHelpers extends TableFileGenerator with OutputHelpers { self: SourceCodeGenerator => def headerComment: String def schemaName: String @@ -31,30 +30,21 @@ trait TableOutputHelpers extends TableFileGenerator with OutputHelpers { |} |""".stripMargin.trim() - def writeTablesToFile(profile: String, - folder: String, - pkg: String, - fileName: String): Unit = { - writeStringToFile( - content = - packageTableCode(headerComment, pkg, schemaName, imports, profile), - folder = folder, - pkg = s"$pkg.$schemaName", - fileName = fileName) + def writeTablesToFile(profile: String, folder: String, pkg: String, fileName: String): Unit = { + writeStringToFile(content = packageTableCode(headerComment, pkg, schemaName, imports, profile), + folder = folder, + pkg = s"$pkg.$schemaName", + fileName = fileName) } } -trait RowOutputHelpers extends RowFileGenerator with OutputHelpers { - self: SourceCodeGenerator => +trait RowOutputHelpers extends RowFileGenerator with OutputHelpers { self: SourceCodeGenerator => def headerComment: String def schemaName: String def imports: String - def packageRowCode(headerComment: String, - schemaName: String, - pkg: String, - imports: String): String = + def packageRowCode(headerComment: String, schemaName: String, pkg: String, imports: String): String = s"""|${headerComment.trim().lines.map("// " + _).mkString("\n")} |/** Definitions for table rows types of database schema $schemaName */ |package $pkg @@ -67,10 +57,9 @@ trait RowOutputHelpers extends RowFileGenerator with OutputHelpers { def writeRowsToFile(folder: String, pkg: String, fileName: String): Unit = { - writeStringToFile( - content = packageRowCode(headerComment, schemaName, pkg, imports), - folder = folder, - pkg = s"$pkg.$schemaName", - fileName = fileName) + writeStringToFile(content = packageRowCode(headerComment, schemaName, pkg, imports), + folder = folder, + pkg = s"$pkg.$schemaName", + fileName = fileName) } } |