Auto inject task and default settings

1 files changed, 23 insertions, 17 deletions

diff --git a/src/main/scala/CodegenPlugin.scala b/src/main/scala/CodegenPlugin.scala
index 8bde45d..d74eda7 100644
--- a/src/main/scala/CodegenPlugin.scala
+++ b/src/main/scala/CodegenPlugin.scala
@@ -7,28 +7,34 @@ object CodegenPlugin extends AutoPlugin {
   type TableColumn = (String, String)
 
   object autoImport {
-    lazy val codegen = TaskKey[Unit]("gen-tables", "generate slick database schema")
-
     lazy val codegenURI = SettingKey[String]("codegen-uri", "uri for the database configuration")
     lazy val codegenPackage = SettingKey[String]("codegen-package", "package in which to place generated code")
     lazy val codegenOutputPath = SettingKey[String]("codegen-output-path", "directory to with the generated code will be written")
     lazy val codegenSchemaWhitelist = SettingKey[List[String]]("codegen-schema-whitelist", "schemas and tables to process")
     lazy val codegenForeignKeys = SettingKey[Map[TableColumn, TableColumn]]("codegen-foreign-keys", "foreign key references to data models add manually")
 
-    lazy val slickCodeGenTask = Def.task {
-      Generator.run(
-        new java.net.URI(codegenURI.value),
-        codegenPackage.value,
-        Some(codegenSchemaWhitelist.value).filter(_.nonEmpty),
-        codegenOutputPath.value,
-        codegenForeignKeys.value
-      )
-    }
-
-    lazy val baseCodegenSettings: Seq[Def.Setting[_]] = Seq(
-      codegen := slickCodeGenTask.value,
-      codegenSchemaWhitelist := List.empty,
-      codegenForeignKeys := Map.empty
-    )
+    lazy val slickCodeGenTask = TaskKey[Unit]("gen-tables", "generate the table definitions")
+
   }
+
+  import autoImport._
+
+  override lazy val projectSettings = Seq(
+    codegenSchemaWhitelist := List.empty,
+    codegenForeignKeys := Map.empty,
+    slickCodeGenTask := Def.taskDyn {
+      Def.task {
+        Generator.run(
+          new java.net.URI(codegenURI.value),
+          codegenPackage.value,
+          Some(codegenSchemaWhitelist.value).filter(_.nonEmpty),
+          codegenOutputPath.value,
+          codegenForeignKeys.value
+        )
+      }
+    }.value
+
+  )
+
+
 }