com.github.nalukit.nalu.processor.scanner.ShellCompositesAnnotationScanner 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
There is a newer version: 2.20.3
/*
 * 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.component.annotation.Composite;
import com.github.nalukit.nalu.client.component.annotation.Composite.Scope;
import com.github.nalukit.nalu.client.component.annotation.Composites;
import com.github.nalukit.nalu.processor.model.intern.ClassNameModel;
import com.github.nalukit.nalu.processor.model.intern.ShellAndControllerCompositeModel;
import com.github.nalukit.nalu.processor.model.intern.ShellModel;

import javax.annotation.processing.ProcessingEnvironment;
import javax.annotation.processing.RoundEnvironment;
import javax.lang.model.element.Element;
import javax.lang.model.element.TypeElement;
import javax.lang.model.type.MirroredTypeException;
import java.util.Objects;

public class ShellCompositesAnnotationScanner {

  private final ProcessingEnvironment processingEnvironment;

  private final ShellModel shellModel;

  private final Element shellElement;

  @SuppressWarnings("unused")
  private ShellCompositesAnnotationScanner(Builder builder) {
    super();
    this.processingEnvironment = builder.processingEnvironment;
    this.shellModel            = builder.shellModel;
    this.shellElement          = builder.shellElement;
    setUp();
  }

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

  private void setUp() {
  }

  @SuppressWarnings("unused")
  public ShellModel scan(RoundEnvironment roundEnvironment) {
    Composites annotation = this.shellElement.getAnnotation(Composites.class);
    if (annotation != null) {
      for (Composite composite : annotation.value()) {
        shellModel.getComposites()
                  .add(new ShellAndControllerCompositeModel(composite.name(),
                                                            new ClassNameModel(Objects.requireNonNull(getCompositeTypeElement(composite))
                                                                                      .toString()),
                                                            composite.selector(),
                                                            new ClassNameModel(Objects.requireNonNull(getCompositeConditionElement(composite))
                                                                                      .toString()),
                                                            Scope.GLOBAL == composite.scope()));
      }
    }
    return this.shellModel;
  }

  private TypeElement getCompositeTypeElement(Composite annotation) {
    try {
      annotation.compositeController();
    } catch (MirroredTypeException exception) {
      return (TypeElement) this.processingEnvironment.getTypeUtils()
                                                     .asElement(exception.getTypeMirror());
    }
    return null;
  }

  private TypeElement getCompositeConditionElement(Composite annotation) {
    try {
      annotation.condition();
    } catch (MirroredTypeException exception) {
      return (TypeElement) this.processingEnvironment.getTypeUtils()
                                                     .asElement(exception.getTypeMirror());
    }
    return null;
  }

  public static class Builder {

    ProcessingEnvironment processingEnvironment;

    ShellModel shellModel;

    Element shellElement;

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

    public Builder shellModel(ShellModel shellModel) {
      this.shellModel = shellModel;
      return this;
    }

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

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

  }

}