Application: gvSIG desktop

/* gvSIG. Sistema de Informaci�n Geogr�fica de la Generalitat Valenciana

*

* Copyright (C) 2004 IVER T.I. and Generalitat Valenciana.

*

* This program is free software; you can redistribute it and/or

* modify it under the terms of the GNU General Public License

* as published by the Free Software Foundation; either version 2

* of the License, or (at your option) any later version.

*

* This program is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

* GNU General Public License for more details.

*

* You should have received a copy of the GNU General Public License

* along with this program; if not, write to the Free Software

* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

*

* For more information, contact:

*

*  Generalitat Valenciana

*   Conselleria d'Infraestructures i Transport

*   Av. Blasco Ib��ez, 50

*   46010 VALENCIA

*   SPAIN

*

*      +34 963862235

*   gvsig@gva.es

*      www.gvsig.gva.es

*

*    or

*

*   IVER T.I. S.A

*   Salamanca 50

*   46005 Valencia

*   Spain

*

*   +34 963163400

*   dac@iver.es

*/

package es.gva.cit.catalogClient.csw.drivers;

import es.gva.cit.catalogClient.csw.parsers.CswCapabilitiesParser;

import es.gva.cit.catalogClient.csw.parsers.CswSupportedProtocolOperations;

import es.gva.cit.catalogClient.drivers.AbstractCatalogServiceDriver;

import es.gva.cit.catalogClient.metadataXML.XMLNode;

import es.gva.cit.catalogClient.metadataXML.XMLTree;

import es.gva.cit.catalogClient.metadataXML.XMLTreeNumberOfRecordsAnswer;

import es.gva.cit.catalogClient.protocols.HTTPGetProtocol;

import es.gva.cit.catalogClient.protocols.HTTPPostProtocol;

import es.gva.cit.catalogClient.protocols.SOAPProtocol;

import es.gva.cit.catalogClient.querys.Query;

import es.gva.cit.catalogClient.schemas.Schemas;

import org.apache.commons.httpclient.NameValuePair;

import java.net.URL;

/**

 * This class implements the CSW protocol.

 * 

 * @see http://portal.opengeospatial.org/files/?artifact_id=5929&version=1

 * @author Jorge Piera Llodra (piera_jor@gva.es)

 */

public class CSWCatalogServiceDriver extends AbstractCatalogServiceDriver {

    private String responseHandler = null;

    private String hopCount = null;

    private String distributedSearch = null;

    private String constraint = null;

    private String[] CONSTRAINTLANGUAGE = null;

    private String[] elementSetName = null;

    private String[] typeNames = null;

    private String[] resultType = null;

    private String[] NAMESPACE = null;

  private String version = "2.0.0";

    private CswSupportedProtocolOperations operations = null;

    /*

     * (non-Javadoc)

     *

     * @see ICatalogServerDriver#GetCapabilities(java.lang.String,

     *      java.lang.String)

     */

    public XMLNode[] getCapabilities(URL url) {

        XMLNode[] nodes = null;

        CSWMessages messages = new CSWMessages(this);

        CswCapabilitiesParser parser = new CswCapabilitiesParser(this,url);

        System.out.println("**************GET*************");

        nodes = new HTTPGetProtocol().doQuery(url, messages.getHTTPGETCapabilities(true), 0);

        if ((nodes != null) && (parser.getExceptionCode(nodes[0]).equals("8001")) &&

                (parser.getExceptionSubCode(nodes[0]).equals("5002")))

            nodes = new HTTPGetProtocol().doQuery(url, messages.getHTTPGETCapabilities(false), 0);

        if (nodes == null) {

            System.out.println("**************POST*************");

            nodes = new HTTPPostProtocol().doQuery(url,

                    messages.getHTTPPOSTCapabilities(), 0);

            if (nodes == null) {

                System.out.println("**************SOAP*************");

                nodes = new SOAPProtocol().doQuery(url,

                        messages.getSOAPCapabilities(), 0);

                if (nodes == null) {

                    return null;

                } else {

                    setCommunicationProtocol("SOAP");

                }

            } else {

                setCommunicationProtocol("POST");

            }

        } else {

            setCommunicationProtocol("GET");

        }

        return nodes;

    }

    /**

     * this method implements the describeRecords operation

     * @param url

     * @return

     */

    public XMLNode[] describeRecords(URL url) {

        //return new HTTPPostProtocol().doQuery(url, getMessageDescribeRecords(),0);

        return null;

    }

    /*

     * (non-Javadoc)

     *

     * @see ICatalogServerDriver#GetRecords(java.lang.String, java.lang.String,

     *      java.lang.String, int, int, java.lang.String, java.lang.String)

     */

