From a0d200c6b042f1c7669bdba11a7607476c16ecf4 Mon Sep 17 00:00:00 2001
From: Stewart Stewart <stewinsalot@gmail.com>
Date: Thu, 1 Dec 2016 15:42:44 -0800
Subject: add parameter for type replacements (decouple driver Time)

---
 src/main/scala/CodegenPlugin.scala | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'src/main/scala/CodegenPlugin.scala')

diff --git a/src/main/scala/CodegenPlugin.scala b/src/main/scala/CodegenPlugin.scala
index 1286919..b3a97ca 100644
--- a/src/main/scala/CodegenPlugin.scala
+++ b/src/main/scala/CodegenPlugin.scala
@@ -32,6 +32,10 @@ object CodegenPlugin extends AutoPlugin {
       "codegen-schema-imports",
       "A list of things to import into each schema definition"
     )
+    lazy val codegenTypeReplacements = SettingKey[Map[String, String]](
+      "codegen-type-replacements",
+      "A map of types to find and replace"
+    )
 
     lazy val slickCodeGenTask =
       TaskKey[Unit]("gen-tables", "generate the table definitions")
@@ -46,6 +50,7 @@ object CodegenPlugin extends AutoPlugin {
     codegenSchemaBaseClassParts := List.empty,
     codegenIdType := Option.empty,
     codegenSchemaImports := List.empty,
+    codegenTypeReplacements := Map.empty,
     slickCodeGenTask := Def.taskDyn {
       Def.task {
         Generator.run(
@@ -62,7 +67,8 @@ object CodegenPlugin extends AutoPlugin {
             case parts => parts.mkString(" with ")
           },
           codegenIdType.value,
-          codegenSchemaImports.value
+          codegenSchemaImports.value,
+          codegenTypeReplacements.value
         )
       }
     }.value
-- 
cgit v1.2.3