package org.inferred.freebuilder.processor;

import static org.inferred.freebuilder.processor.model.ModelUtils.findAnnotationMirror;

import com.google.common.collect.ImmutableSet;

import org.inferred.freebuilder.processor.property.Property;
import org.inferred.freebuilder.processor.source.Excerpts;
import org.inferred.freebuilder.processor.source.QualifiedName;

import java.util.Optional;
import java.util.Set;

import javax.lang.model.element.AnnotationMirror;
import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.Name;
import javax.lang.model.element.TypeElement;
import javax.lang.model.util.Elements;

class JacksonSupport {

  private enum GenerateAnnotation {
    DEFAULT,
    JSON_ANY,
    NONE
  }

  private static final String JSON_DESERIALIZE =
      "com.fasterxml.jackson.databind.annotation.JsonDeserialize";
  private static final QualifiedName JSON_PROPERTY =
      QualifiedName.of("com.fasterxml.jackson.annotation", "JsonProperty");
  private static final String JACKSON_XML_ANNOTATION_PACKAGE =
      "com.fasterxml.jackson.dataformat.xml.annotation";
  private static final QualifiedName JSON_ANY_GETTER =
          QualifiedName.of("com.fasterxml.jackson.annotation", "JsonAnyGetter");
  private static final QualifiedName JSON_ANY_SETTER =
          QualifiedName.of("com.fasterxml.jackson.annotation", "JsonAnySetter");
  /** Annotations which disable automatic generation of JsonProperty annotations. */
  private static final Set<QualifiedName> DISABLE_PROPERTY_ANNOTATIONS = ImmutableSet.of(
      QualifiedName.of("com.fasterxml.jackson.annotation", "JsonIgnore"),
      QualifiedName.of("com.fasterxml.jackson.annotation", "JsonUnwrapped"),
      QualifiedName.of("com.fasterxml.jackson.annotation", "JsonValue"));

  public static Optional<JacksonSupport> create(TypeElement userValueType, Elements elements) {
    return findAnnotationMirror(userValueType, JSON_DESERIALIZE)
        .map($ -> new JacksonSupport(elements));
  }

  private final Elements elements;

  private JacksonSupport(Elements elements) {
    this.elements = elements;
  }

  public void addJacksonAnnotations(
      Property.Builder resultBuilder,
      ExecutableElement getterMethod) {
    Optional<AnnotationMirror> jsonPropertyAnnotation = findAnnotationMirror(getterMethod,
            JSON_PROPERTY);
    if (jsonPropertyAnnotation.isPresent()) {
      resultBuilder.addAccessorAnnotations(Excerpts.add("%s%n", jsonPropertyAnnotation.get()));
    } else {
      switch (generateDefaultAnnotations(getterMethod)) {
        case DEFAULT:
          resultBuilder.addAccessorAnnotations(Excerpts.add(
                  "@%s(\"%s\")%n", JSON_PROPERTY, resultBuilder.getName()));
          break;
        case JSON_ANY:
          resultBuilder.addPutAnnotations(Excerpts.add("@%s%n", JSON_ANY_SETTER));
          resultBuilder.addGetterAnnotations(Excerpts.add("@%s%n", JSON_ANY_GETTER));
          break;
        case NONE:
        default:
          break;
      }
    }

    getterMethod
        .getAnnotationMirrors()
        .stream()
        .filter(this::isXmlAnnotation)
        .forEach(annotation -> {
          resultBuilder.addAccessorAnnotations(code -> code.addLine("%s", annotation));
        });
  }

  private boolean isXmlAnnotation(AnnotationMirror mirror) {
    Name pkg = elements.getPackageOf(mirror.getAnnotationType().asElement()).getQualifiedName();
    return pkg.contentEquals(JACKSON_XML_ANNOTATION_PACKAGE);
  }

  private static GenerateAnnotation generateDefaultAnnotations(ExecutableElement getterMethod) {
    for (AnnotationMirror annotationMirror : getterMethod.getAnnotationMirrors()) {
      TypeElement annotationTypeElement =
          (TypeElement) (annotationMirror.getAnnotationType().asElement());
      QualifiedName annotationType = QualifiedName.of(annotationTypeElement);
      if (DISABLE_PROPERTY_ANNOTATIONS.contains(annotationType)) {
        return GenerateAnnotation.NONE;
      }
      if (JSON_ANY_GETTER.equals(annotationType)) {
        return GenerateAnnotation.JSON_ANY;
      }
    }
    return GenerateAnnotation.DEFAULT;
  }
}

@@ -1153,36 +1153,32 @@
 _GETTER =%0A      
-    
 QualifiedName.of
@@ -1287,20 +1287,16 @@
 ETTER =%0A
-    
       Qu
@@ -2267,16 +2267,24 @@
 tation =
+%0A       
  findAnn
@@ -2310,28 +2310,16 @@
 rMethod,
-%0A           
  JSON_PR
@@ -2613,24 +2613,9 @@
 ons(
-Excerpts.add(%0A  
+%0A
     
@@ -2624,18 +2624,29 @@
         
-  
+Excerpts.add(
 %22@%25s(%5C%22%25

Clean up some formatting nits