com.github.nalukit.nalu.processor.scanner.ModulesAnnotationScanner Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nalu-processor Show documentation
Nalu's Processor
The newest version!
/*
 * Copyright (c) 2018 - Frank Hossfeld
 *
 *  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 com.github.nalukit.nalu.processor.scanner;

import com.github.nalukit.nalu.client.module.annotation.Modules;
import com.github.nalukit.nalu.processor.model.MetaModel;
import com.github.nalukit.nalu.processor.model.intern.ClassNameModel;

import javax.annotation.processing.ProcessingEnvironment;
import javax.annotation.processing.RoundEnvironment;
import javax.lang.model.element.Element;
import javax.lang.model.type.TypeMirror;
import java.util.Arrays;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

public class ModulesAnnotationScanner {

  private final ProcessingEnvironment processingEnvironment;

  private final MetaModel metaModel;

  private final Element modulesElement;

  @SuppressWarnings("unused")
  private ModulesAnnotationScanner(Builder builder) {
    super();
    this.processingEnvironment = builder.processingEnvironment;
    this.metaModel             = builder.metaModel;
    this.modulesElement        = builder.modulesElement;
    setUp();
  }

  public static Builder builder() {
    return new Builder();
  }

  private void setUp() {
  }

  public void scan(RoundEnvironment roundEnvironment) {
    TypeMirror modulesTypeMirror = this.processingEnvironment.getElementUtils()
                                                             .getTypeElement(Modules.class.getName())
                                                             .asType();

    List moduleClasses = this.modulesElement.getAnnotationMirrors()
                                                    .stream()
                                                    .filter(annotationMirror -> this.processingEnvironment.getTypeUtils()
                                                                                                          .isSameType(annotationMirror.getAnnotationType(),
                                                                                                                      modulesTypeMirror))
                                                    .flatMap(annotationMirror -> annotationMirror.getElementValues()
                                                                                                 .entrySet()
                                                                                                 .stream())
                                                    .findFirst()
                                                    .map(entry -> Arrays.stream(entry.getValue()
                                                                                     .toString()
                                                                                     .replace("{",
                                                                                              "")
                                                                                     .replace("}",
                                                                                              "")
                                                                                     .replace(" ",
                                                                                              "")
                                                                                     .split(","))
                                                                        .filter(c -> c.contains(".class"))
                                                                        .map((v) -> v.substring(0,
                                                                                                v.indexOf(".class")))
                                                                        .collect(Collectors.toList()))
                                                    .orElse(null);
    this.metaModel.getModules()
                  .clear();
    if (!Objects.isNull(moduleClasses)) {
      this.metaModel.getModules()
                    .addAll(moduleClasses.stream()
                                         .map(ClassNameModel::new)
                                         .collect(Collectors.toList()));
    }
  }

  public static class Builder {

    ProcessingEnvironment processingEnvironment;

    MetaModel metaModel;

    Element modulesElement;

    public Builder processingEnvironment(ProcessingEnvironment processingEnvironment) {
      this.processingEnvironment = processingEnvironment;
      return this;
    }

    public Builder metaModel(MetaModel metaModel) {
      this.metaModel = metaModel;
      return this;
    }

    public Builder modulesElement(Element modulesElement) {
      this.modulesElement = modulesElement;
      return this;
    }

    public ModulesAnnotationScanner build() {
      return new ModulesAnnotationScanner(this);
    }

  }

}