com.freeletics.mad.whetstone.codegen.util.Annotations.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of whetstone-compiler Show documentation

Whetstone Compiler

There is a newer version: 0.14.1

@file:OptIn(ExperimentalAnvilApi::class)
package com.freeletics.mad.whetstone.codegen.util

import com.squareup.anvil.annotations.ExperimentalAnvilApi
import com.squareup.anvil.compiler.internal.reference.AnnotatedReference
import com.squareup.anvil.compiler.internal.reference.AnnotationReference
import com.squareup.anvil.compiler.internal.reference.AnvilCompilationExceptionAnnotationReference
import com.squareup.anvil.compiler.internal.reference.ClassReference
import com.squareup.anvil.compiler.internal.reference.argumentAt
import com.squareup.anvil.compiler.internal.reference.asClassName
import com.squareup.kotlinpoet.ClassName
import org.jetbrains.kotlin.descriptors.ModuleDescriptor
import org.jetbrains.kotlin.name.FqName
import org.jetbrains.kotlin.psi.KtConstantExpression

internal fun AnnotatedReference.findAnnotation(fqName: FqName): AnnotationReference? {
    return annotations.find { it.fqName == fqName }
}

internal fun AnnotationReference.requireClassArgument(name: String, index: Int): ClassName {
    return optionalClassArgument(name, index) ?:
        throw AnvilCompilationExceptionAnnotationReference(this, "Couldn't find $name for $fqName")
}

internal fun AnnotationReference.optionalClassArgument(name: String, index: Int): ClassName? {
    return argumentAt(name, index)?.value()?.asClassName()
}

internal fun AnnotationReference.requireEnumArgument(name: String, index: Int): String {
    return argumentAt(name, index)?.value()?.shortName()?.asString() ?:
        throw AnvilCompilationExceptionAnnotationReference(this, "Couldn't find $name for $fqName")
}

internal fun AnnotationReference.optionalBooleanArgument(name: String, index: Int): Boolean? {
    return argumentAt(name, index)?.value()
}