    public XMLNode[] getRecords(URL url, Query query, int firstRecord) {

        setQuery(query);

        CSWMessages messages = new CSWMessages(this);

        XMLNode[] answerNodes = null;

        XMLNode[] nodes = null;

        if (getOperations().getGetRecords()==null)

                return null;

        if (getServerProfile().equals(Schemas.ISO19115)){

            firstRecord = firstRecord - 1;

        }

        for (int i=0 ; i<getOperations().getGetRecords().length; i++){

            /*

            if (getOperations().getGetRecords()[i].equals("GET")){

                    System.out.println("**************GET*************");

                    nodes = new HTTPGetProtocol().doQuery(url,

                        messages.getHTTPGETRecords(getQuery() , firstRecord), firstRecord);

                    setCommunicationProtocol("GET");                    

                break;

            }

            */

            if (getOperations().getGetRecords()[i].equals("POST")){

                    System.out.println("**************POST*************");

                System.out.println(messages.getHTTPPOSTRecords(getQuery(), firstRecord));                    

                    nodes = new HTTPPostProtocol().doQuery(url,

                        messages.getHTTPPOSTRecords(getQuery(), firstRecord), firstRecord);

                setCommunicationProtocol("POST");                

                break;

            }

            /*

            if (getOperations().getGetRecords()[i].equals("SOAP")){

                         System.out.println("**************SOAP*************");

                         nodes = new SOAPProtocol().doQuery(url,

                        messages.getSOAPRecords(getQuery(), firstRecord), firstRecord);

                         setCommunicationProtocol("SOAP");

                 //We must cut the SOAP Head

                 nodes[0] = SOAPMessageParser.cutHead(nodes[0]);

                         break;

                } */ 

        }

        if (nodes == null) 

                return null;

        int numberOfRecords = getNumberOfRecords(nodes[0]);

        if (numberOfRecords == -1) {

            return null;

        }

        answerNodes = createAnswerTree(numberOfRecords, firstRecord, nodes[0],

                url);

        return answerNodes;

    }

    /**

     * It returns an array of nodes with:

     *         1st position -> tree with information of the answer

     *  Next positions -> One result for each position

     * @param numberOfRecords

     * Number of records returned by the query

     * @param firstRecord

     * Number of the first record

     * @param node

     * Tree returned by the server

     * @param url

     * Server URL

     * @return

     */

    private XMLNode[] createAnswerTree(int numberOfRecords, int firstRecord,

        XMLNode node, URL url) {

        XMLNode[] answerNodes = null;

        if (numberOfRecords > 10) {

            answerNodes = new XMLNode[11];

        } else {

            answerNodes = new XMLNode[numberOfRecords + 1];

        }

        answerNodes[0] = XMLTreeNumberOfRecordsAnswer.getNode(numberOfRecords,

                firstRecord, firstRecord + numberOfRecords);

        XMLNode[] auxNodes = cutMetadata(node);

        if (getServerProfile().equals(Schemas.EBRIM)) {

            auxNodes = getEbRIMNodes(auxNodes, url);

        }

        for (int i = 1;

                (i <= numberOfRecords) && (i <= 10) &&

                (i <= (numberOfRecords - firstRecord + 1)); i++)

            answerNodes[i] = auxNodes[i - 1];

        return answerNodes;

    }

    /**

     * This function retrieve the nodes for one ebRIM answer

     * @param nodes

     * Nodes of the ebRIM answer

     * @param nodes

     * Server URL

     * @return

     * Medatada Nodes.

     */

    private XMLNode[] getEbRIMNodes(XMLNode[] nodes, URL url) {

        XMLNode[] auxNodes = new XMLNode[nodes.length];

        for (int i = 0; i < nodes.length; i++) {

            String id = XMLTree.searchAtribute(nodes[i], "id");

            auxNodes[i] = new HTTPGetProtocol().doQuery(url,

                    getEbRIMRequestParameters(id), 0)[0];

        }

        return auxNodes;

    }

    /**

     * It Returns the parameters needed by getExtrinsicContent

     * @param id

     * Record id

     * @return

     */

    private NameValuePair[] getEbRIMRequestParameters(String id) {

        NameValuePair nvp1 = new NameValuePair("request", "getExtrinsicContent");

        NameValuePair nvp2 = new NameValuePair("id", id);

        return new NameValuePair[] { nvp1, nvp2 };

    }

    /**

     * This function returns the number of records that have been retrieved.

     * It Reads a Node value.

     * @param node

     * The answer tree

     * @return

     * The number of records

     */

    private int getNumberOfRecords(XMLNode node){

            int numberOfRecords = getNumberOfRecords(node,"csw:SearchResults","numberOfRecordsMatched");

            if (numberOfRecords == -1)

                    numberOfRecords = getNumberOfRecords(node,"SearchResults","numberOfRecordsMatched");

            return numberOfRecords;

    }

    /**

     * This function finds and separates metadata from a full tree

     * @param node

     * full tree

     * @return

     * An array of trees that contain metadata.

     */

