Areas.scala

package domain.slick

import domain.model.{Area, Client, Id}
import slick.collection.heterogeneous.HNil
import slick.collection.heterogeneous.syntax._
import slick.driver.PostgresDriver.api._
import slick.lifted.{ProvenShape, Query, TableQuery, Tag}

class Areas(tag: Tag) extends Table[Area](tag, "areas") {

  def areaId: Rep[Int] = column[Int]("area_id", O.PrimaryKey)
  def clientId: Rep[Int] = column[Int]("client_id")
  def areaName: Rep[String] = column[String]("area_name")
  def deleted: Rep[Boolean] = column[Boolean]("deleted")

  import Areas._

  override def * : ProvenShape[Area] = (areaId :: clientId :: areaName :: deleted :: HNil) <>(applyArea, unapplyArea)

}

object Areas {

  protected type AreaHList = Int :: Int :: String :: Boolean :: HNil

  protected def applyArea(data: AreaHList): Area = data match {
    case areaId :: clientId :: areaName :: deleted :: HNil =>
      Area(areaId, clientId, areaName, deleted)
  }

  protected def unapplyArea(s: Area): Option[AreaHList] =
    Some(s.areaId :: s.clientId :: s.name :: s.deleted :: HNil)

  def table: Query[Areas, Area, Seq] = TableQuery[Areas]

  def byClient(clientId: Int): Query[Areas, Area, Seq] =
    table.filter(_.clientId === clientId)

  def byId(areaId: Int): Query[Areas, Area, Seq] =
    table.filter(_.areaId === areaId)

}