com.appland.appmap.transform.annotations.HookAnnotatedSystem Maven / Gradle / Ivy

Go to download
package com.appland.appmap.transform.annotations;

import java.util.Map;
import java.util.Set;

import com.appland.appmap.config.AppMapConfig;
import com.appland.appmap.transform.annotations.AnnotationUtil.AnnotatedBehavior;

import javassist.CtBehavior;

public class HookAnnotatedSystem extends SourceMethodSystem {
  private String annotationClass;

  private HookAnnotatedSystem(CtBehavior behavior, String annotationClass) {
    super(behavior, HookAnnotated.class);

    this.annotationClass = annotationClass;
  }

  /**
   * Factory method. Reads any relevant annotation information and caches it.
   * @param behavior The hook behavior
   * @return A new {@code HookAnnotatedSystem} if {@link HookAnnotated} is found. Otherwise,
   *         {@code null}.
   */
  public static ISystem from(CtBehavior behavior) {
    String annotatedClass = (String) AnnotationUtil.getValue(behavior, HookAnnotated.class, null);
    if (annotatedClass == null) {
      return null;
    }
    return new HookAnnotatedSystem(behavior, annotatedClass);
  }

  @Override
  public Boolean match(CtBehavior behavior, Map hookContext) {
    boolean ret = doMatch(behavior, hookContext);
    if (ret) {
      AnnotationUtil.setAnnotation(new AnnotatedBehavior(behavior), AppMapAgentMethod.class);
    }
    return ret;
  }

  @SuppressWarnings("unchecked")
  private Boolean doMatch(CtBehavior behavior, Map hookContext) {
    final Boolean isExplicitlyExcluded = AppMapConfig.get().excludes(behavior);

    Set annotations = (Set)hookContext.get(Hook.ANNOTATIONS);
    return !isExplicitlyExcluded && annotations != null && annotations.contains(annotationClass);
  }
}