dagger.hilt.processor.internal.DaggerModels.kt Maven / Gradle / Ivy

Go to download

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

A fast dependency injector for Android and Java.

There is a newer version: 2.52

/*
 * Copyright (C) 2023 The Dagger Authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package dagger.hilt.processor.internal

import com.google.auto.common.MoreTypes
import com.google.devtools.ksp.symbol.KSAnnotated
import com.google.devtools.ksp.symbol.KSDeclaration
import com.squareup.javapoet.ClassName
import dagger.spi.model.DaggerAnnotation
import dagger.spi.model.DaggerElement
import dagger.spi.model.DaggerProcessingEnv
import dagger.spi.model.DaggerProcessingEnv.Backend.JAVAC
import dagger.spi.model.DaggerProcessingEnv.Backend.KSP
import dagger.spi.model.DaggerType


fun DaggerType.hasAnnotation(className: ClassName): Boolean =
  when (checkNotNull(backend())) {
    JAVAC -> {
      val javac = javac()
      Processors.hasAnnotation(MoreTypes.asTypeElement(javac), className)
    }
    KSP -> ksp().declaration.hasAnnotation(className.canonicalName())
  }

fun KSDeclaration.hasAnnotation(annotationName: String): Boolean =
  annotations.any {
    it.annotationType.resolve().declaration.qualifiedName?.asString().equals(annotationName)
  }

fun DaggerElement.hasAnnotation(className: ClassName) =
  when (checkNotNull(backend())) {
    JAVAC -> {
      val javac = javac()
      Processors.hasAnnotation(javac, className)
    }
    KSP -> ksp().hasAnnotation(className)
  }

fun DaggerAnnotation.getQualifiedName() =
  when (checkNotNull(backend())) {
    JAVAC -> {
      val javac = javac()
      MoreTypes.asTypeElement(javac.annotationType).qualifiedName.toString()
    }
    KSP -> ksp().annotationType.resolve().declaration.qualifiedName!!.asString()
  }

private fun KSAnnotated.hasAnnotation(className: ClassName) =
  annotations.any {
    it.annotationType.resolve().declaration.qualifiedName!!.asString() == className.canonicalName()
  }