aboutsummaryrefslogtreecommitdiff
path: root/plugins/mathjax/src/main/kotlin/MathjaxPlugin.kt
blob: 9b3c25c28ef002496e23122544698f235d6a6534 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
package  org.jetbrains.dokka.mathjax

import model.doc.CustomTag
import org.jetbrains.dokka.CoreExtensions
import org.jetbrains.dokka.pages.ModulePageNode
import org.jetbrains.dokka.pages.PageNode
import org.jetbrains.dokka.plugability.DokkaContext
import org.jetbrains.dokka.plugability.DokkaPlugin
import org.jetbrains.dokka.transformers.pages.PageNodeTransformer

class MathjaxPlugin : DokkaPlugin() {
    val transformer by extending {
        CoreExtensions.pageTransformer with MathjaxTransformer
    }
}

private const val ANNOTATION = "@usesMathJax"
private const val LIB_PATH = "https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.6/MathJax.js?config=TeX-AMS_SVG&latest"

object MathjaxTransformer : PageNodeTransformer {
    override fun invoke(input: ModulePageNode, dokkaContext: DokkaContext) = input.modified(
        children = input.children.map { transform(it) }
    )

    private fun transform(input: PageNode): PageNode = input.modified(
        embeddedResources = input.embeddedResources + if (input.isNeedingMathjax) listOf(LIB_PATH) else emptyList(),
        children = input.children.map { transform(it) }
    )


    private val PageNode.isNeedingMathjax
        get() = documentable?.platformInfo
            ?.flatMap { it.documentationNode.children }
            .orEmpty()
            .filterIsInstance<CustomTag>()
            .let { true }
//            .any { it.name.contains("ref") && it.root.children.any {it.name} }


}