JsFrontendUtil.kt

/*
 * Copyright (c) 2021. JetBrains s.r.o.
 * Use of this source code is governed by the MIT license that can be found in the LICENSE file.
 */

package org.jetbrains.letsPlot.frontend

import jetbrains.datalore.plot.PlotHtmlHelper
import kotlinx.browser.document
import kotlinx.dom.createElement
import org.jetbrains.letsPlot.Figure
import org.jetbrains.letsPlot.intern.toSpec
import org.w3c.dom.*

object JsFrontendUtil {
    fun createPlotDiv(plot: Figure): HTMLDivElement {
        val spec = plot.toSpec()
        val html = PlotHtmlHelper.getStaticDisplayHtmlForRawSpec(spec)
        val divElement = document.createElement("div")
        setInnerHtml(divElement, html)
        return divElement as HTMLDivElement
    }

    fun setInnerHtml(element: Element, html: String) {
        element.innerHTML = html

        // Force browser to execute scripts in the "html".
        val allScripts = element.querySelectorAll("script")
        for (i in 0 until allScripts.length) {
            val oldScript = allScripts.get(i) as Element
            val newScript = document.createElement("script") {
                this is HTMLScriptElement
                oldScript.attributes.asList().forEach { attr ->
                    this.setAttribute(attr.name, attr.value)
                }

                this.appendChild(document.createTextNode(oldScript.innerHTML))
            }

            (oldScript.parentNode as Element).replaceChild(newScript, oldScript)
        }
    }
}