1 /* 2 * Copyright (C) 2023 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.processor.internal; 18 19 import androidx.room.compiler.processing.XProcessingEnv; 20 import androidx.room.compiler.processing.XProcessingStep; 21 import androidx.room.compiler.processing.XRoundEnv; 22 import androidx.room.compiler.processing.ksp.KspBasicAnnotationProcessor; 23 import com.google.common.collect.ImmutableList; 24 import com.google.devtools.ksp.processing.SymbolProcessorEnvironment; 25 26 /** A KspBasicAnnotationProcessor that contains a single BaseProcessingStep. */ 27 public abstract class KspBaseProcessingStepProcessor extends KspBasicAnnotationProcessor { 28 private BaseProcessingStep processingStep; 29 KspBaseProcessingStepProcessor(SymbolProcessorEnvironment symbolProcessorEnvironment)30 public KspBaseProcessingStepProcessor(SymbolProcessorEnvironment symbolProcessorEnvironment) { 31 super(symbolProcessorEnvironment, HiltProcessingEnvConfigs.CONFIGS); 32 } 33 34 @Override initialize(XProcessingEnv env)35 public void initialize(XProcessingEnv env) { 36 HiltCompilerOptions.checkWrongAndDeprecatedOptions(env); 37 processingStep = processingStep(); 38 } 39 processingStep()40 protected abstract BaseProcessingStep processingStep(); 41 42 @Override preRound(XProcessingEnv env, XRoundEnv round)43 public void preRound(XProcessingEnv env, XRoundEnv round) { 44 processingStep.preRoundProcess(env, round); 45 } 46 47 @Override processingSteps()48 public final ImmutableList<XProcessingStep> processingSteps() { 49 return ImmutableList.of(processingStep); 50 } 51 52 @Override postRound(XProcessingEnv env, XRoundEnv round)53 public void postRound(XProcessingEnv env, XRoundEnv round) { 54 processingStep.postRoundProcess(env, round); 55 } 56 } 57