org.walkmod.javalang.compiler.Compiler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of javalang-compiler Show documentation
Library of compiler components to processs Java code.
There is a newer version: 2.3.10
/*
 Copyright (C) 2015 Raquel Pau and Albert Coroleu.
 
Walkmod is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

Walkmod is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public License
along with Walkmod.  If not, see .*/
package org.walkmod.javalang.compiler;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import javax.tools.JavaCompiler;
import javax.tools.JavaFileObject;
import javax.tools.StandardJavaFileManager;
import javax.tools.ToolProvider;

import org.walkmod.javalang.ASTManager;
import org.walkmod.javalang.ParseException;
import org.walkmod.javalang.ast.CompilationUnit;
import org.walkmod.javalang.ast.PackageDeclaration;
import org.walkmod.javalang.ast.body.TypeDeclaration;

public class Compiler {

	public void compile(File compilationDir, File... files) throws IOException {
		JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
		List optionList = new ArrayList();

		boolean result = (compilationDir.mkdirs() || compilationDir.exists())
				&& compilationDir.canWrite();
		if (result) {
			optionList.addAll(Arrays.asList("-d",
					compilationDir.getAbsolutePath()));
			StandardJavaFileManager sjfm = compiler.getStandardFileManager(
					null, null, null);
			Iterable fileObjects = sjfm
					.getJavaFileObjects(files);
			JavaCompiler.CompilationTask task = compiler.getTask(null, null,
					null, optionList, null, fileObjects);
			task.call();
			sjfm.close();
		} else {
			throw new IOException("The system cannot compile in the "
					+ compilationDir.getAbsolutePath() + " directory");
		}
	}

	public void compile(File compilationDir, File sourcesDir, String... codes)
			throws IOException, ParseException {

		File[] sources = new File[codes.length];
		int i = 0;
		for (String code : codes) {
			CompilationUnit cu = ASTManager.parse(code);

			List types = cu.getTypes();
			String name = null;
			if (types != null) {

				TypeDeclaration next = types.get(0);
				
				name = next.getName() + ".java";

			}
			PackageDeclaration pd = cu.getPackage();

			if (pd != null) {
				String pckg = pd.getName().toString();
				sourcesDir = new File(sourcesDir, pckg.replaceAll(".", "//"));
			}

			if ((sourcesDir.mkdirs() || sourcesDir.exists())
					&& sourcesDir.canWrite()) {

				File tmpClass = new File(sourcesDir, name);
				FileWriter fw = new FileWriter(tmpClass);
				fw.write(code);
				fw.flush();
				fw.close();
				sources[i] = tmpClass;

			}
			i++;
		}
		compile(compilationDir, sources);
	}

}