br.com.objectos.code.MethodInfoArtifactGenerator Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2016 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.code;

import javax.lang.model.element.Element;
import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.TypeElement;

/**
 * @author [email protected] (Marcio Endo)
 */
@FunctionalInterface
public interface MethodInfoArtifactGenerator extends ArtifactGenerator {

  Artifact generate(TypeInfo typeInfo, MethodInfo methodInfo);

  @Override
  default Artifact generateArtifact(ProcessingEnvironmentWrapper processingEnv, Element element) {
    Artifact artifact = Artifact.empty();

    Element enclosingElement = element.getEnclosingElement();
    if (Apt.isType(enclosingElement)) {
      TypeInfo typeInfo = TypeInfoTypeElement.wrap(processingEnv, (TypeElement) enclosingElement);
      TypeParameterInfoMap typeParameterInfoMap = typeInfo.typeParameterInfoMap();

      MethodInfo methodInfo;
      methodInfo = MethodInfoExecutableElement.wrap(processingEnv, (ExecutableElement) element, typeParameterInfoMap);

      artifact = generate(typeInfo, methodInfo);
    }

    return artifact;
  }

  @Override
  default boolean test(Element element) {
    return Util.isMethodInfoKind(element);
  }

}