net.autobuilder.core.Parameter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of auto-builder
There is a newer version: 2.9.3
package net.autobuilder.core;

import com.squareup.javapoet.CodeBlock;
import com.squareup.javapoet.FieldSpec;
import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.ParameterSpec;
import net.autobuilder.core.cases.AsRegularParameterCases;
import net.autobuilder.core.cases.AsSetterParameterCases;
import net.autobuilder.core.cases.CodeInsideSetterCases;
import net.autobuilder.core.cases.ExtraFieldCases;
import net.autobuilder.core.cases.ExtraMethodsCases;
import net.autobuilder.core.cases.ExtractCases;

import java.util.List;
import java.util.Optional;
import java.util.function.Function;

/**
 * Represents one parameter of the generated auto-value constructor.
 */
public abstract class Parameter {

  static  Function asFunction(ParamCases cases) {
    return parameter -> parameter.accept(cases, null);
  }

  abstract  R accept(ParamCases cases, P p);

  public final RegularParameter asRegularParameter() {
    return AS_REGULAR_PARAMETER.apply(this);
  }

  private static final Function AS_REGULAR_PARAMETER =
      asFunction(new AsRegularParameterCases());

  public ParameterSpec asSetterParameter() {
    return AS_SETTER_PARAMETER.apply(this);
  }

  private static final Function AS_SETTER_PARAMETER =
      asFunction(new AsSetterParameterCases());

  /**
   * Creates the code inside one of the builder's setter methods.
   */
  CodeBlock codeInsideSetter() {
    return CODE_INSIDE_SETTER.apply(this);
  }

  private static final Function CODE_INSIDE_SETTER =
      asFunction(new CodeInsideSetterCases());

  final CodeBlock extract() {
    return EXTRACT.apply(this);
  }

  private static final Function EXTRACT =
      asFunction(new ExtractCases());

  public final Optional extraField() {
    return EXTRA_FIELD.apply(this);
  }

  private static final Function> EXTRA_FIELD =
      asFunction(new ExtraFieldCases());

  List getExtraMethods(Model model) {
    return accept(EXTRA_METHODS, model);
  }

  private static final ParamCases, Model> EXTRA_METHODS =
      new ExtraMethodsCases();

  CodeBlock cleanupCode() {
    CodeBlock.Builder builder = CodeBlock.builder();
    if (!asRegularParameter().variableElement.asType().getKind().isPrimitive()) {
      builder.addStatement("$N = null", asRegularParameter().asField());
    }
    extraField().ifPresent(field ->
        builder.addStatement("$N = null", field));
    return builder.build();
  }
}