@Override
public String resolveType(final String type) {
String original = type;
String result = type;
// Strip away any characters that might hinder the type matching process
if (Types.isArray(result)) {
original = Types.stripArray(result);
result = Types.stripArray(result);
}
if (Types.isGeneric(result)) {
original = Types.stripGenerics(result);
result = Types.stripGenerics(result);
}
if (Types.isPrimitive(result)) {
return result;
}
// Check for direct import matches first since they are the fastest and least work-intensive
if (Types.isSimpleName(result)) {
if (!hasImport(type) && Types.isJavaLang(type)) {
result = "java.lang." + result;
}
if (result.equals(original)) {
for (Import imprt : getImports()) {
if (Types.areEquivalent(result, imprt.getQualifiedName())) {
result = imprt.getQualifiedName();
break;
}
}
}
}
// If we didn't match any imports directly, we might have a wild-card/on-demand import.
if (Types.isSimpleName(result)) {
for (Import imprt : getImports()) {
if (imprt.isWildcard()) {
// TODO warn if no wild-card resolvers are configured
// TODO Test wild-card/on-demand import resolving
for (WildcardImportResolver r : getImportResolvers()) {
result = r.resolve(this, result);
if (Types.isQualified(result)) break;
}
}
}
}
// No import matches and no wild-card/on-demand import matches means this class is in the same
// package.
if (Types.isSimpleName(result)) {
if (getPackage() != null) result = getPackage() + "." + result;
}
return result;
}
@Override
public boolean hasImport(final String type) {
String resultType = type;
if (Types.isArray(type)) {
resultType = Types.stripArray(type);
}
if (Types.isGeneric(type)) {
resultType = Types.stripGenerics(type);
}
return getImport(resultType) != null;
}
@Override
public boolean requiresImport(final String type) {
String resultType = type;
if (Types.isArray(resultType)) {
resultType = Types.stripArray(type);
}
if (Types.isGeneric(resultType)) {
resultType = Types.stripGenerics(resultType);
}
if (!validImport(resultType) || hasImport(resultType) || Types.isJavaLang(resultType)) {
return false;
}
return true;
}
@Override
public Import addImport(final String className) {
String strippedClassName = Types.stripGenerics(Types.stripArray(className));
Import imprt;
if (Types.isSimpleName(strippedClassName) && !hasImport(strippedClassName)) {
throw new IllegalArgumentException(
"Cannot import class without a package [" + strippedClassName + "]");
}
if (!hasImport(strippedClassName) && validImport(strippedClassName)) {
imprt = new ImportImpl(this).setName(strippedClassName);
unit.imports().add(imprt.getInternal());
} else if (hasImport(strippedClassName)) {
imprt = getImport(strippedClassName);
} else {
throw new IllegalArgumentException(
"Attempted to import the illegal type [" + strippedClassName + "]");
}
return imprt;
}