From 792f2401e29c728ce6730f54da15382bb567bd04 Mon Sep 17 00:00:00 2001
From: Petro Verkhogliad <vpetro@gmail.com>
Date: Thu, 17 Nov 2016 02:31:10 -0500
Subject: Auto inject task and default settings

---
 src/main/scala/CodegenPlugin.scala | 40 ++++++++++++++++++++++----------------
 1 file changed, 23 insertions(+), 17 deletions(-)

(limited to 'src')

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
+
+  )
+
+
 }
-- 
cgit v1.2.3