-package com.indexdata.mkjsf.pazpar2.sp;\r
-\r
-import static com.indexdata.mkjsf.utils.Utils.nl;\r
-\r
-import java.io.BufferedReader;\r
-import java.io.File;\r
-import java.io.FileReader;\r
-import java.io.IOException;\r
-import java.util.ArrayList;\r
-import java.util.HashMap;\r
-import java.util.List;\r
-import java.util.Map;\r
-import java.util.StringTokenizer;\r
-\r
-import org.apache.http.Header;\r
-import org.apache.http.HttpEntity;\r
-import org.apache.http.HttpResponse;\r
-import org.apache.http.StatusLine;\r
-import org.apache.http.client.ClientProtocolException;\r
-import org.apache.http.client.HttpClient;\r
-import org.apache.http.client.ResponseHandler;\r
-import org.apache.http.client.methods.HttpGet;\r
-import org.apache.http.client.methods.HttpPost;\r
-import org.apache.http.conn.ClientConnectionManager;\r
-import org.apache.http.conn.scheme.PlainSocketFactory;\r
-import org.apache.http.conn.scheme.Scheme;\r
-import org.apache.http.conn.scheme.SchemeRegistry;\r
-import org.apache.http.entity.ByteArrayEntity;\r
-import org.apache.http.entity.FileEntity;\r
-import org.apache.http.impl.client.DefaultHttpClient;\r
-import org.apache.http.impl.conn.PoolingClientConnectionManager;\r
-import org.apache.http.util.EntityUtils;\r
-import org.apache.log4j.Logger;\r
-\r
-import com.indexdata.mkjsf.config.Configuration;\r
-import com.indexdata.mkjsf.config.ConfigurationReader;\r
-import com.indexdata.mkjsf.errors.ConfigurationException;\r
-import com.indexdata.mkjsf.pazpar2.CommandResponse;\r
-import com.indexdata.mkjsf.pazpar2.SearchClient;\r
-import com.indexdata.mkjsf.pazpar2.commands.CommandParameter;\r
-import com.indexdata.mkjsf.pazpar2.commands.Pazpar2Command;\r
-import com.indexdata.mkjsf.pazpar2.commands.sp.AuthCommand;\r
-import com.indexdata.mkjsf.pazpar2.data.CommandError;\r
-import com.indexdata.mkjsf.pazpar2.sp.auth.ServiceProxyUser;\r
-import com.indexdata.mkjsf.utils.Utils;\r
-\r
-public class ServiceProxyClient implements SearchClient {\r
- \r
- private static final long serialVersionUID = -4031644009579840277L;\r
- private static Logger logger = Logger.getLogger(ServiceProxyClient.class);\r
- public static final String MODULENAME = "proxyclient";\r
- public static final String SERVICE_PROXY_URL = "SERVICE_PROXY_URL";\r
- public static final String SP_INIT_DOC_PATHS = "SP_INIT_DOC_PATHS";\r
- private String selectedServiceUrl = "";\r
- private List<String> serviceUrls = new ArrayList<String>();\r
- private List<String> initDocPaths = null;\r
- private Configuration config = null;\r
- \r
- ProxyPz2ResponseHandler handler = new ProxyPz2ResponseHandler();\r
- private transient HttpClient client; \r
- private Pazpar2Command checkAuth = null;\r
- private Pazpar2Command ipAuth = null;\r
-\r
- public ServiceProxyClient () {\r
- SchemeRegistry schemeRegistry = new SchemeRegistry();\r
- schemeRegistry.register(new Scheme("http", 80, PlainSocketFactory.getSocketFactory()));\r
- ClientConnectionManager cm = new PoolingClientConnectionManager(schemeRegistry);\r
- client = new DefaultHttpClient(cm);\r
- }\r
- \r
- @Override\r
- public void configure (ConfigurationReader configReader) {\r
- logger.info(Utils.objectId(this) + " is configuring using the provided " + Utils.objectId(configReader));\r
- try {\r
- config = configReader.getConfiguration(this); \r
- serviceUrls = getMultiProperty(config.get(SERVICE_PROXY_URL),",");\r
- if (serviceUrls.size()==1) {\r
- selectedServiceUrl = serviceUrls.get(0);\r
- }\r
- this.initDocPaths = getMultiProperty(config.get(SP_INIT_DOC_PATHS),",");\r
- checkAuth = new AuthCommand(null);\r
- checkAuth.setParameterInState(new CommandParameter("action","=","check"));\r
- ipAuth = new AuthCommand(null);\r
- ipAuth.setParameterInState(new CommandParameter("action","=","ipauth"));\r
- } catch (ConfigurationException c) {\r
- // TODO: \r
- c.printStackTrace();\r
- } \r
- }\r
- \r
- private List<String> getMultiProperty(String prop, String separator) {\r
- List<String> props = new ArrayList<String>();\r
- if (prop != null) { \r
- StringTokenizer tokenizer = new StringTokenizer(prop,separator);\r
- while (tokenizer.hasMoreElements()) {\r
- props.add(tokenizer.nextToken());\r
- } \r
- }\r
- return props;\r
- }\r
- \r
- public boolean authenticate (ServiceProxyUser user) {\r
- logger.info("Authenticating [" + user.getProperty("name") + "]"); \r
- Pazpar2Command auth = new AuthCommand(null);\r
- auth.setParametersInState(new CommandParameter("action","=","login"), \r
- new CommandParameter("username","=",user.getProperty("name")), \r
- new CommandParameter("password","=",user.getProperty("password"))); \r
- ServiceProxyCommandResponse commandResponse = send(auth);\r
- String responseStr = commandResponse.getResponseString();\r
- logger.info(responseStr); \r
- if (responseStr.contains("FAIL")) {\r
- user.credentialsAuthenticationSucceeded(false);\r
- return false;\r
- } else {\r
- user.credentialsAuthenticationSucceeded(true);\r
- return true;\r
- } \r
- }\r
- \r
- public boolean checkAuthentication (ServiceProxyUser user) { \r
- ServiceProxyCommandResponse commandResponse = send(checkAuth); \r
- String responseStr = commandResponse.getResponseString(); \r
- logger.info(responseStr);\r
- if (responseStr.contains("FAIL")) { \r
- user.authenticationCheckFailed();\r
- return false;\r
- } else { \r
- return true;\r
- } \r
- }\r
- \r
- public boolean ipAuthenticate (ServiceProxyUser user) {\r
- ServiceProxyCommandResponse commandResponse = send(ipAuth); \r
- String responseStr = commandResponse.getResponseString();\r
- logger.info(responseStr);\r
- if (responseStr.contains("FAIL")) {\r
- user.ipAuthenticationSucceeded(false); \r
- return false;\r
- } else {\r
- user.ipAuthenticationSucceeded(true);\r
- return true;\r
- } \r
- }\r
- \r
- public boolean isAuthenticatingClient () {\r
- return true;\r
- }\r
- \r
- public boolean isAuthenticated (ServiceProxyUser user) {\r
- if (user.getProperty("name") != null && user.getProperty("password") != null) {\r
- return checkAuthentication(user);\r
- } else {\r
- return false;\r
- }\r
- }\r
- \r
- /**\r
- * Makes the request\r
- * @param request\r
- * @return HTTP response as a String\r
- * @throws ClientProtocolException\r
- * @throws IOException\r
- */\r
- private ServiceProxyCommandResponse send(Pazpar2Command command) {\r
- ServiceProxyCommandResponse commandResponse = null;\r
- String url = selectedServiceUrl + "?" + command.getEncodedQueryString(); \r
- logger.info("Sending request "+url); \r
- HttpGet httpget = new HttpGet(url); \r
- byte[] response = null;\r
- try {\r
- response = client.execute(httpget, handler);\r
- if (handler.getStatusCode()==200) {\r
- commandResponse = new ServiceProxyCommandResponse(handler.getStatusCode(),response,handler.getContentType());\r
- } else {\r
- logger.error("Service Proxy status code: " + handler.getStatusCode());\r
- commandResponse = new ServiceProxyCommandResponse(handler.getStatusCode(),CommandError.insertPazpar2ErrorXml(command.getCommandName(), "Service Proxy error occurred", new String(response,"UTF-8")),"text/xml"); \r
- } \r
- } catch (Exception e) {\r
- e.printStackTrace();\r
- commandResponse = new ServiceProxyCommandResponse(-1,CommandError.createErrorXml(command.getCommandName(), e.getClass().getSimpleName(), (e.getMessage()!= null ? e.getMessage() : "") + (e.getCause()!=null ? e.getCause().getMessage() : "")),"text/xml");\r
- }\r
- return commandResponse; \r
- }\r
- \r
- public class ProxyPz2ResponseHandler implements ResponseHandler<byte[]> {\r
- private StatusLine statusLine = null;\r
- private Header contentType = null;\r
- public byte[] handleResponse(HttpResponse response) throws ClientProtocolException, IOException {\r
- byte[] resp = null;\r
- HttpEntity entity = response.getEntity(); \r
- statusLine = response.getStatusLine();\r
- if (entity != null) { \r
- resp = EntityUtils.toByteArray(entity); \r
- contentType = response.getEntity().getContentType(); \r
- } \r
- EntityUtils.consume(entity); \r
- return resp;\r
- }\r
- public int getStatusCode() {\r
- return statusLine.getStatusCode();\r
- } \r
- public String getReasonPhrase() {\r
- return statusLine.getReasonPhrase();\r
- }\r
- public String getContentType () {\r
- return (contentType != null ? contentType.getValue() : "Content-Type not known"); \r
- }\r
- }\r
-\r
- public int getStatusCode () {\r
- return handler.getStatusCode();\r
- }\r
- \r
- public String getReasonPhrase() {\r
- return handler.getReasonPhrase();\r
- }\r
-\r
- @Override\r
- public void setSearchCommand(Pazpar2Command command) {\r
- // Do nothing, Service Proxy is handling this \r
- }\r
-\r
- @Override\r
- public CommandResponse executeCommand(Pazpar2Command command) {\r
- return send(command);\r
- }\r
-\r
- public ServiceProxyClient cloneMe() {\r
- logger.debug("Cloning Pz2Client");\r
- ServiceProxyClient clone = new ServiceProxyClient();\r
- clone.client = this.client;\r
- clone.serviceUrls = this.serviceUrls;\r
- clone.selectedServiceUrl = this.selectedServiceUrl;\r
- clone.initDocPaths = this.initDocPaths;\r
- return clone;\r
- }\r
-\r
- @Override\r
- public Map<String, String> getDefaults() { \r
- return new HashMap<String,String>();\r
- }\r
-\r
- @Override\r
- public String getModuleName() {\r
- return MODULENAME;\r
- }\r
- \r
- @Override\r
- public List<String> documentConfiguration () {\r
- List<String> doc = new ArrayList<String>();\r
- doc.add(nl+ MODULENAME + " was configured to access the Pazpar2 service proxy at: " + (selectedServiceUrl.length()>0 ? selectedServiceUrl : "[not defined yet]"));\r
- return null;\r
- }\r
- \r
- public ServiceProxyCommandResponse postInitDoc (String filePath) throws IOException {\r
- logger.info("Looking to post the file in : [" + filePath +"]");\r
- HttpPost post = new HttpPost(selectedServiceUrl+"?command=init&includeDebug=yes");\r
- File initDoc = new File(filePath);\r
- logger.info("Posting to SP: ");\r
- if (logger.isDebugEnabled()) {\r
- BufferedReader reader = new BufferedReader(new FileReader(initDoc));\r
- String line;\r
- while ( (line = reader.readLine()) != null) {\r
- System.out.println(line);\r
- }\r
- reader.close();\r
- }\r
- post.setEntity(new FileEntity(initDoc));\r
- byte[] response = client.execute(post, handler);\r
- logger.debug("Response on POST was: " + new String(response,"UTF-8")); \r
- return new ServiceProxyCommandResponse(handler.getStatusCode(),response,handler.getContentType()); \r
- }\r
- \r
- public List<String> getInitDocPaths () {\r
- logger.debug("Get init doc paths ");\r
- logger.debug("length: " + initDocPaths.size());\r
- return initDocPaths;\r
- }\r
- \r
- public ServiceProxyCommandResponse postInitDoc(byte[] initDoc, boolean includeDebug) throws IOException {\r
- HttpPost post = new HttpPost(selectedServiceUrl+"?command=init" + (includeDebug? "&includeDebug=yes" : ""));\r
- post.setEntity(new ByteArrayEntity(initDoc));\r
- byte[] response = client.execute(post, handler);\r
- logger.debug("Response on POST was: " + new String(response,"UTF-8")); \r
- return new ServiceProxyCommandResponse(handler.getStatusCode(),response,handler.getContentType()); \r
- }\r
- \r
- public void setServiceProxyUrl (String url) {\r
- selectedServiceUrl = url;\r
- }\r
- \r
- public String getServiceProxyUrl () {\r
- return selectedServiceUrl;\r
- }\r
- \r
- public List<String> getServiceProxyUrls () {\r
- return serviceUrls;\r
- }\r
- \r
- public Configuration getConfiguration () {\r
- return config;\r
- }\r
- \r
-}\r