jlengrand/kotlin
1
0
Fork 0
mirror of https://github.com/jlengrand/kotlin.git synced 2026-04-02 00:21:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

[FIR] Add ability to register user defined annotations for plugins

Browse Source
This commit is contained in:
Dmitriy Novozhilov
2020-04-27 18:02:09 +03:00
parent 8ecd3d8efe
commit 3d30ba9c19
2 changed files with 32 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
compiler/fir/resolve/src/org/jetbrains/kotlin/fir/extensions/FirExtensionPointService.kt
View File

@@ -7,15 +7,16 @@ package org.jetbrains.kotlin.fir.extensions
import com.google.common.collect.LinkedHashMultimap
import com.google.common.collect.Multimap
import org.jetbrains.kotlin.descriptors.ClassKind
import org.jetbrains.kotlin.fir.FirAnnotationContainer
import org.jetbrains.kotlin.fir.FirSession
import org.jetbrains.kotlin.fir.FirSessionComponent
import org.jetbrains.kotlin.fir.declarations.FirRegularClass
import org.jetbrains.kotlin.fir.expressions.FirAnnotationCall
import org.jetbrains.kotlin.fir.resolve.firSymbolProvider
import org.jetbrains.kotlin.fir.resolve.getSymbolByTypeRef
import org.jetbrains.kotlin.fir.symbols.impl.FirRegularClassSymbol
import org.jetbrains.kotlin.fir.utils.ComponentArrayOwner
import org.jetbrains.kotlin.fir.utils.ComponentTypeRegistry
import org.jetbrains.kotlin.fir.utils.*
import org.jetbrains.kotlin.name.FqName
import kotlin.properties.ReadOnlyProperty
import kotlin.reflect.KClass
@@ -58,6 +59,9 @@ class FirExtensionPointService(
val extensionsWithAllInAnnotatedMode = createMultimap<AnnotationFqn, P>()
for (extension in extensions) {
_metaAnnotations += extension.metaAnnotations
for (metaAnnotation in extension.metaAnnotations) {
extensionsWithMetaAnnotations.put(metaAnnotation, extension)
}
_annotations += extension.annotations
when (extension.mode) {
FirExtensionPoint.Mode.ANNOTATED_ELEMENT -> {
@@ -83,6 +87,25 @@ class FirExtensionPointService(
)
}
fun registerUserDefinedAnnotation(annotation: FirRegularClass) {
require(annotation.classKind == ClassKind.ANNOTATION_CLASS)
val fqName = annotation.symbol.classId.asSingleFqName()
_annotations += fqName
val extensions = extensionsWithMetaAnnotations[fqName]
if (extensions.isEmpty()) return
for (extension in extensions) {
val registeredExtensions = this[extension::class]
@Suppress("UNCHECKED_CAST")
val map = when (extension.mode) {
FirExtensionPoint.Mode.ANNOTATED_ELEMENT -> registeredExtensions.extensionsWithAnnotatedMode
FirExtensionPoint.Mode.ALL_IN_ANNOTATED_ELEMENT -> registeredExtensions.extensionsWithAllInAnnotatedMode
FirExtensionPoint.Mode.ALL -> throw IllegalStateException("Extension with mode ALL can't be subscribed to meta annotation")
} as Multimap<AnnotationFqn, FirExtensionPoint>
map.put(fqName, extension)
}
}
val annotations: Set<AnnotationFqn>
get() = _annotations
private val _annotations: MutableSet<AnnotationFqn> = mutableSetOf()
@@ -91,6 +114,8 @@ class FirExtensionPointService(
get() = _metaAnnotations
private val _metaAnnotations: MutableSet<AnnotationFqn> = mutableSetOf()
private val extensionsWithMetaAnnotations: Multimap<AnnotationFqn, FirExtensionPoint> = createMultimap()
class ExtensionsAccessor<P : FirExtensionPoint>(
private val session: FirSession,
private val extensions: FirRegisteredExtension<P>

6
compiler/fir/tree/src/org/jetbrains/kotlin/fir/utils/ComponentArray.kt
View File

@@ -16,6 +16,10 @@ abstract class ComponentArrayOwner<K : Any, V : Any> {
protected fun registerComponent(tClass: KClass<out K>, value: V) {
componentArray[(typeRegistry.getId(tClass))] = value
}
protected operator fun get(tClass: KClass<out K>): V {
return componentArray[typeRegistry.getId(tClass)]
}
}
@@ -60,7 +64,7 @@ class ComponentArray<T : Any> {
data[index] = value
}
operator fun get(index: Int): T? {
operator fun get(index: Int): T {
@Suppress("UNCHECKED_CAST")
return data.getOrNull(index) as T
}