/*
 * This file is part of LibrePlan
 *
 * Copyright (C) 2009-2010 Fundación para o Fomento da Calidade Industrial e
 *                         Desenvolvemento Tecnolóxico de Galicia
 * Copyright (C) 2010-2011 Igalia, S.L.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.libreplan.business.resources.bootstrap;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import org.libreplan.business.common.daos.IEntitySequenceDAO;
import org.libreplan.business.common.entities.EntityNameEnum;
import org.libreplan.business.resources.daos.ICriterionDAO;
import org.libreplan.business.resources.daos.ICriterionTypeDAO;
import org.libreplan.business.resources.entities.Criterion;
import org.libreplan.business.resources.entities.CriterionType;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;

/**
 * Loads all {@link ICriterionTypeProvider} and if there is any criterion that
 * doesn't exist, creates them.<br />
 * @author Óscar González Fernández <ogonzalez@igalia.com>
 * @author Diego Pino García <dpino@igalia.com>
 */
@Component
@Scope(BeanDefinition.SCOPE_SINGLETON)
public class CriterionsBootstrap implements ICriterionsBootstrap {

    @Autowired
    private ICriterionDAO criterionDAO;

    @Autowired
    private ICriterionTypeDAO criterionTypeDAO;

    @Autowired
    private IEntitySequenceDAO entitySequenceDAO;

    @Autowired
    private List<ICriterionTypeProvider> providers;

    public CriterionsBootstrap() {
    }

    @Override
    @Transactional
    public void loadRequiredData() {
        if (criterionTypeDAO.findAll().isEmpty()) {
            Map<CriterionType, List<String>> typesWithCriterions = getTypesWithCriterions();
            // Insert predefined criterions
            for (Entry<CriterionType, List<String>> entry : typesWithCriterions.entrySet()) {
                CriterionType criterionType = retrieveOrCreate(entry.getKey());
                // Create predefined criterions for criterionType
                for (String criterionName : entry.getValue()) {
                    ensureCriterionExists(criterionName, criterionType);
                }
            }
        }
    }

    private void ensureCriterionExists(String criterionName, CriterionType criterionType) {
        Criterion criterion = Criterion.createPredefined(criterionName, criterionType);

        if (!criterionDAO.existsPredefinedCriterion(criterion)) {
            int numberOfDigits = getNumberOfDigitsCode();
            criterionType.setGenerateCode(criterion, numberOfDigits);
            criterionDAO.save(criterion);
        }
    }

    private CriterionType retrieveOrCreate(CriterionType criterionType) {
        if (!criterionTypeDAO.existsPredefinedType(criterionType)) {
            criterionType
                    .setCode(entitySequenceDAO
                            .getNextEntityCodeWithoutTransaction(EntityNameEnum.CRITERION));
            criterionType.setCodeAutogenerated(true);
            criterionTypeDAO.save(criterionType);
            return criterionType;
        }

        return criterionTypeDAO.findPredefined(criterionType);
    }

    private Map<CriterionType, List<String>> getTypesWithCriterions() {
        HashMap<CriterionType, List<String>> result = new HashMap<>();
        for (ICriterionTypeProvider provider : providers) {
            for (Entry<CriterionType, List<String>> entry : provider.getRequiredCriterions().entrySet()) {
                if (!result.containsKey(entry.getKey())) {
                    result.put(entry.getKey(), new ArrayList<String>());
                }
                result.get(entry.getKey()).addAll(entry.getValue());
            }
        }

        return result;
    }

    protected Integer getNumberOfDigitsCode() {
        return entitySequenceDAO.getNumberOfDigitsCode(EntityNameEnum.CRITERION);
    }

}