/**

  * Copyright 2010 The Kuali Foundation Licensed under the

  * Educational Community License, Version 2.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.osedu.org/licenses/ECL-2.0

  *

  * 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.statement.config.context;

 import java.util.ArrayList;

 import java.util.List;

 import java.util.Map;

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

 import org.kuali.student.common.versionmanagement.dto.VersionDisplayInfo;

 import org.kuali.student.core.statement.dto.ReqComponentInfo;

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

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

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

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

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

 import org.kuali.student.lum.statement.config.context.util.NLCluSet;

 import org.kuali.student.lum.statement.typekey.ReqComponentFieldTypes;

 /**

  * This class creates the template context for course list types.

  */

 public class LuContextImpl extends BasicContextImpl {

     /**

      * Learning unit service.

      */

         private LuService luService;

         /**

          * <code>clu</code> token (key) references a Clu object used in templates.

          * e.g. 'Student must have completed all of 

          * $clu.getOfficialIdentifier().getShortName()'

          */

         public final static String CLU_TOKEN = "clu";

         public final static String COURSE_CLU_TOKEN = "courseClu";

         public final static String PROGRAM_CLU_TOKEN = "programClu";

         public final static String TEST_CLU_TOKEN = "testClu";

         /**

          * <code>cluSet</code> token (key) references a Clu set object

          * used in templates.

          * e.g. 'Student must have completed all of $cluSet.getCluSetAsCode()'

          */

         public final static String CLU_SET_TOKEN = "cluSet";

         public final static String COURSE_CLU_SET_TOKEN = "courseCluSet";

         public final static String PROGRAM_CLU_SET_TOKEN = "programCluSet";

         public final static String TEST_CLU_SET_TOKEN = "testCluSet";

         /**

          * Sets the LU service.

          *

          * @param luService LU service

          */

     public void setLuService(LuService luService) {

                 this.luService = luService;

         }

         /**

      * Gets a CLU.

      *

      * @param cluId CLU id

      * @return CLU

      * @throws OperationFailedException If retrieving CLU fails

      */

     public CluInfo getCluInfo(String cluId) throws OperationFailedException {

                 if (cluId == null) {

                         return null;

                 }

                 try {

                         VersionDisplayInfo versionInfo = luService.getCurrentVersion(LuServiceConstants.CLU_NAMESPACE_URI, cluId);

                         CluInfo clu = this.luService.getClu(versionInfo.getId());

                         return clu;

                 } catch(Exception e) {

                         throw new OperationFailedException(e.getMessage(), e);

                 }

     }

     private CluInfo getClu(ReqComponentInfo reqComponent, String key) throws OperationFailedException {

         Map<String, String> map = getReqComponentFieldMap(reqComponent);

         if(map.containsKey(key)) {

                     String cluId = map.get(key);

                     return getCluInfo(cluId);

         }

         return null;

     }

     /**

      * Gets a CLU set.

      *

      * @param cluSetId CLU set id

      * @return CLU set

      * @throws OperationFailedException If retrieving CLU set fails

      */

     public CluSetInfo getCluSetInfo(String cluSetId) throws OperationFailedException {

                 if (cluSetId == null) {

                         return null;

                 }

                 try {

                     CluSetInfo cluSet = this.luService.getCluSetInfo(cluSetId);

                     return cluSet;

                 } catch(Exception e) {

                         throw new OperationFailedException(e.getMessage(), e);

                 }

     }

     /**

      * Gets the CLU set.

      *

      * @param cluSetId CLU set id

      * @return CLU set

      * @throws OperationFailedException If building a custom CLU set fails

      */

     public NLCluSet getCluSet(String cluSetId) throws OperationFailedException {

                 if (cluSetId == null) {

                         return null;

                 }

             CluSetInfo cluSet = getCluSetInfo(cluSetId);

                 try {

                     List<CluInfo> list = new ArrayList<CluInfo>();

                     CluSetTreeViewInfo tree = luService.getCluSetTreeView(cluSetId);

                     findClusInCluSet(tree, list);

                     return new NLCluSet(cluSet.getId(), list, cluSet);

                 } catch(Exception e) {

                         throw new OperationFailedException(e.getMessage(), e);

                 }

     }

     private static void findClusInCluSet(CluSetTreeViewInfo tree, List<CluInfo> cluList) {

             if (tree.getCluSets() != null) {

                         for (CluSetTreeViewInfo cluSet : tree.getCluSets()) {

                                 findClusInCluSet(cluSet, cluList);

                         }

                 } else {

                         for (CluInfo clu : tree.getClus()) {

                                 if (!containsClu(cluList, clu)) {

                                         cluList.add(clu);

                                 }

                         }

                 }

         }

         private static boolean containsClu(List<CluInfo> cluList, CluInfo clu) {

                 for (CluInfo clu2 : cluList) {

                         if (clu2.getId().equals(clu.getId())) {

                                 return true;

                         }

                 }

                 return false;

         }

         /**

      * Gets a new CLU set from comma separated list of CLU ids.

      *

      * @param cluIds Comma separated list of CLU ids

      * @return A new CLU set

      * @throws OperationFailedException If building a custom CLU set fails

      */

     /*public NLCluSet getClusAsCluSet(String cluIds) throws OperationFailedException {

             String[] cluIdArray = cluIds.split("\\s*,\\s*");

             List<CluInfo> list = new ArrayList<CluInfo>();

             for(String cluId : cluIdArray) {

                     CluInfo clu = getCluInfo(cluId);

                     list.add(clu);

             }

             return new NLCluSet(null, list);

     }*/

     /**

      * Gets a custom CLU set from a requirement component.

      *

      * @param reqComponent Requirement component

      * @return custom CLU set

      * @throws OperationFailedException If building a custom CLU set fails

      */

     public NLCluSet getCluSet(ReqComponentInfo reqComponent, String key) throws OperationFailedException {

         Map<String, String> map = getReqComponentFieldMap(reqComponent);

             NLCluSet cluSet = null;

             if(map.containsKey(key)) {

                 String cluSetId = map.get(key);

             cluSet = getCluSet(cluSetId);

         }

             return cluSet;

     }

     /**

      * Creates the context map (template data) for the requirement component.

      * 

      * @param reqComponent Requirement component

      * @throws OperationFailedException Creating context map fails

      */

     public Map<String, Object> createContextMap(ReqComponentInfo reqComponent) throws OperationFailedException {

         Map<String, Object> contextMap = super.createContextMap(reqComponent);

         CluInfo clu = getClu(reqComponent, ReqComponentFieldTypes.CLU_KEY.getId());

         if(clu != null) {

                 contextMap.put(CLU_TOKEN, clu);

         }

         CluInfo courseClu = getClu(reqComponent, ReqComponentFieldTypes.COURSE_CLU_KEY.getId());

         if(courseClu != null) {

                 contextMap.put(COURSE_CLU_TOKEN, courseClu);

         }

         CluInfo programClu = getClu(reqComponent, ReqComponentFieldTypes.PROGRAM_CLU_KEY.getId());

         if(programClu != null) {

                 contextMap.put(PROGRAM_CLU_TOKEN, programClu);

         }

         CluInfo testClu = getClu(reqComponent, ReqComponentFieldTypes.TEST_CLU_KEY.getId());

         if(testClu != null) {

                 contextMap.put(TEST_CLU_TOKEN, testClu);

         }

         NLCluSet cluSet = getCluSet(reqComponent, ReqComponentFieldTypes.CLUSET_KEY.getId());

         if(cluSet != null) {

                 contextMap.put(CLU_SET_TOKEN, cluSet);

         }

         NLCluSet courseCluSet = getCluSet(reqComponent, ReqComponentFieldTypes.COURSE_CLUSET_KEY.getId());

         if(courseCluSet != null) {

                 contextMap.put(COURSE_CLU_SET_TOKEN, courseCluSet);

         }

         NLCluSet programCluSet = getCluSet(reqComponent, ReqComponentFieldTypes.PROGRAM_CLUSET_KEY.getId());

         if(programCluSet != null) {

                 contextMap.put(PROGRAM_CLU_SET_TOKEN, programCluSet);

         }

         NLCluSet testCluSet = getCluSet(reqComponent, ReqComponentFieldTypes.TEST_CLUSET_KEY.getId());

         if(testCluSet != null) {

                 contextMap.put(TEST_CLU_SET_TOKEN, testCluSet);

         }

         return contextMap;

     }

 }