aboutsummaryrefslogtreecommitdiff
path: root/plugins/base/src/main/kotlin/renderers/preprocessors.kt
blob: b64d2e1fa683d1a4f3c99b7f077760aa1ebf55a9 (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.base.renderers

import org.jetbrains.dokka.base.resolvers.shared.LinkFormat
import org.jetbrains.dokka.pages.*
import org.jetbrains.dokka.plugability.DokkaContext
import org.jetbrains.dokka.transformers.pages.PageTransformer

object RootCreator : PageTransformer {
    override fun invoke(input: RootPageNode) =
        RendererSpecificRootPage("", listOf(input), RenderingStrategy.DoNothing)
}

class PackageListCreator(
    val context: DokkaContext,
    val format: LinkFormat,
    val outputFilesNames: List<String> = listOf("package-list"),
    val removeModulePrefix: Boolean = true
) : PageTransformer {
    override fun invoke(input: RootPageNode) =
        input.modified(children = input.children.map {
            it.takeUnless { it is ModulePage }
                ?: it.modified(children = it.children + packageList(input, it as ModulePage))
        })


    private fun packageList(rootPageNode: RootPageNode, module: ModulePage): List<RendererSpecificPage> {
        val content = PackageListService(context, rootPageNode).createPackageList(
            module,
            format.formatName,
            format.linkExtension
        )
        return outputFilesNames.map { fileName ->
            RendererSpecificResourcePage(
                "${rootPageNode.name}/${fileName}",
                emptyList(),
                RenderingStrategy.Write(content)
            )
        }
    }
}