/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.geronimo.cxf.builder;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.jar.JarFile;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBElement;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Unmarshaller;
import javax.xml.transform.stream.StreamSource;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.apache.cxf.jaxws.javaee.HandlerChainsType;
import org.apache.cxf.jaxws.javaee.PortComponentType;
import org.apache.cxf.jaxws.javaee.ServiceImplBeanType;
import org.apache.cxf.jaxws.javaee.WebserviceDescriptionType;
import org.apache.cxf.jaxws.javaee.WebservicesType;
import org.apache.cxf.jaxws.support.JaxWsImplementorInfo;
import org.apache.geronimo.common.DeploymentException;
import org.apache.geronimo.cxf.pojo.POJOWebServiceContainerFactoryGBean;
import org.apache.geronimo.gbean.GBeanData;
import org.apache.geronimo.gbean.GBeanInfo;
import org.apache.geronimo.gbean.GBeanInfoBuilder;
import org.apache.geronimo.j2ee.deployment.Module;
import org.apache.geronimo.j2ee.deployment.WebServiceBuilder;
import org.apache.geronimo.j2ee.j2eeobjectnames.NameFactory;
import org.apache.geronimo.jaxws.JAXWSUtils;
import org.apache.geronimo.jaxws.PortInfo;
import org.apache.geronimo.jaxws.builder.JAXWSServiceBuilder;
import org.apache.geronimo.jaxws.builder.WARWebServiceFinder;
import org.apache.geronimo.jaxws.wsdl.WsdlGenerator;
import org.apache.geronimo.jaxws.wsdl.WsdlGeneratorOptions;
import org.apache.geronimo.kernel.repository.Environment;
public class CXFBuilder extends JAXWSServiceBuilder {
private static final Logger LOG = LoggerFactory.getLogger(CXFBuilder.class);
/**
* This property if enabled will cause the Sun wsgen tool to be used to
* generate the WSDL for servies without WSDL. By default CXF tooling
* will be used the generate the WSDL.
*/
private static final String USE_WSGEN_PROPERTY =
"org.apache.geronimo.cxf.use.wsgen";
protected Collection<WsdlGenerator> wsdlGenerators;
public CXFBuilder() {
super(null);
}
public CXFBuilder(Environment defaultEnvironment, Collection<WsdlGenerator> wsdlGenerators) {
super(defaultEnvironment);
this.wsdlGenerators = wsdlGenerators;
this.webServiceFinder = new WARWebServiceFinder();
}
protected GBeanInfo getContainerFactoryGBeanInfo() {
return POJOWebServiceContainerFactoryGBean.GBEAN_INFO;
}
protected Map<String, PortInfo> parseWebServiceDescriptor(InputStream in,
URL wsDDUrl,
JarFile moduleFile,
boolean isEJB,
Map correctedPortLocations)
throws DeploymentException {
LOG.debug("Parsing descriptor " + wsDDUrl);
Map<String, PortInfo> map = null;
try {
JAXBContext ctx = JAXBContext.newInstance(WebservicesType.class);
Unmarshaller unmarshaller = ctx.createUnmarshaller();
Object obj = unmarshaller.unmarshal(new StreamSource(in), WebservicesType.class);
if (obj instanceof JAXBElement) {
obj = ((JAXBElement) obj).getValue();
}
if (!(obj instanceof WebservicesType)) {
return map;
}
WebservicesType wst = (WebservicesType) obj;
for (WebserviceDescriptionType desc : wst.getWebserviceDescription()) {
String wsdlFile = null;
if (desc.getWsdlFile() != null) {
wsdlFile = getString(desc.getWsdlFile().getValue());
}
String serviceName = desc.getWebserviceDescriptionName().getValue();
for (PortComponentType port : desc.getPortComponent()) {
PortInfo portInfo = new PortInfo();
String serviceLink = null;
ServiceImplBeanType beanType = port.getServiceImplBean();
if (beanType.getEjbLink() != null) {
serviceLink = beanType.getEjbLink().getValue();
} else if (beanType.getServletLink().getValue() != null) {
serviceLink = beanType.getServletLink().getValue();
}
portInfo.setServiceLink(serviceLink);
if (port.getServiceEndpointInterface() != null) {
String sei = port.getServiceEndpointInterface().getValue();
portInfo.setServiceEndpointInterfaceName(sei);
}
String portName = port.getPortComponentName().getValue();
portInfo.setPortName(portName);
portInfo.setProtocolBinding(port.getProtocolBinding());
portInfo.setServiceName(serviceName);
portInfo.setWsdlFile(wsdlFile);
if (port.getEnableMtom() != null) {
portInfo.setEnableMTOM(port.getEnableMtom().isValue());
}
portInfo.setHandlers(HandlerChainsType.class, port.getHandlerChains());
if (port.getWsdlPort() != null) {
portInfo.setWsdlPort(port.getWsdlPort().getValue());
}
if (port.getWsdlService() != null) {
portInfo.setWsdlService(port.getWsdlService().getValue());
}
String location = (String) correctedPortLocations.get(serviceLink);
portInfo.setLocation(location);
if (map == null) {
map = new HashMap<String, PortInfo>();
}
map.put(serviceLink, portInfo);
}
}
return map;
} catch (FileNotFoundException e) {
return Collections.emptyMap();
} catch (IOException ex) {
throw new DeploymentException("Unable to read " + wsDDUrl, ex);
} catch (JAXBException ex) {
throw new DeploymentException("Unable to parse " + wsDDUrl, ex);
} catch (Exception ex) {
throw new DeploymentException("Unknown deployment error", ex);
} finally {
try {
in.close();
} catch (IOException e) {
// ignore
}
}
}
private static String getString(String in) {
if (in != null) {
in = in.trim();
if (in.length() == 0) {
return null;
}
}
return in;
}
@Override
protected void initialize(GBeanData targetGBean, Class serviceClass, PortInfo portInfo, Module module)
throws DeploymentException {
if (Boolean.getBoolean(USE_WSGEN_PROPERTY)) {
generateWSDL(serviceClass, portInfo, module);
}
}
protected WsdlGenerator getWsdlGenerator() throws DeploymentException {
if (this.wsdlGenerators == null || this.wsdlGenerators.isEmpty()) {
throw new DeploymentException("Wsdl generator not found");
} else {
return this.wsdlGenerators.iterator().next();
}
}
private void generateWSDL(Class serviceClass, PortInfo portInfo, Module module)
throws DeploymentException {
String serviceName = (portInfo.getServiceName() == null ? serviceClass.getName() : portInfo.getServiceName());
if (isWsdlSet(portInfo, serviceClass)) {
LOG.debug("Service " + serviceName + " has WSDL.");
return;
}
if (isHTTPBinding(portInfo, serviceClass)) {
LOG.debug("Service " + serviceName + " has HTTPBinding.");
return;
}
if (JAXWSUtils.isWebServiceProvider(serviceClass)) {
throw new DeploymentException("WSDL must be specified for @WebServiceProvider service " + serviceName);
}
LOG.debug("Service " + serviceName + " does not have WSDL. Generating WSDL...");
WsdlGenerator wsdlGenerator = getWsdlGenerator();
WsdlGeneratorOptions options = new WsdlGeneratorOptions();
options.setSAAJ(WsdlGeneratorOptions.SAAJ.SUN);
JaxWsImplementorInfo serviceInfo = new JaxWsImplementorInfo(serviceClass);
// set wsdl service
if (portInfo.getWsdlService() == null) {
options.setWsdlService(serviceInfo.getServiceName());
} else {
options.setWsdlService(portInfo.getWsdlService());
}
// set wsdl port
if (portInfo.getWsdlPort() != null) {
options.setWsdlPort(portInfo.getWsdlPort());
}
String wsdlFile = wsdlGenerator.generateWsdl(module, serviceClass.getName(), module.getEarContext(), options);
portInfo.setWsdlFile(wsdlFile);
LOG.debug("Generated " + wsdlFile + " for service " + serviceName);
}
public static final GBeanInfo GBEAN_INFO;
static {
GBeanInfoBuilder infoBuilder = GBeanInfoBuilder.createStatic(CXFBuilder.class, NameFactory.MODULE_BUILDER);
infoBuilder.addInterface(WebServiceBuilder.class);
infoBuilder.addAttribute("defaultEnvironment", Environment.class, true, true);
infoBuilder.addReference("WsdlGenerator", WsdlGenerator.class, GBeanInfoBuilder.DEFAULT_J2EE_TYPE);
infoBuilder.setConstructor(new String[]{"defaultEnvironment", "WsdlGenerator"});
GBEAN_INFO = infoBuilder.getBeanInfo();
}
public static GBeanInfo getGBeanInfo() {
return GBEAN_INFO;
}
}