diff options
| -rw-r--r-- | core/src/main/kotlin/plugability/DokkaContext.kt | 33 |
1 files changed, 28 insertions, 5 deletions
diff --git a/core/src/main/kotlin/plugability/DokkaContext.kt b/core/src/main/kotlin/plugability/DokkaContext.kt index a5bcd695..b917ff99 100644 --- a/core/src/main/kotlin/plugability/DokkaContext.kt +++ b/core/src/main/kotlin/plugability/DokkaContext.kt @@ -8,7 +8,8 @@ import kotlin.reflect.KClass import kotlin.reflect.full.createInstance interface DokkaContext { - operator fun <T : Any, E : ExtensionPoint<T>> get(point: E): List<Extension<T>> + operator fun <T, E> get(point: E, askDefault: AskDefault = AskDefault.WhenEmpty): List<Extension<T>> + where T : Any, E : ExtensionPoint<T> fun <T : DokkaPlugin> plugin(kclass: KClass<T>): T? @@ -16,7 +17,7 @@ interface DokkaContext { companion object { fun create(pluginsClasspath: Iterable<File>, logger: DokkaLogger): DokkaContext = - DokkaContextConfigurationImpl(logger).apply { + DokkaContextConfigurationImpl(logger, DefaultContext(logger)).apply { pluginsClasspath.map { it.relativeTo(File(".").absoluteFile).toURI().toURL() } .toTypedArray() .let { URLClassLoader(it, this.javaClass.classLoader) } @@ -32,7 +33,8 @@ interface DokkaContextConfiguration { } private class DokkaContextConfigurationImpl( - override val logger: DokkaLogger + override val logger: DokkaLogger, + private val defaultContext: DokkaContext? ) : DokkaContext, DokkaContextConfiguration { private val plugins = mutableMapOf<KClass<*>, DokkaPlugin>() @@ -41,10 +43,17 @@ private class DokkaContextConfigurationImpl( internal val extensions = mutableMapOf<ExtensionPoint<*>, MutableList<Extension<*>>>() @Suppress("UNCHECKED_CAST") - override operator fun <T : Any, E : ExtensionPoint<T>> get(point: E) = extensions[point] as List<Extension<T>> + override operator fun <T, E> get(point: E, askDefault: AskDefault) where T : Any, E : ExtensionPoint<T> = + when (askDefault) { + AskDefault.Never -> extensions[point].orEmpty() + AskDefault.Always -> extensions[point].orEmpty() + defaultContext?.get(point, askDefault).orEmpty() + AskDefault.WhenEmpty -> + extensions[point]?.takeIf { it.isNotEmpty() } ?: defaultContext?.get(point, askDefault).orEmpty() + } as List<Extension<T>> @Suppress("UNCHECKED_CAST") - override fun <T : DokkaPlugin> plugin(kclass: KClass<T>) = (plugins[kclass] ?: pluginStubFor(kclass)) as T + override fun <T : DokkaPlugin> plugin(kclass: KClass<T>) = + (plugins[kclass] ?: defaultContext?.plugin(kclass) ?: pluginStubFor(kclass)) as T private fun <T : DokkaPlugin> pluginStubFor(kclass: KClass<T>): DokkaPlugin = pluginStubs.getOrPut(kclass) { kclass.createInstance().also { it.context = this } } @@ -78,4 +87,18 @@ private fun checkClasspath(classLoader: URLClassLoader) { "Please fix your plugins dependencies or exclude dokka api artifact from plugin classpath" ) } +} + +class DefaultContext(override val logger: DokkaLogger) : DokkaContext { + override fun <T : Any, E : ExtensionPoint<T>> get(point: E, askDefault: AskDefault): List<Extension<T>> = + when (point) { + + else -> emptyList() + } + + override fun <T : DokkaPlugin> plugin(kclass: KClass<T>): Nothing? = null +} + +enum class AskDefault { + Always, Never, WhenEmpty }
\ No newline at end of file |