package yalp.classloading;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.lang.annotation.Annotation;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javassist.ClassPool;
import javassist.CtClass;
import yalp.Logger;
import yalp.Yalp;
import yalp.YalpPlugin;
import yalp.classloading.enhancers.Enhancer;
import yalp.exceptions.UnexpectedException;
import yalp.vfs.VirtualFile;
/**
* Application classes container.
*/
public class ApplicationClasses {
/**
* Reference to the eclipse compiler.
*/
ApplicationCompiler compiler = new ApplicationCompiler(this);
/**
* Cache of all compiled classes
*/
Map<String, ApplicationClass> classes = new HashMap<String, ApplicationClass>();
/**
* Clear the classes cache
*/
public void clear() {
classes = new HashMap<String, ApplicationClass>();
}
/**
* Get a class by name
*
* @param name The fully qualified class name
* @return The ApplicationClass or null
*/
public ApplicationClass getApplicationClass(String name) {
if (!classes.containsKey(name) && getJava(name) != null) {
classes.put(name, new ApplicationClass(name));
}
return classes.get(name);
}
/**
* Retrieve all application classes assignable to this class.
*
* @param clazz The superclass, or the interface.
* @return A list of application classes.
*/
public List<ApplicationClass> getAssignableClasses(Class<?> clazz) {
List<ApplicationClass> results = new ArrayList<ApplicationClass>();
if (clazz != null) {
for (ApplicationClass applicationClass : new ArrayList<ApplicationClass>(classes.values())) {
if (!applicationClass.isClass()) {
continue;
}
try {
Yalp.classloader.loadClass(applicationClass.name);
} catch (ClassNotFoundException ex) {
throw new UnexpectedException(ex);
}
try {
if (clazz.isAssignableFrom(applicationClass.javaClass) && !applicationClass.javaClass.getName().equals(clazz.getName())) {
results.add(applicationClass);
}
} catch (Exception e) {
}
}
}
return results;
}
/**
* Retrieve all application classes with a specific annotation.
*
* @param clazz The annotation class.
* @return A list of application classes.
*/
public List<ApplicationClass> getAnnotatedClasses(Class<? extends Annotation> clazz) {
List<ApplicationClass> results = new ArrayList<ApplicationClass>();
for (ApplicationClass applicationClass : classes.values()) {
if (!applicationClass.isClass()) {
continue;
}
try {
Yalp.classloader.loadClass(applicationClass.name);
} catch (ClassNotFoundException ex) {
throw new UnexpectedException(ex);
}
if (applicationClass.javaClass != null && applicationClass.javaClass.isAnnotationPresent(clazz)) {
results.add(applicationClass);
}
}
return results;
}
/**
* All loaded classes.
*
* @return All loaded classes
*/
public List<ApplicationClass> all() {
return new ArrayList<ApplicationClass>(classes.values());
}
/**
* Put a new class to the cache.
*/
public void add(ApplicationClass applicationClass) {
classes.put(applicationClass.name, applicationClass);
}
/**
* Remove a class from cache
*/
public void remove(ApplicationClass applicationClass) {
classes.remove(applicationClass.name);
}
public void remove(String applicationClass) {
classes.remove(applicationClass);
}
/**
* Does this class is already loaded ?
*
* @param name The fully qualified class name
*/
public boolean hasClass(String name) {
return classes.containsKey(name);
}
/**
* Represent a application class
*/
public static class ApplicationClass {
/**
* The fully qualified class name
*/
public String name;
/**
* A reference to the java source file
*/
public VirtualFile javaFile;
/**
* The Java source
*/
public String javaSource;
/**
* The compiled byteCode
*/
public byte[] javaByteCode;
/**
* The enhanced byteCode
*/
public byte[] enhancedByteCode;
/**
* The in JVM loaded class
*/
public Class<?> javaClass;
/**
* The in JVM loaded package
*/
public Package javaPackage;
/**
* Last time than this class was compiled
*/
public Long timestamp = 0L;
/**
* Is this class compiled
*/
boolean compiled;
/**
* Signatures checksum
*/
public int sigChecksum;
public ApplicationClass() {
}
public ApplicationClass(String name) {
this.name = name;
this.javaFile = getJava(name);
this.refresh();
}
/**
* Need to refresh this class !
*/
public void refresh() {
if (this.javaFile != null) {
this.javaSource = this.javaFile.contentAsString();
}
this.javaByteCode = null;
this.enhancedByteCode = null;
this.compiled = false;
this.timestamp = 0L;
}
static final ClassPool enhanceChecker_classPool = Enhancer.newClassPool();
static final CtClass ctYalpPluginClass = enhanceChecker_classPool.makeClass(YalpPlugin.class.getName());
/**
* Enhance this class
*
* @return the enhanced byteCode
*/
public byte[] enhance() {
this.enhancedByteCode = this.javaByteCode;
if (isClass()) {
// before we can start enhancing this class we must make sure it is not a YalpPlugin.
// YalpPlugins can be included as regular java files in a Yalp-application.
// If a YalpPlugin is present in the application, it is loaded when other plugins are loaded.
// All plugins must be loaded before we can start enhancing.
// This is a problem when loading YalpPlugins bundled as regular app-class since it uses the same classloader
// as the other (soon to be) enhanched yalp-app-classes.
boolean shouldEnhance = true;
try {
CtClass ctClass = enhanceChecker_classPool.makeClass(new ByteArrayInputStream(this.enhancedByteCode));
if (ctClass.subclassOf(ctYalpPluginClass)) {
shouldEnhance = false;
}
} catch (Exception e) {
// nop
}
if (shouldEnhance) {
Yalp.pluginCollection.enhance(this);
}
}
if (System.getProperty("precompile") != null) {
try {
// emit bytecode to standard class layout as well
File f = Yalp.getFile("precompiled/java/" + (name.replace(".", "/")) + ".class");
f.getParentFile().mkdirs();
FileOutputStream fos = new FileOutputStream(f);
fos.write(this.enhancedByteCode);
fos.close();
} catch (Exception e) {
e.printStackTrace();
}
}
return this.enhancedByteCode;
}
/**
* Is this class already compiled but not defined ?
*
* @return if the class is compiled but not defined
*/
public boolean isDefinable() {
return compiled && javaClass != null;
}
public boolean isClass() {
return isClass(this.name);
}
public static boolean isClass(String name) {
return !name.endsWith("package-info");
}
public String getPackage() {
int dot = name.lastIndexOf('.');
return dot > -1 ? name.substring(0, dot) : "";
}
/**
* Compile the class from Java source
*
* @return the bytes that comprise the class file
*/
public byte[] compile() {
long start = System.currentTimeMillis();
Yalp.classes.compiler.compile(new String[]{this.name});
if (Logger.isTraceEnabled()) {
Logger.trace("%sms to compile class %s", System.currentTimeMillis() - start, name);
}
return this.javaByteCode;
}
/**
* Unload the class
*/
public void uncompile() {
this.javaClass = null;
}
/**
* Call back when a class is compiled.
*
* @param code The bytecode.
*/
public void compiled(byte[] code) {
javaByteCode = code;
enhancedByteCode = code;
compiled = true;
this.timestamp = this.javaFile.lastModified();
}
@Override
public String toString() {
return name + " (compiled:" + compiled + ")";
}
}
// ~~ Utils
/**
* Retrieve the corresponding source file for a given class name.
* It handles innerClass too !
*
* @param name The fully qualified class name
* @return The virtualFile if found
*/
public static VirtualFile getJava(String name) {
String fileName = name;
if (fileName.contains("$")) {
fileName = fileName.substring(0, fileName.indexOf("$"));
}
fileName = fileName.replace(".", "/") + ".java";
for (VirtualFile path : Yalp.javaPath) {
VirtualFile javaFile = path.child(fileName);
if (javaFile.exists()) {
return javaFile;
}
}
return null;
}
@Override
public String toString() {
return classes.toString();
}
}