trait OOutputHelpers extends slick.codegen.OutputHelpers {
def imports: String
def headerComment: String = ""
override def packageCode(profile: String,
pkg: String,
container: String,
parentType: Option[String]): String = {
s"""|${headerComment.trim().lines.map("// " + _).mkString("\n")}
|package $pkg
|
|$imports
|
|/** Stand-alone Slick data model for immediate use */
|package object $container 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()
}
}
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)
}
}