io.javalin.rendering.JavalinRenderer.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hudi-spark-bundle Show documentation
The newest version!
/*
 * Javalin - https://javalin.io
 * Copyright 2017 David Åse
 * Licensed under Apache 2.0: https://github.com/tipsy/javalin/blob/master/LICENSE
 */

package io.javalin.rendering

import io.javalin.rendering.markdown.JavalinCommonmark
import io.javalin.rendering.template.*
import org.slf4j.LoggerFactory

object JavalinRenderer {

    private val log = LoggerFactory.getLogger(JavalinRenderer.javaClass)

    private val extensions = mutableMapOf()

    init {
        register(JavalinVelocity, ".vm", ".vtl")
        register(JavalinFreemarker, ".ftl")
        register(JavalinMustache, ".mustache")
        register(JavalinJtwig, ".jtwig", ".twig", ".html.twig")
        register(JavalinPebble, ".peb", ".pebble")
        register(JavalinThymeleaf, ".html", ".tl", ".thyme", ".thymeleaf")
        register(JavalinCommonmark, ".md", ".markdown")
    }

    fun renderBasedOnExtension(filePath: String, model: Map): String {
        val extension = if (filePath.hasTwoDots) filePath.doubleExtension else filePath.extension
        val renderer = extensions[extension]
                ?: extensions[filePath.extension] // fallback to a non-double extension
                ?: throw IllegalArgumentException("No Renderer registered for extension '${filePath.extension}'.")
        return renderer.render(filePath, model)
    }

    @JvmStatic
    fun register(fileRenderer: FileRenderer, vararg ext: String) = ext.forEach {
        if (extensions[it] != null) {
            log.info("'$it' is already registered to ${extensions[it]!!.javaClass}. Overriding.")
        }
        extensions[it] = fileRenderer
    }

    private val String.extension: String get() = this.replaceBeforeLast(".", "")
    private val String.doubleExtension: String get() = this.substringBeforeLast(".", "").extension + this.extension
    private val String.hasTwoDots: Boolean get() = this.count { it == '.' } > 1
}