From f78a33927998c1cc44f5f09e165c2839ea51949c Mon Sep 17 00:00:00 2001
From: Naftoli Gugenheim <naftoligug@gmail.com>
Date: Thu, 12 Mar 2020 17:32:33 -0400
Subject: [PATCH] Add TableModule and EntityTableModule

To save some boilerplate when defining slick table schemas
---
 .../slick/additions/AdditionsProfile.scala    | 30 +++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/src/main/scala/slick/additions/AdditionsProfile.scala b/src/main/scala/slick/additions/AdditionsProfile.scala
index 589a9aa..b4236bc 100644
--- a/src/main/scala/slick/additions/AdditionsProfile.scala
+++ b/src/main/scala/slick/additions/AdditionsProfile.scala
@@ -141,5 +141,35 @@ trait AdditionsProfile { this: JdbcProfile =>
     trait AutoNameSnakify extends AutoName { this: Table[_] =>
       protected implicit def nameStyle: NameStyle = NameStyle.Snakify
     }
+
+    trait TableModule {
+      type Rec
+      type Row <: Table[Rec]
+      val Q: TableQuery[Row]
+    }
+
+    abstract class EntityTableModule[K: BaseColumnType, V](tableName: String) extends TableModule {
+      type Rec = KeyedEntity[K, V]
+
+      abstract class BaseEntRow(tag: Tag) extends EntityTable[K, V](tag, tableName) with AutoNameSnakify {
+        def tableQuery = EntityTableModule.this.Q
+      }
+
+      type Row <: BaseEntRow
+
+      val mkRow: Tag => Row = {
+        import scala.reflect.runtime.universe._
+        val m = runtimeMirror(this.getClass.getClassLoader)
+        val thisAsSymbol = m.moduleSymbol(this.getClass)
+        val rowClassMirror = m.reflectClass(thisAsSymbol.info.member(TypeName("Row")).asClass)
+        val ctor = rowClassMirror.reflectConstructor(rowClassMirror.symbol.primaryConstructor.asMethod)
+
+        tag => ctor.apply(tag).asInstanceOf[Row]
+      }
+
+      class TableQuery extends EntTableQuery[K, V, Row](mkRow)
+
+      val Q: TableQuery = new TableQuery
+    }
   }
 }