package fr.ifremer.quadrige3.core.dao.administration.strategy;
   
   /*-
    * #%L
    * Quadrige3 Core :: Quadrige3 Client Core
    * $Id:$
    * $HeadURL:$
    * %%
    * Copyright (C) 2017 Ifremer
   * %%
   * 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 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/>.
   * #L%
   */
  import com.google.common.base.Function;
  import fr.ifremer.quadrige3.core.dao.referential.AcquisitionLevel;
  import fr.ifremer.quadrige3.core.dao.referential.AcquisitionLevelImpl;
  import fr.ifremer.quadrige3.core.dao.referential.PrecisionTypeImpl;
  import fr.ifremer.quadrige3.core.dao.referential.pmfm.PmfmImpl;
  import fr.ifremer.quadrige3.core.dao.technical.Assert;
  import fr.ifremer.quadrige3.core.dao.technical.Daos;
  import fr.ifremer.quadrige3.core.vo.administration.strategy.PmfmStrategyVO;
  import org.apache.commons.collections4.CollectionUtils;
  import org.apache.commons.lang3.ArrayUtils;
  import org.hibernate.Query;
  import org.hibernate.SessionFactory;
  import org.hibernate.type.IntegerType;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.context.annotation.Lazy;
  import org.springframework.stereotype.Repository;
  
  import javax.annotation.Nullable;
  import java.sql.Timestamp;
  import java.util.Arrays;
  import java.util.Collection;
  import java.util.List;
  
  /**
   * <p>
   * PmfmStrategyDaoImpl class.
   * </p>
   * 
   * @see PmfmStrategy
   */
  @Repository("pmfmStrategyDao")
  @Lazy
  public class PmfmStrategyDaoImpl
  		extends PmfmStrategyDaoBase
  {
  	/**
  	 * Constructor used by Spring
  	 * 
  	 * @param sessionFactory
  	 *            a {@link org.hibernate.SessionFactory} object.
  	 */
  	@Autowired
  	public PmfmStrategyDaoImpl(SessionFactory sessionFactory) {
  		super();
  		setSessionFactory(sessionFactory);
  	}
  
  	/** {@inheritDoc} */
  	public void toPmfmStrategyVO(
  			PmfmStrategy source,
  			PmfmStrategyVO target)
  	{
  		super.toPmfmStrategyVO(source, target);
  
  		// Strategy
  		if (source.getStrategy() == null) {
  			target.setStratId(null);
  		}
  		else {
  			target.setStratId(source.getStrategy().getStratId());
  		}
  
  		// Pmfm
  		if (source.getPmfm() == null) {
  			target.setPmfmId(null);
  		}
  		else {
  			target.setPmfmId(source.getPmfm().getPmfmId());
  		}
  
  		// Precision type
  		if (source.getPrecisionType() == null) {
  			target.setPrecTypeId(null);
  		}
  		else {
 			target.setPrecTypeId(source.getPrecisionType().getPrecTypeId());
 		}
 
 		// Acquisition levels
 		if (CollectionUtils.isEmpty(source.getAcquisitionLevels())) {
 			target.setAcquisLevelCds(null);
 		}
 		else {
 			target.setAcquisLevelCds(source.getAcquisitionLevels().stream().map(AcquisitionLevel::getAcquisLevelCd).toArray(String[]::new));
 		}
 
 		// UI functions
 		if (CollectionUtils.isEmpty(source.getUiFunctions())) {
 			target.setUiFunctionCds(null);
 		}
 		else {
 			target.setUiFunctionCds(source.getUiFunctions().stream().map(UiFunction::getUiFunctionCd).toArray(String[]::new));
 		}
 
 		// Qualitative values
 		target.setQualValueIds(getQualitativeValueIds(target.getPmfmStratId()).toArray(new Integer[0]));
 	}
 
 	/**
 	 * Retrieves the entity object that is associated with the specified value object
 	 * from the object store. If no such entity object exists in the object store,
 	 * a new, blank entity is created
 	 */
 	private PmfmStrategy loadPmfmStrategyFromPmfmStrategyVO(PmfmStrategyVO pmfmStrategyVO)
 	{
 		PmfmStrategy pmfmStrategy = null;
 		if (pmfmStrategyVO.getPmfmStratId() != null) {
 			pmfmStrategy = this.get(pmfmStrategyVO.getPmfmStratId());
 		}
 		if (pmfmStrategy == null)
 		{
 			pmfmStrategy = PmfmStrategy.Factory.newInstance();
 		}
 		return pmfmStrategy;
 	}
 
 	/** {@inheritDoc} */
 	public PmfmStrategy pmfmStrategyVOToEntity(PmfmStrategyVO pmfmStrategyVO)
 	{
 		PmfmStrategy entity = this.loadPmfmStrategyFromPmfmStrategyVO(pmfmStrategyVO);
 		this.pmfmStrategyVOToEntity(pmfmStrategyVO, entity, true);
 		return entity;
 	}
 
 	/** {@inheritDoc} */
 	@Override
 	public void pmfmStrategyVOToEntity(
 			PmfmStrategyVO source,
 			PmfmStrategy target,
 			boolean copyIfNull)
 	{
 		super.pmfmStrategyVOToEntity(source, target, copyIfNull);
 
 		// Strategy
 		if (copyIfNull || source.getStratId() != null || source.getStrategyVO() != null) {
 			if (source.getStratId() == null && source.getStrategyVO() == null) {
 				target.setStrategy(null);
 			}
 			else {
 				Integer stratId = source.getStratId();
 				if (stratId == null) {
 					stratId = source.getStrategyVO().getStratId();
 				}
 				target.setStrategy(load(StrategyImpl.class, stratId));
 			}
 		}
 
 		// Pmfm
 		if (copyIfNull || source.getPmfmId() != null) {
 			if (source.getPmfmId() == null) {
 				target.setPmfm(null);
 			}
 			else {
 				target.setPmfm(load(PmfmImpl.class, source.getPmfmId()));
 			}
 		}
 
 		// Precision type
 		if (copyIfNull || source.getPrecTypeId() != null) {
 			if (source.getPrecTypeId() == null) {
 				target.setPrecisionType(null);
 			}
 			else {
 				target.setPrecisionType(load(PrecisionTypeImpl.class, source.getPrecTypeId()));
 			}
 		}
 
 		// Acquisition levels
 		if (copyIfNull || ArrayUtils.isNotEmpty(source.getAcquisLevelCds())) {
 			if (ArrayUtils.isEmpty(source.getAcquisLevelCds())) {
 				target.getAcquisitionLevels().clear();
 			}
 			else {
 				Daos.replaceEntities(
 						target.getAcquisitionLevels(),
 						source.getAcquisLevelCds(),
 						new Function<String, AcquisitionLevel>() {
 							@Nullable
 							@Override
 							public AcquisitionLevel apply(@Nullable String acquisLevelCd) {
 								return load(AcquisitionLevelImpl.class, acquisLevelCd);
 							}
 						}
 						);
 			}
 		}
 
 		// UI functions
 		if (copyIfNull || ArrayUtils.isNotEmpty(source.getUiFunctionCds())) {
 			if (ArrayUtils.isEmpty(source.getUiFunctionCds())) {
 				target.getUiFunctions().clear();
 			}
 			else {
 				Daos.replaceEntities(
 						target.getUiFunctions(),
 						source.getUiFunctionCds(),
 						new Function<String, UiFunction>() {
 							@Nullable
 							@Override
 							public UiFunction apply(@Nullable String uiFunctionCd) {
 								return load(UiFunctionImpl.class, uiFunctionCd);
 							}
 						}
 						);
 			}
 		}
 
 	}
 
 	/** {@inheritDoc} */
 	@Override
 	protected PmfmStrategyVO handleSave(PmfmStrategyVO source, Timestamp updateDt) {
 		Assert.notNull(source);
 		Assert.notNull(source.getStratId());
 
 		// Load parent
 		Strategy parent = get(StrategyImpl.class, source.getStratId());
 
 		// Load entity
 		PmfmStrategy entity = null;
 		boolean isNew = false;
 		if (source.getPmfmStratId() != null) {
 			entity = get(source.getPmfmStratId());
 		}
 		if (entity == null) {
 			entity = PmfmStrategy.Factory.newInstance();
 			parent.addPmfmStrategies(entity);
 			entity.setStrategy(parent);
 			entity.setPmfmStratId(source.getPmfmStratId()); // assigned id
 			isNew = true;
 		}
 
 		// update update_dt
 		// NOT NEED on a client database
 
 		// VO -> Entity
 		pmfmStrategyVOToEntity(source, entity, true);
 
 		// Save entity
 		if (isNew) {
 			Integer pmfmStratId = (Integer) getSession().save(entity);
 			source.setPmfmStratId(pmfmStratId);
 		} else {
 			getSession().update(entity);
 		}
 
 		getSession().flush();
 
 		// save qualitative values
 		saveQualitativeValues(source);
 
 		return source;
 	}
 
 	@Override
 	public void remove(PmfmStrategy pmfmStrategy) {
 
 		// Remove PMFM_APPLIED_STRATEGY by query
 		queryUpdate("deletePmfmAppliedStrategiesByPmfmStrategyId",
 			"pmfmStrategyId", IntegerType.INSTANCE, pmfmStrategy.getPmfmStratId());
 
 		deleteQualitativeValues(pmfmStrategy.getPmfmStratId());
 		super.remove(pmfmStrategy);
 	}
 
 	@Override
 	public void remove(Collection<PmfmStrategy> entities) {
 		entities.forEach(pmfmStrategy -> deleteQualitativeValues(pmfmStrategy.getPmfmStratId()));
 		super.remove(entities);
 	}
 
 	@Override
 	public void remove(Integer pmfmStratId) {
 		deleteQualitativeValues(pmfmStratId);
 		super.remove(pmfmStratId);
 	}
 
 	public List<Integer> getQualitativeValueIds(int pmfmStrategyId) {
 		return queryListTyped("qualitativeValueIdsByPmfmStrategyId", "pmfmStrategyId", IntegerType.INSTANCE, pmfmStrategyId);
 	}
 
 	public void deleteQualitativeValues(int pmfmStrategyId) {
 		Query query = createSQLQuery("deleteQualitativeValuesByPmfmStrategyId");
 		query.setParameter("pmfmStrategyId", pmfmStrategyId);
 		query.executeUpdate();
 	}
 
 	public void deleteQualitativeValues(Collection<Integer> pmfmStrategyIds) {
 		Query query = createSQLQuery("deleteQualitativeValuesByPmfmStrategyIds");
 		query.setParameterList("pmfmStrategyIds", pmfmStrategyIds);
 		query.executeUpdate();
 	}
 
 	private void saveQualitativeValues(PmfmStrategyVO pmfmStrategy) {
 		// Delete first
 		deleteQualitativeValues(pmfmStrategy.getPmfmStratId());
 		// then insert
 		if (ArrayUtils.isNotEmpty(pmfmStrategy.getQualValueIds())) {
 			Arrays.stream(pmfmStrategy.getQualValueIds())
 				.forEach(qualitativeValueId -> saveQualitativeValue(pmfmStrategy.getPmfmStratId(), pmfmStrategy.getPmfmId(), qualitativeValueId));
 		}
 	}
 
 	public void saveQualitativeValue(int pmfmStrategyId, int pmfmId, int qualitativeValueId) {
 		Query query = createSQLQuery("addQualitativeValueByPmfmStrategyId");
 		query.setParameter("pmfmStrategyId", pmfmStrategyId);
 		query.setParameter("pmfmId", pmfmId);
 		query.setParameter("qualitativeValueId", qualitativeValueId);
 		query.executeUpdate();
 	}
 }