.circumflex-orm-i18n.0.3.source-code.i18n.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of circumflex-orm-i18n
The newest version!
package ru.circumflex.orm.i18n

import ru.circumflex.orm._
import ru.circumflex.core._
import java.lang.String
import java.util.Locale
import ORM._
import ORMI18N._
import java.sql.{PreparedStatement, Connection}

class LocalizedTransaction extends StatefulTransaction
        with JDBCHelper{
  protected var _currentLanguage: String = null

  def setLanguage(lang: String): this.type = {
    cleanup()
    _currentLanguage = lang
    val q = i18nDialect.setLangQuery(_currentLanguage)
    auto(connection.prepareStatement(q))(st => {
      sqlLog.debug(q)
      st.executeUpdate
    })
    return this
  }

  def setLanguage(locale: Locale): this.type = setLanguage(locale.getLanguage)

  def currentLanguage: Option[String] =
    if (_currentLanguage == null) None
    else Some(_currentLanguage)

  def languageSet_?(): Boolean = _currentLanguage != null

  override def sql[A](sql: String)(actions: PreparedStatement => A): A = {
    if (!languageSet_?) setLanguage(ORMI18N.defaultLanguage)
    super.sql(sql)(actions)
  }
  
  override def dml[A](actions: Connection => A): A = {
    if (!languageSet_?) setLanguage(ORMI18N.defaultLanguage)
    super.dml(actions)
  }
}

class I18nTransactionManager extends TransactionManager {
  def openLocalizedTransaction: LocalizedTransaction = new LocalizedTransaction()
  def getLocalizedTransaction = getTransaction.asInstanceOf[LocalizedTransaction]
  override def openTransaction() = openLocalizedTransaction
}

/**
 * ORM Internationalization features configuration.
 */
object ORMI18N {
  val langSetting = Circumflex.cfg("orm.i18n.langSetting") match {
    case Some(s: String) => s
    case _ => "cx.lang"
  }
  val defaultLanguage = Circumflex.cfg("orm.i18n.defaultLanguage") match {
    case Some(l: String) => l.trim.toLowerCase
    case _ => "en"
  }
  val i18nDialect: I18NDialect = ORM.dialect match {
    case d: I18NDialect => d
    case _ => throw new ORMException("'orm.dialect' must point to valid I18NDialect.")
  }
  val i18nTransactionManager = ORM.transactionManager match {
    case tm: I18nTransactionManager => tm
    case _ => throw new ORMException("'orm.transactionManager' must point to valid I18nTransactionManager.")
  }
  def ltx: LocalizedTransaction = i18nTransactionManager.getLocalizedTransaction
}