extra fix for typeparameters in classes

This commit is contained in:
Laurent SCHOELENS 2023-03-28 13:14:18 +02:00
parent 7147eaa5ed
commit 24e5822d7b
2 changed files with 37 additions and 10 deletions

View File

@ -162,18 +162,41 @@ class ClassGraph {
linePostfix = opt.compact ? "" : "\n"; linePostfix = opt.compact ? "" : "\n";
} }
/** Return the class's name, possibly by stripping the leading path */ /** Return the class's name, possibly by stripping the leading path
private String qualifiedName(Options opt, Name className) { * @param list */
if (opt.hideGenerics) { private String qualifiedName(Options opt, Name className, List<? extends TypeParameterElement> typeParameters) {
className = removeTemplate(elementUtils, className); String genericsInfo = "";
if (opt.hideGenerics || typeParameters == null || typeParameters.isEmpty()) {
// nothing to do
} else {
for (int i = 0; i < typeParameters.size(); i++) {
if (i == 0) {
genericsInfo += "<";
}
TypeParameterElement type = typeParameters.get(i);
List<? extends TypeMirror> mirrors = type.getBounds();
if (opt.showQualifiedGenerics && mirrors != null && !mirrors.isEmpty()) {
for (TypeMirror m : mirrors) {
genericsInfo += ElementUtil.getQualifiedName(types, m);
}
} else {
genericsInfo += type.getSimpleName();
}
if (i < typeParameters.size() - 1) {
genericsInfo += ", ";
}
if (i == typeParameters.size() - 1) {
genericsInfo += ">";
}
}
} }
// Fast path - nothing to do:
if (opt.showQualified && (opt.showQualifiedGenerics || className.toString().indexOf('<') < 0)) { if (opt.showQualified) {
return className.toString(); return className.toString() + genericsInfo;
} }
StringBuilder buf = new StringBuilder(className.length()); StringBuilder buf = new StringBuilder(className.length());
qualifiedNameInner(opt, className, buf, 0, !opt.showQualified); qualifiedNameInner(opt, className, buf, 0, !opt.showQualified);
return buf.toString(); return buf.toString() + genericsInfo;
} }
private static int qualifiedNameInner(Options opt, Name r, StringBuilder buf, int last, boolean strip) { private static int qualifiedNameInner(Options opt, Name r, StringBuilder buf, int last, boolean strip) {
@ -428,7 +451,7 @@ class ClassGraph {
} }
stereotype(opt, c, Align.CENTER); stereotype(opt, c, Align.CENTER);
Font font = c.getModifiers().contains(Modifier.ABSTRACT) && c.getKind() != ElementKind.INTERFACE ? Font.CLASS_ABSTRACT : Font.CLASS; Font font = c.getModifiers().contains(Modifier.ABSTRACT) && c.getKind() != ElementKind.INTERFACE ? Font.CLASS_ABSTRACT : Font.CLASS;
String qualifiedName = qualifiedName(opt, c.getQualifiedName()); String qualifiedName = qualifiedName(opt, c.getQualifiedName(), c.getTypeParameters());
int idx = splitPackageClass(qualifiedName); int idx = splitPackageClass(qualifiedName);
if (opt.showComment) { if (opt.showComment) {
tableLine(Align.LEFT, Font.CLASS.wrap(opt, htmlNewline(escape(TagUtil.getComment(docTrees, c))))); tableLine(Align.LEFT, Font.CLASS.wrap(opt, htmlNewline(escape(TagUtil.getComment(docTrees, c)))));
@ -680,7 +703,7 @@ class ClassGraph {
w.print(linePrefix + info.name + "[label="); w.print(linePrefix + info.name + "[label=");
externalTableStart(opt, className, classToUrl(className)); externalTableStart(opt, className, classToUrl(className));
innerTableStart(); innerTableStart();
String qualifiedName = qualifiedName(opt, className); String qualifiedName = qualifiedName(opt, className, null);
int startTemplate = qualifiedName.indexOf('<'); int startTemplate = qualifiedName.indexOf('<');
int idx = qualifiedName.lastIndexOf('.', startTemplate < 0 ? qualifiedName.length() - 1 : startTemplate); int idx = qualifiedName.lastIndexOf('.', startTemplate < 0 ? qualifiedName.length() - 1 : startTemplate);
if (opt.postfixPackage && idx > 0 && idx < (qualifiedName.length() - 1)) { if (opt.postfixPackage && idx > 0 && idx < (qualifiedName.length() - 1)) {

View File

@ -15,6 +15,7 @@ import javax.lang.model.type.ArrayType;
import javax.lang.model.type.DeclaredType; import javax.lang.model.type.DeclaredType;
import javax.lang.model.type.TypeKind; import javax.lang.model.type.TypeKind;
import javax.lang.model.type.TypeMirror; import javax.lang.model.type.TypeMirror;
import javax.lang.model.type.TypeVariable;
import javax.lang.model.util.ElementFilter; import javax.lang.model.util.ElementFilter;
import javax.lang.model.util.Elements; import javax.lang.model.util.Elements;
import javax.lang.model.util.Types; import javax.lang.model.util.Types;
@ -68,6 +69,9 @@ public class ElementUtil {
if (element instanceof TypeElement) { if (element instanceof TypeElement) {
return ((TypeElement) element).getQualifiedName(); return ((TypeElement) element).getQualifiedName();
} }
if (element instanceof TypeVariable) {
return element.getSimpleName();
}
return ""; return "";
} }