org.jaxdb.jsql.generator.GeneratorUtil Maven / Gradle / Ivy

Go to download
/* Copyright (c) 2014 JAX-DB
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * You should have received a copy of The MIT License (MIT) along with this
 * program. If not, see .
 */

package org.jaxdb.jsql.generator;

import java.lang.reflect.Field;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.LocalTime;

import org.jaxdb.jsql.GenerateOn;
import org.jaxdb.jsql.data;
import org.jaxdb.www.ddlx_0_5.xLygluGCXAA.$Documented;
import org.libj.lang.Strings;
import org.libj.math.BigInt;
import org.openjax.xml.datatype.HexBinary;
import org.w3.www._2001.XMLSchema;

final class GeneratorUtil {
  static final Object THIS = new Object();
  static final Object MUTABLE = new Object();
  static final Object PRIMARY_KEY = new Object();
  static final Object KEY_FOR_UPDATE = new Object();
  static final Object COMMIT_UPDATE_CHANGE = new Object();

  private static StringBuilder indent(final StringBuilder b, final int depth) {
    for (int i = 0, i$ = depth * 2; i < i$; ++i) // [N]
      b.append(' ');

    return b;
  }

  private static StringBuilder appendDocInfo(final StringBuilder b, final String type, final String name, final int depth, final Object[] args) {
    if (args.length == 0)
      return b;

    indent(b, depth).append(" * \n");
    indent(b, depth).append(" * \n");
    indent(b, depth).append(" * \n");
    for (int i = 0, i$ = args.length; i < i$;) { // [A]
      final Object property = args[i++];
      final Object value = args[i++];
      indent(b, depth).append(" * \n");
    }
    indent(b, depth).append(" * ").append(type).append(" \"").append(name).append("\"").append(property).append(" ").append(value).append("\n");
    indent(b, depth).append(" * \n");

    return b;
  }

  static String getDoc(final $Documented documented, final int depth, final char start, final char end, final String type, final String name, final Object ... args) {
    String str = "";
    final XMLSchema.yAA.$String doc = documented.getDocumentation();
    if ((doc == null || (str = doc.text().trim()).length() == 0) && args.length == 0)
      return "";

    final StringBuilder b = new StringBuilder();
    if (start != '\0')
      b.append(start);

    indent(b, depth).append("/**\n");
    if (str.length() > 0) {
      indent(b, depth).append(" * ").append(str).append('\n');
      indent(b, depth).append(" *\n");
    }

    appendDocInfo(b, type, name, depth, args);
    indent(b, depth).append(" */");
    if (end != '\0')
      b.append(end);

    return b.toString();
  }

  @SuppressWarnings("rawtypes")
  static String compile(final Object object, final Class type) {
    if (object == null)
      return "null";

    if (object instanceof StringBuilder)
      return object.toString();

    if (object instanceof String) {
      if (type == data.DATETIME.class) {
        final String[] dateTime = Strings.split(((String)object), ' ');
        final String[] date = Strings.split(dateTime[0], '-');
        final String[] time = Strings.split(dateTime[1], ':');
        final int d = time[2].indexOf('.');
        String nanos = null;
        if (d > -1) {
          nanos = String.format("%-9s", time[2].substring(d + 1)).replace(' ', '0');
          time[2] = time[2].substring(0, d);
        }

        return LocalDateTime.class.getName() + ".of(" + date[0] + ", " + date[1] + ", " + date[2] + ", " + time[0] + ", " + time[1] + ", " + time[2] + (nanos != null ? ", " + nanos : "") + ")";
      }

      if (type == data.DATE.class) {
        final String[] date = Strings.split((String)object, '-');
        return LocalDate.class.getName() + ".of(" + date[0] + ", " + date[1] + ", " + date[2] + ")";
      }

      if (type == data.TIME.class) {
        final String[] time = Strings.split(((String)object), ':');
        final int d = time[2].indexOf('.');
        String nanos = null;
        if (d > -1) {
          nanos = String.format("%-9s", time[2].substring(d + 1)).replace(' ', '0');
          time[2] = time[2].substring(0, d);
        }

        return LocalTime.class.getName() + ".of(" + time[0] + ", " + time[1] + ", " + time[2] + (nanos != null ? ", " + nanos : "") + ")";
      }

      return "\"" + ((String)object).replace("\"", "\\\"").replace("\n", "\\n") + "\"";
    }

    if (object instanceof HexBinary) {
      final StringBuilder builder = new StringBuilder("new byte[] ");
      for (final byte b : ((HexBinary)object).getBytes()) // [A]
        builder.append(String.format(" (byte)0x%02x,", b));

      builder.setCharAt(11, '{');
      builder.setCharAt(builder.length() - 1, '}');
      return builder.toString();
    }

    if (object instanceof Byte)
      return "(byte)" + object;

    if (object instanceof Short)
      return "(short)" + object;

    if (object instanceof Float)
      return object + "f";

    if (object instanceof Double)
      return object + "d";

    if (object instanceof BigInteger)
      return "new " + BigInt.class.getName() + "(\"" + object + "\")";

    if (object instanceof BigDecimal)
      return "new " + BigDecimal.class.getName() + "(\"" + object + "\")";

    if (object instanceof Long)
      return object + "L";

    if (object instanceof GenerateOn) {
      try {
        final Field[] fields = GenerateOn.class.getDeclaredFields();
        for (final Field field : fields) // [A]
          if (field.get(null) == object)
            return GenerateOn.class.getName() + "." + field.getName();

        throw new IllegalStateException("Did not find the desired field");
      }
      catch (final IllegalAccessException e) {
        throw new IllegalStateException(e);
      }
    }

    return String.valueOf(object);
  }

  private GeneratorUtil() {
  }
}