AndroidXHiltProcessor.kt

/*
 * Copyright 2020 The Android Open Source Project
 *
 * 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 androidx.hilt

import androidx.hilt.work.WorkerStep
import com.google.auto.service.AutoService
import net.ltgt.gradle.incap.IncrementalAnnotationProcessor
import net.ltgt.gradle.incap.IncrementalAnnotationProcessorType.ISOLATING
import javax.annotation.processing.AbstractProcessor
import javax.annotation.processing.Processor
import javax.annotation.processing.RoundEnvironment
import javax.lang.model.SourceVersion
import javax.lang.model.element.Element
import javax.lang.model.element.TypeElement

/**
 * Annotation processor for the various AndroidX Hilt extensions.
 */
@AutoService(Processor::class)
@IncrementalAnnotationProcessor(ISOLATING)
class AndroidXHiltProcessor : AbstractProcessor() {

    override fun getSupportedAnnotationTypes() = setOf(
        ClassNames.HILT_WORKER.canonicalName()
    )

    override fun getSupportedSourceVersion() = SourceVersion.latest()

    override fun process(
        annotations: MutableSet<out TypeElement>,
        roundEnv: RoundEnvironment
    ): Boolean {
        getSteps().forEach { step ->
            annotations.firstOrNull { it.qualifiedName.contentEquals(step.annotation()) }?.let {
                step.process(roundEnv.getElementsAnnotatedWith(it))
            }
        }
        return false
    }

    private fun getSteps() = listOf(
        WorkerStep(processingEnv)
    )

    interface Step {
        fun annotation(): String
        fun process(annotatedElements: Set<Element>)
    }
}