| gvSIG desktop 1

/*

 * Created on 27-abr-2005

 *

 * TODO To change the template for this generated file go to

 * Window - Preferences - Java - Code Style - Code Templates

 */

package es.gva.cit.catalogClient.protocols;

import java.io.ByteArrayInputStream;

import java.io.InputStream;

import java.net.URL;

import java.util.Properties;

import java.util.StringTokenizer;

import java.util.Vector;

import org.w3c.dom.Document;

import org.w3c.dom.Node;

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

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

import es.gva.cit.catalogClient.utils.Strings;

import com.k_int.IR.IRQuery;

import com.k_int.IR.InformationFragment;

import com.k_int.IR.InformationFragmentSource;

import com.k_int.IR.RecordFormatSpecification;

import com.k_int.IR.SearchException;

import com.k_int.IR.SearchTask;

import com.k_int.IR.Searchable;

import com.k_int.IR.TimeoutExceededException;

import com.k_int.gen.Z39_50_APDU_1995.InitializeResponse_type;

/**

 * @author Administrador

 *

 * TODO To change the template for this generated type comment go to

 * Window - Preferences - Java - Code Style - Code Templates

 */

public class Z3950Protocol extends Z3950OriginBean implements IProtocols {

        public static final char ISO2709_RS = 035;

        public static final char ISO2709_FS = 036;

        public static final char ISO2709_IDFS = 037;

        private static final String PREFIX_QUERY_TYPE = "PREFIX";

        private static final String CCL_QUERY_TYPE = "CCL";;

        private int auth_type = 0; // 0=none, 1=anonymous, 2=open, 3=idpass

        private String principal = null;

        private String group = null;

        private String credentials = null;

        public Node[] doQuery(URL url,Object object,int firstRecord){

                String message = (String) object;

                Node[] nodes; 

                //Search properties

                Properties props = new Properties();

                props.setProperty("ServiceHost",url.getHost());

                props.setProperty("ServicePort",String.valueOf(url.getPort()));

                //props.setProperty("service_short_name","");

            //props.setProperty("service_long_name","");

            //props.setProperty("default_record_syntax","xml");

                //props.setProperty("service_user_principal","");

            //props.setProperty("service_user_group","");

            //props.setProperty("service_user_credentials","");

                //Search Object

            Searchable zServer = new com.k_int.z3950.IRClient.Z3950Origin();

            zServer.init(props);

                //Create the query object

            IRQuery e = new IRQuery();

                e.collections = new Vector();

                e.collections.add(getDatabase(url));

            e.query = new com.k_int.IR.QueryModels.PrefixString(new String(message));

            //create the search task

            SearchTask st = (SearchTask) zServer.createTask(e,null);

            try {

                        // time to wait

                        st.evaluate(1000000);

                        RecordFormatSpecification spec = new RecordFormatSpecification("xml",null,"f");

                        InformationFragmentSource ifs = st.getTaskResultSet();

                        InformationFragment frag = null;

                        Document[] doms = new Document[ifs.getFragmentCount()];

                        int lastRecord;

                        if (ifs.getFragmentCount() > 10){

                                nodes = new Node[11];

                                if ((ifs.getFragmentCount() - firstRecord) < 10)

                                        lastRecord = ifs.getFragmentCount();

                                else

                                        lastRecord = firstRecord + 9;

                        }else{

                                nodes = new Node[ifs.getFragmentCount() + 1];

                                lastRecord = ifs.getFragmentCount();

                        }

                        nodes[0] = XMLTreeNumberOfRecordsAnswer.getNode(ifs.getFragmentCount(),

                                        firstRecord,lastRecord);

                        int index = 1;                

                        for ( int i = firstRecord ; i<= lastRecord; i++ ){

                                //Fragment

                                frag = ifs.getFragment(i,spec);

                                String cadena = frag.toString();

                                //This is a strange character

                                cadena = Strings.replace(cadena,"?","");

                                //Change the format.

                                cadena = Strings.replace(cadena,"<?xml version=\"1.0\" encoding=\"UTF-8\"?>",

                                "<?xml version=\"1.0\" encoding=\"ISO-8859-1\"?>");

                                //System.out.println(cadena);

                                InputStream buffer = new ByteArrayInputStream(cadena.getBytes());

                                nodes[index] = XMLTree.XMLToTree(buffer);

                                index++;

                        }

           } catch (TimeoutExceededException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                        return null;

                } catch (SearchException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                        return null;

                } catch (Exception e2) {

                        //TODO Salta cuando no nos devuelve nada

                        e2.printStackTrace();

                        return null;

                }

                zServer.destroy();

                zServer = null;        

            return nodes;

    }

