package org.kuali.student.lum.program.client.major.edit;

 import java.util.ArrayList;

 import java.util.Iterator;

 import java.util.List;

 import java.util.Set;

 import java.util.TreeSet;

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

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

 import org.kuali.student.common.assembly.data.Data.Property;

 import org.kuali.student.common.ui.client.application.Application;

 import org.kuali.student.common.ui.client.application.KSAsyncCallback;

 import org.kuali.student.common.ui.client.application.ViewContext;

 import org.kuali.student.common.ui.client.configurable.mvc.views.SectionView;

 import org.kuali.student.common.ui.client.mvc.Callback;

 import org.kuali.student.common.ui.client.mvc.DataModel;

 import org.kuali.student.common.ui.client.mvc.HasCrossConstraints;

 import org.kuali.student.common.ui.client.mvc.ModelRequestCallback;

 import org.kuali.student.common.ui.client.mvc.View;

 import org.kuali.student.common.ui.client.mvc.history.HistoryManager;

 import org.kuali.student.common.ui.client.service.DataSaveResult;

 import org.kuali.student.common.ui.client.validator.ValidatorClientUtils;

 import org.kuali.student.common.ui.client.widgets.KSButton;

 import org.kuali.student.common.ui.client.widgets.KSButtonAbstract;

 import org.kuali.student.common.ui.client.widgets.notification.KSNotification;

 import org.kuali.student.common.ui.client.widgets.notification.KSNotifier;

 import org.kuali.student.common.ui.shared.IdAttributes;

 import org.kuali.student.common.ui.shared.IdAttributes.IdType;

 import org.kuali.student.common.validation.dto.ValidationResultInfo;

 import org.kuali.student.lum.common.client.configuration.LUMViews;

 import org.kuali.student.lum.common.client.helpers.RecentlyViewedHelper;

 import org.kuali.student.lum.common.client.widgets.AppLocations;

 import org.kuali.student.lum.program.client.ProgramConstants;

 import org.kuali.student.lum.program.client.ProgramRegistry;

 import org.kuali.student.lum.program.client.ProgramSections;

 import org.kuali.student.lum.program.client.ProgramStatus;

 import org.kuali.student.lum.program.client.ProgramUtils;

 import org.kuali.student.lum.program.client.events.AddSpecializationEvent;

 import org.kuali.student.lum.program.client.events.AfterSaveEvent;

 import org.kuali.student.lum.program.client.events.ChangeViewEvent;

 import org.kuali.student.lum.program.client.events.MetadataLoadedEvent;

 import org.kuali.student.lum.program.client.events.ModelLoadedEvent;

 import org.kuali.student.lum.program.client.events.SpecializationCreatedEvent;

 import org.kuali.student.lum.program.client.events.SpecializationSaveEvent;

 import org.kuali.student.lum.program.client.events.SpecializationUpdateEvent;

 import org.kuali.student.lum.program.client.events.StateChangeEvent;

 import org.kuali.student.lum.program.client.events.StoreRequirementIDsEvent;

 import org.kuali.student.lum.program.client.events.UpdateEvent;

 import org.kuali.student.lum.program.client.major.MajorController;

 import org.kuali.student.lum.program.client.properties.ProgramProperties;

 import org.kuali.student.lum.program.client.rpc.AbstractCallback;

 import org.kuali.student.lum.program.client.widgets.ProgramSideBar;

 import com.google.gwt.core.client.GWT;

 import com.google.gwt.event.dom.client.ClickEvent;

 import com.google.gwt.event.dom.client.ClickHandler;

 import com.google.gwt.event.shared.HandlerManager;

 import com.google.gwt.user.client.Window;

 /**

  * @author Igor

  */

 public class MajorEditController extends MajorController {

     private final KSButton saveButton = new KSButton(ProgramProperties.get().common_save());

     private final KSButton cancelButton = new KSButton(ProgramProperties.get().common_cancel(), KSButtonAbstract.ButtonStyle.ANCHOR_LARGE_CENTERED);

     private final Set<String> existingVariationIds = new TreeSet<String>();

     private ProgramStatus previousState;

     /**

      * Constructor.

      *

      * @param programModel

      */

     public MajorEditController(DataModel programModel, ViewContext viewContext, HandlerManager eventBus) {

         super(programModel, viewContext, eventBus);

         configurer = GWT.create(MajorEditConfigurer.class);

         sideBar.setState(ProgramSideBar.State.EDIT);

         initHandlers();

     }

     @Override

     protected void configureView() {

         super.configureView();

         if (!initialized) {

             eventBus.fireEvent(new MetadataLoadedEvent(programModel.getDefinition(), this));

             List<Enum<?>> excludedViews = new ArrayList<Enum<?>>();

             excludedViews.add(ProgramSections.PROGRAM_REQUIREMENTS_EDIT);

             excludedViews.add(ProgramSections.SUPPORTING_DOCUMENTS_EDIT);

             excludedViews.add(ProgramSections.SUMMARY);

             addCommonButton(ProgramProperties.get().program_menu_sections(), saveButton, excludedViews);

             addCommonButton(ProgramProperties.get().program_menu_sections(), cancelButton, excludedViews);

             initialized = true;

         }

     }

     private void initHandlers() {

         saveButton.addClickHandler(new ClickHandler() {

             @Override

             public void onClick(ClickEvent event) {

                 doSave();

             }

         });

         cancelButton.addClickHandler(new ClickHandler() {

             @Override

             public void onClick(ClickEvent event) {

                 doCancel();

             }

         });

         eventBus.addHandler(UpdateEvent.TYPE, new UpdateEvent.Handler() {

             @Override

             public void onEvent(UpdateEvent event) {

                 Enum<?> view = event.getCurrentView();

                 if (view != null) {

                     setCurrentViewEnum(view);

                 }

                 doSave(event.getOkCallback());

             }

         });

         eventBus.addHandler(StateChangeEvent.TYPE, new StateChangeEvent.Handler() {

             @Override

             public void onEvent(final StateChangeEvent event) {

                     //FIXME: The proper way of doing this would be a single server side call to validate next state

                     //which would retrieve warnings & required for next state, instead of re-validating warnings for

                     //current state server side and validating required for next state client side.

                     programRemoteService.validate(programModel.getRoot(), new KSAsyncCallback<List<ValidationResultInfo>>(){

                                         @Override

                                         public void onSuccess(final List<ValidationResultInfo> currentStateResults) {

                                 programModel.validateNextState(new Callback<List<ValidationResultInfo>>() {

                                     @Override

                                     public void exec(List<ValidationResultInfo> nextStateResults) {

                                             //Update validation warnings and process for all screens

                                             Application.getApplicationContext().clearValidationWarnings();

                                             Application.getApplicationContext().addValidationWarnings(currentStateResults);

                                             isValid(Application.getApplicationContext().getValidationWarnings(), false);

                                             boolean isSectionValid = isValid(nextStateResults, true) 

                                                 && Application.getApplicationContext().getValidationWarnings().isEmpty();

                                         if (isSectionValid) {

                                             Callback<Boolean> callback = new Callback<Boolean>() {

                                                 @Override

                                                 public void exec(Boolean result) {

                                                     if (result) {

                                                         reloadMetadata = true;

                                                         loadMetadata(new Callback<Boolean>() {

                                                             @Override

                                                             public void exec(Boolean result) {

                                                                 if (result) {

                                                                     ProgramUtils.syncMetadata(configurer, programModel.getDefinition());

                                                                     HistoryManager.navigate(AppLocations.Locations.VIEW_PROGRAM.getLocation(), context);

                                                                 }

                                                             }

                                                         });

                                                     }

                                                 }

                                             };

                                             previousState = ProgramStatus.of(programModel);

                                             ProgramUtils.setStatus(programModel, event.getProgramStatus().getValue());

                                             saveData(callback);

                                         } else {

                                             KSNotifier.add(new KSNotification("Unable to save, please check fields for errors.", false, true, 5000));

                                         }

                                     }

                                 });

                                         }

                     });                

             }

         });

         eventBus.addHandler(SpecializationSaveEvent.TYPE, new SpecializationSaveEvent.Handler() {

             @Override

             public void onEvent(SpecializationSaveEvent event) {

                 Data currentVariations = getDataProperty(ProgramConstants.VARIATIONS);

                 existingVariationIds.clear();

                 for (Data.Property prop : currentVariations) {

                     String existingId = (String) ((Data) prop.getValue()).get(ProgramConstants.ID);

                     existingVariationIds.add(existingId);

                 }

                 String updatedId = event.getData().get(ProgramConstants.ID);

                 Integer updatedKey = null;

                 //FIXME: This is ugly but gets us past a  blocker issue. 

                 // Theres something wrong with the way the models are

                 // handled in the major and variation controllers  so they get out of sync.

                 // This is a temporary workaround

                 if (updatedId != null) { // this is an update of an existing variation

                     for (Data.Property prop : currentVariations) {

                         String id = (String) ((Data) prop.getValue()).get(ProgramConstants.ID);

                         if (updatedId.equals(id)) {

                             updatedKey = prop.getKey();

                             Data currentMetaInfo = ((Data) prop.getValue()).get("metaInfo");

                             String latestVersionInd = currentMetaInfo.get("versionInd");

                             Data newMetaInfo = event.getData().get("metaInfo");

                             if (newMetaInfo == null) {

                                 newMetaInfo = new Data();

                                 event.getData().set("metaInfo", newMetaInfo);

                             }

                             newMetaInfo.set("versionInd", latestVersionInd);

                             break;

                         }

                     }

                     currentVariations.set(updatedKey, event.getData());

                 } else {

                     currentVariations.add(event.getData());

                 }

                 doSave();

             }

         });

         eventBus.addHandler(AddSpecializationEvent.TYPE, new AddSpecializationEvent.Handler() {

             @Override

             public void onEvent(AddSpecializationEvent event) {

                 String id = getStringProperty(ProgramConstants.ID);

                 ProgramRegistry.setRow((getDataProperty(ProgramConstants.VARIATIONS)).size());

                 ProgramRegistry.setData(ProgramUtils.createNewSpecializationBasedOnMajor(programModel));

                 ViewContext viewContext = new ViewContext();

                 viewContext.setId(id);

                 viewContext.setIdType(IdAttributes.IdType.OBJECT_ID);

                 HistoryManager.navigate(AppLocations.Locations.EDIT_VARIATION.getLocation(), viewContext);

             }

         });

         eventBus.addHandler(StoreRequirementIDsEvent.TYPE, new StoreRequirementIDsEvent.Handler() {

             @Override

             public void onEvent(StoreRequirementIDsEvent event) {

                 List<String> ids = event.getProgramRequirementIds();

                 programModel.set(QueryPath.parse(ProgramConstants.PROGRAM_REQUIREMENTS), new Data());

                 Data programRequirements = programModel.get(ProgramConstants.PROGRAM_REQUIREMENTS);

                 if (programRequirements == null) {

                     Window.alert("Cannot find program requirements in data model.");

                     GWT.log("Cannot find program requirements in data model", null);

                     return;

                 }

                 for (String id : ids) {

                     programRequirements.add(id);

                 }

                 doSave();

             }

         });

         eventBus.addHandler(ChangeViewEvent.TYPE, new ChangeViewEvent.Handler() {

             @Override

             public void onEvent(ChangeViewEvent event) {

                 showView(event.getViewToken());

             }

         });

     }

     @Override

     protected void loadModel(ModelRequestCallback<DataModel> callback) {

         ViewContext viewContext = getViewContext();

         if (viewContext.getIdType() == IdType.COPY_OF_OBJECT_ID) {

             createNewVersionAndLoadModel(callback, viewContext);

         } else {

             super.loadModel(callback);

         }

     }

     protected void createNewVersionAndLoadModel(final ModelRequestCallback<DataModel> callback, final ViewContext viewContext) {

         Data data = new Data();

         Data versionData = new Data();

         versionData.set(new Data.StringKey("versionIndId"), getViewContext().getId());

         versionData.set(new Data.StringKey("versionComment"), "Major Disicpline Version");

         data.set(new Data.StringKey("versionInfo"), versionData);

         programRemoteService.saveData(data, new AbstractCallback<DataSaveResult>(ProgramProperties.get().common_retrievingData()) {

             @Override

             public void onSuccess(DataSaveResult result) {

                 super.onSuccess(result);

                 programModel.setRoot(result.getValue());

                 viewContext.setId(ProgramUtils.getProgramId(programModel));

                 viewContext.setIdType(IdType.OBJECT_ID);

                 setHeaderTitle();

                 setStatus();

                 callback.onModelReady(programModel);

                 eventBus.fireEvent(new ModelLoadedEvent(programModel));

             }

             @Override

             public void onFailure(Throwable caught) {

                 super.onFailure(caught);

                 callback.onRequestFail(caught);

             }

         });

     }

     private void doSave(final Callback<Boolean> okCallback) {

         requestModel(new ModelRequestCallback<DataModel>() {

             @Override

             public void onModelReady(DataModel model) {

                 MajorEditController.this.updateModelFromCurrentView();

                 model.validate(new Callback<List<ValidationResultInfo>>() {

                     @Override

                     public void exec(List<ValidationResultInfo> result) {

                         boolean isSectionValid = isValid(result, true);

                         if (isSectionValid) {

                             saveData(okCallback);

                         } else {

                             okCallback.exec(false);

                             KSNotifier.add(new KSNotification("Unable to save, please check fields for errors.", false, true, 5000));

                         }

                     }

                 });

             }

             @Override

             public void onRequestFail(Throwable cause) {

                 GWT.log("Unable to retrieve model for validation and save", cause);

             }

         });

     }

     private void doCancel() {

         showView(ProgramSections.SUMMARY);

     }

     @Override

     protected void doSave() {

         doSave(NO_OP_CALLBACK);

     }

     private void saveData(final Callback<Boolean> okCallback) {

         programRemoteService.saveData(programModel.getRoot(), new AbstractCallback<DataSaveResult>(ProgramProperties.get().common_savingData()) {

             @Override

             public void onSuccess(DataSaveResult result) {

                 super.onSuccess(result);

                 //Clear warning states on field and any warnings stored in ApplicationContext;

                 clearAllWarnings();

                 Application.getApplicationContext().clearValidationWarnings();

                 List<ValidationResultInfo> validationResults = result.getValidationResults();

                 Application.getApplicationContext().addValidationWarnings(validationResults);

                 if (ValidatorClientUtils.hasErrors(validationResults)) {

                     if (previousState != null) {

                         ProgramUtils.setStatus(programModel, previousState.getValue());

                     }

                     ProgramUtils.retrofitValidationResults(validationResults);

                     isValid(validationResults, false, true);

                     ProgramUtils.handleValidationErrorsForSpecializations(validationResults, programModel);

                     //Clean up anything created by earlier code

                     Data currentVariations = getDataProperty(ProgramConstants.VARIATIONS);

                     existingVariationIds.clear();

                     for (Iterator<Property> iter = currentVariations.iterator();iter.hasNext();) {

                             Property prop = iter.next();

                         String existingId = (String) ((Data) prop.getValue()).get(ProgramConstants.ID);

                         if(existingId==null){

                                 iter.remove();

                         }else{

                                 existingVariationIds.add(existingId);

                         }

                     }

                     okCallback.exec(false);

                 } else {

                     Data data = result.getValue();

                     if (data != null) {

                         programModel.setRoot(result.getValue());

                     }

                     previousState = null;

                     setHeaderTitle();

                     setStatus();

                     resetFieldInteractionFlag();

                     configurer.applyPermissions();

                     handleSpecializations();

                     throwAfterSaveEvent();

                     HistoryManager.logHistoryChange();

                     ViewContext viewContext = getViewContext();

                     viewContext.setId(getStringProperty(ProgramConstants.ID));

                     viewContext.setIdType(IdType.OBJECT_ID);

                                     if (ValidatorClientUtils.hasWarnings(validationResults)){

                                             //Show validation warnings for major

                                             isValid(result.getValidationResults(), false, true);                                            

                                             KSNotifier.show("Saved with Warnings");

                                     } else {

                         KSNotifier.show(ProgramProperties.get().common_successfulSave());

                                     }                                  

                     // add to recently viewed now that we're sure to know the program's id

                     ViewContext docContext = new ViewContext();

                     docContext.setId(getStringProperty(ProgramConstants.ID));

                     docContext.setIdType(IdType.OBJECT_ID);

                     docContext.setAttribute(ProgramConstants.TYPE, ProgramConstants.MAJOR_LU_TYPE_ID + '/' + ProgramSections.PROGRAM_DETAILS_VIEW);

                     RecentlyViewedHelper.addDocument(getProgramName(),

                             HistoryManager.appendContext(AppLocations.Locations.VIEW_PROGRAM.getLocation(), docContext));

                     okCallback.exec(true);

                     processCurrentView();

                 }

             }

         });

     }

     private void processCurrentView() {

         Enum<?> currentView = getCurrentViewEnum();

         if (currentView.name().equals(ProgramSections.VIEW_ALL.name())) {

             HistoryManager.navigate(AppLocations.Locations.VIEW_PROGRAM.getLocation(), getViewContext());

         } else {

             showView(currentView);

         }

     }

     /**

      * Handles after save work for specializations.

      */

     private void handleSpecializations() {

         String newVariationId = null;

         Data variations = programModel.get(ProgramConstants.VARIATIONS);

         for (Data.Property prop : variations) {

             String varId = (String) ((Data) prop.getValue()).get(ProgramConstants.ID);

             if (!existingVariationIds.contains(varId)) {

                 newVariationId = varId;

                 existingVariationIds.add(newVariationId);

                 break;

             }

         }

         if (newVariationId != null) {

             eventBus.fireEvent(new SpecializationCreatedEvent(newVariationId));

         } else {

             eventBus.fireEvent(new SpecializationUpdateEvent(variations));

         }

     }

     private void throwAfterSaveEvent() {

         eventBus.fireEvent(new AfterSaveEvent(programModel, this));

     }

     @Override

     public void onModelLoadedEvent() {

         Enum<?> changeSection = ProgramRegistry.getSection();

         if (changeSection != null) {

             showView(changeSection);

             ProgramRegistry.setSection(null);

         } else {

             String id = getStringProperty(ProgramConstants.ID);

             if (id == null) {

                 showView(ProgramSections.PROGRAM_DETAILS_EDIT);

             } else {

                 showView(ProgramSections.SUMMARY);

             }

         }

     }

         @Override

         public void beforeShow(final Callback<Boolean> onReadyCallback) {

                 if(!initialized){

                         Application.getApplicationContext().clearCrossConstraintMap(null);

                         Application.getApplicationContext().clearPathToFieldMapping(null);

                 }

                 super.beforeShow(onReadyCallback);

         }

         //Before show is called before the model is bound to the widgets. We need to update cross constraints after widget binding

         //This gets called twice which is not optimal

         @Override

         public <V extends Enum<?>> void showView(V viewType,

                         final Callback<Boolean> onReadyCallback) {

                 Callback<Boolean> updateCrossConstraintsCallback = new Callback<Boolean>(){

                         public void exec(Boolean result) {

                                 onReadyCallback.exec(result);

                         for(HasCrossConstraints crossConstraint:Application.getApplicationContext().getCrossConstraints(null)){

                                 crossConstraint.reprocessWithUpdatedConstraints();

                         }

                         showWarnings();        

                         }

         };

                 super.showView(viewType, updateCrossConstraintsCallback);

         }

         //Ensure that the managing bodies section view is updated before the user edits specializations

         @Override

         public void beforeViewChange(Enum<?> viewChangingTo, final Callback<Boolean> okToChangeCallback){

                 if(LUMViews.VARIATION_EDIT.equals(viewChangingTo)){

                         getView(ProgramSections.MANAGE_BODIES_EDIT, new Callback<View>(){

                                 @Override

                                 public void exec(final View view) {

                                         if(view!=null && view instanceof SectionView){

                                                 requestModel(new ModelRequestCallback<DataModel>(){

                                                         public void onModelReady(DataModel model) {

                                                                 ((SectionView)view).updateWidgetData(model);

                                                                 okToChangeCallback.exec(true);

                                                         }

                                                         public void onRequestFail(Throwable cause) {

                                                                 okToChangeCallback.exec(false);

                                                         }

                                                 });

                                         }else{

                                                 okToChangeCallback.exec(true);

                                         }

                                 }});

                 }else{

                         okToChangeCallback.exec(true);

                 }

                 this.showExport(isExportButtonActive());        // KSLAB-1916

         }

 }