src/main/kotlin/net/pterodactylus/sone/template/ParserFilter.kt

   1 package net.pterodactylus.sone.template
   2
   3 import net.pterodactylus.sone.core.Core
   4 import net.pterodactylus.sone.data.Sone
   5 import net.pterodactylus.sone.text.Part
   6 import net.pterodactylus.sone.text.SoneTextParser
   7 import net.pterodactylus.sone.text.SoneTextParserContext
   8 import net.pterodactylus.util.template.Filter
   9 import net.pterodactylus.util.template.TemplateContext
  10 import javax.inject.Inject
  11 import javax.inject.Singleton
  12
  13 /**
  14  * Parses a [String] into a number of [Part]s.
  15  */
  16 @Singleton
  17 class ParserFilter @Inject constructor(private val core: Core, private val soneTextParser: SoneTextParser) : Filter {
  18
  19         override fun format(templateContext: TemplateContext?, data: Any?, parameters: MutableMap<String, Any?>?): Any? {
  20                 val text = data?.toString() ?: return listOf<Part>()
  21                 val soneParameter = parameters?.get("sone")
  22                 val sone = when (soneParameter) {
  23                         is String -> core.getSone(soneParameter)
  24                         is Sone -> soneParameter
  25                         else -> null
  26                 }
  27                 val context = SoneTextParserContext(sone)
  28                 return soneTextParser.parse(text, context)
  29         }
  30
  31 }