com.antwerkz.critter.CritterContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of critter-maven Show documentation

There is a newer version: 3.0.5

package com.antwerkz.critter;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

import static java.lang.String.format;
import org.jboss.forge.roaster.model.source.JavaClassSource;

public class CritterContext {
  private Map classes = new HashMap<>();

  private String criteriaPkg;

  public CritterContext(final String criteriaPkg) {

    this.criteriaPkg = criteriaPkg;
  }

  public void add(CritterClass critterClass) {
    classes.put(format("%s.%s", critterClass.getPackage(), critterClass.getName()), critterClass);
    critterClass.setPackage(criteriaPkg);
  }

  public CritterClass get(String name) {
    return classes.get(name);
  }

  public Collection getClasses() {
    return classes.values();
  }

  public boolean isEmbedded(final JavaClassSource clazz) {
    final CritterClass critterClass = get(clazz.getName());
    return critterClass != null && critterClass.isEmbedded();
  }
}