private static ClassHierarchyProto.Node serializeNode(Node n) {
List<ClassHierarchyProto.Node> children = new ArrayList<>();
for (Node child : n.getChildren()) {
children.add(serializeNode(child));
}
if (n instanceof ClassNode) {
ClassNode<?> cn = (ClassNode<?>) n;
ConstructorDef<?>[] injectable = cn.getInjectableConstructors();
ConstructorDef<?>[] all = cn.getAllConstructors();
List<ConstructorDef<?>> others = new ArrayList<>(Arrays.asList(all));
others.removeAll(Arrays.asList(injectable));
List<ClassHierarchyProto.ConstructorDef> injectableConstructors = new ArrayList<>();
for (ConstructorDef<?> inj : injectable) {
injectableConstructors.add(serializeConstructorDef(inj));
}
List<ClassHierarchyProto.ConstructorDef> otherConstructors = new ArrayList<>();
for (ConstructorDef<?> other : others) {
otherConstructors.add(serializeConstructorDef(other));
}
List<String> implFullNames = new ArrayList<>();
for (ClassNode<?> impl : cn.getKnownImplementations()) {
implFullNames.add(impl.getFullName());
}
return newClassNode(
cn.getName(),
cn.getFullName(),
cn.isInjectionCandidate(),
cn.isExternalConstructor(),
cn.isUnit(),
injectableConstructors,
otherConstructors,
implFullNames,
children);
} else if (n instanceof NamedParameterNode) {
NamedParameterNode<?> np = (NamedParameterNode<?>) n;
return newNamedParameterNode(
np.getName(),
np.getFullName(),
np.getSimpleArgName(),
np.getFullArgName(),
np.isSet(),
np.isList(),
np.getDocumentation(),
np.getShortName(),
np.getDefaultInstanceAsStrings(),
children);
} else if (n instanceof PackageNode) {
return newPackageNode(n.getName(), n.getFullName(), children);
} else {
throw new IllegalStateException("Encountered unknown type of Node: " + n);
}
}
@Override
public ClassHierarchy merge(ClassHierarchy ch) {
if (this == ch) {
return this;
}
if (!(ch instanceof ProtocolBufferClassHierarchy)) {
throw new UnsupportedOperationException(
"Cannot merge with class hierarchies of type: " + ch.getClass().getName());
}
final ProtocolBufferClassHierarchy pch = (ProtocolBufferClassHierarchy) ch;
for (final String key : pch.lookupTable.keySet()) {
if (!this.lookupTable.containsKey(key)) {
this.lookupTable.put(key, pch.lookupTable.get(key));
}
for (final Node n : ch.getNamespace().getChildren()) {
if (!this.namespace.contains(n.getFullName())) {
if (n instanceof NamedParameter) {
final NamedParameterNode np = (NamedParameterNode) n;
new NamedParameterNodeImpl<>(
this.namespace,
np.getName(),
np.getFullName(),
np.getFullArgName(),
np.getSimpleArgName(),
np.isSet(),
np.isList(),
np.getDocumentation(),
np.getShortName(),
np.getDefaultInstanceAsStrings());
} else if (n instanceof ClassNode) {
final ClassNode cn = (ClassNode) n;
new ClassNodeImpl(
namespace,
cn.getName(),
cn.getFullName(),
cn.isUnit(),
cn.isInjectionCandidate(),
cn.isExternalConstructor(),
cn.getInjectableConstructors(),
cn.getAllConstructors(),
cn.getDefaultImplementation());
}
}
}
}
return this;
}
/**
* @param args
* @throws FileNotFoundException
* @throws MalformedURLException
*/
public static void main(final String[] args)
throws FileNotFoundException, MalformedURLException, UnsupportedEncodingException {
int i = 0;
String doc = null;
String jar = null;
boolean tangTests = false;
while (i < args.length) {
if (args[i].equals("--doc")) {
i++;
doc = args[i];
} else if (args[i].equals("--jar")) {
i++;
jar = args[i];
} else if (args[i].equals("--tang-tests")) {
tangTests = true;
}
i++;
}
final Tint t;
if (jar != null) {
final File f = new File(jar);
if (!f.exists()) {
throw new FileNotFoundException(jar);
}
t = new Tint(new URL[] {f.toURI().toURL()}, tangTests);
} else {
t = new Tint(new URL[0], tangTests);
}
if (doc != null) {
try (final PrintStream out = new PrintStream(doc, "UTF-8")) {
out.println("<html><head><title>TangDoc</title>");
out.println("<style>");
out.println(
"body { font-family: 'Segoe UI', 'Comic Sans MS'; font-size:12pt; font-weight: 200; "
+ "margin: 1em; column-count: 2; }");
out.println(".package { font-size:18pt; font-weight: 500; column-span: all; }");
// out.println(".class { break-after: never; }");
// out.println(".doc { break-before: never; }");
out.println(".decl-margin { padding: 8pt; break-inside: avoid; }");
out.println(".module-margin { padding: 8pt; column-span: all; break-inside: avoid; }");
out.println(".decl { background-color: aliceblue; padding: 6pt;}");
out.println(".fullName { font-size: 11pt; font-weight: 400; }");
out.println(".simpleName { font-size: 11pt; font-weight: 400; }");
out.println(".constructorArg { padding-left: 16pt; }");
out.println("." + SETTERS + " { padding-top: 6pt; font-size: 10pt; }");
out.println("." + USES + " { padding-top: 6pt; font-size: 10pt; }");
out.println("pre { font-size: 10pt; }");
out.println("</style>");
out.println("</head><body>");
String currentPackage = "";
for (final Node n : t.getNamesUsedAndSet()) {
final String fullName = n.getFullName();
final String[] tok = fullName.split("\\.");
final StringBuffer sb = new StringBuffer(tok[0]);
for (int j = 1; j < tok.length; j++) {
if (tok[j].matches("^[A-Z].*") || j > 4) {
break;
} else {
sb.append("." + tok[j]);
}
}
final String pack = sb.toString();
if (!currentPackage.equals(pack)) {
currentPackage = pack;
out.println(t.endPackage());
out.println(t.startPackage(currentPackage));
}
if (n instanceof NamedParameterNode<?>) {
out.println(t.toHtmlString((NamedParameterNode<?>) n, currentPackage));
} else if (n instanceof ClassNode<?>) {
out.println(t.toHtmlString((ClassNode<?>) n, currentPackage));
} else {
throw new IllegalStateException();
}
}
out.println("</div>");
out.println(t.endPackage());
out.println("<div class='package'>Module definitions</div>");
for (final Field f : t.modules.keySet()) {
final String moduleName = ReflectionUtilities.getFullName(f);
// String declaringClassName =
// ReflectionUtilities.getFullName(f.getDeclaringClass());
out.println(
"<div class='module-margin' id='"
+ moduleName
+ "'><div class='decl'><span class='fullName'>"
+ moduleName
+ "</span>");
out.println("<pre>");
final String conf = t.modules.get(f).toPrettyString();
final String[] tok = conf.split("\n");
for (final String line : tok) {
out.println(stripPrefix(line, "no.such.prefix")); // t.modules.get(f).toPrettyString());
}
// List<Entry<String,String>> lines = t.modules.get(f).toStringPairs();
// for(Entry<String,String> line : lines) {
// String k = t.stripPrefix(line.getKey(), declaringClassName);
// String v = t.stripPrefix(line.getValue(), declaringClassName);
// out.println(k+"="+v);
// }
out.println("</pre>");
out.println("</div></div>");
}
out.println("<div class='package'>Interfaces and injectable classes</div>");
for (final ClassNode<?> c : t.knownClasses) {
if (t.classFilter(tangTests, c.getFullName())) {
Class<?> clz = null;
try {
clz = t.ch.classForName(c.getFullName());
} catch (final ClassNotFoundException e) {
// TODO[JIRA REEF-864] Clarify handling in this case
e.printStackTrace();
}
final String typ =
clz == null ? "undefined" : clz.isInterface() ? "interface" : "class";
out.println(
"<div class='module-margin' id='"
+ c.getFullName()
+ "'><div class='decl'>"
+ "<span class='fullName'>"
+ typ
+ " "
+ c.getFullName()
+ "</span>");
for (final ConstructorDef<?> d : c.getInjectableConstructors()) {
out.println("<div class='uses'>" + c.getFullName() + "(");
for (final ConstructorArg a : d.getArgs()) {
if (a.getNamedParameterName() != null) {
out.print(
"<div class='constructorArg'><a href='#"
+ a.getType()
+ "'>"
+ stripPrefix(a.getType(), "xxx")
+ "</a> <a href='#"
+ a.getNamedParameterName()
+ "'>"
+ a.getNamedParameterName()
+ "</a></div>");
} else {
out.print(
"<div class='constructorArg'><a href='#"
+ a.getType()
+ "'>"
+ stripPrefix(a.getType(), "xxx")
+ "</a></div>");
}
}
out.println(")</div>");
}
out.println("</div></div>");
}
/*
out.println("<h1>Default usage of classes and constants</h1>");
for(String s : t.usages.keySet()) {
out.println("<h2>" + s + "</h2>");
for(Node n : t.usages.getValuesForKey(s)) {
out.println("<p>" + n.getFullName() + "</p>");
}
} */
}
out.println("</body></html>");
}
}
}
private Node register(final String s) {
final Class<?> c;
try {
c = classForName(s);
} catch (final ClassNotFoundException e1) {
return null;
}
try {
final Node n = getAlreadyBoundNode(c);
return n;
} catch (final NameResolutionException e) {
// node not bound yet
}
// First, walk up the class hierarchy, registering all out parents. This
// can't be loopy.
if (c.getSuperclass() != null) {
register(ReflectionUtilities.getFullName(c.getSuperclass()));
}
for (final Class<?> i : c.getInterfaces()) {
register(ReflectionUtilities.getFullName(i));
}
// Now, we'd like to register our enclosing classes. This turns out to be
// safe.
// Thankfully, Java doesn't allow:
// class A implements A.B { class B { } }
// It also doesn't allow cycles such as:
// class A implements B.BB { interface AA { } }
// class B implements A.AA { interface BB { } }
// So, even though grafting arbitrary DAGs together can give us cycles, Java
// seems
// to have our back on this one.
final Class<?> enclosing = c.getEnclosingClass();
if (enclosing != null) {
register(ReflectionUtilities.getFullName(enclosing));
}
// Now register the class. This has to be after the above so we know our
// parents (superclasses and enclosing packages) are already registered.
final Node n = registerClass(c);
// Finally, do things that might introduce cycles that invlove c.
// This has to be below registerClass, which ensures that any cycles
// this stuff introduces are broken.
for (final Class<?> innerClass : c.getDeclaredClasses()) {
register(ReflectionUtilities.getFullName(innerClass));
}
if (n instanceof ClassNode) {
final ClassNode<?> cls = (ClassNode<?>) n;
for (final ConstructorDef<?> def : cls.getInjectableConstructors()) {
for (final ConstructorArg arg : def.getArgs()) {
register(arg.getType());
if (arg.getNamedParameterName() != null) {
final NamedParameterNode<?> np =
(NamedParameterNode<?>) register(arg.getNamedParameterName());
try {
// TODO: When handling sets, need to track target of generic parameter, and check the
// type here!
if (!np.isSet()
&& !np.isList()
&& !ReflectionUtilities.isCoercable(
classForName(arg.getType()), classForName(np.getFullArgName()))) {
throw new ClassHierarchyException(
"Named parameter type mismatch in "
+ cls.getFullName()
+ ". Constructor expects a "
+ arg.getType()
+ " but "
+ np.getName()
+ " is a "
+ np.getFullArgName());
}
} catch (final ClassNotFoundException e) {
throw new ClassHierarchyException(
"Constructor refers to unknown class " + arg.getType(), e);
}
}
}
}
} else if (n instanceof NamedParameterNode) {
final NamedParameterNode<?> np = (NamedParameterNode<?>) n;
register(np.getFullArgName());
}
return n;
}