ohair@286: /*
alanb@368:  * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved.
ohair@286:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
ohair@286:  *
ohair@286:  * This code is free software; you can redistribute it and/or modify it
ohair@286:  * under the terms of the GNU General Public License version 2 only, as
ohair@286:  * published by the Free Software Foundation.  Oracle designates this
ohair@286:  * particular file as subject to the "Classpath" exception as provided
ohair@286:  * by Oracle in the LICENSE file that accompanied this code.
ohair@286:  *
ohair@286:  * This code is distributed in the hope that it will be useful, but WITHOUT
ohair@286:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
ohair@286:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
ohair@286:  * version 2 for more details (a copy is included in the LICENSE file that
ohair@286:  * accompanied this code).
ohair@286:  *
ohair@286:  * You should have received a copy of the GNU General Public License version
ohair@286:  * 2 along with this work; if not, write to the Free Software Foundation,
ohair@286:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
ohair@286:  *
ohair@286:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@286:  * or visit www.oracle.com if you need additional information or have any
ohair@286:  * questions.
ohair@286:  */
ohair@286: 
ohair@286: package com.sun.tools.internal.ws.wsdl.parser;
ohair@286: 
ohair@286: import com.sun.tools.internal.ws.wsdl.document.jaxws.JAXWSBindingsConstants;
ohair@286: import com.sun.tools.internal.xjc.reader.internalizer.LocatorTable;
ohair@286: import com.sun.xml.internal.bind.marshaller.SAX2DOMEx;
ohair@286: import org.w3c.dom.Comment;
ohair@286: import org.w3c.dom.Document;
ohair@286: import org.w3c.dom.Element;
ohair@286: import org.w3c.dom.Node;
ohair@286: import org.xml.sax.Attributes;
ohair@286: import org.xml.sax.Locator;
ohair@286: import org.xml.sax.SAXException;
ohair@286: import org.xml.sax.ext.LexicalHandler;
ohair@286: 
ohair@286: import java.util.Set;
ohair@286: 
ohair@286: /**
ohair@286:  * Builds DOM while keeping the location information.
ohair@286:  *
ohair@286:  * <p>
ohair@286:  * This class also looks for outer most &lt;jaxws:bindings>
ohair@286:  * customizations.
ohair@286:  *
ohair@286:  * @author
ohair@286:  *     Kohsuke Kawaguchi (kohsuke.kawaguchi@sun.com)
ohair@286:  *     Vivek Pandey
ohair@286:  */
ohair@286: class DOMBuilder extends SAX2DOMEx implements LexicalHandler {
ohair@286:     /**
ohair@286:      * Grows a DOM tree under the given document, and
ohair@286:      * stores location information to the given table.
ohair@286:      *
ohair@286:      * @param outerMostBindings
ohair@286:      *      This set will receive newly found outermost
ohair@286:      *      jaxb:bindings customizations.
ohair@286:      */
ohair@286:     public DOMBuilder( Document dom, LocatorTable ltable, Set outerMostBindings ) {
ohair@286:         super( dom );
ohair@286:         this.locatorTable = ltable;
ohair@286:         this.outerMostBindings = outerMostBindings;
ohair@286:     }
ohair@286: 
ohair@286:     /** Location information will be stored into this object. */
ohair@286:     private final LocatorTable locatorTable;
ohair@286: 
ohair@286:     private final Set outerMostBindings;
ohair@286: 
ohair@286:     private Locator locator;
ohair@286: 
ohair@286:     public void setDocumentLocator(Locator locator) {
ohair@286:         this.locator = locator;
ohair@286:         super.setDocumentLocator(locator);
ohair@286:     }
ohair@286: 
ohair@286: 
ohair@286:     public void startElement(String namespaceURI, String localName, String qName, Attributes atts) {
ohair@286:         super.startElement(namespaceURI, localName, qName, atts);
ohair@286: 
ohair@286:         Element e = getCurrentElement();
ohair@286:         locatorTable.storeStartLocation( e, locator );
ohair@286: 
ohair@286:         // check if this element is an outer-most <jaxb:bindings>
ohair@286:         if( JAXWSBindingsConstants.JAXWS_BINDINGS.getNamespaceURI().equals(e.getNamespaceURI())
ohair@286:         &&  "bindings".equals(e.getLocalName()) ) {
ohair@286: 
ohair@286:             // if this is the root node (meaning that this file is an
ohair@286:             // external binding file) or if the parent is XML Schema element
ohair@286:             // (meaning that this is an "inlined" external binding)
ohair@286:             Node p = e.getParentNode();
ohair@286:             if( p instanceof Document) {
ohair@286:                 outerMostBindings.add(e);   // remember this value
ohair@286:             }
ohair@286:         }
ohair@286:     }
ohair@286: 
ohair@286:     public void endElement(String namespaceURI, String localName, String qName) {
ohair@286:         locatorTable.storeEndLocation( getCurrentElement(), locator );
ohair@286:         super.endElement(namespaceURI, localName, qName);
ohair@286:     }
ohair@286: 
ohair@286:     public void startDTD(String name, String publicId, String systemId) throws SAXException {}
ohair@286: 
ohair@286:     public void endDTD() throws SAXException {}
ohair@286: 
ohair@286:     public void startEntity(String name) throws SAXException {}
ohair@286: 
ohair@286:     public void endEntity(String name) throws SAXException {}
ohair@286: 
ohair@286:     public void startCDATA() throws SAXException {}
ohair@286: 
ohair@286:     public void endCDATA() throws SAXException {}
ohair@286: 
ohair@286:     public void comment(char[] ch, int start, int length) throws SAXException {
ohair@286:         //Element e = getCurrentElement(); // does not work as the comments at the top of the document would return Document Node
ohair@286:         // instead of Element
ohair@286:         Node parent = nodeStack.peek();
ohair@286:         Comment comment = document.createComment(new String(ch,start,length));
ohair@286:         parent.appendChild(comment);
ohair@286:     }
ohair@286: }