    private XMLNode[] cutMetadata(XMLNode node) {

        XMLNode[] auxNodes = XMLTree.searchMultipleNode(node,

                "csw:SearchResults->brief:MD_Metadata");

        if (auxNodes.length == 0) {

            auxNodes = XMLTree.searchMultipleNode(node,

                    "SearchResults->ebrim:Organization");

        }

        if (auxNodes.length == 0) {

            auxNodes = XMLTree.searchMultipleNode(node,

                    "SearchResults->wrs:WRSExtrinsicObject");

        }

        if (auxNodes.length == 0) {

            auxNodes = XMLTree.searchMultipleNode(node,

                    "SearchResults->WRSExtrinsicObject");

        }        

        if (auxNodes.length == 0) {

            auxNodes = XMLTree.searchMultipleNode(node,

                    "csw:SearchResults->dc:metadata");

        }

        if (auxNodes.length == 0) {

            auxNodes = XMLTree.searchMultipleNode(node,

                    "csw:SearchResults->iso19115:MD_Metadata");

        }

        return auxNodes;

    }

    /*

     * (non-Javadoc)

     *

     * @see catalogClient.ICatalogServerDriver#setParameters(java.util.Properties)

     */

    public boolean setParameters(XMLNode[] nodes,URL url) {

        return new CswCapabilitiesParser(this,url).parse(nodes[0]);

    }

    /**

     * @param typeNames

     * The typeNames to set.

     */

    public void setTypeNames(String[] typeNames) {

        this.typeNames = typeNames;

    }

    /* (non-Javadoc)

     * @see es.gva.cit.catalogClient.drivers.ICatalogServiveDriver#isTheProtocol(java.net.URL)

     */

    public boolean isProtocolSupported(URL url) {

        return true;

    }

    /**

     * @return Returns the distributedSearch.

     */

    public String getDistributedSearch() {

        return distributedSearch;

    }

    /**

     * @param distributedSearch

     *            The distributedSearch to set.

     */

    public void setDistributedSearch(String distributedSearch) {

        this.distributedSearch = distributedSearch;

    }

    /**

     * @return Returns the hopCount.

     */

    public String getHopCount() {

        return hopCount;

    }

    /**

     * @param hopCount

     *            The hopCount to set.

     */

    public void setHopCount(String hopCount) {

        this.hopCount = hopCount;

    }

    /**

     * @return Returns the nAMESPACE.

     */

    public String[] getNAMESPACE() {

        return NAMESPACE;

    }

    /**

     * @param namespace

     *            The nAMESPACE to set.

     */

    public void setNAMESPACE(String[] namespace) {

        NAMESPACE = namespace;

    }

    /**

     * @return Returns the responseHandler.

     */

    public String getResponseHandler() {

        return responseHandler;

    }

    /**

     * @param responseHandler

     *            The responseHandler to set.

     */

    public void setResponseHandler(String responseHandler) {

        this.responseHandler = responseHandler;

    }

    /**

     * @return Returns the resultType.

     */

    public String[] getResultType() {

        return resultType;

    }

    /**

     * @param resultType

     *            The resultType to set.

     */

    public void setResultType(String[] resultType) {

        this.resultType = resultType;

    }

    /**

     * @return Returns the typeNames.

     */

    public String[] getTypeNames() {

        return typeNames;

    }

    /**

     * @return Returns the version.

     */

    public String getVersion() {

        return version;

    }

    /**

     * @param version

     *            The version to set.

     */

    public void setVersion(String version) {

        this.version = version;

    }

    /**

     * @return Returns the constraint.

     */

    public String getConstraint() {

        return constraint;

    }

    /**

     * @param constraint

     *            The constraint to set.

     */

    public void setConstraint(String constraint) {

        this.constraint = constraint;

    }

    /**

     * @return Returns the cONSTRAINTLANGUAGE.

     */

    public String[] getCONSTRAINTLANGUAGE() {

        return CONSTRAINTLANGUAGE;

    }

    /**

     * @param constraintlanguage

     *            The cONSTRAINTLANGUAGE to set.

     */

    public void setCONSTRAINTLANGUAGE(String[] constraintlanguage) {

        CONSTRAINTLANGUAGE = constraintlanguage;

    }

    /**

     * @return Returns the elementSetName.

     */

    public String[] getElementSetName() {

        return elementSetName;

    }

    /**

     * @param elementSetName

     *            The elementSetName to set.

     */

    public void setElementSetName(String[] elementSetName) {

        this.elementSetName = elementSetName;

    }

        /**

         * @return Returns the operations.

         */

        public CswSupportedProtocolOperations getOperations() {

                return operations;

        }

        /**

         * @param operations The operations to set.

         */

        public void setOperations(CswSupportedProtocolOperations operations) {

                this.operations = operations;

        }

    /* (non-Javadoc)

     * @see es.gva.cit.catalogClient.drivers.ICatalogServiceDriver#getServerPofile()

     */

    public String getQueryProfile() {

        // TODO Auto-generated method stub

        return getServerProfile();

    }

}