aboutsummaryrefslogtreecommitdiff
path: root/src/main/scala/CodegenPlugin.scala
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/scala/CodegenPlugin.scala')
-rw-r--r--src/main/scala/CodegenPlugin.scala8
1 files changed, 7 insertions, 1 deletions
diff --git a/src/main/scala/CodegenPlugin.scala b/src/main/scala/CodegenPlugin.scala
index f9351c1..92e595f 100644
--- a/src/main/scala/CodegenPlugin.scala
+++ b/src/main/scala/CodegenPlugin.scala
@@ -24,6 +24,10 @@ object CodegenPlugin extends AutoPlugin {
lazy val codegenSchemaBaseClassParts = SettingKey[List[String]](
"codegen-schema-base-class-parts",
"parts inherited by each generated schema object")
+ lazy val codegenIdType = SettingKey[String](
+ "codegen-id-type",
+ "The in-scope type `T` of kind `T[TableRow]` to apply in place T for id columns."
+ )
lazy val slickCodeGenTask =
TaskKey[Unit]("gen-tables", "generate the table definitions")
@@ -36,6 +40,7 @@ object CodegenPlugin extends AutoPlugin {
codegenSchemaWhitelist := List.empty,
codegenForeignKeys := Map.empty,
codegenSchemaBaseClassParts := List.empty,
+ codegenIdType := "Id",
slickCodeGenTask := Def.taskDyn {
Def.task {
Generator.run(
@@ -47,7 +52,8 @@ object CodegenPlugin extends AutoPlugin {
codegenSchemaBaseClassParts.value match {
case Nil => "AnyRef"
case parts => parts.mkString(" with ")
- }
+ },
+ codegenIdType.value
)
}
}.value