/*

  * Copyright 2007 The Kuali Foundation

  *

  * Licensed under the Educational Community License, Version 1.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.opensource.org/licenses/ecl1.php

  *

  * 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.kuali.student.lum.program.service.assembler;

 import java.util.ArrayList;

 import java.util.List;

 import java.util.Map;

 import org.apache.log4j.Logger;

 import org.kuali.student.common.assembly.BOAssembler;

 import org.kuali.student.common.assembly.BaseDTOAssemblyNode;

 import org.kuali.student.common.assembly.BaseDTOAssemblyNode.NodeOperation;

 import org.kuali.student.common.assembly.data.AssemblyException;

 import org.kuali.student.common.dto.AmountInfo;

 import org.kuali.student.common.dto.DtoConstants;

 import org.kuali.student.common.exceptions.DataValidationErrorException;

 import org.kuali.student.common.exceptions.DoesNotExistException;

 import org.kuali.student.common.exceptions.InvalidParameterException;

 import org.kuali.student.common.exceptions.MissingParameterException;

 import org.kuali.student.common.exceptions.OperationFailedException;

 import org.kuali.student.lum.course.service.assembler.CourseAssembler;

 import org.kuali.student.lum.lu.dto.CluCluRelationInfo;

 import org.kuali.student.lum.lu.dto.CluInfo;

 import org.kuali.student.lum.lu.service.LuService;

 import org.kuali.student.lum.program.dto.CoreProgramInfo;

 import org.kuali.student.lum.program.dto.MajorDisciplineInfo;

 import org.kuali.student.lum.program.dto.ProgramVariationInfo;

 import org.kuali.student.lum.service.assembler.CluAssemblerUtils;

 /**

  * @author KS TODO - Much of this should be shared with ProgramVariationAssembler (and probably other Program Assemblers to

  *         come). AssemblerUtils?

  */

 public class MajorDisciplineAssembler implements BOAssembler<MajorDisciplineInfo, CluInfo> {

     final static Logger LOG = Logger.getLogger(CourseAssembler.class);

     private LuService luService;

     private ProgramVariationAssembler programVariationAssembler;

     private CoreProgramAssembler coreProgramAssembler;

     private CluAssemblerUtils cluAssemblerUtils;

     private ProgramAssemblerUtils programAssemblerUtils;

     @Override

     public MajorDisciplineInfo assemble(CluInfo clu, MajorDisciplineInfo majorDiscipline, boolean shallowBuild) throws AssemblyException {

         MajorDisciplineInfo mdInfo = (null != majorDiscipline) ? majorDiscipline : new MajorDisciplineInfo();

         // Copy all the data from the clu to the majordiscipline

         programAssemblerUtils.assembleBasics(clu, mdInfo);

         programAssemblerUtils.assembleIdentifiers(clu, mdInfo);

         programAssemblerUtils.assembleBasicAdminOrgs(clu, mdInfo);

         programAssemblerUtils.assembleFullOrgs(clu, mdInfo);

         programAssemblerUtils.assembleAtps(clu, mdInfo);

         programAssemblerUtils.assembleLuCodes(clu, mdInfo);

         mdInfo.setIntensity((null != clu.getIntensity()) ? clu.getIntensity().getUnitType() : null);

         mdInfo.setStdDuration(clu.getStdDuration());

         mdInfo.setPublishedInstructors(clu.getInstructors());

         mdInfo.setCampusLocations(clu.getCampusLocations());        

         mdInfo.setAccreditingAgencies(clu.getAccreditations());

         mdInfo.setEffectiveDate(clu.getEffectiveDate());

         mdInfo.setDescr(clu.getDescr());

         mdInfo.setVersionInfo(clu.getVersionInfo());

         mdInfo.setNextReviewPeriod(clu.getNextReviewPeriod());

         if (!shallowBuild) {

                 programAssemblerUtils.assembleRequirements(clu, mdInfo);

             mdInfo.setCredentialProgramId(programAssemblerUtils.getCredentialProgramID(clu.getId()));

             mdInfo.setResultOptions(programAssemblerUtils.assembleResultOptions(clu.getId()));

             mdInfo.setLearningObjectives(cluAssemblerUtils.assembleLos(clu.getId(), shallowBuild));

             mdInfo.setVariations(assembleVariations(clu.getId(), shallowBuild));

             mdInfo.setOrgCoreProgram(assembleCoreProgram(clu.getId(), shallowBuild));

             programAssemblerUtils.assemblePublications(clu, mdInfo);

         }

        return mdInfo;

     }

     private CoreProgramInfo assembleCoreProgram(String cluId, boolean shallowBuild) throws AssemblyException {

         CoreProgramInfo coreProgramInfo = null;

         try {

             List<CluInfo> corePrograms = luService.getRelatedClusByCluId(cluId, ProgramAssemblerConstants.HAS_CORE_PROGRAM);

             // TODO - is it an error if there's more than one core program?

             if (corePrograms.size() == 1) {

                 coreProgramInfo = coreProgramAssembler.assemble(corePrograms.get(0), null, shallowBuild);

             } else if (corePrograms.size() > 1) {

                 throw new AssemblyException(new DataValidationErrorException("MajorDiscipline has more than one associated Core Program"));

             }

         } catch (Exception e) {

             throw new AssemblyException(e);

         }

         return coreProgramInfo;

     }

     private List<ProgramVariationInfo> assembleVariations(String cluId, boolean shallowBuild) throws AssemblyException {

         List<ProgramVariationInfo> variations = new ArrayList<ProgramVariationInfo>();

         try {

                 Map<String, CluCluRelationInfo> currentRelations = null;

                 currentRelations = programAssemblerUtils.getCluCluActiveRelations(cluId, ProgramAssemblerConstants.HAS_PROGRAM_VARIATION);

                 if(currentRelations != null && !currentRelations.isEmpty()){

                         for (String variationId : currentRelations.keySet()) {

                                 CluInfo variationClu = luService.getClu(variationId);

                                 variations.add(programVariationAssembler.assemble(variationClu, null, shallowBuild));

                         }

                 }

         } catch (Exception e) {

             throw new AssemblyException(e);

         }

         return variations;

     }

     @Override

     public BaseDTOAssemblyNode<MajorDisciplineInfo, CluInfo> disassemble(MajorDisciplineInfo major, NodeOperation operation) throws AssemblyException {

                 if (major == null) {

                     LOG.error("Major for  disassemble is null!");

                         throw new AssemblyException("Major cannot be null");

                 }

         //TODO   IDs for objects w/o ids

                 BaseDTOAssemblyNode<MajorDisciplineInfo, CluInfo> result = new BaseDTOAssemblyNode<MajorDisciplineInfo, CluInfo>(

                                 this);

                 CluInfo clu;

                 try {

                         clu = (NodeOperation.UPDATE == operation) ? luService.getClu(major.getId()) : new CluInfo();

         } catch (Exception e) {

                         throw new AssemblyException("Error getting existing learning unit during major update", e);

         } 

         boolean stateChanged = NodeOperation.UPDATE == operation && major.getState() != null && !major.getState().equals(clu.getState());

         programAssemblerUtils.disassembleBasics(clu, major);

         if (major.getId() == null)

             major.setId(clu.getId());

         programAssemblerUtils.disassembleLuCodes(clu, major, operation);

         programAssemblerUtils.disassembleAdminOrgs(clu, major, operation);

         programAssemblerUtils.disassembleAtps(clu, major, operation);

         programAssemblerUtils.disassembleIdentifiers(clu, major, operation);

         programAssemblerUtils.disassemblePublications(clu, major, operation, result);

         if(major.getProgramRequirements() != null && !major.getProgramRequirements().isEmpty()) {

                 programAssemblerUtils.disassembleRequirements(clu, major, operation, result, stateChanged);                

         }

         if (major.getVariations() != null && !major.getVariations().isEmpty()) {

             try {

                                 disassembleVariations(major, operation, result);

                         } catch (Exception e) {

                                 throw new AssemblyException("Error diassembling Variations during major update", e);

                         } 

         }

         if (major.getOrgCoreProgram() != null ) {

             disassembleCoreProgram(major, operation, result);

         }

         if (major.getCredentialProgramId() != null) {

             disassembleCredentialProgram(major, operation, result);

         }

         if (major.getResultOptions() != null) {

             disassembleResultOptions(major, operation, result);

         }

         if (major.getLearningObjectives() != null) {

             disassembleLearningObjectives(major, operation, result);

         }

         AmountInfo intensity = new AmountInfo();

         intensity.setUnitType(major.getIntensity());

                 clu.setIntensity(intensity);

         clu.setStdDuration(major.getStdDuration());

         clu.setInstructors(major.getPublishedInstructors());

         clu.setNextReviewPeriod(major.getNextReviewPeriod());

         clu.setEffectiveDate(major.getEffectiveDate());

         clu.setCampusLocations(major.getCampusLocations());

         clu.setDescr(major.getDescr());

         clu.setAccreditations(major.getAccreditingAgencies());

         clu.setNextReviewPeriod(major.getNextReviewPeriod());

                 // Add the Clu to the result

                 result.setNodeData(clu);

                 result.setOperation(operation);

                 result.setBusinessDTORef(major);

             return result;

     }

     private void disassembleLearningObjectives(MajorDisciplineInfo major, NodeOperation operation, BaseDTOAssemblyNode<MajorDisciplineInfo, CluInfo> result) throws AssemblyException {

         try {

             List<BaseDTOAssemblyNode<?, ?>> loResults = cluAssemblerUtils.disassembleLos(major.getId(), major.getState(),  major.getLearningObjectives(), operation);

             if (loResults != null) {

                 result.getChildNodes().addAll(loResults);

             }

         } catch (DoesNotExistException e) {

         } catch (Exception e) {

             throw new AssemblyException("Error while disassembling los", e);

         }

     }

     private void disassembleResultOptions(MajorDisciplineInfo major, NodeOperation operation, BaseDTOAssemblyNode<MajorDisciplineInfo, CluInfo> result) throws AssemblyException {

         //TODO Check for ProgramAssemblerConstants.CERTIFICATE_RESULTS too

         BaseDTOAssemblyNode<?, ?> degreeResults = cluAssemblerUtils.disassembleCluResults(

                 major.getId(), major.getState(), major.getResultOptions(), operation, ProgramAssemblerConstants.DEGREE_RESULTS, "Result options", "Result option");

         if (degreeResults != null) {

             result.getChildNodes().add(degreeResults);

         }

     }

     private void disassembleCredentialProgram(MajorDisciplineInfo major, NodeOperation operation, BaseDTOAssemblyNode<MajorDisciplineInfo, CluInfo> result) throws AssemblyException {

         List<BaseDTOAssemblyNode<?,?>> credentialResults;

         try {

             credentialResults = programAssemblerUtils.disassembleCredentialProgram(major, operation, ProgramAssemblerConstants.HAS_MAJOR_PROGRAM);

             if (credentialResults != null) {

                 result.getChildNodes().addAll(credentialResults);

             }

         } catch (Exception e) {

             throw new AssemblyException("Error while disassembling Credential program", e);

         }

     }

     private void disassembleVariations(MajorDisciplineInfo major, NodeOperation operation, BaseDTOAssemblyNode<MajorDisciplineInfo, CluInfo> result) throws AssemblyException, DoesNotExistException, InvalidParameterException, MissingParameterException, OperationFailedException {

             Map<String, CluCluRelationInfo> currentRelations = null;

             List<BaseDTOAssemblyNode<?, ?>> nodes = new ArrayList<BaseDTOAssemblyNode<?, ?>>();

             if (!NodeOperation.CREATE.equals(operation)){

                     currentRelations = programAssemblerUtils.getCluCluActiveRelations(major.getId(), ProgramAssemblerConstants.HAS_PROGRAM_VARIATION);

             }

             // Loop through all the variations in this MD

         for (ProgramVariationInfo variation : major.getVariations()) {

             BaseDTOAssemblyNode<?,?> variationNode;

             try {

                     if (NodeOperation.UPDATE.equals(operation) && variation.getId() != null

                                                 && (currentRelations != null && currentRelations.containsKey(variation.getId()))) {

                         // If the relationship already exists update it

                         // remove this entry from the map so we can tell what needs to be deleted at the end

                             variationNode = programVariationAssembler.disassemble(variation, operation);

                             if (variationNode != null) nodes.add(variationNode);

                             currentRelations.remove(variation.getId());  

                     } else if (!NodeOperation.DELETE.equals(operation)) {

                                         // the variation does not exist, so create variation & cluclurelation

                             variationNode = programVariationAssembler.disassemble(variation, NodeOperation.CREATE);

                             if (variationNode != null) nodes.add(variationNode);

                                         programAssemblerUtils.addCreateRelationNode(major.getId(), variation.getId(), ProgramAssemblerConstants.HAS_PROGRAM_VARIATION, nodes);

                                 }

             } catch (Exception e) {

                 throw new AssemblyException("Error while disassembling Variation", e);

             } 

         }

         // Now any leftover variation ids are no longer needed, so inactive them

         if(currentRelations != null && currentRelations.size() > 0){

                 programAssemblerUtils.addInactiveRelationNodes(currentRelations, nodes);          

                 addInactivateVariationNodes(currentRelations, nodes);

         }

         result.getChildNodes().addAll(nodes);

     }

     private void addInactivateVariationNodes(Map<String, CluCluRelationInfo> currentRelations, List<BaseDTOAssemblyNode<?, ?>> nodes) throws AssemblyException{

             for (String variationId : currentRelations.keySet()) {

                         CluInfo variationClu;

                         try {

                                 variationClu = luService.getClu(variationId);

                                 ProgramVariationInfo delVariation = programVariationAssembler.assemble(variationClu, null, true);

                                 delVariation.setState(DtoConstants.STATE_INACTIVE);

                                 BaseDTOAssemblyNode<?,?> variationNode = programVariationAssembler.disassemble(delVariation , NodeOperation.UPDATE);

                                 if (variationNode != null) nodes.add(variationNode);

                         } catch (Exception e) {

                                 throw new AssemblyException("Error while disassembling variation, deactivateVariations", e);

                         }

             }

     }

     private void disassembleCoreProgram(MajorDisciplineInfo major, NodeOperation operation, BaseDTOAssemblyNode<MajorDisciplineInfo, CluInfo> result) throws AssemblyException {

         BaseDTOAssemblyNode<?,?> coreResults;

         try {

             coreResults = coreProgramAssembler.disassemble(major.getOrgCoreProgram(), operation);

             if (coreResults != null) {

                 result.getChildNodes().add(coreResults);

             }

         } catch (Exception e) {

             throw new AssemblyException("Error while disassembling Core program", e);

         }

     }

     // Setters for Spring

     public void setLuService(LuService luService) {

         this.luService = luService;

     }

         public void setProgramVariationAssembler(ProgramVariationAssembler programVariationAssembler) {

         this.programVariationAssembler = programVariationAssembler;

     }

     public ProgramVariationAssembler getProgramVariationAssembler() {

         return programVariationAssembler;

     }

     public void setCoreProgramAssembler(CoreProgramAssembler coreProgramAssembler) {

         this.coreProgramAssembler = coreProgramAssembler;

     }

     public void setCluAssemblerUtils(CluAssemblerUtils cluAssemblerUtils) {

         this.cluAssemblerUtils = cluAssemblerUtils;

     }

     public void setProgramAssemblerUtils(ProgramAssemblerUtils programAssemblerUtils) {

         this.programAssemblerUtils = programAssemblerUtils;

     }

 }