bundles/org.eclipse.jst.ws.cxf.creation.core/src/org/eclipse/jst/ws/internal/cxf/creation/core/commands/CXFDeployCommand.java

/*******************************************************************************
 * Copyright (c) 2008 IONA Technologies PLC
 * 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:
 * IONA Technologies PLC - initial API and implementation
 *******************************************************************************/
package org.eclipse.jst.ws.internal.cxf.creation.core.commands;

import java.io.IOException;
import java.util.List;

import javax.wsdl.Definition;
import javax.wsdl.extensions.soap.SOAPAddress;

import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.runtime.CoreException;
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.jst.j2ee.model.IModelProvider;
import org.eclipse.jst.j2ee.model.ModelProviderManager;
import org.eclipse.jst.ws.internal.common.ServerUtils;
import org.eclipse.jst.ws.internal.cxf.core.model.CXFDataModel;
import org.eclipse.jst.ws.internal.cxf.core.utils.FileUtils;
import org.eclipse.jst.ws.internal.cxf.core.utils.SpringUtils;
import org.eclipse.jst.ws.internal.cxf.core.utils.WSDLUtils;
import org.eclipse.jst.ws.internal.cxf.creation.core.CXFCreationCorePlugin;
import org.eclipse.wst.common.frameworks.datamodel.AbstractDataModelOperation;
import org.eclipse.wst.server.core.IServer;
import org.eclipse.wst.server.core.ServerCore;
import org.eclipse.wst.ws.internal.wsrt.IWebService;
import org.eclipse.wst.ws.internal.wsrt.WebServiceInfo;

/**
 * 
 * @author sclarke
 */
@SuppressWarnings("restriction")
public class CXFDeployCommand extends AbstractDataModelOperation {
    private static final String CXF_SERVLET = "org.apache.cxf.transport.servlet.CXFServlet"; //$NON-NLS-1$
    private IProject project;
    private IWebService webService;
    private CXFDataModel model;
    
    public CXFDeployCommand(String projectName, IWebService webService) {
        this.project = FileUtils.getProject(projectName);
        this.webService = webService;
    }
    
    @Override
    public IStatus execute(IProgressMonitor monitor, IAdaptable info) throws ExecutionException {
        IStatus status = Status.OK_STATUS;
        
        try {
            WebServiceInfo webServiceInfo = webService.getWebServiceInfo();
            
            String serverInstanceId = webServiceInfo.getServerInstanceId();
            String serverFactoryId = webServiceInfo.getServerFactoryId();
            if (serverInstanceId != null) {
                IServer server = ServerCore.findServer(serverInstanceId);
                String webCobComponentURL = ServerUtils.getEncodedWebComponentURL(project, serverFactoryId,
                        server);
                
                String urlPattern = ""; //$NON-NLS-1$
                
                IModelProvider provider = ModelProviderManager.getModelProvider(project);
                Object modelProvider = provider.getModelObject();
                // jst.web 2.5
                if (modelProvider instanceof org.eclipse.jst.javaee.web.WebApp) {
                    org.eclipse.jst.javaee.web.WebApp javaeeWebApp = 
                        (org.eclipse.jst.javaee.web.WebApp) modelProvider;
                    
                    urlPattern = getURLPattern(javaeeWebApp);
                }
                
                // jst.web 2.4
                if (modelProvider instanceof org.eclipse.jst.j2ee.webapplication.WebApp) {
                    org.eclipse.jst.j2ee.webapplication.WebApp webApp = 
                        (org.eclipse.jst.j2ee.webapplication.WebApp) modelProvider;
                    
                    urlPattern = getURLPattern(webApp);
                }
                
                urlPattern = urlPattern.substring(0, urlPattern.lastIndexOf("/")); //$NON-NLS-1$
                String jaxwsEndpointAddress = SpringUtils.getEndpointAddress(project, 
                        model.getConfigId());
                
                String wsdlAddress = webCobComponentURL + urlPattern + jaxwsEndpointAddress;
                String wsdlURL = wsdlAddress + "?wsdl"; //$NON-NLS-1$
                
                webService.getWebServiceInfo().setWsdlURL(wsdlURL);
                
                Definition definition = model.getWsdlDefinition();
                SOAPAddress soapAddress = WSDLUtils.getEndpointAddress(definition);
                if (soapAddress != null) {
                   soapAddress.setLocationURI(wsdlAddress);
                   WSDLUtils.writeWSDL(model);
                }              
            }
        } catch (IOException ioe) {
            status = new Status(IStatus.ERROR, CXFCreationCorePlugin.PLUGIN_ID, ioe.getLocalizedMessage());
            CXFCreationCorePlugin.log(status);
        } catch (CoreException ce) {
            status = ce.getStatus();
            CXFCreationCorePlugin.log(status);
        }
    
        return status;
    }
    
    @SuppressWarnings("unchecked")
    private String getURLPattern(org.eclipse.jst.javaee.web.WebApp javaeeWebApp) {
        List<org.eclipse.jst.javaee.web.Servlet> servlets = javaeeWebApp.getServlets();
        for (org.eclipse.jst.javaee.web.Servlet servlet : servlets) {
            if (servlet.getServletClass().equals(CXF_SERVLET)) {
                List<org.eclipse.jst.javaee.web.ServletMapping> servletMappings = 
                    javaeeWebApp.getServletMappings();
                for (org.eclipse.jst.javaee.web.ServletMapping servletMapping : servletMappings) {
                    if (servletMapping.getServletName().equals(servlet.getServletName())) {
                        List<org.eclipse.jst.javaee.core.UrlPatternType> urlPatterns = 
                            servletMapping.getUrlPatterns();
                        if (urlPatterns.size() > 0) {
                            String value = ((org.eclipse.jst.javaee.core.UrlPatternType)
                                    urlPatterns.get(0)).getValue();
                            return value;
                        }
                    }
                }
            }
        }
        return ""; //$NON-NLS-1$
    }
    
    @SuppressWarnings("unchecked")
    private String getURLPattern(org.eclipse.jst.j2ee.webapplication.WebApp webApp) {
        List<org.eclipse.jst.j2ee.webapplication.Servlet> servlets = webApp.getServlets();
        for (org.eclipse.jst.j2ee.webapplication.Servlet servlet : servlets) {
            if (servlet.getServletClass().getJavaName().equals(CXF_SERVLET)) {
                List<org.eclipse.jst.j2ee.webapplication.ServletMapping> servletMappings = 
                    webApp.getServletMappings();
                for (org.eclipse.jst.j2ee.webapplication.ServletMapping servletMapping : servletMappings) {
                    if (servletMapping.getServlet().getServletName().equals(servlet.getServletName())) {
                        return servletMapping.getUrlPattern();
                    }
                }
            }
        }
        return ""; //$NON-NLS-1$
    }
    
    public void setCXFDataModel(CXFDataModel model) {
        this.model = model;
    }
    
    public String getClientComponentType() {
        return "template.cxf.core"; //$NON-NLS-1$
    }
}