/**
 * Copyright (c) 2017-present, Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under the BSD-style license found in the
 * LICENSE file in the root directory of this source tree. An additional grant
 * of patent rights can be found in the PATENTS file in the same directory.
 */

package com.facebook.litho.specmodels.processor;

import javax.annotation.Nullable;
import javax.lang.model.element.TypeElement;
import javax.lang.model.util.Elements;

import java.lang.annotation.Annotation;
import java.util.ArrayList;
import java.util.List;

import com.facebook.litho.specmodels.internal.ImmutableList;
import com.facebook.litho.annotations.LayoutSpec;
import com.facebook.litho.annotations.OnCreateTreeProp;
import com.facebook.litho.annotations.ShouldUpdate;
import com.facebook.litho.specmodels.model.DelegateMethodDescriptions;
import com.facebook.litho.specmodels.model.DependencyInjectionHelper;
import com.facebook.litho.specmodels.model.LayoutSpecGenerator;
import com.facebook.litho.specmodels.model.DefaultLayoutSpecGenerator;
import com.facebook.litho.specmodels.model.LayoutSpecModel;

/**
 * Factory for creating {@link LayoutSpecModel}s.
 */
public class LayoutSpecModelFactory {
  private static final List<Class<? extends Annotation>> INTER_STAGE_INPUT_ANNOTATIONS =
      new ArrayList<>();
  private static final List<Class<? extends Annotation>> DELEGATE_METHOD_ANNOTATIONS =
      new ArrayList<>();
  static {
    DELEGATE_METHOD_ANNOTATIONS.addAll(
        DelegateMethodDescriptions.LAYOUT_SPEC_DELEGATE_METHODS_MAP.keySet());
    DELEGATE_METHOD_ANNOTATIONS.add(OnCreateTreeProp.class);
    DELEGATE_METHOD_ANNOTATIONS.add(ShouldUpdate.class);
  }

  /**
   * Create a {@link LayoutSpecModel} from the given {@link TypeElement} and an optional
   * {@link DependencyInjectionHelper}.
   */
  public static LayoutSpecModel create(
      Elements elements,
      TypeElement element,
      @Nullable DependencyInjectionHelper dependencyInjectionHelper) {
    return create(
        elements,
        element,
        dependencyInjectionHelper,
        DELEGATE_METHOD_ANNOTATIONS,
        new DefaultLayoutSpecGenerator());
  }

  public static LayoutSpecModel create(
      Elements elements,
      TypeElement element,
      @Nullable DependencyInjectionHelper dependencyInjectionHelper,
      List<Class<? extends Annotation>> delegateMethodAnnotations,
      LayoutSpecGenerator layoutSpecGenerator) {
    return new LayoutSpecModel(
        element.getQualifiedName().toString(),
        getValue(element),
        DelegateMethodExtractor.getDelegateMethods(
            element,
            delegateMethodAnnotations,
            INTER_STAGE_INPUT_ANNOTATIONS),
        EventMethodExtractor.getOnEventMethods(
            elements, element, INTER_STAGE_INPUT_ANNOTATIONS),
        UpdateStateMethodExtractor.getOnUpdateStateMethods(
            element,
            INTER_STAGE_INPUT_ANNOTATIONS),
        ImmutableList.copyOf(TypeVariablesExtractor.getTypeVariables(element)),
        ImmutableList.copyOf(PropDefaultsExtractor.getPropDefaults(element)),
        EventDeclarationsExtractor.getEventDeclarations(elements, element, LayoutSpec.class),
        JavadocExtractor.getClassJavadoc(elements, element),
        JavadocExtractor.getPropJavadocs(elements, element),
        isPublic(element),
        dependencyInjectionHelper,
        isPureRender(element),
        element,
        layoutSpecGenerator);
  }

  private static String getValue(TypeElement element) {
    final LayoutSpec layoutSpec = element.getAnnotation(LayoutSpec.class);
    return layoutSpec != null ? layoutSpec.value() : "";
  }

  private static boolean isPublic(TypeElement element) {
    final LayoutSpec layoutSpec = element.getAnnotation(LayoutSpec.class);
    return layoutSpec == null || layoutSpec.isPublic();
  }

  private static boolean isPureRender(TypeElement element) {
    final LayoutSpec layoutSpec = element.getAnnotation(LayoutSpec.class);
    return layoutSpec != null && layoutSpec.isPureRender();
  }
}