/**

	 * This method takes a dd file and returns the document root of the file

	 * @param file - IFile which is a dd file

	 * @return DocumentRoot of the file.

	 */

	public static DocumentRoot getDocRoot(IFile file)

	{

		URI uri = URI.createPlatformResourceURI(file.getFullPath().toString());

		if (uri == null)

			return null;

		ensureMusePackagePresent();

		ResourceSet resourceSet = new ResourceSetImpl();

		resourceSet.getResourceFactoryRegistry().getExtensionToFactoryMap()

				.put(Resource.Factory.Registry.DEFAULT_EXTENSION,

						new MyDescriptorResourceFactoryImpl());

		resourceSet.getPackageRegistry().put(DescriptorPackage.eNS_URI,

				DescriptorPackage.eINSTANCE);

		Resource resource = resourceSet.getResource(uri, true);

		DocumentRoot root = (DocumentRoot) resource.getContents().get(0);

		return root;

	}

/*******************************************************************************

 * Copyright (c) 2007 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:

 * 	Balan Subramanian (bsubram@us.ibm.com)

 *  Andrew Eberbach (aeberbac@us.ibm.com)

 *     IBM Corporation - initial API and implementation

 *******************************************************************************/

package org.eclipse.tptp.wsdm.tooling.codegen.dd.internal;

import org.eclipse.core.resources.IFile;

import org.eclipse.jface.action.IAction;

import org.eclipse.jface.viewers.ISelection;

import org.eclipse.jface.viewers.StructuredSelection;

import org.eclipse.jface.wizard.WizardDialog;

import org.eclipse.tptp.wsdm.tooling.codegen.mrt.internal.NewProjectWizard;

import org.eclipse.ui.IObjectActionDelegate;

import org.eclipse.ui.IWorkbenchPart;

/**

 * This class provides the action to generate smashed WSDL and Java code from a

 * Deployment Descriptore file. 

 */

public class Generation implements IObjectActionDelegate {

	private IWorkbenchPart _targetPart;

	private StructuredSelection _selection;

	/**

	 * Shows the <code>NewProjectWizards</code> after wrapping the selected

	 * Deployment Descriptor file in a <code>DescriptorHelper</code>.

	 */

	public void run(IAction action) {

		//Get the file from the selected element

		//It should be the first element in the selection

		IFile ddFile = (IFile) _selection.getFirstElement();	

		//Create a code generation delegate which will

		//some of the heavy-lifting during code generation so

		//the UI thread can stay responsive

		DDCodeGenerationDelegate delegate = new DDCodeGenerationDelegate(ddFile);		

		//Open up the wizard dialog for code generation

		NewProjectWizard npw = new NewProjectWizard(delegate);

		WizardDialog wd = new WizardDialog(_targetPart.getSite().getShell(),npw);

		wd.open();

	}

	public void setActivePart(IAction action, IWorkbenchPart targetPart) {

		_targetPart = targetPart;

	}

	public void selectionChanged(IAction action, ISelection selection) {

		if (selection instanceof StructuredSelection) {

			_selection = (StructuredSelection) selection;

		}

	}

}

/*******************************************************************************

 * Copyright (c) 2007 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:

 * 	Andrew Eberbach (aeberbac@us.ibm.com)

 *     IBM Corporation - initial API and implementation

 *******************************************************************************/

package org.eclipse.tptp.wsdm.tooling.codegen.dd.internal;

import java.io.IOException;

import java.util.ArrayList;

import java.util.Collections;

import java.util.List;

import org.apache.ws.muse.descriptor.CapabilityType;

import org.apache.ws.muse.descriptor.DescriptorPackage;

import org.apache.ws.muse.descriptor.DocumentRoot;

import org.apache.ws.muse.descriptor.MuseType;

import org.apache.ws.muse.descriptor.ResourceTypeType;

import org.apache.ws.muse.descriptor.RootType;

import org.eclipse.core.resources.IFile;

import org.eclipse.core.runtime.SubProgressMonitor;

import org.eclipse.emf.common.util.URI;

import org.eclipse.emf.ecore.resource.Resource;

import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;

import org.eclipse.tptp.wsdm.tooling.codegen.mrt.internal.CodeGenerationDelegate;

import org.eclipse.tptp.wsdm.tooling.codegen.mrt.internal.DescriptorHelper;

import org.eclipse.tptp.wsdm.tooling.codegen.mrt.internal.SubscriptionManagerInspector;

import org.eclipse.tptp.wsdm.tooling.editor.dde.util.internal.DdeUtil;

import org.eclipse.tptp.wsdm.tooling.model.capabilities.Capability;

import org.eclipse.tptp.wsdm.tooling.model.manageableResourceType.ManageableResourceType;

import org.eclipse.tptp.wsdm.tooling.nls.messages.dde.internal.Messages;

import org.eclipse.tptp.wsdm.tooling.util.internal.MrtUtils;

import org.eclipse.tptp.wsdm.tooling.util.internal.MyDescriptorResourceFactoryImpl;

