source: MDService2/trunk/MDService2/src/eu/clarin/cmdi/mdservice/action/Pz2ProxyAction.java @ 1488

package eu.clarin.cmdi.mdservice.action;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.StringReader;
import java.io.StringWriter;
import java.net.MalformedURLException;
import java.net.URISyntaxException;
import java.net.URL;
import java.util.HashMap;
import java.io.ByteArrayInputStream;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import javax.xml.xpath.XPath;
import javax.xml.xpath.XPathConstants;
import javax.xml.xpath.XPathExpression;
import javax.xml.xpath.XPathExpressionException;
import javax.xml.xpath.XPathFactory;

import org.w3c.dom.Document;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;

import eu.clarin.cmdi.mdservice.model.Query;

/**
 
 */
public class Pz2ProxyAction extends GenericProxyAction {

        private static final long serialVersionUID = 1L;

        private String proxy_key = "pazpar2";
        
        private String command;
        private String query;
        private String sessionID;
        private String sort;
        private String block;
        
        public void setCommand(String command) {
                this.command = command;
        }
        public String getCommand() {
                return command;
        }
        public void setQuery(String query) {
                this.query = query;
        }
        public String getQuery() {
                return query;
        }
        public void setSessionID(String sessionID) {
                this.sessionID = sessionID;
        }
        public String getSessionID() {
                return sessionID;
        }

        public void setSort(String sort) {
                this.sort = sort;
        }
        public String getSort() {
                return sort;
        }
        public void setBlock(String block) {
                this.block = block;
        }
        public String getBlock() {
                return block;
        }
        
        @Override
        protected void  initialize(){
                 super.initialize();
                 this.setActionkey("pazpar2");                          
         }
        
        //TODO cache after analysis
        @Override
        public String getCache() {
                return Cache.SKIP;
        }
        
        public String createURLParams(){
                String params = "?";
                if (!(command == null)){
                        params = params + "command=" + command;
                }
                if (!(sessionID == null)){
                        params = params + "&session=" + sessionID;
                }
                if (!(query == null)){
                        params = params + "&query=" + query;
                }
                if (!(getStartItem() == null)){
                        params = params + "&start=" + getStartItem();
                }
                if (!(getMaximumItems() == null)){
                        params = params + "&num=" + getMaximumItems();
                }
                if (!(sort == null)){
                        params = params + "&sort=" + sort;
                        if (!(block == null)){
                                params = params + ":" + block;
                        }
                }
                
                return params;
        }
        @Override
        public String getBaseURI() {            
                String uri = getRepositoryPath();               
                return uri;
        }
        
        @Override
        public URL getTargetRequest() throws IOException {
        URL targetURL = null;           
                        
                targetURL =new URL( getBaseURL(), createURLParams() );
                Admin.notifyUser("Pz2.targetURL:" + targetURL);
            return targetURL;
        }
        

        @Override
        public InputStream getSourceStream() throws IOException, NoStylesheetException {
                if (getCommand() == null){
                        try {
                                return getSourcePz2();
                        } catch (Exception e) {
                                // TODO Auto-generated catch block
                                e.printStackTrace();
                        }
                } 
                        
                return super.getSourceStream();
        }
        
        public Document getDocument(InputStream is){
                Document doc = null;
                DocumentBuilderFactory docFactory = DocumentBuilderFactory.newInstance();
        DocumentBuilder docBuilder;
                try {
                        docBuilder = docFactory.newDocumentBuilder();
                        doc = docBuilder.parse(is);
                } catch (ParserConfigurationException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                } catch (SAXException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                } catch (IOException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                }       
                return doc;
        }
        public String getDocumentData(InputStream is, String expression){
                String sessionid = "";
        
                XPathFactory factory = XPathFactory.newInstance(); 
            XPath xpath = factory.newXPath(); 
            XPathExpression expr;
                try {
                        expr = xpath.compile(expression);
                        sessionid = (String) expr.evaluate(getDocument(is), XPathConstants.STRING);
                } catch (XPathExpressionException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                }
                        
                return sessionid;
                
        }
        public InputStream getSourcePz2() throws Exception {
                
                InputStream is = null;
                
                //init
                setCommand("init");
                String sessionID = getDocumentData(this.getSourceStream(), "//init/session");
                // get result
                setCommand("search");
                setSessionID(sessionID);
                setQuery(this.getSquery());
                String ok = getDocumentData(this.getSourceStream(),"//search/status");
                
                String activeclients = "1";
                setCommand("show");
                setSort("relevance");
                setBlock("1");
                while (Integer.parseInt(activeclients) > 0){
                        //activeclients = getDocumentData(this.getSourceStream(), "//show/activeclients");
                        is = this.getSourceStream();
                        activeclients = getDocumentData(is, "//show/activeclients");
                        
                }
                is = this.getSourceStream();
                return is;
        }

        
}