net.sf.staccatocommons.instrument.context.internal.AbstractAnnotationContext Maven / Gradle / Ivy

Go to download
/**
 *  Copyright (c) 2010-2012, The StaccatoCommons Team
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as published by
 *  the Free Software Foundation; version 3 of the License.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Lesser General Public License for more details.
 */

package net.sf.staccatocommons.instrument.context.internal;

import java.util.Set;

import javassist.ClassPool;
import javassist.CtClass;
import javassist.NotFoundException;
import net.sf.staccatocommons.check.Ensure;
import net.sf.staccatocommons.collections.stream.Streams;
import net.sf.staccatocommons.instrument.context.AnnotationContext;
import net.sf.staccatocommons.lang.SoftException;
import net.sf.staccatocommons.lang.function.AbstractFunction;
import net.sf.staccatocommons.restrictions.check.NonNull;

import org.slf4j.Logger;

/**
 * @author flbulgarelli
 * 
 */
public abstract class AbstractAnnotationContext implements AnnotationContext {

  private final Logger logger;
  private final ClassPool classPool;

  private Set presentAnnotations;

  /**
   * Creates a new {@link AbstractAnnotationContext}
   */
  public AbstractAnnotationContext(@NonNull ClassPool pool, @NonNull Logger logger) {
    Ensure.isNotNull("pool", pool);
    Ensure.isNotNull("logger", logger);
    this.logger = logger;
    this.classPool = pool;
  }

  /**
   * @return the logger
   */
  public Logger getLogger() {
    return logger;
  }

  @Override
  public void logDebugMessage(String message, Object... arguments) {
    getLogger().debug(message, arguments);
  }

  @Override
  public void logInfoMessage(String message, Object... arguments) {
    getLogger().info(message, arguments);
  }

  @Override
  public void logWarnMessage(String message, Object... arguments) {
    getLogger().warn(message, arguments);
  }

  @Override
  public void logErrorMessage(String message, Object... arguments) {
    getLogger().error(message, arguments);
  }

  public final ClassPool getClassPool() {
    return classPool;
  }

  public CtClass getClass(String className) throws NotFoundException {
    Ensure.isNotNull("className", className);
    return getClassPool().get(className);
  }

  /**
   * @return the presentAnnotations
   */
  public Set getPresentAnnotationsTypes(Object[] annotations) {
    if (presentAnnotations == null) {
      presentAnnotations = Streams.cons(annotations).map(new AbstractFunction() {
        public String apply(Object arg) {
          return arg.getClass().getName();
        }
      }).toSet();
    }
    return presentAnnotations;
  }

  @Override
  public String getPackage() {
    try {
      return getDeclaringClass().getPackageName();
    } catch (NotFoundException e) {
      throw SoftException.soften(e);
    }
  }
}