package space.kscience.visionforge.gdml.jupyter

import org.w3c.dom.Element
import react.child
import space.kscience.dataforge.context.AbstractPlugin
import space.kscience.dataforge.context.Context
import space.kscience.dataforge.context.PluginFactory
import space.kscience.dataforge.context.PluginTag
import space.kscience.dataforge.meta.Meta
import space.kscience.dataforge.names.Name
import space.kscience.dataforge.names.asName
import space.kscience.visionforge.ElementVisionRenderer
import space.kscience.visionforge.Vision
import space.kscience.visionforge.solid.Solid
import space.kscience.visionforge.solid.three.ThreePlugin
import kotlin.reflect.KClass

class ThreeWithControls : AbstractPlugin(), ElementVisionRenderer {
    val three by require(ThreePlugin)

    override val tag: PluginTag get() = Companion.tag

    override fun rateVision(vision: Vision): Int =
        if (vision is Solid) ElementVisionRenderer.DEFAULT_RATING * 2 else ElementVisionRenderer.ZERO_RATING

    override fun render(element: Element, vision: Vision, meta: Meta) {
        react.dom.render(element) {
            child(GdmlView) {
                attrs {
                    this.context = this@ThreeWithControls.context
                    this.rootVision = vision
                }
            }
        }
    }

    override fun content(target: String): Map<Name, Any> {
        return when (target) {
            ElementVisionRenderer.TYPE -> mapOf("three.withControls".asName() to this)
            else -> super.content(target)
        }
    }

    companion object : PluginFactory<ThreeWithControls> {
        override val tag: PluginTag = PluginTag("vision.threejs.withControls", PluginTag.DATAFORGE_GROUP)
        override val type: KClass<ThreeWithControls> = ThreeWithControls::class
        override fun invoke(meta: Meta, context: Context): ThreeWithControls = ThreeWithControls()
    }
}