skinny.task.generator.ReverseGenerator.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of skinny-task_2.11 Show documentation

skinny-task

There is a newer version: 3.1.0

package skinny.task.generator

import scalikejdbc.DB
import scalikejdbc.metadata.{ Table, ForeignKey }

trait ReverseGenerator extends CodeGenerator {

  def extractAssociationParams(tableName: String, pkName: Option[String], cachedTables: Seq[Table]): Seq[String] = {

    // belongsTo associations
    val foreignKeys: Seq[ForeignKey] = {
      val table: Option[Table] = {
        cachedTables.find((t) => t.name.toLowerCase == tableName.toLowerCase).orElse(DB.getTable(tableName))
      }
      table.map(_.foreignKeys.toSeq).getOrElse(Nil)
    }
    val belongsToAssociations: Seq[String] = foreignKeys.map { fk =>
      val name = toCamelCase(fk.name.toLowerCase).replace(toFirstCharUpper(fk.foreignColumnName.toLowerCase), "")
      s"${name}:Option[${toFirstCharUpper(name)}]"
    }

    // hasMany associations
    val hasManyAssociations: Seq[String] = {
      if (cachedTables.isEmpty) Nil
      else {
        val expectedFkName = s"${toFirstCharLower(toCamelCase(tableName))}${pkName.map(toFirstCharUpper).getOrElse("Id")}"
        val hasManyTables: Seq[Table] = cachedTables.filter { table =>
          table.foreignKeys.exists { (fk) =>
            fk.foreignTableName.toLowerCase == tableName.toLowerCase &&
              toCamelCase(fk.name.toLowerCase) == expectedFkName
          }
        }
        hasManyTables.map { table =>
          val name = toCamelCase(table.name.toLowerCase)
          s"${name}s:Seq[${toClassName(name)}]"
        }
      }
    }

    belongsToAssociations ++ hasManyAssociations
  }

}