D-Net

/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

 * 

 * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.

 * 

 * The contents of this file are subject to the terms of either the GNU

 * General Public License Version 2 only ("GPL") or the Common Development

 * and Distribution License("CDDL") (collectively, the "License").  You

 * may not use this file except in compliance with the License. You can obtain

 * a copy of the License at https://glassfish.dev.java.net/public/CDDL+GPL.html

 * or glassfish/bootstrap/legal/LICENSE.txt.  See the License for the specific

 * language governing permissions and limitations under the License.

 * 

 * When distributing the software, include this License Header Notice in each

 * file and include the License file at glassfish/bootstrap/legal/LICENSE.txt.

 * Sun designates this particular file as subject to the "Classpath" exception

 * as provided by Sun in the GPL Version 2 section of the License file that

 * accompanied this code.  If applicable, add the following below the License

 * Header, with the fields enclosed by brackets [] replaced by your own

 * identifying information: "Portions Copyrighted [year]

 * [name of copyright owner]"

 * 

 * Contributor(s):

 * 

 * If you wish your version of this file to be governed by only the CDDL or

 * only the GPL Version 2, indicate your decision by adding "[Contributor]

 * elects to include this software in this distribution under the [CDDL or GPL

 * Version 2] license."  If you don't indicate a single choice of license, a

 * recipient has the option to distribute your version of this file under

 * either the CDDL, the GPL Version 2 or to extend the choice of license to

 * its licensees as provided above.  However, if you add GPL Version 2 code

 * and therefore, elected the GPL Version 2 license, then the option applies

 * only if the new code is made subject to such option by the copyright

 * holder.

 */

package eu.dnetlib.data.search.resources;

import javax.xml.namespace.QName;

import javax.xml.stream.XMLInputFactory;

import javax.xml.stream.XMLStreamReader;

import javax.xml.stream.events.XMLEvent;

import java.io.FileInputStream;

/**

 * MyStreamFilter sample is used to demonstrates the use

 * of STAX stream filter api's. This filter accepts only 

 * StartElement and EndElement events and filters out rest of

 * the events.

 */

public class MAin implements javax.xml.stream.StreamFilter{

    private static String filename = null;

    private static void printUsage() {

        System.out.println("usage: java -Djava.endorsed.dirs=<jaxp dist/lib directory> stax.filter.MyStreamFilter -f <xmlfile>");

    }

    /**

     *

     * @param args

     * @throws Exception

     */

    public static void main(String[] args) throws Exception {

        try {

            if(args[0].equals("-f")){

                filename = args[1];

            }

            else{

                printUsage() ;

            }

        } catch (ArrayIndexOutOfBoundsException aioobe){

            printUsage();

            System.exit(0);

        } catch (Exception ex){

            printUsage();

            ex.printStackTrace() ;

        }

        XMLInputFactory xmlif = null ;

        try{

            xmlif = XMLInputFactory.newInstance();

            xmlif.setProperty(XMLInputFactory.IS_REPLACING_ENTITY_REFERENCES,Boolean.TRUE);

            xmlif.setProperty(XMLInputFactory.IS_SUPPORTING_EXTERNAL_ENTITIES,Boolean.FALSE);

            xmlif.setProperty(XMLInputFactory.IS_NAMESPACE_AWARE , Boolean.TRUE);

            xmlif.setProperty(XMLInputFactory.IS_COALESCING , Boolean.TRUE);

        }catch(Exception ex){

            ex.printStackTrace();

        }

        System.out.println("FACTORY: " + xmlif);

        System.out.println("filename = "+ filename);

        try{

            FileInputStream fis = new FileInputStream(filename);

            XMLStreamReader xmlr = xmlif.createFilteredReader(xmlif.createXMLStreamReader(fis), new MAin());

            int eventType = xmlr.getEventType();

            printEventType(eventType);

            while(xmlr.hasNext()){

                eventType = xmlr.next();

                printEventType(eventType);

                printName(xmlr,eventType);

                printText(xmlr);

                if(xmlr.isStartElement()){

                    printAttributes(xmlr);

                }

                printPIData(xmlr);

                System.out.println("-----------------------------");

            }

        }catch(Exception ex){

            ex.printStackTrace();

        }

    }

    /**

     * @param eventType

     * @return

     */

    public final static String getEventTypeString(int eventType) {

        switch (eventType){

            case XMLEvent.START_ELEMENT:

                return "START_ELEMENT";

            case XMLEvent.END_ELEMENT:

                return "END_ELEMENT";

            case XMLEvent.PROCESSING_INSTRUCTION:

                return "PROCESSING_INSTRUCTION";

            case XMLEvent.CHARACTERS:

                return "CHARACTERS";

            case XMLEvent.COMMENT:

                return "COMMENT";

            case XMLEvent.START_DOCUMENT:

                return "START_DOCUMENT";

            case XMLEvent.END_DOCUMENT:

                return "END_DOCUMENT";

            case XMLEvent.ENTITY_REFERENCE:

                return "ENTITY_REFERENCE";

            case XMLEvent.ATTRIBUTE:

                return "ATTRIBUTE";

            case XMLEvent.DTD:

                return "DTD";

            case XMLEvent.CDATA:

                return "CDATA";

        }

        return "UNKNOWN_EVENT_TYPE";

    }

    private static void printEventType(int eventType) {

        System.out.print("EVENT TYPE("+eventType+"):");

        System.out.println(getEventTypeString(eventType));

    }

    private static void printName(XMLStreamReader xmlr,int eventType){

        if(xmlr.hasName()){

            System.out.println("HAS NAME: " + xmlr.getLocalName());

        } else {

            System.out.println("HAS NO NAME");

        }

    }

    private static void printText(XMLStreamReader xmlr){

        if(xmlr.hasText()){

            System.out.println("HAS TEXT: " + xmlr.getText());

        } else {

            System.out.println("HAS NO TEXT");

        }

    }

    private static void printPIData(XMLStreamReader xmlr){

        if (xmlr.getEventType() == XMLEvent.PROCESSING_INSTRUCTION){

            System.out.println(" PI target = " + xmlr.getPITarget() ) ;

            System.out.println(" PI Data = " + xmlr.getPIData() ) ;

        }

    }

    private static void printAttributes(XMLStreamReader xmlr){

        if(xmlr.getAttributeCount() > 0){

            System.out.println("\nHAS ATTRIBUTES: ");

            int count = xmlr.getAttributeCount() ;

            for(int i = 0 ; i < count ; i++) {

                QName name = xmlr.getAttributeName(i) ;

                String namespace = xmlr.getAttributeNamespace(i) ;

                String  type = xmlr.getAttributeType(i) ;

                String prefix = xmlr.getAttributePrefix(i) ;

                String value = xmlr.getAttributeValue(i) ;

                System.out.println("ATTRIBUTE-PREFIX: " + prefix );

                System.out.println("ATTRIBUTE-NAMESP: " + namespace );

                System.out.println("ATTRIBUTE-NAME:   " + name.toString() );

                System.out.println("ATTRIBUTE-VALUE:  " + value );

                System.out.println("ATTRIBUTE-TYPE:  " + type );

                System.out.println();

            }

        } else {

            System.out.println("HAS NO ATTRIBUTES");

        }

    }

    /**

     * Accept only StartElement and EndElement events,Filters out rest of the events.

     * @param reader

     * @return

     */

    public boolean accept(XMLStreamReader reader) {

        if(!reader.isStartElement() && !reader.isEndElement())

            return false;

        else

            return true;

    }

}