/**
* 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.jibx.tools;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.IdentityHashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import javax.wsdl.Definition;
import javax.wsdl.Import;
import javax.wsdl.Types;
import javax.wsdl.extensions.UnknownExtensibilityElement;
import javax.wsdl.extensions.schema.Schema;
import javax.wsdl.extensions.schema.SchemaImport;
import org.w3c.dom.Attr;
import org.w3c.dom.Element;
import org.apache.cxf.Bus;
import org.apache.cxf.catalog.CatalogXmlSchemaURIResolver;
import org.apache.cxf.common.xmlschema.SchemaCollection;
import org.apache.cxf.helpers.CastUtils;
import org.apache.ws.commons.schema.XmlSchema;
public class JibxSchemaHelper {
private final Map<String, Element> schemaList;
private List<JibxSchemaResolver> resolvers;
private final Map<String, String> catalogResolved = new HashMap<String, String>();
private final Bus bus;
public JibxSchemaHelper(final Bus b, final Map<String, Element> s) {
this.bus = b;
this.schemaList = s;
}
public void getSchemas(final Definition def, final SchemaCollection schemaCol,
List<JibxSchemaResolver> r) {
this.resolvers = r;
List<Definition> defList = new ArrayList<Definition>();
parseImports(def, defList);
extractSchema(def, schemaCol);
// added
getSchemaList(def);
Map<Definition, Definition> done = new IdentityHashMap<Definition, Definition>();
done.put(def, def);
for (Definition def2 : defList) {
if (!done.containsKey(def2)) {
extractSchema(def2, schemaCol);
// added
getSchemaList(def2);
done.put(def2, def2);
}
}
}
private void extractSchema(Definition def, SchemaCollection schemaCol) {
Types typesElement = def.getTypes();
if (typesElement != null) {
int schemaCount = 1;
for (Object obj : typesElement.getExtensibilityElements()) {
org.w3c.dom.Element schemaElem = null;
if (obj instanceof Schema) {
Schema schema = (Schema)obj;
schemaElem = schema.getElement();
} else if (obj instanceof UnknownExtensibilityElement) {
org.w3c.dom.Element elem = ((UnknownExtensibilityElement)obj).getElement();
if (elem.getLocalName().equals("schema")) {
schemaElem = elem;
}
}
if (schemaElem != null) {
synchronized (schemaElem.getOwnerDocument()) {
for (Object prefix : def.getNamespaces().keySet()) {
String ns = (String)def.getNamespaces().get(prefix);
if ("".equals(prefix)) {
if (!schemaElem.hasAttribute("xmlns")) {
Attr attr = schemaElem.getOwnerDocument()
.createAttributeNS(javax.xml.XMLConstants.XMLNS_ATTRIBUTE_NS_URI,
"xmlns");
attr.setValue(ns);
schemaElem.setAttributeNodeNS(attr);
}
} else if (!schemaElem.hasAttribute("xmlns:" + prefix)) {
String namespace = javax.xml.XMLConstants.XMLNS_ATTRIBUTE_NS_URI;
Attr attr = schemaElem.getOwnerDocument()
.createAttributeNS(namespace, "xmlns:" + prefix);
attr.setValue(ns);
schemaElem.setAttributeNodeNS(attr);
}
}
String systemId = def.getDocumentBaseURI() + "#types" + schemaCount;
schemaCol.setBaseUri(def.getDocumentBaseURI());
CatalogXmlSchemaURIResolver schemaResolver = new CatalogXmlSchemaURIResolver(bus);
schemaCol.setSchemaResolver(schemaResolver);
XmlSchema xmlSchema = schemaCol.read(schemaElem, systemId);
catalogResolved.putAll(schemaResolver.getResolvedMap());
JibxSchemaResolver resolver
= new JibxSchemaResolver(systemId, xmlSchema, schemaCol, schemaElem);
resolvers.add(resolver);
// SchemaInfo schemaInfo = new SchemaInfo(xmlSchema.getTargetNamespace());
// schemaInfo.setSchema(xmlSchema);
// schemaInfo.setSystemId(systemId);
// schemaInfo.setElement(schemaElem);
// schemaInfos.add(schemaInfo);
schemaCount++;
}
}
}
}
}
private void parseImports(Definition def, List<Definition> defList) {
List<Import> importList = new ArrayList<Import>();
Collection<List<Import>> ilist = CastUtils.cast(def.getImports().values());
for (List<Import> list : ilist) {
importList.addAll(list);
}
for (Import impt : importList) {
if (!defList.contains(impt.getDefinition())) {
defList.add(impt.getDefinition());
parseImports(impt.getDefinition(), defList);
}
}
}
// Workaround for getting the elements
private void getSchemaList(Definition def) {
Types typesElement = def.getTypes();
if (typesElement != null) {
Iterator ite = typesElement.getExtensibilityElements().iterator();
while (ite.hasNext()) {
Object obj = ite.next();
if (obj instanceof Schema) {
Schema schema = (Schema)obj;
addSchema(schema.getDocumentBaseURI(), schema);
}
}
}
}
private void addSchema(String docBaseURI, Schema schema) {
// String docBaseURI = schema.getDocumentBaseURI();
Element schemaEle = schema.getElement();
String name = docBaseURI;
for (JibxSchemaResolver r : resolvers) {
if (schemaEle == r.getElement()) {
name = r.getId();
}
}
if (schemaList.get(name) == null) {
schemaList.put(name, schemaEle);
} else if (schemaList.get(docBaseURI) != null && schemaList.containsValue(schemaEle)) {
// do nothing
} else {
String tns = schema.getDocumentBaseURI() + "#"
+ schema.getElement().getAttribute("targetNamespace");
if (schemaList.get(tns) == null) {
schemaList.put(tns, schema.getElement());
}
}
Map<String, List> imports = CastUtils.cast(schema.getImports());
if (imports != null && imports.size() > 0) {
Collection<String> importKeys = imports.keySet();
for (String importNamespace : importKeys) {
List<SchemaImport> schemaImports = CastUtils.cast(imports.get(importNamespace));
for (SchemaImport schemaImport : schemaImports) {
Schema tempImport = schemaImport.getReferencedSchema();
String key = schemaImport.getSchemaLocationURI();
if (importNamespace == null && tempImport != null) {
importNamespace = tempImport.getDocumentBaseURI();
}
if (tempImport != null && !catalogResolved.containsKey(key)) {
key = tempImport.getDocumentBaseURI();
}
if (tempImport != null && !isSchemaParsed(key, importNamespace)
&& !schemaList.containsValue(tempImport.getElement())) {
addSchema(key, tempImport);
}
}
}
}
}
private boolean isSchemaParsed(String baseUri, String ns) {
if (schemaList.get(baseUri) != null) {
Element ele = schemaList.get(baseUri);
String tns = ele.getAttribute("targetNamespace");
if (ns.equals(tns)) {
return true;
}
}
return false;
}
}