br.com.objectos.way.code.MethodInfoMethodSpecWriter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of code-core Show documentation
The newest version!
/*
 * Copyright 2015 Objectos, Fábrica de Software LTDA.
 *
 * 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 br.com.objectos.way.code;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

import javax.lang.model.element.Modifier;

import com.squareup.javapoet.CodeBlock;
import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.MethodSpec.Builder;
import com.squareup.javapoet.ParameterSpec;
import com.squareup.javapoet.TypeName;

/**
 * @author [email protected] (Marcio Endo)
 */
abstract class MethodInfoMethodSpecWriter> {

  private final MethodInfo methodInfo;
  private final List parameterList;
  private final List statementList = new ArrayList<>();

  private AccessInfo accessInfo;
  private TypeName returnTypeName;

  public MethodInfoMethodSpecWriter(MethodInfo methodInfo) {
    this.methodInfo = methodInfo;
    parameterList = methodInfo.parameterInfoList()
        .stream()
        .map(ParameterInfo::parameterSpec)
        .collect(Collectors.toList());

    accessInfo = methodInfo.accessInfo();
    returnTypeName = methodInfo
        .returnTypeInfo()
        .typeName();
  }

  public T accessInfo(AccessInfo accessInfo) {
    this.accessInfo = accessInfo;
    return self();
  }

  public T addCode(CodeBlock statement) {
    statementList.add(statement);
    return self();
  }

  public T addStatement(String format, Object... args) {
    CodeBlock code = CodeBlock.builder()
        .addStatement(format, args)
        .build();
    return addCode(code);
  }

  public T returns(SimpleTypeInfo typeInfo) {
    returnTypeName = typeInfo.typeName();
    return self();
  }

  public T returns(TypeName typeName) {
    returnTypeName = typeName;
    return self();
  }

  public MethodSpec write() {
    Builder b = MethodSpec.methodBuilder(methodInfo.name())
        .addModifiers(modifiers())
        .addParameters(parameterList)
        .returns(returnTypeName)
        .varargs(methodInfo.varargs());

    for (CodeBlock codeBlock : statementList) {
      b.addCode(codeBlock);
    }

    return build(b);
  }

  T addParameter(TypeName type, String name, Modifier... modifiers) {
    parameterList.add(ParameterSpec.builder(type, name, modifiers).build());
    return self();
  }

  MethodSpec build(MethodSpec.Builder builder) {
    return builder.build();
  }

  abstract T self();

  private Modifier[] modifiers() {
    return accessInfo.modifiers();
  }

}