aboutsummaryrefslogtreecommitdiff
path: root/dokka-android-gradle-plugin/src/main/kotlin/main.kt
blob: 054ed358d1948d32d9fad71bf9b554c6522e90c9 (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
package org.jetbrains.dokka.gradle

import com.android.build.gradle.AppExtension
import com.android.build.gradle.LibraryExtension
import com.android.build.gradle.api.BaseVariant
import org.gradle.api.Plugin
import org.gradle.api.Project
import org.gradle.api.tasks.SourceSet
import java.io.File

open class DokkaAndroidPlugin : Plugin<Project> {
    val allVariantsClassPath = mutableSetOf<File>()

    override fun apply(project: Project) {
        project.tasks.create("dokka", DokkaAndroidTask::class.java).apply {
            moduleName = project.name
            outputDirectory = File(project.buildDir, "dokka").absolutePath
        }

        if (project.hasAndroidPlugin()) {
            project.afterEvaluate {
                collectClasspath(project)
            }
        }
        else {
            project.plugins.whenPluginAdded {
                if (project.hasAndroidPlugin()) {
                    collectClasspath(project)
                }
            }
        }
    }

    private fun collectClasspath(project: Project) {
        val variants = project.collectAllVariants()
        variants.flatMapTo(allVariantsClassPath) { it.javaCompiler.classpath.files }
    }
}

open class DokkaAndroidTask : DokkaTask() {
    override val sdkProvider: SdkProvider? = AndroidSdkProvider(project)
}

private fun Project.hasAndroidPlugin() = plugins.hasPlugin("com.android.library") || plugins.hasPlugin("com.android.application")

private fun Project.findDokkaAndroidPlugin() = plugins.findPlugin(DokkaAndroidPlugin::class.java)

private fun Project.collectAllVariants(): Collection<BaseVariant> {
    extensions.findByType(LibraryExtension::class.java)?.let {
        return it.libraryVariants
    }
    extensions.findByType(AppExtension::class.java)?.let {
        return it.applicationVariants
    }
    return emptyList()
}

private class AndroidSdkProvider(private val project: Project) : SdkProvider {
    private val ext by lazy {
        project.extensions.findByType(LibraryExtension::class.java) ?: project.extensions.findByType(AppExtension::class.java)
    }

    override val name: String = "android"

    override val isValid: Boolean
        get() = project.hasAndroidPlugin()

    override val classpath: List<File>
        get() = ext.bootClasspath + (project.findDokkaAndroidPlugin()?.allVariantsClassPath ?: emptyList<File>())

    override val sourceDirs: Set<File>?
        get() {
            val sourceSet = ext?.sourceSets?.findByName(SourceSet.MAIN_SOURCE_SET_NAME)
            return sourceSet?.java?.srcDirs
        }
}