/**
* 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.java2wsdl.generator.jaxws;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.wsdl.Definition;
import javax.wsdl.Types;
import javax.wsdl.extensions.ExtensionRegistry;
import javax.wsdl.extensions.schema.Schema;
import javax.xml.bind.SchemaOutputResolver;
import javax.xml.namespace.QName;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.apache.cxf.common.i18n.Message;
import org.apache.cxf.common.logging.LogUtils;
import org.apache.cxf.tools.common.ToolContext;
import org.apache.cxf.tools.common.ToolException;
import org.apache.cxf.tools.common.WSDLConstants;
import org.apache.cxf.tools.common.model.WSDLModel;
import org.apache.cxf.tools.java2wsdl.processor.JavaToWSDLProcessor;
public class TypesGenerator {
private static final Logger LOG = LogUtils.getL7dLogger(JavaToWSDLProcessor.class);
private WSDLModel wmodel;
private Definition definition;
private ExtensionRegistry extensionRegistry;
private ToolContext env;
public TypesGenerator(WSDLModel model , ToolContext penv) {
this.definition = model.getDefinition();
this.wmodel = model;
env = penv;
extensionRegistry = definition.getExtensionRegistry();
}
public void generate() {
Message msg = new Message("GENERATE_TYPES_ERROR", LOG);
try {
wmodel.createJAXBContext();
} catch (Exception e) {
LOG.log(Level.SEVERE, "Create jaxbContext error");
throw new ToolException(msg, e);
}
SchemaOutputResolver resolver = new WSDLOutputResolver(env, wmodel);
try {
wmodel.getJaxbContext().generateSchema(resolver);
} catch (Exception e2) {
throw new ToolException(msg, e2);
}
Types types = definition.createTypes();
try {
Schema schema;
schema = (Schema)extensionRegistry.createExtension(Types.class,
new QName("http://www.w3.org/2001/XMLSchema",
"schema"));
DocumentBuilder docBuilder;
DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
factory.setNamespaceAware(true);
docBuilder = factory.newDocumentBuilder();
Document doc = docBuilder.newDocument();
Element element = doc.createElement("xsd:schema");
Map<String, String> schemafileMap = wmodel.getSchemaNSFileMap();
Set<Map.Entry<String, String>> entryset = schemafileMap.entrySet();
Iterator<Map.Entry<String, String>> ite = entryset.iterator();
while (ite.hasNext()) {
Map.Entry<String, String> entry = (Map.Entry<String, String>)ite.next();
Element importelement = doc.createElement("xsd:import");
importelement.setAttribute("namespace", entry.getKey());
importelement.setAttribute("schemaLocation", entry.getValue());
element.appendChild(importelement);
}
schema.setElement(element);
types.addExtensibilityElement(schema);
definition.setTypes(types);
} catch (Exception e) {
throw new ToolException(msg, e);
}
definition.setTargetNamespace(wmodel.getTargetNameSpace());
definition.addNamespace(WSDLConstants.WSDL_PREFIX, WSDLConstants.NS_WSDL);
definition.addNamespace(WSDLConstants.XSD_PREFIX, WSDLConstants.XSD_NAMESPACE);
definition.addNamespace(WSDLConstants.SOAP_PREFIX, WSDLConstants.SOAP11_NAMESPACE);
definition.addNamespace(WSDLConstants.TNS_PREFIX, wmodel.getTargetNameSpace());
int i = 0;
for (String s : wmodel.getJaxbContext().getKnownNamespaceURIs()) {
if (!"".equals(s)) {
definition.addNamespace("ns" + (++i), s);
}
}
}
}