/*******************************************************************************
 * Copyright (c) 2004, 2006 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 * IBM Corporation - initial API and implementation
 * yyyymmdd bug      Email and other contact information
 * -------- -------- -----------------------------------------------------------
 * 20060404 134913   sengpl@ca.ibm.com - Seng Phung-Lu  
 * 20060517   142339 sengpl@ca.ibm.com - Seng Phung-Lu
 *******************************************************************************/
package org.eclipse.jst.ws.internal.consumption.ui.widgets;

import java.util.ArrayList;
import java.util.Enumeration;
import java.util.Hashtable;
import java.util.List;

import org.eclipse.core.resources.IProject;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.jem.java.JavaClass;
import org.eclipse.jst.j2ee.applicationclient.componentcore.util.AppClientArtifactEdit;
import org.eclipse.jst.j2ee.ejb.componentcore.util.EJBArtifactEdit;
import org.eclipse.jst.j2ee.web.componentcore.util.WebArtifactEdit;
import org.eclipse.jst.j2ee.webservice.wsclient.Handler;
import org.eclipse.jst.j2ee.webservice.wsclient.ServiceRef;
import org.eclipse.jst.j2ee.webservice.wsclient.Webservice_clientFactory;
import org.eclipse.jst.j2ee.webservice.wsclient.internal.impl.Webservice_clientFactoryImpl;
import org.eclipse.jst.ws.internal.common.J2EEUtils;
import org.eclipse.jst.ws.internal.common.JavaMOFUtils;
import org.eclipse.jst.ws.internal.consumption.ui.ConsumptionUIMessages;
import org.eclipse.jst.ws.internal.consumption.ui.common.HandlerServiceRefHolder;
import org.eclipse.jst.ws.internal.consumption.ui.widgets.object.HandlerTableItem;
import org.eclipse.wst.command.internal.env.core.common.StatusUtils;
import org.eclipse.wst.common.componentcore.ArtifactEdit;
import org.eclipse.wst.common.frameworks.datamodel.AbstractDataModelOperation;
import org.eclipse.wst.common.internal.emf.utilities.EtoolsCopyUtility;

/*
 * Provide a way to externalize the edited fields and create new handlers
 *  
 */
public class ClientHandlersWidgetOutputCommand extends AbstractDataModelOperation 
{

  private HandlerServiceRefHolder[] handlerServiceRefHolder_;
  private Hashtable oldWSServiceRefsToHandlersTable_;
  private Hashtable newWSServiceRefsToHandlersTable_;
  private IProject project_;
  private boolean isMultipleSelection_;

  public IStatus execute( IProgressMonitor monitor, IAdaptable adaptable )
  {
    IStatus status = Status.OK_STATUS;

    try {
      
      if (isMultipleSelection_) {
        if (handlerServiceRefHolder_ != null) {
          // add new handler to internal model if it doesn't already exist
          for (int i = 0; i < handlerServiceRefHolder_.length; i++) {
            handlerServiceRefHolder_[i].addHandlerToServiceRef();
            saveEditModel(handlerServiceRefHolder_[i].getProject());
          }
          
        }
        
      }
      else {
        oldWSServiceRefsToHandlersTable_ = new Hashtable();
        newWSServiceRefsToHandlersTable_ = new Hashtable();
        
        if (handlerServiceRefHolder_ != null) {
  
          for (int i = 0; i < handlerServiceRefHolder_.length; i++) {
  
            ServiceRef wsServiceRef =handlerServiceRefHolder_[i].getServiceRef(); 
        
            List wsHandlers = wsServiceRef.getHandlers();
            oldWSServiceRefsToHandlersTable_.put(wsServiceRef, wsHandlers);
            newWSServiceRefsToHandlersTable_.put(wsServiceRef, new ArrayList());
          }
  
          //Enumeration e = handlersTable_.keys();
          for (int j=0;j<handlerServiceRefHolder_.length;j++) {

            List handlerTableItems_ =  handlerServiceRefHolder_[j].getHandlerList();
            if (handlerTableItems_ != null) {
              // form Handler tables
              for (int i = 0; i < handlerTableItems_.size(); i++) {
  
                HandlerTableItem hti = (HandlerTableItem) handlerTableItems_.get(i);
                Object wsModelRef = hti.getWsDescRef();
                if (wsModelRef != null && wsModelRef instanceof ServiceRef) {
                  ServiceRef wsRef = (ServiceRef) wsModelRef;
  
                  Object handler = hti.getHandler();
                  if (handler != null && handler instanceof Handler) {
                    // clone it
                    Handler clonedHandler = (Handler) EtoolsCopyUtility.createCopy((Handler) handler);
                    ((List) newWSServiceRefsToHandlersTable_.get(wsRef)).add(clonedHandler);
                  }
                  else {
                    // create it
                    Webservice_clientFactory wsClientFactory = new Webservice_clientFactoryImpl();
                    Handler newHandler = wsClientFactory.createHandler();
                    newHandler.setHandlerName(hti.getHandlerName());
  
                    JavaClass javaClass = JavaMOFUtils.getJavaClass(hti.getHandlerClassName(), project_);
                    if (javaClass != null) {
                      newHandler.setHandlerClass(javaClass);
                    }
  
                    ((List) newWSServiceRefsToHandlersTable_.get(wsRef)).add(newHandler);
                  }
  
                }
  
              }
            }
          }
 
          // add handlers to ports
          addHandlersToServiceRefs();
        }
        
        // save the artifact edit model
        saveEditModel();
      }
      
    }
    catch (Exception e) 
    {
      return StatusUtils.errorStatus(ConsumptionUIMessages.MSG_ERROR_TASK_EXCEPTED, e);
    }

    return status;
  }

  private void saveEditModel() {
      ArtifactEdit artifactEdit = null;
      try {
          if (J2EEUtils.isWebComponent(project_)) {
                artifactEdit = WebArtifactEdit.getWebArtifactEditForWrite(project_);
          }
          else if (J2EEUtils.isEJBComponent(project_)){
                artifactEdit = EJBArtifactEdit.getEJBArtifactEditForWrite(project_);
            }
          else if (J2EEUtils.isAppClientComponent(project_)){
                artifactEdit = AppClientArtifactEdit.getAppClientArtifactEditForWrite(project_);
          }
      }
      finally {
          if (artifactEdit!=null) {
              artifactEdit.save(null);
              artifactEdit.dispose();
          }
      }
  }
  
  private void saveEditModel(IProject project){
	  project_ = project;
	  saveEditModel();
  }
  
  private void addHandlersToServiceRefs() {
    try {
      Enumeration refsToHandlers = newWSServiceRefsToHandlersTable_.keys();
      while (refsToHandlers.hasMoreElements()) {
        ServiceRef serviceRef = (ServiceRef) refsToHandlers.nextElement();
        if (serviceRef != null) {
          List handlers = (List) newWSServiceRefsToHandlersTable_.get(serviceRef);
          List modelHandlers = (List) oldWSServiceRefsToHandlersTable_.get(serviceRef);
          modelHandlers.clear();
          modelHandlers.addAll(handlers);
        }
      }
    }
    catch (Exception e) {
      e.printStackTrace();
    }
  }


  public void setClientProject(IProject project) {
    this.project_ = project;
  }
  
  /**
   * An array of HandlerDescriptionHolders
   * @return
   */
  public void setHandlerServiceRefHolder(HandlerServiceRefHolder[] handlerHolders){
    this.handlerServiceRefHolder_ = handlerHolders;
  }
  
  public void setIsMultipleSelection(boolean isMulitpleSelection) {
    this.isMultipleSelection_ = isMulitpleSelection;
  }

}