diff options
Diffstat (limited to 'src/main/scala/OutputHelpers.scala')
-rw-r--r-- | src/main/scala/OutputHelpers.scala | 57 |
1 files changed, 56 insertions, 1 deletions
diff --git a/src/main/scala/OutputHelpers.scala b/src/main/scala/OutputHelpers.scala index 3a184a4..2a3641e 100644 --- a/src/main/scala/OutputHelpers.scala +++ b/src/main/scala/OutputHelpers.scala @@ -1,4 +1,4 @@ -trait OutputHelpers extends slick.codegen.OutputHelpers { +trait OOutputHelpers extends slick.codegen.OutputHelpers { def imports: String @@ -25,3 +25,58 @@ trait OutputHelpers extends slick.codegen.OutputHelpers { |}""".stripMargin.trim() } } + +import slick.codegen.{SourceCodeGenerator, OutputHelpers} + +trait TableFileGenerator { self: SourceCodeGenerator => + def writeTablesToFile(folder:String, pkg: String, fileName: String): Unit +} + +trait RowFileGenerator { self: SourceCodeGenerator => + def writeRowsToFile(folder:String, pkg: String, fileName: String): Unit +} + +// Dirty work to hide OutputHelpers +trait TableOutputHelpers extends TableFileGenerator with OutputHelpers { self: SourceCodeGenerator => + + def packageTableCode: String = + """|${headerComment.trim().lines.map("// " + _).mkString("\n")} + |package $pkg + |package $schemaName + | + |$imports + | + |/** Stand-alone Slick data model for immediate use */ + |// TODO: change this to `object tables` + |package object $schemaName 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._ + | ${indent(code)} | + |""".stripMargin.trim() + + def writeTablesToFile(folder:String, pkg: String, fileName: String): Unit = { + writeStringToFile(content = packageTableCode, folder = folder, pkg = pkg, fileName = fileName) + } +} + +trait RowOutputHelpers extends RowFileGenerator with OutputHelpers { self: SourceCodeGenerator => + + def packageRowCode: String = + """|${headerComment.trim().lines.map("// " + _).mkString("\n")} + |/** Definitions for table rows types of database schema $schemaName */ + |package $pkg + |package $schemaName + | + |$imports + | + |$code + |""".stripMargin.trim() + + def writeRowsToFile(folder:String, pkg: String, fileName: String): Unit = { + writeStringToFile(content = packageRowCode, folder = folder, pkg = pkg, fileName = fileName) + } +} |