ManyToManyProperty.java

package joist.codegen.dtos;

import java.util.List;

import joist.codegen.Config;
import joist.codegen.InformationSchemaColumn;
import joist.util.Inflector;

import org.apache.commons.lang.StringUtils;

public class ManyToManyProperty {

  private Config config;
  private String myKeyColumnName;
  private Entity joinTable;
  private Entity mySide;
  private Entity targetTable;
  private ManyToManyProperty other;

  public ManyToManyProperty(Config config, Entity joinTable, Entity mySide, Entity otherSide, InformationSchemaColumn column) {
    this.config = config;
    this.joinTable = joinTable;
    this.mySide = mySide;
    this.targetTable = otherSide;
    this.myKeyColumnName = column.name;
  }

  public String getCapitalVariableNameSingular() {
    return Inflector.camelize(this.other.myKeyColumnName.replaceAll("_id", "")); // rolled_id -> Rolled
  }

  public String getCapitalVariableName() {
    return this.getCapitalVariableNameSingular() + "s";
  }

  public String getJavaType() {
    return "List<" + this.getTargetJavaType() + ">";
  }

  public String getDefaultJavaString() {
    return "new ArrayList<" + this.getTargetJavaType() + ">()";
  }

  public String getTargetJavaType() {
    return this.targetTable.getClassName();
  }

  public List<String> getCustomRules() {
    return this.config.getCustomRules(this.mySide.getClassName(), this.getJavaType(), this.getVariableName());
  }

  public String getVariableName() {
    return StringUtils.uncapitalize(this.getCapitalVariableName());
  }

  public Entity getTargetTable() {
    return this.targetTable;
  }

  public ManyToManyProperty getOther() {
    return this.other;
  }

  public void setOther(ManyToManyProperty other) {
    if (other == this) {
      throw new IllegalStateException("Wrong other");
    }
    this.other = other;
  }

  public ManyToOneProperty getMySideManyToOne() {
    for (ManyToOneProperty mtop : this.joinTable.getManyToOneProperties()) {
      if (mtop.getColumnName().equals(this.myKeyColumnName)) {
        return mtop;
      }
    }
    throw new RuntimeException("No ManyToOne found for " + this.myKeyColumnName);
  }

  public OneToManyProperty getMySideOneToMany() {
    for (OneToManyProperty otom : this.mySide.getOneToManyProperties()) {
      if (otom.getManySide().equals(this.joinTable)) {
        return otom;
      }
    }
    throw new RuntimeException("No OneToMany found for " + this.myKeyColumnName);
  }

  public Entity getJoinTable() {
    return this.joinTable;
  }

  public String toString() {
    return this.mySide.getClassName() + "." + this.getVariableName() + " -> " + this.joinTable.getClassName() + "." + this.myKeyColumnName;
  }
}