src/main/kotlin/net/pterodactylus/sone/freenet/L10nFilter.kt

   1 /*
   2  * Sone - L10nFilter.java - Copyright © 2010–2019 David Roden
   3  *
   4  * This program is free software: you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation, either version 3 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16  */
  17
  18 package net.pterodactylus.sone.freenet
  19
  20 import net.pterodactylus.util.template.*
  21 import java.text.*
  22
  23 /**
  24  * [Filter] implementation replaces [String] values with their
  25  * translated equivalents.
  26  */
  27 class L10nFilter(private val translation: Translation) : Filter {
  28
  29         override fun format(templateContext: TemplateContext?, data: Any?, parameters: Map<String, Any>?): String {
  30                 val parameterValues = getParameters(data, parameters)
  31                 val text = getText(data)
  32                 return if (parameterValues.isEmpty()) {
  33                         translation.translate(text)
  34                 } else
  35                         MessageFormat(translation.translate(text), translation.currentLocale).format(parameterValues.toTypedArray())
  36         }
  37
  38         private fun getText(data: Any?) = (data as? L10nText)?.text ?: data.toString()
  39
  40         private fun getParameters(data: Any?, parameters: Map<String, Any>?) =
  41                         if (data is L10nText)
  42                                 data.parameters
  43                         else
  44                                 (parameters ?: emptyMap()).let { params ->
  45                                         generateSequence(0) { it + 1 }
  46                                                         .takeWhile { it.toString() in params }
  47                                                         .map { params[it.toString()] }
  48                                                         .toList()
  49                                 }
  50
  51 }