1 /*
<lambda>null2  * Copyright (C) 2020 The Dagger Authors.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package dagger.hilt.android.processor.internal.viewmodel
18 
19 import com.google.auto.common.MoreElements
20 import com.google.auto.service.AutoService
21 import dagger.hilt.android.processor.internal.AndroidClassNames
22 import dagger.hilt.processor.internal.BaseProcessor
23 import javax.annotation.processing.Processor
24 import javax.annotation.processing.RoundEnvironment
25 import javax.lang.model.SourceVersion
26 import javax.lang.model.element.Element
27 import javax.lang.model.element.TypeElement
28 import net.ltgt.gradle.incap.IncrementalAnnotationProcessor
29 import net.ltgt.gradle.incap.IncrementalAnnotationProcessorType
30 
31 /**
32  * Annotation processor for @ViewModelInject.
33  */
34 @AutoService(Processor::class)
35 @IncrementalAnnotationProcessor(IncrementalAnnotationProcessorType.ISOLATING)
36 class ViewModelProcessor : BaseProcessor() {
37 
38   private val parsedElements = mutableSetOf<TypeElement>()
39 
40   override fun getSupportedAnnotationTypes() = setOf(
41     AndroidClassNames.HILT_VIEW_MODEL.toString()
42   )
43 
44   override fun getSupportedSourceVersion() = SourceVersion.latest()
45 
46   override fun processEach(annotation: TypeElement, element: Element) {
47     val typeElement = MoreElements.asType(element)
48     if (parsedElements.add(typeElement)) {
49       ViewModelMetadata.create(
50         processingEnv,
51         typeElement,
52       )?.let { viewModelMetadata ->
53         ViewModelModuleGenerator(
54           processingEnv,
55           viewModelMetadata
56         ).generate()
57       }
58     }
59   }
60 
61   override fun postRoundProcess(roundEnv: RoundEnvironment?) {
62     parsedElements.clear()
63   }
64 }
65