Synnefo » Pithos Android Client

package com.rackspace.cloud.files.api.client;

import java.io.File;

import java.io.IOException;

import java.io.StringReader;

import java.net.MalformedURLException;

import java.net.URI;

import java.net.URISyntaxException;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Map.Entry;

import javax.xml.parsers.FactoryConfigurationError;

import javax.xml.parsers.ParserConfigurationException;

import javax.xml.parsers.SAXParser;

import javax.xml.parsers.SAXParserFactory;

import org.apache.http.HttpResponse;

import org.apache.http.client.ClientProtocolException;

import org.apache.http.client.methods.HttpDelete;

import org.apache.http.client.methods.HttpGet;

import org.apache.http.client.methods.HttpPost;

import org.apache.http.client.methods.HttpPut;

import org.apache.http.client.methods.HttpRequestBase;

import org.apache.http.entity.FileEntity;

import org.apache.http.impl.client.BasicResponseHandler;

import org.apache.http.protocol.RequestExpectContinue;

import org.xml.sax.InputSource;

import org.xml.sax.SAXException;

import org.xml.sax.XMLReader;

import android.content.Context;

import android.util.Log;

import com.rackspace.cloud.files.api.client.parsers.ContainerObjectXMLparser;

import com.rackspace.cloud.files.api.client.parsers.ObjectVersionXMLParser;

import com.rackspace.cloud.servers.api.client.Account;

import com.rackspace.cloud.servers.api.client.CloudServersException;

import com.rackspace.cloud.servers.api.client.EntityManager;

import com.rackspace.cloud.servers.api.client.http.HttpBundle;

import com.rackspace.cloud.servers.api.client.parsers.CloudServersFaultXMLParser;

/**

 * 

 * @author Phillip Toohill

 * 

 */

public class ContainerObjectManager extends EntityManager {

        public final static int GET=0;

        public final static int PUT=1;

        public final static int POST=2;

        public final static int DELETE=3;

        public String LOG = "ContainerObjectManager";

        private Context context;

        public static final String storageToken = Account.getAccount()

                        .getStorageToken();

        public ContainerObjectManager(Context context) {

                this.context = context;

        }

        private List<ContainerObjects> executeGet(String containerName, String url) throws CloudServersException{

                CustomHttpClient httpclient = new CustomHttpClient(context);

                HttpGet get = new HttpGet(url);

                ArrayList<ContainerObjects> files = new ArrayList<ContainerObjects>();

                get.addHeader("Content-Type", "application/xml");

                get.addHeader(CustomHttpClient.X_STORAGE_TOKEN, storageToken);

                try {

                        HttpResponse resp = httpclient.execute(get);

                        BasicResponseHandler responseHandler = new BasicResponseHandler();

                        String body = responseHandler.handleResponse(resp);

                        if (resp.getStatusLine().getStatusCode() == 200

                                        || resp.getStatusLine().getStatusCode() == 203) {

                                ContainerObjectXMLparser filesXMLParser = new ContainerObjectXMLparser();

                                SAXParser saxParser = SAXParserFactory.newInstance()

                                                .newSAXParser();

                                XMLReader xmlReader = saxParser.getXMLReader();

                                xmlReader.setContentHandler(filesXMLParser);

                                xmlReader.parse(new InputSource(new StringReader(body)));

                                files = filesXMLParser.getViewFiles();

                                for(ContainerObjects o :files)

                                        o.setContainerName(containerName);

                        } else {

                                CloudServersFaultXMLParser parser = new CloudServersFaultXMLParser();

                                SAXParser saxParser = SAXParserFactory.newInstance()

                                                .newSAXParser();

                                XMLReader xmlReader = saxParser.getXMLReader();

                                xmlReader.setContentHandler(parser);

                                xmlReader.parse(new InputSource(new StringReader(body)));

                                CloudServersException cse = parser.getException();

                                throw cse;

                        }

                } catch (ClientProtocolException e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                } catch (IOException e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                } catch (ParserConfigurationException e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                } catch (SAXException e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                } catch (FactoryConfigurationError e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                }

                return files;

        }

        public List<ContainerObjects> createList(String containerName,

                        boolean shared, String prefix) throws CloudServersException {

                Log.i(LOG,"Create List:"+containerName+ " prefix "+prefix);

                while(prefix.endsWith("/"))

                        prefix = prefix.substring(0,prefix.length()-1);

                String url = getSafeURL(Account.getAccount().getStorageUrl(), containerName)

                                + "?format=xml&shared="+shared+"&prefix="+prefix+"&delimiter=/";

                Log.i(LOG," Getting "+containerName+":"+shared+" "+prefix);

                Log.i(LOG," URL:"+url);

                return executeGet(containerName, url);

        }

