package org.openrdf.elmo.dynacode;
   
   import java.io.ByteArrayInputStream;
   import java.io.File;
   import java.io.FileOutputStream;
   import java.io.IOException;
   import java.io.InputStream;
   import java.net.MalformedURLException;
   import java.net.URL;
  import java.util.concurrent.ConcurrentHashMap;
  import java.util.concurrent.ConcurrentMap;
  
  import javassist.CannotCompileException;
  import javassist.ClassPool;
  import javassist.CtClass;
  import javassist.LoaderClassPath;
  import javassist.NotFoundException;
  import javassist.bytecode.Descriptor;
  
  import org.openrdf.elmo.exceptions.ElmoCompositionException;
  
  public class ClassFactory extends ClassLoader {
  	private static final URL exists;
  
  	static {
  		try {
  			exists = new URL("http://java/"
  					+ ClassFactory.class.getName().replace('.', '/')
  					+ "#exists");
  		} catch (MalformedURLException e) {
  			throw new AssertionError(e);
  		}
  	}
  
  	private static CtClass getPrimitive(Class<?> type, ClassFactory cp) {
  		if (type.equals(Boolean.TYPE))
  			return CtClass.booleanType;
  		if (type.equals(Byte.TYPE))
  			return CtClass.byteType;
  		if (type.equals(Character.TYPE))
  			return CtClass.charType;
  		if (type.equals(Double.TYPE))
  			return CtClass.doubleType;
  		if (type.equals(Float.TYPE))
  			return CtClass.floatType;
  		if (type.equals(Integer.TYPE))
  			return CtClass.intType;
  		if (type.equals(Long.TYPE))
  			return CtClass.longType;
  		if (type.equals(Short.TYPE))
  			return CtClass.shortType;
  		if (type.equals(Void.TYPE))
  			return CtClass.voidType;
  		throw new ElmoCompositionException("Unknown primative type: "
  				+ type.getName());
  	}
  
  	private ConcurrentMap<String, byte[]> bytecodes;
  
  	private ClassPool cp;
  
  	private File target;
  
  	/**
  	 * Creates a new Class Factory using the current context class loader.
  	 */
  	public ClassFactory() {
  		this(Thread.currentThread().getContextClassLoader());
  	}
  
  	/**
  	 * Create a given Class Factory with the given class loader.
  	 * 
  	 * @param parent
  	 */
  	public ClassFactory(ClassLoader parent) {
  		super(parent);
  		cp = new ClassPool();
  		cp.appendClassPath(new LoaderClassPath(this));
  		bytecodes = new ConcurrentHashMap<String, byte[]>();
  		String property = System.getProperty("elmobeans.target");
  		if (property != null) {
  			target = new File(property);
  		}
  	}
  
  	/**
  	 * Create the new Java Class from this template.
  	 * 
  	 * @param template
  	 * @return new Java Class Object
  	 * @throws ElmoCompositionException
  	 */
  	public Class<?> createClass(ClassTemplate template)
  			throws ElmoCompositionException {
  		CtClass cc = template.getCtClass();
  		String name = cc.getName();
  		try {
  			byte[] bytecode = cc.toBytecode();
 			cc.detach();
 			return defineClass(name, bytecode);
 		} catch (IOException e) {
 			throw new ElmoCompositionException(e);
 		} catch (CannotCompileException e) {
 			throw new ElmoCompositionException(e);
 		}
 	}
 
 	/**
 	 * Create a new Class template, which can later be used to create a Java
 	 * class.
 	 * 
 	 * @param className
 	 * @return temporary Class template
 	 */
 	public ClassTemplate createClassTemplate(String className) {
 		return new ClassTemplate(cp.makeClass(className), this);
 	}
 
 	/**
 	 * Create a new Class template, which can later be used to create a Java
 	 * class.
 	 * 
 	 * @param name
 	 * @param class1 super class
 	 * @return temporary Class template
 	 */
 	public ClassTemplate createClassTemplate(String name, Class<?> class1) {
 		try {
 			CtClass cc = cp.makeClass(name, cp.get(class1.getName()));
 			return new ClassTemplate(cc, this);
 		} catch (NotFoundException e) {
 			throw new ElmoCompositionException(e);
 		}
 	}
 
 	@Override
 	public URL getResource(String name) {
 		if (bytecodes.containsKey(name))
 			return exists;
 		return super.getResource(name);
 	}
 
 	@Override
 	public InputStream getResourceAsStream(String name) {
 		if (bytecodes.containsKey(name)) {
 			byte[] b = bytecodes.get(name);
 			return new ByteArrayInputStream(b);
 		}
 		return super.getResourceAsStream(name);
 	}
 
 	/**
 	 * Causes all created Java Classes to be saved into this directory.
 	 * @param folder
 	 */
 	public void setTraget(File folder) {
 		target = folder;
 	}
 
 	CtClass get(Class<?> type) throws ElmoCompositionException {
 		if (type.isPrimitive()) {
 			return getPrimitive(type, this);
 		}
 		try {
 			if (type.isArray())
 				return Descriptor.toCtClass(type.getName(), cp);
 			return cp.get(type.getName());
 		} catch (NotFoundException e) {
 			throw new ElmoCompositionException(e);
 		}
 	}
 
 	private Class defineClass(String name, byte[] bytecode) {
 		String resource = name.replace('.', '/') + ".class";
 		if (target != null) {
 			saveResource(resource, bytecode);
 		}
 		bytecodes.putIfAbsent(resource, bytecode);
 		return defineClass(name, bytecode, 0, bytecode.length);
 	}
 
 	private void saveResource(String fileName, byte[] bytecode) {
 		try {
 			File file = new File(target, fileName);
 			file.getParentFile().mkdirs();
 			FileOutputStream out = new FileOutputStream(file);
 			try {
 				out.write(bytecode);
 			} finally {
 				out.close();
 			}
 		} catch (Exception e) {
 		}
 	}
 
 	Class<?> getJavaClass(CtClass cc) throws ClassNotFoundException {
 		if (cc.isPrimitive()) {
 			if (cc.equals(CtClass.booleanType))
 				return Boolean.TYPE;
 			if (cc.equals(CtClass.byteType))
 				return Byte.TYPE;
 			if (cc.equals(CtClass.charType))
 				return Character.TYPE;
 			if (cc.equals(CtClass.doubleType))
 				return Double.TYPE;
 			if (cc.equals(CtClass.floatType))
 				return Float.TYPE;
 			if (cc.equals(CtClass.intType))
 				return Integer.TYPE;
 			if (cc.equals(CtClass.longType))
 				return Long.TYPE;
 			if (cc.equals(CtClass.shortType))
 				return Short.TYPE;
 			throw new AssertionError();
 		}
 		String name = Descriptor.toJavaName(Descriptor.toJvmName(cc));
 		return Class.forName(name, true, this);
 	}
 }