aboutsummaryrefslogtreecommitdiff
path: root/plugins/xml/src/main/kotlin/XmlPlugin.kt
blob: 4a93206279175098f83a762f554f251c12546308 (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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
package org.jetbrains.dokka.xml

import org.jetbrains.dokka.CoreExtensions
import org.jetbrains.dokka.DefaultExtra
import org.jetbrains.dokka.Model.DocumentationNode
import org.jetbrains.dokka.Model.dfs
import org.jetbrains.dokka.links.DRI
import org.jetbrains.dokka.pages.*
import org.jetbrains.dokka.plugability.DokkaContext
import org.jetbrains.dokka.plugability.DokkaPlugin
import org.jetbrains.dokka.transformers.PageNodeTransformer

class XmlPlugin : DokkaPlugin() {
    val transformer by extending {
        CoreExtensions.pageTransformer with XmlTransformer
    }
}

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 XmlTransformer : PageNodeTransformer {
    enum class XMLKind : Kind {
        Main, XmlList
    }

    override fun invoke(original: ModulePageNode, dokkaContext: DokkaContext): ModulePageNode {
        original.findAll { it is ClassPageNode }.forEach { node ->
            val refs = node.documentationNode?.extra?.filterIsInstance<DefaultExtra>()?.filter { it.key == "@attr ref"}.orEmpty()
            val elementsToAdd = mutableListOf<DocumentationNode<*>>()

            refs.forEach { ref ->
                val toFind = DRI.from(ref.value)
                original.documentationNode?.dfs { it.dri == toFind }?.let { elementsToAdd.add(it) }
            }

            val refTable = ContentGroup(
                listOf(
                    ContentHeader(
                        listOf(
                            ContentText(
                                "XML Attributes",
                                DCI(node.dri, XMLKind.XmlList),
                                node.platforms().toSet(), emptySet(), emptySet()
                            )
                        ),
                        2,
                        DCI(node.dri, XMLKind.Main),
                        node.platforms().toSet(), emptySet(), emptySet()
                    ),
                    ContentTable(
                        emptyList(),
                        elementsToAdd.map {
                            ContentGroup(
                                listOf(
                                    ContentDRILink(
                                        listOf(
                                            ContentText(
                                                it.descriptors.first().name.toString(),
                                                DCI(node.dri, XMLKind.Main),
                                                node.platforms().toSet(), emptySet(), emptySet()
                                            )
                                        ),
                                        it.dri,
                                        DCI(it.dri, XMLKind.XmlList),
                                        it.platformData.toSet(), emptySet(), emptySet()
                                    ),
                                    ContentText(
                                        it.briefDocstring,
                                        DCI(it.dri, XMLKind.XmlList),
                                        it.platformData.toSet(), emptySet(), emptySet()
                                    )
                                ),
                                DCI(node.dri, XMLKind.XmlList),
                                node.platforms().toSet(), emptySet(), emptySet()
                            )
                        },
                        DCI(node.dri, XMLKind.XmlList),
                        node.platforms().toSet(), emptySet(), emptySet()
                    )
                ),
                DCI(node.dri, XMLKind.XmlList),
                node.platforms().toSet(), emptySet(), emptySet()
            )

            val content = node.content as ContentGroup
            val children = (node.content as ContentGroup).children
            node.content = content.copy(children = children + refTable)
        }
        return original
    }

    private fun PageNode.findAll(predicate: (PageNode) -> Boolean): Set<PageNode> {
        val found = mutableSetOf<PageNode>()
        if (predicate(this)) {
            found.add(this)
        } else {
            this.children.asSequence().mapNotNull { it.findAll(predicate) }.forEach { found.addAll(it) }
        }
        return found
    }

    private fun PageNode.platforms() = this.content.platforms.toList()
}