        public List<ContainerObjects> createOtherList(String containerName,

                        String user) throws CloudServersException {

                Log.i(LOG,"Create List:"+containerName);                

                String url = getSafeURL(Account.getAccount().getStorageUrl().replaceAll(Account.getAccount().getUsername(), user), containerName)

                                + "?format=xml";

                return executeGet(containerName, url);

        }

        public List<ContainerObjects> createList(String containerName,

                        String prefix) throws CloudServersException {

                while(prefix.endsWith("/"))

                        prefix = prefix.substring(0,prefix.length()-1);

                String url = getSafeURL(Account.getAccount().getStorageUrl(), containerName)

                                + "?format=xml&prefix="+prefix+"&delimiter=/";

                Log.i(LOG,url);

                return executeGet(containerName, url);

        }

        public List<ObjectVersion> getObjectVersions(String Container, String Object)

                        throws CloudServersException {

                CustomHttpClient httpclient = new CustomHttpClient(context);

                String url = Account.getAccount().getStorageUrl()+"/"+ Container

                                + "/" + Object+"?format=xml&version=list";

                Log.i(LOG,url);

                HttpGet get = new HttpGet(url);

                List<ObjectVersion> versions =null;

                get.addHeader("Content-Type", "application/xml");

                get.addHeader(CustomHttpClient.X_STORAGE_TOKEN, storageToken);

                try {

                        HttpResponse resp = httpclient.execute(get);

                        BasicResponseHandler responseHandler = new BasicResponseHandler();

                        String body = responseHandler.handleResponse(resp);

                        if (resp.getStatusLine().getStatusCode() == 200

                                        || resp.getStatusLine().getStatusCode() == 203) {

                                ObjectVersionXMLParser filesXMLParser = new ObjectVersionXMLParser();

                                SAXParser saxParser = SAXParserFactory.newInstance()

                                                .newSAXParser();

                                XMLReader xmlReader = saxParser.getXMLReader();

                                xmlReader.setContentHandler(filesXMLParser);

                                xmlReader.parse(new InputSource(new StringReader(body)));

                                versions = filesXMLParser.getContainers();

                        } else {

                                CloudServersFaultXMLParser parser = new CloudServersFaultXMLParser();

                                SAXParser saxParser = SAXParserFactory.newInstance()

                                                .newSAXParser();

                                XMLReader xmlReader = saxParser.getXMLReader();

                                xmlReader.setContentHandler(parser);

                                xmlReader.parse(new InputSource(new StringReader(body)));

                                CloudServersException cse = parser.getException();

                                throw cse;

                        }

                } catch (ClientProtocolException e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                } catch (IOException e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                } catch (ParserConfigurationException e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                } catch (SAXException e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                } catch (FactoryConfigurationError e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                }

                return versions;

        }

        public List<ContainerObjects> createListMyShared(

                        String passName, List<Container> containers, String prefix) throws CloudServersException {

                ArrayList<ContainerObjects> files = new ArrayList<ContainerObjects>();

                for(Container con :containers ){

                        Log.i(LOG,"GEETING FILES OF COntainer:"+con.getName());

                        if(con.getName().equalsIgnoreCase("trash")||con.getName().equals(Container.MYSHARED)||con.getName().equals(Container.OTHERS)){}

                        else

                        try{

                                List<ContainerObjects> temp = createList(con.getName(), true, prefix);

                                for(ContainerObjects o : temp){

                                        Log.i(LOG,o.getCName()+" "+o.isShared());

                                        if(o.isShared()||o.isSubDir()){

                                                o.setContainerName(Container.MYSHARED);

                                                files.add(o);

                                        }

                                }

                        }

                        catch(CloudServersException e){

                        }

                }

                return files;

        }

        private HttpBundle executeMethod(int methodName ,String url) throws CloudServersException{

                return executeMethod(methodName, url, null, null);

        }

        private HttpBundle executeMethod(int methodName ,String url,String contentType) throws CloudServersException{

                return executeMethod(methodName, url, contentType, null);

        }

