godot.entrygenerator.extension.PropertyDescriptorExt.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of godot-kotlin-entry-generator Show documentation
Godot Kotlin entry code generator.
The newest version!
package godot.entrygenerator.extension

import godot.entrygenerator.EntryGenerator
import godot.entrygenerator.model.REGISTER_PROPERTY_ANNOTATION
import org.jetbrains.kotlin.descriptors.PropertyDescriptor
import org.jetbrains.kotlin.descriptors.annotations.AnnotationDescriptor
import org.jetbrains.kotlin.name.FqName
import org.jetbrains.kotlin.psi.KtExpression
import org.jetbrains.kotlin.psi.KtProperty
import org.jetbrains.kotlin.resolve.descriptorUtil.fqNameSafe
import org.jetbrains.kotlin.resolve.source.KotlinSourceElement
import org.jetbrains.kotlin.serialization.deserialization.descriptors.DeserializedPropertyDescriptor

fun PropertyDescriptor.getPropertyHintAnnotation(): AnnotationDescriptor? {
    val propertyHintAnnotations = propertyHintAnnotations
        .map { FqName(it) }
        .filter { annotations.hasAnnotation(it) }
        .map { annotations.findAnnotation(it) }

    if (propertyHintAnnotations.size > 1) {
        throw IllegalStateException("The property ${this.name} has multiple PropertyHintAnnotations. You can only specify one! Defined Annotations: $propertyHintAnnotations")
    }

    if (propertyHintAnnotations.isNotEmpty() && !annotations.hasAnnotation(FqName(REGISTER_PROPERTY_ANNOTATION))) {
        throw IllegalStateException("The property ${this.name} has a ${propertyHintAnnotations.first()?.fqName} annotation but is not annotated with @RegisterProperty. Add the @RegisterProperty annotation or remove the ${propertyHintAnnotations.first()?.fqName} annotation")
    }

    return propertyHintAnnotations.firstOrNull()
}

val PropertyDescriptor.assignmentPsi: KtExpression
    get() = if (this is DeserializedPropertyDescriptor) { //incremental compilation
        val psiKtClassWithMembers = EntryGenerator
            .psiClassesWithMembers
            .firstOrNull { psiClassWithMembers -> psiClassWithMembers.ktClass.fqName?.asString() == containingDeclaration.fqNameSafe.asString() }

        val ktProperty = psiKtClassWithMembers
            ?.properties
            ?.firstOrNull { ktProperty -> ktProperty.name == this.name.asString() }
            ?: psiKtClassWithMembers
                ?.signals
                ?.firstOrNull { ktProperty -> ktProperty.name == this.name.asString() }

        requireNotNull(
            ktProperty
                ?.initializer
                ?: ktProperty
                    ?.delegateExpression
        ) { "Property assignment of property $fqNameSafe cannot be null" }
    } else {
        requireNotNull(
            ((this
                .source as KotlinSourceElement)
                .psi as KtProperty)
                .delegateExpressionOrInitializer
        ) { "Property assignment of property $fqNameSafe cannot be null" }
    }


private val propertyHintAnnotations: List =
    listOf(
        "godot.annotation.ColorNoAlpha",
        "godot.annotation.Dir",
        "godot.annotation.DoubleRange",
        "godot.annotation.EnumFlag",
        "godot.annotation.ExpEasing",
        "godot.annotation.ExpRange",
        "godot.annotation.File",
        "godot.annotation.Flags",
        "godot.annotation.FloatRange",
        "godot.annotation.ImageCompressLossLess",
        "godot.annotation.ImageCompressLossy",
        "godot.annotation.IntFlag",
        "godot.annotation.IntIsObjectId",
        "godot.annotation.IntRange",
        "godot.annotation.Layers2DPhysics",
        "godot.annotation.Layers2DRender",
        "godot.annotation.Layers3DPhysics",
        "godot.annotation.Layers3DRender",
        "godot.annotation.Lenght",
        "godot.annotation.Max",
        "godot.annotation.MethodOfBaseType",
        "godot.annotation.MethodOfInstance",
        "godot.annotation.MethodOfScript",
        "godot.annotation.MethodOfVariantType",
        "godot.annotation.MultilineText",
        "godot.annotation.NodePathToEditedNode",
        "godot.annotation.NodePathValidTypes",
        "godot.annotation.ObjectId",
        "godot.annotation.ObjectTooBig",
        "godot.annotation.PlaceHolderText",
        "godot.annotation.PropertyOfBaseType",
        "godot.annotation.PropertyOfInstance",
        "godot.annotation.PropertyOfScript",
        "godot.annotation.PropertyOfVariantType",
        "godot.annotation.ResourceType",
        "godot.annotation.SaveFile",
        "godot.annotation.TypeString"
    )