diff options
Diffstat (limited to 'src/main/scala/CodegenPlugin.scala')
-rw-r--r-- | src/main/scala/CodegenPlugin.scala | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/main/scala/CodegenPlugin.scala b/src/main/scala/CodegenPlugin.scala index 8b54887..01286ee 100644 --- a/src/main/scala/CodegenPlugin.scala +++ b/src/main/scala/CodegenPlugin.scala @@ -28,6 +28,10 @@ object CodegenPlugin extends AutoPlugin { "codegen-id-type", "The in-scope type `T` of kind `T[TableRow]` to apply in place T for id columns" ) + lazy val codegenSchemaImports = SettingKey[List[String]]( + "codegen-schema-imports", + "A list of things to import into each schema definition" + ) lazy val slickCodeGenTask = TaskKey[Unit]("gen-tables", "generate the table definitions") @@ -41,6 +45,7 @@ object CodegenPlugin extends AutoPlugin { codegenForeignKeys := Map.empty, codegenSchemaBaseClassParts := List.empty, codegenIdType := Option.empty, + codegenSchemaImports := List.empty, slickCodeGenTask := Def.taskDyn { Def.task { Generator.run( @@ -53,7 +58,8 @@ object CodegenPlugin extends AutoPlugin { case Nil => "AnyRef" case parts => parts.mkString(" with ") }, - codegenIdType.value + codegenIdType.value, + codegenSchemaImports.value ) } }.value |