FieldToSerialize.java

package com.cedarsoft.serialization.generator.intellij.model;

import com.intellij.openapi.project.Project;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.psi.PsiField;
import com.intellij.psi.PsiMethod;
import com.intellij.psi.PsiType;
import com.intellij.psi.codeStyle.JavaCodeStyleManager;
import com.intellij.psi.codeStyle.VariableKind;
import com.intellij.psi.util.PropertyUtil;
import com.intellij.psi.util.TypeConversionUtil;

import javax.annotation.Nonnull;

/**
 * Describes a field that has been serialized
 *
 * @author Johannes Schneider (<a href="mailto:js@cedarsoft.com">js@cedarsoft.com</a>)
 */
public class FieldToSerialize {
  @Nonnull
  private final PsiType fieldType;
  @Nonnull
  private final PsiField field;
  @Nonnull
  private final String fieldName;
  @Nonnull
  private final FieldSetter fieldSetter;
  @Nonnull
  private final String accessor;
  @Nonnull
  private final String propertyConstant;
  @Nonnull
  private final String defaultValue;

  public FieldToSerialize( @Nonnull PsiType fieldType, @Nonnull PsiField field, @Nonnull FieldSetter fieldSetter ) {
    this.fieldType = fieldType;
    this.field = field;
    this.fieldName = field.getName();
    this.fieldSetter = fieldSetter;

    this.accessor = findGetterName( field );
    this.propertyConstant = "PROPERTY_" + JavaCodeStyleManager.getInstance( getProject() ).suggestVariableName( VariableKind.STATIC_FINAL_FIELD, field.getName(), null, fieldType ).names[0];

    defaultValue = getDefaultValue( fieldType );
  }

  @Nonnull
  public PsiField getField() {
    return field;
  }

  @Nonnull
  public FieldSetter getFieldSetter() {
    return fieldSetter;
  }

  @Nonnull
  public PsiType getFieldType() {
    return fieldType;
  }

  @Nonnull
  public String getFieldName() {
    return fieldName;
  }

  @Nonnull
  public String getAccessor() {
    return accessor;
  }

  @Nonnull
  public String getPropertyConstantName() {
    return propertyConstant;
  }

  @Nonnull
  public String getDefaultValue() {
    return defaultValue;
  }

  public boolean shallVerifyDeserialized() {
    return !PsiType.BOOLEAN.equals( fieldType );
  }

  @Nonnull
  private Project getProject() {
    return field.getProject();
  }

  @Nonnull
  public String getFieldTypeBoxed() {
    return DelegatingSerializer.box( getFieldType() );
  }

  public boolean isPrimitive() {
    return isPrimitive( getFieldType() );
  }

  /**
   * Returns the default value for a given type
   *
   * @param fieldType the field type
   * @return the default value
   */
  @Nonnull
  private static String getDefaultValue( @Nonnull PsiType fieldType ) {
    if ( isPrimitive( fieldType ) ) {
      if ( TypeConversionUtil.isBooleanType( fieldType ) ) {
        return "false";
      }
      if ( PsiType.CHAR.equals( fieldType ) ) {
        return "(char)-1";
      }

      return "-1";
    } else {
      return "null";
    }
  }

  /**
   * Whether the given field type is a primitive
   *
   * @param fieldType the field type
   * @return true if the given type is a primitive, false otherwise
   */
  public static boolean isPrimitive( @Nonnull PsiType fieldType ) {
    return TypeConversionUtil.isPrimitiveAndNotNull( fieldType );
  }

  /**
   * Returns the getter for a given field
   *
   * @param field the field
   * @return the getter name
   */
  @Nonnull
  public static String findGetterName( @Nonnull PsiField field ) {
    PsiMethod getter = PropertyUtil.findGetterForField( field );
    if ( getter != null ) {
      return getter.getName() + "()";

    }
    return "get" + StringUtil.capitalizeWithJavaBeanConvention( field.getName() ) + "()";
  }
}