1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
import slick.codegen.{SourceCodeGenerator, OutputHelpers}
trait TableOutputHelpers extends TableFileGenerator with OutputHelpers { self: SourceCodeGenerator =>
def headerComment: String
def schemaName: String
def imports: String
def packageTableCode(headerComment: String, pkg: String, schemaName: String, imports: String): String =
s"""|${headerComment.trim().lines.map("// " + _).mkString("\n")}
|package $pkg
|package $schemaName
|
|$imports
|
|/** 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(headerComment, pkg, schemaName, imports),
folder = folder,
pkg = s"$pkg.$schemaName",
fileName = fileName)
}
}
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 =
s"""|${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(headerComment, schemaName, pkg, imports),
folder = folder,
pkg = s"$pkg.$schemaName",
fileName = fileName)
}
}
|