81 lines
3.0 KiB
Kotlin
81 lines
3.0 KiB
Kotlin
|
package space.kscience.visionforge.jupyter
|
||
|
|
|
||
|
|
import io.ktor.server.engine.ApplicationEngine
|
||
|
|
import kotlinx.html.FORM
|
||
|
|
import kotlinx.html.p
|
||
|
|
import kotlinx.html.stream.createHTML
|
||
|
|
import kotlinx.html.style
|
||
|
|
import org.jetbrains.kotlinx.jupyter.api.HTML
|
||
|
|
import org.jetbrains.kotlinx.jupyter.api.MimeTypedResult
|
||
|
|
import space.kscience.dataforge.context.Context
|
||
|
|
import space.kscience.dataforge.context.ContextAware
|
||
|
|
import space.kscience.dataforge.context.info
|
||
|
|
import space.kscience.dataforge.context.logger
|
||
|
|
import space.kscience.dataforge.meta.get
|
||
|
|
import space.kscience.dataforge.meta.int
|
||
|
|
import space.kscience.dataforge.meta.string
|
||
|
|
import space.kscience.visionforge.html.HtmlFormFragment
|
||
|
|
import space.kscience.visionforge.html.HtmlFragment
|
||
|
|
import space.kscience.visionforge.html.HtmlVisionFragment
|
||
|
|
import space.kscience.visionforge.html.visionFragment
|
||
|
|
import space.kscience.visionforge.three.server.VisionServer
|
||
|
|
import space.kscience.visionforge.three.server.serve
|
||
|
|
import space.kscience.visionforge.visionManager
|
||
|
|
|
||
|
|
public class VisionForgeServerHandler(override val context: Context) : ContextAware {
|
||
|
|
private var counter = 0
|
||
|
|
|
||
|
|
private var engine: ApplicationEngine? = null
|
||
|
|
private var server: VisionServer? = null
|
||
|
|
|
||
|
|
public var isolateFragments: Boolean = false
|
||
|
|
|
||
|
|
public fun legacyMode() {
|
||
|
|
isolateFragments = true
|
||
|
|
}
|
||
|
|
|
||
|
|
public fun startServer(
|
||
|
|
host: String = context.properties["visionforge.host"].string ?: "localhost",
|
||
|
|
port: Int = context.properties["visionforge.port"].int ?: VisionServer.DEFAULT_PORT,
|
||
|
|
configuration: VisionServer.() -> Unit = {},
|
||
|
|
): HtmlFragment {
|
||
|
|
engine?.stop(1000, 2000)
|
||
|
|
engine = context.visionManager.serve(host, port) {
|
||
|
|
configuration()
|
||
|
|
server = this
|
||
|
|
}.start()
|
||
|
|
return {
|
||
|
|
if(server!= null){
|
||
|
|
p {
|
||
|
|
style = "color: red;"
|
||
|
|
+"Stopping current VisionForge server"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
p {
|
||
|
|
style = "color: blue;"
|
||
|
|
+"Starting VisionForge server on http://$host:$port"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public fun stopServer() {
|
||
|
|
engine?.apply {
|
||
|
|
logger.info { "Stopping VisionForge server" }
|
||
|
|
}?.stop(1000, 2000)
|
||
|
|
}
|
||
|
|
|
||
|
|
private fun produceHtmlString(
|
||
|
|
fragment: HtmlVisionFragment,
|
||
|
|
): String = server?.serveVisionsFromFragment("content[${counter++}]", fragment)
|
||
|
|
?: createHTML().apply {
|
||
|
|
visionFragment(context.visionManager, fragment = fragment)
|
||
|
|
}.finalize()
|
||
|
|
|
||
|
|
public fun produceHtml(isolated: Boolean? = null, fragment: HtmlVisionFragment): MimeTypedResult =
|
||
|
|
HTML(produceHtmlString(fragment), isolated ?: isolateFragments)
|
||
|
|
|
||
|
|
public fun fragment(body: HtmlVisionFragment): MimeTypedResult = produceHtml(fragment = body)
|
||
|
|
public fun page(body: HtmlVisionFragment): MimeTypedResult = produceHtml(true, body)
|
||
|
|
|
||
|
|
public fun form(builder: FORM.() -> Unit): HtmlFormFragment = HtmlFormFragment("form[${counter++}]", builder = builder)
|
||
|
|
}
|