io.crysknife.generator.context.GenerationContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of crysknife-processor Show documentation
apt processor for crysknife-annotations project
The newest version!
/*
 * Copyright © 2020 Treblereel
 *
 * 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 io.crysknife.generator.context;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import javax.annotation.processing.ProcessingEnvironment;
import javax.annotation.processing.RoundEnvironment;
import javax.lang.model.element.TypeElement;
import javax.lang.model.util.Elements;
import javax.lang.model.util.Types;
import javax.tools.Diagnostic;

import io.crysknife.annotation.Application;
import io.crysknife.generator.context.oracle.ResourceOracle;
import io.crysknife.generator.context.oracle.ResourceOracleImpl;
import io.crysknife.logger.TreeLogger;
import io.github.classgraph.ClassGraph;
import io.github.classgraph.ScanResult;


/**
 * @author Dmitrii Tikhomirov Created by treblereel 2/21/19
 */
public class GenerationContext {

  private final TypeElement application;
  private final RoundEnvironment roundEnvironment;
  private final ProcessingEnvironment processingEnvironment;
  private final ScanResult scanResult;
  private final ResourceOracle resourceOracle = new ResourceOracleImpl(this);
  private ExecutionEnv executionEnv = ExecutionEnv.J2CL;

  private final String[] disabled = {"io.crysknife.generator", "io.crysknife.validation",
      "io.crysknife.util", "io.crysknife.task", "io.crysknife.processor", "io.crysknife.logger",
      "io.crysknife.definition", "io.crysknife.com.google", "io.crysknife.client.utils"};

  private final String[] enabled = {"io.crysknife"};

  public GenerationContext(TypeElement application, RoundEnvironment roundEnvironment,
      ProcessingEnvironment processingEnvironment, TreeLogger logger) {
    this.application = application;
    this.roundEnvironment = roundEnvironment;
    this.processingEnvironment = processingEnvironment;

    Application applicationAnnotation = application.getAnnotation(Application.class);

    if (applicationAnnotation.packages().length > 0) {
      List packages = new ArrayList<>(Arrays.asList(applicationAnnotation.packages()));
      packages.addAll(Arrays.asList(enabled));
      scanResult = new ClassGraph().acceptPackages(packages.toArray(new String[packages.size()]))
          .rejectPackages(disabled).enableMethodInfo().enableFieldInfo().enableClassInfo()
          .enableAnnotationInfo().scan();
    } else {
      scanResult = new ClassGraph().enableMethodInfo().enableFieldInfo().enableClassInfo()
          .enableAnnotationInfo().rejectPackages(disabled).scan();
    }

    /*    scanResult.getAllClasses().stream().forEach(classInfo -> {
      logger.branch(TreeLogger.Type.INFO, classInfo.getName());
    });*/


    logger.log(TreeLogger.Type.DEBUG,
        String.format("found classes   : %s", scanResult.getAllClasses().size()));
    logger.log(TreeLogger.Type.DEBUG,
        String.format("found resources : %s", scanResult.getAllResources().size()));
    logger.log(TreeLogger.Type.DEBUG,
        String.format("found interfaces: %s", scanResult.getAllInterfaces().size()));

    try {
      Class.forName("org.aspectj.lang.ProceedingJoinPoint");
      executionEnv = ExecutionEnv.JRE;
      processingEnvironment.getMessager().printMessage(Diagnostic.Kind.WARNING,
          "JRE generation mode.");
    } catch (ClassNotFoundException e) {

    }
    logger.log(TreeLogger.INFO, "Current generation mode: " + executionEnv);
  }

  public ExecutionEnv getExecutionEnv() {
    return executionEnv;
  }

  public Elements getElements() {
    return processingEnvironment.getElementUtils();
  }

  public Types getTypes() {
    return processingEnvironment.getTypeUtils();
  }

  public RoundEnvironment getRoundEnvironment() {
    return roundEnvironment;
  }

  public ProcessingEnvironment getProcessingEnvironment() {
    return processingEnvironment;
  }

  public ResourceOracle getResourceOracle() {
    return resourceOracle;
  }

  public ScanResult getScanResult() {
    return scanResult;
  }

  public TypeElement getApplication() {
    return application;
  }
}