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

There is a newer version: 1.5

package net.beanstandard.compiler;

import javax.lang.model.element.ExecutableElement;
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.ElementKind.METHOD;
import static javax.lang.model.element.Modifier.ABSTRACT;
import static javax.lang.model.element.Modifier.PRIVATE;
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;

final class Arity1 {

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

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

  private static void addEnclosedMethods(TypeElement type, Map methods) {
    methodsIn(type.getEnclosedElements())
        .stream()
        .filter(method -> method.getParameters().size() == 1)
        .filter(method -> method.getReturnType().getKind() == VOID)
        .forEach(method -> {
          if (method.getKind() == METHOD
              && !method.getModifiers().contains(STATIC)
              && !method.getModifiers().contains(ABSTRACT)
              && !method.getModifiers().contains(PRIVATE)) {
            methods.computeIfAbsent(SetterSignature.ofSetter(method),
                __ -> method);
          }
        });
  }
}