play.api.db.evolutions.EvolutionsHelper.scala Maven / Gradle / Ivy

/*
 * Copyright (C) from 2022 The Play Framework Contributors , 2011-2021 Lightbend Inc. 
 */

package play.api.db.evolutions

import java.util.regex.Matcher
import java.util.regex.Pattern

private[evolutions] object EvolutionsHelper {

  def substituteVariables(
      sql: String,
      substitutionsMappings: Map[String, String],
      prefix: String,
      suffix: String,
      escape: Boolean
  ): String = {
    var result: String = sql;
    for ((k, v) <- substitutionsMappings) yield {
      result =
        result.replaceAll("(?i)(^|[^!])" + Pattern.quote(prefix + k + suffix), "$1" + Matcher.quoteReplacement(v))
    }
    if (escape) {
      result.replaceAll(
        "(?i)" + Pattern.quote("!" + prefix) + "([^" + Pattern.quote(suffix) + "]*)" + Pattern.quote(suffix),
        Matcher.quoteReplacement(prefix) + "$1" + Matcher.quoteReplacement(suffix)
      )
    } else {
      result
    }
  }

  def applySchemaAndTable(sql: String, schema: String, table: String): String = {
    val withSchema = applySchema(sql, schema)
    applyTableName(withSchema, table)
  }

  def applyConfig(sql: String, config: EvolutionsDatasourceConfig): String = {
    applySchemaAndTable(sql, schema = config.schema, table = config.metaTable)
  }

  def applySchema(sql: String, schema: String): String = {
    sql.replaceAll("\\$\\{schema}", Option(schema).filter(_.trim.nonEmpty).map(_.trim + ".").getOrElse(""))
  }

  def applyTableName(sql: String, table: String): String = {
    sql.replaceAll("\\$\\{evolutions_table}", getPreparedTableName(table))
  }

  private def getPreparedTableName(tableName: String): String = {
    Option(tableName).filter(_.trim.nonEmpty).getOrElse("play_evolutions")
  }
}