utype can return the mirror (#118)

Now can get the mirror of the Utype

Author: SentryMan

prism-core/src/main/java/io/avaje/prism/internal/UTypeWriter.java

@@ -43,6 +43,13 @@ public static void write(PrintWriter out, String packageName) {
             + "  Set<String> importTypes();\n"
             + "\n"
             + "  /**\n"
+            + "   * Return the type mirror of this Utype.\n"
+            + "   *\n"
+            + "   * @return the typemirror\n"
+            + "   */\n"
+            + "  TypeMirror mirror();\n"
+            + "\n"
+            + "  /**\n"
             + "   * Return the full type as a code safe string. (with annotations if present)\n"
             + "   *\n"
             + "   * @return the full typeName\n"

prism-core/src/main/java/io/avaje/prism/internal/VisitorWriter.java

@@ -40,6 +40,7 @@ public static void write(PrintWriter out, String packageName) {
             + "\n"
             + "  private final Map<TypeVariable, String> typeVariables;\n"
             + "  private Set<String> allTypes = new HashSet<>();\n"
+            + "  private TypeMirror typeMirror;\n"
             + "  private String mainType;\n"
             + "  private String fullType;\n"
             + "  private final List<UType> params = new ArrayList<>();\n"
@@ -75,6 +76,11 @@ public static void write(PrintWriter out, String packageName) {
             + "    return allTypes.stream().filter(this::notJavaLang).collect(toSet());\n"
             + "  }\n"
             + "\n"
+            + "  @Override\n"
+            + "  public TypeMirror mirror() {\n"
+            + "    return typeMirror;\n"
+            + "  }\n"
+            + "\n"
             + "  private boolean notJavaLang(String type) {\n"
             + "    return !type.startsWith(\"java.lang.\") || Character.isLowerCase(type.charAt(10));\n"
             + "  }\n"
@@ -163,6 +169,7 @@ public static void write(PrintWriter out, String packageName) {
             + "\n"
             + "  @Override\n"
             + "  public StringBuilder visitPrimitive(PrimitiveType t, StringBuilder p) {\n"
+            + "    typeMirror = t;\n"
             + "    kind = t.getKind();\n"
             + "    if (includeAnnotations) {\n"
             + "      for (final var ta : t.getAnnotationMirrors()) {\n"
@@ -182,11 +189,13 @@ public static void write(PrintWriter out, String packageName) {
             + "\n"
             + "  @Override\n"
             + "  public StringBuilder visitNull(NullType t, StringBuilder p) {\n"
+            + "    typeMirror = t;\n"
             + "    return p;\n"
             + "  }\n"
             + "\n"
             + "  @Override\n"
             + "  public StringBuilder visitArray(ArrayType t, StringBuilder p) {\n"
+            + "    typeMirror = t;\n"
             + "    kind = t.getKind();\n"
             + "    boolean mainUnset = this.mainType == null;\n"
             + "    final var ct = t.getComponentType();\n"
@@ -212,6 +221,7 @@ public static void write(PrintWriter out, String packageName) {
             + "\n"
             + "  @Override\n"
             + "  public StringBuilder visitDeclared(DeclaredType t, StringBuilder p) {\n"
+            + "    typeMirror = t;\n"
             + "    kind = t.getKind();\n"
             + "    final String fqn = fullyQualifiedName(t, includeAnnotations);\n"
             + "    var trimmed = fullyQualifiedName(t, false);\n"
@@ -303,13 +313,15 @@ public static void write(PrintWriter out, String packageName) {
             + "\n"
             + "  @Override\n"
             + "  public StringBuilder visitError(ErrorType t, StringBuilder p) {\n"
+            + "    typeMirror = t;\n"
             + "    kind = t.getKind();\n"
             + "    mainType = t.toString();\n"
             + "    return p.append(mainType);\n"
             + "  }\n"
             + "\n"
             + "  @Override\n"
             + "  public StringBuilder visitTypeVariable(TypeVariable t, StringBuilder p) {\n"
+            + "    typeMirror = t;\n"
             + "    kind = t.getKind();\n"
             + "    // Types can be recursive so we have to check if we have already done this type.\n"
             + "    final String previous = typeVariables.get(t);\n"
@@ -344,6 +356,7 @@ public static void write(PrintWriter out, String packageName) {
             + "\n"
             + "  @Override\n"
             + "  public StringBuilder visitWildcard(WildcardType t, StringBuilder p) {\n"
+            + "    typeMirror = t;\n"
             + "    kind = t.getKind();\n"
             + "    final var extendsBound = t.getExtendsBound();\n"
             + "    final var superBound = t.getSuperBound();\n"
@@ -371,13 +384,15 @@ public static void write(PrintWriter out, String packageName) {
             + "\n"
             + "  @Override\n"
             + "  public StringBuilder visitNoType(NoType t, StringBuilder p) {\n"
+            + "    typeMirror = t;\n"
             + "    kind = t.getKind();\n"
             + "    mainType = t.toString();\n"
             + "    return p.append(mainType);\n"
             + "  }\n"
             + "\n"
             + "  @Override\n"
             + "  public StringBuilder visitIntersection(IntersectionType t, StringBuilder p) {\n"
+            + "    typeMirror = t;\n"
             + "    kind = t.getKind();\n"
             + "    boolean first = true;\n"
             + "    for (final var b : t.getBounds()) {\n"
@@ -393,6 +408,7 @@ public static void write(PrintWriter out, String packageName) {
             + "\n"
             + "  @Override\n"
             + "  public StringBuilder visitUnion(UnionType t, StringBuilder p) {\n"
+            + "    typeMirror = t;\n"
             + "    kind = t.getKind();\n"
             + "    boolean first = true;\n"
             + "    for (final var b : t.getAlternatives()) {\n"