org.http4k.template.RockerTemplates.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of http4k-template-rocker Show documentation

Http4k Rocker templating support

The newest version!

package org.http4k.template

import com.fizzed.rocker.Rocker
import com.fizzed.rocker.RockerModel
import com.fizzed.rocker.runtime.DefaultRockerModel
import com.fizzed.rocker.runtime.RockerRuntime

/**
 * Use this class as the extension class for all Template classes in the Rocker generation step
 */
abstract class RockerViewModel : DefaultRockerModel(), ViewModel {
    override fun template() = super.template() + ".rocker.html"
}

@Suppress("PARAMETER_NAME_CHANGED_ON_OVERRIDE")
class RockerTemplates : Templates {

    override fun CachingClasspath(unused: String) = object : TemplateRenderer {
        override fun invoke(p1: ViewModel): String = renderIt(p1)
    }

    override fun Caching(unused: String) = object : TemplateRenderer {
        override fun invoke(p1: ViewModel): String = renderIt(p1)
    }

    override fun HotReload(unused: String) = object : TemplateRenderer {
        init {
            RockerRuntime.getInstance().isReloading = true
        }

        override fun invoke(p1: ViewModel) = kotlin.runCatching {
            Rocker.template(p1.template())
        }.onFailure { throw ViewNotFound(p1) }
            .map {
                it
                    .bind(
                        p1.javaClass.declaredFields.mapNotNull {
                            try {
                                it.name to p1.javaClass.getDeclaredMethod(it.name)(p1)
                            } catch (e: NoSuchMethodException) {
                                null
                            }
                        }.toMap()
                    ).render().toString()
            }.getOrThrow()
    }

    private fun renderIt(p1: ViewModel) = if (p1 is RockerModel) p1.render().toString()
    else throw ViewNotFound(p1)
}