/*
    * Copyright 2001-2005 The Apache Software Foundation.
    *
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *      http://www.apache.org/licenses/LICENSE-2.0
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.apache.logging.log4j.catalog.jpa.converter;
  
  import java.util.HashSet;
  import java.util.Map;
  import java.util.Set;
  import java.util.function.Function;
  import java.util.stream.Collectors;
  
  import org.apache.logging.log4j.LogManager;
  import org.apache.logging.log4j.Logger;
  import org.apache.logging.log4j.catalog.api.Attribute;
  import org.apache.logging.log4j.catalog.api.Constraint;
  import org.apache.logging.log4j.catalog.jpa.model.AttributeModel;
  import org.apache.logging.log4j.catalog.jpa.model.ConstraintModel;
  import org.apache.logging.log4j.catalog.jpa.service.AttributeService;
  import org.modelmapper.AbstractConverter;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.stereotype.Component;
  
  /**
   *
   */
  @Component
  public class AttributeConverter extends AbstractConverter<Attribute, AttributeModel> {
      private static Logger LOGGER = LogManager.getLogger(AttributeConverter.class);
  
      @Autowired
      private AttributeService attributeService;
  
      public  AttributeModel convert(Attribute attribute) {
          LOGGER.traceEntry(attribute.getName());
          AttributeModel model;
          if (attribute.getId() != null) {
              model = attributeService.getAttribute(attribute.getId()).orElseGet(AttributeModel::new);
          } else {
              model = new AttributeModel();
          }
          model.setName(attribute.getName());
          model.setAliases(attribute.getAliases());
          model.setDescription(attribute.getDescription());
          model.setDisplayName(attribute.getDisplayName());
          model.setDataType(attribute.getDataType());
          model.setId(attribute.getId());
          model.setCatalogId(attribute.getCatalogId());
          model.setIndexed(attribute.isIndexed());
          model.setRequestContext(attribute.isRequestContext());
          model.setRequired(attribute.isRequired());
          model.setSortable(attribute.isSortable());
          model.setExamples(attribute.getExamples());
          Set<ConstraintModel> constraintModels = model.getConstraints() != null ? model.getConstraints() :
                  new HashSet<>();
          Map<Long, ConstraintModel> constraintMap =
              constraintModels.stream().collect(Collectors.toMap(ConstraintModel::getId, Function.identity()));
          if (attribute.getConstraints() != null) {
              constraintModels.removeIf(a -> attribute.getConstraints().stream().noneMatch(b -> b.getId().equals(a.getId())));
              for (Constraint constraint : attribute.getConstraints()) {
                  ConstraintModel constraintModel;
                  if (constraint.getId() != null) {
                      constraintModel = constraintMap.get(constraint.getId());
                      constraintModel.setConstraintType(constraint.getConstraintType().getName());
                      constraintModel.setValue(constraint.getValue());
                  } else {
                      constraintModel = new ConstraintModel();
                      constraintModel.setConstraintType(constraint.getConstraintType().getName());
                      constraintModel.setValue(constraint.getValue());
                      constraintModels.add(constraintModel);
                  }
              }
          }
          model.setConstraints(constraintModels);
          return LOGGER.traceExit(model);
      }
  }