querydsl/querydsl-jpa-codegen/src/test/java/com/querydsl/jpa/codegen/CompileUtils.java

package com.querydsl.jpa.codegen;

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

import org.junit.Assert;

import com.mysema.codegen.SimpleCompiler;

public final class CompileUtils {

    private CompileUtils() { }

    private static final SimpleCompiler compiler = new SimpleCompiler();

    private static List<String> getFiles(String path) {
        List<String> classes = new ArrayList<String>();
        for (File file : new File(path).listFiles()) {
            if (file.getName().endsWith(".java")) {
                classes.add(file.getPath());
            } else if (file.isDirectory() && !file.getName().startsWith(".")) {
                classes.addAll(getFiles(file.getAbsolutePath()));
            }
        }
        return classes;
    }

    public static void compile(String target) throws IOException {
        List<String> options = new ArrayList<String>();
        options.addAll(getFiles(target));

        int compilationResult = compiler.run(null, null, null, options.toArray(new String[options.size()]));

        if (compilationResult != 0) {
            Assert.fail("Compilation Failed");
        }
    }
}