/**
 * Copyright (c) 2014-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.model;

import javax.lang.model.element.Modifier;

import com.facebook.litho.specmodels.generator.BuilderGenerator;
import com.facebook.litho.specmodels.generator.ComponentImplGenerator;
import com.facebook.litho.specmodels.generator.DelegateMethodGenerator;
import com.facebook.litho.specmodels.generator.EventGenerator;
import com.facebook.litho.specmodels.generator.JavadocGenerator;
import com.facebook.litho.specmodels.generator.PreambleGenerator;
import com.facebook.litho.specmodels.generator.PureRenderGenerator;
import com.facebook.litho.specmodels.generator.StateGenerator;
import com.facebook.litho.specmodels.generator.TreePropGenerator;
import com.facebook.litho.specmodels.generator.TypeSpecDataHolder;

import com.squareup.javapoet.TypeSpec;

public class DefaultLayoutSpecGenerator implements LayoutSpecGenerator {

  @Override
  public TypeSpec generate(LayoutSpecModel layoutSpecModel) {
    final TypeSpec.Builder typeSpec =
        TypeSpec.classBuilder(layoutSpecModel.getComponentName())
            .superclass(ClassNames.COMPONENT_LIFECYCLE)
            .addTypeVariables(layoutSpecModel.getTypeVariables());

    if (layoutSpecModel.isPublic()) {
      typeSpec.addModifiers(Modifier.PUBLIC);
    }

    if (layoutSpecModel.hasInjectedDependencies()) {
      layoutSpecModel.getDependencyInjectionHelper().generate(layoutSpecModel).addToTypeSpec(typeSpec);
    } else {
      typeSpec.addModifiers(Modifier.FINAL);
    }

    TypeSpecDataHolder.newBuilder()
        .addTypeSpecDataHolder(JavadocGenerator.generate(layoutSpecModel))
        .addTypeSpecDataHolder(PreambleGenerator.generate(layoutSpecModel))
        .addTypeSpecDataHolder(ComponentImplGenerator.generate(layoutSpecModel))
        .addTypeSpecDataHolder(TreePropGenerator.generate(layoutSpecModel))
        .addTypeSpecDataHolder(DelegateMethodGenerator.generateDelegates(
            layoutSpecModel,
            DelegateMethodDescriptions.LAYOUT_SPEC_DELEGATE_METHODS_MAP))
        .addTypeSpecDataHolder(PureRenderGenerator.generate(layoutSpecModel))
        .addTypeSpecDataHolder(EventGenerator.generate(layoutSpecModel))
        .addTypeSpecDataHolder(StateGenerator.generate(layoutSpecModel))
        .addTypeSpecDataHolder(BuilderGenerator.generate(layoutSpecModel))
        .build()
        .addToTypeSpec(typeSpec);

    return typeSpec.build();
  }
}