io.javalin.rendering.template.JavalinJte.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of javalin-rendering Show documentation

There is a newer version: 6.4.0

/*
 * 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.template

import gg.jte.ContentType
import gg.jte.TemplateEngine
import gg.jte.output.StringOutput
import gg.jte.resolve.DirectoryCodeResolver
import io.javalin.http.Context
import io.javalin.http.servlet.isLocalhost
import io.javalin.rendering.FileRenderer
import io.javalin.rendering.JavalinRenderer
import io.javalin.rendering.util.RenderingDependency.JTE
import io.javalin.rendering.util.RenderingDependency.JTE_KOTLIN
import io.javalin.rendering.util.Util
import java.io.File

class JavalinJte @JvmOverloads constructor(
    private val templateEngine: TemplateEngine = defaultJteEngine(),
    private val isDevFunction: (Context) -> Boolean = { ctx -> ctx.isLocalhost() }
) : FileRenderer {

    private var isDev: Boolean? = null // cached and easily accessible, is set on first request (can't be configured directly by end user)

    override fun render(filePath: String, model: Map, ctx: Context): String {
        if (isDev == true && filePath.endsWith(".kte")) {
            Util.throwIfNotAvailable(JTE_KOTLIN)
        }
        isDev = isDev ?: isDevFunction(ctx)
        val stringOutput = StringOutput()
        templateEngine.render(filePath, model, stringOutput)
        return stringOutput.toString()
    }

    companion object {
        val extensions = arrayOf(".jte", ".kte")

        @JvmStatic
        @JvmOverloads
        fun init(templateEngine: TemplateEngine? = null, isDevFunction: ((Context) -> Boolean)? = null) {
            Util.throwIfNotAvailable(JTE)
            val fileRenderer = JavalinJte(templateEngine ?: defaultJteEngine(), isDevFunction ?: { it.isLocalhost() })
            JavalinRenderer.register(fileRenderer, *extensions)
        }

        private fun defaultJteEngine() = TemplateEngine.create(DirectoryCodeResolver(File("src/main/jte").toPath()), ContentType.Html)
    }

    class Loader : JavalinRenderer.FileRendererLoader {
        override fun load() = if (!JavalinRenderer.hasRenderer(*extensions) && JTE.exists()) init() else Unit
    }

}