jsclassloader.classes.ClassFileSet Maven / Gradle / Ivy

Go to download
package jsclassloader.classes;

import java.io.File;
import java.io.IOException;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.logging.Logger;

import jsclassloader.Config;

import com.esotericsoftware.wildcard.Paths;

/**
 * Given a list of root folders this class will locate all js class files and store them, converting their
 * folder paths and file names into fully qualified class names.
 */
public class ClassFileSet {
	
	private final static Logger LOG = Logger.getLogger("JS-Class-Loader");
	
	private Path basePath;
	private final List rootDirs;
	private Map classnameToFilelookup;
	private Map classnameToSrcDirLookup;
	private Map fileToClassnamelookup;
	private List allJsClasses = new ArrayList();

	
	
	public ClassFileSet(Config config) throws IOException {
		
		basePath = java.nio.file.Paths.get(new File(config.getProperty(Config.PROP_BASE_FOLDER)).getCanonicalPath());
		
		classnameToFilelookup = new HashMap();
		classnameToSrcDirLookup = new HashMap();
		fileToClassnamelookup = new HashMap();
		
		this.rootDirs = this.generateSourceFolderList(config);
		if (this.rootDirs.size() == 0) {
			throw new RuntimeException("Error, the source paths that you have provided don't match anything.\n"
					+ "base path: " + config.getProperty(Config.PROP_BASE_FOLDER) + "\n"
					+ "source paths: " + config.getProperty(Config.PROP_SOURCE_PATHS));
		}
		this.initialize();
	}
	
	public int size() {
		return allJsClasses.size();
	}

	public List getAllJsClasses() {
		return allJsClasses;
	}

	public File getFileFromClassname(String classname) {
		return classnameToFilelookup.get(classname);
	}
	
	public String getSrcDirFromClassname(String classname) {
		return classnameToSrcDirLookup.get(classname);
	}
	
	public String getClassnameFromFile(File file) {
		return fileToClassnamelookup.get(file);
	}
	
	private void initialize() throws IOException {
		for (File rootDir : rootDirs) {
			LOG.info("Source path: " + rootDir.getAbsolutePath());
			int rootFilenameLength = rootDir.getAbsolutePath().length();
			
			Path srcRootPath = basePath.relativize(java.nio.file.Paths.get(rootDir.getCanonicalPath()));
			
			findJsFiles(rootDir, rootFilenameLength, srcRootPath.toString());
		}
	}

	private void findJsFiles(File directory, int rootFilenameLength, String rootDir) {

		File[] files = directory.listFiles();
		if (files == null) { return; }
		for (int i = 0; i < files.length; i++) {
			if (files[i].isDirectory()) {
				findJsFiles(files[i], rootFilenameLength, rootDir);
			}
			if (files[i].getName().endsWith(".js")) {
				String classname = generateClassnameFromFile(files[i], rootFilenameLength);
				allJsClasses.add(classname);
				classnameToFilelookup.put(classname, files[i]);
				classnameToSrcDirLookup.put(classname, rootDir);
				fileToClassnamelookup.put(files[i], classname);
			}
		}
	}

	private String generateClassnameFromFile(File file, int rootFilenameLength) {
		String name = file.getAbsolutePath();
		int endPos = name.length() - 3;
		String relative = name.substring(rootFilenameLength + 1 , endPos);
		String result  = relative.replace(File.separatorChar, '.');
		return result;
	}
	
	private List generateSourceFolderList(Config config) {
		Paths paths = new Paths();
		
		String sourcePaths = config.getProperty(Config.PROP_SOURCE_PATHS);
		String basePath = config.getProperty(Config.PROP_BASE_FOLDER);
		
		for (String path : sourcePaths.split(",")) {
			LOG.info("Source folder string: " + basePath + ", " + path);
			paths.glob(basePath, path);
		}
		return paths.dirsOnly().getFiles();
	}
}