azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

New J2K: handle more types of nullability annotations

Browse Source
This commit is contained in:
Ilya Kirillov
2019-04-08 18:45:57 +03:00
parent d20e4bac1d
commit af17db5f45
1 changed files with 30 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nj2k/src/org/jetbrains/kotlin/nj2k/conversions/JetbrainsNullableAnnotationsConverter.kt
+30 -30
View File
@@ -6,6 +6,8 @@
package org.jetbrains.kotlin.nj2k.conversions
import org.jetbrains.kotlin.j2k.ast.Nullability
import org.jetbrains.kotlin.load.java.NOT_NULL_ANNOTATIONS
import org.jetbrains.kotlin.load.java.NULLABLE_ANNOTATIONS
import org.jetbrains.kotlin.nj2k.ConversionContext
import org.jetbrains.kotlin.nj2k.tree.*
import org.jetbrains.kotlin.nj2k.tree.impl.JKTypeElementImpl
@@ -14,40 +16,38 @@ import org.jetbrains.kotlin.nj2k.tree.impl.JKTypeElementImpl
class JetbrainsNullableAnnotationsConverter(private val context: ConversionContext) : RecursiveApplicableConversionBase() {
override fun applyToElement(element: JKTreeElement): JKTreeElement {
if (element !is JKAnnotationListOwner) return recurse(element)
val nullableAnnotationSymbol =
context.symbolProvider.provideByFqName("org.jetbrains.annotations.Nullable")
val notNullAnnotationSymbol =
context.symbolProvider.provideByFqName("org.jetbrains.annotations.NotNull")
val nullableAnnotation =
element.annotationList.annotations.firstOrNull { it.classSymbol == nullableAnnotationSymbol }
val notNullAnnotation =
element.annotationList.annotations.firstOrNull { it.classSymbol == notNullAnnotationSymbol }
when (element) {
is JKVariable -> {
if (nullableAnnotation != null) {
element.annotationList.annotations -= nullableAnnotation
element.type =
JKTypeElementImpl(element.type.type.updateNullability(Nullability.Nullable))
val annotationsToRemove = mutableListOf<JKAnnotation>()
for (annotation in element.annotationList.annotations) {
val nullability = annotation.annotationNullability() ?: continue
when (element) {
is JKVariable -> {
annotationsToRemove += annotation
element.type = JKTypeElementImpl(element.type.type.updateNullability(nullability))
}
if (notNullAnnotation != null) {
element.annotationList.annotations -= notNullAnnotation
element.type =
JKTypeElementImpl(element.type.type.updateNullability(Nullability.NotNull))
}
}
is JKMethod -> {
if (nullableAnnotation != null) {
element.annotationList.annotations -= nullableAnnotation
element.returnType =
JKTypeElementImpl(element.returnType.type.updateNullability(Nullability.Nullable))
}
if (notNullAnnotation != null) {
element.annotationList.annotations -= notNullAnnotation
element.returnType =
JKTypeElementImpl(element.returnType.type.updateNullability(Nullability.NotNull))
is JKMethod -> {
annotationsToRemove += annotation
element.returnType = JKTypeElementImpl(element.returnType.type.updateNullability(nullability))
}
}
}
element.annotationList.annotations -= annotationsToRemove
return recurse(element)
}
private fun JKAnnotation.annotationNullability(): Nullability? =
when {
classSymbol.fqName in nullableAnnotationsFqNames -> Nullability.Nullable
classSymbol.fqName in notNullAnnotationsFqNames -> Nullability.NotNull
else -> null
}
companion object {
val nullableAnnotationsFqNames =
NULLABLE_ANNOTATIONS.map { it.asString() }.toSet()
val notNullAnnotationsFqNames =
NOT_NULL_ANNOTATIONS.map { it.asString() }.toSet()
}
}