import org.eclipse.tptp.wsdm.tooling.util.internal.WsdmToolingLog;

/**

 * A Deployment Descriptor-specifc resolver. This will go through the Deployment

 * Descriptor and find all of the MCap files and resolve them and merge them

 * into WSDL documents.

 */

public class DDCodeGenerationDelegate implements CodeGenerationDelegate {

	/**

	 * The IFile of the unparsed descriptor file

	 */

	private IFile _descriptorFile;

	public DDCodeGenerationDelegate(IFile descriptorFile) {

		_descriptorFile = descriptorFile;

	}

	public DescriptorHelper run(SubProgressMonitor progressMonitor) throws Exception {		

		inspectForSubscriptionManager();

		return new DescriptorHelper(_descriptorFile);

	}

	private void inspectForSubscriptionManager()

	{

		DocumentRoot docRoot = DdeUtil.getDocRoot(_descriptorFile);

		SubscriptionManagerInspector inspector = new SubscriptionManagerInspector(docRoot);

		inspector.setMrtParentDirectory(_descriptorFile.getParent().getFullPath()

				.toString());

		RootType root = docRoot.getRoot();

		MuseType mt = root.getMuse();

		List rts = mt.getResourceType();

		if(rts == null || rts.size() == 0)

		{

			return;

		}

		List mrts = subManagerAdded(rts);// This method returns a list of mrts if subscriptionmanager is not already added. else null

		if(mrts == null || mrts.size() == 0) 

		{

			return;

		}

		for(int i = 0; i < mrts.size(); i++)

		{

			ManageableResourceType mrt = (ManageableResourceType) mrts.get(i);

			Capability[] capDefs = getMrtCapabilities(mrt);

			ResourceTypeType subMgr = inspector.inspect(capDefs);

			if(subMgr == null)

			{

				continue;

			}

			else

			{

				mt.getResourceType().add(subMgr);

				root.setMuse(mt);

				docRoot.setRoot(root);

				save(docRoot);

				break;

			}

		}

	}

	private Capability[] getMrtCapabilities(ManageableResourceType mrt) {

		Capability[] capList = null;

		try {

			capList = MrtUtils.getCapabilities(mrt);

		} catch (Exception e) {

			WsdmToolingLog

					.logError(

							org.eclipse.tptp.wsdm.tooling.nls.messages.mrt.internal.Messages.MRT_ERROR_CANNOT_TRACE_CAPS,

							e);

		}

		if (capList != null)

			return capList;

		return new Capability[0];

	}

	private void save(DocumentRoot root)

	{

		ensureMusePackagePresent();

		URI ddFileURI = URI.createPlatformResourceURI(_descriptorFile.getFullPath().toString());

		ResourceSetImpl rsImpl = new ResourceSetImpl();

		rsImpl.getResourceFactoryRegistry().getExtensionToFactoryMap()

		.put(Resource.Factory.Registry.DEFAULT_EXTENSION,

				new MyDescriptorResourceFactoryImpl());

		rsImpl.getPackageRegistry().put(DescriptorPackage.eNS_URI,

		DescriptorPackage.eINSTANCE);

		Resource ddRes = rsImpl.getResource(ddFileURI, true);

		ddRes.getContents().remove(0);

		ddRes.getContents().add(root);

		try {

			ddRes.save(Collections.EMPTY_MAP);

		} catch (IOException e) {

			e.printStackTrace();

		}

	}

	private void ensureMusePackagePresent()

	{

		// Ensure EMF knows about Muse descriptor

		if (DescriptorPackage.eINSTANCE == null)

		{

			WsdmToolingLog.logError(Messages.FAILED_TO_INITIALIZE_MUSE_DESCRIPTOR_ERROR_, new Throwable());

		}

	}

	/**

	 * This method takes alist of resourceTypes and checks whether subscriptionmanager is already present.<br>

	 * if so returns null. else converts the resourcetypes into mrts and returns the list.

	 * @param rts

	 * @return

	 */

	private List subManagerAdded(List rts)

	{

		List mrts = new ArrayList();

		if(rts == null || rts.size() == 0)

		{

			return null;

		}

		for(int i = 0; i < rts.size(); i++)

		{

			ResourceTypeType rt = (ResourceTypeType) rts.get(i);

			IFile mrtFile = DdeUtil.getIFileFromName(DdeUtil

					.getMrtFileName(rt.getWsdl().getWsdlFile()), "mrt"); 

			if(mrtFile == null)

			{

				continue;

			}

			ManageableResourceType mrt = DdeUtil.loadMrtFile(mrtFile);

			mrts.add(mrt);

			List capList = DdeUtil.getCapabilitiesFromMRT(mrt);

			if(capList == null || capList.size() == 0)

			{

				continue;

			}

			for(int j = 0; j < capList.size(); j++)

			{

				CapabilityType ctype = (CapabilityType) capList.get(j);

				if(ctype.getCapabilityUri().equals("http://docs.oasis-open.org/wsn/bw-2/SubscriptionManager"))

				{

					return null;

				}

			}

		}

		return mrts;

	}

}