plugins/base/src/main/kotlin/resolvers/local/DefaultLocationProviderFactory.kt


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

package org.jetbrains.dokka.base.resolvers.local

import org.jetbrains.dokka.pages.MultimoduleRootPageNode
import org.jetbrains.dokka.pages.RootPageNode
import org.jetbrains.dokka.plugability.DokkaContext
import java.util.*
import java.util.concurrent.ConcurrentHashMap

class DefaultLocationProviderFactory(private val context: DokkaContext) : LocationProviderFactory {

    private val cache = ConcurrentHashMap<CacheWrapper, LocationProvider>()

    override fun getLocationProvider(pageNode: RootPageNode) = cache.computeIfAbsent(CacheWrapper(pageNode)) {
        if (pageNode.children.first() is MultimoduleRootPageNode) MultimoduleLocationProvider(pageNode, context)
        else DefaultLocationProvider(pageNode, context)
    }
}

private class CacheWrapper(val pageNode: RootPageNode) {
    override fun equals(other: Any?) = other is CacheWrapper && other.pageNode == this.pageNode

    override fun hashCode() = System.identityHashCode(pageNode)
}