/**
* 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.net.URL;
import java.util.Collections;
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.helpers.IOUtils;
import org.apache.cxf.jaxrs.ext.codegen.SourceGenerator;
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.util.URIParserUtil;
import org.apache.cxf.tools.wadlto.WadlToolConstants;
import org.apache.cxf.tools.wadlto.jaxb.CustomizationParser;
public class JAXRSContainer extends AbstractCXFToolContainer {
private static final Map<String, String> DEFAULT_TYPES_MAP;
private static final String TOOL_NAME = "wadl2java";
static {
// Should we have a common code which checks W3C EPR bindings in tools-common ?
DEFAULT_TYPES_MAP = Collections.singletonMap("org.w3._2005._08.addressing.EndpointReference",
"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<String> getArrayKeys() {
return new HashSet<String>();
}
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<InputSource> bindingFiles = parser.getJaxbBindings();
sg.setBindingFiles(bindingFiles);
sg.setCompilerArgs(parser.getCompilerArgs());
List<InputSource> schemaPackageFiles = parser.getSchemaPackageFiles();
sg.setSchemaPackageFiles(schemaPackageFiles);
sg.setSchemaPackageMap(context.getNamespacePackageMap());
// sg.setSchemaPackageName((String)context.get(WadlToolConstants.CFG_TOOLS_PACKAGENAME)));
// TODO: consider introducing an option too for users be able to
// supply custom type mappings
sg.setSchemaTypesMap(DEFAULT_TYPES_MAP);
if (context.optionSet(WadlToolConstants.CFG_GENERATE_ENUMS)) {
sg.setGenerateEnums(true);
}
// 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<String> generatedServiceClasses = sg.getGeneratedServiceClasses();
for (String className : generatedServiceClasses) {
int index = className.lastIndexOf(".");
collector.addServiceClassName(className.substring(0, index),
className.substring(index + 1),
className);
}
List<String> 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);
return IOUtils.toString(url.openStream());
} 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);
}
}
}
}