        public String getDatabase(URL url){

                StringTokenizer sti = new StringTokenizer(url.getPath(),"/");

                if (sti.countTokens() == 0)

                        return "Default";

                else

                        return sti.nextToken();

        }

        public String openConnection(URL url){

            String hostname = url.getHost();

                  String port = String.valueOf(url.getPort());

                String cadena = "";

                try{

                    InitializeResponse_type resp = connect(hostname,port,auth_type,principal,group,credentials);

                    if (resp.result.booleanValue() == true ){

                            if ( resp.referenceId != null )

                                    cadena = "Reference ID : " + new String(resp.referenceId) + "\n";

                            cadena = cadena + "Implementation ID : " + resp.implementationId + "\n";

                            cadena = cadena + "Implementation Name : " + resp.implementationName + "\n" ;

                            cadena = cadena + "Implementation Version : " + resp.implementationVersion + "\n";

                    }else{

                            System.out.println(" Failed to establish association");

                            return null;

                    }

            }catch ( Exception e ){

                    e.printStackTrace();

                    return null;

            }

            return cadena;

        }

        public static boolean isProtocolSupported(URL url){

                try{

                        //Search properties

                        Properties props = new Properties();

                        props.setProperty("ServiceHost",url.getHost());

                        props.setProperty("ServicePort",String.valueOf(url.getPort()));

                        //Search Object

                        Searchable zServer = new com.k_int.z3950.IRClient.Z3950Origin();

                        zServer.init(props);

                        //        Create the query object

                    IRQuery e = new IRQuery();

                    e.collections = new Vector();

                    e.collections.add(url.getHost()+":"+String.valueOf(url.getPort()));

                    e.query = new com.k_int.IR.QueryModels.PrefixString("@attrset geo @attr 1=4 @attr 4=6 @attr 2=3 \"a\"" );

                    //create the search task

                    SearchTask st = (SearchTask) zServer.createTask(e,null);

                          st.evaluate(100000);

                }catch(Exception e){

                        System.err.println(e.toString());

                        return false;

                }

                return true;

        }

         private void DisplayISO2709(byte[] octets)

          {

            String orig = new String(octets);

            int record_length = Integer.parseInt(orig.substring(0,5));   // Positions 0-4 are length of record

            char record_status = orig.charAt(5);

            char type_of_record = orig.charAt(6);

            int indicator_length = 2;                                    // Default indicator length

            int subfield_code_length = 2;                                // Default indicator length

            char character_coding_scheme = orig.charAt(9);

            int base_address = Integer.parseInt(orig.substring(12,17));  //

            int length_field_length = Character.digit(orig.charAt(20),10);

            int length_entry = Character.digit(orig.charAt(21),10);

            if ( Character.isDigit(orig.charAt(10)) )

              indicator_length = Character.digit(orig.charAt(10), 10);

            if ( Character.isDigit(orig.charAt(11)) )

              subfield_code_length = Character.digit(orig.charAt(11), 10);

            int offset = 24;

            while ( orig.charAt(offset) != ISO2709_FS )

            {

              // int tag = Integer.parseInt(orig.substring(offset, offset+3));

              String tag = orig.substring(offset, offset+3);

              System.out.print(tag+" ");

              offset += 3;

              int data_length = Integer.parseInt(orig.substring(offset, offset+length_field_length));

              offset += length_field_length;

              int data_offset = Integer.parseInt(orig.substring(offset, offset+length_entry));

              offset += length_entry;

              String this_tag = orig.substring(base_address+data_offset+2, base_address+data_offset+data_length);

              StringTokenizer st = new StringTokenizer(this_tag, ""+ISO2709_RS+ISO2709_FS+ISO2709_IDFS, true);

              String subfield = null;

              while ( st.hasMoreTokens() )

              {

                subfield = st.nextToken();

                if ( subfield.charAt(0) == ISO2709_RS )

                {

                  // System.err.print("Record sep ");

                }

                else if ( subfield.charAt(0) == ISO2709_FS )

                {

                  // System.err.print("Field sep ");

                }

                else if ( subfield.charAt(0) == ISO2709_IDFS )

                {

                  subfield = st.nextToken();

                  System.out.print("$"+subfield.charAt(0)+" "+subfield.substring(1, subfield.length()));

                }

                else

                {

                  System.out.print(subfield);

                }

              }

              System.out.println("");

            }

          }

          private void displayGRS(Vector v)

          {

            com.k_int.IR.Syntaxes.GRS1 grs_rec = new com.k_int.IR.Syntaxes.GRS1("Repository",

                                                                                "Coll", 

                                                                                "", 

                                                                                v,

                                                                                null);

            System.out.println(grs_rec.toString());

          }

}