SystemTypeLoader.java

package nebula.lang;

import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.Enumeration;

public class SystemTypeLoader extends TypeLoader {
	final ResourcePoolTail source;

	public SystemTypeLoader() {
		super(BootstrapTypeLoader.getInstance());
		this.source = new ResourcePoolTail();
		this.init("nebula.properties");
	}

	public SystemTypeLoader(TypeLoader parent) {
		super(parent);
		this.source = new ResourcePoolTail();
	}

	public ResourcePath insertResourcePath(ResourcePath cp) {
		return source.insertResourcePath(cp);
	}

	public ResourcePath appendResourcePath(ResourcePath cp) {
		return source.appendResourcePath(cp);
	}

	public ResourcePath insertResourcePath(String pathname) {
		return source.insertResourcePath(pathname);
	}

	public ResourcePath appendResourcePath(String pathname) {
		return source.appendResourcePath(pathname);
	}

	private void init(String name) {
		try {

			Enumeration<URL> resources = this.getClass().getClassLoader().getResources(name);

			while (resources.hasMoreElements()) {
				URL url = resources.nextElement();
				if ("jar".equals(url.getProtocol())) {
					source.appendResourcePath(new JarClassPath(url.getPath().substring(5).split("!")[0]));
				} else {
					source.appendResourcePath(new DirResourcePath(new File(url.getPath()).getParentFile().getPath()));
				}
			}

			// resources = this.getClass().getClassLoader().getResources(name);
			// while (resources.hasMoreElements()) {
			// URL url = resources.nextElement();
			// if ("jar".equals(url.getProtocol())) {
			// loadJar(new File(url.getPath().substring(5).split("!")[0]));
			// } else {
			// loadFolder(new File(url.getPath()).getParentFile());
			// }
			// }

		} catch (IOException e) {
			e.printStackTrace();
		}
	}

	// private void loadJar(File f) {
	// if (log.isTraceEnabled()) {
	// log.trace(f.getPath());
	// }
	// try {
	// JarFile jf = new JarFile(f);
	// Enumeration<JarEntry> entries = jf.entries();
	// while (entries.hasMoreElements()) {
	// JarEntry entry = entries.nextElement();
	// if (entry.getName().endsWith(".nebula")) {
	// if (log.isTraceEnabled()) {
	// log.trace("** Load type from " + entry.getName());
	// }
	// List<Type> typeList = super.defineNebula(jf.getInputStream(entry));
	// for (Type type : typeList) {
	// type.underlyingSource = entry;
	// }
	// }
	// }
	// } catch (Exception e) {
	// throw new RuntimeException(e);
	// }
	// }

	public void loadFolder(File root) {
		source.appendResourcePath(new DirResourcePath(root.getAbsolutePath()));
		if (log.isTraceEnabled()) {
			log.trace(root.getPath());
		}
		loadFolder(root, root);
	}

	private void loadFolder(File root, File d) {
		try {
			if (!d.exists() || !d.isDirectory())
				return;

			for (File f : d.listFiles()) {
				if (f.isFile() && f.getName().endsWith(".nebula")) {
					this.defineNebula(f.toURI().toURL());
				} else if (f.isDirectory()) {
					loadFolder(root, f);
				}
			}
		} catch (Exception e) {
			throw new RuntimeException(e);
		}
	}

	@Override
	protected URL loadClassData(String name) {
		URL url = source.find(name, ".nebula");
		return url;
	}

}