package space.kscience.visionforge.jupyter

import kotlinx.html.*
import org.jetbrains.kotlinx.jupyter.api.libraries.resources
import space.kscience.dataforge.context.Context
import space.kscience.dataforge.misc.DFExperimental
import space.kscience.gdml.Gdml
import space.kscience.plotly.Plot
import space.kscience.plotly.PlotlyPage
import space.kscience.plotly.StaticPlotlyRenderer
import space.kscience.tables.*
import space.kscience.visionforge.gdml.toVision
import space.kscience.visionforge.html.HtmlFragment
import space.kscience.visionforge.html.VisionPage
import space.kscience.visionforge.markup.MarkupPlugin
import space.kscience.visionforge.plotly.PlotlyPlugin
import space.kscience.visionforge.plotly.asVision
import space.kscience.visionforge.solid.Solids
import space.kscience.visionforge.tables.TableVisionPlugin
import space.kscience.visionforge.tables.toVision
import space.kscience.visionforge.visionManager


@DFExperimental
public class JupyterCommonIntegration : VisionForgeIntegration(CONTEXT.visionManager) {

    override fun Builder.afterLoaded() {

        resources {
            js("visionforge-common") {
                classPath("js/visionforge-jupyter-common.js")
            }
        }

        import(
            "space.kscience.gdml.*",
            "space.kscience.visionforge.solid.*",
            "space.kscience.tables.*",
            "space.kscience.dataforge.meta.*",
            "space.kscience.plotly.*",
            "space.kscience.plotly.models.*",
            "space.kscience.visionforge.plotly.plotly"
        )

        render<Gdml> { gdmlModel ->
            handler.produceHtml {
                vision { gdmlModel.toVision() }
            }
        }

        render<Table<*>> { table ->
            handler.produceHtml {
                vision { table.toVision() }
            }
        }

        render<Plot> { plot ->
            handler.produceHtml {
                vision { plot.asVision() }
            }
        }


        render<PlotlyPage> { plotlyPage ->
            val headers = plotlyPage.headers.associate { plotlyFragment ->
                plotlyFragment.hashCode().toString(16) to HtmlFragment {
                    plotlyFragment.visit(this)
                }

            }
            VisionPage(visionManager, headers) {
                div{
                    p { +"Plotly page renderer is not recommended in VisionForge, use `vf.page{}`" }
                }
                div {
                    plotlyPage.fragment.render.invoke(this, StaticPlotlyRenderer)
                }
            }
        }
    }

    public companion object {
        private val CONTEXT: Context = Context("Jupyter-common") {
            plugin(Solids)
            plugin(PlotlyPlugin)
            plugin(TableVisionPlugin)
            plugin(MarkupPlugin)
        }
    }
}