@Override
public void visit(final CompilationUnit n, final A arg) {
visitComment(n.getComment(), arg);
if (n.getPackage() != null) {
n.getPackage().accept(this, arg);
}
if (n.getImports() != null) {
for (final ImportDeclaration i : n.getImports()) {
i.accept(this, arg);
}
}
if (n.getTypes() != null) {
for (final TypeDeclaration<?> typeDeclaration : n.getTypes()) {
typeDeclaration.accept(this, arg);
}
}
}
@Override
public Boolean visit(final CompilationUnit n1, final Node arg) {
final CompilationUnit n2 = (CompilationUnit) arg;
if (!nodeEquals(n1.getPackage(), n2.getPackage())) {
return Boolean.FALSE;
}
if (!nodesEquals(n1.getImports(), n2.getImports())) {
return Boolean.FALSE;
}
if (!nodesEquals(n1.getTypes(), n2.getTypes())) {
return Boolean.FALSE;
}
if (!nodesEquals(n1.getComments(), n2.getComments())) {
return Boolean.FALSE;
}
return Boolean.TRUE;
}
@Nullable
private static String extractClassName(String code, PrintStream err) {
InputStream inputStream = new ByteArrayInputStream(code.getBytes(StandardCharsets.UTF_8));
try {
CompilationUnit compilationUnit = JavaParser.parse(inputStream);
List<TypeDeclaration> types = compilationUnit.getTypes();
if (types.size() == 1) {
String simpleType = types.get(0).getName();
return Optional.ofNullable(compilationUnit.getPackage())
.map(PackageDeclaration::getPackageName)
.map(it -> it + "." + simpleType)
.orElse(simpleType);
} else if (types.size() == 0) {
err.println("No class definition found");
} else {
err.println("Too many class definitions found. Only one class can be defined at a time.");
}
} catch (ParseException e) {
// ignore error, let the compiler provide an error message
return "Err";
}
return null;
}