i18n.i18n.scala Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of scala-activerecord_2.11 Show documentation

scala-activerecord

There is a newer version: 0.5.1

Show newest version

package com.github.aselab.activerecord.i18n

import java.util._
import java.io.InputStreamReader
import java.text.MessageFormat
import mojolly.inflector.InflectorImports._

trait Translator {
  def apply(key: String, args: Any*)(implicit locale: Locale): String =
    get(key, args:_*).getOrElse(key)

  def get(key: String, args: Any*)(implicit locale: Locale): Option[String]

  def field(model: Class[_], name: String)(implicit locale: Locale): String = {
    val key = "activerecord.models." + model.getSimpleName + "." + name
    get(key).getOrElse(name.titleize)
  }
}

object DefaultTranslator extends Translator {
  val utf8Control = new CharsetResourceBundleControl

  def bundle(implicit locale: Locale): ResourceBundle =
    ResourceBundle.getBundle("activerecord", locale, utf8Control)

  def get(key: String, args: Any*)(implicit locale: Locale): Option[String] = try {
    Option(bundle.getString(key)).map(msg =>
      MessageFormat.format(msg, args.map(_.asInstanceOf[AnyRef]):_*)
    )
  } catch {
    case e: MissingResourceException => None
  }
}

class CharsetResourceBundleControl(charset: String = "UTF8")
  extends ResourceBundle.Control
{
  override def newBundle(
    baseName: String, locale: Locale, format: String,
    loader: ClassLoader, reload: Boolean
  ): ResourceBundle = {
    val bundleName = toBundleName(baseName, locale)
    val resourceName = toResourceName(bundleName, "properties")

    val stream = if (reload) {
      Option(loader.getResource(resourceName)) map { url =>
        val conn = url.openConnection
        conn.setUseCaches(false)
        conn.getInputStream
      }
    } else {
      Option(loader.getResourceAsStream(resourceName))
    }

    stream.map { s =>
      val bundle = new PropertyResourceBundle(new InputStreamReader(s, charset))
      s.close
      bundle
    }.orNull
  }
}