/**
* 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.axis2.builder;
import org.apache.axis2.jaxws.javaee.HandlerChainsType;
import org.apache.axis2.jaxws.javaee.PortComponentType;
import org.apache.axis2.jaxws.javaee.ServiceImplBeanType;
import org.apache.axis2.jaxws.javaee.WebserviceDescriptionType;
import org.apache.axis2.jaxws.javaee.WebservicesType;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.geronimo.axis2.Axis2WebServiceContainerFactoryGBean;
import org.apache.geronimo.axis2.client.Axis2ServiceReference;
import org.apache.geronimo.common.DeploymentException;
import org.apache.geronimo.deployment.DeploymentContext;
import org.apache.geronimo.deployment.util.DeploymentUtil;
import org.apache.geronimo.gbean.AbstractName;
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.WebModule;
import org.apache.geronimo.j2ee.deployment.WebServiceBuilder;
import org.apache.geronimo.j2ee.j2eeobjectnames.NameFactory;
import org.apache.geronimo.jaxws.PortInfo;
import org.apache.geronimo.jaxws.builder.EndpointInfoBuilder;
import org.apache.geronimo.jaxws.builder.JAXWSServiceBuilder;
import org.apache.geronimo.jaxws.client.EndpointInfo;
import org.apache.geronimo.kernel.GBeanNotFoundException;
import org.apache.geronimo.kernel.repository.Environment;
import org.apache.geronimo.xbeans.geronimo.naming.GerServiceRefType;
import org.apache.geronimo.xbeans.javaee.ServiceRefHandlerChainsType;
import javax.wsdl.Definition;
import javax.wsdl.factory.WSDLFactory;
import javax.wsdl.xml.WSDLReader;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBElement;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Unmarshaller;
import javax.xml.namespace.QName;
import javax.xml.transform.stream.StreamSource;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.StringWriter;
import java.net.URI;
import java.net.URL;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.jar.JarFile;
public class Axis2Builder extends JAXWSServiceBuilder {
private static final Log log = LogFactory.getLog(Axis2Builder.class);
private Definition wsdlDefinition = null;
public Axis2Builder(Environment defaultEnviroment) {
super(defaultEnviroment);
}
public Axis2Builder(){
super(null);
}
protected GBeanInfo getContainerFactoryGBeanInfo() {
return Axis2WebServiceContainerFactoryGBean.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()) {
org.apache.geronimo.axis2.PortInfo portInfo = new org.apache.geronimo.axis2.PortInfo();
if(wsdlFile != null && !wsdlFile.equals("")){
URL wsdlURL = DeploymentUtil.createJarURL(moduleFile, wsdlFile);
InputStream wsdlStream = wsdlURL.openStream();
if(wsdlStream == null){
throw new DeploymentException("unable to read descriptor "+wsdlURL);
}else {
try {
WSDLFactory factory = WSDLFactory.newInstance();
WSDLReader reader = factory.newWSDLReader();
reader.setFeature("javax.wsdl.importDocuments", true);
reader.setFeature("javax.wsdl.verbose", false);
wsdlDefinition = reader.readWSDL(wsdlURL.toString());
portInfo.setWsdlDefinition(wsdlDefinition);
wsdlStream.close();
} catch (RuntimeException e) {
throw new DeploymentException("invalid WSDL provided "+wsdlURL);
}
}
}
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.EMPTY_MAP;
} 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
}
}
}
public boolean configurePOJO(GBeanData targetGBean,
String servletName,
Module module,
String seiClassName,
DeploymentContext context)
throws DeploymentException {
boolean status = super.configurePOJO(targetGBean, servletName, module, seiClassName, context);
if(!status) {
return false;
}
//change the URL
Map sharedContext = ((WebModule) module).getSharedContext();
String contextRoot = ((WebModule) module).getContextRoot();
Map portInfoMap = (Map) sharedContext.get(getKey());
if(portInfoMap != null && portInfoMap.get(servletName) != null){
org.apache.geronimo.jaxws.PortInfo portInfo = (org.apache.geronimo.jaxws.PortInfo) portInfoMap.get(servletName);
processURLPattern(contextRoot, portInfo);
}
return status;
}
public Object createService(Class serviceInterface,
Class serviceReference,
URI wsdlURI,
QName serviceQName,
Map<Class, String> portComponentRefMap,
ServiceRefHandlerChainsType handlerChains,
GerServiceRefType serviceRefType,
Module module,
ClassLoader cl) throws DeploymentException {
EndpointInfoBuilder builder = new EndpointInfoBuilder(serviceInterface,
serviceRefType, portComponentRefMap, module.getModuleFile(),
wsdlURI, serviceQName);
builder.build();
wsdlURI = builder.getWsdlURI();
serviceQName = builder.getServiceQName();
Map<Object, EndpointInfo> seiInfoMap = builder.getEndpointInfo();
String handlerChainsXML = null;
try {
handlerChainsXML = getHanderChainAsString(handlerChains);
} catch (IOException e) {
// this should not happen
log.warn("Failed to serialize handler chains", e);
}
String serviceReferenceName = (serviceReference == null) ? null : serviceReference.getName();
return new Axis2ServiceReference(serviceInterface.getName(), serviceReferenceName, wsdlURI,
serviceQName, module.getModuleName(), handlerChainsXML, seiInfoMap);
}
private static String getHanderChainAsString(ServiceRefHandlerChainsType handlerChains)
throws IOException {
String xml = null;
if (handlerChains != null) {
StringWriter w = new StringWriter();
handlerChains.save(w);
xml = w.toString();
}
return xml;
}
private static String getString(String in) {
if (in != null) {
in = in.trim();
if (in.length() == 0) {
return null;
}
}
return in;
}
private void processURLPattern(String contextRoot, org.apache.geronimo.jaxws.PortInfo portInfo) throws DeploymentException {
//if the user specifies a url-pattern, set it here.
String oldup = portInfo.getLocation();
if (oldup == null || oldup.length() == 0) {
//if we cannot grab a valid urlpattern, default it to the port-component-name.
oldup = portInfo.getPortName();
} else {
int i = oldup.indexOf(contextRoot);
oldup = oldup.substring(i + contextRoot.length() + 1);
oldup = oldup.trim();
if (oldup.indexOf("*") > 0) {
//uncomment this before we fix this issue. workarond by assuming * is at the end.
//throw new DeploymentException("Per JSR 109, the url-pattern should not contain an asterisk.");
oldup = oldup.substring(0, oldup.length() - 1);
}
//trim the forward slashes at the beginning or end.
if (oldup.substring(0, 1).equalsIgnoreCase("/")) {
oldup = oldup.substring(1);
}
if (oldup.substring(oldup.length()-1).equalsIgnoreCase("/")) {
oldup = oldup.substring(0, oldup.length() - 1);
}
}
portInfo.setLocation(oldup);
}
public static final GBeanInfo GBEAN_INFO;
static {
GBeanInfoBuilder infoBuilder = GBeanInfoBuilder.createStatic(Axis2Builder.class, NameFactory.MODULE_BUILDER);
infoBuilder.addInterface(WebServiceBuilder.class);
infoBuilder.addAttribute("defaultEnvironment", Environment.class, true, true);
infoBuilder.setConstructor(new String[]{"defaultEnvironment"});
GBEAN_INFO = infoBuilder.getBeanInfo();
}
public static GBeanInfo getGBeanInfo() {
return GBEAN_INFO;
}
}