dagger.hilt.processor.internal.BaseProcessingStep 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.
The newest version!
/*
 * 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 static dagger.internal.codegen.extension.DaggerStreams.toImmutableSet;

import androidx.room.compiler.processing.XElement;
import androidx.room.compiler.processing.XProcessingEnv;
import androidx.room.compiler.processing.XProcessingStep;
import androidx.room.compiler.processing.XRoundEnv;
import com.google.common.collect.ImmutableSet;
import com.squareup.javapoet.ClassName;
import java.util.Map;
import java.util.Set;

/**
 * Implements default configurations for ProcessingSteps, and provides structure for exception
 * handling.
 *
 * In each round it will do the following:
 *
 * 

 *   #preProcess()
 *   
foreach element:
 *       
 *         #processEach()
 *       
 *   
#postProcess()
 * 
 *
 * #processEach() allows each element to be processed, even if exceptions are thrown. Due to the
 * non-deterministic ordering of the processed elements, this is needed to ensure a consistent set
 * of exceptions are thrown with each build.
 */
public abstract class BaseProcessingStep implements XProcessingStep {
  private final ProcessorErrorHandler errorHandler;
  boolean isAnnotationClassNamesOverridden = true;

  private final XProcessingEnv processingEnv;

  public BaseProcessingStep(XProcessingEnv env) {
    errorHandler = new ProcessorErrorHandler(env);
    processingEnv = env;
  }

  protected final XProcessingEnv processingEnv() {
    return processingEnv;
  }

  @Override
  public final ImmutableSet annotations() {
    ImmutableSet annotationClassNames = annotationClassNames();
    if (!isAnnotationClassNamesOverridden) {
      return ImmutableSet.of("*");
    }
    if (annotationClassNames == null || annotationClassNames.isEmpty()) {
      throw new IllegalStateException("annotationClassNames() should return one or more elements.");
    } else {
      return annotationClassNames.stream().map(ClassName::canonicalName).collect(toImmutableSet());
    }
  }

  // When this method is not implemented by users, all annotated elements will processed by this
  // processing step.
  protected ImmutableSet annotationClassNames() {
    isAnnotationClassNamesOverridden = false;
    return ImmutableSet.of();
  }

  protected void processEach(ClassName annotation, XElement element) throws Exception {}

  protected void preProcess(XProcessingEnv env, XRoundEnv round) {}

  protected void postProcess(XProcessingEnv env, XRoundEnv round) throws Exception {}

  public final void preRoundProcess(XProcessingEnv env, XRoundEnv round) {
    preProcess(env, round);
  }

  public final void postRoundProcess(XProcessingEnv env, XRoundEnv round) {
    if (errorHandler.isEmpty()) {
      try {
        postProcess(env, round);
      } catch (Exception e) {
        errorHandler.recordError(e);
      }
    }
    if (!delayErrors() || round.isProcessingOver()) {
      errorHandler.checkErrors();
    }
  }

  @Override
  public final ImmutableSet process(
      XProcessingEnv env,
      Map> elementsByAnnotation,
      boolean isLastRound) {
    ImmutableSet.Builder elementsToReprocessBuilder = ImmutableSet.builder();
    for (ClassName annotationName : annotationClassNames()) {
      Set elements = elementsByAnnotation.get(annotationName.canonicalName());
      if (elements != null) {
        for (XElement element : elements) {
          try {
            processEach(annotationName, element);
          } catch (Exception e) {
            if (e instanceof ErrorTypeException && !isLastRound) {
              // Allow an extra round to reprocess to try to resolve this type.
              elementsToReprocessBuilder.add(element);
            } else {
              errorHandler.recordError(e);
            }
          }
        }
      }
    }
    return elementsToReprocessBuilder.build();
  }

  /**
   * Returns true if you want to delay errors to the last round. Useful if the processor generates
   * code for symbols used a lot in the user code. Delaying allows as much code to compile as
   * possible for correctly configured types and reduces error spam.
   */
  protected boolean delayErrors() {
    return false;
  }
}