SourceCompiler.java

/**
 *  The MIT License (MIT)
 *
 *  Copyright (c) 2014 momokan (http://lwjgfont.chocolapod.net)
 *
 *  Permission is hereby granted, free of charge, to any person obtaining a copy
 *  of this software and associated documentation files (the "Software"), to deal
 *  in the Software without restriction, including without limitation the rights
 *  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 *  copies of the Software, and to permit persons to whom the Software is
 *  furnished to do so, subject to the following conditions:
 *
 *  The above copyright notice and this permission notice shall be included in all
 *  copies or substantial portions of the Software.
 *
 *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 *  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 *  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 *  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 *  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 *  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 *  SOFTWARE.
 */
package net.chocolapod.lwjgfont.packager;

import static javax.tools.JavaFileObject.Kind.SOURCE;
import static javax.tools.StandardLocation.SOURCE_PATH;
import static net.chocolapod.lwjgfont.packager.LwjgFontUtil.CHARSET_UTF8;

import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.net.URLClassLoader;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Iterator;
import java.util.List;
import java.util.Locale;

import javax.annotation.processing.Processor;
import javax.tools.Diagnostic;
import javax.tools.DiagnosticListener;
import javax.tools.JavaCompiler;
import javax.tools.JavaFileManager;
import javax.tools.JavaFileObject;
import javax.tools.StandardJavaFileManager;
import javax.tools.StandardLocation;
import javax.tools.ToolProvider;
import javax.tools.JavaCompiler.CompilationTask;

import net.chocolapod.lwjgfont.exception.LwjgFontException;

import static net.chocolapod.lwjgfont.exception.LwjgFontErrorMessage.SYSTEM_COMPILER_NOT_FOUND;

public class SourceCompiler {
	
	protected static final JavaCompiler		COMPILER = ToolProvider.getSystemJavaCompiler();
//	protected static final CompileOption	MAIN_COMPILE_OPTION = new CompileOption("src/main/java/", TestResource.MAIN_BASE_DIR, "target/classes/");
//	protected static final CompileOption	TEST_COMPILE_OPTION = new CompileOption("src/test/java/", TestResource.TEST_BASE_DIR, "target/test-classes/");
	
	private String	srcDir = "";
	private String	resourceDir = "";
	private String	targetDir = "";

	public void compile(List<String> classCanonicalNames) throws IOException {
		JavaCompiler					compiler = ToolProvider.getSystemJavaCompiler();

		if (compiler == null) {
			throw new LwjgFontException(SYSTEM_COMPILER_NOT_FOUND);
		}

		StandardJavaFileManager		fileManager = compiler.getStandardFileManager(null, Locale.getDefault(), CHARSET_UTF8);
		List<File>					classPaths = getClassPathAsFileList(fileManager);

		fileManager.setLocation(StandardLocation.SOURCE_PATH, Arrays.asList(new File(srcDir)));
		fileManager.setLocation(StandardLocation.CLASS_PATH, classPaths);
		fileManager.setLocation(StandardLocation.CLASS_OUTPUT, Arrays.asList(new File(targetDir)));

		try {
			List<JavaFileObject>	files = new ArrayList<>();

			for (String classCanonicalName: classCanonicalNames) {
				files.add(fileManager.getJavaFileForInput(SOURCE_PATH, classCanonicalName, SOURCE));
			}

			compiler.getTask(null, fileManager, new DiagnosticsReporter(), null, null, files).call();
		} finally {
			fileManager.close();
		}
	}

	private List<File> getClassPathAsFileList(StandardJavaFileManager fileManager) {
		List<File>					classPaths = new ArrayList<>();
		Iterator<? extends File>	iterator = fileManager.getLocation(StandardLocation.CLASS_PATH).iterator();

		classPaths.add(new File(resourceDir));
		while (iterator.hasNext()) {
			classPaths.add(iterator.next());
		}

		return classPaths;
	}

	public void setSourceDir(String srcDir) {
		this.srcDir = srcDir;
	}
	public void setResourceDir(String resourceDir) {
		this.resourceDir = resourceDir;
	}
	public void setTargetDir(String targetDir) {
		this.targetDir = targetDir;
	}

	class DiagnosticsReporter implements DiagnosticListener<JavaFileObject> {

		@Override
		public void report(Diagnostic<? extends JavaFileObject> diagnostic) {
//			if (LwjgFontProcessor.isDebug()) {
				System.out.println(diagnostic);
//			}
			/*
			System.out.println(diagnostic.getCode());
			System.out.println(diagnostic.getKind());
			System.out.println(diagnostic.getPosition());
			System.out.println(diagnostic.getStartPosition());
			System.out.println(diagnostic.getEndPosition());
			System.out.println(diagnostic.getSource());
			System.out.println(diagnostic.getMessage(null));
			*/
		}
		
	}
}