| /* |
| * Copyright 2004,2005 The Apache Software Foundation. |
| * |
| * Licensed 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.axis2.tool.ant; |
| |
| import org.apache.axis2.util.CommandLineOption; |
| import org.apache.axis2.util.CommandLineOptionConstants; |
| import org.apache.axis2.util.CommandLineOptionParser; |
| import org.apache.axis2.util.URLProcessor; |
| import org.apache.axis2.wsdl.codegen.CodeGenerationEngine; |
| import org.apache.axis2.wsdl.util.ConfigPropertyFileLoader; |
| import org.apache.tools.ant.AntClassLoader; |
| import org.apache.tools.ant.BuildException; |
| import org.apache.tools.ant.Task; |
| import org.apache.tools.ant.types.Path; |
| import org.apache.tools.ant.types.Reference; |
| |
| import java.util.HashMap; |
| import java.util.Map; |
| |
| public class AntCodegenTask extends Task { |
| |
| private String wsdlFileName = null; |
| private String output = "."; |
| private String language = ConfigPropertyFileLoader.getDefaultLanguage(); |
| private String packageName = URLProcessor.DEFAULT_PACKAGE; |
| private String databindingName=ConfigPropertyFileLoader.getDefaultDBFrameworkName(); |
| private String portName=null; |
| private String serviceName=null; |
| |
| private boolean asyncOnly = false; |
| private boolean syncOnly = false; |
| private boolean serverSide = false; |
| private boolean testcase = false; |
| private boolean generateServiceXml = false; |
| private boolean generateAllClasses = false; |
| private boolean unpackClasses = false; |
| private boolean serverSideInterface = false; |
| |
| private String repositoryPath = null; |
| private String externalMapping = null; |
| private String wsdlVersion = null; |
| private String targetSourceFolderLocation = null; |
| private String targetResourcesFolderLocation = null; |
| private boolean unwrap = false; |
| |
| private String namespaceToPackages = null; |
| |
| private Path classpath; |
| |
| |
| |
| |
| /** |
| * |
| */ |
| public AntCodegenTask() { |
| super(); |
| } |
| |
| |
| /** |
| * Sets the classpath. |
| * @return Returns Path. |
| */ |
| public Path createClasspath() { |
| if (classpath == null) { |
| classpath = new Path(getProject()); |
| classpath = classpath.concatSystemClasspath(); |
| } |
| return classpath.createPath(); |
| } |
| |
| /** |
| * Set the reference to an optional classpath |
| * |
| * @param r the id of the Ant path instance to act as the classpath |
| */ |
| public void setClasspathRef(Reference r) { |
| createClasspath().setRefid(r); |
| } |
| |
| /** |
| * |
| * @return |
| */ |
| public boolean isServerSideInterface() { |
| return serverSideInterface; |
| } |
| |
| /** |
| * |
| * @param serverSideInterface |
| */ |
| public void setServerSideInterface(boolean serverSideInterface) { |
| this.serverSideInterface = serverSideInterface; |
| } |
| |
| /** |
| * Fills the option map. This map is passed onto |
| * the code generation API to generate the code. |
| */ |
| private Map fillOptionMap() { |
| Map optionMap = new HashMap(); |
| |
| //////////////////////////////////////////////////////////////// |
| //WSDL file name |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.WSDL_LOCATION_URI_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.WSDL_LOCATION_URI_OPTION, |
| getStringArray(wsdlFileName))); |
| |
| //WSDL version |
| if (wsdlVersion != null) { |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.WSDL_VERSION_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.WSDL_VERSION_OPTION, |
| getStringArray(wsdlVersion))); |
| } |
| |
| // repository path |
| if (repositoryPath != null) { |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.REPOSITORY_PATH_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.REPOSITORY_PATH_OPTION, |
| getStringArray(repositoryPath))); |
| } |
| |
| // external mapping |
| if (externalMapping != null) { |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.EXTERNAL_MAPPING_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.EXTERNAL_MAPPING_OPTION, |
| getStringArray(externalMapping))); |
| } |
| |
| // target source folder location |
| if (targetSourceFolderLocation != null) { |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.SOURCE_FOLDER_NAME_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.SOURCE_FOLDER_NAME_OPTION, |
| getStringArray(targetSourceFolderLocation))); |
| } |
| |
| // target source folder location |
| if (targetResourcesFolderLocation != null) { |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.RESOURCE_FOLDER_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.RESOURCE_FOLDER_OPTION, |
| getStringArray(targetResourcesFolderLocation))); |
| } |
| |
| // target source folder location |
| if (unwrap) { |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.UNWRAP_PARAMETERS, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.UNWRAP_PARAMETERS, |
| new String[0])); |
| } |
| |
| //output location |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.OUTPUT_LOCATION_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.OUTPUT_LOCATION_OPTION, |
| getStringArray(output))); |
| ////////////////////////////////////////////////////////////////// |
| // Databinding type |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.DATA_BINDING_TYPE_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.DATA_BINDING_TYPE_OPTION, |
| getStringArray(databindingName))); |
| |
| // Async only option - forcing to generate async methods only |
| if (asyncOnly) { |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.CODEGEN_ASYNC_ONLY_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.CODEGEN_ASYNC_ONLY_OPTION, |
| new String[0])); |
| } |
| // Sync only option - forcing to generate Sync methods only |
| if (syncOnly) { |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.CODEGEN_SYNC_ONLY_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.CODEGEN_SYNC_ONLY_OPTION, |
| new String[0])); |
| } |
| |
| //Package |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.PACKAGE_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.PACKAGE_OPTION, |
| getStringArray(packageName))); |
| |
| //stub language |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.STUB_LANGUAGE_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.STUB_LANGUAGE_OPTION, |
| getStringArray(language))); |
| |
| |
| //server side and generate services.xml options |
| if (serverSide) { |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.SERVER_SIDE_CODE_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.SERVER_SIDE_CODE_OPTION, |
| new String[0])); |
| |
| //services XML generation - effective only when specified as the server side |
| if (generateServiceXml) { |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants |
| .GENERATE_SERVICE_DESCRIPTION_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants |
| .GENERATE_SERVICE_DESCRIPTION_OPTION, |
| new String[0])); |
| } |
| //generate all option - Only valid when generating serverside code |
| if (generateAllClasses) { |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.GENERATE_ALL_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.GENERATE_ALL_OPTION, |
| new String[0])); |
| } |
| |
| } |
| |
| //generate the test case |
| if (testcase) { |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.GENERATE_TEST_CASE_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.GENERATE_TEST_CASE_OPTION, |
| new String[0])); |
| } |
| |
| //Unwrap classes option - this determines whether the generated classes are inside the stub/MR |
| //or gets generates as seperate classes |
| if (unpackClasses) { |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.UNPACK_CLASSES_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.UNPACK_CLASSES_OPTION, |
| new String[0])); |
| } |
| |
| //server side interface option |
| if (serverSideInterface) { |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.SERVER_SIDE_INTERFACE_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.SERVER_SIDE_INTERFACE_OPTION, |
| new String[0])); |
| } |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.SERVICE_NAME_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.SERVICE_NAME_OPTION, |
| new String[]{serviceName})); |
| |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.PORT_NAME_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.PORT_NAME_OPTION, |
| new String[]{portName})); |
| // set the namespaces |
| optionMap.put( |
| CommandLineOptionConstants.WSDL2JavaConstants.NAME_SPACE_TO_PACKAGE_OPTION, |
| new CommandLineOption( |
| CommandLineOptionConstants.WSDL2JavaConstants.NAME_SPACE_TO_PACKAGE_OPTION, |
| new String[]{namespaceToPackages})); |
| |
| return optionMap; |
| } |
| |
| /** |
| * Utility method to convert a string into a single item string[] |
| * @param value |
| * @return Returns String[]. |
| */ |
| private String[] getStringArray(String value) { |
| String[] values = new String[1]; |
| values[0] = value; |
| return values; |
| } |
| |
| public void execute() throws BuildException { |
| try { |
| /* |
| * This needs the ClassLoader we use to load the task |
| * have all the dependancies set, hope that |
| * is ok for now |
| */ |
| |
| AntClassLoader cl = new AntClassLoader( |
| getClass().getClassLoader(), |
| getProject(), |
| classpath == null ? createClasspath() : classpath, |
| false); |
| |
| Thread.currentThread().setContextClassLoader(cl); |
| cl.addPathElement(output); |
| |
| Map commandLineOptions = this.fillOptionMap(); |
| CommandLineOptionParser parser = |
| new CommandLineOptionParser(commandLineOptions); |
| new CodeGenerationEngine(parser).generate(); |
| } catch (Throwable e) { |
| throw new BuildException(e); |
| } |
| |
| } |
| |
| public void setPortName(String portName) { |
| this.portName = portName; |
| } |
| |
| public void setServiceName(String serviceName) { |
| this.serviceName = serviceName; |
| } |
| |
| public void setGenerateAllClasses(boolean generateAllClasses) { |
| this.generateAllClasses = generateAllClasses; |
| } |
| |
| public void setUnpackClasses(boolean unpackClasses) { |
| this.unpackClasses = unpackClasses; |
| } |
| |
| public void setWsdlFileName(String wsdlFileName) { |
| this.wsdlFileName = wsdlFileName; |
| } |
| |
| public void setOutput(String output) { |
| this.output = output; |
| } |
| |
| public void setPackageName(String packageName) { |
| this.packageName = packageName; |
| } |
| |
| public void setLanguage(String language) { |
| this.language = language; |
| } |
| |
| public void setAsyncOnly(boolean asyncOnly) { |
| this.asyncOnly = asyncOnly; |
| } |
| |
| public void setSyncOnly(boolean syncOnly) { |
| this.syncOnly = syncOnly; |
| } |
| |
| public void setServerSide(boolean serverSide) { |
| this.serverSide = serverSide; |
| } |
| |
| public void setTestcase(boolean testcase) { |
| this.testcase = testcase; |
| } |
| |
| public void setGenerateServiceXml(boolean generateServiceXml) { |
| this.generateServiceXml = generateServiceXml; |
| } |
| |
| public void setRepositoryPath(String repositoryPath) { |
| this.repositoryPath = repositoryPath; |
| } |
| |
| public void setExternalMapping(String externalMapping) { |
| this.externalMapping = externalMapping; |
| } |
| |
| public void setWsdlVersion(String wsdlVersion) { |
| this.wsdlVersion = wsdlVersion; |
| } |
| |
| public void setTargetSourceFolderLocation(String targetSourceFolderLocation) { |
| this.targetSourceFolderLocation = targetSourceFolderLocation; |
| } |
| |
| public void setTargetResourcesFolderLocation(String targetResourcesFolderLocation) { |
| this.targetResourcesFolderLocation = targetResourcesFolderLocation; |
| } |
| |
| public void setUnwrap(boolean unwrap) { |
| this.unwrap = unwrap; |
| } |
| |
| /** |
| * @return Returns Path. |
| */ |
| public Path getClasspath() { |
| return classpath; |
| } |
| |
| /** |
| * @param path |
| */ |
| public void setClasspath(Path path) { |
| classpath = path; |
| } |
| |
| public String getDatabindingName() { |
| return databindingName; |
| } |
| |
| public void setDatabindingName(String databindingName) { |
| this.databindingName = databindingName; |
| } |
| |
| public String getNamespaceToPackages() { |
| return namespaceToPackages; |
| } |
| |
| public void setNamespaceToPackages(String namespaceToPackages) { |
| this.namespaceToPackages = namespaceToPackages; |
| } |
| } |
