private CompilationUnit getCompilationUnit(SourceFile sourceFile) {
CompilationUnit compilationUnit;
if (Files.exists(sourceFile.getOutputPath())) {
try {
compilationUnit = JavaParser.parse(sourceFile.getOutputPath().toFile());
} catch (Throwable t) {
throw new RuntimeException(t);
}
} else {
compilationUnit = new CompilationUnit();
compilationUnit.setComment(
new LineComment(" Generated by GraphWalker (http://www.graphwalker.org)"));
if (!"".equals(sourceFile.getPackageName())) {
compilationUnit.setPackage(createPackageDeclaration(sourceFile));
}
compilationUnit.setImports(
Arrays.asList(
new ImportDeclaration(
new NameExpr("org.graphwalker.java.annotation.Model"), false, false),
new ImportDeclaration(
new NameExpr("org.graphwalker.java.annotation.Vertex"), false, false),
new ImportDeclaration(
new NameExpr("org.graphwalker.java.annotation.Edge"), false, false)));
ASTHelper.addTypeDeclaration(compilationUnit, getInterfaceName(sourceFile));
}
return compilationUnit;
}
private PackageDeclaration createPackageDeclaration(SourceFile sourceFile) {
return new PackageDeclaration(ASTHelper.createNameExpr(sourceFile.getPackageName()));
}