plugins/org.eclipse.jst.j2ee/earproject/org/eclipse/jst/j2ee/application/internal/operations/AddComponentToEnterpriseApplicationOp.java

package org.eclipse.jst.j2ee.application.internal.operations;

import java.util.List;
import java.util.Properties;

import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.jem.util.logger.proxy.Logger;
import org.eclipse.jst.j2ee.application.Application;
import org.eclipse.jst.j2ee.application.Module;
import org.eclipse.jst.j2ee.application.WebModule;
import org.eclipse.jst.j2ee.componentcore.util.EARArtifactEdit;
import org.eclipse.jst.j2ee.internal.J2EEConstants;
import org.eclipse.jst.j2ee.internal.application.ApplicationPackage;
import org.eclipse.wst.common.componentcore.ComponentCore;
import org.eclipse.wst.common.componentcore.datamodel.properties.ICreateReferenceComponentsDataModelProperties;
import org.eclipse.wst.common.componentcore.internal.operation.CreateReferenceComponentsOp;
import org.eclipse.wst.common.componentcore.internal.util.IModuleConstants;
import org.eclipse.wst.common.componentcore.resources.ComponentHandle;
import org.eclipse.wst.common.componentcore.resources.IVirtualComponent;
import org.eclipse.wst.common.frameworks.datamodel.IDataModel;

public class AddComponentToEnterpriseApplicationOp extends CreateReferenceComponentsOp {
	public static final String metaInfFolderDeployPath = "/"; //$NON-NLS-1$

	public AddComponentToEnterpriseApplicationOp(IDataModel model) {
		super(model);
	}
	
	public IStatus execute(IProgressMonitor monitor, IAdaptable info) throws ExecutionException {
		addReferencedComponents(monitor);
		updateEARDD(monitor);
		return OK_STATUS;
	}

	protected void updateEARDD(IProgressMonitor monitor){
		
		EARArtifactEdit earEdit = null;
       	try {
			ComponentHandle handle = (ComponentHandle) model.getProperty(ICreateReferenceComponentsDataModelProperties.SOURCE_COMPONENT_HANDLE);
			earEdit = EARArtifactEdit.getEARArtifactEditForWrite(handle);
			if(earEdit != null){
				Application application = earEdit.getApplication();
				List list = (List)model.getProperty(ICreateReferenceComponentsDataModelProperties.TARGET_COMPONENTS_HANDLE_LIST);
				if (list != null && list.size() > 0) {
					for (int i = 0; i < list.size(); i++) {
						ComponentHandle tmphandle = (ComponentHandle)list.get(i);
						IVirtualComponent wc = ComponentCore.createComponent(tmphandle.getProject(), tmphandle.getName());
						addModule(application, wc);	
					}
				}
			}	
			earEdit.saveIfNecessary(monitor);
       	}
       	catch(Exception e){
       		Logger.getLogger().logError(e);
       	} finally {			
       		if(earEdit != null)
				earEdit.dispose();
       	}		
	}
	protected Module createNewModule(IVirtualComponent wc) {
		
		String type = wc.getComponentTypeId();
		if ( type.equals(IModuleConstants.JST_WEB_MODULE) ){
			return ((ApplicationPackage) EPackage.Registry.INSTANCE.getEPackage(ApplicationPackage.eNS_URI)).getApplicationFactory().createWebModule();
		}else if ( type.equals(IModuleConstants.JST_EJB_MODULE)) {
			return ((ApplicationPackage) EPackage.Registry.INSTANCE.getEPackage(ApplicationPackage.eNS_URI)).getApplicationFactory().createEjbModule();
		}else if ( type.equals(IModuleConstants.JST_APPCLIENT_MODULE)) {
			return ((ApplicationPackage) EPackage.Registry.INSTANCE.getEPackage(ApplicationPackage.eNS_URI)).getApplicationFactory().createJavaClientModule();
		}else if ( type.equals(IModuleConstants.JST_CONNECTOR_MODULE)) {
			return ((ApplicationPackage) EPackage.Registry.INSTANCE.getEPackage(ApplicationPackage.eNS_URI)).getApplicationFactory().createConnectorModule();
		}
		return null;
	}
	
	protected void addModule(Application application, IVirtualComponent wc) {
		Application dd = application;
		
		Module m = createNewModule(wc);

		String name = wc.getName();
		String type = wc.getComponentTypeId();
		
		if ( type.equals(IModuleConstants.JST_WEB_MODULE) ){
			name += ".war"; //$NON-NLS-1$
		}else if ( type.equals(IModuleConstants.JST_EJB_MODULE)) {
			name += ".jar"; //$NON-NLS-1$
		}else if ( type.equals(IModuleConstants.JST_APPCLIENT_MODULE)) {
			name += ".jar"; //$NON-NLS-1$
		}else if ( type.equals(IModuleConstants.JST_CONNECTOR_MODULE)) {
			name += ".rar"; //$NON-NLS-1$
		}
		
		if( m!= null){
			m.setUri(name);
			if (m instanceof WebModule) {

					
			Properties props = wc.getMetaProperties();
			String contextroot = ""; //$NON-NLS-1$
			if(( props != null ) && ( props.containsKey(J2EEConstants.CONTEXTROOT)))
				contextroot = props.getProperty(J2EEConstants.CONTEXTROOT);
				((WebModule) m).setContextRoot(contextroot);
			}		
			dd.getModules().add(m);	
		}	
	}

	public IStatus redo(IProgressMonitor monitor, IAdaptable info) throws ExecutionException {
		// TODO Auto-generated method stub
		return null;
	}

	public IStatus undo(IProgressMonitor monitor, IAdaptable info) throws ExecutionException {
		// TODO Auto-generated method stub
		return null;
	}

}