package com.stanfy.helium.handler.codegen.java.entity;

import com.stanfy.helium.handler.codegen.java.JavaGeneratorOptions;

import java.util.Arrays;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;

import javax.lang.model.element.Modifier;

import static com.stanfy.helium.handler.codegen.java.entity.Writers.WriterWrapper;

/**
 * Options for POJO generator.
 */
public class EntitiesGeneratorOptions extends JavaGeneratorOptions {

  private static final long serialVersionUID = 1;

  /** Field modifiers. */
  private Set<Modifier> fieldModifiers = Collections.emptySet();

  /** Whether to generate setters. */
  private boolean addSetters;

  /** Whether to generate getters. */
  private boolean addGetters;

  /** Whether to generate toString() method. */
  private boolean addToString = true;

  /** Writer  */
  private WriterWrapper writerWrapper = Writers.chain();

  /** Set of java class names that can be used as parents
   *  in generating entities from messages.
   *
   *  @see com.stanfy.helium.model.Message#parent
   */
  private Set<String> externalParentClasses = new HashSet<String>();

  /** Default options. */
  public static EntitiesGeneratorOptions defaultOptions(final String packageName) {
    EntitiesGeneratorOptions options = new EntitiesGeneratorOptions();
    options.setFieldModifiers(new HashSet<Modifier>(Arrays.asList(Modifier.PUBLIC)));
    options.setPackageName(packageName);
    return options;
  }

  public Set<Modifier> getFieldModifiers() {
    return fieldModifiers;
  }

  public void setFieldModifiers(final Set<Modifier> fieldModifiers) {
    if (fieldModifiers == null) {
      throw new IllegalArgumentException("Field modifiers cannot be null. Provide empty set instead.");
    }
    this.fieldModifiers = fieldModifiers;
  }

  public boolean isAddSetters() {
    return addSetters;
  }

  public void setAddSetters(final boolean addSetters) {
    this.addSetters = addSetters;
  }

  public boolean isAddGetters() {
    return addGetters;
  }

  public void setAddGetters(final boolean addGetters) {
    this.addGetters = addGetters;
  }

  public boolean isAddToString() {
    return addToString;
  }

  public void setAddToString(final boolean addToString) {
    this.addToString = addToString;
  }

  public WriterWrapper getWriterWrapper() {
    return writerWrapper;
  }

  public void setWriterWrapper(final WriterWrapper writerWrapper) {
    if (writerWrapper == null) {
      throw new IllegalArgumentException("Writer wrapper cannot be null");
    }
    this.writerWrapper = writerWrapper;
  }

  public Set<String> getExternalParentClasses() {
    return externalParentClasses;
  }

  public void setExternalParentClasses(final Set<String> externalParentClasses) {
    this.externalParentClasses = externalParentClasses;
  }
}

@@ -906,16 +906,21 @@
 ;%0A%0A  /**
+%0A   *
  Set of 
@@ -968,17 +968,16 @@
 nts%0A   *
- 
  in gene
@@ -1016,17 +1016,16 @@
   *%0A   *
- 
  @see co

Fix javadocs according to comments.