package fr.ifremer.quadrige2.core.dao.administration.strategy;
   
   /*-
    * #%L
    * Quadrige2 Core :: Quadrige2 Server 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 com.google.common.base.Preconditions;
  import com.google.common.collect.Lists;
  import fr.ifremer.quadrige2.core.dao.referential.AcquisitionLevel;
  import fr.ifremer.quadrige2.core.dao.referential.AcquisitionLevelImpl;
  import fr.ifremer.quadrige2.core.dao.referential.PrecisionTypeImpl;
  import fr.ifremer.quadrige2.core.dao.referential.pmfm.PmfmImpl;
  import fr.ifremer.quadrige2.core.exception.BadUpdateDtException;
  import fr.ifremer.quadrige2.core.dao.technical.Daos;
  import fr.ifremer.quadrige2.core.dao.technical.Dates;
  import fr.ifremer.quadrige2.core.vo.administration.strategy.PmfmStrategyVO;
  import org.apache.commons.collections4.CollectionUtils;
  import org.apache.commons.lang3.ArrayUtils;
  import org.hibernate.SessionFactory;
  import org.nuiton.i18n.I18n;
  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.List;
  import java.util.Objects;
  
  /**
   * <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 {
 			List<String> acquisLevelCds = Lists.transform(Lists.newArrayList(source.getAcquisitionLevels()),
 					new Function<AcquisitionLevel, String>() {
 						@Nullable
 						@Override
 						public String apply(AcquisitionLevel source) {
 							return source.getAcquisLevelCd();
 						}
 					});
 			target.setAcquisLevelCds(acquisLevelCds.toArray(new String[acquisLevelCds.size()]));
 		}
 
 		// UI functions
 		if (CollectionUtils.isEmpty(source.getUiFunctions())) {
 			target.setUiFunctionCds(null);
 		}
 		else {
 			List<String> uiFunctionCds = Lists.transform(Lists.newArrayList(source.getUiFunctions()),
 					new Function<UiFunction, String>() {
 						@Nullable
 						@Override
 						public String apply(UiFunction source) {
 							return source.getUiFunctionCd();
 						}
 					});
 			target.setUiFunctionCds(uiFunctionCds.toArray(new String[uiFunctionCds.size()]));
 		}
 	}
 
 	/**
 	 * 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(),
 						acquisLevelCd -> 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(),
 						uiFunctionCd -> load(UiFunctionImpl.class, uiFunctionCd));
 			}
 		}
 
 	}
 
 	/** {@inheritDoc} */
 	@Override
 	protected PmfmStrategyVO handleSave(PmfmStrategyVO source, Timestamp updateDt) throws Exception {
 		Preconditions.checkNotNull(source);
 		Preconditions.checkNotNull(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);
 			isNew = true;
 		}
 
 		// Check update_dt
 		if (!isNew) {
 			if (entity.getUpdateDt() != null) {
 				Timestamp serverUpdateDtNoMillisecond = Dates.resetMillisecond(entity.getUpdateDt());
 				Timestamp sourceUpdateDtNoMillisecond = Dates.resetMillisecond(source.getUpdateDt());
 				if (!Objects.equals(sourceUpdateDtNoMillisecond, serverUpdateDtNoMillisecond)) {
 					throw new BadUpdateDtException(I18n.t("quadrige2.dao.pmfmStrategy.badUpdateDt", source.getStratId(), serverUpdateDtNoMillisecond,
 							sourceUpdateDtNoMillisecond));
 				}
 			}
 		}
 
 		// update update_dt
 		Timestamp newUpdateDt = updateDt;
 		if (newUpdateDt == null) {
 			newUpdateDt = getDatabaseCurrentTimestamp();
 		}
 		source.setUpdateDt(newUpdateDt);
 
 		// VO -> Entity
 		pmfmStrategyVOToEntity(source, entity, true);
 
 		// Save entity
 		if (isNew) {
 			Integer pmfmStratId = (Integer) getSession().save(entity);
 			source.setPmfmStratId(pmfmStratId);
 		} else {
 			getSession().update(entity);
 		}
 
 		return source;
 	}
 }