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
  11 /**
  12  * Parses a [String] into a number of [Part]s.
  13  */
  14 class ParserFilter(private val core: Core, private val soneTextParser: SoneTextParser) : Filter {
  15
  16         override fun format(templateContext: TemplateContext?, data: Any?, parameters: MutableMap<String, Any?>?): Any? {
  17                 val text = data?.toString() ?: return listOf<Part>()
  18                 val soneParameter = parameters?.get("sone")
  19                 val sone = when (soneParameter) {
  20                         is String -> core.getSone(soneParameter).orNull()
  21                         is Sone -> soneParameter
  22                         else -> null
  23                 }
  24                 val context = SoneTextParserContext(sone as? Sone)
  25                 return soneTextParser.parse(text, context)
  26         }
  27
  28 }