Schema.java

/*******************************************************************************
 * Copyright (c) 2006-2010 eBay Inc. All Rights Reserved.
 * 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
 *******************************************************************************/
package org.ebayopensource.turmeric.tools.codegen.external.wsdl.parser.schema;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;

import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;

/**
 * A class to represent the tpo level <schema> element of a schema
 * 
 * @author Owen Burroughs <owenb@apache.org>
 */
public class Schema extends Annotation implements Serializable {
	
	static final long serialVersionUID = 1L;	
	
	private String targetNamespace = "";	
	private List<SchemaType> types = new ArrayList<SchemaType>();
	@SuppressWarnings("unchecked")
	private ArrayList iaiLocations = new ArrayList();
	private List<Element> iaiElements = new ArrayList<Element>();

	/**
	 * Constructor
	 * @param el The dom element for this schema
	 */	
	@SuppressWarnings("unchecked")
	public Schema(Element el) {
		this(el, el.getAttribute("targetNamespace"));
		targetNamespace = el.getAttribute("targetNamespace");
        NodeList children = el.getChildNodes();
		for (int i=0; i<children.getLength(); i++) {
			Node child = children.item(i);
			if (child.getNodeType() == Node.ELEMENT_NODE) {
				Element subEl = (Element) child;
				String elType = subEl.getLocalName();
				if (elType.equals("complexType")) {
					types.add(new ComplexType(subEl, targetNamespace));
				} else if (elType.equals("simpleType")) {
					types.add(new SimpleType(subEl, targetNamespace));
				} else if (elType.equals("element")) {
					types.add(new ElementType(subEl, targetNamespace));	
				} else if( elType.equals("attributeGroup") ){ 
					types.add(new AttributeGroupType(subEl, targetNamespace));
				} else if( elType.equals("attribute") ){ 
					types.add(new Attribute(subEl, targetNamespace));
				} else if( elType.equals("group") ){ 
					types.add(new GroupType(subEl, targetNamespace));	
				} else if (elType.equals("import") || elType.equals("include")) {
					// If either an import or an include is defined, we need to get
					// the referenced file so store its location (if appropriate)
					String loc = subEl.getAttribute("schemaLocation");
					if (loc != null && !loc.equals("")) {
						iaiLocations.add(loc);
					}
					iaiElements.add(subEl);
				} else {
					//ignore all other types
				}
			}			
		}
	}
	
	private Schema(Element el, String targetNameSpace) {
		super(el, targetNameSpace);
		this.targetNamespace = targetNameSpace;
	}

	
	/**
	 * Get a list of all the types within this schema
	 * @return A list of SchemaType objects
	 */
	public List<SchemaType> getTypes() {
		return types;
	}

	/**
	 * Get the "targetNamespace" attribute for this schema
	 * @return The "targetNamespace" attribute
	 */	
	public String getTargetNamespace() {
		return targetNamespace;
	}

	/**
	 * Get all the locations of imported/included schemas so that they can also be retrieved
	 * @return An array of all the import/include schemaLocations
	 */		
	@SuppressWarnings("unchecked")
	String[] getImportsAndIncludes() {		
        return (String[]) iaiLocations.toArray(new String[iaiLocations.size()]);		
	}
	
	/**
	 * Get the import and include Elements as a List of Elements for the current Schema
	 * @return
	 * @author arajmony
	 */
	public List<Element> getImportsAndIncludesElements(){
		return iaiElements;
	}
}