bundles/org.eclipse.jst.ws.consumption.ui/src/org/eclipse/jst/ws/internal/consumption/ui/widgets/test/wssample/AddModuleDependenciesCommand.java - webservices/webtools.webservices - Git at Google

 /*******************************************************************************
  * Copyright (c) 2003, 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
  * -------- -------- -----------------------------------------------------------
  * 20060324   122799 rsinha@ca.ibm.com - Rupam Kuehner
  *******************************************************************************/
 package org.eclipse.jst.ws.internal.consumption.ui.widgets.test.wssample;

 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.NullProgressMonitor;
 import org.eclipse.core.runtime.Status;
 import org.eclipse.jdt.core.IClasspathEntry;
 import org.eclipse.jdt.core.IJavaProject;
 import org.eclipse.jdt.core.JavaCore;
 import org.eclipse.jdt.core.JavaModelException;
 import org.eclipse.jem.util.emf.workbench.ProjectUtilities;
 import org.eclipse.jst.j2ee.project.facet.IJavaProjectMigrationDataModelProperties;
 import org.eclipse.jst.j2ee.project.facet.JavaProjectMigrationDataModelProvider;
 import org.eclipse.jst.ws.internal.common.J2EEUtils;
 import org.eclipse.jst.ws.internal.consumption.command.common.AddModuleToServerCommand;
 import org.eclipse.jst.ws.internal.consumption.command.common.AssociateModuleWithEARCommand;
 import org.eclipse.jst.ws.internal.consumption.command.common.CreateModuleCommand;
 import org.eclipse.jst.ws.internal.consumption.ui.command.StartServerCommand;
 import org.eclipse.wst.common.componentcore.ComponentCore;
 import org.eclipse.wst.common.componentcore.datamodel.properties.ICreateReferenceComponentsDataModelProperties;
 import org.eclipse.wst.common.componentcore.internal.operation.CreateReferenceComponentsDataModelProvider;
 import org.eclipse.wst.common.componentcore.internal.util.ComponentUtilities;
 import org.eclipse.wst.common.componentcore.resources.IVirtualComponent;
 import org.eclipse.wst.common.environment.IEnvironment;
 import org.eclipse.wst.common.frameworks.datamodel.AbstractDataModelOperation;
 import org.eclipse.wst.common.frameworks.datamodel.DataModelFactory;
 import org.eclipse.wst.common.frameworks.datamodel.IDataModel;
 import org.eclipse.wst.ws.internal.wsrt.TestInfo;

 public class AddModuleDependenciesCommand extends AbstractDataModelOperation
 {

   private TestInfo testInfo;

   private IProject sampleIProject;
   private IProject clientIProject;
   private IProject sampleEARIProject;
   private String sampleEARProject;
   private String sampleEARModule;

   public AddModuleDependenciesCommand(TestInfo testInfo)
   {
     this.testInfo = testInfo;
   }

   /**
    * Execute WebServerDefaultingTask Set the default server name and id given a
    * deployable.
    */
   public IStatus execute( IProgressMonitor monitor, IAdaptable adaptable )
   {
     IEnvironment env = getEnvironment();
       createSampleProjects(env, monitor );
 	  clientIProject = ProjectUtilities.getProject(testInfo.getClientProject());

       if (clientIProject != null && !J2EEUtils.isWebComponent(clientIProject))
       {
         if (J2EEUtils.isJavaComponent(clientIProject)){
           addJavaProjectAsUtilityJar(clientIProject, sampleEARIProject,monitor);
           addJavaProjectAsUtilityJar(clientIProject, sampleIProject,monitor);
           try
           {
         	  addBuildPath(sampleIProject, clientIProject);
           } catch (JavaModelException jme)
           {
         	  //Do nothing in this catch block. The worst that
         	  //will happen is that the sample Web project
         	  //will show some compile errors. The sample will
         	  //likely still launch successfully on the server
         	  //and the user will be able to use it.
           }
 		}
 	  }

     return Status.OK_STATUS;
   }

   private void addJavaProjectAsUtilityJar(IProject javaProject, IProject earProject,IProgressMonitor monitor)
   {
 	  try {
 		  IDataModel migrationdm = DataModelFactory.createDataModel(new JavaProjectMigrationDataModelProvider());
 		  migrationdm.setProperty(IJavaProjectMigrationDataModelProperties.PROJECT_NAME, javaProject.getName());
 		  migrationdm.getDefaultOperation().execute(monitor, null);


 		  IDataModel refdm = DataModelFactory.createDataModel(new CreateReferenceComponentsDataModelProvider());
 		  List targetCompList = (List) refdm.getProperty(ICreateReferenceComponentsDataModelProperties.TARGET_COMPONENT_LIST);

 		  IVirtualComponent targetcomponent = ComponentCore.createComponent(javaProject);
 		  IVirtualComponent sourcecomponent = ComponentUtilities.getComponent(earProject.getName());
 		  targetCompList.add(targetcomponent);

 		  refdm.setProperty(ICreateReferenceComponentsDataModelProperties.SOURCE_COMPONENT,sourcecomponent );
 		  refdm.setProperty(ICreateReferenceComponentsDataModelProperties.TARGET_COMPONENT_LIST, targetCompList);
 		  refdm.setProperty(ICreateReferenceComponentsDataModelProperties.TARGET_COMPONENTS_DEPLOY_PATH,  "/WEB-INF/lib");
 		  refdm.getDefaultOperation().execute(monitor, null);
 	  }catch (Exception e) {

 	  }
   }


   private void addBuildPath(IProject referencingProject, IProject referencedProject) throws JavaModelException
   {
     IJavaProject javaProject = JavaCore.create(referencingProject);
     if (javaProject != null)
     {
       IClasspathEntry[] oldCp = javaProject.getRawClasspath();
 	  IClasspathEntry[] newCp = new IClasspathEntry[oldCp.length + 1];
 	  for (int i = 0; i < oldCp.length; i++)
         newCp[i] = oldCp[i];
 	  newCp[newCp.length - 1] = JavaCore.newProjectEntry(referencedProject.getFullPath());
 	  javaProject.setRawClasspath(newCp, new NullProgressMonitor());
     }
   }


   public static final String DEFAULT_SAMPLE_EAR_PROJECT_EXT = "EAR";

   private void createSampleProjects(IEnvironment env, IProgressMonitor monitor )
   {
 	  sampleIProject = ProjectUtilities.getProject(testInfo.getGenerationProject());
 	  clientIProject = ProjectUtilities.getProject(testInfo.getClientProject());

 	if (testInfo.getClientNeedEAR()) {
 	  if(testInfo.getClientEARProject() == null || testInfo.getClientEARProject().length() == 0){
 		  sampleEARProject = testInfo.getGenerationProject() + DEFAULT_SAMPLE_EAR_PROJECT_EXT;
 	  }
 	  else{
 	      sampleEARProject = testInfo.getClientEARProject();
 	  }

 	  sampleEARIProject  = ProjectUtilities.getProject(sampleEARProject);
 	  if(sampleEARIProject == null || !sampleEARIProject.isOpen()){

 	    CreateModuleCommand createEAR = new CreateModuleCommand();
 	    createEAR.setProjectName(sampleEARProject);
 	    createEAR.setModuleName(sampleEARModule);
 	    createEAR.setServerInstanceId(testInfo.getClientExistingServer().getId());
 	    createEAR.setServerFactoryId(testInfo.getClientServerTypeID());
 	    createEAR.setModuleType(CreateModuleCommand.EAR);
 	    createEAR.setJ2eeLevel(J2EEUtils.getJ2EEVersionAsString(clientIProject));
 		createEAR.setEnvironment( env );
 		  IStatus status = createEAR.execute( monitor, null );
 	    if (status.getSeverity()==Status.ERROR)
         {
           env.getStatusHandler().reportError(status);
         }

 		AddModuleToServerCommand modToServer = new AddModuleToServerCommand();
 		modToServer.setModule(sampleEARModule);
 		modToServer.setProject(sampleEARProject);
 		modToServer.setServerInstanceId(testInfo.getClientExistingServer().getId());
 		modToServer.setEnvironment( env );
 		status = modToServer.execute( monitor, null );
 		if (status.getSeverity()==Status.ERROR)
 	    {
 	      env.getStatusHandler().reportError(status);
 	    }

 	  }
 	}

 	  if (!sampleIProject.isOpen())
       {
         CreateModuleCommand createSample = new CreateModuleCommand();
         createSample.setProjectName(testInfo.getGenerationProject());
 		createSample.setModuleType(CreateModuleCommand.WEB);
 		createSample.setServerInstanceId(testInfo.getClientExistingServer().getId());
         createSample.setServerFactoryId(testInfo.getClientServerTypeID());
         createSample.setJ2eeLevel(J2EEUtils.getJ2EEVersionAsString(clientIProject));
         createSample.setEnvironment( env );
 		IStatus status = createSample.execute( monitor, null );

 	   if (testInfo.getClientNeedEAR()) {
 //		Associate the client module and service EAR
 	    AssociateModuleWithEARCommand associateCommand = new AssociateModuleWithEARCommand();
 	    associateCommand.setProject(testInfo.getGenerationProject());
 	    associateCommand.setEARProject(sampleEARProject);
 	    associateCommand.setEar(sampleEARModule);
 		associateCommand.setEnvironment( env );
 	    status = associateCommand.execute( monitor, null );
 	    if (status.getSeverity()==Status.ERROR)
 	    {
 	      env.getStatusHandler().reportError(status);
 	    }
 	   }
        else
        {
          // If no EAR is needed, the sample project needs to be added to
          // to the server directly.
         AddModuleToServerCommand addToServer = new AddModuleToServerCommand();
         addToServer.setModule(testInfo.getGenerationProject());
         addToServer.setProject(testInfo.getGenerationProject());
         addToServer.setServerInstanceId(testInfo.getClientExistingServer().getId());
         addToServer.setEnvironment( env );
         status = addToServer.execute( monitor, null );
         if (status.getSeverity()==Status.ERROR)
         {
           env.getStatusHandler().reportError(status);
         }
        }

 		StartServerCommand startServer = new StartServerCommand(true);
 		startServer.setServerInstanceId(testInfo.getClientExistingServer().getId());
 		startServer.setEnvironment( env );
 		status = startServer.execute( monitor, null );
 	    if (status.getSeverity()==Status.ERROR)
 	    {
 	      env.getStatusHandler().reportError(status);
 	    }

 	  }

   }


 }
	/*******************************************************************************
	* Copyright (c) 2003, 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
	* -------- -------- -----------------------------------------------------------
	* 20060324 122799 rsinha@ca.ibm.com - Rupam Kuehner
	*******************************************************************************/
	package org.eclipse.jst.ws.internal.consumption.ui.widgets.test.wssample;

	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.NullProgressMonitor;
	import org.eclipse.core.runtime.Status;
	import org.eclipse.jdt.core.IClasspathEntry;
	import org.eclipse.jdt.core.IJavaProject;
	import org.eclipse.jdt.core.JavaCore;
	import org.eclipse.jdt.core.JavaModelException;
	import org.eclipse.jem.util.emf.workbench.ProjectUtilities;
	import org.eclipse.jst.j2ee.project.facet.IJavaProjectMigrationDataModelProperties;
	import org.eclipse.jst.j2ee.project.facet.JavaProjectMigrationDataModelProvider;
	import org.eclipse.jst.ws.internal.common.J2EEUtils;
	import org.eclipse.jst.ws.internal.consumption.command.common.AddModuleToServerCommand;
	import org.eclipse.jst.ws.internal.consumption.command.common.AssociateModuleWithEARCommand;
	import org.eclipse.jst.ws.internal.consumption.command.common.CreateModuleCommand;
	import org.eclipse.jst.ws.internal.consumption.ui.command.StartServerCommand;
	import org.eclipse.wst.common.componentcore.ComponentCore;
	import org.eclipse.wst.common.componentcore.datamodel.properties.ICreateReferenceComponentsDataModelProperties;
	import org.eclipse.wst.common.componentcore.internal.operation.CreateReferenceComponentsDataModelProvider;
	import org.eclipse.wst.common.componentcore.internal.util.ComponentUtilities;
	import org.eclipse.wst.common.componentcore.resources.IVirtualComponent;
	import org.eclipse.wst.common.environment.IEnvironment;
	import org.eclipse.wst.common.frameworks.datamodel.AbstractDataModelOperation;
	import org.eclipse.wst.common.frameworks.datamodel.DataModelFactory;
	import org.eclipse.wst.common.frameworks.datamodel.IDataModel;
	import org.eclipse.wst.ws.internal.wsrt.TestInfo;

	public class AddModuleDependenciesCommand extends AbstractDataModelOperation
	{

	private TestInfo testInfo;

	private IProject sampleIProject;
	private IProject clientIProject;
	private IProject sampleEARIProject;
	private String sampleEARProject;
	private String sampleEARModule;

	public AddModuleDependenciesCommand(TestInfo testInfo)
	{
	this.testInfo = testInfo;
	}

	/**
	* Execute WebServerDefaultingTask Set the default server name and id given a
	* deployable.
	*/
	public IStatus execute( IProgressMonitor monitor, IAdaptable adaptable )
	{
	IEnvironment env = getEnvironment();
	createSampleProjects(env, monitor );
	clientIProject = ProjectUtilities.getProject(testInfo.getClientProject());

	if (clientIProject != null && !J2EEUtils.isWebComponent(clientIProject))
	{
	if (J2EEUtils.isJavaComponent(clientIProject)){
	addJavaProjectAsUtilityJar(clientIProject, sampleEARIProject,monitor);
	addJavaProjectAsUtilityJar(clientIProject, sampleIProject,monitor);
	try
	{
	addBuildPath(sampleIProject, clientIProject);
	} catch (JavaModelException jme)
	{
	//Do nothing in this catch block. The worst that
	//will happen is that the sample Web project
	//will show some compile errors. The sample will
	//likely still launch successfully on the server
	//and the user will be able to use it.
	}
	}
	}

	return Status.OK_STATUS;
	}

	private void addJavaProjectAsUtilityJar(IProject javaProject, IProject earProject,IProgressMonitor monitor)
	{
	try {
	IDataModel migrationdm = DataModelFactory.createDataModel(new JavaProjectMigrationDataModelProvider());
	migrationdm.setProperty(IJavaProjectMigrationDataModelProperties.PROJECT_NAME, javaProject.getName());
	migrationdm.getDefaultOperation().execute(monitor, null);


	IDataModel refdm = DataModelFactory.createDataModel(new CreateReferenceComponentsDataModelProvider());
	List targetCompList = (List) refdm.getProperty(ICreateReferenceComponentsDataModelProperties.TARGET_COMPONENT_LIST);

	IVirtualComponent targetcomponent = ComponentCore.createComponent(javaProject);
	IVirtualComponent sourcecomponent = ComponentUtilities.getComponent(earProject.getName());
	targetCompList.add(targetcomponent);

	refdm.setProperty(ICreateReferenceComponentsDataModelProperties.SOURCE_COMPONENT,sourcecomponent );
	refdm.setProperty(ICreateReferenceComponentsDataModelProperties.TARGET_COMPONENT_LIST, targetCompList);
	refdm.setProperty(ICreateReferenceComponentsDataModelProperties.TARGET_COMPONENTS_DEPLOY_PATH, "/WEB-INF/lib");
	refdm.getDefaultOperation().execute(monitor, null);
	}catch (Exception e) {

	}
	}


	private void addBuildPath(IProject referencingProject, IProject referencedProject) throws JavaModelException
	{
	IJavaProject javaProject = JavaCore.create(referencingProject);
	if (javaProject != null)
	{
	IClasspathEntry[] oldCp = javaProject.getRawClasspath();
	IClasspathEntry[] newCp = new IClasspathEntry[oldCp.length + 1];
	for (int i = 0; i < oldCp.length; i++)
	newCp[i] = oldCp[i];
	newCp[newCp.length - 1] = JavaCore.newProjectEntry(referencedProject.getFullPath());
	javaProject.setRawClasspath(newCp, new NullProgressMonitor());
	}
	}


	public static final String DEFAULT_SAMPLE_EAR_PROJECT_EXT = "EAR";

	private void createSampleProjects(IEnvironment env, IProgressMonitor monitor )
	{
	sampleIProject = ProjectUtilities.getProject(testInfo.getGenerationProject());
	clientIProject = ProjectUtilities.getProject(testInfo.getClientProject());

	if (testInfo.getClientNeedEAR()) {
	if(testInfo.getClientEARProject() == null \|\| testInfo.getClientEARProject().length() == 0){
	sampleEARProject = testInfo.getGenerationProject() + DEFAULT_SAMPLE_EAR_PROJECT_EXT;
	}
	else{
	sampleEARProject = testInfo.getClientEARProject();
	}

	sampleEARIProject = ProjectUtilities.getProject(sampleEARProject);
	if(sampleEARIProject == null \|\| !sampleEARIProject.isOpen()){

	CreateModuleCommand createEAR = new CreateModuleCommand();
	createEAR.setProjectName(sampleEARProject);
	createEAR.setModuleName(sampleEARModule);
	createEAR.setServerInstanceId(testInfo.getClientExistingServer().getId());
	createEAR.setServerFactoryId(testInfo.getClientServerTypeID());
	createEAR.setModuleType(CreateModuleCommand.EAR);
	createEAR.setJ2eeLevel(J2EEUtils.getJ2EEVersionAsString(clientIProject));
	createEAR.setEnvironment( env );
	IStatus status = createEAR.execute( monitor, null );
	if (status.getSeverity()==Status.ERROR)
	{
	env.getStatusHandler().reportError(status);
	}

	AddModuleToServerCommand modToServer = new AddModuleToServerCommand();
	modToServer.setModule(sampleEARModule);
	modToServer.setProject(sampleEARProject);
	modToServer.setServerInstanceId(testInfo.getClientExistingServer().getId());
	modToServer.setEnvironment( env );
	status = modToServer.execute( monitor, null );
	if (status.getSeverity()==Status.ERROR)
	{
	env.getStatusHandler().reportError(status);
	}

	}
	}

	if (!sampleIProject.isOpen())
	{
	CreateModuleCommand createSample = new CreateModuleCommand();
	createSample.setProjectName(testInfo.getGenerationProject());
	createSample.setModuleType(CreateModuleCommand.WEB);
	createSample.setServerInstanceId(testInfo.getClientExistingServer().getId());
	createSample.setServerFactoryId(testInfo.getClientServerTypeID());
	createSample.setJ2eeLevel(J2EEUtils.getJ2EEVersionAsString(clientIProject));
	createSample.setEnvironment( env );
	IStatus status = createSample.execute( monitor, null );

	if (testInfo.getClientNeedEAR()) {
	// Associate the client module and service EAR
	AssociateModuleWithEARCommand associateCommand = new AssociateModuleWithEARCommand();
	associateCommand.setProject(testInfo.getGenerationProject());
	associateCommand.setEARProject(sampleEARProject);
	associateCommand.setEar(sampleEARModule);
	associateCommand.setEnvironment( env );
	status = associateCommand.execute( monitor, null );
	if (status.getSeverity()==Status.ERROR)
	{
	env.getStatusHandler().reportError(status);
	}
	}
	else
	{
	// If no EAR is needed, the sample project needs to be added to
	// to the server directly.
	AddModuleToServerCommand addToServer = new AddModuleToServerCommand();
	addToServer.setModule(testInfo.getGenerationProject());
	addToServer.setProject(testInfo.getGenerationProject());
	addToServer.setServerInstanceId(testInfo.getClientExistingServer().getId());
	addToServer.setEnvironment( env );
	status = addToServer.execute( monitor, null );
	if (status.getSeverity()==Status.ERROR)
	{
	env.getStatusHandler().reportError(status);
	}
	}

	StartServerCommand startServer = new StartServerCommand(true);
	startServer.setServerInstanceId(testInfo.getClientExistingServer().getId());
	startServer.setEnvironment( env );
	status = startServer.execute( monitor, null );
	if (status.getSeverity()==Status.ERROR)
	{
	env.getStatusHandler().reportError(status);
	}

	}

	}








	}