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
56
57
58
59
60
61
62
63
64
65
|
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,
profile: String): String =
s"""|${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`
|object `package` 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(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 =>
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)
}
}
|
