package fr.ifremer.quadrige3.core.dao.data.survey;
   
   /*-
    * #%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 fr.ifremer.quadrige3.core.dao.administration.user.DepartmentImpl;
  import fr.ifremer.quadrige3.core.dao.administration.user.QuserImpl;
  import fr.ifremer.quadrige3.core.dao.technical.Assert;
  import fr.ifremer.quadrige3.core.dao.technical.Beans;
  import fr.ifremer.quadrige3.core.dao.technical.Daos;
  import fr.ifremer.quadrige3.core.exception.QuadrigeTechnicalException;
  import fr.ifremer.quadrige3.core.vo.data.survey.CampaignVO;
  import fr.ifremer.quadrige3.core.vo.data.survey.OccasionVO;
  import org.apache.commons.collections4.CollectionUtils;
  import org.apache.commons.lang3.ArrayUtils;
  import org.hibernate.HibernateException;
  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.Resource;
  import java.util.Collection;
  import java.util.List;
  
  /**
   * <p>
   * CampaignDaoImpl class.
   * </p>
   *
   * @see fr.ifremer.quadrige3.core.dao.data.survey.Campaign
   */
  @Repository("campaignDao")
  @Lazy
  public class CampaignDaoImpl
          extends CampaignDaoBase {
      @Resource
      private OccasionDao occasionDao;
  
      /**
       * Constructor used by Spring
       *
       * @param sessionFactory a {@link org.hibernate.SessionFactory} object.
       */
      @Autowired
      public CampaignDaoImpl(SessionFactory sessionFactory) {
          super();
          setSessionFactory(sessionFactory);
      }
  
      @Override
      public void remove(Collection<Campaign> entities) {
          Assert.notEmpty(entities);
  
          for (Campaign entity : entities) {
              remove(entity);
          }
      }
  
      /**
       * {@inheritDoc}
       */
      @Override
      public void remove(Campaign entity) {
  
          // Remove occasions
          if (CollectionUtils.isNotEmpty(entity.getOccasions())) {
              occasionDao.remove(entity.getOccasions());
              entity.getOccasions().clear();
          }
  
          getSession().flush();
  
          // Remove link to program
          if (CollectionUtils.isNotEmpty(entity.getPrograms())) {
              entity.getPrograms().clear();
          }
  
         // Remove link to moratorium
         if (CollectionUtils.isNotEmpty(entity.getMoratoria())) {
             entity.getMoratoria().clear();
         }
 
         super.remove(entity);
     }
 
     @Override
     protected CampaignVO handleSave(CampaignVO source) {
 
         Assert.notNull(source.getQuserId());
         Assert.notNull(source.getRecDepId());
         Assert.notNull(source.getCampaignStartDt());
         Assert.notNull(source.getCampaignNm());
 
         // Already saved
         Campaign entity = null;
         if (source.getCampaignId() != null) {
             entity = get(source.getCampaignId());
         }
 
         // Load (or create) the entity
         boolean isNew = false;
         if (entity == null) {
             entity = Campaign.Factory.newInstance();
             isNew = true;
         }
 
         // Check update_dt
         // NOT NEED on a client database
 
         // Update update_dt
         // NOT NEED on a client database
 
         // VO -> Entity
         campaignVOToEntity(source, entity, true, false /* do not convert occasions */);
 
         // Save entity
         if (isNew) {
             Integer campaignId = (Integer) getSession().save(entity);
             source.setCampaignId(campaignId);
         } else {
             getSession().update(entity);
         }
 
         List<Integer> occasionsIdsToRemove = Beans.collectProperties(entity.getOccasions(), "occasId");
 
         // Save occasions
         if (ArrayUtils.isNotEmpty(source.getOccasionVOs())) {
             for (OccasionVO occasionVO : source.getOccasionVOs()) {
                 occasionVO.setCampaignId(entity.getCampaignId());
                 occasionVO = occasionDao.save(occasionVO, null);
 
                 occasionsIdsToRemove.remove(occasionVO.getOccasId());
             }
         }
 
         getSession().flush();
         getSession().clear();
 
         // remove unused occasions
         if (CollectionUtils.isNotEmpty(occasionsIdsToRemove)) {
             occasionDao.removeByIds(occasionsIdsToRemove);
         }
 
         return source;
     }
 
     @Override
     protected Long handleCountSurveyUsage(int campaignId) {
         try
         {
             return queryCount("countSurveyUsingCampaign",
                     "campaignId", IntegerType.INSTANCE, campaignId);
 
         } catch (HibernateException he) {
             throw new QuadrigeTechnicalException(String.format("Could not get the campaign usage in surveys : %s", he.getMessage()), he);
         }
     }
 
     /*
     @Override
     protected Integer[] handleGetWritableCampaignIdsByUserId(Integer quserId) throws Exception {
 
         // Get writable programs for user
         List<ProgramVO> writablePrograms = programStrategyJdbcDao.getWritableProgramsByUserId(quserId);
 
         try {
 
             Query query = createQuery("writableCampaignIdsByQuserId",
                     "quserId", IntegerType.INSTANCE, quserId);
 
             // Limit to active users
             query.setParameterList("statusCodes", ImmutableList.of(StatusCode.ENABLE.getValue()));
 
             // Writable programs for user (if any)
             if (CollectionUtils.isEmpty(writablePrograms)) {
                 query.setParameter("progCds", null);
                 query.setParameter("hasProgCds", null);
             } else {
                 query.setParameterList("progCds", Lists.transform(writablePrograms, ProgramVO::getProgCd));
                 query.setParameter("hasProgCds", true);
             }
 
             List<Integer> campaignIds = (List<Integer>) query.list();
 
             return campaignIds.toArray(new Integer[campaignIds.size()]);
 
         } catch (HibernateException he) {
             throw new QuadrigeTechnicalException(String.format("Could not get the writable campaign Ids for user: %s", he.getMessage()));
         }
     }
 */
     @Override
     public void toCampaignVO(Campaign source, CampaignVO target) {
 
         // copy base attributes
         super.toCampaignVO(source, target);
 
         // copy user and recorder department attributes
         target.setQuserId(source.getQuser() != null ? source.getQuser().getQuserId() : null);
         target.setRecDepId(source.getRecorderDepartment() != null ? source.getRecorderDepartment().getDepId() : null);
 
         // Occasions
         if (CollectionUtils.isEmpty(source.getOccasions())) {
             target.setOccasionVOs(null);
         } else {
             target.setOccasionVOs(occasionDao.toOccasionVOArray(source.getOccasions()));
         }
     }
 
     /**
      * 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 Campaign loadCampaignFromCampaignVO(CampaignVO campaignVO) {
         Campaign campaign = null;
         if (campaignVO.getCampaignId() != null) {
             campaign = get(campaignVO.getCampaignId());
         }
         if (campaign == null) {
             campaign = Campaign.Factory.newInstance();
         }
         return campaign;
     }
 
     @Override
     public Campaign campaignVOToEntity(CampaignVO campaignVO) {
 
         Campaign campaign = loadCampaignFromCampaignVO(campaignVO);
         campaignVOToEntity(campaignVO, campaign, true);
         return campaign;
     }
 
     @Override
     public void campaignVOToEntity(CampaignVO source, Campaign target, boolean copyIfNull) {
 
         campaignVOToEntity(source, target, copyIfNull, false);
 
     }
 
     protected void campaignVOToEntity(CampaignVO source, Campaign target, boolean copyIfNull, boolean withOccasions) {
 
         // copy base attributes
         super.campaignVOToEntity(source, target, copyIfNull);
 
         // campaign id
         if (copyIfNull || source.getCampaignId() != null) {
             target.setCampaignId(source.getCampaignId());
         }
 
         // quser
         if (copyIfNull || source.getQuserId() != null) {
             target.setQuser(source.getQuserId() != null ? load(QuserImpl.class, source.getQuserId()) : null);
         }
 
         // recorder department
         if (copyIfNull || source.getRecDepId() != null) {
             target.setRecorderDepartment(source.getRecDepId() != null ? load(DepartmentImpl.class, source.getRecDepId()) : null);
         }
 
         // Strategies
         if (withOccasions) {
             // Save linked occasions
             fillEntityOccasions(source, target, copyIfNull);
         }
     }
 
     /**
      * <p>
      * fil occasion linked with campaign.
      * </p>
      *
      * @param source     a {@link fr.ifremer.quadrige3.core.vo.data.survey.CampaignVO} object.
      * @param target     a {@link fr.ifremer.quadrige3.core.dao.data.survey.Campaign} object.
      * @param copyIfNull a boolean.
      */
     protected void fillEntityOccasions(CampaignVO source, final Campaign target, boolean copyIfNull) {
         if (copyIfNull || source.getOccasionVOs() != null) {
             if (source.getOccasionVOs() == null) {
                 target.getOccasions().clear();
             } else {
                 Daos.replaceEntities(target.getOccasions(),
                         source.getOccasionVOs(),
                         occasionVO -> {
                             Occasion occasion = occasionDao.occasionVOToEntity(occasionVO);
                             occasion.setCampaign(target);
                             return occasion;
                         });
             }
         }
     }
 
 }