• Home
• io.github.nafg
• slick-additions-codegen_2.13
• 0.13.1
• source code
• MonocleLensesModelsCodeGenerator.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

slick.additions.codegen.extra.monocle.MonocleLensesModelsCodeGenerator.scala Maven / Gradle / Ivy

package slick.additions.codegen.extra.monocle

import scala.meta._

import slick.additions.codegen.ScalaMetaDsl._
import slick.additions.codegen.{ModelsCodeGenerator, TableConfig}


/** Annotates model classes with Monocle's `@Lenses`.
  *
  * Generated code requires `monocle-macro`
  */
trait MonocleLensesModelsCodeGenerator extends ModelsCodeGenerator {
  override protected def modelClass(tableConfig: TableConfig) =
    super
      .modelClass(tableConfig)
      .withMod(Mod.Annot(init(typ"Lenses")))

  override protected def rowStats(tableConfig: TableConfig) =
    List(
      modelClass(tableConfig),
      defObject(Term.Name(tableConfig.modelClassName))()
    )

  override protected def imports(strings: List[String]) = super.imports(strings :+ "monocle.macros.Lenses")
}