ingo@1: /*
ingo@1:  * Copyright (c) 2010 by Intevation GmbH
ingo@1:  *
ingo@1:  * This program is free software under the LGPL (>=v2.1)
ingo@1:  * Read the file LGPL.txt coming with the software for details
ingo@1:  * or visit http://www.gnu.org/licenses/ if it does not exist.
ingo@1:  */
ingo@1: package de.intevation.artifacts.httpclient.http;
ingo@1: 
ingo@1: import java.io.InputStream;
ingo@1: import java.io.IOException;
ingo@1: import java.io.OutputStream;
ingo@10: import java.util.ArrayList;
ingo@10: import java.util.List;
ingo@1: 
ingo@1: import org.apache.log4j.Logger;
ingo@1: 
ingo@1: import org.restlet.Client;
ingo@1: import org.restlet.Request;
ingo@1: import org.restlet.Response;
ingo@10: import org.restlet.data.ClientInfo;
ingo@10: import org.restlet.data.Language;
ingo@1: import org.restlet.data.MediaType;
ingo@1: import org.restlet.data.Method;
ingo@10: import org.restlet.data.Preference;
ingo@1: import org.restlet.data.Protocol;
ingo@1: import org.restlet.data.Status;
ingo@1: import org.restlet.ext.xml.DomRepresentation;
ingo@1: import org.restlet.representation.Representation;
ingo@1: 
ingo@1: import org.w3c.dom.Document;
ingo@1: 
ingo@1: import de.intevation.artifacts.httpclient.exceptions.ConnectionException;
ingo@1: import de.intevation.artifacts.httpclient.http.response.DocumentResponseHandler;
ingo@1: import de.intevation.artifacts.httpclient.http.response.ResponseHandler;
ingo@1: import de.intevation.artifacts.httpclient.http.response.StreamResponseHandler;
ingo@1: import de.intevation.artifacts.httpclient.objects.Artifact;
ingo@1: import de.intevation.artifacts.httpclient.objects.ArtifactFactory;
ingo@1: import de.intevation.artifacts.httpclient.utils.ArtifactProtocolUtils;
ingo@2: import de.intevation.artifacts.httpclient.utils.ArtifactCreator;
ingo@2: 
ingo@1: 
ingo@1: /**
ingo@1:  * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a>
ingo@1:  */
ingo@1: public class HttpClientImpl implements HttpClient {
ingo@1: 
ingo@1:     private static final Logger logger = Logger.getLogger(HttpClient.class);
ingo@1: 
ingo@4:     /** The URL part of the resource to list the existing users of the server.*/
ingo@4:     public static final String PATH_LIST_USERS = "/list-users";
ingo@4: 
ingo@7:     /** The URL part of the resource to list the Collections owned by a specific
ingo@7:      * user.*/
ingo@7:     public static final String PATH_USER_COLLECTIONS = "/list-collections";
ingo@7: 
ingo@5:     /** The URL part of the resource to call a specific service.*/
ingo@5:     public static final String PATH_SERVICE = "/service";
ingo@5: 
ingo@3:     /** The URL path of the resource to create new artifact collections.*/
ingo@3:     public static final String PATH_CREATE_COLLECTION = "/create-collection";
ingo@3: 
ingo@6:     /** The URL path of the resource to work with an artifact collections.*/
ingo@6:     public static final String PATH_ACTION_COLLECTION = "/collection";
ingo@6: 
ingo@9:     /** The URL path of the resource to work with an artifact collections.*/
ingo@9:     public static final String PATH_OUT_COLLECTION = "/collection";
ingo@9: 
ingo@1:     private String serverUrl;
ingo@1: 
ingo@10:     private String localeString;
ingo@10: 
ingo@1: 
ingo@1:     public HttpClientImpl(String serverUrl) {
ingo@1:         this.serverUrl = serverUrl;
ingo@1:     }
ingo@1: 
ingo@1: 
ingo@10:     /**
ingo@10:      * This constructor might be used to modify the request's locale manually.
ingo@10:      * E.g. the localization should not be based on the configured browser
ingo@10:      * locale, but site specific configuration - than you are able to set the
ingo@10:      * locale in this constructor.
ingo@10:      *
ingo@10:      * @param serverUrl The url that is used for the request.
ingo@10:      * @param localeString The string representation of the desired locale.
ingo@10:      */
ingo@10:     public HttpClientImpl(String serverUrl, String localeString) {
ingo@10:         this(serverUrl);
ingo@10: 
ingo@10:         this.localeString = localeString;
ingo@10:     }
ingo@10: 
ingo@10: 
ingo@1:     @Override
ingo@1:     public ArtifactFactory[] getArtifactFactories()
ingo@1:     throws ConnectionException
ingo@1:     {
ingo@1:         ResponseHandler handler = new DocumentResponseHandler();
ingo@1: 
ingo@1:         try {
ingo@1:             String    url   = serverUrl + "/factories";
ingo@1:             Document result = (Document) handler.handle(doGet(url));
ingo@1: 
ingo@1:             return ArtifactProtocolUtils.extractArtifactFactories(result);
ingo@1:         }
ingo@1:         catch (IOException ioe) {
ingo@1:             throw new ConnectionException(
ingo@1:                 "Connection to server failed. No Factories recieved.");
ingo@1:         }
ingo@1:     }
ingo@1: 
ingo@1: 
ingo@2:     /**
ingo@2:      * This method creates a new artifact in the artifact server and returns
ingo@2:      * this artifact. The new artifact is created using <i>creator</i>. If no
ingo@2:      * {@link ArtifactCreator} is given (null), an {@link Artifact} is returned.
ingo@2:      *
ingo@2:      * @param doc The CREATE document.
ingo@2:      * @param creator The {@link ArtifactCreator} that is used to extract the
ingo@2:      * new artifact from response document of the server.
ingo@2:      *
ingo@2:      * @return the new artifact.
ingo@2:      */
ingo@1:     @Override
ingo@2:     public Object create(Document doc, ArtifactCreator creator)
ingo@2:     throws ConnectionException
ingo@2:     {
ingo@1:         ResponseHandler handler = new DocumentResponseHandler();
ingo@1: 
ingo@1:         try {
ingo@1:             String   url    = serverUrl + "/create";
ingo@1:             Document result = (Document) handler.handle(doPost(url, doc));
ingo@1: 
ingo@2:             return creator == null
ingo@2:                 ? ArtifactProtocolUtils.extractArtifact(result)
ingo@2:                 : creator.create(result);
ingo@1:         }
ingo@1:         catch (IOException ioe) {
ingo@1:             throw new ConnectionException(
ingo@1:                 "Connection to server failed. No Artifact created.");
ingo@1:         }
ingo@1:     }
ingo@1: 
ingo@1: 
ingo@1:     @Override
ingo@1:     public Object describe(
ingo@1:         Artifact        artifact,
ingo@1:         Document        doc,
ingo@1:         ResponseHandler handler)
ingo@1:     throws ConnectionException
ingo@1:     {
ingo@1:         try {
ingo@1:             String   url    = serverUrl + "/artifact/" + artifact.getUuid();
ingo@1:             return handler.handle(doPost(url, doc));
ingo@1:         }
ingo@1:         catch (IOException ioe) {
ingo@1:             throw new ConnectionException(
ingo@1:                 "Connection to server failed: " + ioe.getMessage());
ingo@1:         }
ingo@1:     }
ingo@1: 
ingo@1: 
ingo@1:     @Override
ingo@1:     public Object feed(Artifact artifact, Document doc, ResponseHandler handler)
ingo@1:     throws ConnectionException
ingo@1:     {
ingo@1:         try {
ingo@1:             String   url    = serverUrl + "/artifact/" + artifact.getUuid();
ingo@1:             Document result = (Document) handler.handle(doPost(url, doc));
ingo@1: 
ingo@1:             return result;
ingo@1:         }
ingo@1:         catch (IOException ioe) {
ingo@1:             throw new ConnectionException(
ingo@1:                 "Connection to server failed: " + ioe.getMessage());
ingo@1:         }
ingo@1:     }
ingo@1: 
ingo@1: 
ingo@1:     @Override
ingo@1:     public Object advance(Artifact artifact, Document doc, ResponseHandler handler)
ingo@1:     throws ConnectionException
ingo@1:     {
ingo@1:         try {
ingo@1:             String   url    = serverUrl + "/artifact/" + artifact.getUuid();
ingo@1:             Document result = (Document) handler.handle(doPost(url, doc));
ingo@1: 
ingo@1:             return result;
ingo@1:         }
ingo@1:         catch (IOException ioe) {
ingo@1:             throw new ConnectionException(
ingo@1:                 "Connection to server failed: " + ioe.getMessage());
ingo@1:         }
ingo@1:     }
ingo@1: 
ingo@1: 
ingo@1:     @Override
ingo@1:     public void out(
ingo@1:         Artifact     artifact,
ingo@1:         Document     doc,
ingo@1:         String       target,
ingo@1:         OutputStream out)
ingo@1:     throws ConnectionException
ingo@1:     {
ingo@1:         try {
ingo@1:             String url =
ingo@1:                 serverUrl
ingo@1:                 + "/artifact/"
ingo@1:                 + artifact.getUuid()
ingo@1:                 + "/" + target;
ingo@1: 
ingo@1:             ResponseHandler handler = new StreamResponseHandler();
ingo@1: 
ingo@1:             InputStream stream = (InputStream) handler.handle(doPost(url, doc));
sascha@17:             try {
sascha@17:                 byte[] b = new byte[4096];
sascha@17:                 int i;
sascha@17:                 while ((i = stream.read(b)) >= 0) {
sascha@17:                     out.write(b, 0, i);
sascha@17:                 }
sascha@17:             }
sascha@17:             finally {
sascha@17:                 stream.close();
ingo@1:             }
ingo@1:         }
ingo@1:         catch (IOException ioe) {
ingo@1:             throw new ConnectionException(
ingo@1:                 "Connection to server failed: " + ioe.getMessage());
ingo@1:         }
ingo@1:     }
ingo@1: 
ingo@1: 
ingo@10:     //==============================
ingo@10:     // HTTP specific methods
ingo@10:     //==============================
ingo@10: 
ingo@1:     private Response doPost(String url, Document body) throws IOException {
ingo@1:         logger.info("Start HTTP-POST request to: "+ url);
ingo@1: 
ingo@1:         Client client   = new Client(Protocol.HTTP);
ingo@10:         Request request = prepareRequest(Method.POST, url);
ingo@1: 
ingo@1:         Representation representation = new DomRepresentation(
ingo@1:             MediaType.APPLICATION_XML,
ingo@1:             body);
ingo@1: 
ingo@1:         request.setEntity(representation);
ingo@1:         Response response = client.handle(request);
ingo@1: 
ingo@1:         Status status = response.getStatus();
ingo@1:         if (status.getCode() != 200) {
ingo@1:             logger.error("Response status: " + status.getCode());
ingo@1:             throw new IOException(status.getDescription());
ingo@1:         }
ingo@1: 
ingo@1:         return response;
ingo@1:     }
ingo@1: 
ingo@1: 
ingo@1:     private Response doGet(String url) throws IOException {
ingo@1:         logger.info("Start HTTP-POST request to: "+ url);
ingo@1: 
ingo@1:         Client client   = new Client(Protocol.HTTP);
ingo@10:         Request request = prepareRequest(Method.GET, url);
ingo@1: 
ingo@1:         Response response = client.handle(request);
ingo@1: 
ingo@1:         Status status = response.getStatus();
ingo@1:         if (status.getCode() != 200) {
ingo@1:             logger.error("Response status: " + status.getCode());
ingo@1:             throw new IOException(status.getDescription());
ingo@1:         }
ingo@1: 
ingo@1:         return response;
ingo@1:     }
ingo@4: 
ingo@4: 
ingo@10:     /**
ingo@10:      * This method prepares the request object.
ingo@10:      *
ingo@10:      * @param method The HTTP method (GET,POST).
ingo@10:      * @param url The URL used for the request.
ingo@10:      *
ingo@10:      * @return the request object.
ingo@10:      */
ingo@10:     private Request prepareRequest(Method method, String url) {
ingo@10:         Request request = new Request(method, url);
ingo@10: 
ingo@10:         ClientInfo info = request.getClientInfo();
ingo@10: 
ingo@10:         setLocale(info);
ingo@10: 
ingo@10:         request.setClientInfo(info);
ingo@10: 
ingo@10:         return request;
ingo@10:     }
ingo@10: 
ingo@10: 
ingo@10:     /**
ingo@10:      * This method is called to set the request's locale.
ingo@10:      *
ingo@10:      * @param info The ClientInfo that is used to provide request information.
ingo@10:      */
ingo@10:     private void setLocale(ClientInfo info) {
ingo@11:         if (localeString == null) {
ingo@10:             return;
ingo@10:         }
ingo@10: 
ingo@10:         List<Preference<Language>> accepted =
ingo@10:             new ArrayList<Preference<Language>>();
ingo@10: 
ingo@10:         Language lang = Language.valueOf(localeString);
ingo@10: 
ingo@10:         if (lang != null) {
ingo@11:             logger.info("Set locale of the request object: " + lang.toString());
ingo@11: 
ingo@10:             Preference<Language> pref = new Preference<Language>();
ingo@10:             pref.setMetadata(lang);
ingo@10:             accepted.add(pref);
ingo@10: 
ingo@10:             info.setAcceptedLanguages(accepted);
ingo@10:         }
ingo@10:     }
ingo@10: 
ingo@10: 
ingo@6:     //==============================
ingo@6:     // Collection API
ingo@6:     //==============================
ingo@6: 
ingo@6:     /**
ingo@6:      * This method triggers the artifact servers resource to create a new
ingo@6:      * artifact collection.
ingo@6:      *
ingo@6:      * @param create The CREATE document for the collection.
ingo@6:      * @param ownerId The uuid of the creator.
ingo@6:      * @param handler The handler that is used to create the result object.
ingo@6:      *
ingo@6:      * @return a result object created by <i>handler</i>.
ingo@6:      */
ingo@6:     public Object createCollection(
ingo@6:         Document        create,
ingo@6:         String          ownerId,
ingo@6:         ResponseHandler handler)
ingo@6:     throws ConnectionException
ingo@6:     {
ingo@6:         String url = serverUrl + PATH_CREATE_COLLECTION + "/" + ownerId;
ingo@6: 
ingo@6:         try {
ingo@6:             return handler.handle(doPost(url, create));
ingo@6:         }
ingo@6:         catch (IOException ioe) {
ingo@6:             throw new ConnectionException(ioe.getMessage(), ioe);
ingo@6:         }
ingo@6:     }
ingo@6: 
ingo@6: 
ingo@6:     /**
ingo@6:      * This method might be used to trigger a collection specific action. The
ingo@6:      * action that is executed depends on the document <i>actionDoc</i>.
ingo@6:      *
ingo@6:      * @param actionDoc The document that describes the action to be executed.
ingo@6:      * @param uuid      The uuid of the collection.
ingo@6:      * @param handler   The handler that is used to create the result object.
ingo@6:      *
ingo@6:      * @return a result object created by <i>handler</i>.
ingo@6:      */
ingo@6:     public Object doCollectionAction(
ingo@6:         Document        actionDoc,
ingo@6:         String          uuid,
ingo@6:         ResponseHandler handler)
ingo@6:     throws ConnectionException
ingo@6:     {
ingo@6:         String url = serverUrl + PATH_ACTION_COLLECTION + "/" + uuid;
ingo@6: 
ingo@6:         try {
ingo@6:             return handler.handle(doPost(url, actionDoc));
ingo@6:         }
ingo@6:         catch (IOException ioe) {
ingo@6:             throw new ConnectionException(ioe.getMessage(), ioe);
ingo@6:         }
ingo@6:     }
ingo@6: 
ingo@6: 
ingo@9:     /**
ingo@9:      * This method triggers the out() operation of a Collection. The result of
ingo@9:      * this operation is written to <i>out</i> directly - there is no return
ingo@9:      * value.
ingo@9:      *
ingo@9:      * @param doc The request document for the out() operation.
ingo@9:      * @param uuid The identifier of the Collection.
ingo@9:      * @param type The name of the output type.
ingo@9:      * @param out The OutputStream.
ingo@9:      */
ingo@9:     public void collectionOut(
ingo@9:         Document     doc,
ingo@9:         String       uuid,
ingo@9:         String       type,
ingo@9:         OutputStream out)
ingo@9:     throws ConnectionException
ingo@9:     {
ingo@9:         try {
ingo@12:             InputStream stream = collectionOut(doc, uuid, type);
ingo@9: 
ingo@9:             byte[] b = new byte[4096];
sascha@17:             try {
sascha@17:                 int i;
sascha@17:                 while ((i = stream.read(b)) >= 0) {
sascha@17:                     out.write(b, 0, i);
sascha@17:                 }
sascha@17:             }
sascha@17:             finally {
sascha@17:                 stream.close();
ingo@9:             }
ingo@9:         }
ingo@9:         catch (IOException ioe) {
ingo@9:             throw new ConnectionException(ioe.getMessage(), ioe);
ingo@9:         }
ingo@9:     }
ingo@9: 
ingo@9: 
ingo@12:     /**
ingo@12:      * This method triggers the out() operation of a Collection. The result of
ingo@12:      * this operation is returned as an InputStream.
ingo@12:      *
ingo@12:      * @param doc The request document for the out() operation.
ingo@12:      * @param uuid The identifier of the Collection.
ingo@12:      * @param type The name of the output type.
ingo@12:      *
ingo@12:      * @return an InputStream.
ingo@12:      */
ingo@12:     public InputStream collectionOut(
ingo@12:         Document    doc,
ingo@12:         String      uuid,
ingo@12:         String      type)
ingo@12:     throws ConnectionException
ingo@12:     {
ingo@12:         String url = serverUrl + PATH_OUT_COLLECTION + "/" + uuid + "/" + type;
ingo@12: 
ingo@12:         ResponseHandler handler = new StreamResponseHandler();
ingo@12: 
ingo@12:         try {
ingo@12:             return (InputStream) handler.handle(doPost(url, doc));
ingo@12:         }
ingo@12:         catch (IOException ioe) {
ingo@12:             throw new ConnectionException(ioe.getMessage(), ioe);
ingo@12:         }
ingo@12:     }
ingo@12: 
ingo@12: 
ingo@4:     /*******************************
ingo@5:      * Service API
ingo@5:      *******************************/
ingo@5: 
ingo@5:      public Document callService(String url, String service, Document input)
ingo@5:      throws ConnectionException
ingo@5:      {
ingo@5:         logger.info("Start service call to '" + service + "'");
ingo@5: 
ingo@5:         DocumentResponseHandler handler = new DocumentResponseHandler();
ingo@5: 
ingo@5:         try {
ingo@5:             String serverUrl = url + PATH_SERVICE + "/" + service;
ingo@5:             return (Document) handler.handle(doPost(serverUrl, input));
ingo@5:         }
ingo@5:         catch (IOException ioe) {
ingo@5:             throw new ConnectionException(
ingo@5:                 "Connection to server failed: " + ioe.getMessage());
ingo@5:         }
ingo@5:      }
ingo@5: 
ingo@5: 
ingo@5:     /*******************************
ingo@4:      * Users API
ingo@4:      *******************************/
ingo@5: 
ingo@4:     public Document listUsers()
ingo@4:     throws ConnectionException
ingo@4:     {
ingo@4:         ResponseHandler handler = new DocumentResponseHandler();
ingo@4:         String    url           = serverUrl + PATH_LIST_USERS;
ingo@4: 
ingo@4:         try {
ingo@4:             return (Document) handler.handle(doGet(url));
ingo@4:         }
ingo@4:         catch (IOException ioe) {
ingo@4:             throw new ConnectionException(ioe.getMessage(), ioe);
ingo@4:         }
ingo@4:     }
ingo@7: 
ingo@7: 
ingo@7:     public Document listUserCollections(String userid)
ingo@7:     throws ConnectionException
ingo@7:     {
ingo@7:         ResponseHandler handler = new DocumentResponseHandler();
ingo@7: 
ingo@7:         String url = serverUrl + PATH_USER_COLLECTIONS + "/" + userid;
ingo@7: 
ingo@7:         try {
ingo@7:             return (Document) handler.handle(doGet(url));
ingo@7:         }
ingo@7:         catch (IOException ioe) {
ingo@7:             throw new ConnectionException(ioe.getMessage(), ioe);
ingo@7:         }
ingo@7:     }
ingo@1: }
ingo@1: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8: