skinny.task.generator.DBMigrationFileGenerator.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 java.io.File
import org.joda.time.DateTime

/**
 * DB migration file generator.
 */
object DBMigrationFileGenerator extends DBMigrationFileGenerator

trait DBMigrationFileGenerator extends CodeGenerator {

  private[this] def showUsage = {
    showSkinnyGenerator()
    println("""  Usage: sbt "task/run generate:migration Create_members_table 'create table members(id bigserial not null primary key, name varchar(255) not null);'" """)
    println("")
  }

  def run(args: List[String]) {
    args.toList match {
      case name :: sqlParts =>
        showSkinnyGenerator()
        generate(name, sqlParts.map(_.replaceFirst("^'", "").replaceFirst("'$", "")).mkString(" "))
        println("")
      case _ => showUsage
    }
  }

  def generate(name: String, sql: String): Unit = {
    val version = DateTime.now.toString("yyyyMMddHHmmss")
    val file = new File(s"${resourceDir}/db/migration/V${version}__${name}.sql")
    writeIfAbsent(file, sql)
  }

}