source: MDService2/trunk/MDService2/src/eu/clarin/cmdi/mdservice/action/SRUProxyAction.java @ 1184

package eu.clarin.cmdi.mdservice.action;

import java.io.BufferedInputStream;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.MalformedURLException;
import java.net.URISyntaxException;
import java.net.URL;
import java.util.HashMap;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import javax.xml.transform.TransformerException;

import net.sf.json.JSON;
import net.sf.json.JSONArray;
import net.sf.json.JSONObject;
import net.sf.json.JSONSerializer;


import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringEscapeUtils;
import org.apache.struts2.interceptor.ServletRequestAware;

import com.opensymphony.xwork2.ActionSupport;

/**
 * main Struts 2 controller 
 * responds to requests (collections, model, recordset) 
 * by dispatching the requests to appropriate internal methods and filling the inputStream with the result
 *  
 * @author vronk
 *
 */
public class SRUProxyAction extends ActionSupport  
{

        private static final long serialVersionUID = 1L;
        
        /** 
         * Properties to be filled by Struts with according request-parameters
         */     
        private String version;
        private String operation;
        private String query;
        private int startRecord;
        private int maximumRecords;
        private String recordPacking;
        private String recordSchema;
        private String recordXPath;
        private int resultSetTTL;
        private String sortKeys;
        private String stylesheet;      
        private String extraRequestData;

        private MDRepoProxyAction mdrepoproxyaction;
        private InputStream resultStream;
        
        private final static HashMap<String,String> hash_operations = new HashMap<String,String>();
         static
         {       
                 hash_operations.put("searchRetrieve", "recordset");
         } 
         
         public SRUProxyAction(){
                 super();
                 initialize();
         }
         
         protected void  initialize(){
                 //repository = 1;
         }
         
         public InputStream getResultStream() {
             return resultStream;
         }
         public void setResultStream(InputStream _resultStream){
                resultStream = _resultStream;
         }
        
        public String getVersion() {
                return version;
        }

        public void setVersion(String version) {
                this.version = version;
        }

        public String getOperation() {
                return operation;
        }

        public void setOperation(String operation) {
                this.operation = operation;
        }
        
        public String getQuery() {
                return query;
        }

        public void setQuery(String query) {
                this.query = query;
        }
        
        public int getStartRecord() {
                return startRecord;
        }

        public void setStartRecord(int startRecord) {
                this.startRecord = startRecord;
        }
        public int getMaximumRecords() {
                return maximumRecords;
        }

        public void setMaximumRecords(int maximumRecords) {
                this.maximumRecords = maximumRecords;
        }
        
        public String getRecordPacking() {
                return recordPacking;
        }

        public void setRecordPacking(String recordPacking) {
                this.recordPacking = recordPacking;
        }

        public String getRecordSchema() {
                return recordSchema;
        }

        public void setRecordSchema(String recordSchema) {
                this.recordSchema = recordSchema;
        }

        public String getRecordXPath() {
                return recordXPath;
        }

        public void setRecordXPath(String recordXPath) {
                this.recordXPath = recordXPath;
        }

        public int getResultSetTTL() {
                return resultSetTTL;
        }

        public void setResultSetTTL(int resultSetTTL) {
                this.resultSetTTL = resultSetTTL;
        }
        
        public String getSortKeys() {
                return sortKeys;
        }

        public void setSortKeys(String sortKeys) {
                this.sortKeys = sortKeys;
        }
                
        public String getStylesheet() {
                return stylesheet;
        }

        public void setStylesheet(String stylesheet) {
                this.stylesheet = stylesheet;
        }
        
        public String getExtraRequestData() {
                return extraRequestData;
        }

        public void setExtraRequestData(String extraRequestData) {
                this.extraRequestData = extraRequestData;
        }


        /**
         * primitive identification of the target-proxy
         * base for finding the right base_url in the props
         * subclass has to override with its specific proxykey
         * @return
         */
        public String getProxyKey() {
                return "";
        }
        
        public void prepare(){
                mdrepoproxyaction = new MDRepoProxyAction();
                mdrepoproxyaction.setRepository(2);
                
                mdrepoproxyaction.setActionkey(hash_operations.get(this.getOperation()));
                mdrepoproxyaction.setSquery("");
                mdrepoproxyaction.setQ(getQuery());
                mdrepoproxyaction.setFormat("xml");
                
                if (getStartRecord() > 0 ){
                        mdrepoproxyaction.setStartItem(Integer.toString(getStartRecord()));
                }
                if (getMaximumRecords() > 0 ) {
                        mdrepoproxyaction.setMaximumItems(Integer.toString(this.getMaximumRecords()));
                }
                
        }
        public static String inputStreamAsString(InputStream stream)
                throws IOException {
                BufferedReader br = new BufferedReader(new InputStreamReader(stream));
                StringBuilder sb = new StringBuilder();
                String line = null;

                while ((line = br.readLine()) != null) {
                sb.append(line + "\n");
                }

                br.close();
                return sb.toString();
        }

        public void postprocess(){
                //recordPacking
                if (this.getRecordPacking() != null){
                        if (this.getRecordPacking().toLowerCase().equals("string")){
                                try {
                                        resultStream = IOUtils.toInputStream(StringEscapeUtils.escapeXml(inputStreamAsString(resultStream)));
                                } catch (IOException e) {
                                        // TODO Auto-generated catch block
                                        e.printStackTrace();
                                }

                        }
                }
        }
        /**
         * default Action method
         */
        public String execute() throws Exception {

                prepare();
                if (this.mdrepoproxyaction.execute() == SUCCESS){
                        resultStream = this.mdrepoproxyaction.getResultStream();
                        postprocess();
                        return SUCCESS;
                }
                return ERROR;
        }


}