        private HttpBundle executeMethod(int methodName ,String url, String contentType, Map<String,String> headers) throws CloudServersException{

                HttpResponse resp = null;

                CustomHttpClient httpclient = new CustomHttpClient(context);

                Log.d(LOG, url);

                HttpRequestBase addObject = null;

                switch (methodName) {

                case GET:

                        addObject = new HttpGet(url);

                        break;

                case PUT:

                        addObject = new HttpPut(url);

                        break;

                case POST:

                        addObject = new HttpPost(url);

                        break;

                case DELETE:

                        addObject = new HttpDelete(url);

                        break;

                default:

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage("Invalid method");

                        throw cse;

                }

                addObject

                                .addHeader("X-Auth-Token", Account.getAccount().getAuthToken());

                if(contentType!=null)

                        addObject.addHeader("Content-Type", contentType);

                //addObject.addHeader("Content-Type",t);

                if(headers!=null)

                        for(Entry<String,String> entry : headers.entrySet()){

                                addObject.addHeader(entry.getKey(),entry.getValue());

                        }

                httpclient.removeRequestInterceptorByClass(RequestExpectContinue.class);

                HttpBundle bundle = new HttpBundle();

                bundle.setCurlRequest(addObject);

                try {

                        resp = httpclient.execute(addObject);

                        bundle.setHttpResponse(resp);

                } catch (ClientProtocolException e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                } catch (IOException e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                } catch (FactoryConfigurationError e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                }

                return bundle;

        }

        public HttpBundle deleteObject(String Container, String Object)

                        throws CloudServersException {

                String url = getSafeURL(Account.getAccount().getStorageUrl(), Container

                                + "/" + Object);

                return executeMethod(DELETE, url);

        }

        public HttpBundle trashObject(String container, String Object)

                        throws CloudServersException {

                String url = getSafeURL(Account.getAccount().getStorageUrl(), Container.TRASH

                                + "/" + Object);

                Map<String,String> headers = new HashMap<String, String>();

                headers.put("X-Move-From","/"+container+"/"+Object);

                return executeMethod(PUT, url,"text/plain; charset=UTF-8",headers);

        }

        public HttpBundle restoreObject(String container, String Object)

                        throws CloudServersException {

                String url = getSafeURL(Account.getAccount().getStorageUrl(), Container.PITHOS

                                + "/" + Object);

                Map<String,String> headers = new HashMap<String, String>();

                headers.put("X-Move-From","/"+container+"/"+Object);

                return executeMethod(PUT, url,"text/plain; charset=UTF-8",headers);

        }

        public HttpBundle getObject(String Container, String Object)

                        throws CloudServersException {

                String url = getSafeURL(Account.getAccount().getStorageUrl(), Container

                                + "/" + Object);

                return executeMethod(GET, url);

        }

        public HttpBundle addObject(String Container, String Path, String Object,

                        String type) throws CloudServersException {

                String url = getSafeURL(Account.getAccount().getStorageUrl(), Container

                                + "/" + Path + Object);

                return executeMethod(DELETE, url,type);

        }

        public HttpBundle updateObject(String Container, String Path, String Object,

                        String type, Map<String,String> headers) throws CloudServersException {

                String url;

                if(Container !=null)

                        url = getSafeURL(Account.getAccount().getStorageUrl(), Container

                                + "/" + Path + Object);

                else{

                        url = getSafeURL(Account.getAccount().getStorageUrl(),"");

                        if(url.endsWith("/"))

                                url = url.substring(0, url.length()-1);

                }

                url =url+"?update=";

                Log.d(LOG, url);

                return executeMethod(DELETE, url,  "        text/plain; charset=UTF-8", headers);

        }

        /*

         * used for adding sdcard files, requires an extra parameter to store the data

         * for the file

         */

        public HttpBundle addFileObject(String Container, String Path, String Object,

                        String type,  String file) throws CloudServersException {

                HttpResponse resp = null;

                CustomHttpClient httpclient = new CustomHttpClient(context);

                String url = getSafeURL(Account.getAccount().getStorageUrl(), Container

                                + "/" + Path + Object);

                HttpPut addObject = new HttpPut(url);

                addObject

                                .addHeader("X-Auth-Token", Account.getAccount().getAuthToken());

                addObject.addHeader("Content-Type", type);

                httpclient.removeRequestInterceptorByClass(RequestExpectContinue.class);

                if (file.startsWith("file://")) {

            file = file.substring(7);

        }

                FileEntity tmp = null;

                        tmp = new FileEntity(new File(file),type);

                addObject.setEntity(tmp);

                HttpBundle bundle = new HttpBundle();

                bundle.setCurlRequest(addObject);

                try {

                        resp = httpclient.execute(addObject);

                        bundle.setHttpResponse(resp);

                } catch (ClientProtocolException e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                } catch (IOException e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                } catch (FactoryConfigurationError e) {

                        CloudServersException cse = new CloudServersException();

                        cse.setMessage(e.getLocalizedMessage());

                        throw cse;

                }

                return bundle;

        }

        private String getSafeURL(String badURL, String name) {

                URI uri = null;

                try {

                        uri = new URI("https", badURL.substring(8), "/" + name + "/", "");

                } catch (URISyntaxException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                }

                String url = null;

                try {

                        url = uri.toURL().toString();

                } catch (MalformedURLException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                }

                return url.substring(0, url.length() - 2);

        }

}