net.beanstandard.compiler.Arity1 Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bean-standard Show documentation
The newest version!
package net.beanstandard.compiler;

import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.PackageElement;
import javax.lang.model.element.TypeElement;
import javax.lang.model.type.TypeKind;
import javax.lang.model.type.TypeMirror;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import static javax.lang.model.element.Modifier.ABSTRACT;
import static javax.lang.model.element.Modifier.PRIVATE;
import static javax.lang.model.element.Modifier.PUBLIC;
import static javax.lang.model.element.Modifier.STATIC;
import static javax.lang.model.type.TypeKind.VOID;
import static javax.lang.model.util.ElementFilter.methodsIn;
import static net.beanstandard.compiler.LessTypes.asTypeElement;
import static net.beanstandard.compiler.LessTypes.getPackage;

final class Arity1 {

  static List singleParameterMethodsReturningVoid(
      TypeElement type) {
    PackageElement ourPackage = getPackage(type);
    Map methods = new LinkedHashMap<>();
    addFromSuperclass(type, methods, ourPackage);
    return new ArrayList<>(methods.values());
  }

  private static void addFromSuperclass(
      TypeElement type,
      Map methods,
      PackageElement ourPackage) {
    addEnclosedMethods(type, methods, ourPackage);
    TypeMirror superclass = type.getSuperclass();
    if (superclass.getKind() != TypeKind.DECLARED) {
      return;
    }
    addFromSuperclass(
        asTypeElement(superclass), methods, ourPackage);
  }

  private static void addEnclosedMethods(
      TypeElement type,
      Map methods,
      PackageElement ourPackage) {
    methodsIn(type.getEnclosedElements())
        .stream()
        .filter(method -> method.getParameters().size() == 1)
        .filter(method -> method.getReturnType().getKind() == VOID)
        .filter(method -> !method.getModifiers().contains(ABSTRACT))
        .filter(method -> !method.getModifiers().contains(STATIC))
        .filter(method -> !method.getModifiers().contains(PRIVATE))
        .filter(method -> method.getModifiers().contains(PUBLIC) ||
            getPackage(method).equals(ourPackage))
        .forEach(method -> methods.putIfAbsent(
            SetterSignature.ofSetter(method), method));
  }
}