org.apache.cxf.tools.wadlto.jaxrs.JAXRSContainer Maven / Gradle / Ivy
/**
* 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.cxf.tools.wadlto.jaxrs;
import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Reader;
import java.net.URL;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.xml.sax.InputSource;
import org.apache.cxf.common.util.URIParserUtil;
import org.apache.cxf.helpers.IOUtils;
import org.apache.cxf.tools.common.AbstractCXFToolContainer;
import org.apache.cxf.tools.common.ClassUtils;
import org.apache.cxf.tools.common.ToolException;
import org.apache.cxf.tools.common.toolspec.ToolSpec;
import org.apache.cxf.tools.common.toolspec.parser.BadUsageException;
import org.apache.cxf.tools.util.ClassCollector;
import org.apache.cxf.tools.wadlto.WadlToolConstants;
import org.apache.cxf.tools.wadlto.jaxb.CustomizationParser;
public class JAXRSContainer extends AbstractCXFToolContainer {
private static final Map DEFAULT_JAVA_TYPE_MAP;
private static final String TOOL_NAME = "wadl2java";
private static final String EPR_TYPE_KEY = "org.w3._2005._08.addressing.EndpointReference";
static {
DEFAULT_JAVA_TYPE_MAP = Collections.singletonMap(EPR_TYPE_KEY,
"javax.xml.ws.wsaddressing.W3CEndpointReference");
}
public JAXRSContainer(ToolSpec toolspec) throws Exception {
super(TOOL_NAME, toolspec);
}
public void execute() throws ToolException {
if (hasInfoOption()) {
return;
}
buildToolContext();
processWadl();
}
public void execute(boolean exitOnFinish) throws ToolException {
try {
if (getArgument() != null) {
super.execute(exitOnFinish);
}
execute();
} catch (ToolException ex) {
if (ex.getCause() instanceof BadUsageException) {
printUsageException(TOOL_NAME, (BadUsageException)ex.getCause());
}
throw ex;
} catch (Exception ex) {
throw new ToolException(ex);
} finally {
tearDown();
}
}
public void buildToolContext() {
getContext();
context.addParameters(getParametersMap(getArrayKeys()));
if (context.get(WadlToolConstants.CFG_OUTPUTDIR) == null) {
context.put(WadlToolConstants.CFG_OUTPUTDIR, ".");
}
setPackageAndNamespaces();
}
public Set getArrayKeys() {
Set set = new HashSet();
set.add(WadlToolConstants.CFG_BINDING);
set.add(WadlToolConstants.CFG_SCHEMA_PACKAGENAME);
set.add(WadlToolConstants.CFG_SCHEMA_TYPE_MAP);
set.add(WadlToolConstants.CFG_MEDIA_TYPE_MAP);
return set;
}
private void processWadl() {
File outDir = new File((String)context.get(WadlToolConstants.CFG_OUTPUTDIR));
String wadlURL = getAbsoluteWadlURL();
String wadl = readWadl(wadlURL);
SourceGenerator sg = new SourceGenerator();
sg.setBus(getBus());
boolean generateImpl = context.optionSet(WadlToolConstants.CFG_IMPL);
sg.setGenerateImplementation(generateImpl);
if (generateImpl) {
sg.setGenerateInterfaces(context.optionSet(WadlToolConstants.CFG_INTERFACE));
}
sg.setPackageName((String)context.get(WadlToolConstants.CFG_PACKAGENAME));
sg.setResourceName((String)context.get(WadlToolConstants.CFG_RESOURCENAME));
String wadlNs = (String)context.get(WadlToolConstants.CFG_WADL_NAMESPACE);
if (wadlNs != null) {
sg.setWadlNamespace(wadlNs);
}
sg.setSupportMultipleXmlReps(context.optionSet(WadlToolConstants.CFG_MULTIPLE_XML_REPS));
// set the base path
sg.setWadlPath(wadlURL);
CustomizationParser parser = new CustomizationParser(context);
parser.parse(context);
List bindingFiles = parser.getJaxbBindings();
sg.setBindingFiles(bindingFiles);
sg.setCompilerArgs(parser.getCompilerArgs());
List schemaPackageFiles = parser.getSchemaPackageFiles();
sg.setSchemaPackageFiles(schemaPackageFiles);
sg.setSchemaPackageMap(context.getNamespacePackageMap());
sg.setJavaTypeMap(DEFAULT_JAVA_TYPE_MAP);
sg.setSchemaTypeMap(getSchemaTypeMap());
sg.setMediaTypeMap(getMediaTypeMap());
sg.setGenerateEnums(context.optionSet(WadlToolConstants.CFG_GENERATE_ENUMS));
sg.setInheritResourceParams(context.optionSet(WadlToolConstants.CFG_INHERIT_PARAMS));
sg.setSkipSchemaGeneration(context.optionSet(WadlToolConstants.CFG_NO_TYPES));
boolean noVoidForEmptyResponses = context.optionSet(WadlToolConstants.CFG_NO_VOID_FOR_EMPTY_RESPONSES);
if (noVoidForEmptyResponses) {
sg.setUseVoidForEmptyResponses(false);
}
// generate
String codeType = context.optionSet(WadlToolConstants.CFG_TYPES)
? SourceGenerator.CODE_TYPE_GRAMMAR : SourceGenerator.CODE_TYPE_PROXY;
sg.generateSource(wadl, outDir, codeType);
// compile
if (context.optionSet(WadlToolConstants.CFG_COMPILE)) {
ClassCollector collector = createClassCollector();
List generatedServiceClasses = sg.getGeneratedServiceClasses();
for (String className : generatedServiceClasses) {
int index = className.lastIndexOf(".");
collector.addServiceClassName(className.substring(0, index),
className.substring(index + 1),
className);
}
List generatedTypeClasses = sg.getGeneratedTypeClasses();
for (String className : generatedTypeClasses) {
int index = className.lastIndexOf(".");
collector.addTypesClassName(className.substring(0, index),
className.substring(index + 1),
className);
}
context.put(ClassCollector.class, collector);
new ClassUtils().compile(context);
}
}
protected String readWadl(String wadlURI) {
try {
URL url = new URL(wadlURI);
Reader reader = new InputStreamReader(url.openStream(), "UTF-8");
return IOUtils.toString(reader);
} catch (IOException e) {
throw new ToolException(e);
}
}
protected String getAbsoluteWadlURL() {
String wadlURL = (String)context.get(WadlToolConstants.CFG_WADLURL);
String absoluteWadlURL = URIParserUtil.getAbsoluteURI(wadlURL);
context.put(WadlToolConstants.CFG_WADLURL, absoluteWadlURL);
return absoluteWadlURL;
}
//TODO: this belongs to JAXB Databinding, should we just reuse
// org.apache.cxf.tools.wsdlto.databinding.jaxb ?
private void setPackageAndNamespaces() {
String[] schemaPackageNamespaces = new String[]{};
Object value = context.get(WadlToolConstants.CFG_SCHEMA_PACKAGENAME);
if (value != null) {
schemaPackageNamespaces = value instanceof String ? new String[]{(String)value}
: (String[])value;
}
for (int i = 0; i < schemaPackageNamespaces.length; i++) {
int pos = schemaPackageNamespaces[i].indexOf("=");
String packagename = schemaPackageNamespaces[i];
if (pos != -1) {
String ns = schemaPackageNamespaces[i].substring(0, pos);
packagename = schemaPackageNamespaces[i].substring(pos + 1);
context.addNamespacePackageMap(ns, packagename);
} else {
// this is the default schema package name
// if CFG_PACKAGENAME is set then it's only used for JAX-RS resource
// classes
context.put(WadlToolConstants.CFG_SCHEMA_PACKAGENAME, packagename);
}
}
}
private Map getSchemaTypeMap() {
return getMap(WadlToolConstants.CFG_SCHEMA_TYPE_MAP);
}
private Map getMediaTypeMap() {
return getMap(WadlToolConstants.CFG_MEDIA_TYPE_MAP);
}
private Map getMap(String parameterName) {
String[] typeToClasses = new String[]{};
Object value = context.get(parameterName);
if (value != null) {
typeToClasses = value instanceof String ? new String[]{(String)value}
: (String[])value;
}
Map typeMap = new HashMap();
for (int i = 0; i < typeToClasses.length; i++) {
int pos = typeToClasses[i].indexOf("=");
if (pos != -1) {
String type = typeToClasses[i].substring(0, pos);
if (type.contains("%3D")) {
type = type.replace("%3D", "=");
}
String clsName = typeToClasses[i].substring(pos + 1);
typeMap.put(type, clsName);
}
}
return typeMap;
}
}