io.avaje.prism.internal.PrismWriter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of avaje-prism-core Show documentation
generator for avaje prisms
The newest version!
package io.avaje.prism.internal;

import java.io.PrintWriter;

import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.TypeElement;

class PrismWriter {
  String name;
  String mirrorType;
  String prismType;
  boolean arrayed;
  ExecutableElement m;

  PrismWriter(ExecutableElement m, boolean arrayed) {
    this.m = m;
    this.arrayed = arrayed;
    this.name = m.getSimpleName().toString();
  }

  public void setPrismType(String prismType) {
    this.prismType = prismType;
  }

  public void setMirrorType(String mirrorType) {
    this.mirrorType = mirrorType;
  }

  void writeField(String indent, PrintWriter out) {
    out.format("%s  /** store prism value of %s */\n", indent, name);
    if (arrayed) {
      out.format("%s  private final List<%s> _%s;\n\n", indent, prismType, name);
    } else {
      out.format("%s  private final %s _%s;\n\n", indent, prismType, name);
    }
  }

  /* return source code that converts an expr of mirrorType to prismType. */
  String mirror2prism(String expr) {
    return String.format(m2pFormat, expr);
  }

  String m2pFormat = "undefinedConverter(%s)";

  public void setM2pFormat(String m2pFormat) {
    this.m2pFormat = m2pFormat;
  }

  void writeInitializer(String indent, PrintWriter out) {
    if (arrayed) {
      if (mirrorType.equals(prismType)) {
        out.format("%s    _%s = getArrayValues(\"%s\", %s.class);\n", indent, name, name, prismType);
      } else {
        out.format("%s    List<%s> %sMirrors = getArrayValues(\"%s\", %s.class);\n", indent, mirrorType, name, name, mirrorType);
        out.format("%s     _%s = new ArrayList<%s>(%sMirrors.size());\n", indent, name, prismType, name);
        out.format("%s    for(%s %sMirror : %sMirrors) {\n", indent, mirrorType, name, name);
        out.format("%s        _%s.add(%s);\n", indent, name, mirror2prism(name + "Mirror"));
        out.format("%s    }\n", indent);
      }
    } else if (mirrorType.equals(prismType)) {
      out.format("%s    _%s = getValue(\"%s\", %s.class);\n", indent, name, name, prismType);
    } else {
      out.format("%s    %s %sMirror = getValue(\"%s\", %s.class);\n", indent, mirrorType, name, name, mirrorType);
      out.format("%s    valid = valid && %sMirror != null;\n", indent, name);
      out.format("%s    _%s = %sMirror == null ? null : %s;\n",
          indent, name, name, mirror2prism(name + "Mirror"));
    }
  }

  void writeMethod(String indent, PrintWriter out) {
    if (arrayed) {
      out.format("%s  /** \n", indent);
      out.format("%s   * Returns a List<%s> representing the value of the {@code %s} member of the Annotation.\n", indent, prismType, m);
      out.format("%s   * @see %s#%s()\n", indent, ((TypeElement) m.getEnclosingElement()).getQualifiedName(), name);
      out.format("%s   */ \n", indent);
      out.format("%s  public List<%s> %s() { return _%s; }\n\n", indent, prismType, name, name);
    } else {
      out.format("%s  /** \n", indent);
      out.format("%s   * Returns a %s representing the value of the {@code %s %s} member of the Annotation.\n", indent, prismType, m.getReturnType(), m);
      out.format("%s   * @see %s#%s()\n", indent, ((TypeElement) m.getEnclosingElement()).getQualifiedName(), name);
      out.format("%s   */ \n", indent);
      out.format("%s  public %s %s() { return _%s; }\n\n", indent, prismType, name, name);
    }
  }
}